Add geometry argument

1 files changed, 17 insertions, 0 deletions

diff --git a/boltzgen/geometry.py b/boltzgen/geometry.py
index 4b61982..193cbe6 100644
--- a/boltzgen/geometry.py
+++ b/boltzgen/geometry.py
@@ -1,3 +1,4 @@
+import re
 from utility.ndindex import ndindex
 
 class Geometry:
@@ -7,6 +8,22 @@ class Geometry:
         self.size_z = size_z
         self.volume = size_x * size_y * size_z
 
+    @classmethod
+    def parse(self, s):
+        args = re.search(r'([0-9]+):([0-9]+):?([0-9]+)?', s)
+        if args.group(3) == None:
+            return self(int(args.group(1)), int(args.group(2)))
+        else:
+            return self(int(args.group(1)), int(args.group(2)), int(args.group(3)))
+
+    def dimension(self):
+        if self.size_x > 1 and self.size_y > 1 and self.size_z == 1:
+            return 2
+        elif self.size_x > 1 and self.size_y > 1 and self.size_z > 1:
+            return 3
+        else:
+            raise Exception('Geometry malformed')
+
     def inner_cells(self):
         for idx in numpy.ndindex(self.inner_size()):
             yield tuple(map(lambda i: i + 1, idx))