dump2rotorang.pyをargpase対応にした

dump2rotorang.py

@@ -1,4 +1,5 @@
 #!/usr/bin/env python
+import argparse
 import numpy as np
 import io
 import re
@@ -86,13 +87,17 @@ def run(trjfile):
     dset[temp]["orientation"] = df
 
 
-for k in ["D3d-o", "D3d-p"]:
+par = argparse.ArgumentParser(description="test")
-    dset = {}
+par.add_argument('-f', '--files')
-    files = sorted(glob.glob(f"{k}_*K.lammpstrj"))
+args = par.parse_args()
-    for f in files:
+
-        print(f)
+# for k in ["D3d-o", "D3d-p"]:
-        run(f)
+#     dset = {}
-    outfile = f"{k}.pkl"
+#     files = sorted(glob.glob(f"{k}_*K.lammpstrj"))
-    with open(outfile, "wb") as f:
+#     for f in files:
-        pickle.dump(dset, f)
+#         print(f)
-    print(f"{outfile} was created.")
+#         run(f)
+#     outfile = f"{k}.pkl"
+#     with open(outfile, "wb") as f:
+#         pickle.dump(dset, f)
+#     print(f"{outfile} was created.")