Merge pull request #76 from abel-research/aopFixes

Removed zipper effect on aop save due to poor interpolation
abel-research · Nov 1, 2022 · 8bf7184 · 8bf7184
2 parents 7a3d5f8 + 8277739
commit 8bf7184
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/ampscan/core.py b/ampscan/core.py
@@ -755,12 +755,14 @@ def save_aop(self, filename, slices=100, spokes=72, sliceInterval = None, spokeI
             z = slices[i] - minSl
             rPoly = ((x ** 2) + (y ** 2)) ** 0.5
             tPoly = np.rad2deg(np.arctan2(y, x))
-            tPoly[tPoly < -90] += 360
+            #  Increase range for interpolation
+            idx = tPoly < 0
+            rPoly = np.append(rPoly, rPoly[idx])
+            tPoly = np.append(tPoly, tPoly[idx] + 360)
+            # Sort by angle
             idx = np.argsort(tPoly)
             rPoly = rPoly[idx]
-            np.append(rPoly, rPoly[0])
             tPoly = tPoly[idx]
-            np.append(tPoly, 270)
             rs = np.interp(spokes, tPoly, rPoly)
             rs = np.flip(rs)
             for j, r in enumerate(rs):