mirror of
https://github.com/OPM/ResInsight.git
synced 2025-01-23 23:13:39 -06:00
139 lines
4.8 KiB
C++
139 lines
4.8 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 "RicAsciiExportSummaryPlotFeature.h"
|
|
|
|
#include "RiaApplication.h"
|
|
#include "RiaPreferences.h"
|
|
|
|
#include "RimMainPlotCollection.h"
|
|
#include "RimProject.h"
|
|
#include "RimSummaryCase.h"
|
|
#include "RimSummaryCurveFilter.h"
|
|
#include "RimSummaryPlot.h"
|
|
#include "RimSummaryPlotExportSettings.h"
|
|
#include "RimSummaryPlotCollection.h"
|
|
|
|
#include "RiuMainPlotWindow.h"
|
|
|
|
#include "cafSelectionManager.h"
|
|
|
|
#include <QAction>
|
|
|
|
#include "cvfAssert.h"
|
|
#include <vector>
|
|
#include "cafPdmUiPropertyViewDialog.h"
|
|
#include "RiuMainWindow.h"
|
|
#include "QMessageBox"
|
|
#include "QFileInfo"
|
|
#include "QDebug"
|
|
#include "RiaLogging.h"
|
|
#include "cafProgressInfo.h"
|
|
|
|
|
|
CAF_CMD_SOURCE_INIT(RicAsciiExportSummaryPlotFeature, "RicAsciiExportSummaryPlotFeature");
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
bool RicAsciiExportSummaryPlotFeature::isCommandEnabled()
|
|
{
|
|
return true;
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
void RicAsciiExportSummaryPlotFeature::onActionTriggered(bool isChecked)
|
|
{
|
|
RimProject* project = RiaApplication::instance()->project();
|
|
CVF_ASSERT(project);
|
|
|
|
std::vector<RimSummaryPlot*> selectedSummaryPlots;
|
|
caf::SelectionManager::instance()->objectsByType(&selectedSummaryPlots);
|
|
|
|
|
|
|
|
RimSummaryPlotExportSettings exportSettings;
|
|
|
|
RiaApplication* app = RiaApplication::instance();
|
|
QString projectFolder = app->currentProjectPath();
|
|
|
|
QString defaultDir = RiaApplication::instance()->lastUsedDialogDirectoryWithFallback("SUMMARYPLOT_ASCIIEXPORT_DIR", projectFolder);
|
|
QString defaultFilename = QString("SummaryPlotExport"); //TODO - take from summery plot user name
|
|
|
|
QString outputFileName = defaultDir + "/" + defaultFilename;
|
|
exportSettings.fileName = outputFileName;
|
|
|
|
caf::PdmUiPropertyViewDialog propertyDialog(RiuMainWindow::instance(), &exportSettings, "Export Summary Plots to Ascii", "");
|
|
if (propertyDialog.exec() == QDialog::Accepted)
|
|
{
|
|
RiaApplication::instance()->setLastUsedDialogDirectory("SUMMARYPLOT_ASCIIEXPORT_DIR", QFileInfo(exportSettings.fileName).absolutePath());
|
|
|
|
bool isOk = writeAsciiExportForSummaryPlots(exportSettings.fileName(), selectedSummaryPlots);
|
|
|
|
if (!isOk)
|
|
{
|
|
QMessageBox::critical(NULL, "File export", "Failed to exported current result to " + exportSettings.fileName);
|
|
}
|
|
}
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
void RicAsciiExportSummaryPlotFeature::setupActionLook(QAction* actionToSetup)
|
|
{
|
|
actionToSetup->setText("Export Summary Plot Data");
|
|
actionToSetup->setIcon(QIcon(":/SummaryPlot16x16.png"));
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
bool RicAsciiExportSummaryPlotFeature::writeAsciiExportForSummaryPlots(const QString& fileName, const std::vector<RimSummaryPlot*>& selectedSummaryPlots)
|
|
{
|
|
RiaLogging::info(QString("Writing ascii values for summary plot(s) to file: %1").arg(fileName));
|
|
|
|
QFile file(fileName);
|
|
if (!file.open(QIODevice::WriteOnly | QIODevice::Text))
|
|
{
|
|
return false;
|
|
}
|
|
|
|
caf::ProgressInfo pi(selectedSummaryPlots.size(), QString("Writing data to file %1").arg(fileName));
|
|
size_t progress = 0;
|
|
|
|
QTextStream out(&file);
|
|
out << "\n";
|
|
out << "-- Exported from ResInsight" << "\n";
|
|
|
|
for (auto* summaryPlot : selectedSummaryPlots)
|
|
{
|
|
progress++;
|
|
pi.setProgress(progress);
|
|
}
|
|
|
|
out << "\n";
|
|
|
|
RiaLogging::info(QString("Competed writing COMPDAT data to file %1").arg(fileName));
|
|
return true;
|
|
}
|
|
|
|
|