ResInsight/ApplicationCode/GeoMech/GeoMechDataModel/RigFemPartGrid.h

104 lines
3.4 KiB
C++

/////////////////////////////////////////////////////////////////////////////////
//
// 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();
virtual bool ijkFromCellIndex(size_t cellIndex, size_t* i, size_t* j, size_t* k) const;
virtual size_t cellIndexFromIJK(size_t i, size_t j, size_t k) const;
virtual size_t gridPointCountI() const;
virtual size_t gridPointCountJ() const;
virtual size_t gridPointCountK() const;
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
virtual bool isCellValid(size_t i, size_t j, size_t k) const;
virtual cvf::Vec3d minCoordinate() const;
virtual cvf::Vec3d maxCoordinate() const;
virtual bool cellIJKNeighbor(size_t i, size_t j, size_t k, FaceType face, size_t* neighborCellIndex) const;
virtual bool cellIJKFromCoordinate(const cvf::Vec3d& coord, size_t* i, size_t* j, size_t* k) const;
virtual void cellCornerVertices(size_t cellIndex, cvf::Vec3d vertices[8]) const;
virtual cvf::Vec3d cellCentroid(size_t cellIndex) const;
virtual void cellMinMaxCordinates(size_t cellIndex, cvf::Vec3d* minCoordinate, cvf::Vec3d* maxCoordinate) const;
virtual size_t gridPointIndexFromIJK(size_t i, size_t j, size_t k) const;
virtual cvf::Vec3d gridPointCoordinate(size_t i, size_t j, size_t k) const;
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;
};