topods_utils.h

Go to the documentation of this file.

/****************************************************************************
**  FougTools
**  Copyright Fougue (30 Mar. 2015)
**  contact@fougue.pro
**
** This software is a computer program whose purpose is to provide utility
** tools for the C++ language and the Qt toolkit.
**
** This software is governed by the CeCILL-C license under French law and
** abiding by the rules of distribution of free software.  You can  use,
** modify and/ or redistribute the software under the terms of the CeCILL-C
** license as circulated by CEA, CNRS and INRIA at the following URL
** "http://www.cecill.info".
****************************************************************************/

#pragma once

#include "occtools.h"

#include <BRep_Builder.hxx>
#include <Handle_ShapeExtend_WireData.hxx>
#include <TopAbs_ShapeEnum.hxx>
#include <TopExp_Explorer.hxx>
#include <TopoDS_Compound.hxx>
#include <TopoDS_Face.hxx>
#include <TopoDS_Wire.hxx>
#include <gp_Vec.hxx>

#include <string>

namespace occ {

class OCCTOOLS_EXPORT TopoDsUtils
{
public:
    // Make facilities
    template<typename FWD_ITERATOR>
    static TopoDS_Compound makeCompoundFromShapeRange(
            FWD_ITERATOR iBegin, FWD_ITERATOR iEnd);

    template<typename CONTAINER>
    static TopoDS_Compound makeCompoundFromShapeContainer(CONTAINER cnter);

    template<typename FWD_ITERATOR>
    static TopoDS_Wire makeWireFromEdgeRange(
            FWD_ITERATOR iBegin, FWD_ITERATOR iEnd);

    // TopoDS_Shape <-> std::string
    static std::string shapeToString(const TopoDS_Shape& shape);
    static TopoDS_Shape shapeFromString(const std::string& str);

    // TopExp_Explorer facilities
    template<typename FUNC>
    static void forEach(
            const TopoDS_Shape& shape, TopAbs_ShapeEnum shapeType, FUNC fn);

    template<typename FUNC>
    static void forEach(TopExp_Explorer& explorer, FUNC fn);

    template<typename FUNC>
    static void forEachVertex(const TopoDS_Shape& shape, FUNC fn);

    template<typename FUNC>
    static void forEachEdge(const TopoDS_Shape& shape, FUNC fn);

    template<typename FUNC>
    static void forEachWire(const TopoDS_Shape& shape, FUNC fn);

    template<typename FUNC>
    static void forEachFace(const TopoDS_Shape& shape, FUNC fn);

    // Misc
    static gp_Vec normalToFaceAtUV(
            const TopoDS_Face& face, Standard_Real u, Standard_Real v);

private:
    static Handle_ShapeExtend_WireData createShapeExtendWireData();
    static void addEdge(
            const Handle_ShapeExtend_WireData& wireData,
            const TopoDS_Edge& edge);
    static TopoDS_Wire fixedWire(const Handle_ShapeExtend_WireData& wireData);
};



// --
// -- Implementation
// --

template<typename FWD_ITERATOR>
TopoDS_Compound TopoDsUtils::makeCompoundFromShapeRange(
        FWD_ITERATOR iBegin, FWD_ITERATOR iEnd)
{
    TopoDS_Compound cmpd;
    BRep_Builder builder;
    builder.MakeCompound(cmpd);

    while (iBegin != iEnd) {
        builder.Add(cmpd, *iBegin);
        ++iBegin;
    }

    return cmpd;
}

template<typename CONTAINER>
TopoDS_Compound TopoDsUtils::makeCompoundFromShapeContainer(CONTAINER cnter)
{
    return TopoDsUtils::makeCompoundFromShapeRange(cnter.begin(), cnter.end());
}

template<typename FWD_ITERATOR>
TopoDS_Wire TopoDsUtils::makeWireFromEdgeRange(
        FWD_ITERATOR iBegin, FWD_ITERATOR iEnd)
{
    Handle_ShapeExtend_WireData wireData =
            TopoDsUtils::createShapeExtendWireData();
    while (iBegin != iEnd) {
        TopoDsUtils::addEdge(wireData, *iBegin);
        ++iBegin;
    }
    return TopoDsUtils::fixedWire(wireData);
}

template<typename FUNC>
void TopoDsUtils::forEach(
        const TopoDS_Shape &shape, TopAbs_ShapeEnum shapeType, FUNC fn)
{
    TopoDsUtils::forEach(TopExp_Explorer(shape, shapeType), std::move(fn));
}

template<typename FUNC>
void TopoDsUtils::forEach(TopExp_Explorer &explorer, FUNC fn)
{
    while (explorer.More()) {
        fn(explorer.Current());
        explorer.Next();
    }
}

template<typename FUNC>
void TopoDsUtils::forEachVertex(const TopoDS_Shape& shape, FUNC fn)
{
    TopoDsUtils::forEach(shape, TopAbs_VERTEX, std::move(fn));
}

template<typename FUNC>
void TopoDsUtils::forEachEdge(const TopoDS_Shape& shape, FUNC fn)
{
    TopoDsUtils::forEach(shape, TopAbs_EDGE, std::move(fn));
}

template<typename FUNC>
void TopoDsUtils::forEachWire(const TopoDS_Shape& shape, FUNC fn)
{
    TopoDsUtils::forEach(shape, TopAbs_WIRE, std::move(fn));
}

template<typename FUNC>
void TopoDsUtils::forEachFace(const TopoDS_Shape& shape, FUNC fn)
{
    TopoDsUtils::forEach(shape, TopAbs_FACE, std::move(fn));
}

} // namespace occ