From 4ba22ae672625e447ec472c8547fbb3e51da7c50 Mon Sep 17 00:00:00 2001 From: David Minton Date: Thu, 21 Oct 2021 10:09:20 -0400 Subject: [PATCH] Updated python write methods to include the new parameter variables. --- python/swiftest/swiftest/io.py | 15 ++++++++++++++- python/swiftest/swiftest/simulation_class.py | 8 +++++--- 2 files changed, 19 insertions(+), 4 deletions(-) diff --git a/python/swiftest/swiftest/io.py b/python/swiftest/swiftest/io.py index 3c2151413..ca0f29374 100644 --- a/python/swiftest/swiftest/io.py +++ b/python/swiftest/swiftest/io.py @@ -58,6 +58,11 @@ def read_swiftest_param(param_file_name, param): param['ISTEP_OUT'] = int(param['ISTEP_OUT']) param['ISTEP_DUMP'] = int(param['ISTEP_DUMP']) + param['OUT_TYPE'] = param['OUT_TYPE'].upper() + param['OUT_FORM'] = param['OUT_FORM'].upper() + param['OUT_STAT'] = param['OUT_STAT'].upper() + param['IN_TYPE'] = param['IN_TYPE'].upper() + param['IN_FORM'] = param['IN_FORM'].upper() param['T0'] = real2float(param['T0']) param['TSTOP'] = real2float(param['TSTOP']) param['DT'] = real2float(param['DT']) @@ -74,11 +79,14 @@ def read_swiftest_param(param_file_name, param): param['RHILL_PRESENT'] = param['RHILL_PRESENT'].upper() param['FRAGMENTATION'] = param['FRAGMENTATION'].upper() if param['FRAGMENTATION'] == 'YES' and param['PARTICLE_OUT'] == '': - param['PARTICLE_OUT'] = 'particle.dat' + if param['OUT_TYPE'] == 'REAL8' or param['OUT_TYPE'] == 'REAL4': + param['PARTICLE_OUT'] = 'particle.dat' param['ROTATION'] = param['ROTATION'].upper() param['TIDES'] = param['TIDES'].upper() param['ENERGY'] = param['ENERGY'].upper() param['GR'] = param['GR'].upper() + param['INTERACTION_LOOPS'] = param['INTERACTION_LOOPS'].upper() + param['ENCOUNTER_CHECK'] = param['ENCOUNTER_CHECK'].upper() if 'GMTINY' in param: param['GMTINY'] = real2float(param['GMTINY']) except IOError: @@ -1517,6 +1525,11 @@ def swiftest2swifter_param(swiftest_param, J2=0.0, J4=0.0): swifter_param['OUT_TYPE'] = 'REAL4' if swifter_param['OUT_FORM'] == 'XVEL': swifter_param['OUT_FORM'] = 'XV' + IN_FORM = swifter_param.pop("IN_FORM", None) + INTERACTION_LOOPS = swifter_param.pop("INTERACTION_LOOPS", None) + ENCOUNTER_CHECK = swifter_param.pop("ENCOUNTER_CHECK", None) + ENCOUNTER_CHECK_PLPL = swifter_param.pop("ENCOUNTER_CHECK_PLPL", None) + ENCOUNTER_CHECK_PLTP = swifter_param.pop("ENCOUNTER_CHECK_PLTP", None) swifter_param['! VERSION'] = "Swifter parameter file converted from Swiftest" return swifter_param diff --git a/python/swiftest/swiftest/simulation_class.py b/python/swiftest/swiftest/simulation_class.py index b2736c96d..79c17b086 100644 --- a/python/swiftest/swiftest/simulation_class.py +++ b/python/swiftest/swiftest/simulation_class.py @@ -34,12 +34,12 @@ def __init__(self, codename="Swiftest", param_file=""): 'CHK_QMIN': "-1.0", 'CHK_QMIN_COORD': "HELIO", 'CHK_QMIN_RANGE': "-1.0 -1.0", - 'ENC_OUT': "enc.dat", + 'ENC_OUT': "", 'MU2KG': constants.MSun, 'TU2S': constants.JD2S, 'DU2M': constants.AU2M, 'EXTRA_FORCE': "NO", - 'DISCARD_OUT': "discard.out", + 'DISCARD_OUT': "", 'PARTICLE_OUT' : "", 'BIG_DISCARD': "NO", 'CHK_CLOSE': "YES", @@ -48,7 +48,9 @@ def __init__(self, codename="Swiftest", param_file=""): 'ROTATION': "NO", 'TIDES': "NO", 'ENERGY': "NO", - 'GR': "NO", + 'GR': "YES", + 'INTERACTION_LOOPS': "ADAPTIVE", + 'ENCOUNTER_CHECK': "ADAPTIVE" } self.codename = codename if param_file != "" :