# coding=utf-8
from nom_client.nom_client import NoMClient
from nom_client.utils.file_util import read_file_into_array

import matplotlib.pyplot as plt


nom_client = NoMClient(project_name="GREET example",
                       default_server_id="ext_rest_server")


greet_model = nom_client.get_model('trajectory')

greet_model.set_params(orbitSpecificationCode=210, 
                       epoch="2020-01-01 00:00:00", 
                       startDate="2020-01-01 00:00:00",
                       missionDuration=30,
                       naifCode=599, 
                       circularAltitude=200000)


greet_result = nom_client.run_model(greet_model)
print(greet_result)


iso_15390_model = nom_client.get_model('iso_15390')
iso_15390_model.set_external_input(external_input_name="trajectory", external_input=greet_result)
iso_15390_results = nom_client.run_model(iso_15390_model)
print(iso_15390_results)

print(iso_15390_results["particle_spectrum"]["integral_flux"].values)

print(greet_result.file_outputs.keys())

trajectory = greet_result.get_model_result_by_name(result_name="trajectory")

mjd = trajectory.get_variable_data(variable_name="mjd")
altitude = trajectory.get_variable_data(variable_name="altitude")

plt.semilogy(mjd.T, altitude.T, linewidth=2.0, markersize=12.0, label='Altitude')

plt.title('Altitude', fontsize=18)
plt.grid(True)
plt.xlabel("mjd")
plt.ylabel("Altitude (km)")
plt.show()