-
Notifications
You must be signed in to change notification settings - Fork 60
/
parse_palabos_output.py
41 lines (28 loc) · 1.36 KB
/
parse_palabos_output.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
import numpy as np
from .pore_utils import find_line_in_file
def create_pressure_data_file(inputs):
sim_dir = inputs['input output']['simulation directory']
output_dir = inputs['input output']['output folder']
# Parse pressure data
pc_data_file = sim_dir + '/' + output_dir + 'output.dat'
pc_data = find_line_in_file(pc_data_file, 'Pressure difference =', 3)
pc_data = np.insert(pc_data, 0, 0)
np.savetxt(sim_dir + '/' + output_dir + 'data_Pc.txt', pc_data)
return
def create_relperm_data_file(inputs):
sim_dir = inputs['input output']['simulation directory']
output_dir = inputs['input output']['output folder']
num_runs = inputs['rel perm']['num_geoms']
# print((num_runs-1)/2)
# Parse rel perm data
kr_data_file = sim_dir + '/' + output_dir + r'4relperm/relPerm&vels.txt'
k_absolute = find_line_in_file(kr_data_file, 'Absolute Permeability', 3)
kr_data = find_line_in_file(kr_data_file, 'Relative Permeability', 3)
divide_kr_data = int((num_runs-1)/2) + 1
krw_data = kr_data[0:divide_kr_data]
krnw_data = kr_data[divide_kr_data:]
krnw_data = np.insert(krnw_data, 0, 0)
np.savetxt(sim_dir + '/' + output_dir + 'data_krw.txt', krw_data)
np.savetxt(sim_dir + '/' + output_dir + 'data_krnw.txt', krnw_data)
np.savetxt(sim_dir + '/' + output_dir + 'data_k.txt', k_absolute)
return