added first implementation

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)

+