process in collect_in_cluster

Author: burrbull

CHANGELOG-rust.md

@@ -5,6 +5,8 @@ This changelog tracks the Rust `svdtools` project. See
 
 ## [Unreleased]
 
+* Allow to modify registers in `_cluster:`
+
 ## [v0.3.11] 2024-03-06
 
 * Add `_expand_array`

src/patch/peripheral.rs

@@ -29,6 +29,12 @@ pub trait PeripheralExt: InterruptExt + RegisterBlockExt {
 pub trait ClusterExt: RegisterBlockExt {
     /// Work through a cluster, handling all registers
     fn process(&mut self, peripheral: &Hash, pname: &str, config: &Config) -> PatchResult;
+
+    /// Work through a cluster, handling all registers
+    fn pre_process(&mut self, peripheral: &Hash, pname: &str, config: &Config) -> PatchResult;
+
+    /// Work through a cluster, handling all registers
+    fn post_process(&mut self, peripheral: &Hash, pname: &str, config: &Config) -> PatchResult;
 }
 
 /// Collecting methods for processing peripheral interrupt contents
@@ -710,7 +716,7 @@ impl RegisterBlockExt for Peripheral {
 }
 
 impl ClusterExt for Cluster {
-    fn process(&mut self, pmod: &Hash, _pname: &str, config: &Config) -> PatchResult {
+    fn pre_process(&mut self, pmod: &Hash, _pname: &str, config: &Config) -> PatchResult {
         // Handle deletions
         if let Some(deletions) = pmod.get(&"_delete".to_yaml()) {
             match deletions {
@@ -881,6 +887,12 @@ impl ClusterExt for Cluster {
             }
         }
 
+        Ok(())
+    }
+
+    fn process(&mut self, pmod: &Hash, pname: &str, config: &Config) -> PatchResult {
+        self.pre_process(pmod, pname, config)?;
+
         // Handle registers
         for (rspec, register) in pmod {
             let rspec = rspec.str()?;
@@ -890,6 +902,10 @@ impl ClusterExt for Cluster {
             }
         }
 
+        self.post_process(pmod, pname, config)
+    }
+
+    fn post_process(&mut self, pmod: &Hash, _pname: &str, config: &Config) -> PatchResult {
         // Expand register arrays
         for (rspec, rmod) in pmod.hash_iter("_expand_array") {
             let rspec = rspec.str()?;
@@ -1365,7 +1381,7 @@ fn collect_in_cluster(
 
     for (rspec, rmod) in cmod {
         let rspec = rspec.str()?;
-        if rspec == "description" || rspec == "dimIncrement" {
+        if rspec == "description" || rspec == "dimIncrement" || rspec.starts_with('_') {
             continue;
         }
         let mut registers = Vec::new();
@@ -1493,7 +1509,7 @@ fn collect_in_cluster(
         .address_offset(address_offset);
     let mut config = config.clone();
     config.update_fields = true;
-    let cluster = if single {
+    let mut cluster = if single {
         for (_, (rmod, mut registers)) in rdict.into_iter() {
             let mut reg = registers.swap_remove(0);
             let rmod = rmod.hash()?;
@@ -1538,6 +1554,8 @@ fn collect_in_cluster(
                 .build(VAL_LVL)?,
         )
     };
+    cluster.pre_process(cmod, cname, &config)?;
+    cluster.post_process(cmod, cname, &config)?;
     regs.insert(place, RegisterCluster::Cluster(cluster));
     Ok(())
 }