vtkvectorfunction.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 VTK_VECTOR_FUNCTION_HH
#define VTK_VECTOR_FUNCTION_HH

#include <ewoms/io/baseoutputwriter.hh>

#include <dune/grid/io/file/vtk/function.hh>
#include <dune/istl/bvector.hh>
#include <dune/common/fvector.hh>
#include <dune/common/version.hh>

#include <opm/common/Exceptions.hpp>
#include <opm/common/ErrorMacros.hpp>

#include <string>
#include <limits>
#include <vector>

namespace Ewoms {

template <class GridView, class Mapper>
class VtkVectorFunction : public Dune::VTKFunction<GridView>
{
    enum { dim = GridView::dimension };
    typedef typename GridView::ctype ctype;
    typedef typename GridView::template Codim<0>::Entity Element;

    typedef BaseOutputWriter::VectorBuffer VectorBuffer;

public:
    VtkVectorFunction(std::string name,
                      const GridView &gridView,
                      const Mapper &mapper,
                      const VectorBuffer &buf,
                      int codim)
        : name_(name)
        , gridView_(gridView)
        , mapper_(mapper)
        , buf_(buf)
        , codim_(codim)
    { assert(int(buf_.size()) == mapper_.size()); }

    virtual std::string name() const
    { return name_; }

    virtual int ncomps() const
    { return buf_[0].size(); }

    virtual double evaluate(int mycomp,
                            const Element &e,
                            const Dune::FieldVector<ctype, dim> &xi) const
    {
        int idx;
        if (codim_ == 0) {
            // cells. map element to the index
#if DUNE_VERSION_NEWER(DUNE_COMMON, 2, 4)
            idx = mapper_.index(e);
#else
            idx = mapper_.map(e);
#endif
        }
        else if (codim_ == dim) {
            // find vertex which is closest to xi in local
            // coordinates. This code is based on Dune::P1VTKFunction
            double min = 1e100;
            int imin = -1;
            Dune::GeometryType gt = e.type();
#if DUNE_VERSION_NEWER(DUNE_COMMON, 2, 4)
            int n = e.subEntities(dim);
#else
            int n = e.template count<dim>();
#endif
            for (int i = 0; i < n; ++i) {
#if DUNE_VERSION_NEWER(DUNE_COMMON, 2, 3)
                Dune::FieldVector<ctype, dim> local =
                    Dune::ReferenceElements<ctype, dim>::general(gt).position(i, dim);
#else
                Dune::FieldVector<ctype, dim> local =
                    Dune::GenericReferenceElements<ctype, dim>::general(gt).position(i, dim);
#endif

                local -= xi;
                if (local.infinity_norm() < min) {
                    min = local.infinity_norm();
                    imin = i;
                }
            }

            // map vertex to an index
#if DUNE_VERSION_NEWER(DUNE_COMMON, 2, 4)
            idx = mapper_.subIndex(e, imin, codim_);
#else
            idx = mapper_.map(e, imin, codim_);
#endif
        }
        else
            OPM_THROW(std::logic_error, "Only element and vertex based vector "
                                        " fields are supported so far.");

        double val = buf_[idx][mycomp];
        if (std::abs(val) < std::numeric_limits<float>::min())
            val = 0;

        return val;
    }

private:
    const std::string name_;
    const GridView gridView_;
    const Mapper &mapper_;
    const VectorBuffer &buf_;
    int codim_;
};

} // namespace Ewoms

#endif