mirror of
https://github.com/OPM/ResInsight.git
synced 2025-02-10 23:46:00 -06:00
120 lines
4.7 KiB
C++
120 lines
4.7 KiB
C++
/////////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
// Copyright (C) 2019- 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 "RicfExportFlowCharacteristics.h"
|
|
|
|
#include "RiaApplication.h"
|
|
#include "RiaLogging.h"
|
|
|
|
#include "RicfApplicationTools.h"
|
|
|
|
#include "RimEclipseResultCase.h"
|
|
#include "RimFlowCharacteristicsPlot.h"
|
|
#include "RimFlowPlotCollection.h"
|
|
#include "RimMainPlotCollection.h"
|
|
#include "RimProject.h"
|
|
|
|
#include <QDir>
|
|
#include <QFile>
|
|
#include <QTextStream>
|
|
|
|
CAF_PDM_SOURCE_INIT(RicfExportFlowCharacteristics, "exportFlowCharacteristics");
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
RicfExportFlowCharacteristics::RicfExportFlowCharacteristics()
|
|
{
|
|
RICF_InitField(&m_caseId, "caseId", -1, "Case ID", "", "", "");
|
|
RICF_InitField(&m_selectedTimeSteps, "timeSteps", std::vector<int>(), "Selected Time Steps", "", "", "");
|
|
RICF_InitField(&m_injectors, "injectors", std::vector<QString>(), "Injectors", "", "", "");
|
|
RICF_InitField(&m_producers, "producers", std::vector<QString>(), "Producers", "", "", "");
|
|
RICF_InitField(&m_fileName, "fileName", QString(), "Export File Name", "", "", "");
|
|
RICF_InitField(&m_minCommunication, "minimumCommunication", 0.0, "Minimum Communication", "", "", "");
|
|
RICF_InitField(&m_maxPvFraction, "aquiferCellThreshold", 0.1, "Aquifer Cell Threshold", "", "", "");
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
RicfCommandResponse RicfExportFlowCharacteristics::execute()
|
|
{
|
|
using TOOLS = RicfApplicationTools;
|
|
|
|
auto eclipseCase = dynamic_cast<RimEclipseResultCase*>(TOOLS::caseFromId(m_caseId()));
|
|
if (!eclipseCase)
|
|
{
|
|
QString error = QString("exportFlowCharacteristics: Could not find case with ID %1.").arg(m_caseId());
|
|
RiaLogging::error(error);
|
|
return RicfCommandResponse(RicfCommandResponse::COMMAND_ERROR, error);
|
|
}
|
|
|
|
{
|
|
QString exportFileName = m_fileName();
|
|
QFileInfo fi(exportFileName);
|
|
if (!fi.isAbsolute())
|
|
{
|
|
QString relativePath = fi.path();
|
|
|
|
QString exportFolder = RiaApplication::instance()->createAbsolutePathFromProjectRelativePath(relativePath);
|
|
|
|
QDir exportDir(exportFolder);
|
|
if (!exportDir.exists())
|
|
{
|
|
if (!exportDir.mkpath("."))
|
|
{
|
|
QString msg = QString("Failed to create folder - %1").arg(exportFolder);
|
|
return RicfCommandResponse(RicfCommandResponse::COMMAND_ERROR, msg);
|
|
}
|
|
}
|
|
|
|
exportFileName = exportFolder + "/" + fi.fileName();
|
|
}
|
|
|
|
RimFlowPlotCollection* flowPlotColl = RiaApplication::instance()->project()->mainPlotCollection->flowPlotCollection();
|
|
if (flowPlotColl)
|
|
{
|
|
RimFlowCharacteristicsPlot* plot = flowPlotColl->defaultFlowCharacteristicsPlot();
|
|
plot->setFromFlowSolution(eclipseCase->defaultFlowDiagSolution());
|
|
plot->setTimeSteps(m_selectedTimeSteps);
|
|
plot->setInjectorsAndProducers(m_injectors, m_producers);
|
|
plot->setAquiferCellThreshold(m_maxPvFraction);
|
|
plot->setMinimumCommunication(m_minCommunication);
|
|
|
|
plot->loadDataAndUpdate();
|
|
|
|
{
|
|
QString content = plot->curveDataAsText();
|
|
|
|
QFile file(exportFileName);
|
|
if (file.open(QIODevice::WriteOnly | QIODevice::Text))
|
|
{
|
|
QTextStream textstream(&file);
|
|
textstream << content;
|
|
}
|
|
else
|
|
{
|
|
QString msg = QString("Failed to export file - %1").arg(exportFileName);
|
|
return RicfCommandResponse(RicfCommandResponse::COMMAND_ERROR, msg);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return RicfCommandResponse();
|
|
}
|