-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathdata_structures.py
186 lines (137 loc) · 4.86 KB
/
data_structures.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
class CameraSettings:
def __init__(self):
# the 4 slider values
self.brightness = 240
self.gamma = 1
self.exposure = 0.0333
self.gain = 0
self.exposure_mode = True
self.iris = 1
self.video_format = 'RGB24 (1216x1024) [Skipping 2x]'
self.roi = [0, 0, 1216, 1024]
# Getter and Setter methods for brightness
def get_brightness(self):
return self.brightness
def set_brightness(self, brightness):
self.brightness = brightness
def reset_brightness(self):
self.brightness = 240
# Getter and Setter methods for gamma
def get_gamma(self):
return self.gamma
def set_gamma(self, gamma):
self.gamma = gamma
def reset_gamma(self):
self.gamma = 1
# Getter and Setter methods for iris
def get_iris(self):
return self.iris
def set_iris(self, iris):
self.iris = iris
def reset_iris(self):
self.iris = 1
# Getter and Setter methods for exposure
def get_exposure(self):
return self.exposure
def set_exposure(self, exposure):
self.exposure = exposure
def reset_exposure(self):
self.exposure = 0.0333
# Getter and Setter methods for exposure_mode
def get_exposure_mode(self):
return self.exposure_mode
def set_exposure_mode(self, exposure_mode):
self.exposure_mode = exposure_mode
def auto_exposure_mode(self):
self.exposure_mode = True
def reset_exposure_mode(self):
self.exposure_mode = True
def manual_exposure_mode(self):
self.exposure_mode = False
# Getter and Setter methods for gain
def get_gain(self):
return self.gain
def set_gain(self, gain):
self.gain = gain
# Getter and Setter methods for video format
def get_video_format(self):
return self.video_format
def set_video_format(self, video_format):
self.video_format = video_format
# Getter and Setter methods for roi
def get_roi(self):
return self.roi
def set_roi(self, roi):
self.roi = roi
class CalibrationSettings:
def __init__(self):
self.pixel_calibration = 49.5517721152208 # 1
self.TCAmmX = 3.500 # 2
self.TCAmmY = 3.500 # 3
self.tolerated_pupil_dist= 0.150 # 4
# Getter and Setter methods pixel_calibration
def get_pixel_calibration(self):
return self.pixel_calibration
def set_pixel_calibration(self, pixel_calibration):
self.pixel_calibration = pixel_calibration
# Getter and Setter methods TCAmmX
def get_TCAmmX(self):
return self.TCAmmX
def set_TCAmmX(self, TCAmmX):
self.TCAmmX = TCAmmX
# Getter and Setter methods TCAmmY
def get_TCAmmY(self):
return self.TCAmmY
def set_TCAmmY(self, TCAmmY):
self.TCAmmY = TCAmmY
# Getter and Setter methods tolerated_pupil_dist
def get_tolerated_pupil_dist(self):
return self.tolerated_pupil_dist
def set_tolerated_pupil_dist(self, tolerated_pupil_dist):
self.tolerated_pupil_dist = tolerated_pupil_dist
class SysParams:
def __init__(self):
self.pupil_tracker = 0
self.pupil_duration = 0
self.pupil_camera_fps = 0
self.pupil_diff_x = []
self.pupil_diff_y = []
self.pixel_per_deg = 0
self.real_system = 0
self.net_comm_obj = None
# Add other parameters as needed
# Getter and Setter for pupil_duration
def get_pupil_duration(self):
return self.pupil_duration
def set_pupil_duration(self, pupil_duration):
self.pupil_duration = pupil_duration
# Getter and Setter for pupil_camera_fps
def get_pupil_camera_fps(self):
return self.pupil_camera_fps
def set_pupil_camera_fps(self, pupil_camera_fps):
self.pupil_camera_fps = pupil_camera_fps
# Getter and Setter for pupil_diff_x
def get_pupil_diff_x(self):
return self.pupil_diff_x
def set_pupil_diff_x(self, pupil_diff_x):
self.pupil_diff_x = pupil_diff_x
# Getter and Setter for pupil_diff_y
def get_pupil_diff_y(self):
return self.pupil_diff_y
def set_pupil_diff_y(self, pupil_diff_y):
self.pupil_diff_y = pupil_diff_y
# Getter and Setter for pixel_per_deg
def get_pixel_per_deg(self):
return self.pixel_per_deg
def set_pixel_per_deg(self, pixel_per_deg):
self.pixel_per_deg = pixel_per_deg
# Getter and Setter for real_system
def get_real_system(self):
return self.real_system
def set_real_system(self, real_system):
self.real_system = real_system
# Getter and Setter for net_comm_obj
def get_net_comm_obj(self):
return self.net_comm_obj
def set_net_comm_obj(self, net_comm_obj):
self.net_comm_obj = net_comm_obj