GridHelpers.hpp

Go to the documentation of this file.

/*
  Copyright 2014, 2015 Dr. Markus Blatt - HPC-Simulation-Software & Services
  Copyright 2014 Statoil AS
  Copyright 2015

  This file is part of the Open Porous Media project (OPM).

  OPM 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.

  OPM 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 for more details.

  You should have received a copy of the GNU General Public License
  along with OPM.  If not, see <http://www.gnu.org/licenses/>.
*/
#ifndef OPM_CORE_GRIDHELPERS_HEADER_INCLUDED
#define OPM_CORE_GRIDHELPERS_HEADER_INCLUDED
#include <opm/core/grid.h>
#include <boost/range/iterator_range.hpp>

namespace Opm
{
namespace UgGridHelpers
{

class SparseTableView
{
public:
    class IntRange : public boost::iterator_range<const int*>
    {
    public:
        typedef boost::iterator_range<const int*> BaseRowType;
        typedef BaseRowType::size_type size_type;
        typedef int value_type;
        
        IntRange(const int* start_arg, const int* end_arg)
            : BaseRowType(start_arg, end_arg)
        {}
    };
    typedef boost::iterator_range<const int*> row_type;

    SparseTableView(int* data, int *offset, std::size_t size_arg)
        : data_(data), offset_(offset), size_(size_arg)
    {}

    row_type operator[](std::size_t row) const
    {
        assert(row<=size());
        return row_type(data_ + offset_[row], data_ + offset_[row+1]);
    }
    
    std::size_t size() const
    {
        return size_;
    }
    
    std::size_t noEntries() const
    {
        return offset_[size_];
    }

private:
    const int* data_;
    const int* offset_;
    std::size_t size_;
};

int numCells(const UnstructuredGrid& grid);

int numFaces(const UnstructuredGrid& grid);

int dimensions(const UnstructuredGrid& grid);

int numCellFaces(const UnstructuredGrid& grid);

const int* cartDims(const UnstructuredGrid& grid);

const int* globalCell(const UnstructuredGrid& grid);

template<class G>
struct CellCentroidTraits
{
};

template<>
struct CellCentroidTraits<UnstructuredGrid>
{
    typedef const double* IteratorType;
    typedef const double* ValueType;
};

CellCentroidTraits<UnstructuredGrid>::IteratorType
beginCellCentroids(const UnstructuredGrid& grid);


double cellCenterDepth(const UnstructuredGrid& grid, int cell_index);


double cellCentroidCoordinate(const UnstructuredGrid& grid, int cell_index,
                                 int coordinate);


const double* cellCentroid(const UnstructuredGrid& grid, int cell_index);


double cellVolume(const UnstructuredGrid& grid, int cell_index);

template<class T>
struct CellVolumeIteratorTraits
{
};

template<>
struct CellVolumeIteratorTraits<UnstructuredGrid>
{
    typedef const double* IteratorType;
};

const double* beginCellVolumes(const UnstructuredGrid& grid);

const double* endCellVolumes(const UnstructuredGrid& grid);

const double* faceCentroid(const UnstructuredGrid& grid, int face_index);


template<class G>
struct FaceCentroidTraits
{
};

template<>
struct FaceCentroidTraits<UnstructuredGrid>
{
    typedef const double* IteratorType;
    typedef const double* ValueType;
};

FaceCentroidTraits<UnstructuredGrid>::IteratorType
beginFaceCentroids(const UnstructuredGrid& grid);

FaceCentroidTraits<UnstructuredGrid>::ValueType
faceCentroid(const UnstructuredGrid& grid, int face_index);

const double* faceNormal(const UnstructuredGrid& grid, int face_index);

double faceArea(const UnstructuredGrid& grid, int face_index);

int faceTag(const UnstructuredGrid& grid, boost::iterator_range<const int*>::const_iterator cell_face);

template<class T>
struct Cell2FacesTraits
{
};

template<>
struct Cell2FacesTraits<UnstructuredGrid>
{
    typedef SparseTableView Type;
};

template<class T>
struct Face2VerticesTraits
{
};

template<>
struct Face2VerticesTraits<UnstructuredGrid>
{
    typedef SparseTableView Type;
};

Cell2FacesTraits<UnstructuredGrid>::Type 
cell2Faces(const UnstructuredGrid& grid);

Face2VerticesTraits<UnstructuredGrid>::Type 
face2Vertices(const UnstructuredGrid& grid);

const double* vertexCoordinates(const UnstructuredGrid& grid, int index);

class FaceCellsProxy
{
public:
    FaceCellsProxy(const UnstructuredGrid& grid)
    : face_cells_(grid.face_cells)
    {}
    int operator()(int face_index, int local_index) const
    {
        return face_cells_[2*face_index+local_index];
    }
private:
    const int* face_cells_;
};

template<class T>
struct FaceCellTraits
{};

template<>
struct FaceCellTraits<UnstructuredGrid>
{
    typedef FaceCellsProxy Type;
};

FaceCellTraits<UnstructuredGrid>::Type faceCells(const UnstructuredGrid& grid);

template<class T>
T* increment(T* cc, int i, int dim)
{
    return cc+(i*dim);
}
template<class T>
T increment(const T& t, int i, int)
{
    return t+i;
}

template<class T>
double getCoordinate(T* cc, int i)
{
    return cc[i];
}

template<class T>
double getCoordinate(T t, int i)
{
    return (*t)[i];
}

} // end namespace UGGridHelpers
} // end namespace OPM
#endif