1 files changed, 34 insertions, 0 deletions
diff --git a/Wrappers/Python/ccpi/framework/BlockGeometry.py b/Wrappers/Python/ccpi/framework/BlockGeometry.py
new file mode 100755
index 0000000..87dfe92
--- /dev/null
+++ b/Wrappers/Python/ccpi/framework/BlockGeometry.py
@@ -0,0 +1,34 @@
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import print_function
+from __future__ import unicode_literals
+
+import numpy
+from numbers import Number
+import functools
+#from ccpi.framework import AcquisitionData, ImageData
+#from ccpi.optimisation.operators import Operator, LinearOperator
+ 
+class BlockGeometry(object):
+    '''Class to hold Geometry as column vector'''
+    #__array_priority__ = 1
+    def __init__(self, *args, **kwargs):
+        ''''''
+        self.geometries = args
+        self.index = 0
+        #shape = kwargs.get('shape', None)
+        #if shape is None:
+        #   shape = (len(args),1)
+        shape = (len(args),1)
+        self.shape = shape
+        #print (self.shape)
+        n_elements = functools.reduce(lambda x,y: x*y, shape, 1)
+        if len(args) != n_elements:
+            raise ValueError(
+                    'Dimension and size do not match: expected {} got {}'
+                    .format(n_elements, len(args)))
+
+    def allocate(self):
+        containers = [geom.allocate() for geom in self.geometries]
+        BlockDataContainer(*containers)
+