Update also other metadata in Grid::set_convolution

NNPDF · Aug 27, 2024 · 741dcb9 · 741dcb9
1 parent 091b000
commit 741dcb9
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/pineappl/src/grid.rs b/pineappl/src/grid.rs
@@ -856,6 +856,19 @@ impl Grid {
 
         self.set_key_value(&format!("convolution_type_{}", index + 1), &type_);
         self.set_key_value(&format!("convolution_particle_{}", index + 1), &particle);
+
+        // update the remaining metadata
+        for (index, convolution) in self.convolutions().into_iter().enumerate() {
+            if self
+                .key_values()
+                // UNWRAP: we set some key-values before so there must be a storage
+                .unwrap_or_else(|| unreachable!())
+                .get(&format!("initial_state_{}", index + 1))
+                .is_some()
+            {
+                self.set_convolution(index, convolution);
+            }
+        }
     }
 
     fn increase_shape(&mut self, new_dim: &(usize, usize, usize)) {