Adding spine line angle relative to segment

Author: jtle00

mapmanagercore/layers/line.py

@@ -7,7 +7,8 @@
 import shapely
 import geopandas as gp
 from ..benchmark import timer
-
+import math
+from math import pi as PI
 
 class MultiLineLayer(Layer):
     @Layer.setProperty
@@ -128,6 +129,48 @@ def getSpineSide(line: LineString, spine: Point):
     val = getSide(first, last, spine)
     return val
 
+# abj
+@ timer
+def getSpineAngle(segmentLine: LineString, spineLine: LineString):
+    """ Return the angle between the two Lines
+    Line 1: The line formed between the spine head and the anchor point
+    Line 2: The line formed by two points on the segment tracing. 
+    Grab two points, one “up” and the other “down” the segment from the spine anchor point. 
+    I think the anchor point on the segment tracing is our new “position”.
+
+    Args:
+        segmentLine: segment in the for of a LineString
+        spineLine: Linestring of spine head to anchor point
+    """
+    spineLineCoord0 = Point(spineLine.coords[0])
+    spineLineCoord1 = Point(spineLine.coords[1])
+    sl0x = spineLineCoord0.x
+    sl0y = spineLineCoord0.y
+    sl1x = spineLineCoord1.x
+    sl1y = spineLineCoord1.y
+
+    segmentLineCoord0 = Point(segmentLine.coords[0])
+    segmentLineCoord1 = Point(segmentLine.coords[-1])
+    sgl0x = segmentLineCoord0.x
+    sgl0y = segmentLineCoord0.y
+    sgl1x = segmentLineCoord1.x
+    sgl1y = segmentLineCoord1.y
+    
+    m1 = (sl1y-sl0y)/(sl1x-sl0x)
+    m2 = (sgl1y-sgl0y)/(sgl1x-sgl0x)
+
+    angle_rad = math.atan(m1) - math.atan(m2)
+    angle_deg = angle_rad*180/PI
+
+    # Range: 0 - 360
+    # Check for Negative angle and add 360 degrees to determine counter clockwise value
+    if angle_deg < 0:
+        angle_deg = angle_deg + 360 
+
+    # print("m1", m1, "m2", m2, "degree:", angle_deg)
+
+    return angle_deg
+
 # abb
 @ timer
 def getSpinePositon(line: LineString, origin: Point):

mapmanagercore/schemas/spine.py

@@ -1,6 +1,6 @@
 from shapely.geometry import Point
 import numpy as np
-from ..layers.line import calcSubLine, extend, getSpinePositon, getSpineSide
+from ..layers.line import calcSubLine, extend, getSpinePositon, getSpineSide, getSpineAngle
 import shapely
 from ..lazy_geo_pandas import schema, calculated, LazyGeoFrame
 import geopandas as gp
@@ -114,6 +114,7 @@ class Spine:
     segmentID: int
     point: Point
     anchor: Point
+    spineAngle: LineString
 
     xBackgroundOffset: float
     yBackgroundOffset: float
@@ -175,6 +176,21 @@ def spineSide(frame: LazyGeoFrame):
     @calculated(title="Anchor", dependencies=["anchor", "point"], plot=False)
     def anchorLine(frame: LazyGeoFrame):
         return frame[["anchor", "point"]].apply(lambda x: LineString([x["anchor"], x["point"]]), axis=1)
+    
+    # abj
+    @calculated(title="Spine Angle", dependencies=["point"])
+    def angle(frame: LazyGeoFrame):
+        
+        # do this for all spines
+        segmentFrame = frame.getFrame("Segment")
+        df = frame[["segmentID", "point", "anchorLine"]].join(
+            segmentFrame[["segment"]], on=["segmentID", "t"])
+        print("spineAngle df:", df)
+
+        # # Create a dataframe of 
+        _ret = df.apply(lambda d: getSpineAngle(d["segment"], d["anchorLine"]), axis=1)
+        
+        return _ret
 
     @calculated(tile="ROI Base", dependencies={
         "Spine": ["anchor"],