ttc-data/trackExample.py at main · wazzu-racing/ttc-data · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
from matplotlib import pyplot as plt
import numpy as np
import trackModel


def plot_the_track():

    for i in range(len(points_x)):
        match i % 5:
            case 0: col = 'red'
            case 1: col = 'blue'
            case 2: col = 'green'
            case 3: col = 'black'
            case 4: col = 'violet'
        plt.plot(points_x[i], points_y[i], marker='o', color=col)
        plt.plot(points_x2[i], points_y2[i], marker='o', color=col)
    track.plot_track()
    plt.axis('equal')
    plt.show()

def plot_the_single_point_sim():
    t = track.plot_sim('single point')
    print(f'single point sim time: {t}')
    plt.xlabel('Distance, in')
    plt.ylabel('Velocity, in/s')


def plot_the_four_wheel_sim():
    t = track.plot_sim('four wheel')
    print(f'4 wheel sim time: {t}')
    plt.xlabel('Distance, in')
    plt.ylabel('Velocity, in/s')
    plt.legend()
    plt.show()


car_model = 'C:/Users/maxwe/Downloads/FSAE/2023-2024 Car/Repo/62_car_model.pkl'


points_x = [0,    246.3, 600.2, 668.7, 454.3, 676.5, 284.2, -26,    -178.2, -260.9, -170, -444.4, -554.68, -370]
points_y = [1000, 606.7, 548.9, 412,   206.1, -70,  -192.3, -242.5, -348.9, -92.4,  182.2, 374.7,  460.4,  823.4]

points_x2 = [0,   156.3, 400.2, 868.7, 584.3, 526.5, 284.2,  100,   -378.2, -460.9,  70,  -344.4, -454.68, -520]
points_y2 = [800, 556.7, 548.9, 462,   136.1, -20,   8.3, -342.5, -448.9, -92.4,  232.2, 574.7,  560.4,   973.4]

#points_x = [-700, -600.3, -300.2, -100.7, 200.3, 300.5, 284.2, -26, -178.2, -460.9, -670, -444.4, -504.68, -870]
#points_y = [1000, 832.7, 548.9, 700, 500, -70, -292.3, -242.5, -548.9, -292.4, 182.2, 374.7, 460.4, 623.4]


track = trackModel.track(points_x, points_y, points_x2, points_y2, car_model, 500)

plot_the_track()
track.adjust_course(80)
plot_the_track()
plot_the_single_point_sim()
plot_the_four_wheel_sim()
# y = track.sim_prop('four wheel')
# print(y)