adapt postprocess

2019-04-05 09:46:24 +02:00
parent b852cfec80
commit dca8ab5d41
2 changed files with 149 additions and 97 deletions
@@ -0,0 +1,58 @@
+# coding: utf-8
+import sys
+import numpy as np
+import module_extract as me
+import extract_disk as d
+import pymses
+import os
+import argparse
+import disk_postprocess as dp
+
+
+storage_in = "/drf/projets/alfven-data/"
+storage_out = "/dsm/anais/storageA/"
+
+parser = argparse.ArgumentParser()
+parser.add_argument("runs", help="name of runs", nargs="*", default=["015_iso"])
+parser.add_argument(
+    "-f", "--first_output", help="id of first output", type=int, default=1
+)
+parser.add_argument(
+    "-l", "--last_output", help="id of last output", type=int, default=100
+)
+parser.add_argument("-s", "--step", help="step between two output", type=int, default=1)
+args = parser.parse_args()
+
+
+user = "nbrucy"
+folder = "simus"
+project = "disk"
+runs = args.runs
+
+first = args.first_output
+last = args.last_output
+step = args.step
+
+for run in runs:
+    path_suffix = user + "/" + folder + "/" + project + "/" + run
+    path_in = storage_in + path_suffix
+    path_out = storage_out + path_suffix
+
+    if not os.path.exists(path_out):
+        os.makedirs(path_out)
+
+    for i in range(first, last + 1, step):
+        me.make_image_zoom(
+            path_in,
+            i,
+            [0.5],
+            sinks=False,
+            force=False,
+            path_out=path_out,
+            tag=run + "_",
+            cpuamr=False,
+            mag_im=False,
+            AU_units=False,
+        )
+        dp.disk_prop(path_in, i, path_out=path_out, rad_ext=50000)
+        dp.plot_disk_prop(path_out, i, tag=run + "_")