def main(): root = Tkinter.Tk() root.withdraw() file_name = tkFileDialog.askopenfilename(filetypes = (("All files", "*"), ("Template files", "*.type"))) # file_name = "Team 174 Robot Logs - RobotLog_20150210_054531870_log.csv" converted_dict = load_csv_file(file_name, {'Date and Time': parse_datetime}) image_dir = "images/" if not os.path.exists(image_dir): os.mkdir(image_dir) plt.figure(1) plot_driver_joysticks(converted_dict) plt.savefig(image_dir + "DriverSticks.png") plt.figure(2) plot_encoders(converted_dict) plt.savefig(image_dir + "Encoders.png") plt.figure(3) plot_scale_tilt(converted_dict) plt.savefig(image_dir + "Scaling.png") plt.figure(4) plot_extension(converted_dict) plt.savefig(image_dir + "Extension.png") plt.figure(5) plot_harvester(converted_dict) plt.savefig(image_dir + "Harvester.png") plt.show()
def create_plots(filepath, filename): converted_dict = load_csv_file(filepath) if 'Segment Number' not in converted_dict: return segment = converted_dict['Segment Number'] dt = converted_dict['Time'] position = converted_dict['Position'] velocity = converted_dict['Velocity'] acceleration = converted_dict['Acceleration'] time = [] sum_time = 0 for i in range(len(dt)): time.append(sum_time) sum_time += dt[i] plt.figure(filename) plt.subplot(3, 1, 1) plt.title("Position") plt.scatter(time, position, c=segment) plt.subplot(3, 1, 2) plt.title("Velocity") plt.scatter(time, velocity, c=segment) plt.subplot(3, 1, 3) plt.title("Acceleration") plt.scatter(time, acceleration, c=segment)
def create_plots(filepath, filename): converted_dict = load_csv_file(filepath) directions = ["Left", "Right"] if "LeftPos" not in converted_dict: return dt = converted_dict['LeftDt'] position = converted_dict['LeftPos'] velocity = converted_dict['LeftVelocity'] acceleration = converted_dict['LeftAcceleration'] time = [] sum_time = 0 for i in range(len(dt)): time.append(sum_time) sum_time += dt[i] num_rows = 5 num_cols = 2 for i, direction in enumerate(directions): position = converted_dict[direction + 'Pos'] velocity = converted_dict[direction + 'Velocity'] acceleration = converted_dict[direction + 'Acceleration'] plt.figure(filename) plt.subplot(num_rows, num_cols, i + 1) plt.title(direction + " Position") plt.scatter(time, position) plt.subplot(num_rows, num_cols, i + 3) plt.title(direction + " Velocity") plt.scatter(time, velocity) plt.subplot(num_rows, num_cols, i + 5) plt.title(direction + " Acceleration") plt.scatter(time, acceleration) heading = converted_dict['LeftHeading'] left_x = converted_dict['LeftX'] left_y = converted_dict['LeftY'] right_x = converted_dict['RightX'] right_y = converted_dict['RightY'] plt.subplot(num_rows, num_cols, i + 7) plt.title("Heading") plt.scatter(time, heading) plt.subplot(num_rows, num_cols, i + 9) plt.title("Position") plt.scatter(left_x, left_y, c=time) plt.scatter(right_x, right_y, c=time)