OpenCAX/entry/src/main/cpp/include/opencascade/XCAFDoc_AssemblyIterator.hxx
2026-02-19 19:05:41 +08:00

71 lines
2.9 KiB
C++

// Created on: 2022-05-11
// Copyright (c) 2022 OPEN CASCADE SAS
//
// This file is part of Open CASCADE Technology software library.
//
// This library is free software; you can redistribute it and/or modify it under
// the terms of the GNU Lesser General Public License version 2.1 as published
// by the Free Software Foundation, with special exception defined in the file
// OCCT_LGPL_EXCEPTION.txt. Consult the file LICENSE_LGPL_21.txt included in OCCT
// distribution for complete text of the license and disclaimer of any warranty.
//
// Alternatively, this file may be used under the terms of Open CASCADE
// commercial license or contractual agreement.
#ifndef _XCAFDoc_AssemblyIterator_HeaderFile
#define _XCAFDoc_AssemblyIterator_HeaderFile
#include "NCollection_Sequence.hxx"
#include "XCAFDoc_AssemblyItemId.hxx"
class TDF_Label;
class TDocStd_Document;
class XCAFDoc_ShapeTool;
//! Iterator in depth along the assembly tree.
class XCAFDoc_AssemblyIterator
{
public:
//! Constructs iterator starting from assembly roots.
//! \param[in] theDoc - document to iterate.
//! \param [in, opt] theLevel - max level of hierarchy to reach (INT_MAX is for no limit).
Standard_EXPORT XCAFDoc_AssemblyIterator(const Handle(TDocStd_Document)& theDoc,
const Standard_Integer theLevel = INT_MAX);
//! Constructs iterator starting from the specified position in the assembly tree.
//! \param[in] theDoc - document to iterate.
//! \param[in] theRoot - assembly item to start iterating from.
//! \param [in, opt] theLevel - max level of hierarchy to reach (INT_MAX is for no limit).
Standard_EXPORT XCAFDoc_AssemblyIterator(const Handle(TDocStd_Document)& theDoc,
const XCAFDoc_AssemblyItemId& theRoot,
const Standard_Integer theLevel = INT_MAX);
//! \return true if there is still something to iterate, false -- otherwise.
Standard_EXPORT Standard_Boolean More() const;
//! Moves depth-first iterator to the next position.
Standard_EXPORT void Next();
//! \return current item.
Standard_EXPORT XCAFDoc_AssemblyItemId Current() const;
private:
struct AuxAssemblyItem
{
TDF_Label myLabel;
XCAFDoc_AssemblyItemId myItem;
};
void createItem(const TDF_Label& theLabel,
const TColStd_ListOfAsciiString& theParentPath,
AuxAssemblyItem& theAuxItem) const;
private:
Handle(XCAFDoc_ShapeTool) myShapeTool; //!< Document shape tool.
NCollection_Sequence<AuxAssemblyItem> myFringe; //!< Items pending for iteration.
Standard_Integer myMaxLevel; //!< Limit on max depth of iteration.
Standard_Integer mySeedLevel; //!< Level of hierarchy where we start.
};
#endif // _XCAFDoc_AssemblyIterator_HeaderFile