completor 1.1.2__py3-none-any.whl → 1.2.0__py3-none-any.whl

completor/prepare_outputs.py

@@ -120,6 +120,7 @@ def dataframe_tostring(
             "END_MD": "{:.3f}".format,
             Headers.FLOW_COEFFICIENT: "{:.10g}".format,
             "CV": "{:.10g}".format,
+            Headers.CROSS: "{:.3e}".format,
             Headers.FLOW_CROSS_SECTIONAL_AREA: "{:.3e}".format,
             "FLOW_CROSS_SECTIONAL_AREA": "{:.3e}".format,
             Headers.OIL_FLOW_CROSS_SECTIONAL_AREA: "{:.3e}".format,
@@ -135,6 +136,13 @@ def dataframe_tostring(
             Headers.ALPHA_PILOT: "{:.10g}".format,
         }
 
+        # Cast floats to str befor headers are messed up (pandas formatter does not work reliably with MultiIndex headers).
+        for column, formatter in formatters.items():
+            try:
+                df_temp[column] = df_temp[column].map(formatter)
+            except (KeyError, ValueError):
+                pass
+
         if header:
             # Modify headers to reduce width.
             column_splits = [tuple(column.split("_")) for column in df_temp.columns]
@@ -173,9 +181,7 @@ def dataframe_tostring(
                 df_temp.columns = pd.MultiIndex.from_frame(new_cols)
 
     try:
-        output_string = df_temp.to_string(
-            index=False, justify="justify", formatters=formatters, header=header, sparsify=False
-        )
+        output_string = df_temp.to_string(index=False, justify="justify", header=header, sparsify=False)
     except ValueError:
         if df_temp.isnull().values.any():
             raise CompletorError("Got NaN values in table, please report if encountered!")
@@ -184,18 +190,14 @@ def dataframe_tostring(
         df_temp = df_temp.replace("1*", np.nan, inplace=False)
         # Probably find columns where this is the case and cast to numeric after replacing with nan?
         df_temp[columns_with_1_star] = df_temp[columns_with_1_star].astype(np.float64, errors="ignore")
-        output_string = df_temp.to_string(
-            index=False, justify="justify", formatters=formatters, header=header, sparsify=False, na_rep="1*"
-        )
+        output_string = df_temp.to_string(index=False, justify="justify", header=header, sparsify=False, na_rep="1*")
 
     if output_string is None:
         return ""
 
     too_long_lines = check_width_lines(output_string, limit)
     if too_long_lines:
-        output_string = df_temp.to_string(
-            index=False, justify="left", formatters=formatters, header=header, sparsify=False
-        )
+        output_string = df_temp.to_string(index=False, justify="left", header=header, sparsify=False)
         if output_string is None:
             return ""
         too_long_lines2 = check_width_lines(output_string, limit)
@@ -448,13 +450,17 @@ def prepare_device_layer(df_well: pd.DataFrame, df_tubing: pd.DataFrame, device_
                     df_well[Headers.DEVICE_TYPE] == Content.VALVE,
                     "/ -- Valve types",
                     np.where(
-                        df_well[Headers.DEVICE_TYPE] == Content.DENSITY_ACTIVATED_RECOVERY,
-                        "/ -- DAR types",
+                        df_well[Headers.DEVICE_TYPE] == Content.DENSITY,
+                        "/ -- DENSITY types",
                         np.where(
-                            df_well[Headers.DEVICE_TYPE] == Content.AUTONOMOUS_INFLOW_CONTROL_VALVE,
-                            "/ -- AICV types",
+                            df_well[Headers.DEVICE_TYPE] == Content.DUAL_RATE_CONTROLLED_PRODUCTION,
+                            "/ -- DUALRCP types",
                             np.where(
-                                df_well[Headers.DEVICE_TYPE] == Content.INFLOW_CONTROL_VALVE, "/ -- ICV types", ""
+                                df_well[Headers.DEVICE_TYPE] == Content.INJECTION_VALVE,
+                                "/ -- INJV types",
+                                np.where(
+                                    df_well[Headers.DEVICE_TYPE] == Content.INFLOW_CONTROL_VALVE, "/ -- ICV types", ""
+                                ),
                             ),
                         ),
                     ),
@@ -852,6 +858,7 @@ def prepare_completion_segments(
             }
         )
     compseg[Headers.EMPTY] = "/"
+    compseg.sort_values(Headers.START_MEASURED_DEPTH, inplace=True)
     return compseg
 
 
@@ -1067,6 +1074,8 @@ def prepare_autonomous_inflow_control_device(
         wsegaicd[Headers.D] = df_merge[Headers.D].to_numpy()
         wsegaicd[Headers.E] = df_merge[Headers.E].to_numpy()
         wsegaicd[Headers.F] = df_merge[Headers.F].to_numpy()
+        if Headers.Z in df_merge.columns:
+            wsegaicd[Headers.Z] = df_merge[Headers.Z].to_numpy()
         wsegaicd[Headers.EMPTY] = "/"
     return wsegaicd
 
@@ -1249,8 +1258,8 @@ def prepare_inflow_control_valve(
     return wsegicv
 
 
-def prepare_density_activated_recovery(well_name: str, df_well: pd.DataFrame, df_device: pd.DataFrame) -> pd.DataFrame:
-    """Prepare data frame for DAR.
+def prepare_density_driven(well_name: str, df_well: pd.DataFrame, df_device: pd.DataFrame) -> pd.DataFrame:
+    """Prepare data frame for DENSITY.
 
     Args:
         well_name: Well name.
@@ -1258,7 +1267,7 @@ def prepare_density_activated_recovery(well_name: str, df_well: pd.DataFrame, df
         df_device: Device data for this well and lateral.
 
     Returns:
-        DataFrame for DAR.
+        DataFrame for DENSITY.
     """
     df_well = df_well[(df_well[Headers.DEVICE_TYPE] == Content.PERFORATED) | (df_well[Headers.NUMBER_OF_DEVICES] > 0)]
     if df_well.shape[0] == 0:
@@ -1270,34 +1279,84 @@ def prepare_density_activated_recovery(well_name: str, df_well: pd.DataFrame, df
         right_on=[Headers.TUBING_MEASURED_DEPTH],
         direction="nearest",
     )
-    df_merge = df_merge[df_merge[Headers.DEVICE_TYPE] == Content.DENSITY_ACTIVATED_RECOVERY]
-    wsegdar = pd.DataFrame()
+    df_merge = df_merge[df_merge[Headers.DEVICE_TYPE] == Content.DENSITY]
+    wsegdensity = pd.DataFrame()
     if df_merge.shape[0] > 0:
-        wsegdar[Headers.WELL] = [well_name] * df_merge.shape[0]
-        wsegdar[Headers.START_SEGMENT_NUMBER] = df_merge[Headers.START_SEGMENT_NUMBER].to_numpy()
+        wsegdensity[Headers.WELL] = [well_name] * df_merge.shape[0]
+        wsegdensity[Headers.START_SEGMENT_NUMBER] = df_merge[Headers.START_SEGMENT_NUMBER].to_numpy()
         # the Cv is already corrected by the scaling factor
-        wsegdar[Headers.FLOW_COEFFICIENT] = df_merge[Headers.FLOW_COEFFICIENT].to_numpy()
-        wsegdar[Headers.OIL_FLOW_CROSS_SECTIONAL_AREA] = df_merge[Headers.OIL_FLOW_CROSS_SECTIONAL_AREA].to_numpy()
-        wsegdar[Headers.GAS_FLOW_CROSS_SECTIONAL_AREA] = df_merge[Headers.GAS_FLOW_CROSS_SECTIONAL_AREA].to_numpy()
-        wsegdar[Headers.WATER_FLOW_CROSS_SECTIONAL_AREA] = df_merge[Headers.WATER_FLOW_CROSS_SECTIONAL_AREA].to_numpy()
-        wsegdar[Headers.WATER_HOLDUP_FRACTION_LOW_CUTOFF] = df_merge[
+        wsegdensity[Headers.FLOW_COEFFICIENT] = df_merge[Headers.FLOW_COEFFICIENT].to_numpy()
+        wsegdensity[Headers.OIL_FLOW_CROSS_SECTIONAL_AREA] = df_merge[Headers.OIL_FLOW_CROSS_SECTIONAL_AREA].to_numpy()
+        wsegdensity[Headers.GAS_FLOW_CROSS_SECTIONAL_AREA] = df_merge[Headers.GAS_FLOW_CROSS_SECTIONAL_AREA].to_numpy()
+        wsegdensity[Headers.WATER_FLOW_CROSS_SECTIONAL_AREA] = df_merge[
+            Headers.WATER_FLOW_CROSS_SECTIONAL_AREA
+        ].to_numpy()
+        wsegdensity[Headers.WATER_HOLDUP_FRACTION_LOW_CUTOFF] = df_merge[
             Headers.WATER_HOLDUP_FRACTION_LOW_CUTOFF
         ].to_numpy()
-        wsegdar[Headers.WATER_HOLDUP_FRACTION_HIGH_CUTOFF] = df_merge[
+        wsegdensity[Headers.WATER_HOLDUP_FRACTION_HIGH_CUTOFF] = df_merge[
             Headers.WATER_HOLDUP_FRACTION_HIGH_CUTOFF
         ].to_numpy()
-        wsegdar[Headers.GAS_HOLDUP_FRACTION_LOW_CUTOFF] = df_merge[Headers.GAS_HOLDUP_FRACTION_LOW_CUTOFF].to_numpy()
-        wsegdar[Headers.GAS_HOLDUP_FRACTION_HIGH_CUTOFF] = df_merge[Headers.GAS_HOLDUP_FRACTION_HIGH_CUTOFF].to_numpy()
-        wsegdar[Headers.DEFAULTS] = "5*"
-        wsegdar[Headers.MAX_FLOW_CROSS_SECTIONAL_AREA] = wsegdar[Headers.OIL_FLOW_CROSS_SECTIONAL_AREA].to_numpy()
-        wsegdar[Headers.EMPTY] = "/"
-    return wsegdar
+        wsegdensity[Headers.GAS_HOLDUP_FRACTION_LOW_CUTOFF] = df_merge[
+            Headers.GAS_HOLDUP_FRACTION_LOW_CUTOFF
+        ].to_numpy()
+        wsegdensity[Headers.GAS_HOLDUP_FRACTION_HIGH_CUTOFF] = df_merge[
+            Headers.GAS_HOLDUP_FRACTION_HIGH_CUTOFF
+        ].to_numpy()
+        wsegdensity[Headers.DEFAULTS] = "5*"
+        wsegdensity[Headers.MAX_FLOW_CROSS_SECTIONAL_AREA] = wsegdensity[
+            Headers.OIL_FLOW_CROSS_SECTIONAL_AREA
+        ].to_numpy()
+        wsegdensity[Headers.EMPTY] = "/"
+    return wsegdensity
 
 
-def prepare_autonomous_inflow_control_valve(
+def prepare_injection_valve(well_name: str, df_well: pd.DataFrame, df_device: pd.DataFrame) -> pd.DataFrame:
+    """Prepare data frame for INJECTION VALVE.
+
+    Args:
+        well_name: Well name.
+        df_well: Well data.
+        df_device: Device data for this well and lateral.
+
+    Returns:
+        DataFrame for INJECTION VALVE.
+    """
+    df_well = df_well[(df_well[Headers.DEVICE_TYPE] == Content.PERFORATED) | (df_well[Headers.NUMBER_OF_DEVICES] > 0)]
+    if df_well.shape[0] == 0:
+        return pd.DataFrame()
+    df_merge = pd.merge_asof(
+        left=df_device,
+        right=df_well,
+        left_on=[Headers.MEASURED_DEPTH],
+        right_on=[Headers.TUBING_MEASURED_DEPTH],
+        direction="nearest",
+    )
+    df_merge = df_merge[df_merge[Headers.DEVICE_TYPE] == Content.INJECTION_VALVE]
+    wseginjv = pd.DataFrame()
+    if df_merge.shape[0] > 0:
+        wseginjv[Headers.WELL] = [well_name] * df_merge.shape[0]
+        wseginjv[Headers.START_SEGMENT_NUMBER] = df_merge[Headers.START_SEGMENT_NUMBER].to_numpy()
+        # the Cv is already corrected by the scaling factor
+        wseginjv[Headers.FLOW_COEFFICIENT] = df_merge[Headers.FLOW_COEFFICIENT].to_numpy()
+        wseginjv[Headers.PRIMARY_FLOW_CROSS_SECTIONAL_AREA] = df_merge[
+            Headers.PRIMARY_FLOW_CROSS_SECTIONAL_AREA
+        ].to_numpy()
+        wseginjv[Headers.SECONDARY_FLOW_CROSS_SECTIONAL_AREA] = df_merge[
+            Headers.SECONDARY_FLOW_CROSS_SECTIONAL_AREA
+        ].to_numpy()
+        wseginjv[Headers.TRIGGER_PARAMETER] = df_merge[Headers.TRIGGER_PARAMETER].to_numpy()
+        wseginjv[Headers.TRIGGER_VALUE] = df_merge[Headers.TRIGGER_VALUE].to_numpy()
+        wseginjv[Headers.DEFAULTS] = "5*"
+        wseginjv[Headers.MAX_FLOW_CROSS_SECTIONAL_AREA] = wseginjv[Headers.PRIMARY_FLOW_CROSS_SECTIONAL_AREA].to_numpy()
+        wseginjv[Headers.EMPTY] = "/"
+    return wseginjv
+
+
+def prepare_dual_rate_controlled_production(
     well_name: str, df_well: pd.DataFrame, df_device: pd.DataFrame
 ) -> pd.DataFrame:
-    """Prepare data frame for AICV.
+    """Prepare data frame for DUALRCP.
 
     Args:
         well_name: Well name.
@@ -1305,7 +1364,7 @@ def prepare_autonomous_inflow_control_valve(
         df_device: Device data for this well and lateral.
 
     Returns:
-        DataFrame for AICV.
+        DataFrame for DUALRCP.
     """
     df_well = df_well[(df_well[Headers.DEVICE_TYPE] == Content.PERFORATED) | (df_well[Headers.NUMBER_OF_DEVICES] > 0)]
     if df_well.shape[0] == 0:
@@ -1317,53 +1376,53 @@ def prepare_autonomous_inflow_control_valve(
         right_on=[Headers.TUBING_MEASURED_DEPTH],
         direction="nearest",
     )
-    df_merge = df_merge[df_merge[Headers.DEVICE_TYPE] == Content.AUTONOMOUS_INFLOW_CONTROL_VALVE]
-    wsegaicv = pd.DataFrame()
+    df_merge = df_merge[df_merge[Headers.DEVICE_TYPE] == Content.DUAL_RATE_CONTROLLED_PRODUCTION]
+    wsegdualrcp = pd.DataFrame()
     if df_merge.shape[0] > 0:
-        wsegaicv[Headers.WELL] = [well_name] * df_merge.shape[0]
-        wsegaicv[Headers.START_SEGMENT_NUMBER] = df_merge[Headers.START_SEGMENT_NUMBER].to_numpy()
-        wsegaicv[Headers.END_SEGMENT_NUMBER] = df_merge[Headers.START_SEGMENT_NUMBER].to_numpy()
-        wsegaicv[Headers.ALPHA_MAIN] = df_merge[Headers.ALPHA_MAIN].to_numpy()
-        wsegaicv[Headers.SCALE_FACTOR] = df_merge[Headers.SCALE_FACTOR].to_numpy()
-        wsegaicv[Headers.CALIBRATION_FLUID_DENSITY] = df_merge[Headers.AICV_CALIBRATION_FLUID_DENSITY].to_numpy()
-        wsegaicv[Headers.CALIBRATION_FLUID_VISCOSITY] = df_merge[Headers.AICV_FLUID_VISCOSITY].to_numpy()
-        wsegaicv[Headers.DEF] = ["5*"] * df_merge.shape[0]
-        wsegaicv[Headers.X_MAIN] = df_merge[Headers.X_MAIN].to_numpy()
-        wsegaicv[Headers.Y_MAIN] = df_merge[Headers.Y_MAIN].to_numpy()
-        wsegaicv[Headers.FLAG] = [Headers.OPEN] * df_merge.shape[0]
-        wsegaicv[Headers.A_MAIN] = df_merge[Headers.A_MAIN].to_numpy()
-        wsegaicv[Headers.B_MAIN] = df_merge[Headers.B_MAIN].to_numpy()
-        wsegaicv[Headers.C_MAIN] = df_merge[Headers.C_MAIN].to_numpy()
-        wsegaicv[Headers.D_MAIN] = df_merge[Headers.D_MAIN].to_numpy()
-        wsegaicv[Headers.E_MAIN] = df_merge[Headers.E_MAIN].to_numpy()
-        wsegaicv[Headers.F_MAIN] = df_merge[Headers.F_MAIN].to_numpy()
-        wsegaicv[Headers.ALPHA_PILOT] = df_merge[Headers.ALPHA_PILOT].to_numpy()
-        wsegaicv[Headers.X_PILOT] = df_merge[Headers.X_PILOT].to_numpy()
-        wsegaicv[Headers.Y_PILOT] = df_merge[Headers.Y_PILOT].to_numpy()
-        wsegaicv[Headers.A_PILOT] = df_merge[Headers.A_PILOT].to_numpy()
-        wsegaicv[Headers.B_PILOT] = df_merge[Headers.B_PILOT].to_numpy()
-        wsegaicv[Headers.C_PILOT] = df_merge[Headers.C_PILOT].to_numpy()
-        wsegaicv[Headers.D_PILOT] = df_merge[Headers.D_PILOT].to_numpy()
-        wsegaicv[Headers.E_PILOT] = df_merge[Headers.E_PILOT].to_numpy()
-        wsegaicv[Headers.F_PILOT] = df_merge[Headers.F_PILOT].to_numpy()
-        wsegaicv[Headers.AICV_WATER_CUT] = df_merge[Headers.AICV_WATER_CUT].to_numpy()
-        wsegaicv[Headers.AICV_GAS_HOLDUP_FRACTION] = df_merge[Headers.AICV_GAS_HOLDUP_FRACTION].to_numpy()
-        wsegaicv[Headers.EMPTY] = "/"
-    return wsegaicv
-
-
-def print_wsegdar(df_wsegdar: pd.DataFrame, well_number: int) -> str:
-    """Print DAR devices.
+        wsegdualrcp[Headers.WELL] = [well_name] * df_merge.shape[0]
+        wsegdualrcp[Headers.START_SEGMENT_NUMBER] = df_merge[Headers.START_SEGMENT_NUMBER].to_numpy()
+        wsegdualrcp[Headers.END_SEGMENT_NUMBER] = df_merge[Headers.START_SEGMENT_NUMBER].to_numpy()
+        wsegdualrcp[Headers.ALPHA_MAIN] = df_merge[Headers.ALPHA_MAIN].to_numpy()
+        wsegdualrcp[Headers.SCALE_FACTOR] = df_merge[Headers.SCALE_FACTOR].to_numpy()
+        wsegdualrcp[Headers.CALIBRATION_FLUID_DENSITY] = df_merge[Headers.DUALRCP_CALIBRATION_FLUID_DENSITY].to_numpy()
+        wsegdualrcp[Headers.CALIBRATION_FLUID_VISCOSITY] = df_merge[Headers.DUALRCP_FLUID_VISCOSITY].to_numpy()
+        wsegdualrcp[Headers.DEF] = ["5*"] * df_merge.shape[0]
+        wsegdualrcp[Headers.X_MAIN] = df_merge[Headers.X_MAIN].to_numpy()
+        wsegdualrcp[Headers.Y_MAIN] = df_merge[Headers.Y_MAIN].to_numpy()
+        wsegdualrcp[Headers.FLAG] = [Headers.OPEN] * df_merge.shape[0]
+        wsegdualrcp[Headers.A_MAIN] = df_merge[Headers.A_MAIN].to_numpy()
+        wsegdualrcp[Headers.B_MAIN] = df_merge[Headers.B_MAIN].to_numpy()
+        wsegdualrcp[Headers.C_MAIN] = df_merge[Headers.C_MAIN].to_numpy()
+        wsegdualrcp[Headers.D_MAIN] = df_merge[Headers.D_MAIN].to_numpy()
+        wsegdualrcp[Headers.E_MAIN] = df_merge[Headers.E_MAIN].to_numpy()
+        wsegdualrcp[Headers.F_MAIN] = df_merge[Headers.F_MAIN].to_numpy()
+        wsegdualrcp[Headers.ALPHA_PILOT] = df_merge[Headers.ALPHA_PILOT].to_numpy()
+        wsegdualrcp[Headers.X_PILOT] = df_merge[Headers.X_PILOT].to_numpy()
+        wsegdualrcp[Headers.Y_PILOT] = df_merge[Headers.Y_PILOT].to_numpy()
+        wsegdualrcp[Headers.A_PILOT] = df_merge[Headers.A_PILOT].to_numpy()
+        wsegdualrcp[Headers.B_PILOT] = df_merge[Headers.B_PILOT].to_numpy()
+        wsegdualrcp[Headers.C_PILOT] = df_merge[Headers.C_PILOT].to_numpy()
+        wsegdualrcp[Headers.D_PILOT] = df_merge[Headers.D_PILOT].to_numpy()
+        wsegdualrcp[Headers.E_PILOT] = df_merge[Headers.E_PILOT].to_numpy()
+        wsegdualrcp[Headers.F_PILOT] = df_merge[Headers.F_PILOT].to_numpy()
+        wsegdualrcp[Headers.DUALRCP_WATER_CUT] = df_merge[Headers.DUALRCP_WATER_CUT].to_numpy()
+        wsegdualrcp[Headers.DUALRCP_GAS_HOLDUP_FRACTION] = df_merge[Headers.DUALRCP_GAS_HOLDUP_FRACTION].to_numpy()
+        wsegdualrcp[Headers.EMPTY] = "/"
+    return wsegdualrcp
+
+
+def print_wsegdensity(df_wsegdensity: pd.DataFrame, well_number: int) -> str:
+    """Print DENSITY devices.
 
     Args:
-        df_wsegdar: Output from function prepare_wsegdar.
+        df_wsegdensity: Output from function prepare_wsegdensity.
         well_number: Well number.
 
     Returns:
         Formatted actions to be included in the output file.
 
     Raises:
-        CompletorError: If there are to many wells and/or segments with DAR.
+        CompletorError: If there are to many wells and/or segments with DENSITY.
     """
     header = [
         [
@@ -1403,9 +1462,9 @@ def print_wsegdar(df_wsegdar: pd.DataFrame, well_number: int) -> str:
     sign_gas = [">", "<=", "<", ""]
     suvtrig = ["0", "0", "1", "2"]
     action = "UDQ\n"
-    for idx in range(df_wsegdar.shape[0]):
-        segment_number = df_wsegdar[Headers.START_SEGMENT_NUMBER].iloc[idx]
-        well_name = df_wsegdar[Headers.WELL].iloc[idx]
+    for idx in range(df_wsegdensity.shape[0]):
+        segment_number = df_wsegdensity[Headers.START_SEGMENT_NUMBER].iloc[idx]
+        well_name = df_wsegdensity[Headers.WELL].iloc[idx]
         action += f"  ASSIGN SUVTRIG {well_name} {segment_number} 0 /\n"
     action += "/\n\n"
     iaction = 3
@@ -1414,25 +1473,25 @@ def print_wsegdar(df_wsegdar: pd.DataFrame, well_number: int) -> str:
     for itm in header[iaction]:
         header_string += "  " + itm
     action += header_string.rstrip() + "\n"
-    for idx in range(df_wsegdar.shape[0]):
-        segment_number = df_wsegdar[Headers.START_SEGMENT_NUMBER].iloc[idx]
-        print_df = df_wsegdar[df_wsegdar[Headers.START_SEGMENT_NUMBER] == segment_number]
+    for idx in range(df_wsegdensity.shape[0]):
+        segment_number = df_wsegdensity[Headers.START_SEGMENT_NUMBER].iloc[idx]
+        print_df = df_wsegdensity[df_wsegdensity[Headers.START_SEGMENT_NUMBER] == segment_number]
         print_df = print_df[header[iaction]]
         print_df = dataframe_tostring(print_df, True, False, False) + "\n"
         action += print_df
     action += "/\n\n"
-    for idx in range(df_wsegdar.shape[0]):
-        segment_number = df_wsegdar[Headers.START_SEGMENT_NUMBER].iloc[idx]
-        well_name = df_wsegdar[Headers.WELL].iloc[idx]
-        water_holdup_fraction_low_cutoff = df_wsegdar[Headers.WATER_HOLDUP_FRACTION_LOW_CUTOFF].iloc[idx]
-        water_holdup_fraction_high_cutoff = df_wsegdar[Headers.WATER_HOLDUP_FRACTION_HIGH_CUTOFF].iloc[idx]
-        gas_holdup_fraction_low_cutoff = df_wsegdar[Headers.GAS_HOLDUP_FRACTION_LOW_CUTOFF].iloc[idx]
-        gas_holdup_fraction_high_cutoff = df_wsegdar[Headers.GAS_HOLDUP_FRACTION_HIGH_CUTOFF].iloc[idx]
+    for idx in range(df_wsegdensity.shape[0]):
+        segment_number = df_wsegdensity[Headers.START_SEGMENT_NUMBER].iloc[idx]
+        well_name = df_wsegdensity[Headers.WELL].iloc[idx]
+        water_holdup_fraction_low_cutoff = df_wsegdensity[Headers.WATER_HOLDUP_FRACTION_LOW_CUTOFF].iloc[idx]
+        water_holdup_fraction_high_cutoff = df_wsegdensity[Headers.WATER_HOLDUP_FRACTION_HIGH_CUTOFF].iloc[idx]
+        gas_holdup_fraction_low_cutoff = df_wsegdensity[Headers.GAS_HOLDUP_FRACTION_LOW_CUTOFF].iloc[idx]
+        gas_holdup_fraction_high_cutoff = df_wsegdensity[Headers.GAS_HOLDUP_FRACTION_HIGH_CUTOFF].iloc[idx]
         for iaction in range(2):
             act_number = iaction + 1
             act_name = f"D{well_number:03d}{segment_number:03d}{act_number:1d}"
             if len(act_name) > 8:
-                raise CompletorError("Too many wells and/or too many segments with DAR")
+                raise CompletorError("Too many wells and/or too many segments with DENSITY")
             action += (
                 f"ACTIONX\n{act_name} 1000000 /\n"
                 f"SWHF '{well_name}' {segment_number} "
@@ -1442,7 +1501,7 @@ def print_wsegdar(df_wsegdar: pd.DataFrame, well_number: int) -> str:
                 f"SUVTRIG '{well_name}' {segment_number} "
                 f"= {suvtrig[iaction]} /\n/\n\n"
             )
-            print_df = df_wsegdar[df_wsegdar[Headers.START_SEGMENT_NUMBER] == segment_number]
+            print_df = df_wsegdensity[df_wsegdensity[Headers.START_SEGMENT_NUMBER] == segment_number]
             print_df = print_df[header[iaction]]  # type: ignore
             header_string = Keywords.WELL_SEGMENTS_VALVE + "\n--"
             for item in header[iaction]:
@@ -1460,7 +1519,7 @@ def print_wsegdar(df_wsegdar: pd.DataFrame, well_number: int) -> str:
         act_number = iaction + 1
         act_name = f"D{well_number:03d}{segment_number:03d}{act_number:1d}"
         if len(act_name) > 8:
-            raise CompletorError("Too many wells and/or too many segments with DAR")
+            raise CompletorError("Too many wells and/or too many segments with DENSITY")
         action += (
             f"ACTIONX\n{act_name} 1000000 /\n"
             f"SGHF '{well_name}' {segment_number} "
@@ -1468,7 +1527,7 @@ def print_wsegdar(df_wsegdar: pd.DataFrame, well_number: int) -> str:
             f"SUVTRIG '{well_name}' {segment_number} "
             f"= {suvtrig[iaction]} /\n/\n\n"
         )
-        print_df = df_wsegdar[df_wsegdar[Headers.START_SEGMENT_NUMBER] == segment_number]
+        print_df = df_wsegdensity[df_wsegdensity[Headers.START_SEGMENT_NUMBER] == segment_number]
         print_df = print_df[header[iaction]]  # type: ignore
         header_string = Keywords.WELL_SEGMENTS_VALVE + "\n--"
         for item in header[iaction]:
@@ -1483,7 +1542,7 @@ def print_wsegdar(df_wsegdar: pd.DataFrame, well_number: int) -> str:
         act_number = iaction + 1
         act_name = f"D{well_number:03d}{segment_number:03d}{act_number:1d}"
         if len(act_name) > 8:
-            raise CompletorError("Too many wells and/or too many segments with DAR")
+            raise CompletorError("Too many wells and/or too many segments with DENSITY")
         action += (
             f"ACTIONX\n{act_name} 1000000 /\n"
             f"SWHF '{well_name}' {segment_number} "
@@ -1491,7 +1550,7 @@ def print_wsegdar(df_wsegdar: pd.DataFrame, well_number: int) -> str:
             f"SUVTRIG '{well_name}' {segment_number} "
             f"= {suvtrig[iaction]} /\n/\n\n"
         )
-        print_df = df_wsegdar[df_wsegdar[Headers.START_SEGMENT_NUMBER] == segment_number]
+        print_df = df_wsegdensity[df_wsegdensity[Headers.START_SEGMENT_NUMBER] == segment_number]
         print_df = print_df[header[iaction]]  # type: ignore
         header_string = Keywords.WELL_SEGMENTS_VALVE + "\n--"
         for item in header[iaction]:
@@ -1504,18 +1563,180 @@ def print_wsegdar(df_wsegdar: pd.DataFrame, well_number: int) -> str:
     return action
 
 
-def print_wsegaicv(df_wsegaicv: pd.DataFrame, well_number: int) -> str:
-    """Print for AICV devices.
+def print_wseginjv(df_wseginjv: pd.DataFrame, well_number: int) -> str:
+    """Print INJECTION VALVE devices.
+
+    Args:
+        df_wseginjv: Output from function prepare_wseginjv.
+        well_number: Well number.
+
+    Returns:
+        Formatted actions to be included in the output file.
+
+    Raises:
+        CompletorError: If there are to many wells and/or segments with INJECTION VALVE.
+    """
+    header = [
+        [
+            Headers.WELL,
+            Headers.START_SEGMENT_NUMBER,
+            Headers.FLOW_COEFFICIENT,
+            Headers.SECONDARY_FLOW_CROSS_SECTIONAL_AREA,
+            Headers.DEFAULTS,
+            Headers.MAX_FLOW_CROSS_SECTIONAL_AREA,
+        ],
+        [
+            Headers.WELL,
+            Headers.START_SEGMENT_NUMBER,
+            Headers.FLOW_COEFFICIENT,
+            Headers.PRIMARY_FLOW_CROSS_SECTIONAL_AREA,
+            Headers.DEFAULTS,
+            Headers.MAX_FLOW_CROSS_SECTIONAL_AREA,
+        ],
+    ]
+
+    sign = ["<", ">="]
+    suvtrig = ["0", "1"]
+    action = "UDQ\n"
+    for idx in range(df_wseginjv.shape[0]):
+        segment_number = df_wseginjv[Headers.START_SEGMENT_NUMBER].iloc[idx]
+        well_name = df_wseginjv[Headers.WELL].iloc[idx]
+        action += f"  ASSIGN SUVTRIG {well_name} {segment_number} 0 /\n"
+    action += "/\n\n"
+    iaction = 1
+    action += Keywords.WELL_SEGMENTS_VALVE + "\n"
+    header_string = "--"
+    for itm in header[iaction]:
+        header_string += "  " + itm
+    action += header_string.rstrip() + "\n"
+    for idx in range(df_wseginjv.shape[0]):
+        segment_number = df_wseginjv[Headers.START_SEGMENT_NUMBER].iloc[idx]
+        print_df = df_wseginjv[df_wseginjv[Headers.START_SEGMENT_NUMBER] == segment_number]
+        print_df = print_df[header[iaction]]
+        print_df = dataframe_tostring(print_df, True, False, False) + "\n"
+        action += print_df
+    action += "/\n\n"
+    for idx in range(df_wseginjv.shape[0]):
+        segment_number = df_wseginjv[Headers.START_SEGMENT_NUMBER].iloc[idx]
+        well_name = df_wseginjv[Headers.WELL].iloc[idx]
+        # Trigger paramater is segment water rate
+        if df_wseginjv[Headers.TRIGGER_PARAMETER].iloc[idx] == "SWFR":
+            water_segment_rate_cutoff = -1 * df_wseginjv[Headers.TRIGGER_VALUE].iloc[idx]
+            iaction = 0
+            act_number = iaction + 1
+            act_name = f"INJVOP{well_number:03d}{segment_number:03d}{act_number:1d}"
+            if len(act_name) > 13:
+                raise CompletorError("Too many wells and/or too many segments with Injection Valve")
+            action += (
+                f"ACTIONX\n{act_name} 1000000 /\n"
+                f"SWFR '{well_name}' {segment_number} "
+                f"{sign[iaction]} {water_segment_rate_cutoff} AND /\n"
+                f"SUVTRIG '{well_name}' {segment_number} "
+                f"= {suvtrig[iaction]} /\n/\n\n"
+            )
+            print_df = df_wseginjv[df_wseginjv[Headers.START_SEGMENT_NUMBER] == segment_number]
+            print_df = print_df[header[iaction]]  # type: ignore
+            header_string = Keywords.WELL_SEGMENTS_VALVE + "\n--"
+
+            for item in header[iaction]:
+                header_string += "  " + item
+            header_string = header_string.rstrip() + "\n"
+            print_df = header_string + dataframe_tostring(print_df, True, False, False)  # type: ignore
+            print_df += "\n/\n"
+            print_df += f"\nUDQ\n  ASSIGN SUVTRIG {well_name} {segment_number} 1 /\n/\n"
+            action += print_df + "\nENDACTIO\n\n"
+
+            iaction = 1
+            act_number = iaction + 1
+            act_name = f"INJVCL{well_number:03d}{segment_number:03d}{act_number:1d}"
+            if len(act_name) > 13:
+                raise CompletorError("Too many wells and/or too many segments with Injection Valve")
+            action += (
+                f"ACTIONX\n{act_name} 1000000 /\n"
+                f"SWFR '{well_name}' {segment_number} "
+                f"{sign[iaction]} {water_segment_rate_cutoff} AND /\n"
+                f"SUVTRIG '{well_name}' {segment_number} "
+                f"= {suvtrig[iaction]} /\n/\n\n"
+            )
+            print_df = df_wseginjv[df_wseginjv[Headers.START_SEGMENT_NUMBER] == segment_number]
+            print_df = print_df[header[iaction]]  # type: ignore
+            header_string = Keywords.WELL_SEGMENTS_VALVE + "\n--"
+
+            for item in header[iaction]:
+                header_string += "  " + item
+            header_string = header_string.rstrip() + "\n"
+            print_df = header_string + dataframe_tostring(print_df, True, False, False)  # type: ignore
+            print_df += "\n/\n"
+            print_df += f"\nUDQ\n  ASSIGN SUVTRIG {well_name} {segment_number} 0 /\n/\n"
+            action += print_df + "\nENDACTIO\n\n"
+
+        # Trigger parameter is segment pressure drop
+        elif df_wseginjv[Headers.TRIGGER_PARAMETER].iloc[idx] == "SPRD":
+            pressure_drop_cutoff = -1 * df_wseginjv[Headers.TRIGGER_VALUE].iloc[idx]
+            iaction = 0
+            act_number = iaction + 1
+            act_name = f"INJVOP{well_number:03d}{segment_number:03d}{act_number:1d}"
+            if len(act_name) > 13:
+                raise CompletorError("Too many wells and/or too many segments with Injection Valve")
+            action += (
+                f"ACTIONX\n{act_name} 1000000 /\n"
+                f"SPRD '{well_name}' {segment_number} "
+                f"{sign[iaction]} {pressure_drop_cutoff} AND /\n"
+                f"SUVTRIG '{well_name}' {segment_number} "
+                f"= {suvtrig[iaction]} /\n/\n\n"
+            )
+            print_df = df_wseginjv[df_wseginjv[Headers.START_SEGMENT_NUMBER] == segment_number]
+            print_df = print_df[header[iaction]]  # type: ignore
+            header_string = Keywords.WELL_SEGMENTS_VALVE + "\n--"
+
+            for item in header[iaction]:
+                header_string += "  " + item
+            header_string = header_string.rstrip() + "\n"
+            print_df = header_string + dataframe_tostring(print_df, True, False, False)  # type: ignore
+            print_df += "\n/\n"
+            print_df += f"\nUDQ\n  ASSIGN SUVTRIG {well_name} {segment_number} 1 /\n/\n"
+            action += print_df + "\nENDACTIO\n\n"
+
+            iaction = 1
+            act_number = iaction + 1
+            act_name = f"INJVCL{well_number:03d}{segment_number:03d}{act_number:1d}"
+            if len(act_name) > 13:
+                raise CompletorError("Too many wells and/or too many segments with Injection Valve")
+            action += (
+                f"ACTIONX\n{act_name} 1000000 /\n"
+                f"SPRD '{well_name}' {segment_number} "
+                f"{sign[iaction]} {pressure_drop_cutoff} AND /\n"
+                f"SUVTRIG '{well_name}' {segment_number} "
+                f"= {suvtrig[iaction]} /\n/\n\n"
+            )
+            print_df = df_wseginjv[df_wseginjv[Headers.START_SEGMENT_NUMBER] == segment_number]
+            print_df = print_df[header[iaction]]  # type: ignore
+            header_string = Keywords.WELL_SEGMENTS_VALVE + "\n--"
+
+            for item in header[iaction]:
+                header_string += "  " + item
+            header_string = header_string.rstrip() + "\n"
+            print_df = header_string + dataframe_tostring(print_df, True, False, False)  # type: ignore
+            print_df += "\n/\n"
+            print_df += f"\nUDQ\n  ASSIGN SUVTRIG {well_name} {segment_number} 0 /\n/\n"
+            action += print_df + "\nENDACTIO\n\n"
+        else:
+            raise CompletorError("Trigger paramater given is not supported")
+    return action
+
+
+def print_wsegdualrcp(df_wsegdualrcp: pd.DataFrame, well_number: int) -> str:
+    """Print for DUALRCP devices.
 
     Args:
-        df_wsegaicv: Output from function prepare_wsegaicv.
+        df_wsegdualrcp: Output from function prepare_wsegdualrcp.
         well_number: Well number.
 
     Returns:
         Formatted actions to be included in the output file.
 
     Raises:
-        CompletorError: If there are too many wells and/or segments with AICV.
+        CompletorError: If there are too many wells and/or segments with DUALRCP.
     """
     header = [
         [
@@ -1583,17 +1804,17 @@ def print_wsegaicv(df_wsegaicv: pd.DataFrame, well_number: int) -> str:
     sign_gas = ["<", ">="]
     operator = ["AND", "OR"]
     action = ""
-    for idx in range(df_wsegaicv.shape[0]):
-        segment_number = df_wsegaicv[Headers.START_SEGMENT_NUMBER].iloc[idx]
-        well_name = df_wsegaicv[Headers.WELL].iloc[idx]
-        wct = df_wsegaicv[Headers.AICV_WATER_CUT].iloc[idx]
-        ghf = df_wsegaicv[Headers.AICV_GAS_HOLDUP_FRACTION].iloc[idx]
+    for idx in range(df_wsegdualrcp.shape[0]):
+        segment_number = df_wsegdualrcp[Headers.START_SEGMENT_NUMBER].iloc[idx]
+        well_name = df_wsegdualrcp[Headers.WELL].iloc[idx]
+        wct = df_wsegdualrcp[Headers.DUALRCP_WATER_CUT].iloc[idx]
+        ghf = df_wsegdualrcp[Headers.DUALRCP_GAS_HOLDUP_FRACTION].iloc[idx]
         # LOWWCT_LOWGHF
         for iaction in range(2):
             act_number = iaction + 1
             act_name = f"V{well_number:03d}{segment_number:03d}{act_number:1d}"
             if len(act_name) > 8:
-                raise CompletorError("Too many wells and/or too many segments with AICV")
+                raise CompletorError("Too many wells and/or too many segments with DUALRCP")
             action += (
                 f"ACTIONX\n{act_name} 1000000 /\n"
                 f"SUWCT '{well_name}' {segment_number} {sign_water[iaction]} "
@@ -1601,7 +1822,7 @@ def print_wsegaicv(df_wsegaicv: pd.DataFrame, well_number: int) -> str:
                 f"SGHF '{well_name}' {segment_number} {sign_gas[iaction]} {ghf} /\n/\n"
             )
 
-            print_df = df_wsegaicv[df_wsegaicv[Headers.START_SEGMENT_NUMBER] == segment_number]
+            print_df = df_wsegdualrcp[df_wsegdualrcp[Headers.START_SEGMENT_NUMBER] == segment_number]
             print_df = print_df[header[iaction]]
             print_df.columns = new_column
             print_df = Keywords.AUTONOMOUS_INFLOW_CONTROL_DEVICE + "\n" + dataframe_tostring(print_df, True)