ResInsight/ApplicationCode/GeoMech/GeoMechDataModel/RigFemPartGrid.h

104 lines
3.5 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:
RigFemPartGrid();
~RigFemPartGrid() override;
void setFemPart( const RigFemPart* femPart );
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;
void cellCornerVertices( size_t cellIndex, cvf::Vec3d vertices[8] ) const override;
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;
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;
};