# check for arguments
if len(sys.argv) < 3:
print('Please provide planet and capsule name.')
exit(1)
planet_name, capsule_name = sys.argv[1:3]
# init simulation environments
# basic = Sim(planet_name, capsule_name, 80000, ignore_buoyancy=True)
realistic = Sim(planet_name, capsule_name, 80000)
# basic_range = get_angle_range(basic, step=0.01)
realistic_range = get_angle_range(realistic, step=0.01)
# print(range_to_tex_table(basic_range, basic, step=0.2))
# print()
# print(range_to_tex_table(realistic_range, realistic, step=0.2))
# draw scheme
realistic.draw_scheme(4.15, 'planet', save_to_file='4_15_scheme.pdf')
# realistic.plot_height_over_time(5.8, save_to_file='5_8_height.pdf')
# realistic.plot_height_over_x(5.8, save_to_file='5_8_height_over_x.pdf')
# realistic.plot_velocity_over_time(5.8, save_to_file='5_8_velocity.pdf')
# plot range for initial velocities
# plot_range_for_velocity(8000, 12001, step=200, save_to_file='range_of_velocity.pdf')
# 4.56 je kratek odboj
# 4.13 se odbije dvakrat
# 4.7 je skor optimaln entry kot
