added geometry option

brosaplanella · May 26, 2020 · fda5342 · fda5342
1 parent 1007c87
commit fda5342
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/pybamm/models/full_battery_models/base_battery_model.py b/pybamm/models/full_battery_models/base_battery_model.py
@@ -181,6 +181,7 @@ def options(self, extra_options):
             "current collector": "uniform",
             "particle": "Fickian diffusion",
             "thermal": "isothermal",
+            "geometry": "arbitrary",
             "external submodels": [],
             "sei": None,
         }
@@ -273,6 +274,10 @@ def options(self, extra_options):
             raise pybamm.OptionError(
                 "Unknown thermal model '{}'".format(options["thermal"])
             )
+        if options["geometry"] not in ["arbitrary", "pouch"]:
+            raise pybamm.OptionError(
+                "Unknown geometry '{}'".format(options["geometry"])
+            )            
         if options["sei"] not in [
             None,
             "constant",
@@ -548,7 +553,7 @@ def set_thermal_submodel(self):
 
         elif self.options["thermal"] == "lumped":
             thermal_submodel = pybamm.thermal.Lumped(
-                self.param, self.options["dimensionality"]
+                self.param, self.options["dimensionality"], self.options["geometry"]
             )
 
         elif self.options["thermal"] == "x-lumped":