BUG: fix offset accumulation rules

andykee · Dec 16, 2020 · 1659da4 · 1659da4
1 parent cd638c4
commit 1659da4
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/lentil/plane.py b/lentil/plane.py
@@ -593,8 +593,11 @@ class and any subclasses by reimplementing any of the following methods:
                     ofst = util.slice_offset(slc[seg], amplitude.shape)
 
                     # if wavefront.offset is empty, we will just append the offset for each slice as we go (even if offset is None)
-                    if not wavefront.offset:
+                    #if not wavefront.offset:
+                    if ofst:
                         offset.append(ofst)
+                    else:
+                        offset.append(None)
 
                     # if len(wavefront.offset) == 1
 
@@ -614,11 +617,14 @@ class and any subclasses by reimplementing any of the following methods:
                     ofst = util.slice_offset(s, amplitude.shape)
 
                     # if wavefront.offset is empty, we will just append the offset for each slice as we go (even if offset is None)
-                    if not wavefront.offset:
+                    # if not wavefront.offset:
+                    if ofst:
                         offset.append(ofst)
+                    else:
+                        offset.append(None)
 
-
-        if offset:
+        if not wavefront.offset or not all(wavefront.offset):
+            # of wavefront.offset is [] or a list of [None]:
              wavefront.offset = offset
 
         # TODO: verify this should really be extend and not append