Added time history plot for geo mech models

ApplicationCode/GeoMech/GeoMechDataModel/CMakeLists.txt

@@ -32,6 +32,8 @@ add_library( ${PROJECT_NAME}
 	RigFemPartGrid.cpp
 	RigFemResultAddress.h
 	RigFemResultPosEnum.h
+    RigFemTimeHistoryResultAccessor.h
+    RigFemTimeHistoryResultAccessor.cpp
 )
 
 target_link_libraries( ${PROJECT_NAME} LibCore cafTensor ResultStatisticsCache)

ApplicationCode/GeoMech/GeoMechDataModel/RigFemTimeHistoryResultAccessor.cpp

@@ -0,0 +1,148 @@
+/////////////////////////////////////////////////////////////////////////////////
+//
+//  Copyright (C) Statoil ASA
+//  Copyright (C) Ceetron Solutions AS
+// 
+//  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 "RigFemTimeHistoryResultAccessor.h"
+
+#include "RigFemPart.h"
+#include "RigFemPartCollection.h"
+#include "RigFemPartGrid.h"
+#include "RigFemPartResultsCollection.h"
+#include "RigFemTypes.h"
+#include "RigGeoMechCaseData.h"
+
+#include <cmath> // Needed for HUGE_VAL on Linux
+
+//--------------------------------------------------------------------------------------------------
+/// 
+//--------------------------------------------------------------------------------------------------
+RigFemTimeHistoryResultAccessor::RigFemTimeHistoryResultAccessor(RigGeoMechCaseData* geomData, RigFemResultAddress femResultAddress,
+    size_t gridIndex, size_t cellIndex, const cvf::Vec3d& intersectionPoint)
+    : m_geoMechCaseData(geomData),
+    m_femResultAddress(femResultAddress),
+    m_gridIndex(gridIndex),
+    m_cellIndex(cellIndex),
+    m_intersectionPoint(intersectionPoint)
+{
+    computeTimeHistoryData();
+}
+
+//--------------------------------------------------------------------------------------------------
+/// 
+//--------------------------------------------------------------------------------------------------
+QString RigFemTimeHistoryResultAccessor::topologyText() const
+{
+    QString text;
+
+    if (m_geoMechCaseData)
+    {
+        RigFemPart* femPart = m_geoMechCaseData->femParts()->part(m_gridIndex);
+        int elementId = femPart->elmId(m_cellIndex);
+        text += QString("Element : Id[%1]").arg(elementId);
+
+        size_t i = 0;
+        size_t j = 0;
+        size_t k = 0;
+        if (m_geoMechCaseData->femParts()->part(m_gridIndex)->structGrid()->ijkFromCellIndex(m_cellIndex, &i, &j, &k))
+        {
+            // Adjust to 1-based Eclipse indexing
+            i++;
+            j++;
+            k++;
+
+            cvf::Vec3d domainCoord = m_intersectionPoint;
+            text += QString(", ijk[%1, %2, %3]").arg(i).arg(j).arg(k);
+
+            QString formattedText;
+            formattedText.sprintf("Intersection point : [E: %.2f, N: %.2f, Depth: %.2f]", domainCoord.x(), domainCoord.y(), -domainCoord.z());
+
+            text += formattedText;
+        }
+    }
+
+    return text;
+}
+
+//--------------------------------------------------------------------------------------------------
+/// 
+//--------------------------------------------------------------------------------------------------
+std::vector<double> RigFemTimeHistoryResultAccessor::timeHistoryValues() const
+{
+    return m_timeHistoryValues;
+}
+
+//--------------------------------------------------------------------------------------------------
+/// 
+//--------------------------------------------------------------------------------------------------
+void RigFemTimeHistoryResultAccessor::computeTimeHistoryData()
+{
+    m_timeHistoryValues.clear();
+    
+    size_t scalarResultIndex = cvf::UNDEFINED_SIZE_T;
+
+    // Compute scalar result index from geometry
+    {
+        RigFemPart* femPart = m_geoMechCaseData->femParts()->part(m_gridIndex);
+        RigElementType elmType =  femPart->elementType(m_cellIndex);
+        const int* elmentConn = femPart->connectivities(m_cellIndex);
+        int elmNodeCount = RigFemTypes::elmentNodeCount(elmType);
+
+        // Find the closest node
+        int closestLocalNode = -1;
+        float minDist = std::numeric_limits<float>::infinity();
+        for (int lNodeIdx = 0; lNodeIdx < elmNodeCount; ++lNodeIdx)
+        {
+            int nodeIdx = elmentConn[lNodeIdx];
+            cvf::Vec3f nodePos = femPart->nodes().coordinates[nodeIdx];
+            float dist = (nodePos - cvf::Vec3f(m_intersectionPoint)).lengthSquared();
+            if (dist < minDist) 
+            {
+                closestLocalNode = lNodeIdx;
+                minDist = dist;
+            }
+        }
+
+        // Create a text showing the results from the closest node
+        if (closestLocalNode >= 0)
+        {
+            int nodeIdx = elmentConn[closestLocalNode];
+            if (m_femResultAddress.resultPosType == RIG_NODAL)
+            {
+                scalarResultIndex = static_cast<size_t>(nodeIdx);
+            } 
+            else 
+            {
+                scalarResultIndex = femPart->elementNodeResultIdx(static_cast<int>(m_cellIndex), closestLocalNode);
+            }
+        }
+    }
+
+    if (scalarResultIndex == cvf::UNDEFINED_SIZE_T) return;
+    
+    RigFemPartResultsCollection* femPartResultsColl = m_geoMechCaseData->femPartResults();
+    for (int frameIdx = 0; frameIdx < femPartResultsColl->frameCount(); frameIdx++)
+    {
+        const std::vector<float>& scalarResults = m_geoMechCaseData->femPartResults()->resultValues(m_femResultAddress, static_cast<int>(m_gridIndex), frameIdx);
+        if (scalarResults.size())
+        {
+            float scalarValue = scalarResults[scalarResultIndex];
+
+            m_timeHistoryValues.push_back(scalarValue);
+        }
+    }
+}

ApplicationCode/GeoMech/GeoMechDataModel/RigFemTimeHistoryResultAccessor.h

@@ -0,0 +1,54 @@
+/////////////////////////////////////////////////////////////////////////////////
+//
+//  Copyright (C) Statoil ASA
+//  Copyright (C) Ceetron Solutions AS
+// 
+//  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.
+//
+/////////////////////////////////////////////////////////////////////////////////
+
+#pragma once
+
+#include "RigFemResultAddress.h"
+
+#include "cvfStructGrid.h"
+#include "cvfVector3.h"
+
+class RigGeoMechCaseData;
+
+
+class RigFemTimeHistoryResultAccessor
+{
+public:
+    RigFemTimeHistoryResultAccessor(RigGeoMechCaseData* geomData, RigFemResultAddress femResultAddress,
+        size_t gridIndex, size_t cellIndex, const cvf::Vec3d& intersectionPoint);
+
+    QString             topologyText() const;
+    std::vector<double> timeHistoryValues() const;
+
+private:
+    void                computeTimeHistoryData();
+
+private:
+    RigGeoMechCaseData*     m_geoMechCaseData;
+    RigFemResultAddress     m_femResultAddress;
+
+    size_t                  m_gridIndex;
+    size_t                  m_cellIndex;
+    size_t                  m_scalarResultIndex;
+
+    cvf::Vec3d              m_intersectionPoint;
+
+    std::vector<double>     m_timeHistoryValues;
+};
+

ApplicationCode/UserInterface/RiuTimeHistoryQwtPlot.cpp

@@ -51,9 +51,9 @@ RiuTimeHistoryQwtPlot::~RiuTimeHistoryQwtPlot()
 //--------------------------------------------------------------------------------------------------
 /// 
 //--------------------------------------------------------------------------------------------------
-void RiuTimeHistoryQwtPlot::addCurve(const QString& curveName, const std::vector<QDateTime>& dateTimes, const std::vector<double>& yValues)
+void RiuTimeHistoryQwtPlot::addCurve(const QString& curveName, const std::vector<QDateTime>& dateTimes, const std::vector<double>& timeHistoryValues)
 {
-    CVF_ASSERT(dateTimes.size() == yValues.size());
+    CVF_ASSERT(dateTimes.size() == timeHistoryValues.size());
 
     QwtPlotCurve* plotCurve = new QwtPlotCurve("Curve 1");
 
@@ -61,7 +61,7 @@ void RiuTimeHistoryQwtPlot::addCurve(const QString& curveName, const std::vector
     for (int i = 0; i < dateTimes.size(); i++)
     {
         double milliSecSinceEpoch = QwtDate::toDouble(dateTimes[i]);
-        points << QPointF(milliSecSinceEpoch, yValues[i]);
+        points << QPointF(milliSecSinceEpoch, timeHistoryValues[i]);
     }
 
     plotCurve->setSamples(points);
@@ -78,6 +78,21 @@ void RiuTimeHistoryQwtPlot::addCurve(const QString& curveName, const std::vector
     this->replot();
 }
 
+//--------------------------------------------------------------------------------------------------
+/// 
+//--------------------------------------------------------------------------------------------------
+void RiuTimeHistoryQwtPlot::addCurve(const QString& curveName, const std::vector<double>& frameTimes, const std::vector<double>& timeHistoryValues)
+{
+    std::vector<QDateTime> dateTimes;
+
+    for (size_t i = 0; i < frameTimes.size(); i++)
+    {
+        dateTimes.push_back(QwtDate::toDateTime(frameTimes[i]));
+    }
+
+    addCurve(curveName, dateTimes, timeHistoryValues);
+}
+
 //--------------------------------------------------------------------------------------------------
 /// 
 //--------------------------------------------------------------------------------------------------

ApplicationCode/UserInterface/RiuTimeHistoryQwtPlot.h

@@ -34,7 +34,9 @@ public:
     RiuTimeHistoryQwtPlot(QWidget* parent = NULL);
     virtual ~RiuTimeHistoryQwtPlot();
 
-    void addCurve(const QString& curveName, const std::vector<QDateTime>& dateTimes, const std::vector<double>& yValues);
+    void addCurve(const QString& curveName, const std::vector<QDateTime>& dateTimes, const std::vector<double>& timeHistoryValues);
+    void addCurve(const QString& curveName, const std::vector<double>& frameTimes, const std::vector<double>& timeHistoryValues);
+    
     void deleteAllCurves();
 
 private:

ApplicationCode/UserInterface/RiuViewerCommands.cpp

@@ -77,6 +77,7 @@
 #include <QMenu>
 #include <QMouseEvent>
 #include <QStatusBar>
+#include "RigFemTimeHistoryResultAccessor.h"
 
 //==================================================================================================
 //
@@ -513,6 +514,12 @@ void RiuViewerCommands::handlePickAction(int winPosX, int winPosY)
             resultInfo = textBuilder.mainResultText();
 
             pickInfo = textBuilder.topologyText(", ");
+
+            if (geomView->cellResult() &&
+                geomView->cellResult()->hasResult())
+            {
+                addTimeHistoryCurve(geomView, gridIndex, cellIndex, localIntersectionPoint);
+            }
         }
     }
     else
@@ -540,7 +547,6 @@ void RiuViewerCommands::addTimeHistoryCurve(RimEclipseView* eclipseView, size_t
     std::vector<QDateTime> timeStepDates = eclipseView->eclipseCase()->reservoirData()->results(porosityModel)->timeStepDates(eclipseView->cellResult()->scalarResultIndex());
 
     RigTimeHistoryResultAccessor timeHistResultAccessor(eclipseView->eclipseCase()->reservoirData(), gridIndex, cellIndex, eclipseView->cellResult()->scalarResultIndex(), porosityModel);
-    timeHistResultAccessor.computeTimeHistoryData();
 
     QString curveName = eclipseView->eclipseCase()->caseUserDescription();
     curveName += " - Result : ";
@@ -548,13 +554,48 @@ void RiuViewerCommands::addTimeHistoryCurve(RimEclipseView* eclipseView, size_t
     curveName += " - ";
     curveName += timeHistResultAccessor.topologyText();
 
-    std::vector<double> yValues = timeHistResultAccessor.timeHistoryValues();
+    std::vector<double> timeHistoryValues = timeHistResultAccessor.timeHistoryValues();
 
-    CVF_ASSERT(timeStepDates.size() == yValues.size());
+    CVF_ASSERT(timeStepDates.size() == timeHistoryValues.size());
 
     RiuMainWindow* mainWnd = RiuMainWindow::instance();
     
-    mainWnd->timeHistoryPlot()->addCurve(curveName, timeStepDates, yValues);
+    mainWnd->timeHistoryPlot()->addCurve(curveName, timeStepDates, timeHistoryValues);
+}
+
+//--------------------------------------------------------------------------------------------------
+/// 
+//--------------------------------------------------------------------------------------------------
+void RiuViewerCommands::addTimeHistoryCurve(RimGeoMechView* geoMechView, size_t gridIndex, size_t cellIndex, const cvf::Vec3d& localIntersectionPoint)
+{
+    if (geoMechView &&
+        geoMechView->cellResult() &&
+        geoMechView->geoMechCase() &&
+        geoMechView->geoMechCase()->geoMechData())
+    {
+        RigFemTimeHistoryResultAccessor timeHistResultAccessor(geoMechView->geoMechCase()->geoMechData(), geoMechView->cellResult->resultAddress(), gridIndex, cellIndex, localIntersectionPoint);
+
+        QString curveName;
+        curveName.append(geoMechView->geoMechCase()->caseUserDescription() + ", ");
+
+        caf::AppEnum<RigFemResultPosEnum> resPosAppEnum = geoMechView->cellResult()->resultPositionType();
+        curveName.append(resPosAppEnum.uiText() + ", ");
+        curveName.append(geoMechView->cellResult()->resultFieldUiName()+ ", ") ;
+        curveName.append(geoMechView->cellResult()->resultComponentUiName() + ":\n");
+        curveName.append(timeHistResultAccessor.topologyText());
+
+        std::vector<double> timeHistoryValues = timeHistResultAccessor.timeHistoryValues();
+        std::vector<double> frameTimes;
+        for (size_t i = 0; i < timeHistoryValues.size(); i++)
+        {
+            frameTimes.push_back(i);
+        }
+
+        CVF_ASSERT(frameTimes.size() == timeHistoryValues.size());
+
+        RiuMainWindow* mainWnd = RiuMainWindow::instance();
+        mainWnd->timeHistoryPlot()->addCurve(curveName, frameTimes, timeHistoryValues);
+    }
 }
 
 //--------------------------------------------------------------------------------------------------

ApplicationCode/UserInterface/RiuViewerCommands.h

@@ -64,6 +64,7 @@ private:
     void            extractIntersectionData(const cvf::HitItemCollection& hitItems, cvf::Vec3d* localIntersectionPoint, cvf::Part** firstPart, uint* firstPartFaceHit, cvf::Part** nncPart, uint* nncPartFaceHit);
     void            updateSelectionFromPickedPart(cvf::Part* part);
     void            addTimeHistoryCurve(RimEclipseView* eclipseView, size_t gridIndex, size_t cellIndex);
+    void            addTimeHistoryCurve(RimGeoMechView* geoMechView, size_t gridIndex, size_t cellIndex, const cvf::Vec3d& localIntersectionPoint);
 
     size_t m_currentGridIdx;
     size_t m_currentCellIndex;