WooeyScript.py

__author__ = 'cholloway'

import argparse
from OdomProcessor import OdomProcessor

parser = argparse.ArgumentParser(description='Process odometries in bag file.', argument_default=argparse.SUPPRESS)
parser.add_argument('--data_topics', type=list, nargs='+', help='The odometry topic')
parser.add_argument('--labels', type=list, nargs='+', help='The topic labels for the graphs')
parser.add_argument('--foldername', type=list, nargs='?', help='The foldername containing the bags')
parser.add_argument('--filename', type=str, nargs='?', help='The filename of the bag')
parser.add_argument('--vicon_topic', type=str, nargs='?', help='The vicon topic')
parser.add_argument('--results_file', type=str, nargs='?', help="The results file (used for generating comparison graphs)")
parser.add_argument('--heatmap', type=bool, nargs='?', help="Generate heatmaps (increases runtime by a significant margin)")
parser.add_argument('--sql', type=bool, nargs='?', help="Database to stick results into.")
parser.add_argument('--from_csv', type=bool, nargs='?', help="Use CSVs and not bags")
parser.add_argument('--output_foldername', type=str, nargs='?', help="The foldername where graphs and yaml results will be saved")
args = parser.parse_args()

if __name__=="__main__":
    op = OdomProcessor(**vars(args))
    op.run_files()