/////////////////////////////////////////////////////////////////////////////////
//
//  Copyright (C) 2015-     Statoil ASA
//  Copyright (C) 2015-     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 "cvfStructGrid.h"

class RigFemPart;

class RigFemPartGrid : public cvf::StructGridInterface
{
public:
    explicit RigFemPartGrid(const RigFemPart* femPart);
    virtual ~RigFemPartGrid();

    bool        ijkFromCellIndex(size_t cellIndex, size_t* i, size_t* j, size_t* k) const override;
    size_t      cellIndexFromIJK(size_t i, size_t j, size_t k) const override;

    size_t      gridPointCountI() const override;
    size_t      gridPointCountJ() const override;
    size_t      gridPointCountK() const override;
 
    cvf::Vec3i          findMainIJKFaces(int elementIndex) const;
    
    std::pair<cvf::Vec3st, cvf::Vec3st> reservoirIJKBoundingBox() const;

 private:
    void                generateStructGridData();
    int                 findElmIdxForIJK000();
    int                 perpendicularFaceInDirection(cvf::Vec3f direction, int perpFaceIdx, int elmIdx);

    const RigFemPart*  m_femPart;

    std::vector<cvf::Vec3i> m_ijkPrElement;
    cvf::Vec3st         m_elmentIJKCounts;

private: // Unused, Not implemented
    bool        isCellValid(size_t i, size_t j, size_t k) const override;
    cvf::Vec3d  minCoordinate() const override;
    cvf::Vec3d  maxCoordinate() const override;
    bool        cellIJKNeighbor(size_t i, size_t j, size_t k, FaceType face, size_t* neighborCellIndex) const override;


    bool        cellIJKFromCoordinate(const cvf::Vec3d& coord, size_t* i, size_t* j, size_t* k) const override;
    virtual void        cellCornerVertices(size_t cellIndex, cvf::Vec3d vertices[8]) const;
    cvf::Vec3d  cellCentroid(size_t cellIndex) const override;
    void        cellMinMaxCordinates(size_t cellIndex, cvf::Vec3d* minCoordinate, cvf::Vec3d* maxCoordinate) const override;
    size_t      gridPointIndexFromIJK(size_t i, size_t j, size_t k) const override;
    cvf::Vec3d  gridPointCoordinate(size_t i, size_t j, size_t k) const override;

    class IJKArray
    {
    public:
        IJKArray(): m_iCount(0), m_jCount(0){}

        void resize(size_t iCount, size_t jCount, size_t kCount)
        {
            data.resize(iCount*jCount*kCount, cvf::UNDEFINED_SIZE_T);
            m_iCount = iCount;
            m_jCount = jCount;
        }

        size_t& at(size_t i, size_t j, size_t k)
        {
            return data[i + j* m_iCount + k * m_iCount*m_jCount];
        }
        
        size_t at(size_t i, size_t j, size_t k) const
        {
            return data[i + j* m_iCount + k * m_iCount*m_jCount];
        }

    private:
        size_t m_iCount;
        size_t m_jCount;

        std::vector< size_t > data;
    };

    IJKArray m_elmIdxPrIJK; 
    std::pair<cvf::Vec3st, cvf::Vec3st> m_reservoirIJKBoundingBox;
};