@@ -127,7 +127,7 @@ def __init__(self, fs, db_list, signals=[], days_offset=None, verbose=True):
127
127
self .verbose = verbose
128
128
return
129
129
130
- def extract_phys (self , df_raw , tp_type = None ):
130
+ def extract_phys (self , df_raw ):
131
131
"""Given df of raw data and list of decoding databases, create new def with
132
132
physical values (no duplicate signals and optionally filtered/rebaselined)
133
133
"""
@@ -138,15 +138,12 @@ def extract_phys(self, df_raw, tp_type=None):
138
138
for db in self .db_list :
139
139
df_decoder = can_decoder .DataFrameDecoder (db )
140
140
141
- if tp_type != None :
142
- df_phys_tp = pd .DataFrame ()
143
- for length , group in df_raw .groupby ("DataLength" ):
144
- df_phys_group = df_decoder .decode_frame (group )
145
- df_phys_tp = df_phys_tp .append (df_phys_group )
141
+ df_phys_temp = pd .DataFrame ()
142
+ for length , group in df_raw .groupby ("DataLength" ):
143
+ df_phys_group = df_decoder .decode_frame (group )
144
+ df_phys_temp = df_phys_temp .append (df_phys_group )
146
145
147
- df_phys = df_phys .append (df_phys_tp .sort_index ())
148
- else :
149
- df_phys = df_phys .append (df_decoder .decode_frame (df_raw ))
146
+ df_phys = df_phys .append (df_phys_temp .sort_index ())
150
147
151
148
# remove duplicates in case multiple DBC files contain identical signals
152
149
df_phys ["datetime" ] = df_phys .index
@@ -155,20 +152,21 @@ def extract_phys(self, df_raw, tp_type=None):
155
152
156
153
# optionally filter and rebaseline the data
157
154
df_phys = self .filter_signals (df_phys )
158
- df_phys = self .rebaseline_data (df_phys )
155
+
156
+ if not df_phys .empty and type (self .days_offset ) == int :
157
+ df_phys = self .rebaseline_data (df_phys )
159
158
160
159
return df_phys
161
160
162
161
def rebaseline_data (self , df_phys ):
163
162
"""Given a df of physical values, this offsets the timestamp
164
163
to be equal to today, minus a given number of days.
165
164
"""
166
- if not df_phys .empty and type (self .days_offset ) == int :
167
- from datetime import datetime , timezone
168
- import pandas as pd
165
+ from datetime import datetime , timezone
166
+ import pandas as pd
169
167
170
- delta_days = (datetime .now (timezone .utc ) - df_phys .index .min ()).days - self .days_offset
171
- df_phys .index = df_phys .index + pd .Timedelta (delta_days , "day" )
168
+ delta_days = (datetime .now (timezone .utc ) - df_phys .index .min ()).days - self .days_offset
169
+ df_phys .index = df_phys .index + pd .Timedelta (delta_days , "day" )
172
170
173
171
return df_phys
174
172
0 commit comments