From dba0c4089115d25cd9937df48d9d02ac38791e9a Mon Sep 17 00:00:00 2001
From: ignis <ignis@postech.ac.kr>
Date: Wed, 3 Mar 2021 07:24:30 +0900
Subject: [PATCH] extract u and c in x major order

---
 extract_c.py | 27 +++++++++++++++++++++++++++
 extract_u.py | 27 +++++++++++++++++++++++++++
 2 files changed, 54 insertions(+)
 create mode 100644 extract_c.py
 create mode 100644 extract_u.py

diff --git a/extract_c.py b/extract_c.py
new file mode 100644
index 0000000..7dfac11
--- /dev/null
+++ b/extract_c.py
@@ -0,0 +1,27 @@
+import argparse
+import numpy as np
+import dnstool
+
+# Commandline argument parser
+parser = argparse.ArgumentParser()
+parser.add_argument("-c", "--case", help="target case name", required=True)
+args = parser.parse_args()
+params = vars(args)
+
+casename = params["case"]
+field_idx = 4
+
+cases = dnstool.case_library()
+
+case = cases[casename]
+
+nx, ny, nz = case.shape
+
+with open("c.dat", "w+") as ufile:
+
+    storage = np.memmap(ufile, dtype=np.double, shape=(nx, len(case.data_files), ny, nz))
+
+    for i, fname in enumerate(case.data_files):
+        print(i, fname)
+        time, y = case.read_single_field(fname, field_idx)
+        storage[:,i,:,:] = 1. - y.T
diff --git a/extract_u.py b/extract_u.py
new file mode 100644
index 0000000..d25f0a8
--- /dev/null
+++ b/extract_u.py
@@ -0,0 +1,27 @@
+import argparse
+import numpy as np
+import dnstool
+
+# Commandline argument parser
+parser = argparse.ArgumentParser()
+parser.add_argument("-c", "--case", help="target case name", required=True)
+args = parser.parse_args()
+params = vars(args)
+
+casename = params["case"]
+field_idx = 0
+
+cases = dnstool.case_library()
+
+case = cases[casename]
+
+nx, ny, nz = case.shape
+
+with open("u.dat", "w+") as ufile:
+
+    storage = np.memmap(ufile, dtype=np.double, shape=(nx, len(case.data_files), ny, nz))
+
+    for i, fname in enumerate(case.data_files):
+        print(i, fname)
+        time, u = case.read_single_field(fname, field_idx)
+        storage[:,i,:,:] = u.T