def __init__(self, distance_to_sun=1, mass=1, initial_angle=0, sun_mass=1):
self.distance_to_sun = distance_to_sun
self.mass = mass
self.initial_angle = initial_angle
self.angle = initial_angle
self.velocity_magnitude = math.sqrt(sun_mass / self.distance_to_sun)
self.angular_velocity = self.velocity_magnitude / self.distance_to_sun
self.peroid = 2 * math.pi / self.angular_velocity
self.position = Vector()
self.position.x = self.distance_to_sun * math.cos(self.angle)
self.position.y = self.distance_to_sun * math.sin(self.angle)
self.velocity = Vector() # linear velocity
def calc_overall_force(self) -> Vector:
force = Vector()
for planet in self.planets:
distance = self.satellite.position.distance(planet.position)
direction = self.satellite.position.direction(planet.position)
direction.normalize()
this_planet_force_magnitude = planet.mass * G / (distance *
distance)
force += (direction * this_planet_force_magnitude)
# sun force
distance = self.satellite.position.distance(Vector(0, 0))
direction = self.satellite.position.direction(Vector(0, 0))
direction.normalize()
sun_force_magnitude = (self.sun_mass * G) / (distance * distance)
force += (direction * sun_force_magnitude)
return force
def update_stats(self, dt):
self.fly_time += dt
distance_to_destination = self.position.distance(
self.destination_planet.position)
if distance_to_destination < self.closest_encounter:
self.closest_encounter = distance_to_destination
self.closest_encounter_time = self.fly_time + self.cpu.time
self.closest_encounter_position = Vector(self.position.x,
self.position.y)
def __init__(self, start_planet: 'Planet', cpu: 'Cpu',
destination_planet: 'Planet'):
self.start_planet = start_planet
self.destination_planet = destination_planet
self.cpu = cpu
self.velocity = Vector()
self.position = Vector()
self.mass = 1 # mass can be small
self.force = Vector()
self.cut_down = self.cpu.time
self.fly = False
self.position = self.start_planet.position + Vector(
0, self.start_planet.mass)
self.velocity = self.start_planet.velocity
self.fly_time = 0
self.closest_encounter = 10e10
self.closest_encounter_time = 10e10
self.path = []
def step(self, dt: float):
# angular_velocity in constant over time
self.angle += (self.angular_velocity * dt) % self.peroid
self.position.x = self.distance_to_sun * math.cos(self.angle)
self.position.y = self.distance_to_sun * math.sin(self.angle)
velocity_angle = self.angle + math.pi / 2
velocity_normal_vector = Vector(math.cos(velocity_angle),
math.sin(velocity_angle))
self.velocity = velocity_normal_vector * self.velocity_magnitude
def step(self, dt: float):
self.path.append({
"position": Vector(self.position.x, self.position.y),
"force": Vector(self.force.x, self.force.y)
})
if not self.fly:
self.position = self.start_planet.position + Vector(
0, self.start_planet.mass)
self.velocity = self.start_planet.velocity
self.cut_down -= dt
if self.cut_down > 0.0:
return
else:
self.launch()
dv = self.force * dt
self.velocity += dv
dp = self.velocity * dt
self.position += dp
self.update_stats(dt)
class Satellite:
def __init__(self, start_planet: 'Planet', cpu: 'Cpu',
destination_planet: 'Planet'):
self.start_planet = start_planet
self.destination_planet = destination_planet
self.cpu = cpu
self.velocity = Vector()
self.position = Vector()
self.mass = 1 # mass can be small
self.force = Vector()
self.cut_down = self.cpu.time
self.fly = False
self.position = self.start_planet.position + Vector(
0, self.start_planet.mass)
self.velocity = self.start_planet.velocity
self.fly_time = 0
self.closest_encounter = 10e10
self.closest_encounter_time = 10e10
self.path = []
def launch(self):
self.fly = True
self.velocity = self.velocity + self.cpu.get_velocity_vector()
def set_force(self, force: Vector):
self.force = force
def step(self, dt: float):
self.path.append({
"position": Vector(self.position.x, self.position.y),
"force": Vector(self.force.x, self.force.y)
})
if not self.fly:
self.position = self.start_planet.position + Vector(
0, self.start_planet.mass)
self.velocity = self.start_planet.velocity
self.cut_down -= dt
if self.cut_down > 0.0:
return
else:
self.launch()
dv = self.force * dt
self.velocity += dv
dp = self.velocity * dt
self.position += dp
self.update_stats(dt)
def update_stats(self, dt):
self.fly_time += dt
distance_to_destination = self.position.distance(
self.destination_planet.position)
if distance_to_destination < self.closest_encounter:
self.closest_encounter = distance_to_destination
self.closest_encounter_time = self.fly_time + self.cpu.time
self.closest_encounter_position = Vector(self.position.x,
self.position.y)
def get_score(self):
return self.closest_encounter + abs(self.cpu.speed)
def toJSON(self):
l = []
for point in self.path:
l.append({
'position': point['position'].toDict(),
'force': point['force'].toDict()
})
return json.dumps(l)
parser = argparse.ArgumentParser(description='Simulation visualization')
parser.add_argument("-c",
help="config file name",
type=str,
dest="config",
required=True)
parser.add_argument("--speed", type=float, required=True)
parser.add_argument("--angle", type=float, required=True)
parser.add_argument("--time", type=float, required=True)
args = parser.parse_args()
pygame.init()
screen_width = 800
screen_height = 600
vector_mid = Vector(float(screen_width / 2.0), float(screen_height / 2.0))
screen = pygame.display.set_mode((screen_width, screen_height),
HWSURFACE | DOUBLEBUF | RESIZABLE)
clock = pygame.time.Clock()
done = False
(planets, start_planet, destination_planet,
sun_mass) = Simulation.load_from_file(args.config)
time_factor = 0.001
cpu = Cpu(args.speed, args.angle, args.time)
satellite = Satellite(start_planet, cpu, destination_planet)
simulation = Simulation(planets=planets,
def get_velocity_vector(self):
return Vector(math.cos(self.angle), math.sin(self.angle)) * self.speed