DiamondLightSource /
Savu
| @@ 75-85 (lines=11) @@ | ||
| 72 | self.setup_2D() |
|
| 73 | self.process_frames = self.astra_2D_recon |
|
| 74 | ||
| 75 | def setup_2D(self): |
|
| 76 | pData = self.get_plugin_in_datasets()[0] |
|
| 77 | self.dim_detX = \ |
|
| 78 | pData.get_data_dimension_by_axis_label('x', contains=True) |
|
| 79 | self.dim_rot = \ |
|
| 80 | pData.get_data_dimension_by_axis_label('rot', contains=True) |
|
| 81 | ||
| 82 | self.sino_shape = pData.get_shape() |
|
| 83 | self.nDims = len(self.sino_shape) |
|
| 84 | self.nCols = self.sino_shape[self.dim_detX] |
|
| 85 | self.set_mask(self.sino_shape) |
|
| 86 | ||
| 87 | def set_mask(self, shape): |
|
| 88 | l = self.get_plugin_out_datasets()[0].get_shape()[0] |
|
| @@ 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] |
|
