pq2/dofhelper.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:
//
// SPDX-FileCopyrightText: Copyright © DuMux Project contributors, see AUTHORS.md in root folder
// SPDX-License-Identifier: GPL-3.0-or-later
//
#ifndef DUMUX_DISCRETIZATION_PQ2_DOF_HELPER_HH
#define DUMUX_DISCRETIZATION_PQ2_DOF_HELPER_HH
 
#include <dune/common/fvector.hh>
#include <dune/geometry/referenceelements.hh>
 
namespace Dumux {

template<class GridView>
struct PQ2LagrangeDofHelper
{
    using Scalar = typename GridView::ctype;
    static constexpr int dim = GridView::dimension;
    using GlobalPosition = Dune::FieldVector<Scalar, GridView::dimensionworld>;

    template<class DofMapper, class Element, class LocalKey, class IdSet>
    static std::size_t dofIndex(const DofMapper& m, const Element& e,
                                const LocalKey& lk, const IdSet& /*idSet*/)
    { return m.subIndex(e, lk.subEntity(), lk.codim()); }

    template<class DofMapper, class Element, class LocalKey>
    static std::size_t dofIndex(const DofMapper& m, const Element& e, const LocalKey& lk)
    { return m.subIndex(e, lk.subEntity(), lk.codim()); }

    template<class Geometry, class LocalKey>
    static GlobalPosition dofPosition(const Geometry& geo, const LocalKey& lk)
    {
        if (lk.codim() == dim) return geo.corner(lk.subEntity());
        if (lk.codim() == 0)  return geo.center();
        const auto& ref = Dune::referenceElement<Scalar, dim>(geo.type());
        return geo.global(ref.position(lk.subEntity(), lk.codim()));
    }
};
 
} // namespace Dumux
 
#endif // DUMUX_DISCRETIZATION_PQ2_DOF_HELPER_HH