wedge_colors.h

/*****************************************************************************
 * VCLib                                                                     *
 * Visual Computing Library                                                  *
 *                                                                           *
 * Copyright(C) 2021-2025                                                    *
 * Visual Computing Lab                                                      *
 * ISTI - Italian National Research Council                                  *
 *                                                                           *
 * All rights reserved.                                                      *
 *                                                                           *
 * This program is free software; you can redistribute it and/or modify      *
 * it under the terms of the Mozilla Public License Version 2.0 as published *
 * by the Mozilla Foundation; either version 2 of the License, or            *
 * (at your option) any later version.                                       *
 *                                                                           *
 * This program 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              *
 * Mozilla Public License Version 2.0                                        *
 * (https://www.mozilla.org/en-US/MPL/2.0/) for more details.                *
 ****************************************************************************/
 
#ifndef VCL_MESH_COMPONENTS_WEDGE_COLORS_H
#define VCL_MESH_COMPONENTS_WEDGE_COLORS_H
 
#include "bases/container_component.h"
 
#include <vclib/concepts/mesh/components/color.h>
#include <vclib/concepts/mesh/components/wedge_colors.h>
#include <vclib/space/core/color.h>
#include <vclib/space/core/vector.h>
#include <vclib/types/view.h>
 
namespace vcl::comp {
 
template<int N, typename ParentElemType = void, bool OPT = false>
class WedgeColors :
        public ContainerComponent<
            WedgeColors<N, ParentElemType, OPT>,
            CompId::WEDGE_COLORS,
            vcl::Color,
            N,
            void,
            ParentElemType,
            !std::is_same_v<ParentElemType, void>,
            OPT,
            true>
{
    using Base = ContainerComponent<
        WedgeColors<N, ParentElemType, OPT>,
        CompId::WEDGE_COLORS,
        vcl::Color,
        N,
        void,
        ParentElemType,
        !std::is_same_v<ParentElemType, void>,
        OPT,
        true>;
 
public:
    using WedgeColorType = vcl::Color;
 
    /* Iterator Types declaration */
 
    using WedgeColorsIterator      = Base::Iterator;
    using ConstWedgeColorsIterator = Base::ConstIterator;
 
    static const int WEDGE_COLOR_NUMBER = Base::SIZE;
 
    /* Constructors */
 
    WedgeColors() = default;
 
    /* Member functions */
 
    vcl::Color& wedgeColor(uint i) { return colors().at(i); }
 
    const vcl::Color& wedgeColor(uint i) const { return colors().at(i); }
 
    vcl::Color& wedgeColorMod(int i) { return colors().atMod(i); }
 
    const vcl::Color& wedgeColorMod(int i) const { return colors().atMod(i); }
 
    void setWedgeColor(uint i, const vcl::Color& c) { colors().set(i, c); }
 
    template<Range Rng>
    void setWedgeColors(Rng&& r) requires InputRange<Rng, vcl::Color>
    {
        colors().set(r);
    }
 
    /* Iterator Member functions */
 
    WedgeColorsIterator wedgeColorBegin() { return colors().begin(); }
 
    WedgeColorsIterator wedgeColorEnd() { return colors().end(); }
 
    ConstWedgeColorsIterator wedgeColorBegin() const
    {
        return colors().begin();
    }
 
    ConstWedgeColorsIterator wedgeColorEnd() const { return colors().end(); }
 
    View<WedgeColorsIterator> wedgeColors()
    {
        return View(wedgeColorBegin(), wedgeColorEnd());
    }
 
    View<ConstWedgeColorsIterator> wedgeColors() const
    {
        return View(wedgeColorBegin(), wedgeColorEnd());
    }
 
    // dummy member to discriminate between WedgeColors and FaceHalfEdgePointers
    void __wedgeColors() const {}
 
protected:
    // Component interface functions
    template<typename Element>
    void importFrom(const Element& e, bool = true)
    {
        if constexpr (HasWedgeColors<Element>) {
            if (isWedgeColorsAvailableOn(e)) {
                if constexpr (N > 0) {
                    // same static size
                    if constexpr (N == Element::WEDGE_COLOR_NUMBER) {
                        importWedgeColorsFrom(e);
                    }
                    // from dynamic to static, but dynamic size == static size
                    else if constexpr (Element::WEDGE_COLOR_NUMBER < 0) {
                        if (e.vertexNumber() == N) {
                            importWedgeColorsFrom(e);
                        }
                    }
                    else {
                        // do not import in this case: cannot import from
                        // dynamic size != static size
                    }
                }
                else {
                    // from static/dynamic to dynamic size: need to resize
                    // first, then import
                    resize(e.vertexNumber());
                    importWedgeColorsFrom(e);
                }
            }
        }
    }
 
    void serialize(std::ostream& os) const { colors().serialize(os); }
 
    void deserialize(std::istream& is) { colors().deserialize(is); }
 
    // ContainerComponent interface functions
    void resize(uint n) requires (N < 0) { colors().resize(n); }
 
    void pushBack(const vcl::Color& c = vcl::Color()) requires (N < 0)
    {
        colors().pushBack(c);
    }
 
    void insert(uint i, const vcl::Color& c = vcl::Color()) requires (N < 0)
    {
        colors().insert(i, c);
    }
 
    void erase(uint i) requires (N < 0) { colors().erase(i); }
 
    void clear() requires (N < 0) { colors().clear(); }
 
private:
    template<typename Element>
    void importWedgeColorsFrom(const Element& e)
    {
        for (uint i = 0; i < e.vertexNumber(); ++i) {
            wedgeColor(i) = e.wedgeColor(i);
        }
    }
 
    Vector<vcl::Color, N>& colors() { return Base::container(); }
 
    const Vector<vcl::Color, N>& colors() const { return Base::container(); }
};
 
/* Detector function to check if a class has WedgeColors available */
 
bool isWedgeColorsAvailableOn(const ElementConcept auto& element)
{
    return isComponentAvailableOn<CompId::WEDGE_COLORS>(element);
}
 
} // namespace vcl::comp
 
#endif // VCL_MESH_COMPONENTS_WEDGE_COLORS_H