From bde6dc3540edb9b8fa867effbeac62a40364b3de Mon Sep 17 00:00:00 2001
From: Kristian Bendiksen <kristian.bendiksen@gmail.com>
Date: Fri, 19 Mar 2021 13:20:45 +0100
Subject: [PATCH] #7500 StimPlanModelTemplate: Add edit button for pressure
 table.

---
 .../StimPlanModel/RimStimPlanModelTemplate.cpp  | 17 ++++++++++++++++-
 .../StimPlanModel/RimStimPlanModelTemplate.h    |  1 +
 2 files changed, 17 insertions(+), 1 deletion(-)

diff --git a/ApplicationLibCode/ProjectDataModel/StimPlanModel/RimStimPlanModelTemplate.cpp b/ApplicationLibCode/ProjectDataModel/StimPlanModel/RimStimPlanModelTemplate.cpp
index 7f9a3359d1..db77d61e6d 100644
--- a/ApplicationLibCode/ProjectDataModel/StimPlanModel/RimStimPlanModelTemplate.cpp
+++ b/ApplicationLibCode/ProjectDataModel/StimPlanModel/RimStimPlanModelTemplate.cpp
@@ -39,6 +39,8 @@
 #include "RimStimPlanModelPlot.h"
 #include "RimTools.h"
 
+#include "Riu3DMainWindowTools.h"
+
 #include "cafPdmFieldCvfVec3d.h"
 #include "cafPdmFieldScriptingCapabilityCvfVec3d.h"
 
@@ -87,6 +89,9 @@ RimStimPlanModelTemplate::RimStimPlanModelTemplate()
                        "",
                        "" );
     CAF_PDM_InitField( &m_useTableForPressure, "UseForPressure", false, "Use Pressure Table For Pressure", "", "", "" );
+    CAF_PDM_InitField( &m_editPressureTable, "EditPressureTable", false, "Edit", "", "", "" );
+    m_editPressureTable.uiCapability()->setUiEditorTypeName( caf::PdmUiToolButtonEditor::uiEditorTypeName() );
+    m_editPressureTable.uiCapability()->setUiLabelPosition( caf::PdmUiItemInfo::HIDDEN );
 
     CAF_PDM_InitScriptableFieldNoDefault( &m_staticEclipseCase, "StaticEclipseCase", "Static Case", "", "", "" );
 
@@ -232,6 +237,15 @@ void RimStimPlanModelTemplate::fieldChangedByUi( const caf::PdmFieldHandle* chan
             m_timeStep = timeStepCount - 1;
         }
     }
+    else if ( changedField == &m_editPressureTable )
+    {
+        m_editPressureTable = false;
+        if ( m_pressureTable() )
+        {
+            Riu3DMainWindowTools::selectAsCurrentItem( m_pressureTable() );
+        }
+        return;
+    }
 
     changed.send();
 }
@@ -295,7 +309,8 @@ void RimStimPlanModelTemplate::defineUiOrdering( QString uiConfigName, caf::PdmU
     pressureDataSourceGroup->add( &m_dynamicEclipseCase );
     pressureDataSourceGroup->add( &m_timeStep );
     pressureDataSourceGroup->add( &m_initialPressureEclipseCase );
-    pressureDataSourceGroup->add( &m_useTableForInitialPressure );
+    pressureDataSourceGroup->add( &m_useTableForInitialPressure, { true, 2, 1 } );
+    pressureDataSourceGroup->add( &m_editPressureTable, { false, 1, 0 } );
     pressureDataSourceGroup->add( &m_useTableForPressure );
     m_initialPressureEclipseCase.uiCapability()->setUiReadOnly( m_useTableForInitialPressure() );
 
diff --git a/ApplicationLibCode/ProjectDataModel/StimPlanModel/RimStimPlanModelTemplate.h b/ApplicationLibCode/ProjectDataModel/StimPlanModel/RimStimPlanModelTemplate.h
index ae637177e2..6190efb451 100644
--- a/ApplicationLibCode/ProjectDataModel/StimPlanModel/RimStimPlanModelTemplate.h
+++ b/ApplicationLibCode/ProjectDataModel/StimPlanModel/RimStimPlanModelTemplate.h
@@ -148,6 +148,7 @@ private:
     caf::PdmPtrField<RimEclipseCase*>         m_initialPressureEclipseCase;
     caf::PdmField<bool>                       m_useTableForInitialPressure;
     caf::PdmField<bool>                       m_useTableForPressure;
+    caf::PdmField<bool>                       m_editPressureTable;
     caf::PdmPtrField<RimEclipseCase*>         m_staticEclipseCase;
     caf::PdmField<double>                     m_defaultPorosity;
     caf::PdmField<double>                     m_defaultPermeability;