DiamondLightSource /
Savu
| @@ 100-110 (lines=11) @@ | ||
| 97 | self.setup_2D() |
|
| 98 | self.process_frames = self.astra_2D_vector_recon |
|
| 99 | ||
| 100 | def setup_2D(self): |
|
| 101 | pData = self.get_plugin_in_datasets()[0] |
|
| 102 | self.dim_detX = \ |
|
| 103 | pData.get_data_dimension_by_axis_label('x', contains=True) |
|
| 104 | self.dim_rot = \ |
|
| 105 | pData.get_data_dimension_by_axis_label('rot', contains=True) |
|
| 106 | ||
| 107 | self.sino_shape = pData.get_shape() |
|
| 108 | self.nDims = len(self.sino_shape) |
|
| 109 | self.nCols = self.sino_shape[self.dim_detX] |
|
| 110 | self.set_mask(self.sino_shape) |
|
| 111 | ||
| 112 | def setup_3D(self): |
|
| 113 | pData = self.get_plugin_in_datasets()[0] |
|
| @@ 97-107 (lines=11) @@ | ||
| 94 | self.setup_2D() |
|
| 95 | self.process_frames = self.astra_2D_recon |
|
| 96 | ||
| 97 | def setup_2D(self): |
|
| 98 | pData = self.get_plugin_in_datasets()[0] |
|
| 99 | self.dim_detX = \ |
|
| 100 | pData.get_data_dimension_by_axis_label('x', contains=True) |
|
| 101 | self.dim_rot = \ |
|
| 102 | pData.get_data_dimension_by_axis_label('rot', contains=True) |
|
| 103 | ||
| 104 | self.sino_shape = pData.get_shape() |
|
| 105 | self.nDims = len(self.sino_shape) |
|
| 106 | self.nCols = self.sino_shape[self.dim_detX] |
|
| 107 | self.set_mask(self.sino_shape) |
|
| 108 | ||
| 109 | def set_mask(self, shape): |
|
| 110 | l = self.get_plugin_out_datasets()[0].get_shape()[0] |
|
