mirror of
https://github.com/OPM/ResInsight.git
synced 2025-01-24 07:16:53 -06:00
311 lines
12 KiB
C++
311 lines
12 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 "RimFlowCharacteristicsPlot.h"
|
|
|
|
#include "RiuLineSegmentQwtPlotCurve.h"
|
|
#include "RiuQwtPlotWheelZoomer.h"
|
|
#include "RiuQwtPlotZoomer.h"
|
|
#include "RiuResultQwtPlot.h"
|
|
#include "RiuSummaryQwtPlot.h"
|
|
|
|
#include "cvfBase.h"
|
|
#include "cvfColor3.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)
|
|
: m_plotDefinition(plotDefinition),
|
|
QFrame(parent)
|
|
{
|
|
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::Background, 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);
|
|
|
|
RiuSummaryQwtPlot::setCommonPlotBehaviour(m_lorenzPlot);
|
|
new RiuQwtPlotWheelZoomer(m_lorenzPlot);
|
|
addWindowZoom(m_lorenzPlot);
|
|
RiuSummaryQwtPlot::enableDateBasedBottomXAxis(m_lorenzPlot);
|
|
m_lorenzPlot->setTitle("Lorenz Coefficient");
|
|
|
|
RiuSummaryQwtPlot::setCommonPlotBehaviour(m_sweepEffPlot);
|
|
new RiuQwtPlotWheelZoomer(m_sweepEffPlot);
|
|
addWindowZoom(m_sweepEffPlot);
|
|
m_sweepEffPlot->setTitle("Sweep Efficiency");
|
|
|
|
RiuSummaryQwtPlot::setCommonPlotBehaviour(m_flowCapVsStorageCapPlot);
|
|
new RiuQwtPlotWheelZoomer(m_flowCapVsStorageCapPlot);
|
|
addWindowZoom(m_flowCapVsStorageCapPlot);
|
|
m_flowCapVsStorageCapPlot->setTitle("Flow Capacity vs Storage Capacity");
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
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()
|
|
{
|
|
if (m_plotDefinition)
|
|
{
|
|
m_plotDefinition->handleMdiWindowClosed();
|
|
}
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
void RiuFlowCharacteristicsPlot::setLorenzCurve(const QStringList& dateTimeStrings, const std::vector<QDateTime>& dateTimes, const std::vector<double>& timeHistoryValues)
|
|
{
|
|
initializeColors(dateTimes);
|
|
|
|
m_lorenzPlot->detachItems(QwtPlotItem::Rtti_PlotCurve, true);
|
|
|
|
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 RiuLineSegmentQwtPlotCurve
|
|
|
|
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->setSamplesFromDateAndValues(dateTimes, timeHistoryValues, false);
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
RiuLineSegmentQwtPlotCurve* RiuFlowCharacteristicsPlot::createEmptyCurve(QwtPlot* plot, const QString& curveName, const QColor& curveColor )
|
|
{
|
|
RiuLineSegmentQwtPlotCurve* plotCurve = new RiuLineSegmentQwtPlotCurve(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());
|
|
|
|
RiuLineSegmentQwtPlotCurve* plotCurve = createEmptyCurve(m_flowCapVsStorageCapPlot, dateTime.toString(), m_dateToColorMap[dateTime]);
|
|
plotCurve->setSamplesFromTimeAndValues(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());
|
|
|
|
RiuLineSegmentQwtPlotCurve* plotCurve = createEmptyCurve(m_sweepEffPlot, dateTime.toString(), m_dateToColorMap[dateTime]);
|
|
plotCurve->setSamplesFromTimeAndValues(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);
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
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());
|
|
}
|
|
}
|
|
|