extract u and c in x major order
This commit is contained in:
parent
2afedba9d6
commit
dba0c40891
2 changed files with 54 additions and 0 deletions
27
extract_c.py
Normal file
27
extract_c.py
Normal file
|
|
@ -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
|
||||
27
extract_u.py
Normal file
27
extract_u.py
Normal file
|
|
@ -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
|
||||
Loading…
Add table
Reference in a new issue