diff options
Diffstat (limited to 'src/processors/FBP.py')
-rw-r--r-- | src/processors/FBP.py | 32 |
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) |