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 "base/container_component.h"
#include "base/predicates.h"
 
#include <vclib/space/core.h>
#include <vclib/base.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);
 
    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(); }
};
 
/* concepts */
 
template<typename T>
concept HasWedgeColors = ITB::IsDerivedFromSpecializationOfV<T, WedgeColors>;
 
template<typename T>
concept HasOptionalWedgeColors =
    HasWedgeColors<T> &&
    IsOptionalComponent<typename RemoveRef<T>::WedgeColors>;
 
template<typename T>
concept HasRightNumberOfWedgeColors = T::VERTEX_NUMBER == T::WEDGE_COLOR_NUMBER;
 
template<typename T>
concept SanityCheckWedgeColors =
    !HasWedgeColors<T> || HasRightNumberOfWedgeColors<T>;
 
/* importFrom function */
 
template<int N, typename ParentElemType, bool OPT>
template<typename Element>
void WedgeColors<N, ParentElemType, OPT>::importFrom(const Element& e, bool)
{
    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);
            }
        }
    }
}
 
/* Detector function to check if a class has WedgeColors available */
 
bool isWedgeColorsAvailableOn(const auto& element)
{
    return isComponentAvailableOn<CompId::WEDGE_COLORS>(element);
}
 
} // namespace vcl::comp
 
#endif // VCL_MESH_COMPONENTS_WEDGE_COLORS_H