ResInsight/ApplicationCode/ReservoirDataModel/RigReservoirBuilderMock.h

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

class RigEclipseCaseData;
class RigMainGrid;
class RigConnection;
class RigGridBase;
class RigCell;

class QString;

class LocalGridRefinement
{
public:
    LocalGridRefinement(const cvf::Vec3st& mainGridMin, const cvf::Vec3st& mainGridMax, const cvf::Vec3st& singleCellRefinementFactors)
    {
        m_mainGridMinCellPosition = mainGridMin;
        m_mainGridMaxCellPosition = mainGridMax;
        m_singleCellRefinementFactors = singleCellRefinementFactors;
    }

    cvf::Vec3st m_mainGridMinCellPosition;
    cvf::Vec3st m_mainGridMaxCellPosition;
    cvf::Vec3st m_singleCellRefinementFactors;
};

class RigReservoirBuilderMock
{
public:
    RigReservoirBuilderMock();

    void setWorldCoordinates(cvf::Vec3d minWorldCoordinate, cvf::Vec3d maxWorldCoordinate);
    void setGridPointDimensions(const cvf::Vec3st& gridPointDimensions);
    void setResultInfo(size_t resultCount, size_t timeStepCount);
    void enableWellData(bool enableWellData);

    size_t resultCount() const { return m_resultCount; }
    size_t timeStepCount() const { return m_timeStepCount; }
    cvf::Vec3st gridPointDimensions() const { return m_gridPointDimensions; }

    void addLocalGridRefinement(const cvf::Vec3st& minCellPosition, const cvf::Vec3st& maxCellPosition, const cvf::Vec3st& singleCellRefinementFactors);

    void populateReservoir(RigEclipseCaseData* eclipseCase);

    bool inputProperty(RigEclipseCaseData* eclipseCase, const QString& propertyName, std::vector<double>* values );
    bool staticResult(RigEclipseCaseData* eclipseCase, const QString& result, std::vector<double>* values );
    bool dynamicResult(RigEclipseCaseData* eclipseCase, const QString& result, size_t stepIndex, std::vector<double>* values );

private:
    void        addFaults(RigEclipseCaseData* eclipseCase);

    static void addNnc(RigMainGrid* grid, size_t i1, size_t j1, size_t k1, size_t i2, size_t j2, size_t k2, std::vector<RigConnection> &nncConnections);
    void        addWellData(RigEclipseCaseData* eclipseCase, RigGridBase* grid);
    static void appendCells(size_t nodeStartIndex, size_t cellCount, RigGridBase* hostGrid, std::vector<RigCell>& cells);
    
    static void appendNodes(const cvf::Vec3d& min, const cvf::Vec3d& max, const cvf::Vec3st& cubeDimension, std::vector<cvf::Vec3d>& nodes);
    static void appendCubeNodes(const cvf::Vec3d& min, const cvf::Vec3d& max, std::vector<cvf::Vec3d>& nodes);

    size_t cellIndexFromIJK(size_t i, size_t j, size_t k) const
    {
        CVF_TIGHT_ASSERT(i < (m_gridPointDimensions.x() - 1));
        CVF_TIGHT_ASSERT(j < (m_gridPointDimensions.y() - 1));
        CVF_TIGHT_ASSERT(k < (m_gridPointDimensions.z() - 1));

        size_t ci = i + j*(m_gridPointDimensions.x() - 1) + k*((m_gridPointDimensions.x() - 1)*(m_gridPointDimensions.y() - 1));
        return ci;
    }

    cvf::Vec3st cellDimension()
    {
        return cvf::Vec3st(m_gridPointDimensions.x() - 1, m_gridPointDimensions.y() - 1, m_gridPointDimensions.z() - 1);
    }


private:
    cvf::Vec3d m_minWorldCoordinate;
    cvf::Vec3d m_maxWorldCoordinate;
    cvf::Vec3st m_gridPointDimensions;
    size_t      m_resultCount;
    size_t      m_timeStepCount;
    bool        m_enableWellData;

    std::vector<LocalGridRefinement> m_localGridRefinements;
};