#872 - Adding command feature for export of summary plots to ascii (and writing empty export file)

2025-02-25 18:55:39 -06:00 · 2017-03-07 13:48:17 +01:00
parent 6e69bdb81d
commit b888e933b5
8 changed files with 291 additions and 0 deletions
--- a/ApplicationCode/Commands/SummaryPlotCommands/RicAsciiExportSummaryPlotFeature.cpp
+++ b/ApplicationCode/Commands/SummaryPlotCommands/RicAsciiExportSummaryPlotFeature.cpp
@@ -0,0 +1,138 @@
+/////////////////////////////////////////////////////////////////////////////////
+//
+//  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;
+}
+
+