diff options
Diffstat (limited to 'tests/roof/graph.py')
| -rw-r--r-- | tests/roof/graph.py | 10 |
1 files changed, 3 insertions, 7 deletions
diff --git a/tests/roof/graph.py b/tests/roof/graph.py index 62aa44e..0f516bc 100644 --- a/tests/roof/graph.py +++ b/tests/roof/graph.py @@ -1,3 +1,4 @@ +import math import re import gi @@ -57,7 +58,7 @@ class RoofGraph(RoofConfig): params = { 'path': path, 'first': first, 'step': step } if self.args.number: params['number'] = self.args.number - + print ("Reading {} data from {}".format(self.args.read,path)) # FIXME: handle raw data parameters return self.get_task('read', **params) @@ -72,12 +73,7 @@ class RoofGraph(RoofConfig): # Reconstruction from network or simulated data (also generation of flat/dark-fields) build_type = "raw" if self.args.noroof else "sino" if self.check_writer_type_is_raw() else "ufo" - build = self.get_roof_task('roof-build', simulate = self.args.simulate, number = self.args.number, build = build_type) - for id in range(self.streams): - read = self.get_roof_task('roof-read', id = id, simulate = self.args.simulate, path = path, first_file_number = first) - self.graph.connect_nodes(read, build) - build.bind_property('stop', read, 'stop', GObject.BindingFlags.DEFAULT) - + build = self.get_roof_task('roof-build', simulate = self.args.simulate, path = path, first_file_number = first, number = self.args.number, build = build_type) return build def get_writer(self): |