gridcommhandles.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) 2011-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_GRID_COMM_HANDLES_HH
#define EWOMS_GRID_COMM_HANDLES_HH

#include <dune/grid/common/datahandleif.hh>
#include <dune/common/version.hh>

namespace Ewoms {

template <class FieldType, class Container, class EntityMapper, int commCodim>
class GridCommHandleSum
    : public Dune::CommDataHandleIF<GridCommHandleSum<FieldType, Container,
                                                      EntityMapper, commCodim>,
                                    FieldType>
{
public:
    GridCommHandleSum(Container &container, const EntityMapper &mapper)
        : mapper_(mapper), container_(container)
    {}

    bool contains(int dim, int codim) const
    {
        // return true if the codim is the same as the codim which we
        // are asked to communicate with.
        return codim == commCodim;
    }

    bool fixedsize(int dim, int codim) const
    {
        // for each DOF we communicate a single value which has a
        // fixed size
        return true;
    }

    template <class EntityType>
    size_t size(const EntityType &e) const
    {
        // communicate a field type per entity
        return 1;
    }

    template <class MessageBufferImp, class EntityType>
    void gather(MessageBufferImp &buff, const EntityType &e) const
    {
#if DUNE_VERSION_NEWER(DUNE_COMMON, 2, 4)
        int dofIdx = mapper_.index(e);
#else
        int dofIdx = mapper_.map(e);
#endif
        buff.write(container_[dofIdx]);
    }

    template <class MessageBufferImp, class EntityType>
    void scatter(MessageBufferImp &buff, const EntityType &e, size_t n)
    {
#if DUNE_VERSION_NEWER(DUNE_COMMON, 2, 4)
        int dofIdx = mapper_.index(e);
#else
        int dofIdx = mapper_.map(e);
#endif

        FieldType tmp;
        buff.read(tmp);
        container_[dofIdx] += tmp;
    }

private:
    const EntityMapper &mapper_;
    Container &container_;
};

template <class FieldType, class Container, class EntityMapper, int commCodim>
class GridCommHandleGhostSync
    : public Dune::CommDataHandleIF<GridCommHandleGhostSync<FieldType, Container,
                                                            EntityMapper, commCodim>,
                                    FieldType>
{
public:
    GridCommHandleGhostSync(Container &container, const EntityMapper &mapper)
        : mapper_(mapper), container_(container)
    {
    }

    bool contains(int dim, int codim) const
    {
        // return true if the codim is the same as the codim which we
        // are asked to communicate with.
        return codim == commCodim;
    }

    bool fixedsize(int dim, int codim) const
    {
        // for each DOF we communicate a single value which has a
        // fixed size
        return true;
    }

    template <class EntityType>
    size_t size(const EntityType &e) const
    {
        // communicate a field type per entity
        return 1;
    }

    template <class MessageBufferImp, class EntityType>
    void gather(MessageBufferImp &buff, const EntityType &e) const
    {
#if DUNE_VERSION_NEWER(DUNE_COMMON, 2, 4)
        int dofIdx = mapper_.index(e);
#else
        int dofIdx = mapper_.map(e);
#endif
        buff.write(container_[dofIdx]);
    }

    template <class MessageBufferImp, class EntityType>
    void scatter(MessageBufferImp &buff, const EntityType &e, size_t n)
    {
#if DUNE_VERSION_NEWER(DUNE_COMMON, 2, 4)
        int dofIdx = mapper_.index(e);
#else
        int dofIdx = mapper_.map(e);
#endif
        buff.read(container_[dofIdx]);
    }

private:
    const EntityMapper &mapper_;
    Container &container_;
};

template <class FieldType, class Container, class EntityMapper, int commCodim>
class GridCommHandleMax
    : public Dune::CommDataHandleIF<GridCommHandleMax<FieldType, Container,
                                                      EntityMapper, commCodim>,
                                    FieldType>
{
public:
    GridCommHandleMax(Container &container, const EntityMapper &mapper)
        : mapper_(mapper), container_(container)
    {}

    bool contains(int dim, int codim) const
    {
        // return true if the codim is the same as the codim which we
        // are asked to communicate with.
        return codim == commCodim;
    }

    bool fixedsize(int dim, int codim) const
    {
        // for each DOF we communicate a single value which has a
        // fixed size
        return true;
    }

    template <class EntityType>
    size_t size(const EntityType &e) const
    {
        // communicate a field type per entity
        return 1;
    }

    template <class MessageBufferImp, class EntityType>
    void gather(MessageBufferImp &buff, const EntityType &e) const
    {
#if DUNE_VERSION_NEWER(DUNE_COMMON, 2, 4)
        int dofIdx = mapper_.index(e);
#else
        int dofIdx = mapper_.map(e);
#endif
        buff.write(container_[dofIdx]);
    }

    template <class MessageBufferImp, class EntityType>
    void scatter(MessageBufferImp &buff, const EntityType &e, size_t n)
    {
#if DUNE_VERSION_NEWER(DUNE_COMMON, 2, 4)
        int dofIdx = mapper_.index(e);
#else
        int dofIdx = mapper_.map(e);
#endif
        FieldType tmp;
        buff.read(tmp);
        container_[dofIdx] = std::max(container_[dofIdx], tmp);
    }

private:
    const EntityMapper &mapper_;
    Container &container_;
};

template <class FieldType, class Container, class EntityMapper, int commCodim>
class GridCommHandleMin
    : public Dune::CommDataHandleIF<GridCommHandleMin<FieldType, Container,
                                                      EntityMapper, commCodim>,
                                    FieldType>
{
public:
    GridCommHandleMin(Container &container, const EntityMapper &mapper)
        : mapper_(mapper), container_(container)
    {}

    bool contains(int dim, int codim) const
    {
        // return true if the codim is the same as the codim which we
        // are asked to communicate with.
        return codim == commCodim;
    }

    bool fixedsize(int dim, int codim) const
    {
        // for each DOF we communicate a single value which has a
        // fixed size
        return true;
    }

    template <class EntityType>
    size_t size(const EntityType &e) const
    {
        // communicate a field type per entity
        return 1;
    }

    template <class MessageBufferImp, class EntityType>
    void gather(MessageBufferImp &buff, const EntityType &e) const
    {
#if DUNE_VERSION_NEWER(DUNE_COMMON, 2, 4)
        int dofIdx = mapper_.index(e);
#else
        int dofIdx = mapper_.map(e);
#endif
        buff.write(container_[dofIdx]);
    }

    template <class MessageBufferImp, class EntityType>
    void scatter(MessageBufferImp &buff, const EntityType &e, size_t n)
    {
#if DUNE_VERSION_NEWER(DUNE_COMMON, 2, 4)
        int dofIdx = mapper_.index(e);
#else
        int dofIdx = mapper_.map(e);
#endif
        FieldType tmp;
        buff.read(tmp);
        container_[dofIdx] = std::min(container_[dofIdx], tmp);
    }

private:
    const EntityMapper &mapper_;
    Container &container_;
};

} // namespace Ewoms

#endif