fracturemapper.hh

Go to the documentation of this file.

// -*- mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*-
// vi: set et ts=4 sw=4 sts=4:
/*
  Copyright (C) 2012-2013 by Andreas Lauser

  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 2 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 EWOMS_FRACTURE_MAPPER_HH
#define EWOMS_FRACTURE_MAPPER_HH

#include <ewoms/common/propertysystem.hh>

#include <algorithm>
#include <set>

namespace Ewoms {

template <class TypeTag>
class FractureMapper
{
    struct FractureEdge
    {
        FractureEdge(int edgeVertex1Idx, int edgeVertex2Idx)
            : i_(std::min(edgeVertex1Idx, edgeVertex2Idx)),
              j_(std::max(edgeVertex1Idx, edgeVertex2Idx))
        {}

        bool operator<(const FractureEdge &e) const
        { return i_ < e.i_ || (i_ == e.i_ && j_ < e.j_); }

        bool operator==(const FractureEdge &e) const
        { return i_ == e.i_ && j_ == e.j_; }

        int i_;
        int j_;
    };

public:
    FractureMapper()
    {}

    void addFractureEdge(int vertexIdx1, int vertexIdx2)
    {
        fractureEdges_.insert(FractureEdge(vertexIdx1, vertexIdx2));
        fractureVertices_.insert(vertexIdx1);
        fractureVertices_.insert(vertexIdx2);
    }

    bool isFractureVertex(unsigned vertexIdx) const
    { return fractureVertices_.count(vertexIdx) > 0; }

    bool isFractureEdge(unsigned vertex1Idx, unsigned vertex2Idx) const
    {
        FractureEdge tmp(vertex1Idx, vertex2Idx);
        return fractureEdges_.count(tmp) > 0;
    }

private:
    std::set<FractureEdge> fractureEdges_;
    std::set<unsigned> fractureVertices_;
};

} // namespace Ewoms

#endif // EWOMS_FRACTURE_MAPPER_HH