From c4ba41107ef4a07be1f8baa5ed0f84f8adc4c87b Mon Sep 17 00:00:00 2001 From: David Minton Date: Tue, 22 Jun 2021 16:45:35 -0400 Subject: [PATCH] Fixed bug in swiftestio that was preventing rotations from being saved in an input file conversion --- python/swiftestio/swiftestio.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/python/swiftestio/swiftestio.py b/python/swiftestio/swiftestio.py index 9da200cb4..69c88b711 100644 --- a/python/swiftestio/swiftestio.py +++ b/python/swiftestio/swiftestio.py @@ -710,11 +710,10 @@ def swiftest_xr2_infile(ds, config, framenum=-1): print(pli['Radius'].values, file=plfile) print(pli['px'].values, pli['py'].values, pli['pz'].values, file=plfile) print(pli['vx'].values, pli['vy'].values, pli['vz'].values, file=plfile) - if config['ROTATION'] == 'YES': - print(pli['Ip_x'].values, pli['Ip_y'].values, pli['Ip_z'].values, file=plfile) - print(pli['rot_x'].values, pli['rot_y'].values, pli['rot_z'].values, file=plfile) - - plfile.close() + if config['ROTATION'] == 'YES': + print(pli['Ip_x'].values, pli['Ip_y'].values, pli['Ip_z'].values, file=plfile) + print(pli['rot_x'].values, pli['rot_y'].values, pli['rot_z'].values, file=plfile) + plfile.close() # TP file tpfile = open(config['TP_IN'], 'w')