Latest stuff I found from Harishharish

1 files changed, 32 insertions, 0 deletions

diff --git a/src/processors/FBP.py b/src/processors/FBP.py
new file mode 100644
index 0000000..c48c590
--- /dev/null
+++ b/src/processors/FBP.py
@@ -0,0 +1,32 @@
+from cil.framework import DataProcessor
+from cil.framework import DataOrder
+from src.processors.FBP_Standard import FBP_Standard
+from src.processors.FBP_Stacked import FBP_Stacked
+
+class FBP(DataProcessor):
+    def __init__(self, volume_geometry, sinogram_geometry, stacked=False, precision_mode='single', stack_num=2):
+        if stacked==True:
+            processor = FBP_Stacked(volume_geometry, sinogram_geometry, precision_mode, stack_num)
+        else:
+            processor = FBP_Standard(volume_geometry, sinogram_geometry)
+
+        super(FBP, self).__init__(volume_geometry=volume_geometry, sinogram_geometry=sinogram_geometry,
+                                  stacked=stacked,
+                                  precision_mode=precision_mode, stack_num=stack_num, processor=processor)
+
+        self.processor = processor
+
+    def set_input(self, dataset):
+        return self.processor.set_input(dataset)
+
+    def get_input(self):
+        return self.processor.get_input()
+
+    def get_output(self, out=None):
+        return self.processor.get_output(out=None)
+
+    def check_input(self, dataset):
+        return self.processor.check_input(dataset)
+
+    def process(self, out=None):
+        return self.processor.process(out=None)