CRIkitUI.py 文件源码

def deNoiseNRB(self):
        """
        Denoise NRB with Savitky-Golay
        """
        # Range of pixels to perform-over
        rng = self.hsi.freq.op_range_pix

        plugin = _widgetSG(window_length=11, polyorder=3)
        winPlotEffect = _DialogPlotEffect.dialogPlotEffect(self.nrb.mean()[rng],
                                                           x=self.hsi.f,
                                                           plugin=plugin,
                                                           parent=self)
        if winPlotEffect is not None:
            win_size = winPlotEffect.parameters['window_length']
            order = winPlotEffect.parameters['polyorder']

            nrb_denoise = _copy.deepcopy(_np.squeeze(self.nrb.data))
            nrb_denoise[..., rng] = _sg(nrb_denoise[..., rng], win_size, order)

            self.nrb.data = nrb_denoise

            # Backup for Undo
            self.bcpre.add_step(['DenoiseNrbSG',
                                 'Win_size', win_size,
                                 'Order', order])

#            if self.ui.actionUndo_Backup_Enabled.isChecked():
#                try:
#                    _BCPre.backup_pickle(self.hsi, self.bcpre.id_list[-1])
#                except:
#                    print('Error in pickle backup (Undo functionality)')
#                else:
#                    self.bcpre.backed_up()
        self.changeSlider()