First cut implementation, very much still a work in progress

This commit is contained in:
sigurdp
2019-12-11 09:38:25 +01:00
committed by Magne Sjaastad
parent 349886d73e
commit 92bb34197e
7 changed files with 712 additions and 107 deletions

View File

@@ -23,6 +23,7 @@
#include "RimFlowCharacteristicsPlot.h"
#include "RimProject.h"
#include "RimWellAllocationPlot.h"
#include "RimWellDistributionPlot.h"
#include "cafProgressInfo.h"
#include "cvfAssert.h"
@@ -42,6 +43,9 @@ RimFlowPlotCollection::RimFlowPlotCollection()
CAF_PDM_InitFieldNoDefault( &m_defaultWellAllocPlot, "DefaultWellAllocationPlot", "", "", "", "" );
m_defaultWellAllocPlot.uiCapability()->setUiHidden( true );
CAF_PDM_InitFieldNoDefault( &m_wellDistributionPlot, "WellDistributionPlot", "", "", "", "" );
m_wellDistributionPlot.uiCapability()->setUiHidden( true );
CAF_PDM_InitFieldNoDefault( &m_storedWellAllocPlots,
"StoredWellAllocationPlots",
"Stored Well Allocation Plots",
@@ -88,7 +92,7 @@ void RimFlowPlotCollection::closeDefaultPlotWindowAndDeletePlots()
//--------------------------------------------------------------------------------------------------
void RimFlowPlotCollection::loadDataAndUpdate()
{
caf::ProgressInfo plotProgress( m_storedWellAllocPlots.size() + m_storedFlowCharacteristicsPlots.size() + 1, "" );
caf::ProgressInfo plotProgress( m_storedWellAllocPlots.size() + m_storedFlowCharacteristicsPlots.size() + 2, "" );
if ( m_defaultWellAllocPlot ) m_defaultWellAllocPlot->loadDataAndUpdate();
plotProgress.incrementProgress();
@@ -109,6 +113,11 @@ void RimFlowPlotCollection::loadDataAndUpdate()
{
m_flowCharacteristicsPlot->loadDataAndUpdate();
}
if ( m_wellDistributionPlot )
{
m_wellDistributionPlot->loadDataAndUpdate();
}
}
//--------------------------------------------------------------------------------------------------
@@ -185,4 +194,9 @@ void RimFlowPlotCollection::ensureDefaultFlowPlotsAreCreated()
{
m_flowCharacteristicsPlot = new RimFlowCharacteristicsPlot;
}
if ( !m_wellDistributionPlot() )
{
m_wellDistributionPlot = new RimWellDistributionPlot;
}
}

View File

@@ -24,6 +24,7 @@
class RimWellAllocationPlot;
class RimFlowCharacteristicsPlot;
class RimWellDistributionPlot;
//==================================================================================================
///
@@ -50,6 +51,7 @@ public:
private:
caf::PdmChildField<RimFlowCharacteristicsPlot*> m_flowCharacteristicsPlot;
caf::PdmChildField<RimWellAllocationPlot*> m_defaultWellAllocPlot;
caf::PdmChildField<RimWellDistributionPlot*> m_wellDistributionPlot;
caf::PdmChildArrayField<RimWellAllocationPlot*> m_storedWellAllocPlots;
caf::PdmChildArrayField<RimFlowCharacteristicsPlot*> m_storedFlowCharacteristicsPlots;
};

View File

@@ -17,7 +17,28 @@
/////////////////////////////////////////////////////////////////////////////////
#include "RimWellDistributionPlot.h"
#include "RimEclipseResultCase.h"
#include "RimProject.h"
#include "RimFlowDiagSolution.h"
#include "RigEclipseCaseData.h"
#include "RigTofWellDistributionCalculator.h"
#include "RiaColorTools.h"
#include "RiuQwtPlotTools.h"
#include "qwt_plot.h"
#include "qwt_legend.h"
#include "qwt_plot_curve.h"
#include <QWidget>
#include <QGridLayout>
#include <QTextBrowser>
#include "cvfBase.h"
#include "cvfTrace.h"
#include "cvfDebugTimer.h"
//==================================================================================================
@@ -33,7 +54,17 @@ CAF_PDM_SOURCE_INIT(RimWellDistributionPlot, "WellDistributionPlot");
//--------------------------------------------------------------------------------------------------
RimWellDistributionPlot::RimWellDistributionPlot()
{
cvf::Trace::show("RimWellDistributionPlot::RimWellDistributionPlot()");
CAF_PDM_InitObject("Well Distribution Plot", "", "", "");
CAF_PDM_InitFieldNoDefault(&m_case, "Case", "Case", "", "", "");
CAF_PDM_InitField(&m_timeStepIndex, "TimeStepIndex", -1, "Time Step", "", "", "");
CAF_PDM_InitField(&m_wellName, "WellName", QString("None"), "Well", "", "", "");
m_showWindow = false;
setAsPlotMdiWindow();
}
//--------------------------------------------------------------------------------------------------
@@ -48,7 +79,8 @@ RimWellDistributionPlot::~RimWellDistributionPlot()
//--------------------------------------------------------------------------------------------------
QWidget* RimWellDistributionPlot::viewWidget()
{
return nullptr;
cvf::Trace::show("RimWellDistributionPlot::viewWidget()");
return m_myViewWidget;
}
//--------------------------------------------------------------------------------------------------
@@ -56,6 +88,7 @@ QWidget* RimWellDistributionPlot::viewWidget()
//--------------------------------------------------------------------------------------------------
QImage RimWellDistributionPlot::snapshotWindowContent()
{
cvf::Trace::show("RimWellDistributionPlot::snapshotWindowContent()");
return QImage();
}
@@ -64,103 +97,323 @@ QImage RimWellDistributionPlot::snapshotWindowContent()
//--------------------------------------------------------------------------------------------------
void RimWellDistributionPlot::zoomAll()
{
cvf::Trace::show("RimWellDistributionPlot::zoomAll()");
}
RiuQwtPlotWidget* RimWellDistributionPlot::viewer()
{
throw std::logic_error("The method or operation is not implemented.");
}
bool RimWellDistributionPlot::isChecked() const
{
throw std::logic_error("The method or operation is not implemented.");
}
void RimWellDistributionPlot::setChecked(bool checked)
{
throw std::logic_error("The method or operation is not implemented.");
}
//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
QString RimWellDistributionPlot::description() const
{
throw std::logic_error("The method or operation is not implemented.");
}
bool RimWellDistributionPlot::hasCustomFontSizes(RiaDefines::FontSettingType fontSettingType, int defaultFontSize) const
{
throw std::logic_error("The method or operation is not implemented.");
}
bool RimWellDistributionPlot::applyFontSize(RiaDefines::FontSettingType fontSettingType, int oldFontSize, int fontSize, bool forceChange /*= false*/)
{
throw std::logic_error("The method or operation is not implemented.");
}
void RimWellDistributionPlot::setAutoScaleXEnabled(bool enabled)
{
throw std::logic_error("The method or operation is not implemented.");
}
void RimWellDistributionPlot::setAutoScaleYEnabled(bool enabled)
{
throw std::logic_error("The method or operation is not implemented.");
}
void RimWellDistributionPlot::updateAxes()
{
throw std::logic_error("The method or operation is not implemented.");
}
void RimWellDistributionPlot::updateZoomInQwt()
{
throw std::logic_error("The method or operation is not implemented.");
}
void RimWellDistributionPlot::updateZoomFromQwt()
{
throw std::logic_error("The method or operation is not implemented.");
}
void RimWellDistributionPlot::createPlotWidget()
{
throw std::logic_error("The method or operation is not implemented.");
}
void RimWellDistributionPlot::detachAllCurves()
{
throw std::logic_error("The method or operation is not implemented.");
}
caf::PdmObject* RimWellDistributionPlot::findPdmObjectFromQwtCurve(const QwtPlotCurve* curve) const
{
throw std::logic_error("The method or operation is not implemented.");
}
void RimWellDistributionPlot::loadDataAndUpdate()
{
}
void RimWellDistributionPlot::updateLayout()
{
throw std::logic_error("The method or operation is not implemented.");
return uiName();
}
//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
QWidget* RimWellDistributionPlot::createViewWidget(QWidget* mainWindowParent)
{
return nullptr;
cvf::Trace::show("RimWellDistributionPlot::createViewWidget()");
m_myViewWidget = new QWidget(mainWindowParent);
QGridLayout* gridLayout = new QGridLayout(m_myViewWidget);
m_textBrowser = new QTextBrowser;
m_textBrowser->setText("<center><h1>NotYet</h1></center>");
gridLayout->addWidget(m_textBrowser, 0, 0);
m_plotWidgets[0] = constructNewPlotWidget();
m_plotWidgets[1] = constructNewPlotWidget();
m_plotWidgets[2] = constructNewPlotWidget();
gridLayout->addWidget(m_plotWidgets[0], 1, 0);
gridLayout->addWidget(m_plotWidgets[1], 0, 1);
gridLayout->addWidget(m_plotWidgets[2], 1, 1);
onLoadDataAndUpdate();
return m_myViewWidget;
}
//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
QwtPlot* RimWellDistributionPlot::constructNewPlotWidget()
{
class MyQwtPlot : public QwtPlot
{
public:
MyQwtPlot(QWidget* parent) : QwtPlot(parent) {}
QSize sizeHint() const override { return QSize(100, 100); }
QSize minimumSizeHint() const override { return QSize(0, 0); }
};
QwtPlot* plotWidget = new MyQwtPlot(nullptr);
RiuQwtPlotTools::setCommonPlotBehaviour(plotWidget);
QwtLegend* legend = new QwtLegend(plotWidget);
plotWidget->insertLegend(legend, QwtPlot::BottomLegend);
return plotWidget;
}
//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RimWellDistributionPlot::deleteViewWidget()
{
cvf::Trace::show("RimWellDistributionPlot::deleteViewWidget()");
if (m_myViewWidget)
{
m_myViewWidget->deleteLater();
m_myViewWidget = nullptr;
}
}
//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RimWellDistributionPlot::populatePlotWidgetWithCurveData(const RigTofWellDistributionCalculator& calculator, RiaDefines::PhaseType phase, const RimFlowDiagSolution& flowDiagSolution, QwtPlot* plotWidget)
{
cvf::Trace::show("RimWellDistributionPlot::populatePlotWidgetWithCurves(phase=%d '%s')", phase, phase == RiaDefines::OIL_PHASE ? "oil" : (phase == RiaDefines::GAS_PHASE ? "gas" : "water"));
// Currently select this value so that the grid appears on top of the curves
const double baseCurveZValue = 9.5;
plotWidget->detachItems(QwtPlotItem::Rtti_PlotCurve);
plotWidget->setAxisScale(QwtPlot::xBottom, 0, 1);
plotWidget->setAxisScale(QwtPlot::yLeft, 0, 1);
plotWidget->setAxisAutoScale(QwtPlot::xBottom, true);
plotWidget->setAxisAutoScale(QwtPlot::yLeft, true);
const std::vector<double>& tofValuesDays = calculator.sortedUniqueTOFValues();
if (tofValuesDays.size() == 0)
{
cvf::Trace::show("No TOF values!");
return;
}
std::vector<double> tofValuesYears;
for (double tofDays : tofValuesDays)
{
const double tofYears = tofDays/365.2425;
tofValuesYears.push_back(tofYears);
}
cvf::Trace::show("numTofValues: %d (min, max: %f, %f)", static_cast<int>(tofValuesYears.size()), tofValuesYears.front(), tofValuesYears.back());
const size_t numWells = calculator.contributingWellCount();
cvf::Trace::show("numContribWells: %d", static_cast<int>(numWells));
std::vector<double> yVals(tofValuesYears.size(), 0);
for (size_t i = 0; i < numWells; i++)
{
QString wellName = calculator.contributingWellName(i);
const std::vector<double>& volArr = calculator.accumulatedPhaseVolumeForContributingWell(phase, i);
cvf::Color3f cvfClr = flowDiagSolution.tracerColor(wellName);
QColor qtClr = RiaColorTools::toQColor(cvfClr);
cvf::Trace::show("wellName min, max: %s %f, %f", wellName.toStdString().c_str(), volArr.front(), volArr.back());
for (size_t j = 0; j < yVals.size(); j++)
{
yVals[j] += volArr[j];
}
QwtPlotCurve* curve = new QwtPlotCurve;
curve->setTitle(wellName);
curve->setBrush(qtClr);
curve->setZ(baseCurveZValue - i * 0.0001);
curve->setSamples(tofValuesYears.data(), yVals.data(), static_cast<int>(tofValuesYears.size()));
curve->attach(plotWidget);
}
}
//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RimWellDistributionPlot::onLoadDataAndUpdate()
{
cvf::Trace::show("RimWellDistributionPlot::onLoadDataAndUpdate()");
cvf::DebugTimer tim("RimWellDistributionPlot::onLoadDataAndUpdate()");
updateMdiWindowVisibility();
if (!m_myViewWidget)
{
return;
}
CVF_ASSERT(m_textBrowser);
QString str = "<center><h1>MyViewWidget</h1></center>";
str += "<br><br>";
const QString caseName = m_case ? m_case->caseUserDescription() : "N/A";
str += QString("case: %1<br>").arg(caseName);
str += QString("timeStepIndex: %1<br>").arg(m_timeStepIndex);
str += QString("wellName: %1<br>").arg(m_wellName);
m_textBrowser->setText(str);
for (QwtPlot* plotWidget : m_plotWidgets)
{
plotWidget->detachItems(QwtPlotItem::Rtti_PlotCurve);
}
if (m_case)
{
tim.reportLapTimeMS("about to start calc");
RigTofWellDistributionCalculator calc(m_case, m_wellName, m_timeStepIndex);
tim.reportLapTimeMS("calc");
const QString timeStepName = m_case->timeStepName(m_timeStepIndex);
const RimFlowDiagSolution* flowDiagSolution = m_case->defaultFlowDiagSolution();
for (int i = 0; i < m_plotWidgets.size(); i++)
{
const RiaDefines::PhaseType phase = static_cast<RiaDefines::PhaseType>(i);
QwtPlot* plotWidget = m_plotWidgets[i];
populatePlotWidgetWithCurveData(calc, phase, *flowDiagSolution, plotWidget);
QString phaseString = "N/A";
if (phase == RiaDefines::OIL_PHASE) phaseString = "Oil";
else if (phase == RiaDefines::GAS_PHASE) phaseString = "Gas";
else if (phase == RiaDefines::WATER_PHASE) phaseString = "Water";
plotWidget->setAxisTitle(QwtPlot::xBottom, "TOF [years]");
plotWidget->setAxisTitle(QwtPlot::yLeft, "Reservoir Volume [m3]");
const QString plotTitle = QString("%1 Distribution: %2, %3").arg(phaseString).arg(m_wellName).arg(timeStepName);
plotWidget->setTitle(plotTitle);
}
}
for (QwtPlot* plotWidget : m_plotWidgets)
{
plotWidget->replot();
}
}
void RimWellDistributionPlot::updatePlotTitle()
//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RimWellDistributionPlot::defineUiOrdering(QString uiConfigName, caf::PdmUiOrdering& uiOrdering)
{
throw std::logic_error("The method or operation is not implemented.");
caf::PdmUiGroup* group = uiOrdering.addNewGroup("Plot Data");
group->add(&m_case);
group->add(&m_timeStepIndex);
group->add(&m_wellName);
uiOrdering.skipRemainingFields();
}
//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
QList<caf::PdmOptionItemInfo> RimWellDistributionPlot::calculateValueOptions(const caf::PdmFieldHandle* fieldNeedingOptions, bool* /*useOptionsOnly*/)
{
QList<caf::PdmOptionItemInfo> options;
if (fieldNeedingOptions == &m_case)
{
RimProject* ownerProj = nullptr;
firstAncestorOrThisOfType(ownerProj);
if (ownerProj)
{
std::vector<RimEclipseResultCase*> caseArr;
ownerProj->descendantsIncludingThisOfType(caseArr);
for (RimEclipseResultCase* c : caseArr)
{
options.push_back(caf::PdmOptionItemInfo(c->caseUserDescription(), c, true, c->uiIconProvider()));
}
}
}
else if (fieldNeedingOptions == &m_timeStepIndex)
{
if (m_case && m_case->eclipseCaseData())
{
const QStringList timeStepNames = m_case->timeStepStrings();
for (int i = 0; i < timeStepNames.size(); i++)
{
options.push_back(caf::PdmOptionItemInfo(timeStepNames[i], i));
}
}
if (options.size() == 0)
{
options.push_back(caf::PdmOptionItemInfo("None", -1));
}
}
else if (fieldNeedingOptions == &m_wellName)
{
if (m_case && m_case->eclipseCaseData())
{
caf::QIconProvider simWellIcon(":/Well.png");
const std::set<QString> sortedWellNameSet = m_case->eclipseCaseData()->findSortedWellNames();
for (const QString& name : sortedWellNameSet)
{
options.push_back(caf::PdmOptionItemInfo(name, name, true, simWellIcon));
}
}
if (options.size() == 0)
{
options.push_back(caf::PdmOptionItemInfo("None", QVariant()));
}
}
return options;
}
//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RimWellDistributionPlot::fieldChangedByUi(const caf::PdmFieldHandle* changedField, const QVariant& oldValue, const QVariant& newValue)
{
if (changedField == &m_case)
{
fixupDependentFieldsAfterCaseChange();
}
onLoadDataAndUpdate();
}
//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RimWellDistributionPlot::fixupDependentFieldsAfterCaseChange()
{
int newTimeStepIndex = -1;
QString newWellName;
if (m_case)
{
const int timeStepCount = m_case->timeStepStrings().size();
if (timeStepCount > 0)
{
newTimeStepIndex = timeStepCount - 1;
}
const std::set<QString> sortedWellNameSet = m_case->eclipseCaseData()->findSortedWellNames();
if (sortedWellNameSet.size() > 0)
{
newWellName = *sortedWellNameSet.begin();
}
}
m_timeStepIndex = newTimeStepIndex;
m_wellName = newWellName;
}

View File

@@ -19,7 +19,20 @@
#pragma once
#include "RimPlotWindow.h"
#include "RimPlotInterface.h"
#include "RiaDefines.h"
#include "cafPdmPtrField.h"
#include <QPointer>
#include <array>
class RimEclipseResultCase;
class RimFlowDiagSolution;
class RigTofWellDistributionCalculator;
class QTextBrowser;
class QwtPlot;
//==================================================================================================
@@ -27,7 +40,7 @@
//
//
//==================================================================================================
class RimWellDistributionPlot : public RimPlotWindow, public RimPlotInterface
class RimWellDistributionPlot : public RimPlotWindow
{
CAF_PDM_HEADER_INIT;
@@ -35,34 +48,37 @@ public:
RimWellDistributionPlot();
~RimWellDistributionPlot() override;
// RimViewWindow/RimPlotWindow implementations/overrides
virtual QWidget* viewWidget() override;
virtual QImage snapshotWindowContent() override;
virtual void zoomAll() override;
virtual void detachAllCurves() override;
virtual void updateLayout() override;
// RimPlotWindow implementations
virtual QString description() const override;
// RimPlotInterface implementation
virtual RiuQwtPlotWidget* viewer() override;
virtual bool isChecked() const override;
virtual void setChecked(bool checked) override;
virtual QString description() const override;
virtual bool hasCustomFontSizes(RiaDefines::FontSettingType fontSettingType, int defaultFontSize) const override;
virtual bool applyFontSize(RiaDefines::FontSettingType fontSettingType, int oldFontSize, int fontSize, bool forceChange = false) override;
virtual void setAutoScaleXEnabled(bool enabled) override;
virtual void setAutoScaleYEnabled(bool enabled) override;
virtual void updateAxes() override;
virtual void updateZoomInQwt() override;
virtual void updateZoomFromQwt() override;
virtual void createPlotWidget() override;
//virtual void detachAllCurves() override;
virtual caf::PdmObject* findPdmObjectFromQwtCurve(const QwtPlotCurve* curve) const override;
virtual void loadDataAndUpdate() override;
// RimViewWindow implementations
virtual QWidget* viewWidget() override;
virtual QImage snapshotWindowContent() override;
virtual void zoomAll() override;
private:
// RimViewWindow/RimPlotWindow implementations/overrides
// RimPlotWindow overrides
virtual QList<caf::PdmOptionItemInfo> calculateValueOptions(const caf::PdmFieldHandle* fieldNeedingOptions, bool* useOptionsOnly) override;
virtual void fieldChangedByUi(const caf::PdmFieldHandle* changedField, const QVariant& oldValue, const QVariant& newValue) override;
// RimViewWindow implementations
virtual QWidget* createViewWidget(QWidget* mainWindowParent) override;
virtual void deleteViewWidget() override;
virtual void onLoadDataAndUpdate() override;
virtual void updatePlotTitle() override;
private:
virtual void defineUiOrdering(QString uiConfigName, caf::PdmUiOrdering& uiOrdering) override;
void fixupDependentFieldsAfterCaseChange();
static QwtPlot* constructNewPlotWidget();
static void populatePlotWidgetWithCurveData(const RigTofWellDistributionCalculator& calculator, RiaDefines::PhaseType phase, const RimFlowDiagSolution& flowDiagSolution, QwtPlot* plotWidget);
private:
caf::PdmPtrField<RimEclipseResultCase*> m_case;
caf::PdmField<int> m_timeStepIndex;
caf::PdmField<QString> m_wellName;
QPointer<QWidget> m_myViewWidget;
QPointer<QTextBrowser> m_textBrowser;
std::array<QPointer<QwtPlot>, 3> m_plotWidgets;
};