Fixed bug in swiftestio that was preventing rotations from being saved in an input file conversion

MintonGroup · Jun 22, 2021 · c4ba411 · c4ba411
1 parent 4717e42
commit c4ba411
Showing 1 changed file with 4 additions and 5 deletions.
diff --git 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')