def process_class(cls, logs_dir):
print("Processing %s" % cls)
logs=filter(lambda arg: cls in arg, os.listdir(logs_dir))
for log in logs:
signal=Signal.read_3d_csv(os.path.join(logs_dir, log))
chunk=walking_chunk
index, dist=use_dtw(signal, chunk)
print("Log file %s: DTW dist %f" % (log, dist))
def main():
parser=ArgumentParser()
parser.add_argument("--log", "-l", required=True)
parser.add_argument("--output", "-o", required=True)
args=parser.parse_args()
signal=Signal.read_3d_csv(args.log)
print("Length: %d" % len(signal.time))
time_diffs=np.diff(signal.time)
print("dt: %f..%f" % (min(time_diffs), max(time_diffs)))
cutter=LogCutter(signal)
working_figure=plt.gcf()
display_signal(signal, False)
#print(working_figure.number)
cutter.connect(working_figure)
plt.show()
subsignal=Signal.get_subsignal(signal, cutter.left_bound,
cutter.right_bound)
Signal.save_csv(subsignal, args.output)
#! /usr/bin/python3
from core.signal import Signal
import _ucrdtw as dtw
import os
walking_chunk=Signal.read_3d_csv("parse/chunks/walking_chunk.txt")
def process_class(cls, logs_dir):
print("Processing %s" % cls)
logs=filter(lambda arg: cls in arg, os.listdir(logs_dir))
for log in logs:
signal=Signal.read_3d_csv(os.path.join(logs_dir, log))
chunk=walking_chunk
index, dist=use_dtw(signal, chunk)
print("Log file %s: DTW dist %f" % (log, dist))
def use_dtw(signal, chunk):
signal_x=signal.axes_data[0]
chunk_x=chunk.axes_data[0]
index, dist = dtw.ucrdtw(signal_x, chunk_x, 0.05, False)
return index, dist
#! /usr/bin/python3
from argparse import ArgumentParser
from core.signal import Signal, display_signal
import numpy as np
import matplotlib
matplotlib.use('TkAgg')
import pylab as plt
parser=ArgumentParser()
parser.add_argument("--log", "-l", required=True)
parser.add_argument("--show", "-s", required=False, action="store_true")
args=parser.parse_args()
signal=Signal.read_3d_csv(args.log)
print("Length: %d" % len(signal.time))
time_diffs=np.diff(signal.time)
print("dt: %f..%f" % (min(time_diffs), max(time_diffs)))
if args.show:
display_signal(signal, True)
