From 5e72985e2983e9be8117696fe0ec02388e7153f1 Mon Sep 17 00:00:00 2001 From: "Suren A. Chilingaryan" Date: Wed, 27 Jul 2022 00:28:05 +0200 Subject: Latest stuff I found from Harish --- src/processors/FBP.py | 32 ++++++++++++++++++++++++++++++++ 1 file changed, 32 insertions(+) create mode 100644 src/processors/FBP.py (limited to 'src/processors/FBP.py') 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) -- cgit v1.2.3