few bug solving
This commit is contained in:
@@ -1,8 +1,28 @@
|
||||
# coding: utf-8
|
||||
|
||||
import argparse
|
||||
import os
|
||||
from snapshotprocessor import SnapshotProcessor
|
||||
|
||||
|
||||
def convert_to_hdf5(path, snap_num, path_out=".", filename=None, **kwargs):
|
||||
snap = SnapshotProcessor(path=path, num=snap_num, path_out=path_out, **kwargs)
|
||||
snap.convert_hdf5(filename)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
parser = argparse.ArgumentParser(
|
||||
description="Convert Ramses output into a list of cells"
|
||||
)
|
||||
parser.add_argument(
|
||||
"snapshots", help="path to the snapshots", nargs="+", default=[]
|
||||
)
|
||||
parser.add_argument("-o", "--out", help="path to the output", default=".")
|
||||
|
||||
args = parser.parse_args()
|
||||
|
||||
for snap in args.snapshots:
|
||||
path = os.path.dirname(snap)
|
||||
numstem = os.path.basename(snap)
|
||||
snap_num = int(numstem.split("_")[1])
|
||||
convert_to_hdf5(path, snap_num, args.out)
|
||||
|
||||
Reference in New Issue
Block a user