NormalAttrib.hh 5.62 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53
/*===========================================================================*\
 *                                                                           *
 *                            OpenVolumeMesh                                 *
 *        Copyright (C) 2011 by Computer Graphics Group, RWTH Aachen         *
 *                        www.openvolumemesh.org                             *
 *                                                                           *
 *---------------------------------------------------------------------------*
 *  This file is part of OpenVolumeMesh.                                     *
 *                                                                           *
 *  OpenVolumeMesh is free software: you can redistribute it and/or modify   *
 *  it under the terms of the GNU Lesser General Public License as           *
 *  published by the Free Software Foundation, either version 3 of           *
 *  the License, or (at your option) any later version with the              *
 *  following exceptions:                                                    *
 *                                                                           *
 *  If other files instantiate templates or use macros                       *
 *  or inline functions from this file, or you compile this file and         *
 *  link it with other files to produce an executable, this file does        *
 *  not by itself cause the resulting executable to be covered by the        *
 *  GNU Lesser General Public License. This exception does not however       *
 *  invalidate any other reasons why the executable file might be            *
 *  covered by the GNU Lesser General Public License.                        *
 *                                                                           *
 *  OpenVolumeMesh 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 Lesser General Public License for more details.                      *
 *                                                                           *
 *  You should have received a copy of the GNU LesserGeneral Public          *
 *  License along with OpenVolumeMesh.  If not,                              *
 *  see <http://www.gnu.org/licenses/>.                                      *
 *                                                                           *
\*===========================================================================*/

/*===========================================================================*\
 *                                                                           *
 *   $Revision$                                                         *
 *   $Date$                    *
 *   $LastChangedBy$                                                *
 *                                                                           *
\*===========================================================================*/

#ifndef NORMALATTRIB_HH_
#define NORMALATTRIB_HH_

#include <cassert>

#include "../Core/OpenVolumeMeshHandle.hh"
#include "OpenVolumeMeshStatus.hh"
#include "../Core/PropertyDefines.hh"

namespace OpenVolumeMesh {

54
template <class GeomKernelT>
55 56 57
class NormalAttrib {
public:

Jan Möbius's avatar
Jan Möbius committed
58
    explicit NormalAttrib(GeomKernelT& _kernel);
59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77
    virtual ~NormalAttrib();

    /** \brief A simple heuristic to estimate the vertex normals
     *
     * This function takes the vertices' surrounding outside
     * face normals into account and computes an average
     * out of it.
     */
    void update_vertex_normals();

    /** \brief Compute face normals
     *
     * This is accomplished by taking two adjacent half-edges
     * that are incident to the faces and compute their
     * cross product. Note that this method looses accuracy
     * in case the faces in question is not planar.
     */
    void update_face_normals();

78
    const typename GeomKernelT::PointT& operator[](const VertexHandle& _h) const {
Mike Kremer's avatar
Mike Kremer committed
79
        assert((unsigned int)_h.idx() < kernel_.n_vertices());
80 81 82
        return v_normals_[_h.idx()];
    }

83
    const typename GeomKernelT::PointT& operator[](const FaceHandle& _h) const {
Mike Kremer's avatar
Mike Kremer committed
84
        assert((unsigned int)_h.idx() < kernel_.n_faces());
85 86 87
        return f_normals_[_h.idx()];
    }

88
    const typename GeomKernelT::PointT operator[](const HalfFaceHandle& _h) const {
Mike Kremer's avatar
Mike Kremer committed
89 90 91 92 93 94
        assert((unsigned int)_h.idx() < kernel_.n_halffaces());
        double mult = 1.0;
        if(_h.idx() % 2 == 1) mult = -1.0;
        return f_normals_[kernel_.face_handle(_h).idx()] * mult;
    }

95
    typename GeomKernelT::PointT& operator[](const VertexHandle& _h) {
96
        assert((unsigned int)_h.idx() < kernel_.n_vertices());
97 98 99
        return v_normals_[_h.idx()];
    }

100 101 102 103 104
    typename GeomKernelT::PointT& operator[](const FaceHandle& _h) {
        assert((unsigned int)_h.idx() < kernel_.n_faces());
        return f_normals_[_h.idx()];
    }

105
    typename GeomKernelT::PointT operator[](const HalfFaceHandle& _h) {
Mike Kremer's avatar
Mike Kremer committed
106 107 108 109
        assert((unsigned int)_h.idx() < kernel_.n_halffaces());
        double mult = 1.0;
        if(_h.idx() % 2 == 1) mult = -1.0;
        return f_normals_[kernel_.face_handle(_h).idx()] * mult;
110 111 112 113 114 115 116 117
    }

private:

    void compute_vertex_normal(const VertexHandle& _vh);

    void compute_face_normal(const FaceHandle& _fh);

118
    GeomKernelT& kernel_;
119

120 121
    VertexPropertyT<typename GeomKernelT::PointT> v_normals_;
    FacePropertyT<typename GeomKernelT::PointT> f_normals_;
122

123 124 125 126 127 128 129 130 131
};

} // Namespace OpenVolumeMesh

#if defined(INCLUDE_TEMPLATES) && !defined(NORMALATTRIBT_CC)
#include "NormalAttribT.cc"
#endif

#endif /* NORMALATTRIB_HH_ */