DiamondLightSource /
Savu
| @@ 45-54 (lines=10) @@ | ||
| 42 | self.filtercomplex = None |
|
| 43 | self.count = 0 |
|
| 44 | ||
| 45 | def set_filter_padding(self, in_pData, out_pData): |
|
| 46 | in_data = self.get_in_datasets()[0] |
|
| 47 | det_x = in_data.get_data_dimension_by_axis_label('detector_x') |
|
| 48 | det_y = in_data.get_data_dimension_by_axis_label('detector_y') |
|
| 49 | pad_det_y = '%s.%s' % (det_y, self.parameters['Padtopbottom']) |
|
| 50 | pad_det_x = '%s.%s' % (det_x, self.parameters['Padleftright']) |
|
| 51 | mode = self.parameters['Padmethod'] |
|
| 52 | pad_dict = {'pad_directions': [pad_det_x, pad_det_y], 'pad_mode': mode} |
|
| 53 | in_pData[0].padding = pad_dict |
|
| 54 | out_pData[0].padding = pad_dict |
|
| 55 | ||
| 56 | def pre_process(self): |
|
| 57 | self._setup_paganin(*self.get_plugin_in_datasets()[0].get_shape()) |
|
| @@ 44-54 (lines=11) @@ | ||
| 41 | self.Vert_det = None |
|
| 42 | self.pad = None |
|
| 43 | ||
| 44 | @setup_extra_plugin_data_padding |
|
| 45 | def set_filter_padding(self, in_pData, out_pData): |
|
| 46 | self.pad = self.parameters['padding'] |
|
| 47 | in_data = self.get_in_datasets()[0] |
|
| 48 | det_y = in_data.get_data_dimension_by_axis_label('detector_y') |
|
| 49 | pad_det_y = '%s.%s' % (det_y, self.pad) |
|
| 50 | pad_dict = {'pad_directions': [pad_det_y], 'pad_mode': 'edge'} |
|
| 51 | in_pData[0].padding = pad_dict |
|
| 52 | out_pData[0].padding = pad_dict |
|
| 53 | if len(self.get_in_datasets()) > 1: |
|
| 54 | in_pData[1].padding = pad_dict |
|
| 55 | ||
| 56 | @enable_iterative_loop |
|
| 57 | def setup(self): |
|
