blob: da0ad4f2ecd251d5ca645fc339fff404688d0241 [file] [log] [blame]
Maxwell Hendersonf5123fe2023-02-04 13:44:41 -08001import numpy
2
3from graph_tools import *
4
Maxwell Henderson44913482023-02-04 16:37:27 -08005neutral = to_theta_with_circular_index(-0.2, 0.33, circular_index=-1)
6zero = to_theta_with_circular_index(0.0, 0.0, circular_index=-1)
Maxwell Hendersonf5123fe2023-02-04 13:44:41 -08007
Maxwell Henderson44913482023-02-04 16:37:27 -08008neutral_to_cone_1 = to_theta_with_circular_index(0.0, 0.7, circular_index=-1)
9neutral_to_cone_2 = to_theta_with_circular_index(0.2, 0.5, circular_index=-1)
10cone_pos = to_theta_with_circular_index(1.0, 0.4, circular_index=-1)
Maxwell Hendersonf5123fe2023-02-04 13:44:41 -080011
Maxwell Henderson44913482023-02-04 16:37:27 -080012neutral_to_cone_perch_pos_1 = to_theta_with_circular_index(0.4,
13 1.0,
14 circular_index=-1)
15neutral_to_cone_perch_pos_2 = to_theta_with_circular_index(0.7,
16 1.5,
17 circular_index=-1)
18cone_perch_pos = to_theta_with_circular_index(1.0, 2.0, circular_index=-1)
Maxwell Hendersonf5123fe2023-02-04 13:44:41 -080019
Maxwell Henderson44913482023-02-04 16:37:27 -080020segments = [
21 ThetaSplineSegment(neutral, neutral_to_cone_1, neutral_to_cone_2, cone_pos,
22 "NeutralToCone"),
23 ThetaSplineSegment(neutral, neutral_to_cone_perch_pos_1,
24 neutral_to_cone_perch_pos_2, cone_perch_pos,
25 "NeutralToPerchedCone"),
Maxwell Hendersonf5123fe2023-02-04 13:44:41 -080026]