diff options
Diffstat (limited to 'tests/roof/graph.py')
-rw-r--r-- | tests/roof/graph.py | 15 |
1 files changed, 7 insertions, 8 deletions
diff --git a/tests/roof/graph.py b/tests/roof/graph.py index 495c629..62aa44e 100644 --- a/tests/roof/graph.py +++ b/tests/roof/graph.py @@ -32,13 +32,13 @@ class RoofGraph(RoofConfig): def get_roof_task(self, name, **kwargs): kwargs.update(config = self.config_file) return self.get_task(name, **kwargs) - + def get_processor_task(self, stage, name, **kwargs): extra_args = self.get_opt(stage, name + '-options') if extra_args is not None: kwargs.update(extra_args) if (re.compile('roof').match(name)): kwargs.update(config = self.config_file) return self.save_task(stage, name, self.get_task(name, **kwargs)) - + def get_read_task(self, path): params = { 'raw_width': self.fan_bins, 'raw_height': self.fan_projections, 'raw_bitdepth': self.bit_depth } if re.search('\.raw$', path) else { } params['path'] = path @@ -132,11 +132,11 @@ class RoofGraph(RoofConfig): def get_processor(self, head, stage, writer = None): # skip (but not if not already skipped in previous processor) # how to connect readers to ffc? - + filters = self.get_opt(stage, 'filters', roof_filters[stage]) read_here = self.args.read and self.args.read in roof_data_types[stage].keys() write_here = self.args.write and self.args.write in roof_data_types[stage].keys() - + start_pos = 0 if read_here: start_filter = roof_data_types[stage][self.args.read] @@ -159,7 +159,7 @@ class RoofGraph(RoofConfig): if write_here and writer: self.graph.connect_nodes(head, writer) - + return None if write_here else head @@ -172,7 +172,7 @@ class RoofGraph(RoofConfig): # - Write mode: The reconstruction is performed and data is written after the specified step (default) [ -w <all other data types> ] # - Control mode: Control branch and raw data writting [ -c ] # - GUI mode: Visualization in GUI + raw_sinograms are written when enabled in GUI + some control tasks (also when enabled) [ -g ] - + head = reader # Check if we are branching here if (self.args.track or self.args.gui) and (self.get_data_type() is not None): @@ -204,7 +204,6 @@ class RoofGraph(RoofConfig): # if head split to 3 branches.... Otherwise, continue with control branch... except finish: pass - + def run(self): self.scheduler.run(self.graph) -
\ No newline at end of file |