mirror of
https://github.com/OPM/ResInsight.git
synced 2025-02-25 18:55:39 -06:00
44 lines
1.7 KiB
Python
44 lines
1.7 KiB
Python
############################################################################
|
||
# This script will export snapshots for two properties in every loaded case
|
||
# And put them in a snapshots folder in the same folder as the case grid
|
||
############################################################################
|
||
import os
|
||
import rips
|
||
|
||
# Load instance
|
||
resinsight = rips.Instance.find()
|
||
cases = resinsight.project.cases()
|
||
|
||
# Set main window size
|
||
resinsight.commands.set_main_window_size(width=800, height=500)
|
||
|
||
n = 5 # every n-th time_step for snapshot
|
||
property_list = ['SOIL', 'PRESSURE'] # list of parameter for snapshot
|
||
|
||
print ("Looping through cases")
|
||
for case in cases:
|
||
# Get grid path and its folder name
|
||
case_path = case.grid_path()
|
||
folder_name = os.path.dirname(case_path)
|
||
|
||
# create a folder to hold the snapshots
|
||
dirname = os.path.join(folder_name, 'snapshots')
|
||
|
||
if os.path.exists(dirname) is False:
|
||
os.mkdir(dirname)
|
||
|
||
print ("Exporting to folder: " + dirname)
|
||
resinsight.commands.set_export_folder(type='SNAPSHOTS', path=dirname)
|
||
|
||
time_steps = case.time_steps()
|
||
tss_snapshot = range(0, len(time_steps), n)
|
||
print(case.name, case.id, 'Number of time_steps: ' + str(len(time_steps)))
|
||
print('Number of time_steps for snapshoting: ' + str(len(tss_snapshot)))
|
||
|
||
view = case.views()[0]
|
||
for property in property_list:
|
||
view.apply_cell_result(result_type='DYNAMIC_NATIVE', result_variable=property)
|
||
for ts_snapshot in tss_snapshot:
|
||
resinsight.commands.set_time_step(case_id = case.id, time_step = ts_snapshot)
|
||
resinsight.commands.export_snapshots(type='VIEWS', case_id=case.id) # ‘ALL’, ‘VIEWS’ or ‘PLOTS’ default is 'ALL'
|