tri_poly_index_bimap.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_SPACE_COMPLEX_TRI_POLY_INDEX_BIMAP_H
#define VCL_SPACE_COMPLEX_TRI_POLY_INDEX_BIMAP_H
 
#include <vclib/base.h>
 
#include <vector>
 
namespace vcl {
 
class TriPolyIndexBiMap
{
    std::vector<uint> mTriToPoly;
    std::vector<uint> mPolyToTri;
    std::vector<uint> mPolyToTriNumber;
 
public:
    TriPolyIndexBiMap() = default;
 
    uint polygon(uint triangleIndex) const
    {
        assert(triangleIndex < mTriToPoly.size());
        return mTriToPoly[triangleIndex];
    }
 
    uint triangleBegin(uint polygonIndex) const
    {
        assert(polygonIndex < mPolyToTri.size());
        return mPolyToTri[polygonIndex];
    }
 
    uint triangleNumber(uint polygonIndex) const
    {
        // be sure that the current polygon index is valid and the polygon
        // has not been deleted in the mesh
        assert(polygonIndex < mPolyToTri.size());
        assert(polygonIndex < mPolyToTriNumber.size());
        assert(mPolyToTri[polygonIndex] != UINT_NULL);
 
        return mPolyToTriNumber[polygonIndex];
    }
 
    void clear()
    {
        mTriToPoly.clear();
        mPolyToTri.clear();
        mPolyToTriNumber.clear();
    }
 
    void reserve(uint nTriangles, uint nPolygons)
    {
        mTriToPoly.reserve(nTriangles);
        mPolyToTri.reserve(nPolygons);
        mPolyToTriNumber.reserve(nPolygons);
    }
 
    void insert(uint triangleIndex, uint polygonIndex)
    {
        // resize the vectors if needed
        if (triangleIndex >= mTriToPoly.size()) {
            mTriToPoly.resize(triangleIndex + 1, UINT_NULL);
        }
        if (polygonIndex >= mPolyToTri.size()) {
            mPolyToTri.resize(polygonIndex + 1, UINT_NULL);
            mPolyToTriNumber.resize(polygonIndex + 1, 0);
        }
 
        // add the index of the polygon associated to the triangle
        mTriToPoly[triangleIndex] = polygonIndex;
 
        // update the number of triangles associated to the polygon
        mPolyToTriNumber[polygonIndex]++;
        // add the index of the triangle associated to the polygon,
        // but only if it is the first triangle index of the polygon!
        if (mPolyToTri[polygonIndex] == UINT_NULL ||
            triangleIndex < mPolyToTri[polygonIndex])
            mPolyToTri[polygonIndex] = triangleIndex;
    }
 
    uint triangleNumber() const { return mTriToPoly.size(); }
 
    uint polygonNumber() const { return mPolyToTri.size(); }
};
 
} // namespace vcl
 
#endif // VCL_SPACE_COMPLEX_TRI_POLY_INDEX_BIMAP_H