Skip to content
Snippets Groups Projects
Commit 92f985d8 authored by Kien Le's avatar Kien Le
Browse files

Revert changes to get_trimTPSData

parent 0d66c926
No related branches found
No related tags found
1 merge request!33Add tests and refactor trim_downsample_WFChan
...@@ -760,10 +760,35 @@ def get_trimTPSData(chan, startTm, endTm, every_day_5_min_list): ...@@ -760,10 +760,35 @@ def get_trimTPSData(chan, startTm, endTm, every_day_5_min_list):
# preset all 0 for all 5 minutes for each day # preset all 0 for all 5 minutes for each day
tpsData = np.zeros((len(every_day_5_min_list), const.NO_5M_DAY)) tpsData = np.zeros((len(every_day_5_min_list), const.NO_5M_DAY))
zTracesInfo = trim_waveform_data(chan, startTm, endTm) # zoom in to the given range
if not zTracesInfo: chan['startIdx'] = 0
chan['endIdx'] = len(chan['tracesInfo'])
if ((startTm > chan['tracesInfo'][-1]['endTmEpoch']) or
(endTm < chan['tracesInfo'][0]['startTmEpoch'])):
return False return False
indexes = [index for index, tr in enumerate(chan['tracesInfo'])
if tr['startTmEpoch'] > startTm]
if indexes != []:
chan['startIdx'] = indexes[0]
if chan['startIdx'] > 0:
chan['startIdx'] -= 1 # startTm in middle of trace
else:
chan['startIdx'] = 0
indexes = [idx for (idx, tr) in enumerate(chan['tracesInfo'])
if tr['endTmEpoch'] <= endTm]
if indexes != []:
chan['endIdx'] = indexes[-1]
if chan['endIdx'] < len(chan['tracesInfo']) - 1:
chan['endIdx'] += 1 # endTm in middle of trace
else:
chan['endIdx'] = 0
chan['endIdx'] += 1 # a[x:y+1] = [a[x], ...a[y]
zTracesInfo = chan['tracesInfo'][chan['startIdx']:chan['endIdx']]
spr = chan['samplerate'] spr = chan['samplerate']
chan['tps_data'] = [] chan['tps_data'] = []
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment