move calculat_total_drilling_lentghs from economics to wellbores in prepartion for updating it

Author: malcolm-dsider

src/geophires_x/Economics.py

@@ -1,56 +1,13 @@
 import math
 import sys
-import os
 import numpy as np
 import numpy_financial as npf
 import geophires_x.Model as Model
 from geophires_x.OptionList import Configuration, WellDrillingCostCorrelation, EconomicModel, EndUseOptions, PlantType
 from geophires_x.Parameter import intParameter, floatParameter, OutputParameter, ReadParameter, boolParameter, \
     coerce_int_params_to_enum_values
 from geophires_x.Units import *
-
-
-def calculate_total_drilling_lengths_m(Configuration, numnonverticalsections: int, nonvertical_length_km: float,
-                                       InputDepth_km: float, OutputDepth_km: float, nprod:int, ninj:int) -> tuple:
-    """
-    returns the total length, vertical length, and non-vertical lengths, depending on the configuration
-    :param Configuration: Configuration of the well
-    :type Configuration: :class:`~geophires
-    :param numnonverticalsections: number of non-vertical sections
-    :type numnonverticalsections: int
-    :param nonvertical_length_km: length of non-vertical sections in km
-    :type nonvertical_length_km: float
-    :param InputDepth_km: depth of the well in km
-    :type InputDepth_km: float
-    :param OutputDepth_km: depth of the output end of the well in km, if U shaped, and not horizontal
-    :type OutputDepth_km: float
-    :param nprod: number of production wells
-    :type nprod: int
-    :param ninj: number of injection wells
-    :return: total length, vertical length, and horizontal lengths in meters
-    :rtype: tuple
-    """
-    if Configuration == Configuration.ULOOP:
-        # Total drilling depth of both wells and laterals in U-loop [m]
-        vertical_pipe_length_m = (nprod * InputDepth_km * 1000.0) + (ninj * OutputDepth_km * 1000.0)
-        nonvertical_pipe_length_m = numnonverticalsections * nonvertical_length_km * 1000.0
-    elif Configuration == Configuration.COAXIAL:
-        # Total drilling depth of well and lateral in co-axial case [m]
-        vertical_pipe_length_m = (nprod + ninj) * InputDepth_km * 1000.0
-        nonvertical_pipe_length_m = numnonverticalsections * nonvertical_length_km * 1000.0
-    elif Configuration == Configuration.VERTICAL:
-        # Total drilling depth of well in vertical case [m]
-        vertical_pipe_length_m = (nprod + ninj) * InputDepth_km * 1000.0
-        nonvertical_pipe_length_m = 0.0
-    elif Configuration == Configuration.L:
-        # Total drilling depth of well in L case [m]
-        vertical_pipe_length_m = (nprod + ninj) * InputDepth_km * 1000.0
-        nonvertical_pipe_length_m = numnonverticalsections * nonvertical_length_km * 1000.0
-    else:
-        raise ValueError(f'Invalid Configuration: {Configuration}')
-
-    tot_pipe_length_m = vertical_pipe_length_m + nonvertical_pipe_length_m
-    return tot_pipe_length_m, vertical_pipe_length_m, nonvertical_pipe_length_m
+from geophires_x.WellBores import calculate_total_drilling_lengths_m
 
 
 def calculate_cost_of_one_vertical_well(model: Model, depth_m: float, well_correlation: int,
@@ -2230,12 +2187,12 @@ def Calculate(self, model: Model) -> None:
                 model.wellbores.injection_reservoir_depth.value = input_vert_depth_km
 
                 tot_m, tot_vert_m, tot_horiz_m = calculate_total_drilling_lengths_m(model.wellbores.Configuration.value,
-                                                                      model.wellbores.numnonverticalsections.value,
-                                                                      model.wellbores.Nonvertical_length.value / 1000.0,
-                                                                      input_vert_depth_km,
-                                                                      output_vert_depth_km,
-                                                                      model.wellbores.nprod.value,
-                                                                      model.wellbores.ninj.value)
+                                                                                    model.wellbores.numnonverticalsections.value,
+                                                                                    model.wellbores.Nonvertical_length.value / 1000.0,
+                                                                                    input_vert_depth_km,
+                                                                                    output_vert_depth_km,
+                                                                                    model.wellbores.nprod.value,
+                                                                                    model.wellbores.ninj.value)
 
             else:
                 tot_m = tot_vert_m = model.reserv.depth.quantity().to('km').magnitude

src/geophires_x/SBTWellbores.py

@@ -4,8 +4,8 @@
 from pint.facets.plain import PlainQuantity
 
 from geophires_x.WellBores import (WellBores, RameyCalc, WellPressureDrop, \
-    ProdPressureDropsAndPumpingPowerUsingImpedenceModel, ProdPressureDropAndPumpingPowerUsingIndexes)
-from geophires_x.Economics import calculate_total_drilling_lengths_m
+                                   ProdPressureDropsAndPumpingPowerUsingImpedenceModel,
+                                   ProdPressureDropAndPumpingPowerUsingIndexes, calculate_total_drilling_lengths_m)
 from .Parameter import floatParameter, intParameter, boolParameter, OutputParameter, ReadParameter
 from geophires_x.GeoPHIRESUtils import vapor_pressure_water_kPa, quantity, static_pressure_MPa, \
     heat_capacity_water_J_per_kg_per_K

src/geophires_x/WellBores.py

@@ -1363,3 +1363,46 @@ def Calculate(self, model: Model) -> None:
             self.PumpingPower.value = [0. if x < 0. else x for x in self.PumpingPower.value]
 
         model.logger.info(f'complete {self.__class__.__name__}: {__name__}')
+
+
+def calculate_total_drilling_lengths_m(Configuration, numnonverticalsections: int, nonvertical_length_km: float,
+                                       InputDepth_km: float, OutputDepth_km: float, nprod:int, ninj:int) -> tuple:
+    """
+    returns the total length, vertical length, and non-vertical lengths, depending on the configuration
+    :param Configuration: Configuration of the well
+    :type Configuration: :class:`~geophires
+    :param numnonverticalsections: number of non-vertical sections
+    :type numnonverticalsections: int
+    :param nonvertical_length_km: length of non-vertical sections in km
+    :type nonvertical_length_km: float
+    :param InputDepth_km: depth of the well in km
+    :type InputDepth_km: float
+    :param OutputDepth_km: depth of the output end of the well in km, if U shaped, and not horizontal
+    :type OutputDepth_km: float
+    :param nprod: number of production wells
+    :type nprod: int
+    :param ninj: number of injection wells
+    :return: total length, vertical length, and horizontal lengths in meters
+    :rtype: tuple
+    """
+    if Configuration == Configuration.ULOOP:
+        # Total drilling depth of both wells and laterals in U-loop [m]
+        vertical_pipe_length_m = (nprod * InputDepth_km * 1000.0) + (ninj * OutputDepth_km * 1000.0)
+        nonvertical_pipe_length_m = numnonverticalsections * nonvertical_length_km * 1000.0
+    elif Configuration == Configuration.COAXIAL:
+        # Total drilling depth of well and lateral in co-axial case [m]
+        vertical_pipe_length_m = (nprod + ninj) * InputDepth_km * 1000.0
+        nonvertical_pipe_length_m = numnonverticalsections * nonvertical_length_km * 1000.0
+    elif Configuration == Configuration.VERTICAL:
+        # Total drilling depth of well in vertical case [m]
+        vertical_pipe_length_m = (nprod + ninj) * InputDepth_km * 1000.0
+        nonvertical_pipe_length_m = 0.0
+    elif Configuration == Configuration.L:
+        # Total drilling depth of well in L case [m]
+        vertical_pipe_length_m = (nprod + ninj) * InputDepth_km * 1000.0
+        nonvertical_pipe_length_m = numnonverticalsections * nonvertical_length_km * 1000.0
+    else:
+        raise ValueError(f'Invalid Configuration: {Configuration}')
+
+    tot_pipe_length_m = vertical_pipe_length_m + nonvertical_pipe_length_m
+    return tot_pipe_length_m, vertical_pipe_length_m, nonvertical_pipe_length_m