Added inherited object of MimeData to carry model index list

p4#: 20926

ApplicationCode/ProjectDataModel/RimReservoir.cpp

@@ -35,7 +35,9 @@
 #include <QString>
 #include "RimProject.h"
 
-//--------------------------------------------------------------------------------------------------
+CAF_PDM_SOURCE_INIT(RimReservoir, "RimReservoir");
+
+//------------------------------------------------------------------------------------------------
 /// 
 //--------------------------------------------------------------------------------------------------
 RimReservoir::RimReservoir()

ApplicationCode/ProjectDataModel/RimReservoir.h

@@ -40,12 +40,12 @@ class RimCaseCollection;
 //==================================================================================================
 class RimReservoir : public caf::PdmObject
 {
-
+    CAF_PDM_HEADER_INIT;
 public:
     RimReservoir();
     virtual ~RimReservoir();
 
-    virtual bool                openEclipseGridFile() = 0;
+    virtual bool                openEclipseGridFile() { return false;}; // Should be pure virtual but PDM does not allow that.
                                       
     RigEclipseCase*             reservoirData();
     const RigEclipseCase*       reservoirData() const;
@@ -78,5 +78,5 @@ private:
 
 protected:
     cvf::ref<RigEclipseCase>        m_rigEclipseCase;
-};
 
+};

ApplicationCode/ProjectDataModel/RimReservoirCellResultsCacher.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 "RIStdInclude.h"
+#include "RimReservoirCellResultsCacher.h"
+#include "RigReservoirCellResults.h"
+#include "RIApplication.h"
+
+CAF_PDM_SOURCE_INIT(RimReservoirCellResultsCacher, "ReservoirCellResultCacher");
+
+//--------------------------------------------------------------------------------------------------
+/// 
+//--------------------------------------------------------------------------------------------------
+RimReservoirCellResultsCacher::RimReservoirCellResultsCacher()
+    : m_cellResults(NULL)
+{
+    CAF_PDM_InitObject("Cacher", "", "", "");
+
+    CAF_PDM_InitField(&m_resultCacheFileName, "ResultCacheFileName",  QString(), "UiDummyname", "", "" ,"");
+    m_resultCacheFileName.setUiHidden(true);
+    CAF_PDM_InitFieldNoDefault(&m_resultCacheMetaData, "ResultCacheEntries", "UiDummyname", "", "", "");
+    m_resultCacheMetaData.setUiHidden(true);
+
+}
+
+//--------------------------------------------------------------------------------------------------
+/// 
+//--------------------------------------------------------------------------------------------------
+RimReservoirCellResultsCacher::~RimReservoirCellResultsCacher()
+{
+    m_resultCacheMetaData.deleteAllChildObjects();
+}
+
+//--------------------------------------------------------------------------------------------------
+/// 
+//--------------------------------------------------------------------------------------------------
+void RimReservoirCellResultsCacher::setupBeforeSave()
+{
+    if (!m_cellResults) return;
+
+    const std::vector<RigReservoirCellResults::ResultInfo>&  resInfo = m_cellResults->infoForEachResultIndex();
+    m_resultCacheMetaData.deleteAllChildObjects();
+
+    if(resInfo.size())
+    {
+        QDir::root().mkpath(getCacheDirectoryPath());
+
+        QString newValidCacheFileName = getValidCacheFileName();
+        QFile cacheFile(newValidCacheFileName);
+
+        if (!cacheFile.open(QIODevice::WriteOnly)) qWarning() << "Saving project: Can't open the cache file : " + newValidCacheFileName; 
+
+        QDataStream stream(&cacheFile);
+        stream.setVersion(QDataStream::Qt_4_0);
+        stream << (quint32)0xCEECAC4E; // magic number
+        stream << (qint32)1; // Version
+
+        for (int rIdx = 0; rIdx < resInfo.size(); ++rIdx)
+        {
+            RimReservoirCellResultsCacheEntryInfo*  cacheEntry = new RimReservoirCellResultsCacheEntryInfo;
+            m_resultCacheMetaData.push_back(cacheEntry);
+
+            cacheEntry->m_resultType = resInfo[rIdx].m_resultType;
+            cacheEntry->m_resultName = resInfo[rIdx].m_resultName;
+            cacheEntry->m_timeStepDates = resInfo[rIdx].m_timeStepDates;
+
+            cacheEntry->m_filePosition = cacheFile.pos();
+
+            for (int tsIdx = 0; tsIdx < resInfo[rIdx].m_timeStepDates.size() ; ++tsIdx)
+            {
+                std::vector<double>& data = m_cellResults->cellScalarResults(resInfo[rIdx].m_gridScalarResultIndex, tsIdx);
+                if (data.size())
+                {
+                    cacheEntry->m_timeStepHasData.v().push_back(1);
+
+                    for (size_t cIdx = 0; cIdx < data.size(); ++cIdx)
+                    {
+                        stream << data[cIdx];
+                    }
+                }
+                else
+                {
+                     cacheEntry->m_timeStepHasData.v().push_back(0);
+                }
+            }
+        }
+    }
+}
+
+//--------------------------------------------------------------------------------------------------
+/// 
+//--------------------------------------------------------------------------------------------------
+QString RimReservoirCellResultsCacher::getValidCacheFileName()
+{
+    QString cacheFileName;
+    if (m_resultCacheFileName().isEmpty())
+    {
+        QString newCacheDirPath =  getCacheDirectoryPath();
+        QUuid guid = QUuid::createUuid();
+        cacheFileName = newCacheDirPath + "/" + guid.toString();
+    }
+    else
+    {
+        // Make the path correct related to the possibly new project filename
+        QString newCacheDirPath =  getCacheDirectoryPath();
+        QFileInfo oldCacheFile(m_resultCacheFileName());
+
+        cacheFileName = newCacheDirPath + "/" + oldCacheFile.fileName();
+    }
+    return cacheFileName;
+}
+
+//--------------------------------------------------------------------------------------------------
+/// 
+//--------------------------------------------------------------------------------------------------
+QString RimReservoirCellResultsCacher::getCacheDirectoryPath()
+{
+    QString cacheDirPath;
+    QString projectFileName = RIApplication::instance()->project()->fileName();
+    QFileInfo fileInfo(projectFileName);
+    cacheDirPath = fileInfo.canonicalPath();
+    cacheDirPath += "/" + fileInfo.completeBaseName() + "_cache";
+    return cacheDirPath;
+}
+
+
+CAF_PDM_SOURCE_INIT(RimReservoirCellResultsCacheEntryInfo, "ResultCacheEntryInfo");
+
+//--------------------------------------------------------------------------------------------------
+/// 
+//--------------------------------------------------------------------------------------------------
+RimReservoirCellResultsCacheEntryInfo::RimReservoirCellResultsCacheEntryInfo()
+{
+    CAF_PDM_InitObject("Cache Entry", "", "", "");
+
+    CAF_PDM_InitField(&m_resultType, "ResultType",  caf::AppEnum<RimDefines::ResultCatType>(RimDefines::REMOVED), "ResultType", "", "" ,"");
+    CAF_PDM_InitField(&m_resultName, "ResultName",  QString(), "ResultName", "", "" ,"");
+    CAF_PDM_InitFieldNoDefault(&m_timeStepDates, "TimeSteps", "TimeSteps", "", "" ,"");
+    CAF_PDM_InitField(&m_filePosition, "FilePositionDataStart",  qint64(-1), "FilePositionDataStart", "", "" ,"");
+
+}
+
+//--------------------------------------------------------------------------------------------------
+/// 
+//--------------------------------------------------------------------------------------------------
+RimReservoirCellResultsCacheEntryInfo::~RimReservoirCellResultsCacheEntryInfo()
+{
+
+}

ApplicationCode/ProjectDataModel/RimReservoirCellResultsCacher.h

@@ -0,0 +1,69 @@
+/////////////////////////////////////////////////////////////////////////////////
+//
+//  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.
+//
+/////////////////////////////////////////////////////////////////////////////////
+
+#pragma once
+
+#include <QDateTime>
+#include "cafPdmField.h"
+#include "cafPdmObject.h"
+#include "cafAppEnum.h"
+#include "RimDefines.h"
+
+class RimReservoirCellResultsCacheEntryInfo;
+class RigReservoirCellResults;
+
+class RimReservoirCellResultsCacher : public caf::PdmObject
+{
+    CAF_PDM_HEADER_INIT;
+public:
+    RimReservoirCellResultsCacher();
+    virtual ~RimReservoirCellResultsCacher();
+
+    //RigReservoirCellResults* cellResults() const { return m_cellResults; }
+    void setCellResults(RigReservoirCellResults* cellResults) { m_cellResults = cellResults; }
+
+
+    virtual void setupBeforeSave();
+
+    caf::PdmField<QString>                                  m_resultCacheFileName;
+    caf::PdmPointersField<RimReservoirCellResultsCacheEntryInfo*> m_resultCacheMetaData;
+
+    
+
+private:
+    QString getValidCacheFileName();
+    QString getCacheDirectoryPath();
+
+    RigReservoirCellResults* m_cellResults;
+
+};
+
+class RimReservoirCellResultsCacheEntryInfo : public caf::PdmObject
+{
+    CAF_PDM_HEADER_INIT;
+public:
+    RimReservoirCellResultsCacheEntryInfo();
+    virtual ~RimReservoirCellResultsCacheEntryInfo();
+
+    caf::PdmField<caf::AppEnum< RimDefines::ResultCatType> > m_resultType;
+    caf::PdmField<QString>                                   m_resultName;
+    caf::PdmField< std::vector <QDateTime> >                 m_timeStepDates;
+    caf::PdmField< std::vector <int> >                       m_timeStepHasData;
+    caf::PdmField<qint64>                                    m_filePosition;
+};
+

ApplicationCode/ProjectDataModel/RimUiTreeModelPdm.h

@@ -22,6 +22,8 @@
 #include "cafPdmPointer.h"
 #include "cafUiTreeModelPdm.h"
 
+#include <QMimeData>
+
 class QFileSystemWatcher;
 
 class RimCellPropertyFilter;
@@ -31,6 +33,55 @@ class RimInputProperty;
 class RimStatisticalCalculation;
 class RimIdenticalGridCaseGroup;
 
+//--------------------------------------------------------------------------------------------------
+/// MimeData class used to carry a QModelIndexList
+//--------------------------------------------------------------------------------------------------
+class MimeDataWithIndexes : public QMimeData
+{
+    Q_OBJECT
+
+public:
+    MimeDataWithIndexes()
+    {
+    }
+
+    MimeDataWithIndexes(const MimeDataWithIndexes & other)
+    {
+        setIndexes(other.indexes());
+    }
+
+    void setIndexes(const QModelIndexList & indexes)
+    {
+        m_indexes = indexes;
+    }
+
+    const QModelIndexList& indexes() const { return m_indexes; }
+
+    virtual bool hasFormat( const QString &mimetype ) const
+    {
+        return (mimetype == formatName());
+    }
+
+    virtual QStringList formats() const
+    {
+        QStringList supportedFormats = QMimeData::formats();
+        supportedFormats << formatName();
+
+        return supportedFormats;
+    }
+
+    static QString formatName()
+    {
+        return "MimeDataWithIndexes";
+    }
+
+private:
+    QModelIndexList m_indexes;
+};
+
+Q_DECLARE_METATYPE(MimeDataWithIndexes)
+
+    
 //==================================================================================================
 ///
 ///