mirror of
https://github.com/OPM/ResInsight.git
synced 2025-02-25 18:55:39 -06:00
87bc6acd65
Added a new feature for editing style sheets and variable colors and see immediately the result. Made Qwt plots (and items) stylable. Added icons, improved styling possibilities of QMinimizePanel, fixed minor bugs in RicThemeColorEditorFeature.
395 lines
15 KiB
C++
395 lines
15 KiB
C++
/////////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
// 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 "RiuFlowCharacteristicsPlot.h"
|
|
|
|
#include "RiaColorTables.h"
|
|
#include "RiaFeatureCommandContext.h"
|
|
#include "RiaFontCache.h"
|
|
#include "RiaPreferences.h"
|
|
|
|
#include "RimFlowCharacteristicsPlot.h"
|
|
|
|
#include "RiuQwtPlotCurve.h"
|
|
#include "RiuQwtPlotTools.h"
|
|
#include "RiuQwtPlotWheelZoomer.h"
|
|
#include "RiuQwtPlotZoomer.h"
|
|
#include "RiuResultQwtPlot.h"
|
|
|
|
#include "cvfColor3.h"
|
|
|
|
#include "cafCmdFeatureMenuBuilder.h"
|
|
|
|
#include "qwt_date.h"
|
|
#include "qwt_legend.h"
|
|
#include "qwt_plot.h"
|
|
#include "qwt_plot_zoomer.h"
|
|
#include "qwt_symbol.h"
|
|
|
|
#include <QBoxLayout>
|
|
#include <QContextMenuEvent>
|
|
#include <QDateTime>
|
|
#include <QLabel>
|
|
#include <QMenu>
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
RiuFlowCharacteristicsPlot::RiuFlowCharacteristicsPlot( RimFlowCharacteristicsPlot* plotDefinition, QWidget* parent )
|
|
: QFrame( parent )
|
|
, m_plotDefinition( plotDefinition )
|
|
|
|
{
|
|
Q_ASSERT( m_plotDefinition );
|
|
|
|
QGridLayout* mainLayout = new QGridLayout();
|
|
this->setLayout( mainLayout );
|
|
this->layout()->setMargin( 3 );
|
|
this->layout()->setSpacing( 3 );
|
|
|
|
// White background
|
|
QPalette pal = this->palette();
|
|
pal.setColor( QPalette::Window, Qt::white );
|
|
this->setAutoFillBackground( true );
|
|
this->setPalette( pal );
|
|
|
|
m_lorenzPlot = new QwtPlot( this );
|
|
m_flowCapVsStorageCapPlot = new QwtPlot( this );
|
|
m_sweepEffPlot = new QwtPlot( this );
|
|
|
|
mainLayout->addWidget( m_lorenzPlot, 0, 0, 1, 2 );
|
|
mainLayout->addWidget( m_flowCapVsStorageCapPlot, 1, 0 );
|
|
mainLayout->addWidget( m_sweepEffPlot, 1, 1 );
|
|
|
|
RiuQwtPlotTools::setCommonPlotBehaviour( m_lorenzPlot );
|
|
new RiuQwtPlotWheelZoomer( m_lorenzPlot );
|
|
addWindowZoom( m_lorenzPlot );
|
|
|
|
QString dateFormat = RiaPreferences::current()->dateFormat();
|
|
QString timeFormat = RiaPreferences::current()->timeFormat();
|
|
|
|
RiuQwtPlotTools::enableDateBasedBottomXAxis( m_lorenzPlot, dateFormat, timeFormat );
|
|
m_lorenzPlot->setTitle( "Lorenz Coefficient" );
|
|
|
|
RiuQwtPlotTools::setCommonPlotBehaviour( m_sweepEffPlot );
|
|
new RiuQwtPlotWheelZoomer( m_sweepEffPlot );
|
|
addWindowZoom( m_sweepEffPlot );
|
|
m_sweepEffPlot->setTitle( "Sweep Efficiency" );
|
|
|
|
int legendFontSize = caf::FontTools::absolutePointSize( RiaPreferences::current()->defaultPlotFontSize(),
|
|
caf::FontTools::RelativeSize::Small );
|
|
|
|
{
|
|
QwtText axisTitle = m_sweepEffPlot->axisTitle( QwtPlot::xBottom );
|
|
auto font = axisTitle.font();
|
|
font.setPointSize( legendFontSize );
|
|
axisTitle.setFont( font );
|
|
axisTitle.setText( "Dimensionless Time" );
|
|
m_sweepEffPlot->setAxisTitle( QwtPlot::xBottom, axisTitle );
|
|
}
|
|
{
|
|
QwtText axisTitle = m_sweepEffPlot->axisTitle( QwtPlot::yLeft );
|
|
auto font = axisTitle.font();
|
|
font.setPointSize( legendFontSize );
|
|
axisTitle.setFont( font );
|
|
axisTitle.setText( "Sweep Efficiency" );
|
|
m_sweepEffPlot->setAxisTitle( QwtPlot::yLeft, axisTitle );
|
|
}
|
|
|
|
RiuQwtPlotTools::setCommonPlotBehaviour( m_flowCapVsStorageCapPlot );
|
|
new RiuQwtPlotWheelZoomer( m_flowCapVsStorageCapPlot );
|
|
addWindowZoom( m_flowCapVsStorageCapPlot );
|
|
m_flowCapVsStorageCapPlot->setTitle( "Flow Capacity vs Storage Capacity" );
|
|
|
|
{
|
|
QwtText axisTitle = m_flowCapVsStorageCapPlot->axisTitle( QwtPlot::xBottom );
|
|
auto font = axisTitle.font();
|
|
font.setPointSize( legendFontSize );
|
|
axisTitle.setFont( font );
|
|
axisTitle.setText( "Storage Capacity [C]" );
|
|
m_flowCapVsStorageCapPlot->setAxisTitle( QwtPlot::xBottom, axisTitle );
|
|
}
|
|
{
|
|
QwtText axisTitle = m_flowCapVsStorageCapPlot->axisTitle( QwtPlot::yLeft );
|
|
auto font = axisTitle.font();
|
|
font.setPointSize( legendFontSize );
|
|
axisTitle.setFont( font );
|
|
axisTitle.setText( "Flow Capacity [F]" );
|
|
m_flowCapVsStorageCapPlot->setAxisTitle( QwtPlot::yLeft, axisTitle );
|
|
}
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
void RiuFlowCharacteristicsPlot::addWindowZoom( QwtPlot* plot )
|
|
{
|
|
auto zoomer = new RiuQwtPlotZoomer( plot->canvas() );
|
|
// zoomer->setRubberBandPen( QColor( Qt::black ) );
|
|
zoomer->setTrackerMode( QwtPicker::AlwaysOff );
|
|
// zoomer->setTrackerPen( QColor( Qt::black ) );
|
|
zoomer->initMousePattern( 1 );
|
|
}
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
RiuFlowCharacteristicsPlot::~RiuFlowCharacteristicsPlot()
|
|
{
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
void RiuFlowCharacteristicsPlot::setLorenzCurve( const QStringList& dateTimeStrings,
|
|
const std::vector<QDateTime>& dateTimes,
|
|
const std::vector<double>& timeHistoryValues )
|
|
{
|
|
m_lorenzPlot->detachItems( QwtPlotItem::Rtti_PlotCurve, true );
|
|
|
|
if ( !dateTimes.empty() )
|
|
{
|
|
initializeColors( dateTimes );
|
|
|
|
for ( size_t tsIdx = 0; tsIdx < dateTimes.size(); ++tsIdx )
|
|
{
|
|
if ( timeHistoryValues[tsIdx] == HUGE_VAL ) continue;
|
|
|
|
QDateTime dateTime = dateTimes[tsIdx];
|
|
double timeHistoryValue = timeHistoryValues[tsIdx];
|
|
|
|
QString curveName = dateTimeStrings[static_cast<int>( tsIdx )];
|
|
|
|
RiuFlowCharacteristicsPlot::addCurveWithLargeSymbol( m_lorenzPlot,
|
|
curveName,
|
|
m_dateToColorMap[dateTime],
|
|
dateTime,
|
|
timeHistoryValue );
|
|
}
|
|
}
|
|
|
|
m_lorenzPlot->replot();
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
void RiuFlowCharacteristicsPlot::addCurveWithLargeSymbol( QwtPlot* plot,
|
|
const QString& curveName,
|
|
const QColor& color,
|
|
const QDateTime& dateTime,
|
|
double timeHistoryValue )
|
|
{
|
|
auto curve = createEmptyCurve( plot, curveName, color );
|
|
|
|
QwtSymbol::Style style = QwtSymbol::Diamond;
|
|
QwtSymbol* symbol = new QwtSymbol( style );
|
|
|
|
symbol->setSize( 15, 15 );
|
|
symbol->setColor( color );
|
|
|
|
curve->setSymbol( symbol );
|
|
|
|
// Add date and value twice to avoid a cross as symbol generated by RiuQwtPlotCurve
|
|
|
|
std::vector<QDateTime> dateTimes;
|
|
dateTimes.push_back( dateTime );
|
|
dateTimes.push_back( dateTime );
|
|
|
|
std::vector<double> timeHistoryValues;
|
|
timeHistoryValues.push_back( timeHistoryValue );
|
|
timeHistoryValues.push_back( timeHistoryValue );
|
|
|
|
curve->setSamplesFromDatesAndYValues( dateTimes, timeHistoryValues, false );
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
RiuQwtPlotCurve*
|
|
RiuFlowCharacteristicsPlot::createEmptyCurve( QwtPlot* plot, const QString& curveName, const QColor& curveColor )
|
|
{
|
|
RiuQwtPlotCurve* plotCurve = new RiuQwtPlotCurve( curveName );
|
|
|
|
plotCurve->setTitle( curveName );
|
|
plotCurve->setPen( QPen( curveColor ) );
|
|
plotCurve->attach( plot );
|
|
return plotCurve;
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
void RiuFlowCharacteristicsPlot::addFlowCapStorageCapCurve( const QDateTime& dateTime,
|
|
const std::vector<double>& xVals,
|
|
const std::vector<double>& yVals )
|
|
{
|
|
CVF_ASSERT( !m_dateToColorMap.empty() );
|
|
|
|
RiuQwtPlotCurve* plotCurve =
|
|
createEmptyCurve( m_flowCapVsStorageCapPlot, dateTime.toString(), m_dateToColorMap[dateTime] );
|
|
plotCurve->setSamplesFromXValuesAndYValues( xVals, yVals, false );
|
|
m_flowCapVsStorageCapPlot->replot();
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
void RiuFlowCharacteristicsPlot::addSweepEfficiencyCurve( const QDateTime& dateTime,
|
|
const std::vector<double>& xVals,
|
|
const std::vector<double>& yVals )
|
|
{
|
|
CVF_ASSERT( !m_dateToColorMap.empty() );
|
|
|
|
RiuQwtPlotCurve* plotCurve = createEmptyCurve( m_sweepEffPlot, dateTime.toString(), m_dateToColorMap[dateTime] );
|
|
plotCurve->setSamplesFromXValuesAndYValues( xVals, yVals, false );
|
|
|
|
m_sweepEffPlot->replot();
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
void RiuFlowCharacteristicsPlot::removeAllCurves()
|
|
{
|
|
m_lorenzPlot->detachItems( QwtPlotItem::Rtti_PlotCurve, true );
|
|
m_lorenzPlot->replot();
|
|
m_sweepEffPlot->detachItems( QwtPlotItem::Rtti_PlotCurve, true );
|
|
m_sweepEffPlot->replot();
|
|
m_flowCapVsStorageCapPlot->detachItems( QwtPlotItem::Rtti_PlotCurve, true );
|
|
m_flowCapVsStorageCapPlot->replot();
|
|
m_dateToColorMap.clear();
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
void zoomAllInPlot( QwtPlot* plot )
|
|
{
|
|
plot->setAxisAutoScale( QwtPlot::xBottom, true );
|
|
plot->setAxisAutoScale( QwtPlot::yLeft, true );
|
|
plot->replot();
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
void RiuFlowCharacteristicsPlot::zoomAll()
|
|
{
|
|
zoomAllInPlot( m_lorenzPlot );
|
|
zoomAllInPlot( m_sweepEffPlot );
|
|
zoomAllInPlot( m_flowCapVsStorageCapPlot );
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
void RiuFlowCharacteristicsPlot::showLegend( bool show )
|
|
{
|
|
if ( show )
|
|
{
|
|
// Will be released in plot destructor or when a new legend is set
|
|
QwtLegend* legend = new QwtLegend( m_lorenzPlot );
|
|
m_lorenzPlot->insertLegend( legend, QwtPlot::BottomLegend );
|
|
}
|
|
else
|
|
{
|
|
m_lorenzPlot->insertLegend( nullptr );
|
|
}
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
RimFlowCharacteristicsPlot* RiuFlowCharacteristicsPlot::ownerPlotDefinition()
|
|
{
|
|
return m_plotDefinition;
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
RimViewWindow* RiuFlowCharacteristicsPlot::ownerViewWindow() const
|
|
{
|
|
return m_plotDefinition;
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
QSize RiuFlowCharacteristicsPlot::minimumSizeHint() const
|
|
{
|
|
return QSize( 0, 100 );
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
void RiuFlowCharacteristicsPlot::contextMenuEvent( QContextMenuEvent* event )
|
|
{
|
|
if ( m_plotDefinition )
|
|
{
|
|
QString curveDataAsText = m_plotDefinition->curveDataAsText();
|
|
|
|
QString dialogTitle = "Flow Characteristics";
|
|
|
|
RiaFeatureCommandContextTextHelper helper( dialogTitle, curveDataAsText );
|
|
|
|
caf::CmdFeatureMenuBuilder menuBuilder;
|
|
menuBuilder << "RicShowPlotDataFeature";
|
|
|
|
QMenu menu;
|
|
menuBuilder.appendToMenu( &menu );
|
|
|
|
if ( menu.actions().size() > 0 )
|
|
{
|
|
menu.exec( event->globalPos() );
|
|
}
|
|
}
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
QSize RiuFlowCharacteristicsPlot::sizeHint() const
|
|
{
|
|
return QSize( 0, 0 );
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
void RiuFlowCharacteristicsPlot::setDefaults()
|
|
{
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
void RiuFlowCharacteristicsPlot::initializeColors( const std::vector<QDateTime>& dateTimes )
|
|
{
|
|
CVF_ASSERT( m_dateToColorMap.empty() );
|
|
|
|
const caf::ColorTable& palette = RiaColorTables::timestepsPaletteColors();
|
|
cvf::Color3ubArray colorArray = caf::ColorTable::interpolateColorArray( palette.color3ubArray(), dateTimes.size() );
|
|
|
|
for ( size_t tsIdx = 0; tsIdx < dateTimes.size(); ++tsIdx )
|
|
{
|
|
m_dateToColorMap[dateTimes[tsIdx]] = QColor( colorArray[tsIdx].r(), colorArray[tsIdx].g(), colorArray[tsIdx].b() );
|
|
}
|
|
}
|