colindex2 2.9.0__py3-none-any.whl

colindex2/__init__.py

@@ -0,0 +1,11 @@
+from .colindex2 import Detect
+from .commands import detect, track, gen_data_settings, find_track
+from .tracking_overlap2 import Track
+from .get_IDfile import Finder
+from .draw_map import _draw_map
+
+from .modules import great_circle_distance_numba
+from .modules import great_circle_distance  # for arrays
+from .modules import invert_gcd2
+from .modules import moving_direction
+from .modules import bilinear_interp

colindex2/circle_overlap.py

@@ -0,0 +1,122 @@
+#!/usr/bin/env python3
+import numpy as np
+import pandas as pd
+
+
+def area_heron(a, b, c):
+    s = (a+b+c)/2
+    if min(s-a, s-b, s-c) <= 0:
+        return np.nan
+    else:
+        return np.sqrt(s*(s-a)*(s-b)*(s-c))
+
+
+class circle:
+    def __init__(self, r, x):
+        self.r = r
+        self.x = x
+        self.S = np.pi * r * r
+
+
+class circle_overlap:
+    def __init__(self, c1, c2):
+
+        fct = 1
+        d = np.abs(c1.x - c2.x)
+        S = area_heron(c1.r, c2.r, d)
+
+        if np.isnan(S):   # cannot make inner triangle
+
+            alpha1 = np.nan
+            d1 = np.nan
+            h = np.nan
+
+            if d < c1.r:
+                overlap = 1.
+            else:
+                overlap = 0.
+
+        else:  # can make inner triangle
+            
+            alpha1 = np.arcsin(2 * S / c1.r / d)
+            h = S / d * 2
+            d1 = c1.r * np.cos(alpha1)
+
+            fan = c1.r**2 * alpha1 / 2
+            tri = d1 * h / 2
+
+            if c2.r > np.sqrt(c1.r**2+d**2):
+                fct = -1
+                left = fan - tri
+                right = c1.S / 2 - left
+            else:
+                right = fan - tri
+
+            overlap = min(right / c1.S * 2, 1)
+
+        self.d = d
+        self.S = S
+        self.overlap = overlap
+        self.alpha1 = alpha1
+        self.d1 = d1*fct
+        self.h = h
+
+
+def test():
+
+    import matplotlib.pyplot as plt
+    import matplotlib.patches as patches
+
+    r1, r2 = 500, 600
+
+    ds = np.arange(100, 1151, 50)
+    overs = []
+    alpha1s = []
+    d1s = []
+    hs = []
+
+    for d in ds:
+
+        c1 = circle(r=r1, x=0)
+        c2 = circle(r=r2, x=d)
+        o = cirlce_overlap(c1, c2)
+        overs.append(o.overlap)
+        alpha1s.append(o.alpha1)
+        d1s.append(o.d1)
+        hs.append(o.h)
+
+
+    for i, d in enumerate(ds):
+
+        print(f'{d=}')
+        
+        ax = plt.axes()
+        C1 = patches.Circle(xy=(0, 500), radius=r1, fc='none', ec='tab:blue')
+        C2 = patches.Circle(xy=(d, 500), radius=r2, fc='none', ec='tab:blue')
+        ax.add_patch(C1)
+        ax.add_patch(C2)
+        ax.plot(0, 500, marker='.', c='tab:blue')
+        ax.plot(d, 500, marker='.', c='tab:blue')
+        ax.plot([d1s[i], d1s[i]], [500-hs[i], 500+hs[i]], c='r')
+        ax.set_aspect('equal')
+        ax.set_xlim([-700,1250])
+        ax.set_ylim([-200, 1200])
+        ax.set_title(f'O={overs[i]:.3f}')
+        plt.savefig(f'fig/cc_d{d}.png')
+        plt.close()
+
+        ax = plt.axes()
+        ax.plot(ds, overs)
+        ax.plot(ds[i], overs[i], '.r')
+        plt.savefig(f'fig/overs_{d}.png')
+        plt.close()
+
+        '''ax = plt.axes()
+        ax.plot(ds, np.array(alpha1s)/np.pi)
+        ax.plot(ds[i], alpha1s[i]/np.pi, '.r')
+        plt.savefig(f'fig/alpha1s_d{d}.png')
+        plt.close()'''
+        
+
+if __name__ == '__main__':
+    test()