diff --git a/webbpsf/wfirst.py b/webbpsf/wfirst.py index dc400871..b9c84ecd 100644 --- a/webbpsf/wfirst.py +++ b/webbpsf/wfirst.py @@ -477,6 +477,7 @@ def __init__(self): self.pupil_mask_list = self._pupil_controller.pupil_mask_list + self._is_custom_aberrations = False self._detector_npixels = 4096 self._detectors = _load_wfi_detector_aberrations(os.path.join(self._datapath, 'wim_zernikes_cycle8.csv')) assert len(self._detectors.keys()) > 0 @@ -556,6 +557,16 @@ def _unmasked_pupil_path(self): def _masked_pupil_path(self): return self._pupil_controller._masked_pupil_path + def override_aberrations(self, path): + """Override detector aberrations""" + self._detectors = _load_wfi_detector_aberrations(path) + self._is_custom_aberrations = True + + def reset_override_aberrations(self): + """Release detector aberrations override and loads default""" + self._detectors = _load_wfi_detector_aberrations(os.path.join(self._datapath, 'wim_zernikes_cycle8.csv')) + self._is_custom_aberrations = False + class CGI(WFIRSTInstrument): """