Skip to content

Commit

Permalink
#10568 StimPlan Template: Add python api for appending Pressure Table…
Browse files Browse the repository at this point in the history
… Items
  • Loading branch information
kriben committed Sep 6, 2023
1 parent a4e3e26 commit 3442d88
Show file tree
Hide file tree
Showing 4 changed files with 131 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -26,6 +26,7 @@ set(SOURCE_GROUP_HEADER_FILES
${CMAKE_CURRENT_LIST_DIR}/RimcEclipseCase.h
${CMAKE_CURRENT_LIST_DIR}/RimcEclipseStatisticsCase.h
${CMAKE_CURRENT_LIST_DIR}/RimcIdenticalGridCaseGroup.h
${CMAKE_CURRENT_LIST_DIR}/RimcPressureTable.h
)

set(SOURCE_GROUP_SOURCE_FILES
Expand Down Expand Up @@ -56,6 +57,7 @@ set(SOURCE_GROUP_SOURCE_FILES
${CMAKE_CURRENT_LIST_DIR}/RimcEclipseCase.cpp
${CMAKE_CURRENT_LIST_DIR}/RimcEclipseStatisticsCase.cpp
${CMAKE_CURRENT_LIST_DIR}/RimcIdenticalGridCaseGroup.cpp
${CMAKE_CURRENT_LIST_DIR}/RimcPressureTable.cpp
)

list(APPEND CODE_HEADER_FILES ${SOURCE_GROUP_HEADER_FILES})
Expand Down
70 changes: 70 additions & 0 deletions ApplicationLibCode/ProjectDataModelCommands/RimcPressureTable.cpp
Original file line number Diff line number Diff line change
@@ -0,0 +1,70 @@
/////////////////////////////////////////////////////////////////////////////////
//
// Copyright (C) 2023- 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 "RimcPressureTable.h"

#include "RimPressureTable.h"
#include "RimPressureTableItem.h"

#include "cafPdmAbstractFieldScriptingCapability.h"
#include "cafPdmFieldScriptingCapability.h"

CAF_PDM_OBJECT_METHOD_SOURCE_INIT( RimPressureTable, RimcPressureTable_addPressure, "AddPressure" );

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
RimcPressureTable_addPressure::RimcPressureTable_addPressure( caf::PdmObjectHandle* self )
: caf::PdmObjectMethod( self )
{
CAF_PDM_InitObject( "Add pressure data", "", "", "Add pressure data to pressure table." );

CAF_PDM_InitScriptableFieldNoDefault( &m_depth, "Depth", "", "", "", "Depth: TVDMSL [m]" );
CAF_PDM_InitScriptableFieldNoDefault( &m_initialPressure, "InitialPressure", "", "", "", "Initial Pressure [Bar]" );
CAF_PDM_InitScriptableFieldNoDefault( &m_pressure, "Pressure", "", "", "", "Pressure [Bar]" );
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
caf::PdmObjectHandle* RimcPressureTable_addPressure::execute()
{
RimPressureTableItem* pressureTableItem = new RimPressureTableItem;
pressureTableItem->setValues( m_depth, m_initialPressure, m_pressure );

RimPressureTable* pressureTable = self<RimPressureTable>();
pressureTable->insertItem( nullptr, pressureTableItem );

pressureTable->updateAllRequiredEditors();
return pressureTableItem;
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
bool RimcPressureTable_addPressure::resultIsPersistent() const
{
return true;
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
std::unique_ptr<caf::PdmObjectHandle> RimcPressureTable_addPressure::defaultResult() const
{
return std::unique_ptr<caf::PdmObjectHandle>( new RimPressureTableItem );
}
45 changes: 45 additions & 0 deletions ApplicationLibCode/ProjectDataModelCommands/RimcPressureTable.h
Original file line number Diff line number Diff line change
@@ -0,0 +1,45 @@
/////////////////////////////////////////////////////////////////////////////////
//
// Copyright (C) 2023- 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.
//
/////////////////////////////////////////////////////////////////////////////////

#pragma once

#include "RimPressureTable.h"

#include "cafPdmField.h"
#include "cafPdmObjectHandle.h"
#include "cafPdmObjectMethod.h"

//==================================================================================================
///
//==================================================================================================
class RimcPressureTable_addPressure : public caf::PdmObjectMethod
{
CAF_PDM_HEADER_INIT;

public:
RimcPressureTable_addPressure( caf::PdmObjectHandle* self );

caf::PdmObjectHandle* execute() override;
bool resultIsPersistent() const override;
std::unique_ptr<PdmObjectHandle> defaultResult() const override;

private:
caf::PdmField<double> m_depth;
caf::PdmField<double> m_initialPressure;
caf::PdmField<double> m_pressure;
};
Original file line number Diff line number Diff line change
Expand Up @@ -59,6 +59,20 @@
non_net_layers.facies = "Shale"
non_net_layers.update()

# Add some pressure table items
pressure_table = stim_plan_model_template.pressure_table()
pressure_table.add_pressure(depth=2800.0, initial_pressure=260.0, pressure=261.0)
pressure_table.add_pressure(depth=3000.0, initial_pressure=270.0, pressure=273.0)
pressure_table.add_pressure(depth=3400.0, initial_pressure=274.0, pressure=276.0)
pressure_table.add_pressure(depth=3800.0, initial_pressure=276.0, pressure=280.0)

print("Pressure table ({} items)".format(len(pressure_table.items())))
for item in pressure_table.items():
print(
"TDVMSL [m]: {} Initial Pressure: {} Pressure: {}".format(
item.depth, item.initial_pressure, item.pressure
)
)

# Add some scaling factors
elastic_properties = stim_plan_model_template.elastic_properties()
Expand Down

0 comments on commit 3442d88

Please sign in to comment.