-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathdata_loader.py
More file actions
214 lines (209 loc) · 16.3 KB
/
Copy pathdata_loader.py
File metadata and controls
214 lines (209 loc) · 16.3 KB
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
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
# data_loader.py
import numpy as np
def load_coordinate_data() -> tuple[np.ndarray, np.ndarray, np.ndarray]:
"""Loads the S1xyz, S2xyz, and Moxyz coordinate datasets."""
S1xyz = np.array([
[[0.000308, 0.315, 1.41453], [0.273107, 0.1575, 1.41453], [0.545904, 0.315, 1.41453],
[0.818702, 0.1575, 1.41453], [1.0915, 0.315, 1.41453], [1.3643, 0.1575, 1.41453],
[1.6371, 0.315, 1.41453], [1.90989, 0.1575, 1.41453], [2.18269, 0.315, 1.41453],
[2.45549, 0.1575, 1.41453], [2.72829, 0.315, 1.41453], [3.00109, 0.1575, 1.41453],
[3.27389, 0.315, 1.41453], [3.54668, 0.1575, 1.41453], [3.81948, 0.315, 1.41453],
[4.09228, 0.1575, 1.41453]],
[[0.000068, 0.315, 1.45759], [0.271113, 0.1575, 1.45598], [0.542451, 0.315, 1.4514],
[0.814287, 0.1575, 1.44455], [1.08676, 0.315, 1.43644], [1.35991, 0.1575, 1.42831],
[1.63368, 0.315, 1.42139], [1.90792, 0.1575, 1.41677], [2.18239, 0.315, 1.41515],
[2.45683, 0.1575, 1.41678], [2.73096, 0.315, 1.42142], [3.00458, 0.1575, 1.42833],
[3.27755, 0.315, 1.43646], [3.54984, 0.1575, 1.44457], [3.82151, 0.315, 1.45142],
[4.09275, 0.1575, 1.45598]],
[[-0.0001764, 0.315, 1.49261], [0.268972, 0.1575, 1.48943], [0.538705, 0.315, 1.48036],
[0.809459, 0.1575, 1.46673], [1.08149, 0.315, 1.45054], [1.35487, 0.1575, 1.43426],
[1.62949, 0.315, 1.42039], [1.90502, 0.1575, 1.4111], [2.18103, 0.315, 1.40785],
[2.45694, 0.1575, 1.41115], [2.73226, 0.315, 1.42047], [3.00655, 0.1575, 1.43433],
[3.27957, 0.315, 1.45058], [3.55124, 0.1575, 1.46674], [3.82168, 0.315, 1.48036],
[4.09122, 0.1575, 1.48943]],
[[-0.0004171, 0.315, 1.53539], [0.266711, 0.1575, 1.53069], [0.534726, 0.315, 1.5172],
[0.804267, 0.1575, 1.49686], [1.07572, 0.315, 1.47264], [1.34919, 0.1575, 1.4482],
[1.62449, 0.315, 1.42733], [1.90119, 0.1575, 1.41333], [2.17859, 0.315, 1.40843],
[2.45588, 0.1575, 1.41342], [2.73223, 0.315, 1.42748], [3.00705, 0.1575, 1.44833],
[3.27997, 0.315, 1.4727], [3.55089, 0.1575, 1.49686], [3.81997, 0.315, 1.51717],
[4.08768, 0.1575, 1.53066]],
[[-0.0006367, 0.315, 1.57846], [0.265394, 0.1575, 1.57209], [0.531956, 0.315, 1.55399],
[0.799682, 0.1575, 1.52684], [1.06942, 0.315, 1.49463], [1.3419, 0.1575, 1.4621],
[1.6174, 0.315, 1.43418], [1.89551, 0.1575, 1.41533], [2.17509, 0.315, 1.40873],
[2.45451, 0.1575, 1.41547], [2.73222, 0.315, 1.43437], [3.0071, 0.1575, 1.46229],
[3.27881, 0.315, 1.49477], [3.5478, 0.1575, 1.52691], [3.81493, 0.315, 1.55402],
[4.08112, 0.1575, 1.57211]],
[[-0.0008813, 0.315, 1.62103], [0.263419, 0.1575, 1.61313], [0.528222, 0.315, 1.59056],
[0.794243, 0.1575, 1.55676], [1.06254, 0.315, 1.51668], [1.33426, 0.1575, 1.47609],
[1.60998, 0.315, 1.44115], [1.88925, 0.1575, 1.41751], [2.17058, 0.315, 1.40922],
[2.4517, 0.1575, 1.41772], [2.73045, 0.315, 1.44145], [3.0054, 0.1575, 1.47634],
[3.27617, 0.315, 1.5168], [3.54354, 0.1575, 1.55678], [3.80879, 0.315, 1.59053],
[4.07311, 0.1575, 1.61308]],
[[-0.0011095, 0.315, 1.66331], [0.261396, 0.1575, 1.65387], [0.524375, 0.315, 1.62692],
[0.788543, 0.1575, 1.58659], [1.05526, 0.315, 1.53872], [1.32605, 0.1575, 1.49015],
[1.60181, 0.315, 1.44817], [1.88211, 0.1575, 1.41969], [2.16505, 0.315, 1.40972],
[2.44777, 0.1575, 1.42], [2.72745, 0.315, 1.44859], [3.00228, 0.1575, 1.49047],
[3.27195, 0.315, 1.53883], [3.53754, 0.1575, 1.58652], [3.80079, 0.315, 1.62678],
[4.06317, 0.1575, 1.65376]],
[[-0.0013715, 0.315, 1.70529], [0.25932, 0.1575, 1.69436], [0.52045, 0.315, 1.66312],
[0.782719, 0.1575, 1.61627], [1.04767, 0.315, 1.56063], [1.31722, 0.1575, 1.50414],
[1.59273, 0.315, 1.45521], [1.87394, 0.1575, 1.4219], [2.15856, 0.315, 1.4102],
[2.44289, 0.1575, 1.42228], [2.72334, 0.315, 1.45576], [2.99776, 0.1575, 1.50455],
[3.26606, 0.315, 1.56076], [3.52976, 0.1575, 1.61614], [3.79096, 0.315, 1.66288],
[4.05132, 0.1575, 1.69421]],
[[-0.0016186, 0.315, 1.74686], [0.257249, 0.1575, 1.7345], [0.516502, 0.315, 1.69904],
[0.776762, 0.1575, 1.64578], [1.03968, 0.315, 1.58248], [1.30766, 0.1575, 1.51814],
[1.58269, 0.315, 1.46226], [1.86476, 0.1575, 1.42406], [2.15111, 0.315, 1.41062],
[2.43711, 0.1575, 1.42453], [2.71823, 0.315, 1.4629], [2.992, 0.1575, 1.51861],
[3.25863, 0.315, 1.58263], [3.52021, 0.1575, 1.64562], [3.77923, 0.315, 1.69871],
[4.03756, 0.1575, 1.73426]],
[[-0.0018697, 0.315, 1.78802], [0.255202, 0.1575, 1.77421], [0.512514, 0.315, 1.73457],
[0.770565, 0.1575, 1.67505], [1.0312, 0.315, 1.60426], [1.29741, 0.1575, 1.53216],
[1.57181, 0.315, 1.46931], [1.85466, 0.1575, 1.42615], [2.14274, 0.315, 1.41095],
[2.4304, 0.1575, 1.42674], [2.71213, 0.315, 1.47007], [2.9851, 0.1575, 1.53267],
[3.24983, 0.315, 1.60438], [3.50903, 0.1575, 1.67483], [3.7657, 0.315, 1.73417],
[4.02194, 0.1575, 1.77391]],
[[-0.0021325, 0.315, 1.82868], [0.253148, 0.1575, 1.81342], [0.50848, 0.315, 1.76965],
[0.764185, 0.1575, 1.70396], [1.02235, 0.315, 1.62584], [1.28662, 0.1575, 1.54615],
[1.56029, 0.315, 1.47639], [1.84381, 0.1575, 1.42827], [2.1335, 0.315, 1.41129],
[2.42271, 0.1575, 1.42899], [2.70498, 0.315, 1.4773], [2.97706, 0.1575, 1.54673],
[3.23972, 0.315, 1.62594], [3.49633, 0.1575, 1.70367], [3.75051, 0.315, 1.76916],
[4.00458, 0.1575, 1.81308]]
])
S2xyz = np.array([
[[0.0003078, 0.315, 1.71547], [0.273107, 0.1575, 1.71547], [0.545903, 0.315, 1.71547],
[0.818701, 0.1575, 1.71547], [1.0915, 0.315, 1.71547], [1.3643, 0.1575, 1.71547],
[1.6371, 0.315, 1.71547], [1.90989, 0.1575, 1.71547], [2.18269, 0.315, 1.71547],
[2.45549, 0.1575, 1.71547], [2.72829, 0.315, 1.71547], [3.00109, 0.1575, 1.71547],
[3.27389, 0.315, 1.71547], [3.54668, 0.1575, 1.71547], [3.81948, 0.315, 1.71547],
[4.09228, 0.1575, 1.71547]],
[[0.0005483, 0.315, 1.75855], [0.274984, 0.1575, 1.75691], [0.549111, 0.315, 1.75228],
[0.822728, 0.1575, 1.74536], [1.0957, 0.315, 1.73724], [1.36799, 0.1575, 1.72913],
[1.63967, 0.315, 1.72228], [1.9109, 0.1575, 1.71771], [2.18191, 0.315, 1.71611],
[2.45296, 0.1575, 1.71772], [2.7243, 0.315, 1.72229], [2.99613, 0.1575, 1.72914],
[3.26861, 0.315, 1.73726], [3.54176, 0.1575, 1.74539], [3.81553, 0.315, 1.75231],
[4.08977, 0.1575, 1.75693]],
[[0.0007919, 0.315, 1.79363], [0.27671, 0.1575, 1.79033], [0.552026, 0.315, 1.78101],
[0.826319, 0.1575, 1.76714], [1.09934, 0.315, 1.75089], [1.37101, 0.1575, 1.73474],
[1.64145, 0.315, 1.72112], [1.91098, 0.1575, 1.71205], [2.18006, 0.315, 1.70887],
[2.44921, 0.1575, 1.71204], [2.71894, 0.315, 1.72111], [2.98969, 0.1575, 1.73475],
[3.26172, 0.315, 1.75093], [3.53511, 0.1575, 1.76722], [3.80972, 0.315, 1.78109],
[4.08526, 0.1575, 1.79038]],
[[0.0010407, 0.315, 1.83648], [0.278321, 0.1575, 1.83149], [0.554676, 0.315, 1.81744],
[0.829497, 0.1575, 1.79659], [1.10242, 0.315, 1.77221], [1.37334, 0.1575, 1.74806],
[1.64242, 0.315, 1.72775], [1.91013, 0.1575, 1.71426], [2.17714, 0.315, 1.70952],
[2.44426, 0.1575, 1.71423], [2.71228, 0.315, 1.72771], [2.98182, 0.1575, 1.74806],
[3.25327, 0.315, 1.77227], [3.52674, 0.1575, 1.79672], [3.80204, 0.315, 1.81759],
[4.07874, 0.1575, 1.83159]],
[[0.0012621, 0.315, 1.87942], [0.280685, 0.1575, 1.87268], [0.558392, 0.315, 1.85379],
[0.833282, 0.1575, 1.82587], [1.10499, 0.315, 1.79339], [1.37397, 0.1575, 1.76125],
[1.64111, 0.315, 1.73415], [1.9073, 0.1575, 1.71604], [2.17319, 0.315, 1.70969],
[2.43922, 0.1575, 1.71606], [2.70578, 0.315, 1.73417], [2.97351, 0.1575, 1.76133],
[3.24325, 0.315, 1.79353], [3.51571, 0.1575, 1.82606], [3.79123, 0.315, 1.85399],
[4.06934, 0.1575, 1.87282]],
[[0.0015, 0.315, 1.92193], [0.282633, 0.1575, 1.91345], [0.561399, 0.315, 1.88971],
[0.836327, 0.1575, 1.85481], [1.10711, 0.315, 1.81436], [1.37447, 0.1575, 1.77437],
[1.63973, 0.315, 1.74064], [1.90405, 0.1575, 1.71807], [2.16818, 0.315, 1.71012],
[2.43248, 0.1575, 1.71805], [2.6973, 0.315, 1.7406], [2.9633, 0.1575, 1.77439],
[3.23161, 0.315, 1.81448], [3.50332, 0.1575, 1.85506], [3.77904, 0.315, 1.89002],
[4.05832, 0.1575, 1.91365]],
[[0.0017541, 0.315, 1.96414], [0.284481, 0.1575, 1.95386], [0.564146, 0.315, 1.92525],
[0.838976, 0.1575, 1.8834], [1.10866, 0.315, 1.83504], [1.37426, 0.1575, 1.78733],
[1.6375, 0.315, 1.74704], [1.89987, 0.1575, 1.72007], [2.16218, 0.315, 1.71055],
[2.4247, 0.1575, 1.71998], [2.68767, 0.315, 1.74692], [2.95183, 0.1575, 1.78727],
[3.21856, 0.315, 1.83515], [3.48936, 0.1575, 1.88371], [3.76511, 0.315, 1.92566],
[4.0454, 0.1575, 1.95414]],
[[0.0020236, 0.315, 2.00598], [0.286352, 0.1575, 1.99388], [0.566797, 0.315, 1.96041],
[0.841227, 0.1575, 1.91162], [1.10952, 0.315, 1.8554], [1.37322, 0.1575, 1.80002],
[1.63441, 0.315, 1.75328], [1.89478, 0.1575, 1.72198], [2.15518, 0.315, 1.71089],
[2.41586, 0.1575, 1.7218], [2.67699, 0.315, 1.75305], [2.93926, 0.1575, 1.7999],
[3.20421, 0.315, 1.85553], [3.47376, 0.1575, 1.91202], [3.74926, 0.315, 1.96095],
[4.03047, 0.1575, 1.99428]],
[[0.0022649, 0.315, 2.04745], [0.288264, 0.1575, 2.03354], [0.56939, 0.315, 1.99516],
[0.843158, 0.1575, 1.93945], [1.10979, 0.315, 1.87544], [1.37137, 0.1575, 1.81245],
[1.63039, 0.315, 1.75935], [1.88871, 0.1575, 1.7238], [2.14723, 0.315, 1.71121],
[2.40609, 0.1575, 1.72356], [2.66535, 0.315, 1.75903], [2.9256, 0.1575, 1.81228],
[3.18852, 0.315, 1.87559], [3.45651, 0.1575, 1.93993], [3.73154, 0.315, 1.9958],
[4.01359, 0.1575, 2.03401]],
[[0.0025066, 0.315, 2.08855], [0.29017, 0.1575, 2.07276], [0.571903, 0.315, 2.02943],
[0.844868, 0.1575, 1.96682], [1.1096, 0.315, 1.89511], [1.3688, 0.1575, 1.82466],
[1.62548, 0.315, 1.76532], [1.88171, 0.1575, 1.72558], [2.13836, 0.315, 1.71148],
[2.39543, 0.1575, 1.72529], [2.65275, 0.315, 1.76493], [2.9108, 0.1575, 1.82445],
[3.17143, 0.315, 1.89523], [3.43764, 0.1575, 1.96733], [3.71204, 0.315, 2.03018],
[3.99488, 0.1575, 2.07334]],
[[0.0027684, 0.315, 2.12912], [0.291967, 0.1575, 2.1114], [0.574237, 0.315, 2.06308],
[0.846315, 0.1575, 1.99365], [1.10897, 0.315, 1.91444], [1.36558, 0.1575, 1.83674],
[1.61976, 0.315, 1.77127], [1.87385, 0.1575, 1.72735], [2.12863, 0.315, 1.71173],
[2.38389, 0.1575, 1.72697], [2.63923, 0.315, 1.77074], [2.89493, 0.1575, 1.83642],
[3.1531, 0.315, 1.91454], [3.41735, 0.1575, 1.99424], [3.69102, 0.315, 2.06404],
[3.97455, 0.1575, 2.11216]]
])
Moxyz = np.array([
[[0.181254, 0.315, 1.565], [0.45405, 0.1575, 1.565], [0.726847, 0.315, 1.565],
[0.999647, 0.1575, 1.565], [1.27245, 0.315, 1.565], [1.54524, 0.1575, 1.565],
[1.81804, 0.315, 1.565], [2.09084, 0.1575, 1.565], [2.36364, 0.315, 1.565],
[2.63643, 0.1575, 1.565], [2.90923, 0.315, 1.565], [3.18203, 0.1575, 1.565],
[3.45483, 0.315, 1.565], [3.72763, 0.1575, 1.565], [4.00042, 0.315, 1.565],
[4.27322, 0.1575, 1.565]],
[[0.181219, 0.315, 1.60744], [0.453955, 0.1575, 1.60376], [0.726683, 0.315, 1.59753],
[0.999407, 0.1575, 1.58966], [1.27212, 0.315, 1.58137], [1.54484, 0.1575, 1.57391],
[1.81757, 0.315, 1.56841], [2.09032, 0.1575, 1.56574], [2.36306, 0.315, 1.56627],
[2.6358, 0.1575, 1.56993], [2.90853, 0.315, 1.57617], [3.18125, 0.1575, 1.58403],
[3.45397, 0.315, 1.59233], [3.72669, 0.1575, 1.59979], [3.99942, 0.315, 1.60528],
[4.27216, 0.1575, 1.60797]],
[[0.181102, 0.315, 1.64185], [0.453643, 0.1575, 1.63452], [0.726166, 0.315, 1.62205],
[0.998681, 0.1575, 1.60635], [1.27119, 0.315, 1.58979], [1.5437, 0.1575, 1.57488],
[1.81623, 0.315, 1.56393], [2.08878, 0.1575, 1.55856], [2.36134, 0.315, 1.55963],
[2.63388, 0.1575, 1.56695], [2.9064, 0.315, 1.57942], [3.17891, 0.1575, 1.59513],
[3.45142, 0.315, 1.61169], [3.72393, 0.1575, 1.62659], [3.99646, 0.315, 1.63755],
[4.26902, 0.1575, 1.64291]],
[[0.180902, 0.315, 1.68404], [0.45312, 0.1575, 1.67306], [0.725299, 0.315, 1.65439],
[0.997454, 0.1575, 1.63085], [1.2696, 0.315, 1.60604], [1.54177, 0.1575, 1.58372],
[1.81397, 0.315, 1.5673], [2.08621, 0.1575, 1.55928], [2.35846, 0.315, 1.56088],
[2.63067, 0.1575, 1.57185], [2.90285, 0.315, 1.59053], [3.17501, 0.1575, 1.61406],
[3.44716, 0.315, 1.63887], [3.71932, 0.1575, 1.6612], [3.99152, 0.315, 1.67761],
[4.26376, 0.1575, 1.68564]],
[[0.181233, 0.315, 1.72636], [0.453637, 0.1575, 1.71165], [0.725247, 0.315, 1.68668],
[0.996135, 0.1575, 1.65529], [1.26679, 0.315, 1.62227], [1.53783, 0.1575, 1.59251],
[1.80966, 0.315, 1.57049], [2.08224, 0.1575, 1.55967], [2.35506, 0.315, 1.56179],
[2.62746, 0.1575, 1.57651], [2.89907, 0.315, 1.60149], [3.16996, 0.1575, 1.63287],
[3.44061, 0.315, 1.66588], [3.71165, 0.1575, 1.69566], [3.98349, 0.315, 1.71766],
[4.25607, 0.1575, 1.72848]],
[[0.18126, 0.315, 1.76826], [0.453473, 0.1575, 1.74981], [0.724399, 0.315, 1.71859],
[0.994173, 0.1575, 1.67952], [1.26358, 0.315, 1.63838], [1.53364, 0.1575, 1.60127],
[1.80496, 0.315, 1.57377], [2.07746, 0.1575, 1.56023], [2.35033, 0.315, 1.56292],
[2.62254, 0.1575, 1.58136], [2.89346, 0.315, 1.61255], [3.16323, 0.1575, 1.65165],
[3.43265, 0.315, 1.69277], [3.7027, 0.1575, 1.7299], [3.97402, 0.315, 1.7574],
[4.24652, 0.1575, 1.77092]],
[[0.181233, 0.315, 1.80984], [0.453149, 0.1575, 1.78766], [0.723247, 0.315, 1.75029],
[0.991724, 0.1575, 1.70358], [1.25972, 0.315, 1.65444], [1.52863, 0.1575, 1.61002],
[1.79934, 0.315, 1.57703], [2.07167, 0.1575, 1.56079], [2.34453, 0.315, 1.56402],
[2.61644, 0.1575, 1.58617], [2.88654, 0.315, 1.62357], [3.15502, 0.1575, 1.6703],
[3.42302, 0.315, 1.71942], [3.69193, 0.1575, 1.76381], [3.96263, 0.315, 1.79679],
[4.23497, 0.1575, 1.81306]],
[[0.18119, 0.315, 1.85109], [0.452771, 0.1575, 1.82523], [0.721909, 0.315, 1.78172],
[0.98887, 0.1575, 1.72742], [1.2552, 0.315, 1.67036], [1.52278, 0.1575, 1.61871],
[1.79275, 0.315, 1.58027], [2.06491, 0.1575, 1.56129], [2.33773, 0.315, 1.56507],
[2.60931, 0.1575, 1.59093], [2.87845, 0.315, 1.63446], [3.14541, 0.1575, 1.68874],
[3.41174, 0.315, 1.7458], [3.67932, 0.1575, 1.79745], [3.94929, 0.315, 1.83592],
[4.22145, 0.1575, 1.85489]],
[[0.181178, 0.315, 1.892], [0.452399, 0.1575, 1.86246], [0.720472, 0.315, 1.81284],
[0.985709, 0.1575, 1.75106], [1.2501, 0.315, 1.68615], [1.5161, 0.1575, 1.62732],
[1.78523, 0.315, 1.58344], [2.05719, 0.1575, 1.56176], [2.33002, 0.315, 1.56607],
[2.60124, 0.1575, 1.59561], [2.86931, 0.315, 1.64522], [3.13455, 0.1575, 1.707],
[3.39894, 0.315, 1.77192], [3.66494, 0.1575, 1.83075], [3.93407, 0.315, 1.87462],
[4.20603, 0.1575, 1.89631]],
[[0.181186, 0.315, 1.93248], [0.452016, 0.1575, 1.89924], [0.718887, 0.315, 1.84358],
[0.9822, 0.1575, 1.77442], [1.24442, 0.315, 1.70177], [1.50864, 0.1575, 1.63586],
[1.7768, 0.315, 1.58657], [2.04854, 0.1575, 1.56215], [2.32142, 0.315, 1.56702],
[2.59225, 0.1575, 1.60027], [2.85912, 0.315, 1.65592], [3.12243, 0.1575, 1.72507],
[3.38465, 0.315, 1.79772], [3.64887, 0.1575, 1.86364], [3.91702, 0.315, 1.91291],
[4.18877, 0.1575, 1.93734]],
[[0.181139, 0.315, 1.97241], [0.451527, 0.1575, 1.93547], [0.717077, 0.315, 1.87383],
[0.978255, 0.1575, 1.79745], [1.23811, 0.315, 1.71727], [1.50043, 0.1575, 1.64438],
[1.76754, 0.315, 1.58972], [2.03903, 0.1575, 1.56256], [2.31188, 0.315, 1.56797],
[2.58227, 0.1575, 1.60492], [2.84782, 0.315, 1.66654], [3.109, 0.1575, 1.74292],
[3.36885, 0.315, 1.82312], [3.63117, 0.1575, 1.89603], [3.89827, 0.315, 1.95072],
[4.16977, 0.1575, 1.97786]]
])
return S1xyz, S2xyz, Moxyz