wedge_tex_coords.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_TEX_COORDS_H
#define VCL_MESH_COMPONENTS_WEDGE_TEX_COORDS_H
 
#include "bases/container_component.h"
 
#include <vclib/concepts/mesh/components/tex_coord.h>
#include <vclib/concepts/mesh/components/wedge_tex_coords.h>
#include <vclib/space/core/tex_coord.h>
#include <vclib/space/core/vector.h>
#include <vclib/types/view.h>
 
namespace vcl::comp {
 
template<
    typename Scalar,
    int N,
    typename ParentElemType = void,
    bool OPT                = false>
class WedgeTexCoords :
        public ContainerComponent<
            WedgeTexCoords<Scalar, N, ParentElemType, OPT>,
            CompId::WEDGE_TEX_COORDS,
            vcl::TexCoord<Scalar>,
            N,
            ushort,
            ParentElemType,
            !std::is_same_v<ParentElemType, void>,
            OPT,
            true>
{
    using Base = ContainerComponent<
        WedgeTexCoords<Scalar, N, ParentElemType, OPT>,
        CompId::WEDGE_TEX_COORDS,
        vcl::TexCoord<Scalar>,
        N,
        ushort,
        ParentElemType,
        !std::is_same_v<ParentElemType, void>,
        OPT,
        true>;
 
public:
    using WedgeTexCoordType = vcl::TexCoord<Scalar>;
 
    /* Iterator Types declaration */
 
    using WedgeTexCoordsIterator = Vector<vcl::TexCoord<Scalar>, N>::Iterator;
    using ConstWedgeTexCoordsIterator =
        Vector<vcl::TexCoord<Scalar>, N>::ConstIterator;
 
    static const int WEDGE_TEX_COORD_NUMBER = N;
 
    /* Constructors */
 
    WedgeTexCoords() = default;
 
    /* Member functions */
 
    vcl::TexCoord<Scalar>& wedgeTexCoord(uint i) { return texCoords().at(i); }
 
    const vcl::TexCoord<Scalar>& wedgeTexCoord(uint i) const
    {
        return texCoords().at(i);
    }
 
    vcl::TexCoord<Scalar>& wedgeTexCoordMod(int i)
    {
        return texCoords().atMod(i);
    }
 
    const vcl::TexCoord<Scalar>& wedgeTexCoordMod(int i) const
    {
        return texCoords().atMod(i);
    }
 
    void setWedgeTexCoord(uint i, const vcl::TexCoord<Scalar>& t)
    {
        texCoords().set(i, t);
    }
 
    template<Range Rng>
    void setWedgeTexCoords(Rng&& r)
        requires InputRange<Rng, vcl::TexCoord<Scalar>>
    {
        texCoords().set(r);
    }
 
    ushort& textureIndex() { return Base::additionalData(); }
 
    ushort textureIndex() const { return Base::additionalData(); }
 
    /* Iterator Member functions */
 
    WedgeTexCoordsIterator wedgeTexCoordBegin() { return texCoords().begin(); }
 
    WedgeTexCoordsIterator wedgeTexCoordEnd() { return texCoords().end(); }
 
    ConstWedgeTexCoordsIterator wedgeTexCoordBegin() const
    {
        return texCoords().begin();
    }
 
    ConstWedgeTexCoordsIterator wedgeTexCoordEnd() const
    {
        return texCoords().end();
    }
 
    View<WedgeTexCoordsIterator> wedgeTexCoords()
    {
        return View(wedgeTexCoordBegin(), wedgeTexCoordEnd());
    }
 
    View<ConstWedgeTexCoordsIterator> wedgeTexCoords() const
    {
        return View(wedgeTexCoordBegin(), wedgeTexCoordEnd());
    }
 
    // dummy member to discriminate between WedgeTexCoords and
    // FaceHalfEdgePointers
    void __wedgeTexCoords() const {}
 
protected:
    using WedgeTexCoordScalarType = Scalar;
 
    // Component interface functions
    template<typename Element>
    void importFrom(const Element& e, bool = true)
    {
        if constexpr (HasWedgeTexCoords<Element>) {
            if (isWedgeTexCoordsAvailableOn(e)) {
                if constexpr (N > 0) {
                    // same static size
                    if constexpr (N == Element::WEDGE_TEX_COORD_NUMBER) {
                        importWedgeTexCoordsFrom(e);
                    }
                    // from dynamic to static, but dynamic size == static size
                    else if constexpr (Element::WEDGE_TEX_COORD_NUMBER < 0) {
                        if (e.vertexNumber() == N) {
                            importWedgeTexCoordsFrom(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());
                    importWedgeTexCoordsFrom(e);
                }
            }
        }
    }
 
    void serialize(std::ostream& os) const
    {
        texCoords().serialize(os);
        vcl::serialize(os, textureIndex());
    }
 
    void deserialize(std::istream& is)
    {
        texCoords().deserialize(is);
        vcl::deserialize(is, textureIndex());
    }
 
    // ContainerComponent interface functions
    void resize(uint n) requires (N < 0) { texCoords().resize(n); }
 
    void pushBack(const vcl::TexCoord<Scalar>& t = vcl::TexCoord<Scalar>())
        requires (N < 0)
    {
        texCoords().pushBack(t);
    }
 
    void insert(
        uint                         i,
        const vcl::TexCoord<Scalar>& t = vcl::TexCoord<Scalar>())
        requires (N < 0)
    {
        texCoords().insert(i, t);
    }
 
    void erase(uint i) requires (N < 0) { texCoords().erase(i); }
 
    void clear() requires (N < 0) { texCoords().clear(); }
 
private:
    template<typename Element>
    void importWedgeTexCoordsFrom(const Element& e)
    {
        for (uint i = 0; i < e.vertexNumber(); ++i) {
            wedgeTexCoord(i) = e.wedgeTexCoord(i).template cast<Scalar>();
        }
        textureIndex() = e.textureIndex();
    }
 
    Vector<vcl::TexCoord<Scalar>, N>& texCoords() { return Base::container(); }
 
    const Vector<vcl::TexCoord<Scalar>, N>& texCoords() const
    {
        return Base::container();
    }
};
 
/* Detector function to check if a class has WedgeTexCoords available */
 
bool isWedgeTexCoordsAvailableOn(const ElementConcept auto& element)
{
    return isComponentAvailableOn<CompId::WEDGE_TEX_COORDS>(element);
}
 
} // namespace vcl::comp
 
#endif // VCL_MESH_COMPONENTS_WEDGE_TEX_COORDS_H