Merge pull request #7960 from OPM/geomech_WIA4

GeoMech: Well Integrity Analysis
2025-02-25 18:55:39 -06:00 · 2021-09-09 17:34:27 +02:00
parent c603e6fe90
commit a5b80c649b
49 changed files with 2778 additions and 30 deletions
--- a/ApplicationLibCode/ProjectDataModel/WellPath/RimWellIADataAccess.cpp
+++ b/ApplicationLibCode/ProjectDataModel/WellPath/RimWellIADataAccess.cpp
@@ -0,0 +1,130 @@
+/////////////////////////////////////////////////////////////////////////////////
+//
+//  Copyright (C) 2021 -    Equinor ASA
+//
+//  ResInsight is free software: you can redistribute it and/or modify
+//  it under the terms of the GNU General Public License as published by
+//  the Free Software Foundation, either version 3 of the License, or
+//  (at your option) any later version.
+//
+//  ResInsight is distributed in the hope that it will be useful, but WITHOUT ANY
+//  WARRANTY; without even the implied warranty of MERCHANTABILITY or
+//  FITNESS FOR A PARTICULAR PURPOSE.
+//
+//  See the GNU General Public License at <http://www.gnu.org/licenses/gpl.html>
+//  for more details.
+//
+/////////////////////////////////////////////////////////////////////////////////
+
+#include "RimWellIADataAccess.h"
+
+#include "RigFemClosestResultIndexCalculator.h"
+#include "RigFemPartCollection.h"
+#include "RigFemPartResultsCollection.h"
+#include "RigGeoMechCaseData.h"
+#include "RimGeoMechCase.h"
+
+#include "../cafHexInterpolator/cafHexInterpolator.h" // Use relative path, as this is a header only file not part of a library
+
+#include "cvfBoundingBox.h"
+
+//--------------------------------------------------------------------------------------------------
+///
+//--------------------------------------------------------------------------------------------------
+RimWellIADataAccess::RimWellIADataAccess( RimGeoMechCase* thecase )
+    : m_case( thecase )
+    , m_caseData( nullptr )
+{
+    if ( m_case ) m_caseData = m_case->geoMechData();
+}
+
+//--------------------------------------------------------------------------------------------------
+///
+//--------------------------------------------------------------------------------------------------
+RimWellIADataAccess::~RimWellIADataAccess()
+{
+}
+
+//--------------------------------------------------------------------------------------------------
+///
+//--------------------------------------------------------------------------------------------------
+int RimWellIADataAccess::resultIndex( RigFemResultPosEnum resultType, cvf::Vec3d position )
+{
+    int closestCell = elementIndex( position );
+
+    if ( closestCell < 0 ) return -1;
+
+    RigFemClosestResultIndexCalculator closestIndexCalc( m_caseData->femParts()->part( 0 ), resultType, closestCell, -1, position );
+
+    return closestIndexCalc.resultIndexToClosestResult();
+}
+
+//--------------------------------------------------------------------------------------------------
+///
+//--------------------------------------------------------------------------------------------------
+int RimWellIADataAccess::elementIndex( cvf::Vec3d position )
+{
+    cvf::BoundingBox bb;
+    bb.add( position );
+
+    std::vector<size_t> closeCells;
+    m_caseData->femParts()->part( 0 )->findIntersectingCells( bb, &closeCells );
+    if ( closeCells.size() == 0 ) return -1;
+
+    return (int)closeCells[0];
+}
+
+//--------------------------------------------------------------------------------------------------
+///
+//--------------------------------------------------------------------------------------------------
+double RimWellIADataAccess::resultValue( QString             fieldName,
+                                         QString             componentName,
+                                         RigFemResultPosEnum resultType,
+                                         size_t              resultIndex,
+                                         int                 timeStep )
+{
+    RigFemResultAddress address( resultType, fieldName.toStdString(), componentName.toStdString() );
+
+    const std::vector<float>& scalarResults = m_caseData->femPartResults()->resultValues( address, 0, timeStep );
+
+    if ( resultIndex < scalarResults.size() ) return scalarResults[resultIndex];
+
+    return 0.0;
+}
+
+//--------------------------------------------------------------------------------------------------
+///
+//--------------------------------------------------------------------------------------------------
+double RimWellIADataAccess::interpolatedResultValue( QString             fieldName,
+                                                     QString             componentName,
+                                                     RigFemResultPosEnum resultType,
+                                                     cvf::Vec3d          position,
+                                                     int                 timeStep )
+{
+    RigFemResultAddress address( resultType, fieldName.toStdString(), componentName.toStdString() );
+
+    int elmIdx = elementIndex( position );
+
+    RigFemPart*    femPart      = m_caseData->femParts()->part( 0 );
+    RigElementType elmType      = femPart->elementType( elmIdx );
+    const int*     elementConn  = femPart->connectivities( elmIdx );
+    int            elmNodeCount = RigFemTypes::elementNodeCount( elmType );
+
+    const std::vector<float>& scalarResults = m_caseData->femPartResults()->resultValues( address, 0, timeStep );
+
+    std::array<double, 8>     nodeResults;
+    std::array<cvf::Vec3d, 8> nodeCorners;
+
+    for ( int lNodeIdx = 0; lNodeIdx < elmNodeCount; ++lNodeIdx )
+    {
+        int    nodeIdx = elementConn[lNodeIdx];
+        size_t resIdx  = femPart->resultValueIdxFromResultPosType( resultType, elmIdx, lNodeIdx );
+        if ( resIdx >= scalarResults.size() )
+            nodeResults[lNodeIdx] = 0.0;
+        else
+            nodeResults[lNodeIdx] = scalarResults[resIdx];
+        nodeCorners[lNodeIdx] = cvf::Vec3d( femPart->nodes().coordinates[nodeIdx] );
+    }
+
+    return caf::HexInterpolator::interpolateHex( nodeCorners, nodeResults, position );
+}