update: add make slab if not

Exabyte-io · Nov 25, 2024 · bfff098 · bfff098
1 parent c2021eb
commit bfff098
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/py/mat3ra/made/tools/build/slab/__init__.py b/src/py/mat3ra/made/tools/build/slab/__init__.py
@@ -14,3 +14,11 @@ def create_slab(configuration: SlabConfiguration, termination: Optional[Terminat
     builder = SlabBuilder()
     termination = termination or builder.get_terminations(configuration)[0]
     return builder.get_material(configuration, selector_parameters=SlabSelectorParameters(termination=termination))
+
+
+def create_slab_if_not(material: Material, default_slab_configuration: SlabConfiguration):
+    slab = material
+    if not slab.metadata or slab.metadata["build"]["configuration"]["type"] != SlabConfiguration.__name__:
+        print("The material is not a slab. Creating a new slab...")
+        slab = create_slab(default_slab_configuration)
+    return slab