diff --git a/wetb/hawc2/st_file.py b/wetb/hawc2/st_file.py index 135ff4aca7d5e93a286aa13fb42673bbf972a9b5..4e4d53cb67f41d2f509437a92f207824874338fe 100644 --- a/wetb/hawc2/st_file.py +++ b/wetb/hawc2/st_file.py @@ -42,16 +42,16 @@ class StFile(object): The autogenerated methods have the following structure - def xxx(radius=None, mset_nr=1, set_nr=1): + def xxx(radius=None, mset=1, set=1): Parameters: ----------- radius : int, float, array_like or None, optional Radius/radii of interest\n If int, float or array_like: values are interpolated to requested radius/radii If None (default): Values of all radii specified in st file returned - mset_nr : int, optional + mset : int, optional Main set number - set_nr : int, optional + set : int, optional Sub set number @@ -60,9 +60,9 @@ class StFile(object): >>> stfile = StFile(r"tests/test_files/DTU_10MW_RWT_Blade_st.dat") >>> print (stfile.m()) # Interpolated mass at radius 36 [ 1189.51054664 1191.64291781 1202.76694262 ... 15.42438683] - >>> print (st.E(radius=36, mset_nr=1, set_nr=1)) # Elasticity interpolated to radius 36m + >>> print (st.E(radius=36, mset=1, set=1)) # Elasticity interpolated to radius 36m 8722924514.652649 - >>> print (st.E(radius=36, mset_nr=1, set_nr=2)) # Same for stiff blade set + >>> print (st.E(radius=36, mset=1, set=2)) # Same for stiff blade set 8.722924514652648e+17 """ def __init__(self, filename): @@ -83,7 +83,7 @@ class StFile(object): self.main_data_sets[mset_nr] = set_data_dict for i, name in enumerate("r m x_cg y_cg ri_x ri_y x_sh y_sh E G I_x I_y I_p k_x k_y A pitch x_e y_e".split()): - setattr(self, name, lambda radius=None, mset_nr=1, set_nr=1, column=i: self._value(radius, column, mset_nr, set_nr)) + setattr(self, name, lambda radius=None, mset=1, set=1, column=i: self._value(radius, column, mset, set)) def _value(self, radius, column, mset_nr=1, set_nr=1): st_data = self.main_data_sets[mset_nr][set_nr] @@ -91,8 +91,8 @@ class StFile(object): radius = self.radius(None, mset_nr, set_nr) return np.interp(radius, st_data[:, 0], st_data[:, column]) - def radius(self, radius=None, mset_nr=1, set_nr=1): - r = self.main_data_sets[mset_nr][set_nr][:, 0] + def radius(self, radius=None, mset=1, set=1): + r = self.main_data_sets[mset][set][:, 0] if radius is None: return r return r[np.argmin(np.abs(r - radius))] @@ -102,8 +102,8 @@ class StFile(object): if __name__ == "__main__": st = StFile(r"C:\mmpe\HAWC2\models\DTU10MWRef6.0\data/DTU_10MW_RWT_Blade_st.dat") print (st.m()) - print (st.E(radius=36, mset_nr=1, set_nr=1)) # Elastic blade - print (st.E(radius=36, mset_nr=1, set_nr=2)) # stiff blade + print (st.E(radius=36, mset=1, set=1)) # Elastic blade + print (st.E(radius=36, mset=1, set=2)) # stiff blade #print (st.radius()) xyz = np.array([st.x_e(), st.y_e(), st.r()]).T[:40] n = 2