Initial commit of ResInsight version 0.4.8

ApplicationCode/FileInterface/RifReaderEclipseUnifiedRestartFile.cpp

@@ -0,0 +1,164 @@
+/////////////////////////////////////////////////////////////////////////////////
+//
+//  Copyright (C) 2011-2012 Statoil ASA, Ceetron AS
+// 
+//  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 "RifReaderEclipseUnifiedRestartFile.h"
+#include "RifReaderEclipseFileAccess.h"
+
+#ifdef USE_ECL_LIB
+#include <well_state.h>
+#include <well_info.h>
+#include <well_conn.h>
+#include <well_ts.h>
+#endif
+
+//--------------------------------------------------------------------------------------------------
+/// Constructor
+//--------------------------------------------------------------------------------------------------
+RifReaderEclipseUnifiedRestartFile::RifReaderEclipseUnifiedRestartFile(size_t numGrids, size_t numActiveCells)
+    : RifReaderEclipseResultsAccess(numGrids, numActiveCells)
+{
+}
+
+//--------------------------------------------------------------------------------------------------
+/// Destructor
+//--------------------------------------------------------------------------------------------------
+RifReaderEclipseUnifiedRestartFile::~RifReaderEclipseUnifiedRestartFile()
+{
+    close();
+}
+
+//--------------------------------------------------------------------------------------------------
+/// Open file
+//--------------------------------------------------------------------------------------------------
+bool RifReaderEclipseUnifiedRestartFile::open(const QStringList& fileSet)
+{
+#ifdef USE_ECL_LIB
+    QString fileName = fileSet[0];
+
+    cvf::ref<RifReaderEclipseFileAccess> fileAccess = new RifReaderEclipseFileAccess;
+    if (!fileAccess->open(fileName))
+    {
+        return false;
+    }
+
+    m_file = fileAccess;
+
+    return true;
+#else
+    return false;
+#endif
+}
+
+//--------------------------------------------------------------------------------------------------
+/// Close file
+//--------------------------------------------------------------------------------------------------
+void RifReaderEclipseUnifiedRestartFile::close()
+{
+    if (m_file.notNull())
+    {
+        m_file->close();
+        m_file = NULL;
+    }
+}
+
+//--------------------------------------------------------------------------------------------------
+/// Get the number of time steps
+//--------------------------------------------------------------------------------------------------
+size_t RifReaderEclipseUnifiedRestartFile::numTimeSteps()
+{
+    QStringList timeSteps = timeStepsText();
+    return timeSteps.size();
+}
+
+//--------------------------------------------------------------------------------------------------
+/// Get the time step texts
+//--------------------------------------------------------------------------------------------------
+QStringList RifReaderEclipseUnifiedRestartFile::timeStepsText()
+{
+    RifReaderEclipseFileAccess* file = m_file.p();
+    CVF_ASSERT(file != NULL);
+
+    QStringList timeSteps;
+    file->timeStepsText(&timeSteps);
+
+    return timeSteps;
+}
+
+//--------------------------------------------------------------------------------------------------
+/// Get the time steps
+//--------------------------------------------------------------------------------------------------
+QList<QDateTime> RifReaderEclipseUnifiedRestartFile::timeSteps()
+{
+    RifReaderEclipseFileAccess* file = m_file.p();
+    CVF_ASSERT(file != NULL);
+
+    QList<QDateTime> timeSteps;
+    file->timeSteps(&timeSteps);
+
+    return timeSteps;
+}
+
+//--------------------------------------------------------------------------------------------------
+/// Get list of result names
+//--------------------------------------------------------------------------------------------------
+QStringList RifReaderEclipseUnifiedRestartFile::resultNames()
+{
+    // Get the results found on the UNRST file
+    QStringList resultsList;
+    m_file->keywordsOnFile(&resultsList, m_numActiveCells, numTimeSteps());
+
+    return resultsList;
+}
+
+//--------------------------------------------------------------------------------------------------
+/// Get result values for given time step
+//--------------------------------------------------------------------------------------------------
+bool RifReaderEclipseUnifiedRestartFile::results(const QString& resultName, size_t timeStep, std::vector<double>* values)
+{
+    size_t numOccurrences   = m_file->numOccurrences(resultName);
+    size_t startIndex       = timeStep*m_numGrids;
+    CVF_ASSERT(startIndex + m_numGrids <= numOccurrences);
+
+    size_t i;
+    for (i = startIndex; i < startIndex + m_numGrids; i++)
+    {
+        std::vector<double> partValues;
+        if (!m_file->keywordData(resultName, i, &partValues))  // !! don't need to append afterwards
+        {
+            return false;
+        }
+
+        values->insert(values->end(), partValues.begin(), partValues.end());
+    }
+
+    return true;
+}
+
+
+//--------------------------------------------------------------------------------------------------
+///
+//--------------------------------------------------------------------------------------------------
+#ifdef USE_ECL_LIB
+void RifReaderEclipseUnifiedRestartFile::readWellData(well_info_type* well_info)
+{
+    if (!well_info) return;
+
+    well_info_add_UNRST_wells(well_info, m_file->filePointer());
+}
+#endif
+