#1185 Added "Plot Production Rates" for sim well in 3D and project tree

ApplicationCode/Commands/FlowCommands/CMakeLists_files.cmake

@@ -9,6 +9,7 @@ ${CEE_CURRENT_LIST_DIR}RicShowWellAllocationPlotFeature.h
 ${CEE_CURRENT_LIST_DIR}RicAddStoredWellAllocationPlotFeature.h
 ${CEE_CURRENT_LIST_DIR}RicShowContributingWellsFromPlotFeature.h
 ${CEE_CURRENT_LIST_DIR}RicShowContributingWellsFeature.h
+${CEE_CURRENT_LIST_DIR}RicPlotProductionRateFeature.h
 )
 
 set (SOURCE_GROUP_SOURCE_FILES
@@ -16,6 +17,7 @@ ${CEE_CURRENT_LIST_DIR}RicShowWellAllocationPlotFeature.cpp
 ${CEE_CURRENT_LIST_DIR}RicAddStoredWellAllocationPlotFeature.cpp
 ${CEE_CURRENT_LIST_DIR}RicShowContributingWellsFromPlotFeature.cpp
 ${CEE_CURRENT_LIST_DIR}RicShowContributingWellsFeature.cpp
+${CEE_CURRENT_LIST_DIR}RicPlotProductionRateFeature.cpp
 )
 
 list(APPEND CODE_HEADER_FILES

ApplicationCode/Commands/FlowCommands/RicPlotProductionRateFeature.cpp

@@ -0,0 +1,212 @@
+/////////////////////////////////////////////////////////////////////////////////
+//
+//  Copyright (C) 2017 Statoil 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 "RicPlotProductionRateFeature.h"
+
+#include "RiaApplication.h"
+#include "RiaPreferences.h"
+
+#include "RifEclipseSummaryAddress.h"
+
+#include "RigSingleWellResultsData.h"
+
+#include "RimEclipseResultCase.h"
+#include "RimEclipseWell.h"
+#include "RimGridSummaryCase.h"
+#include "RimMainPlotCollection.h"
+#include "RimOilField.h"
+#include "RimProject.h"
+#include "RimSummaryCaseCollection.h"
+#include "RimSummaryCurve.h"
+#include "RimSummaryCurveFilter.h"
+#include "RimSummaryPlot.h"
+#include "RimSummaryPlotCollection.h"
+#include "RimView.h"
+
+#include "RiuMainPlotWindow.h"
+#include "RiuMainWindow.h"
+
+#include "cafSelectionManager.h"
+
+#include <QAction>
+
+CAF_CMD_SOURCE_INIT(RicPlotProductionRateFeature, "RicPlotProductionRateFeature");
+
+//--------------------------------------------------------------------------------------------------
+/// 
+//--------------------------------------------------------------------------------------------------
+bool RicPlotProductionRateFeature::isCommandEnabled()
+{
+    std::vector<RimEclipseWell*> collection;
+    caf::SelectionManager::instance()->objectsByType(&collection);
+
+    for (RimEclipseWell* well : collection)
+    {
+        RimGridSummaryCase* gridSummaryCase = RicPlotProductionRateFeature::gridSummaryCaseForWell(well);
+        if (gridSummaryCase)
+        {
+            return true;
+        }
+    }
+
+    return false;
+}
+
+//--------------------------------------------------------------------------------------------------
+/// 
+//--------------------------------------------------------------------------------------------------
+void RicPlotProductionRateFeature::onActionTriggered(bool isChecked)
+{
+    RimProject* project = RiaApplication::instance()->project();
+    CAF_ASSERT(project);
+
+    RimSummaryCaseCollection* sumCaseColl = project->activeOilField() ? project->activeOilField()->summaryCaseCollection() : nullptr;
+    if (!sumCaseColl) return;
+
+    RimMainPlotCollection* mainPlotColl = project->mainPlotCollection();
+    CAF_ASSERT(mainPlotColl);
+
+    RimSummaryPlotCollection* summaryPlotColl = mainPlotColl->summaryPlotCollection();
+    CAF_ASSERT(summaryPlotColl);
+
+    std::vector<RimEclipseWell*> collection;
+    caf::SelectionManager::instance()->objectsByType(&collection);
+
+    RimSummaryPlot* summaryPlotToSelect = nullptr;
+
+    for (RimEclipseWell* well : collection)
+    {
+        RimGridSummaryCase* gridSummaryCase = RicPlotProductionRateFeature::gridSummaryCaseForWell(well);
+        if (!gridSummaryCase) continue;
+
+        QString curveFilterText = "W*PR:";
+        QString description = "Well Production Rates : ";
+
+        RigSingleWellResultsData* wRes = well->wellResults();
+        if (wRes)
+        {
+            RimView* rimView = nullptr;
+            well->firstAncestorOrThisOfTypeAsserted(rimView);
+
+            int currentTimeStep = rimView->currentTimeStep();
+
+            if (wRes->hasWellResult(currentTimeStep))
+            {
+                const RigWellResultFrame& wrf = wRes->wellResultFrame(currentTimeStep);
+
+                if (   wrf.m_productionType == RigWellResultFrame::OIL_INJECTOR
+                    || wrf.m_productionType == RigWellResultFrame::GAS_INJECTOR
+                    || wrf.m_productionType == RigWellResultFrame::WATER_INJECTOR)
+                {
+                    curveFilterText = "W*IR:";
+                    description = "Well Injection Rates : ";
+                }
+            }
+        }
+
+        curveFilterText += well->name();
+        description += well->name();
+
+        RimSummaryPlot* plot = new RimSummaryPlot();
+        summaryPlotColl->summaryPlots().push_back(plot);
+
+        plot->setDescription(description);
+
+        {
+            RimSummaryCurveFilter* newCurveFilter = new RimSummaryCurveFilter();
+            plot->addCurveFilter(newCurveFilter);
+
+            newCurveFilter->createCurves(gridSummaryCase, curveFilterText);
+        }
+
+
+        {
+            RimSummaryCurve* newCurve = new RimSummaryCurve();
+            plot->addCurve(newCurve);
+
+            newCurve->setSummaryCase(gridSummaryCase);
+
+            RifEclipseSummaryAddress addr(  RifEclipseSummaryAddress::SUMMARY_WELL,
+                                            "WBHP",
+                                            -1,
+                                            -1,
+                                            "",
+                                            well->name().toStdString(),
+                                            -1,
+                                            "",
+                                            -1,
+                                            -1,
+                                            -1);
+
+            newCurve->setSummaryAddress(addr);
+            newCurve->setYAxis(RimDefines::PlotAxis::PLOT_AXIS_RIGHT);
+        }
+
+        summaryPlotColl->updateConnectedEditors();
+        plot->loadDataAndUpdate();
+
+        summaryPlotToSelect = plot;
+    }
+
+    if (summaryPlotToSelect)
+    {
+        RiuMainPlotWindow* mainPlotWindow = RiaApplication::instance()->getOrCreateAndShowMainPlotWindow();
+        if (mainPlotWindow)
+        {
+            mainPlotWindow->selectAsCurrentItem(summaryPlotToSelect);
+            mainPlotWindow->setExpanded(summaryPlotToSelect, true);
+
+            mainPlotWindow->tileWindows();
+        }
+    }
+}
+
+//--------------------------------------------------------------------------------------------------
+/// 
+//--------------------------------------------------------------------------------------------------
+void RicPlotProductionRateFeature::setupActionLook(QAction* actionToSetup)
+{
+    //actionToSetup->setIcon(QIcon(":/WellAllocPlot16x16.png"));
+    actionToSetup->setText("Plot Production Rates");
+}
+
+//--------------------------------------------------------------------------------------------------
+/// 
+//--------------------------------------------------------------------------------------------------
+RimGridSummaryCase* RicPlotProductionRateFeature::gridSummaryCaseForWell(RimEclipseWell* well)
+{
+    RimProject* project = RiaApplication::instance()->project();
+    if (!project) return nullptr;
+
+    RimSummaryCaseCollection* sumCaseColl = project->activeOilField() ? project->activeOilField()->summaryCaseCollection() : nullptr;
+    if (!sumCaseColl) return nullptr;
+
+    RimEclipseResultCase* eclCase = nullptr;
+    well->firstAncestorOrThisOfType(eclCase);
+    if (eclCase)
+    {
+        RimGridSummaryCase* gridSummaryCase = dynamic_cast<RimGridSummaryCase*>(sumCaseColl->findSummaryCaseFromEclipseResultCase(eclCase));
+        if (gridSummaryCase)
+        {
+            return gridSummaryCase;
+        }
+    }
+
+    return nullptr;
+}
+

ApplicationCode/Commands/FlowCommands/RicPlotProductionRateFeature.h

@@ -0,0 +1,45 @@
+/////////////////////////////////////////////////////////////////////////////////
+//
+//  Copyright (C) 2017 Statoil 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.
+//
+/////////////////////////////////////////////////////////////////////////////////
+
+#pragma once
+
+#include "cafCmdFeature.h"
+
+#include "RimFlowDiagSolution.h"
+
+class RimGridSummaryCase;
+class RimEclipseWell;
+
+//==================================================================================================
+/// 
+//==================================================================================================
+class RicPlotProductionRateFeature : public caf::CmdFeature
+{
+    CAF_CMD_HEADER_INIT;
+
+protected:
+    // Overrides
+    virtual bool isCommandEnabled() override;
+    virtual void onActionTriggered( bool isChecked ) override;
+    virtual void setupActionLook( QAction* actionToSetup ) override;
+
+private:
+    static RimGridSummaryCase* gridSummaryCaseForWell(RimEclipseWell* well);
+};
+
+

ApplicationCode/ProjectDataModel/RimContextCommandBuilder.cpp

@@ -420,6 +420,7 @@ QStringList RimContextCommandBuilder::commandsFromSelection()
         }
         else if (dynamic_cast<RimEclipseWell*>(uiItem))
         {
+            commandIds << "RicPlotProductionRateFeature";
             commandIds << "Separator";
             commandIds << "RicEclipseWellShowLabelFeature";
             commandIds << "RicEclipseWellShowHeadFeature";

ApplicationCode/UserInterface/RiuViewerCommands.cpp

@@ -311,6 +311,7 @@ void RiuViewerCommands::displayContextMenu(QMouseEvent* event)
                 commandIds << "RicNewSimWellIntersectionFeature";
                 commandIds << "RicShowWellAllocationPlotFeature";
                 commandIds << "RicShowContributingWellsFeature";
+                commandIds << "RicPlotProductionRateFeature";
             }
         }