summaryrefslogtreecommitdiffstats
path: root/src/processors/FBP.py
diff options
context:
space:
mode:
Diffstat (limited to 'src/processors/FBP.py')
-rw-r--r--src/processors/FBP.py32
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)