-
Notifications
You must be signed in to change notification settings - Fork 0
/
cdip_rose_data.py
186 lines (111 loc) · 5.35 KB
/
cdip_rose_data.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
import netCDF4
import numpy as np
import datetime
import json
class CDIPRoseData:
def __init__(self, station_id, start_date, delta_days, *kwargs):
self.station_id = station_id
# Set up some defaults for the bin sizes
self.radial_bin_count = 16
self.height_bin_count = 10
self.period_bin_count = 10
self.height_max = 5 # meters
self.period_max = 24 # seconds
self.metric = True # change to False to convert to feet
if (self.metric):
self.length_convert_factor = 1.0
else:
self.length_convert_factor = 3.28084
# Validate infput formats
# start_date should be mm/dd/yyyy and delta_days should be int
if (historic):
url = f"http://thredds.cdip.ucsd.edu/thredds/dodsC/cdip/archive/{station_id}/{station_id}_historic.nc"
else:
url = f"https://thredds.cdip.ucsd.edu/thredds/dodsC/cdip/realtime/{station_id}_rt.nc"
self.nc = netCDF4.Dataset(url)
self.ncTime = self.nc.variables['waveTime'][:]
self.Fq = self.nc.variables['waveFrequency'] # Assign variable name - Wave Frequency
self.Tp = self.nc.variables['waveTp'] # Assign variable name - Peak Wave Period
self.Dp = self.nc.variables['waveDp'] # Assign variable name - Peak Wave Direction
self.Hs = self.nc.variables['waveHs'] # Assign variable name - Significant Wave Height
self.start_unix = int(datetime.datetime.strptime(start_date, '%m/%d/%Y').timestamp())
self.end_unix = self.start_unix + (delta_days * 86400)
self.start_index = self.find_index_nearest_time(self.start_unix)
self.end_index = self.find_index_nearest_time(self.end_unix)
self._set_radial_bins()
self._set_height_bins()
self._set_period_bins()
def _validate_input_date(start_date):
return True
def _set_radial_bins(self):
self.radial_bin_width = 360/self.radial_bin_count
self.radial_start = self.radial_bin_width/2
self.radial_start_angles = np.arange(self.radial_start, 360, self.radial_bin_width)
def _set_height_bins(self):
self.height_bin_width = self.height_max/self.height_bin_count
self.height_bin_starts = np.arange(0, self.height_max, self.height_bin_width)
def _set_period_bins(self):
self.period_bin_width = self.period_max/self.period_bin_count
self.period_bin_starts = np.arange(0, self.period_max, self.period_bin_width)
def find_index_nearest_time(self, unixtime):
#nearest = find_nearest(self.ncTime, unixtime) # Find the closest unix timestamp
#near_index = np.where(self.ncTime==nearest)[0][0] # Grab the index number of found date
near_index = (np.abs(self.ncTime - unixtime)).argmin()
return near_index
def find_radial_bin_number(self, angle):
return int((angle - self.radial_start) / self.radial_bin_width) - 1
def find_height_bin_number(self, height):
return int(height/self.height_bin_width) - 1
def find_period_bin_number(self, period):
return int(period/self.period_bin_width) -1
def get_data_source_start():
return 0
def format_rose_data(self, data):
output = [];
for i, row in enumerate(data):
bin = {}
bin['angle'] = self.radial_start_angles[i] + self.radial_bin_width/2
bin['start_angle'] = self.radial_start_angles[i]
bin['end_angle'] = self.radial_start_angles[i] + self.radial_bin_width
for j, entry in enumerate(row):
bin[j] = entry
output.append(bin)
return output
def get_wave_height_rose_data(self):
wave_directions = np.copy(self.Dp[self.start_index:self.end_index])
wave_heights = np.copy(self.Hs[self.start_index:self.end_index])
data = np.array([wave_directions, wave_heights])
rose_counts = np.zeros([self.radial_bin_count, self.height_bin_count], 'int')
for row in data.T:
radial_bin = self.find_radial_bin_number(row[0])
height_bin = self.find_height_bin_number(row[1])
rose_counts[radial_bin, height_bin] += 1
rose_data = rose_counts/np.sum(rose_counts)
return self.format_rose_data(rose_data)
def get_wave_period_rose_data(self):
wave_directions = np.copy(self.Dp[self.start_index:self.end_index])
wave_periods = np.copy(self.Tp[self.start_index:self.end_index])
data = np.array([wave_directions, wave_periods])
rose_counts = np.zeros([self.radial_bin_count, self.period_bin_count], 'int')
for row in data.T:
radial_bin = self.find_radial_bin_number(row[0])
period_bin = self.find_period_bin_number(row[1])
rose_counts[radial_bin, period_bin] += 1
rose_data = rose_counts/np.sum(rose_counts)
return self.format_rose_data(rose_data)
def get_height_and_period_rose_data(self):
output = {}
output['station_id'] = self.station_id
output['start_time'] = self.start_unix
output['end_time'] = self.end_unix
output['wave_height'] = self.get_wave_height_rose_data()
output['wave_period'] = self.get_wave_period_rose_data()
return json.dumps(output)
if __name__ == "__main__":
station_id = '198p1'
start_date = '11/01/2013'
delta_days = 3
rose_data = CDIPRoseData(station_id, start_date, delta_days)
#result = rose_data.get_height_and_period_rose_data()
result = rose_data.format()
print(result)