From edcc714b08797a376724fd8bf346b9f817b8f771 Mon Sep 17 00:00:00 2001
From: "Mads M. Pedersen" <mmpe@dtu.dk>
Date: Wed, 24 May 2017 13:10:13 +0200
Subject: [PATCH] handle at_time_file with rotor name
---
wetb/hawc2/at_time_file.py | 6 ++++--
1 file changed, 4 insertions(+), 2 deletions(-)
diff --git a/wetb/hawc2/at_time_file.py b/wetb/hawc2/at_time_file.py
index 9dbc444..3819ec2 100644
--- a/wetb/hawc2/at_time_file.py
+++ b/wetb/hawc2/at_time_file.py
@@ -49,11 +49,13 @@ class AtTimeFile(object):
self.blade_radius = bladetip_radius
with open(filename, encoding='utf-8') as fid:
lines = fid.readlines()
- self.attribute_names = lines[2].lower().replace("#", "").split()
- data = np.array([[float(l) for l in lines[i].split() ] for i in range(3, len(lines))])
+ atttribute_name_line = [l.startswith("# Radius_s") for l in lines].index(True)
+ self.attribute_names = lines[atttribute_name_line].lower().replace("#", "").split()
+ data = np.array([[float(l) for l in lines[i].split() ] for i in range(atttribute_name_line+1, len(lines))])
self.data = data
def func_factory(column):
def values(l=None, curved_length=False):
+ assert curved_length==True or curved_length==False, "Curved length must be boolean, but is %s"%curved_length
if l is None:
return self.data[:, column]
else:
--
GitLab