ResInsight/ApplicationCode/ReservoirDataModel/RigNNCData.h

/////////////////////////////////////////////////////////////////////////////////
//
//  Copyright (C) Statoil ASA, Ceetron Solutions 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 "cvfBase.h"
#include "cvfObject.h"
#include "cvfVector3.h"

#include <vector>
#include <map>

#include "cvfStructGrid.h"
#include "cafFixedArray.h"

class RigMainGrid;


class RigConnection
{
public:
    RigConnection( ) 
        : m_c1GlobIdx(cvf::UNDEFINED_SIZE_T),
          m_c1Face(cvf::StructGridInterface::NO_FACE),
          m_c2GlobIdx(cvf::UNDEFINED_SIZE_T)
    {}

    bool hasCommonArea() const
    {
        return m_polygon.size() > 0;
    }

    size_t                              m_c1GlobIdx;
    cvf::StructGridInterface::FaceType  m_c1Face;
    size_t                              m_c2GlobIdx;

    std::vector<cvf::Vec3d>             m_polygon;
};


class RigNNCData : public cvf::Object
{
   
public:
    RigNNCData();

    void processConnections(const RigMainGrid& mainGrid);
  
    std::vector<RigConnection>&         connections()        { return m_connections; }
    const std::vector<RigConnection>&   connections() const  { return m_connections; };

    std::vector<double>& makeConnectionScalarResult(size_t scalarResultIndex) 
    { 
        std::vector<double>& results = m_connectionResults[scalarResultIndex]; 
        results.resize(m_connections.size(), HUGE_VAL); 
        return results; 
    }

    const std::vector<double>* connectionScalarResult(size_t scalarResultIndex) const
    {
        std::map<size_t, std::vector<double> >::const_iterator it = m_connectionResults.find(scalarResultIndex);
        if (it != m_connectionResults.end())
            return &(it->second);
        else
            return NULL;
    }

    void setCombTransmisibilityScalarResultIndex(size_t scalarResultIndex)
    {
        std::map<size_t, std::vector<double> >::iterator it = m_connectionResults.find(cvf::UNDEFINED_SIZE_T);
        CVF_ASSERT(it != m_connectionResults.end());

        std::vector<double>& emptyData = m_connectionResults[scalarResultIndex];
        std::vector<double>& realData = m_connectionResults[cvf::UNDEFINED_SIZE_T];
        emptyData.swap(realData);
        m_connectionResults.erase(cvf::UNDEFINED_SIZE_T);
    }

private: // This section is possibly not needed
    //const std::vector<size_t>& findConnectionIndices(size_t reservoirCellIndex, cvf::StructGridInterface::FaceType face) const;
    //typedef std::map<size_t, caf::FixedArray<std::vector<size_t>, 7 > > ConnectionSearchMap;
    //ConnectionSearchMap m_cellIdxToFaceToConnectionIdxMap;

private:
    std::vector<RigConnection> m_connections; 
    std::map<size_t, std::vector<double> > m_connectionResults; ///< scalarResultIndex to value array map
};