From 7c0dd646c8f7c971c25c135fcc40305a40a08377 Mon Sep 17 00:00:00 2001 From: Markus Mirz <mmirz@eonerc.rwth-aachen.de> Date: Mon, 18 Sep 2017 14:51:46 +0200 Subject: [PATCH] added dpsim read function with column names --- dataprocessing/readtools.py | 56 +++++++++++++++++++++++++------------ 1 file changed, 38 insertions(+), 18 deletions(-) diff --git a/dataprocessing/readtools.py b/dataprocessing/readtools.py index cea857a..6bd5a05 100644 --- a/dataprocessing/readtools.py +++ b/dataprocessing/readtools.py @@ -2,40 +2,60 @@ import numpy as np import pandas as pd from .timeseries import * -def read_time_series_Modelica(filename, time_series_names=None): +def read_timeseries_Modelica(filename, timeseries_names=None): from modelicares import SimRes sim = SimRes(filename) - times_series = [] - if time_series_names is None: + timeseries_list = [] + if timeseries_names is None: # No trajectory names specified, thus read in all print('TBD') else: # Read in specified time series - for name in time_series_names: - times_series.append(TimeSeries(name, sim(name).times(), sim(name).values())) - return times_series + for name in timeseries_names: + timeseries_list.append(TimeSeries(name, sim(name).times(), sim(name).values())) + return timeseries_list + +def read_timeseries_PLECS(filename, timeseries_names=None): + pd_df = pd.read_csv(filename) + timeseries_list = [] + if timeseries_names is None: + # No trajectory names specified, thus read in all + timeseries_names = list(pd_df.columns.values) + timeseries_names.remove('Time') + for name in timeseries_names: + timeseries_list.append(TimeSeries(name, pd_df['Time'].values, pd_df[name].values)) + else: + # Read in specified time series + for name in timeseries_names: + timeseries_list.append(TimeSeries(name, pd_df['Time'].values, pd_df[name].values)) + return timeseries_list -def read_time_series_PLECS(filename, time_series_names=None): +def read_timeseries_DPsim(filename, timeseries_names=None): pd_df = pd.read_csv(filename) - times_series = [] - if time_series_names is None: + timeseries_list = [] + + if timeseries_names is None: # No trajectory names specified, thus read in all - time_series_names = list(pd_df.columns.values) - time_series_names.remove('Time') - for name in time_series_names: - times_series.append(TimeSeries(name, pd_df['Time'].values, pd_df[name].values)) + timeseries_names = list(pd_df.columns.values) + timeseries_names.remove('Time') + for name in timeseries_names: + timeseries_list.append(TimeSeries(name, pd_df['Time'].values, pd_df[name].values)) else: # Read in specified time series - for name in time_series_names: - times_series.append(TimeSeries(name, pd_df['Time'].values, pd_df[name].values)) - return times_series + print('no column names specified yet') + + print('DPsim results file length:') + print(len(timeseries_list)) + for result in timeseries_list: + print(result.name) + return timeseries_list -def read_time_series_DPsim(filename, time_series_names=None): +def read_timeseries_DPsim_node_values(filename, timeseries_names=None): pd_df = pd.read_csv(filename, header=None) timeseries_list = [] - if time_series_names is None: + if timeseries_names is None: # No trajectory names specified, thus read in all column_names = list(pd_df.columns.values) column_names.remove(0) -- GitLab