mirror of
https://github.com/OPM/ResInsight.git
synced 2026-11-01 16:43:03 -06:00
Magne Sjaastad
56e98ba1c4
Migrate all assert macros in ApplicationLibCode to CAF_ASSERT and remove every use of cvfAssert.h. CVF_ASSERT is replaced one to one. CVF_TIGHT_ASSERT is also replaced by CAF_ASSERT, which is semantically exact: CVF_ENABLE_TIGHT_ASSERTS is 1 only under _DEBUG, and that is what CAF_ASSERT now does. The two CVF_FAIL_MSG sites become CAF_ASSERT( false && "message" ), preserving the message with the idiom already used elsewhere in the code base. Counts before and after: CVF_ASSERT 1044 to 0, CVF_TIGHT_ASSERT 66 to 0, CVF_FAIL_MSG 2 to 0, cvfAssert.h references 154 to 0. Include handling: files that included cvfAssert.h directly now include cafAssert.h instead, includes left dead by the migration are removed, and files that were relying on cvfAssert.h transitively get an explicit cafAssert.h. Files that reach cafAssert.h through another caf header are left unchanged; a missing include here is a compile error, not a silently disabled assert. ResultStatisticsCache links only LibCore and therefore had no path to cafAssert.h. Add the cafPdmCore directory as a private include path rather than linking the library, since cafAssert.h is header only. Note that this stops these asserts from firing in Release and RelWithDebInfo, where CVF_ASSERT was previously active.
790 lines
32 KiB
C++
790 lines
32 KiB
C++
/////////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
// Copyright (C) 2016 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 "RicSummaryPlotEditorUi.h"
|
|
|
|
#include "RiaColorTables.h"
|
|
#include "RiaCurveSetDefinition.h"
|
|
#include "RiaGuiApplication.h"
|
|
#include "Summary/RiaSummaryCurveDefinition.h"
|
|
#include "Summary/RiaSummaryPlotTools.h"
|
|
|
|
#include "RiuSummaryCurveDefinitionKeywords.h"
|
|
|
|
#include "RimDeltaSummaryEnsemble.h"
|
|
#include "RimEnsembleCurveSet.h"
|
|
#include "RimEnsembleCurveSetCollection.h"
|
|
#include "RimEnsembleCurveSetColorManager.h"
|
|
#include "RimMainPlotCollection.h"
|
|
#include "RimObservedDataCollection.h"
|
|
#include "RimObservedSummaryData.h"
|
|
#include "RimOilField.h"
|
|
#include "RimProject.h"
|
|
#include "RimSummaryCalculationCollection.h"
|
|
#include "RimSummaryCase.h"
|
|
#include "RimSummaryCaseMainCollection.h"
|
|
#include "RimSummaryCurve.h"
|
|
#include "RimSummaryCurveAppearanceCalculator.h"
|
|
#include "RimSummaryCurveCollection.h"
|
|
#include "RimSummaryEnsemble.h"
|
|
#include "RimSummaryMultiPlot.h"
|
|
#include "RimSummaryPlot.h"
|
|
|
|
#include "RiuPlotMainWindow.h"
|
|
#include "RiuPlotMainWindowTools.h"
|
|
#include "RiuSummaryQwtPlot.h"
|
|
#include "RiuSummaryVectorSelectionUi.h"
|
|
#include "RiuTools.h"
|
|
|
|
#include "cafPdmUiComboBoxEditor.h"
|
|
|
|
#include <QInputDialog>
|
|
#include <QMessageBox>
|
|
|
|
#include <algorithm>
|
|
#include <sstream>
|
|
|
|
CAF_PDM_SOURCE_INIT( RicSummaryPlotEditorUi, "RicSummaryCurveCreator" );
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
const size_t ENSEMBLE_CURVE_COUNT_THRESHOLD = 600;
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
const QString RicSummaryPlotEditorUi::CONFIGURATION_NAME = "CurveCreatorCfg";
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
/// Internal functions
|
|
//--------------------------------------------------------------------------------------------------
|
|
int ensembleCurveCount( const std::set<RiaSummaryCurveDefinition>& allCurveDefs );
|
|
template <typename T>
|
|
std::vector<T> toVector( const std::set<T>& set );
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
RicSummaryPlotEditorUi::RicSummaryPlotEditorUi()
|
|
: m_plotContainer( nullptr )
|
|
, m_closeButtonPressed( false )
|
|
{
|
|
CAF_PDM_InitFieldNoDefault( &m_targetPlot, "TargetPlot", "Target Plot" );
|
|
|
|
m_previewPlot = std::make_unique<RimSummaryPlot>();
|
|
m_previewPlot->setLegendPosition( RiuPlotWidget::Legend::TOP );
|
|
|
|
m_summaryCurveSelectionEditor = std::make_unique<RiuSummaryVectorSelectionWidgetCreator>();
|
|
|
|
m_summaryCurveSelectionEditor->summaryAddressSelection()->setFieldChangedHandler( [this]() { selectionEditorFieldChanged(); } );
|
|
m_summaryCurveSelectionEditor->summaryAddressSelection()->setMultiSelectionMode( true );
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
RicSummaryPlotEditorUi::~RicSummaryPlotEditorUi()
|
|
{
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
RimSummaryPlot* RicSummaryPlotEditorUi::previewPlot() const
|
|
{
|
|
return m_previewPlot.get();
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
void RicSummaryPlotEditorUi::updateFromSummaryPlot( RimSummaryPlot* targetPlot, const std::vector<SummarySource*>& defaultSources )
|
|
{
|
|
if ( targetPlot == nullptr || m_targetPlot != targetPlot )
|
|
{
|
|
resetAllFields();
|
|
}
|
|
|
|
m_targetPlot = targetPlot;
|
|
|
|
if ( m_targetPlot )
|
|
{
|
|
m_plotContainer = m_targetPlot->firstAncestorOfType<RimSummaryMultiPlot>();
|
|
|
|
populateCurveCreator( *m_targetPlot );
|
|
syncPreviewCurvesFromUiSelection();
|
|
setInitialCurveVisibility( targetPlot );
|
|
m_previewPlot->loadDataAndUpdate();
|
|
}
|
|
else
|
|
{
|
|
setDefaultCurveSelection( defaultSources );
|
|
m_previewPlot->enableAutoPlotTitle( true );
|
|
m_previewPlot->setPlotTitleVisible( false );
|
|
syncPreviewCurvesFromUiSelection();
|
|
m_plotContainer = nullptr;
|
|
}
|
|
|
|
caf::PdmUiItem::updateConnectedEditors();
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
void RicSummaryPlotEditorUi::updateFromSummaryMultiPlot( RimSummaryMultiPlot* summaryMultiPlot,
|
|
const std::vector<SummarySource*>& defaultSources )
|
|
{
|
|
resetAllFields();
|
|
|
|
m_plotContainer = summaryMultiPlot;
|
|
|
|
setDefaultCurveSelection( defaultSources );
|
|
|
|
caf::PdmUiItem::updateConnectedEditors();
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
QWidget* RicSummaryPlotEditorUi::addressSelectionWidget( QWidget* parent )
|
|
{
|
|
return m_summaryCurveSelectionEditor->getOrCreateWidget( parent );
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
bool RicSummaryPlotEditorUi::isCloseButtonPressed() const
|
|
{
|
|
return m_closeButtonPressed;
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
void RicSummaryPlotEditorUi::clearCloseButton()
|
|
{
|
|
m_closeButtonPressed = false;
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
void RicSummaryPlotEditorUi::fieldChangedByUi( const caf::PdmFieldHandle* changedField, const QVariant& oldValue, const QVariant& newValue )
|
|
{
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
QList<caf::PdmOptionItemInfo> RicSummaryPlotEditorUi::calculateValueOptions( const caf::PdmFieldHandle* fieldNeedingOptions )
|
|
{
|
|
QList<caf::PdmOptionItemInfo> options;
|
|
|
|
if ( fieldNeedingOptions == &m_targetPlot )
|
|
{
|
|
// Create New Plot item
|
|
QString displayName = "( New Plot )";
|
|
options.push_back( caf::PdmOptionItemInfo( displayName, nullptr ) );
|
|
|
|
if ( m_plotContainer )
|
|
{
|
|
m_plotContainer->summaryPlotItemInfos( &options );
|
|
}
|
|
}
|
|
|
|
return options;
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
void RicSummaryPlotEditorUi::defineUiOrdering( QString uiConfigName, caf::PdmUiOrdering& uiOrdering )
|
|
{
|
|
uiOrdering.add( &m_targetPlot );
|
|
uiOrdering.addNewButton( "OK", [this]() { onOkButtonClicked(); } );
|
|
uiOrdering.addNewButton( "Apply", [this]() { onApplyButtonClicked(); }, { .newRow = false } );
|
|
uiOrdering.addNewButton( "Cancel",
|
|
[this]()
|
|
{
|
|
m_closeButtonPressed = true;
|
|
uiCapability()->updateConnectedEditors();
|
|
},
|
|
{ .newRow = false } );
|
|
|
|
uiOrdering.skipRemainingFields( true );
|
|
|
|
syncPreviewCurvesFromUiSelection();
|
|
|
|
m_summaryCurveSelectionEditor->updateUi( uiConfigName );
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
void RicSummaryPlotEditorUi::syncPreviewCurvesFromUiSelection()
|
|
{
|
|
std::vector<RiaSummaryCurveDefinition> allCurveDefinitionsVector =
|
|
m_summaryCurveSelectionEditor->summaryAddressSelection()->allCurveDefinitionsFromSelection();
|
|
|
|
auto curveSetDefs = m_summaryCurveSelectionEditor->summaryAddressSelection()->allCurveSetDefinitionsFromSelections();
|
|
for ( const auto& curveSet : curveSetDefs )
|
|
{
|
|
allCurveDefinitionsVector.emplace_back( curveSet.ensemble(), curveSet.summaryAddress() );
|
|
}
|
|
|
|
std::set<RiaSummaryCurveDefinition> allCurveDefinitions =
|
|
std::set<RiaSummaryCurveDefinition>( allCurveDefinitionsVector.begin(), allCurveDefinitionsVector.end() );
|
|
|
|
std::vector<RimSummaryCurve*> currentCurvesInPreviewPlot = m_previewPlot->summaryAndEnsembleCurves();
|
|
|
|
{
|
|
std::set<RiaSummaryCurveDefinition> currentCurveDefs;
|
|
std::set<RiaSummaryCurveDefinition> newCurveDefs;
|
|
std::set<RimSummaryCurve*> curvesToDelete;
|
|
|
|
for ( const auto& curve : currentCurvesInPreviewPlot )
|
|
{
|
|
currentCurveDefs.insert( curve->curveDefinition() );
|
|
}
|
|
|
|
{
|
|
// Determine which curves to delete from plot
|
|
std::set<RiaSummaryCurveDefinition> deleteCurveDefs;
|
|
std::set_difference( currentCurveDefs.begin(),
|
|
currentCurveDefs.end(),
|
|
allCurveDefinitions.begin(),
|
|
allCurveDefinitions.end(),
|
|
std::inserter( deleteCurveDefs, deleteCurveDefs.end() ) );
|
|
|
|
for ( const auto& curve : currentCurvesInPreviewPlot )
|
|
{
|
|
RiaSummaryCurveDefinition curveDef = curve->curveDefinition();
|
|
if ( deleteCurveDefs.count( curveDef ) > 0 ) curvesToDelete.insert( curve );
|
|
}
|
|
}
|
|
|
|
{
|
|
// Determine which curves are new since last time
|
|
std::set_difference( allCurveDefinitions.begin(),
|
|
allCurveDefinitions.end(),
|
|
currentCurveDefs.begin(),
|
|
currentCurveDefs.end(),
|
|
std::inserter( newCurveDefs, newCurveDefs.end() ) );
|
|
}
|
|
|
|
// Curve sets to delete
|
|
std::set<RimEnsembleCurveSet*> curveSetsToDelete;
|
|
{
|
|
std::vector<RiaCurveSetDefinition> allCurveSetDefinitionsVector =
|
|
m_summaryCurveSelectionEditor->summaryAddressSelection()->allCurveSetDefinitionsFromSelections();
|
|
std::set<RiaCurveSetDefinition> allCurveSetDefinitions =
|
|
std::set<RiaCurveSetDefinition>( allCurveSetDefinitionsVector.begin(), allCurveSetDefinitionsVector.end() );
|
|
std::vector<RimEnsembleCurveSet*> currentCurveSetsInPreviewPlot = m_previewPlot->curveSets();
|
|
std::set<RiaCurveSetDefinition> currentCurveSetDefs;
|
|
|
|
for ( const auto& curveSet : currentCurveSetsInPreviewPlot )
|
|
{
|
|
RimSummaryEnsemble* ensemble = curveSet->summaryEnsemble();
|
|
currentCurveSetDefs.insert( RiaCurveSetDefinition( ensemble, curveSet->summaryAddressY() ) );
|
|
}
|
|
|
|
if ( allCurveSetDefinitions.size() < currentCurveSetsInPreviewPlot.size() )
|
|
{
|
|
// Determine which curves to delete from plot
|
|
std::set<RiaCurveSetDefinition> deleteCurveSetDefs;
|
|
std::set_difference( currentCurveSetDefs.begin(),
|
|
currentCurveSetDefs.end(),
|
|
allCurveSetDefinitions.begin(),
|
|
allCurveSetDefinitions.end(),
|
|
std::inserter( deleteCurveSetDefs, deleteCurveSetDefs.end() ) );
|
|
|
|
for ( const auto& curveSet : currentCurveSetsInPreviewPlot )
|
|
{
|
|
RimSummaryEnsemble* ensemble = curveSet->summaryEnsemble();
|
|
RiaCurveSetDefinition curveSetDef = RiaCurveSetDefinition( ensemble, curveSet->summaryAddressY() );
|
|
if ( deleteCurveSetDefs.count( curveSetDef ) > 0 ) curveSetsToDelete.insert( curveSet );
|
|
}
|
|
}
|
|
}
|
|
|
|
updatePreviewCurvesFromCurveDefinitions( allCurveDefinitions, newCurveDefs, curvesToDelete, curveSetsToDelete );
|
|
}
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
void RicSummaryPlotEditorUi::updatePreviewCurvesFromCurveDefinitions( const std::set<RiaSummaryCurveDefinition>& allCurveDefsToDisplay,
|
|
const std::set<RiaSummaryCurveDefinition>& curveDefsToAdd,
|
|
const std::set<RimSummaryCurve*>& curvesToDelete,
|
|
const std::set<RimEnsembleCurveSet*>& curveSetsToDelete )
|
|
{
|
|
static bool warningDisplayed = false;
|
|
|
|
std::set<RiaSummaryCurveDefinition> summaryCurveDefsToDisplay;
|
|
|
|
// Ignore curve sets when assigning colors to single summary curves
|
|
for ( const auto& def : allCurveDefsToDisplay )
|
|
{
|
|
if ( !def.isEnsembleCurve() ) summaryCurveDefsToDisplay.insert( def );
|
|
}
|
|
|
|
RimSummaryCurveAppearanceCalculator curveLookCalc( summaryCurveDefsToDisplay );
|
|
|
|
// Delete curves
|
|
if ( !curveSetsToDelete.empty() )
|
|
{
|
|
m_previewPlot->ensembleCurveSetCollection()->deleteCurveSets( toVector( curveSetsToDelete ) );
|
|
}
|
|
if ( !curvesToDelete.empty() )
|
|
{
|
|
m_previewPlot->deleteCurves( toVector( curvesToDelete ) );
|
|
}
|
|
|
|
size_t ensembleCurveCnt = ensembleCurveCount( allCurveDefsToDisplay );
|
|
|
|
bool speedCheatsRequired = ensembleCurveCnt > ENSEMBLE_CURVE_COUNT_THRESHOLD;
|
|
bool legendsVisible = m_previewPlot->legendsVisible();
|
|
|
|
// Disable legends when adding curves
|
|
if ( speedCheatsRequired ) m_previewPlot->setLegendsVisible( false );
|
|
|
|
// Add new curves
|
|
std::map<RimSummaryCurve*, std::pair<bool, bool>> stashedErrorBarsAndLegendVisibility;
|
|
for ( const auto& curveDef : curveDefsToAdd )
|
|
{
|
|
RimSummaryCase* currentCase = curveDef.summaryCaseY();
|
|
|
|
if ( curveDef.isEnsembleCurve() )
|
|
{
|
|
// Find curveSet
|
|
RimEnsembleCurveSet* curveSet = nullptr;
|
|
for ( const auto& cs : m_previewPlot->ensembleCurveSetCollection()->curveSets() )
|
|
{
|
|
if ( cs->summaryEnsemble() == curveDef.ensemble() && cs->summaryAddressY() == curveDef.summaryAddressY() )
|
|
{
|
|
curveSet = cs;
|
|
break;
|
|
}
|
|
}
|
|
if ( !curveSet )
|
|
{
|
|
curveSet = new RimEnsembleCurveSet();
|
|
curveSet->disableStatisticCurves();
|
|
curveSet->setSummaryEnsemble( curveDef.ensemble() );
|
|
|
|
// Do not call setSummaryAddressAndStatisticsFlag() here, as the call to m_statistics->updateAllRequiredEditors(); causes a
|
|
// crash in updateUiOrdering. The statistics curves will be created when the curve set is added to the plot.
|
|
curveSet->setSummaryAddressY( curveDef.summaryAddressY() );
|
|
|
|
// Set single curve set color
|
|
auto allCurveSets = m_previewPlot->ensembleCurveSetCollection()->curveSets();
|
|
size_t colorIndex =
|
|
std::count_if( allCurveSets.begin(),
|
|
allCurveSets.end(),
|
|
[]( RimEnsembleCurveSet* curveSet )
|
|
{ return RimEnsembleCurveSetColorManager::hasSameColorForAllRealizationCurves( curveSet->colorMode() ); } );
|
|
curveSet->setColor( RiaColorTables::summaryCurveDefaultPaletteColors().cycledColor3f( colorIndex ) );
|
|
|
|
// Add curve to plot
|
|
m_previewPlot->ensembleCurveSetCollection()->addCurveSet( curveSet );
|
|
|
|
if ( m_previewPlot->ensembleCurveSetCollection()->curveSets().size() > 1 && ensembleCurveCnt > ENSEMBLE_CURVE_COUNT_THRESHOLD )
|
|
{
|
|
// Toggle off new curve set and display warning
|
|
curveSet->showCurves( false );
|
|
|
|
if ( !warningDisplayed )
|
|
{
|
|
QMessageBox mbox;
|
|
mbox.setIcon( QMessageBox::Icon::Warning );
|
|
mbox.setInformativeText( "The new curve set is hidden. Too many visible curve sets may lead to poor performance" );
|
|
mbox.exec();
|
|
warningDisplayed = true;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
auto curve = RiaSummaryPlotTools::createCurve( currentCase, curveDef.summaryAddressY() );
|
|
|
|
if ( speedCheatsRequired )
|
|
{
|
|
stashedErrorBarsAndLegendVisibility[curve] = std::make_pair( curve->errorBarsVisible(), curve->showInLegend() );
|
|
curve->setErrorBarsVisible( false );
|
|
curve->setShowInLegend( false );
|
|
}
|
|
if ( currentCase && currentCase->isObservedData() ) curve->setSymbolSkipDistance( 0 );
|
|
|
|
m_previewPlot->addCurveNoUpdate( curve );
|
|
curveLookCalc.setupCurveLook( curve );
|
|
}
|
|
}
|
|
|
|
// Enable legends if there is not too many curves
|
|
if ( speedCheatsRequired && !warningDisplayed )
|
|
{
|
|
m_previewPlot->setLegendsVisible( legendsVisible );
|
|
|
|
for ( const auto& curveAndVisibilityPair : stashedErrorBarsAndLegendVisibility )
|
|
{
|
|
auto curve = curveAndVisibilityPair.first;
|
|
auto errorBarsAndLegendVisibility = curveAndVisibilityPair.second;
|
|
curve->setErrorBarsVisible( errorBarsAndLegendVisibility.first );
|
|
curve->setShowInLegend( errorBarsAndLegendVisibility.second );
|
|
}
|
|
}
|
|
m_previewPlot->loadDataAndUpdate();
|
|
m_previewPlot->zoomAll();
|
|
m_previewPlot->updateConnectedEditors();
|
|
m_previewPlot->summaryCurveCollection()->updateConnectedEditors();
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
void RicSummaryPlotEditorUi::defineEditorAttribute( const caf::PdmFieldHandle* field, QString uiConfigName, caf::PdmUiEditorAttribute* attribute )
|
|
{
|
|
if ( &m_targetPlot == field )
|
|
{
|
|
caf::PdmUiComboBoxEditorAttribute* attrib = dynamic_cast<caf::PdmUiComboBoxEditorAttribute*>( attribute );
|
|
if ( attrib )
|
|
{
|
|
attrib->adjustWidthToContents = true;
|
|
}
|
|
}
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
/// Populate curve creator from the given curve collection
|
|
//--------------------------------------------------------------------------------------------------
|
|
void RicSummaryPlotEditorUi::populateCurveCreator( const RimSummaryPlot& sourceSummaryPlot )
|
|
{
|
|
std::vector<RiaSummaryCurveDefinition> curveDefs;
|
|
|
|
m_previewPlot->deleteAllSummaryCurves();
|
|
m_previewPlot->ensembleCurveSetCollection()->deleteAllCurveSets();
|
|
|
|
for ( const auto& curve : sourceSummaryPlot.summaryCurves() )
|
|
{
|
|
curveDefs.push_back( curve->curveDefinition() );
|
|
|
|
// Copy curve object to the preview plot
|
|
copyCurveAndAddToPlot( curve, m_previewPlot.get(), true );
|
|
}
|
|
|
|
RimEnsembleCurveSetCollection* previewCurveSetColl = m_previewPlot->ensembleCurveSetCollection();
|
|
for ( const auto& curveSet : sourceSummaryPlot.ensembleCurveSetCollection()->curveSets() )
|
|
{
|
|
RimEnsembleCurveSet* newCurveSet = curveSet->clone();
|
|
newCurveSet->disableStatisticCurves();
|
|
previewCurveSetColl->addCurveSet( newCurveSet );
|
|
|
|
RimSummaryEnsemble* ensemble = curveSet->summaryEnsemble();
|
|
curveDefs.emplace_back( ensemble, curveSet->summaryAddressY() );
|
|
}
|
|
|
|
m_previewPlot->copyAxisPropertiesFromOther( sourceSummaryPlot );
|
|
m_previewPlot->enableAutoPlotTitle( sourceSummaryPlot.autoPlotTitle() );
|
|
m_previewPlot->updatePlotTitle();
|
|
m_previewPlot->updateAxes();
|
|
|
|
if ( curveDefs.empty() )
|
|
{
|
|
auto sumCases = RimProject::current()->allSummaryCases();
|
|
if ( !sumCases.empty() )
|
|
{
|
|
RifEclipseSummaryAddress defaultAdr;
|
|
RiaSummaryCurveDefinition curveDef( sumCases.front(), defaultAdr, false );
|
|
curveDefs.push_back( curveDef );
|
|
}
|
|
}
|
|
|
|
m_summaryCurveSelectionEditor->summaryAddressSelection()->setSelectedCurveDefinitions( curveDefs );
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
/// Copy curves from preview plot to target plot
|
|
//--------------------------------------------------------------------------------------------------
|
|
void RicSummaryPlotEditorUi::updateTargetPlot()
|
|
{
|
|
if ( !m_targetPlot ) return;
|
|
|
|
m_targetPlot->deleteAllSummaryCurves();
|
|
m_targetPlot->ensembleCurveSetCollection()->deleteAllCurveSets();
|
|
|
|
// Add edited curves to target plot
|
|
for ( const auto& editedCurve : m_previewPlot->summaryCurves() )
|
|
{
|
|
if ( !editedCurve->isChecked() )
|
|
{
|
|
continue;
|
|
}
|
|
copyCurveAndAddToPlot( editedCurve, m_targetPlot );
|
|
}
|
|
|
|
for ( const auto& editedCurveSet : m_previewPlot->ensembleCurveSetCollection()->curveSets() )
|
|
{
|
|
if ( !editedCurveSet->isCurvesVisible() )
|
|
{
|
|
continue;
|
|
}
|
|
|
|
RimEnsembleCurveSet* newCurveSet = editedCurveSet->clone();
|
|
m_targetPlot->ensembleCurveSetCollection()->addCurveSet( newCurveSet );
|
|
newCurveSet->setParentPlotNoReplot( m_targetPlot->plotWidget() );
|
|
}
|
|
|
|
m_targetPlot->loadDataAndUpdate();
|
|
|
|
m_targetPlot->updateConnectedEditors();
|
|
m_targetPlot->curvesChanged.send();
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
void RicSummaryPlotEditorUi::copyCurveAndAddToPlot( const RimSummaryCurve* curve, RimSummaryPlot* plot, bool forceVisible )
|
|
{
|
|
auto curveCopy = curve->copyObject<RimSummaryCurve>();
|
|
CAF_ASSERT( curveCopy );
|
|
|
|
if ( forceVisible )
|
|
{
|
|
curveCopy->setCheckState( true );
|
|
}
|
|
|
|
curveCopy->setSummaryCaseY( curve->summaryCaseY() );
|
|
|
|
bool autoAssignAxis = true;
|
|
plot->addCurveNoUpdate( curveCopy, autoAssignAxis );
|
|
|
|
curveCopy->initAfterReadRecursively();
|
|
curveCopy->loadDataAndUpdate( false );
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
void RicSummaryPlotEditorUi::setDefaultCurveSelection( const std::vector<SummarySource*>& defaultSources )
|
|
{
|
|
m_summaryCurveSelectionEditor->summaryAddressSelection()->setDefaultSelection( defaultSources );
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
void RicSummaryPlotEditorUi::resetAllFields()
|
|
{
|
|
std::vector<RiaSummaryCurveDefinition> curveDefinitions;
|
|
m_summaryCurveSelectionEditor->summaryAddressSelection()->setSelectedCurveDefinitions( curveDefinitions );
|
|
|
|
m_previewPlot->deleteAllSummaryCurves();
|
|
m_targetPlot = nullptr;
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
void RicSummaryPlotEditorUi::applyAppearanceToAllPreviewCurves()
|
|
{
|
|
std::set<RiaSummaryCurveDefinition> allCurveDefs = m_previewPlot->summaryAndEnsembleCurveDefinitions();
|
|
|
|
RimSummaryCurveAppearanceCalculator curveLookCalc( allCurveDefs );
|
|
|
|
// Summary curves
|
|
for ( auto& curve : m_previewPlot->summaryCurves() )
|
|
{
|
|
curve->resetAppearance();
|
|
curveLookCalc.setupCurveLook( curve );
|
|
}
|
|
|
|
// Ensemble curve sets
|
|
int colorIndex = 0;
|
|
for ( auto& curveSet : m_previewPlot->ensembleCurveSetCollection()->curveSets() )
|
|
{
|
|
if ( RimEnsembleCurveSetColorManager::hasSameColorForAllRealizationCurves( curveSet->colorMode() ) )
|
|
{
|
|
curveSet->setColor( RiaColorTables::summaryCurveDefaultPaletteColors().cycledColor3f( colorIndex++ ) );
|
|
}
|
|
}
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
void RicSummaryPlotEditorUi::createNewPlot()
|
|
{
|
|
RimSummaryPlot* newSummaryPlot = nullptr;
|
|
|
|
if ( !m_plotContainer )
|
|
{
|
|
std::vector<RimSummaryPlot*> plots;
|
|
m_plotContainer = RiaSummaryPlotTools::createAndAppendSummaryMultiPlot( plots );
|
|
}
|
|
|
|
if ( m_plotContainer )
|
|
{
|
|
newSummaryPlot = new RimSummaryPlot();
|
|
newSummaryPlot->dockAsPlotWindow();
|
|
newSummaryPlot->enableAutoPlotTitle( true );
|
|
m_plotContainer->addPlot( newSummaryPlot );
|
|
}
|
|
|
|
if ( newSummaryPlot )
|
|
{
|
|
newSummaryPlot->loadDataAndUpdate();
|
|
|
|
if ( m_plotContainer )
|
|
{
|
|
m_plotContainer->updateConnectedEditors();
|
|
}
|
|
|
|
m_targetPlot = newSummaryPlot;
|
|
|
|
RiuPlotMainWindow* mainPlotWindow = RiaGuiApplication::instance()->mainPlotWindow();
|
|
mainPlotWindow->updateMultiPlotToolBar();
|
|
}
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
void RicSummaryPlotEditorUi::updateCurveNames()
|
|
{
|
|
for ( RimSummaryCurve* curve : m_previewPlot->summaryCurves() )
|
|
{
|
|
curve->updateCurveNameNoLegendUpdate();
|
|
}
|
|
|
|
if ( m_previewPlot && m_previewPlot->plotWidget() ) m_previewPlot->updateLegend();
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
bool RicSummaryPlotEditorUi::isObservedData( RimSummaryCase* sumCase ) const
|
|
{
|
|
return dynamic_cast<RimObservedSummaryData*>( sumCase ) != nullptr;
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
void RicSummaryPlotEditorUi::selectionEditorFieldChanged()
|
|
{
|
|
syncPreviewCurvesFromUiSelection();
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
void RicSummaryPlotEditorUi::setInitialCurveVisibility( const RimSummaryPlot* targetPlot )
|
|
{
|
|
// Set visibility for imported curves which were not checked in source plot
|
|
std::set<std::pair<RimSummaryCase*, RifEclipseSummaryAddress>> sourceCurveDefs;
|
|
for ( const auto& curve : targetPlot->summaryCurves() )
|
|
{
|
|
sourceCurveDefs.insert( std::make_pair( curve->summaryCaseY(), curve->summaryAddressY() ) );
|
|
}
|
|
|
|
for ( const auto& curve : m_previewPlot->summaryCurves() )
|
|
{
|
|
auto curveDef = std::make_pair( curve->summaryCaseY(), curve->summaryAddressY() );
|
|
if ( sourceCurveDefs.count( curveDef ) == 0 )
|
|
{
|
|
curve->setCheckState( false );
|
|
}
|
|
}
|
|
|
|
std::set<std::pair<RimSummaryEnsemble*, RifEclipseSummaryAddress>> sourceCurveSetDefs;
|
|
for ( const auto& curveSet : targetPlot->ensembleCurveSetCollection()->curveSets() )
|
|
{
|
|
sourceCurveSetDefs.insert( std::make_pair( curveSet->summaryEnsemble(), curveSet->summaryAddressY() ) );
|
|
}
|
|
|
|
for ( const auto& curveSet : m_previewPlot->ensembleCurveSetCollection()->curveSets() )
|
|
{
|
|
auto curveDef = std::make_pair( curveSet->summaryEnsemble(), curveSet->summaryAddressY() );
|
|
if ( sourceCurveSetDefs.count( curveDef ) == 0 )
|
|
{
|
|
curveSet->showCurves( false );
|
|
}
|
|
}
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
void RicSummaryPlotEditorUi::onOkButtonClicked()
|
|
{
|
|
if ( m_targetPlot == nullptr ) createNewPlot();
|
|
updateTargetPlot();
|
|
|
|
m_closeButtonPressed = true;
|
|
|
|
RiuPlotMainWindowTools::showPlotMainWindow();
|
|
RiuPlotMainWindowTools::selectAsCurrentItem( m_targetPlot );
|
|
RiuPlotMainWindowTools::setExpanded( m_targetPlot );
|
|
|
|
caf::PdmField<bool>* field = dynamic_cast<caf::PdmField<bool>*>( m_targetPlot->uiCapability()->objectToggleField() );
|
|
if ( field ) field->setValueWithFieldChanged( true );
|
|
|
|
RiuPlotMainWindow* mainPlotWindow = RiaGuiApplication::instance()->mainPlotWindow();
|
|
mainPlotWindow->updateMultiPlotToolBar();
|
|
|
|
uiCapability()->updateConnectedEditors();
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
void RicSummaryPlotEditorUi::onApplyButtonClicked()
|
|
{
|
|
if ( m_targetPlot == nullptr ) createNewPlot();
|
|
updateTargetPlot();
|
|
|
|
caf::PdmField<bool>* field = dynamic_cast<caf::PdmField<bool>*>( m_targetPlot->uiCapability()->objectToggleField() );
|
|
if ( field ) field->setValueWithFieldChanged( true );
|
|
|
|
RiuPlotMainWindow* mainPlotWindow = RiaGuiApplication::instance()->mainPlotWindow();
|
|
mainPlotWindow->updateMultiPlotToolBar();
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
int ensembleCurveCount( const std::set<RiaSummaryCurveDefinition>& allCurveDefs )
|
|
{
|
|
return std::count_if( allCurveDefs.begin(),
|
|
allCurveDefs.end(),
|
|
[]( const RiaSummaryCurveDefinition& def ) { return def.isEnsembleCurve(); } );
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
template <typename T>
|
|
std::vector<T> toVector( const std::set<T>& set )
|
|
{
|
|
return std::vector<T>( set.begin(), set.end() );
|
|
}
|