Commit 3a01d91e authored by Pierre Paleo's avatar Pierre Paleo
Browse files

Add _allocate_recs method

parent e4f9a741
Pipeline #27103 passed with stages
in 2 minutes and 56 seconds
......@@ -277,8 +277,11 @@ class FullFieldPipeline:
x_s, x_e = options["start_x"], options["end_x"]+1
y_s, y_e = options["start_y"], options["end_y"]+1
self._rec_roi = (x_s, x_e, y_s, y_e)
self.n_slices = self.radios.shape[1] # TODO modify with vertical shifts
self.recs = self._allocate_array((self.n_slices, y_e - y_s, x_e - x_s), "f", name="recs")
self.n_slices = self.radios_shape[1] # TODO modify with vertical shifts
self._allocate_recs(y_e - y_s, x_e - x_s)
def _allocate_recs(self, ny, nx):
self.recs = self._allocate_array((self.n_slices, ny, nx), "f", name="recs")
@use_options("reconstruction", "reconstruction")
......
......@@ -211,6 +211,9 @@ class CudaFullFieldPipelineLimitedMemory(CudaFullFieldPipeline):
self._sinobuilder_output = self._h_sinos # patch self.sino_builder
return self._h_sinos
def _allocate_recs(self, ny, nx):
self.recs = self._allocate_array((self.chunk_size, ny, nx), "f", name="recs")
def _register_callbacks(self):
# No callbacks are registered for this subclass
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment