import math
from trick.unit_test import *

if trick.mc_get_slave_id() == 0:
    trick_utest.unit_tests.enable();
    trick_utest.unit_tests.set_file_name( os.getenv("TRICK_HOME") + "/trick_test/SIM_target_master.xml" )

target_integloop.getIntegrator(trick.Runge_Kutta_Fehlberg_45, 42)

# Set up the Monte Carlo runs. 
#trick.mc_set_dry_run(1)
trick.mc_set_enabled(1)
trick.mc_set_num_runs(200)
var0 = trick.MonteVarCalculated("target.target_state.velocity[0]")
var1 = trick.MonteVarCalculated("target.target_state.velocity[1]","m/s")
var2 = trick.MonteVarCalculated("target.target_state.velocity[2]")
trick_mc.mc.add_variable(var0)
trick_mc.mc.add_variable(var1)
trick_mc.mc.add_variable(var2)

slave0 = trick.MonteSlave("localhost")
#slave0.set_remote_shell(trick.TRICK_SSH)
#slave0.set_sim_path("${HOME}/trick_sims/SIM_target")
slave0.set_S_main_name("./T_main_${TRICK_HOST_CPU}_test.exe")
trick_mc.mc.add_slave(slave0)


# Set the iteration limit. */
iterate.iterate_data.iter_max = 12;

# Set the initial body position. */
target.target_state.position[0] = trick.attach_units( "km", -7378.0)
target.target_state.position[1] = trick.attach_units( "km", 0.0)
target.target_state.position[2] = trick.attach_units( "km", 0.0)

# Compute the initial circular velocity from position. */
#target.state.velocity[0] {km/s} = 0.0, 7.350214, 0.0;
pi = 3.14159265358979323846264338328;
mu = float(target.target_planet.mu);
r_mag = math.sqrt(   (float(target.target_state.position[0]) * float(target.target_state.position[0]))
                   + (float(target.target_state.position[1]) * float(target.target_state.position[1]))
                   + (float(target.target_state.position[2]) * float(target.target_state.position[2])) );

r_3 = r_mag * r_mag * r_mag;
target.target_state.velocity[0] = 0.0;
target.target_state.velocity[1] = math.sqrt( mu / r_mag );
target.target_state.velocity[2] = 0.0;

# Set the desired terminal position. */
target.target_data.position[0] = trick.attach_units( "km", 2*7378.0)
target.target_data.position[1] = trick.attach_units( "km", 0.0)
target.target_data.position[2] = trick.attach_units( "km", 0.0)

# Compute the stop time. */
trick.stop( pi * math.sqrt( r_3 / mu ) )