ColorAttrib.hh 7.36 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 54 55 56 57 58 59 60
/*===========================================================================*\
 *                                                                           *
 *                            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: 36 $                                                         *
 *   $Date: 2012-01-10 18:00:06 +0100 (Di, 10 Jan 2012) $                    *
 *   $LastChangedBy: kremer $                                                *
 *                                                                           *
\*===========================================================================*/

#ifndef COLORATTRIB_HH_
#define COLORATTRIB_HH_

#include <cassert>

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

namespace OpenVolumeMesh {

//== CLASS DEF ================================================================

template <class ColT>
class ColorAttrib {
public:

61
    ColorAttrib(TopologyKernel& _kernel, const ColT _def = ColT());
62 63 64 65 66 67 68 69 70 71 72 73 74

    virtual ~ColorAttrib();

    //==================
    // Vertices
    //==================
    const ColT& operator[](const VertexHandle& _h) const {
        assert((unsigned int)_h.idx() < vcolor_prop_.size());
        return vcolor_prop_[_h.idx()];
    }

    ColT& operator[](const VertexHandle& _h) {
        assert((unsigned int)_h.idx() < kernel_.n_vertices());
75
        vertex_colors_available_ = true;
76 77 78 79 80 81 82 83 84 85 86 87 88
        return vcolor_prop_[_h.idx()];
    }

    //==================
    // Edges
    //==================
    const ColT& operator[](const EdgeHandle& _h) const {
        assert((unsigned int)_h.idx() < ecolor_prop_.size());
        return ecolor_prop_[_h.idx()];
    }

    ColT& operator[](const EdgeHandle& _h) {
        assert((unsigned int)_h.idx() < kernel_.n_edges());
89
        edge_colors_available_ = true;
90 91 92 93 94 95 96 97 98 99 100 101 102
        return ecolor_prop_[_h.idx()];
    }

    //==================
    // Half-Edges
    //==================
    const ColT& operator[](const HalfEdgeHandle& _h) const {
        assert((unsigned int)_h.idx() < hecolor_prop_.size());
        return hecolor_prop_[_h.idx()];
    }

    ColT& operator[](const HalfEdgeHandle& _h) {
        assert((unsigned int)_h.idx() < kernel_.n_halfedges());
103
        halfedge_colors_available_ = true;
104 105 106 107 108 109 110 111 112 113 114 115 116
        return hecolor_prop_[_h.idx()];
    }

    //==================
    // Faces
    //==================
    const ColT& operator[](const FaceHandle& _h) const {
        assert((unsigned int)_h.idx() < fcolor_prop_.size());
        return fcolor_prop_[_h.idx()];
    }

    ColT& operator[](const FaceHandle& _h) {
        assert((unsigned int)_h.idx() < kernel_.n_faces());
117
        face_colors_available_ = true;
118 119 120 121 122 123 124 125 126 127 128 129 130
        return fcolor_prop_[_h.idx()];
    }

    //==================
    // Half-Faces
    //==================
    const ColT& operator[](const HalfFaceHandle& _h) const {
        assert((unsigned int)_h.idx() < hfcolor_prop_.size());
        return hfcolor_prop_[_h.idx()];
    }

    ColT& operator[](const HalfFaceHandle& _h) {
        assert((unsigned int)_h.idx() < kernel_.n_halffaces());
131
        halfface_colors_available_ = true;
132 133 134 135 136 137 138 139 140 141 142 143 144
        return hfcolor_prop_[_h.idx()];
    }

    //==================
    // Cells
    //==================
    const ColT& operator[](const CellHandle& _h) const {
        assert((unsigned int)_h.idx() < ccolor_prop_.size());
        return ccolor_prop_[_h.idx()];
    }

    ColT& operator[](const CellHandle& _h) {
        assert((unsigned int)_h.idx() < kernel_.n_cells());
145
        cell_colors_available_ = true;
146 147 148
        return ccolor_prop_[_h.idx()];
    }

149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164

    bool vertex_colors_available()   { return vertex_colors_available_;   }
    bool halfedge_colors_available() { return halfedge_colors_available_; }
    bool edge_colors_available()     { return edge_colors_available_;     }
    bool halfface_colors_available() { return halfface_colors_available_; }
    bool face_colors_available()     { return face_colors_available_;     }
    bool cell_colors_available()     { return cell_colors_available_;     }

    void clear_vertex_colors();
    void clear_halfedge_colors();
    void clear_edge_colors();
    void clear_halfface_colors();
    void clear_face_colors();
    void clear_cell_colors();


165 166 167 168 169 170 171 172 173 174
private:

    VertexPropertyT<ColT> vcolor_prop_;
    EdgePropertyT<ColT> ecolor_prop_;
    HalfEdgePropertyT<ColT> hecolor_prop_;
    FacePropertyT<ColT> fcolor_prop_;
    HalfFacePropertyT<ColT> hfcolor_prop_;
    CellPropertyT<ColT> ccolor_prop_;

    TopologyKernel& kernel_;
175 176 177 178 179 180 181 182 183 184

    bool vertex_colors_available_;
    bool halfedge_colors_available_;
    bool edge_colors_available_;
    bool halfface_colors_available_;
    bool face_colors_available_;
    bool cell_colors_available_;

    ColT default_color_;

185 186 187 188 189 190 191 192 193
};

} // Namespace OpenVolumeMesh

#if defined(INCLUDE_TEMPLATES) && !defined(COLORATTRIBT_CC)
#include "ColorAttribT.cc"
#endif

#endif /* COLORATTRIB_HH_ */