from nom_client.nom_client import NoMClient



nom_client = NoMClient("equipot example",
                       default_server_id="local_server",
                       debug_output=False)

# ---------------------------------------------------------------------------------------------------------

print("Running equipot")

equipot_model = nom_client.get_model('equipot')


base_dictat_settings = {
"environmentSpecification": 15,
    "structureAtomicNumber": 34
}

single_spec_settings = base_dictat_settings

single_spec_settings["energies"] = [4.00E-02, 7.00E-02, 1.00E-01, 2.50E-01, 5.00E-01, 7.50E-01, 1.00E+00, 1.50E+00,
                                    2.00E+00, 2.50E+00,
                                    3.00E+00, 3.50E+00, 4.00E+00, 4.50E+00, 5.00E+00, 5.50E+00, 6.00E+00, 6.50E+00,
                                    7.00E+00, 7.50E+00,
                                    8.00E+00, 8.50E+00, 1.00E+01]

single_spec_settings["fluxes"] = [3.71E+07, 2.13E+07, 1.36E+07, 4.11E+06, 1.04E+06, 4.08E+05, 2.07E+05, 6.54E+04,
                                  1.98E+04, 6.46E+03, 2.45E+03, 1.05E+03, 5.07E+02, 2.74E+02, 1.68E+02, 1.14E+02,
                                  8.11E+01, 5.79E+01, 4.24E+01, 3.16E+01, 2.27E+01, 1.56E+01, 0.00E+00]



equipot_model.set_params(params=single_spec_settings)


equipot_result = nom_client.run_model(equipot_model)
print(equipot_result)