store slices

CHANGELOG.md

@@ -2,6 +2,8 @@
 
 ## Features
 
+-   Store variable slices in model for inspection ([#925](https://github.com/pybamm-team/PyBaMM/pull/925))
+-   Made t_plus (optionally) a function of electrolyte concentration, and added (1 + dlnf/dlnc) to models ([#921](https://github.com/pybamm-team/PyBaMM/pull/921))
 -   Added `DummySolver` for empty models ([#915](https://github.com/pybamm-team/PyBaMM/pull/915))
 -   Added functionality to broadcast to edges ([#891](https://github.com/pybamm-team/PyBaMM/pull/891))
 -   Reformatted and cleaned up `QuickPlot` ([#886](https://github.com/pybamm-team/PyBaMM/pull/886))

pybamm/discretisations/discretisation.py

@@ -137,6 +137,8 @@ def process_model(self, model, inplace=True, check_model=True):
         # Set the y split for variables
         pybamm.logger.info("Set variable slices for {}".format(model.name))
         self.set_variable_slices(variables)
+        # Keep a record of y_slices in the model
+        model.y_slices = self.y_slices_explicit
 
         # now add extrapolated external variables to the boundary conditions
         # if required by the spatial method
@@ -214,6 +216,7 @@ def set_variable_slices(self, variables):
         """
         # Set up y_slices
         y_slices = defaultdict(list)
+        y_slices_explicit = defaultdict(list)
         start = 0
         end = 0
         # Iterate through unpacked variables, adding appropriate slices to y_slices
@@ -233,14 +236,18 @@ def set_variable_slices(self, variables):
                             submesh = domain_mesh[i]
                             end += submesh.npts_for_broadcast_to_nodes
                         y_slices[child.id].append(slice(start, end))
+                        y_slices_explicit[child].append(slice(start, end))
                         start = end
             else:
                 end += self._get_variable_size(variable)
                 y_slices[variable.id].append(slice(start, end))
+                y_slices_explicit[variable].append(slice(start, end))
                 start = end
 
         # Convert y_slices back to normal dictionary
         self.y_slices = dict(y_slices)
+        # Also keep a record of what the y_slices are, to be stored in the model
+        self.y_slices_explicit = dict(y_slices_explicit)
 
         # reset discretised_symbols
         self._discretised_symbols = {}

tests/unit/test_discretisations/test_discretisation.py

@@ -132,6 +132,8 @@ def test_adding_1D_external_variable(self):
 
         self.assertEqual(disc.y_slices[a.id][0], slice(0, 10, None))
 
+        self.assertEqual(model.y_slices[a][0], slice(0, 10, None))
+
         b_test = np.ones((10, 1))
         np.testing.assert_array_equal(
             model.variables["b"].evaluate(inputs={"b": b_test}), b_test