Make scriptability a CAF-feature

ApplicationCode/GrpcInterface/Python/rips/PythonExamples/create_wbs_plot.py

@@ -6,60 +6,33 @@ import rips
 # Connect to ResInsight instance
 resInsight = rips.Instance.find()
 
-cases = resInsight.project.cases()
-
-well_paths = resInsight.project.import_well_paths(well_path_folder='D:/Projects/ResInsight-regression-test/ModelData/Norne_WellPaths')
-well_log_files = resInsight.project.import_well_log_files(well_log_folder='D:/Projects/ResInsight-regression-test/ModelData/Norne_PLT_LAS')
+# Get all GeoMech cases
+cases = resInsight.project.descendants(rips.GeoMechCase)
 
+# Get all well paths
 well_paths = resInsight.project.well_paths()
 
+# Ensure there's at least one well path
 if len(well_paths) < 1:
     print("No well paths in project")
     exit(1)
 
-well_paths = well_paths[0:4]
-    
-print(well_paths)
-
-# Example of creating parameters for new plots
-wbs_parameters = rips.WbsParameters()
-wbs_parameters.user_poisson_ratio = 0.412347
-wbs_parameters.pore_pressure_outside_reservoir_source = "USER_DEFINED"
-wbs_parameters.user_pp_outside_reservoir = 1.1
-wbs_parameters.fg_shale_source = "PROPORTIONAL_TO_SH"
-wbs_parameters.user_fg_shale = 1.13
-
+# Loop through all cases
 for case in cases:
-    if case.type == "GeoMechCase":
-        min_res_depth, max_res_depth = case.reservoir_depth_range()
-	
-        print (case.id)
-        case_path = case.file_path
-        folder_name = os.path.dirname(case_path)
-        case.import_formation_names(formation_files=['D:/Projects/ResInsight-regression-test/ModelData/norne/Norne_ATW2013.lyr'])
+    assert(isinstance(case, rips.GeoMechCase))
+    min_res_depth, max_res_depth = case.reservoir_depth_range()
 
-        # create a folder to hold the snapshots
-        dirname = os.path.join(folder_name, 'snapshots')
-        print("Exporting to: " + dirname)
+    # Find a good output path
+    case_path = case.file_path
+    folder_name = os.path.dirname(case_path)
 
-        for well_path in well_paths:
-            try:
-                # Create plot with parameters
-                wbsplot = case.create_well_bore_stability_plot(well_path=well_path.name, time_step=0, wbs_parameters=wbs_parameters)
-                # Example of setting parameters for existing plots
-                replace_params = wbsplot.parameters()
-                replace_params.user_poisson_ratio = 0.654321
-                replace_params.user_fg_shale = 1.0321
-                wbsplot.set_parameters(replace_params)
-                # Demonstrate altering general well log plot settings
-                min_depth = max(wbsplot.minimum_depth, min_res_depth)
-                max_depth = min(wbsplot.maximum_depth, max_res_depth)
-                wbsplot.minimum_depth = min_depth
-                wbsplot.maximum_depth = max_depth
-                wbsplot.update()
-                #wbsplot.depth_type = "TRUE_VERTICAL_DEPTH_RKB"
-                
-                wbsplot.export_snapshot(export_folder=dirname)
+    # Import formation names
+    case.import_formation_names(formation_files=['D:/Projects/ResInsight-regression-test/ModelData/norne/Norne_ATW2013.lyr'])
 
-            except grpc.RpcError as e:
-                print("Error: ", e.details())
+    # create a folder to hold the snapshots
+    dirname = os.path.join(folder_name, 'snapshots')
+    print("Exporting to: " + dirname)
+
+    for well_path in well_paths[0:4]: # Loop through the first five well paths
+        # Create plot with parameters
+        wbsplot = case.create_well_bore_stability_plot(well_path=well_path.name, time_step=0)