diff --git a/python/swiftest/swiftest/init_cond.py b/python/swiftest/swiftest/init_cond.py index b5424f7ce..061d6d6d7 100644 --- a/python/swiftest/swiftest/init_cond.py +++ b/python/swiftest/swiftest/init_cond.py @@ -129,7 +129,7 @@ def solar_system_horizons(plname, idval, param, ephemerides_start_date, ds): ephemerides_step = '1d' clab, plab, tlab = swiftest.io.make_swiftest_labels(param) - if param['OUT_FORM'] == 'XV': + if param['OUT_FORM'] == 'EL': plab.append('a') plab.append('e') plab.append('inc') @@ -142,7 +142,7 @@ def solar_system_horizons(plname, idval, param, ephemerides_start_date, ds): tlab.append('capom') tlab.append('omega') tlab.append('capm') - elif param['OUT_FORM'] == 'EL': + elif param['OUT_FORM'] == 'XV': plab.append('xhx') plab.append('xhy') plab.append('xhz') @@ -156,6 +156,32 @@ def solar_system_horizons(plname, idval, param, ephemerides_start_date, ds): tlab.append('vhy') tlab.append('vhz') + elif param['OUT_FORM'] == 'XVEL': + plab.append('xhx') + plab.append('xhy') + plab.append('xhz') + plab.append('vhx') + plab.append('vhy') + plab.append('vhz') + tlab.append('xhx') + tlab.append('xhy') + tlab.append('xhz') + tlab.append('vhx') + tlab.append('vhy') + tlab.append('vhz') + plab.append('a') + plab.append('e') + plab.append('inc') + plab.append('capom') + plab.append('omega') + plab.append('capm') + tlab.append('a') + tlab.append('e') + tlab.append('inc') + tlab.append('capom') + tlab.append('omega') + tlab.append('capm') + dims = ['time', 'id', 'vec'] t = np.array([0.0])