upd files

Author: illarionovEA

deepfield/field/utils/faults_load_utils.py

@@ -1,3 +1,4 @@
+
 import logging
 from typing import cast
 from resdp import DataType

deepfield/field/utils/load_welltrack.py

@@ -316,7 +316,7 @@ def make_patch_spines_invisible(ax):
     for spine in ax.spines.values():
         spine.set_visible(False)
 
-    
+
 def plot_table_1d(table, figsize=None):
     """
     Plot table with 1-dimensional domain.

deepfield/field/utils/plot_utils.py

@@ -1,17 +1,39 @@
+"""Wells utils."""
 import logging
 from collections.abc import Iterable
 from typing import cast
 import warnings
 from numpy.typing import NDArray
 import numpy as np
+import pandas as pd
+
 import resdp
 import resdp.binary
-import pandas as pd
+from resdp import DataType
 
 
+def load_welltrack(data: DataType, binary_data: resdp.binary.BinaryData, logger: logging.Logger) -> pd.DataFrame | None:
+    """Load welltrack."""
+    _ = binary_data, logger
+    section = 'SCHEDULE'
+    res: list[pd.DataFrame] = []
+    if not section in data:
+        return None
+    for key, val in data[section]:
+        if key == 'WELLTRACK':
+            assert isinstance(val, tuple)
+            assert len(val) == 2
+            assert isinstance(val[0], str)
+            assert isinstance(val[1], pd.DataFrame)
+            res.append(cast(pd.DataFrame, val[1]).assign(WELL=cast(str, val[0])))
+    if not res:
+        return None
+    return pd.concat(res)
+
 def load_results(_data: resdp.DataType,
                  binary_data: resdp.binary.BinaryData | None,
                  logger: logging.Logger) -> pd.DataFrame | None:
+    """Load results."""
     _ = _data, logger
     if binary_data is None:
         return None

deepfield/field/utils/load_results.py deepfield/field/utils/wells_utils.pydeepfield/field/utils/load_results.py renamed to deepfield/field/utils/wells_utils.py

@@ -18,8 +18,7 @@
 from .base_component import Attribute, T
 from .grids import OrthogonalGrid
 
-from .utils.load_welltrack import load_welltrack
-from .utils.load_results import load_results
+from .utils.wells_utils import load_results, load_welltrack
 from .utils.grid_raycasting import get_wellblocks_vtk, get_wellblocks_compdat
 from .utils.decorators import apply_to_each_node