27 lines
779 B
Python
27 lines
779 B
Python
import numpy as np
|
|
|
|
|
|
def ball_init(random=False, context_range=None):
|
|
if random:
|
|
dx = np.random.uniform(-0.1, 0.1) # TODO: clarify these numbers?
|
|
dy = np.random.uniform(-0.1, 0.1) # TODO: clarify these numbers?
|
|
dz = np.random.uniform(-0.1, 0.1) # TODO: clarify these numbers?
|
|
|
|
v_x = np.random.uniform(1.7, 1.8)
|
|
v_y = np.random.uniform(0.7, 0.8)
|
|
v_z = np.random.uniform(0.1, 0.2)
|
|
else:
|
|
dx = 1
|
|
dy = 0
|
|
dz = 0.05
|
|
|
|
v_x = 2.5
|
|
v_y = 2
|
|
v_z = 0.5
|
|
|
|
initial_x = 0 + dx - 1.2
|
|
initial_y = -0.2 + dy - 0.6
|
|
initial_z = 0.3 + dz + 1.5
|
|
initial_ball_state = np.array([initial_x, initial_y, initial_z, v_x, v_y, v_z])
|
|
return initial_ball_state
|