cluster of arrays

Author: burrbull

CHANGELOG-rust.md

@@ -6,6 +6,7 @@ This changelog tracks the Rust `svdtools` project. See
 ## [Unreleased]
 
 * Add `_expand_array`
+* Allow register arrays in `collect_in_cluster`
 
 ## [v0.3.10] 2024-02-26
 

src/patch/mod.rs

@@ -693,7 +693,8 @@ fn spec_ind(spec: &str) -> Option<(usize, usize)> {
     use regex::Regex;
     let spec = spec.split(',').next().unwrap_or(spec);
     static RE: Lazy<Regex> = Lazy::new(|| {
-        Regex::new(r"^\w*((?:[\?*]|\[\d+(?:-\d+)?\]|\[[a-zA-Z]+(?:-[a-zA-Z]+)?\])+)\w*$").unwrap()
+        Regex::new(r"^[\w%]*((?:[\?*]|\[\d+(?:-\d+)?\]|\[[a-zA-Z]+(?:-[a-zA-Z]+)?\])+)[\w%]*$")
+            .unwrap()
     });
     let Some(caps) = RE.captures(spec) else {
         return None;

src/patch/peripheral.rs

@@ -1375,8 +1375,8 @@ fn collect_in_cluster(
         let (rspec, ignore) = rspec.spec();
         while i < regs.len() {
             match &regs[i] {
-                RegisterCluster::Register(Register::Single(r)) if matchname(&r.name, rspec) => {
-                    if let RegisterCluster::Register(Register::Single(r)) = regs.remove(i) {
+                RegisterCluster::Register(r) if matchname(&r.name, rspec) => {
+                    if let RegisterCluster::Register(r) = regs.remove(i) {
                         registers.push(r);
                         place = place.min(i);
                     }
@@ -1399,15 +1399,29 @@ fn collect_in_cluster(
             ));
         }
         rspecs.push(rspec.to_string());
+
         if single {
             if registers.len() > 1 {
                 return Err(anyhow!("{path}: more than one registers {rspec} found"));
             }
         } else {
             registers.sort_by_key(|r| r.address_offset);
+            if let Register::Array(_, rdim) = &registers[0] {
+                if !registers
+                    .iter()
+                    .skip(1)
+                    .all(|r| matches!(r, Register::Array(_, d) if d == rdim))
+                {
+                    return Err(anyhow!("`{rspec}` have different dim blocks"));
+                }
+            } else if !registers.iter().skip(1).all(|r| r.is_single()) {
+                return Err(anyhow!(
+                    "Some of `{rspec}` registers are arrays and some are not"
+                ));
+            }
             let bitmasks = registers
                 .iter()
-                .map(RegisterInfo::get_bitmask)
+                .map(|r| RegisterInfo::get_bitmask(r))
                 .collect::<Vec<_>>();
             let new_dim_index = registers
                 .iter()
@@ -1483,7 +1497,7 @@ fn collect_in_cluster(
     config.update_fields = true;
     let cluster = if single {
         for (_, (rmod, mut registers)) in rdict.into_iter() {
-            let mut reg = registers.swap_remove(0).single();
+            let mut reg = registers.swap_remove(0);
             let rmod = rmod.hash()?;
             reg.process(rmod, path, &config)
                 .with_context(|| format!("Processing register `{}`", reg.name))?;
@@ -1497,7 +1511,7 @@ fn collect_in_cluster(
         cinfo.children(children).build(VAL_LVL)?.single()
     } else {
         for (rspec, (rmod, mut registers)) in rdict.into_iter() {
-            let mut reg = registers.swap_remove(0).single();
+            let mut reg = registers.swap_remove(0);
             let rmod = rmod.hash()?;
             reg.process(rmod, path, &config)
                 .with_context(|| format!("Processing register `{}`", reg.name))?;