# coding=utf-8
import os
from pathlib import Path

import nom_client as nc
from nom_client.nom_client import NoMClient

print( 'NoM Client Version: ' + nc.__version__)

cwd = os.getcwd()
homeDir = str(Path.home())

conf = homeDir + '/.nom/nom.cfg'
model_dir = cwd + '/..'

nom_client = NoMClient("eqflux-example", client_configuration=conf, 
                        default_server_id="local_server", 
                        debug_output=False)

trpModel='ae8ap8'
sepModel='esp-psychic-total-fluence'

server_id = 'dev'
models = {}
results = {}

m = 'sapre'
print(m + "...", end="")
models[m] = nom_client.get_model( m, server_id=server_id )
models[m].set_params(
      planet=3,
      missionDurationUnits=0,
      missionDuration=15,
      orbitType='GEN', altitudeSpecification=2, altitude=1000, inclination=60.0, 
      trajectoryOrbits=1, trajectoryDuration=1,
      trajectoryFlag=1
      )
results[m] = nom_client.run_model(model=models[m], server_id=server_id)
print("done")

m = trpModel
print(m + "...", end="")
models[m] = nom_client.get_model( m, server_id=server_id )
models[m].set_params( )
models[m].set_external_input( 
      external_input_name='trajectory',
      external_input=results['sapre']
)
results[m] = nom_client.run_model(model=models[m], server_id=server_id)
print("done")

m = sepModel
print(m + "...", end="")
models[m] = nom_client.get_model( m, server_id=server_id )
models[m].set_params( )
models[m].set_external_input( 
      external_input_name='trajectory',
      external_input=results['sapre']
)
results[m] = nom_client.run_model(model=models[m], server_id=server_id)
print("done")

print("")
print(".... Running EQFLUX ....")

m = 'eqflux'
print(m + "...", end="")
local_model = nom_client.get_model( m, server_id='local_server') 

local_model.set_params( 
      cellType=2,
      multipleJunctionCell=12,
      coverglassThickness=100,
      coverglassThicknessUnit=3
)
local_model.set_external_input( 
                        external_input_name='trappedParticleSpectrum', 
                        external_input=results[trpModel]
)
local_model.set_external_input( 
                        external_input_name='solarParticleSpectrum', 
                        external_input=results[sepModel]
)
#                        
local_results = nom_client.run_model( model=local_model)

print("done")

print(local_results)
outputs = local_results['summary_equivalent_fluences_pmax']
print(outputs)


def list_eqflux( results, title="", param='pmax'):
        op = 'summary_equivalent_fluences_' + param

        thick = results[op].get_variable_data('depth').squeeze()
        source = results[op].get_variable_data('population').squeeze()
        tid = results[op].get_variable_data('fluence').squeeze()
        sc_name = results[source][op].get_metadata()
        print(sc_name)
        print(title + ', Solar Cell, ' + param.capitalize())
        print('{0:8>}'.format('Z[micron]'), end='')
        for j, s in enumerate(source):
            print(', {0:8>}'.format(s), end='')
        print()

        for i, e in enumerate(thick):
            print('{0:8.2f}'.format(e), end='')
            for j, s in enumerate(source):
                print(', {0:8.2e}'.format(tid[j][i]), end='')
            print()

        print()

print("-----------------")
for x in ['pmax','voc','isc']:
   list_eqflux( local_results, param=x)
   print()