running = 1
dt = 0.01 # [seconds]
T = 0.2 # duration of acceleration
# Initialize Target Point Mass ###
random_index = np.random.randint(data.shape[0] - n_test, data.shape[0])
t_position0 = dat[random_index, [4, 5]]
t_p0 = np.array([0.0])
t_v0 = np.array([0.0])
t_a0 = np.array([0.0])
t_angle0 = np.array([dat[random_index, 3]])
t_angle_v0 = np.array([0.0])
t_angle_a0 = np.array([0.0])
target = PointMass(t_position0, t_p0, t_v0, t_a0, t_angle0, t_angle_v0, t_angle_a0)
target.color = 255, 0, 0
ntrials = 1
for itrial in range(ntrials):
t = 0.0 # current time
### Initialize Point Mass ###
position0 = np.array([50.0, 50.0])
p0 = np.array([0.0])
v0 = np.array([0.0])
a0 = np.array([0.0])
angle0 = np.array([dat[random_index, 0]])
angle_v0 = np.array([0.0])
angle_a0 = np.array([0.0])
pm = PointMass(position0, p0, v0, a0, angle0, angle_v0, angle_a0)
dt = 0.01 # [seconds]
T = .2 # duration of acceleration
# Initialize Target Point Mass ###
random_index = np.random.randint(data.shape[0] - n_test, data.shape[0])
t_position0 = dat[random_index, [4, 5]]
t_p0 = np.array([0.])
t_v0 = np.array([0.])
t_a0 = np.array([0.])
t_angle0 = np.array([dat[random_index, 3]])
t_angle_v0 = np.array([0.])
t_angle_a0 = np.array([0.])
target = PointMass(t_position0, t_p0, t_v0, t_a0, t_angle0, t_angle_v0,
t_angle_a0)
target.color = 255, 0, 0
ntrials = 1
for itrial in range(ntrials):
t = 0. # current time
### Initialize Point Mass ###
position0 = np.array([50., 50.])
p0 = np.array([0.])
v0 = np.array([0.])
a0 = np.array([0.])
angle0 = np.array([dat[random_index, 0]])
angle_v0 = np.array([0.])
angle_a0 = np.array([0.])
pm = PointMass(position0, p0, v0, a0, angle0, angle_v0, angle_a0)
