Newer
Older
from __future__ import print_function
from __future__ import unicode_literals
from __future__ import division
from __future__ import absolute_import
from builtins import range
from builtins import int
from future import standard_library
standard_library.install_aliases()
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
import numpy as np
class AEFile(object):
"""Read HAWC2 AE (aerodynamic blade layout) file
examples
--------
>>> aefile = AEFile(r"tests/test_files/NREL_5MW_ae.txt")
>>> print (aefile.thickness(36)) # Interpolated thickness at radius 36
23.78048780487805
>>> print (aefile.chord(36)) # Interpolated chord at radius 36
3.673
>>> print (aefile.pc_set_nr(36)) # pc set number at radius 36
1
"""
def __init__(self, filename):
with open (filename) as fid:
lines = fid.readlines()
nsets = int(lines[0].split()[0])
lptr = 1
self.ae_sets = {}
for _ in range(1, nsets + 1):
for _ in range(nsets):
set_nr, n_rows = [int(v) for v in lines[lptr ].split()[:2]]
lptr += 1
data = np.array([[float(v) for v in l.split()[:4]] for l in lines[lptr:lptr + n_rows]])
self.ae_sets[set_nr] = data
lptr += n_rows
def _value(self, radius, column, set_nr=1):
ae_data = self.ae_sets[set_nr]
return np.interp(radius, ae_data[:, 0], ae_data[:, column])
def chord(self, radius, set_nr=1):
return self._value(radius, 1, set_nr)
def thickness(self, radius, set_nr=1):
return self._value(radius, 2, set_nr)
def pc_set_nr(self, radius, set_nr=1):
ae_data = self.ae_sets[set_nr]
index = np.searchsorted(ae_data[:, 0], radius)
index = max(1, index)
setnr1, setnr2 = ae_data[index - 1:index + 1, 3]
if setnr1 != setnr2:
raise NotImplementedError
return setnr1
if __name__ == "__main__":
ae = AEFile(r"tests/test_files/NREL_5MW_ae.txt")
print (ae.thickness(36))
print (ae.chord(36))
print (ae.pc_set_nr(36))