JP-2037 and JP-1926: Fixing ramp fitting multiprocessing to work with…

… new STCAL interface. (#30) * Fixing ramp fitting multiprocessing to work with new STCAL interface. * Making changes based on code review. * Final DQ array for multiprocessing was incorrectly allocated as dtype=nd.float32, but should be dtype=np.uint32. Changed to allocate an array with the correct dtype. * Changing DQ to be uint32, not float32. * Adding addition of multiprocessing to change log.
spacetelescope · Jul 28, 2021 · 8cf81e0 · 8cf81e0
1 parent f7aef64
commit 8cf81e0
Show file tree

Hide file tree

Showing 3 changed files with 408 additions and 360 deletions.
diff --git a/CHANGES.rst b/CHANGES.rst
@@ -11,6 +11,7 @@
 ramp_fitting
 ------------
 
+- Implemented multiprocessing for OLS. [#30]
 - Added DQ flag parameter to `ramp_fit` [#25]
 
 - Move common ``jump`` code to stcal [#27]

diff --git a/src/stcal/ramp_fitting/constants.py b/src/stcal/ramp_fitting/constants.py
@@ -13,3 +13,11 @@ def update_dqflags(input_flags):
     dqflags["JUMP_DET"] = input_flags["JUMP_DET"]
     dqflags["NO_GAIN_VALUE"] = input_flags["NO_GAIN_VALUE"]
     dqflags["UNRELIABLE_SLOPE"] = input_flags["UNRELIABLE_SLOPE"]
+
+
+def update_dqflags_from_ramp_data(ramp_data):
+    dqflags["DO_NOT_USE"] = ramp_data.flags_do_not_use
+    dqflags["SATURATED"] = ramp_data.flags_saturated
+    dqflags["JUMP_DET"] = ramp_data.flags_jump_det
+    dqflags["NO_GAIN_VALUE"] = ramp_data.flags_no_gain_val
+    dqflags["UNRELIABLE_SLOPE"] = ramp_data.flags_unreliable_slope