mirror of
https://git.dev.opencascade.org/repos/occt.git
synced 2025-04-04 18:06:22 +03:00
973c2be1e1
The copying permission statements at the beginning of source files updated to refer to LGPL. Copyright dates extended till 2014 in advance.
1758 lines
55 KiB
C++
1758 lines
55 KiB
C++
// Created on: 2000-08-03
|
|
// Created by: data exchange team
|
|
// Copyright (c) 2000-2014 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 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.
|
|
|
|
#include <XCAFDoc_ShapeTool.ixx>
|
|
#include <XCAFDoc.hxx>
|
|
|
|
#include <TDF_Tool.hxx>
|
|
#include <TDF_Label.hxx>
|
|
#include <TDF_LabelMap.hxx>
|
|
#include <TDF_LabelSequence.hxx>
|
|
#include <TDF_ChildIterator.hxx>
|
|
#include <TDF_ChildIDIterator.hxx>
|
|
#include <TDF_MapIteratorOfLabelMap.hxx>
|
|
|
|
#include <TDataStd_Name.hxx>
|
|
#include <TDataStd_TreeNode.hxx>
|
|
#include <TDataStd_ChildNodeIterator.hxx>
|
|
#include <TDataStd_UAttribute.hxx>
|
|
|
|
#include <TDocStd_Document.hxx>
|
|
|
|
#include <TNaming_Builder.hxx>
|
|
#include <TNaming_NamedShape.hxx>
|
|
#include <TNaming_Tool.hxx>
|
|
|
|
#include <TopLoc_Location.hxx>
|
|
#include <TopoDS_Iterator.hxx>
|
|
#include <TopoDS_Shape.hxx>
|
|
#include <TopoDS_Compound.hxx>
|
|
#include <BRep_Builder.hxx>
|
|
#include <XCAFDoc_Location.hxx>
|
|
|
|
#include <TCollection_AsciiString.hxx>
|
|
#include <TCollection_HAsciiString.hxx>
|
|
#include <TCollection_ExtendedString.hxx>
|
|
#include <TColStd_SequenceOfHAsciiString.hxx>
|
|
#include <XCAFDoc_GraphNode.hxx>
|
|
#include <TopLoc_IndexedMapOfLocation.hxx>
|
|
#include <gp_Trsf.hxx>
|
|
#include <gp_Pnt.hxx>
|
|
#include <XCAFDoc_ShapeMapTool.hxx>
|
|
|
|
static Standard_Boolean theAutoNaming = Standard_True;
|
|
|
|
// attribute methods //////////////////////////////////////////////////
|
|
|
|
//=======================================================================
|
|
//function : GetID
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
const Standard_GUID& XCAFDoc_ShapeTool::GetID()
|
|
{
|
|
static Standard_GUID ShapeToolID ("efd212ee-6dfd-11d4-b9c8-0060b0ee281b");
|
|
return ShapeToolID;
|
|
}
|
|
|
|
|
|
//=======================================================================
|
|
//function : Set
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
Handle(XCAFDoc_ShapeTool) XCAFDoc_ShapeTool::Set(const TDF_Label& L)
|
|
{
|
|
Handle(XCAFDoc_ShapeTool) A;
|
|
if (!L.FindAttribute (XCAFDoc_ShapeTool::GetID(), A)) {
|
|
A = new XCAFDoc_ShapeTool ();
|
|
L.AddAttribute(A);
|
|
}
|
|
A->Init();
|
|
return A;
|
|
}
|
|
|
|
|
|
//=======================================================================
|
|
//function : Constructor
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
XCAFDoc_ShapeTool::XCAFDoc_ShapeTool()
|
|
{
|
|
hasSimpleShapes = Standard_False;
|
|
}
|
|
|
|
|
|
//=======================================================================
|
|
//function : ID
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
const Standard_GUID& XCAFDoc_ShapeTool::ID() const
|
|
{
|
|
return GetID();
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : Restore
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
void XCAFDoc_ShapeTool::Restore(const Handle(TDF_Attribute)& /*with*/)
|
|
{
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : NewEmpty
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
Handle(TDF_Attribute) XCAFDoc_ShapeTool::NewEmpty() const
|
|
{
|
|
return new XCAFDoc_ShapeTool;
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : Paste
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
void XCAFDoc_ShapeTool::Paste (const Handle(TDF_Attribute)& /*into*/,
|
|
const Handle(TDF_RelocationTable)& /*RT*/) const
|
|
{
|
|
}
|
|
|
|
// Auxiliary methods //////////////////////////////////////////////////
|
|
|
|
//=======================================================================
|
|
//function : SetLabelNameByLink
|
|
//purpose :
|
|
//=======================================================================
|
|
static void SetLabelNameByLink(const TDF_Label L)
|
|
{
|
|
Handle(TDataStd_TreeNode) Node;
|
|
if (! L.FindAttribute(XCAFDoc::ShapeRefGUID(), Node) ||
|
|
! Node->HasFather()) {
|
|
cout<<"Error: XCAFDoc_ShapeTool, SetLabelNameByLink(): NO NODE"<<endl;
|
|
return;
|
|
}
|
|
TCollection_AsciiString Entry;
|
|
TDF_Tool::Entry ( Node->Father()->Label(), Entry );
|
|
Entry.Insert(1, "=>[");
|
|
Entry += "]";
|
|
|
|
TDataStd_Name::Set(L, TCollection_ExtendedString( Entry ));
|
|
}
|
|
|
|
|
|
//=======================================================================
|
|
//function : SetLabelNameByShape
|
|
//purpose :
|
|
//=======================================================================
|
|
static void SetLabelNameByShape(const TDF_Label L)
|
|
{
|
|
TopoDS_Shape S;
|
|
if (XCAFDoc_ShapeTool::GetShape(L, S) &&
|
|
! L.IsAttribute(TDataStd_Name::GetID()) ) {
|
|
Standard_SStream Stream;
|
|
// TopAbs_ShapeEnum Type = S.ShapeType();
|
|
// if (Type == TopAbs_COMPOUND) Stream<<"ASSEMBLY";
|
|
// else
|
|
TopAbs::Print(S.ShapeType(), Stream);
|
|
|
|
#ifdef USE_STL_STREAM
|
|
TCollection_AsciiString aName (Stream.str().c_str());
|
|
#else
|
|
Stream << ends;
|
|
TCollection_AsciiString aName (Stream.str());
|
|
#endif
|
|
TDataStd_Name::Set(L, TCollection_ExtendedString(aName));
|
|
}
|
|
}
|
|
|
|
|
|
//=======================================================================
|
|
//function : SearchUsingMap
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
Standard_Boolean XCAFDoc_ShapeTool::SearchUsingMap(const TopoDS_Shape &S, TDF_Label &L,
|
|
const Standard_Boolean findWithoutLoc,
|
|
const Standard_Boolean findSubShape) const
|
|
{
|
|
|
|
if(myShapeLabels.IsBound(S)) {
|
|
L = myShapeLabels.Find(S);
|
|
return Standard_True;
|
|
}
|
|
TopoDS_Shape S0 = S;
|
|
TopLoc_Location loc;
|
|
S0.Location(loc);
|
|
if(myShapeLabels.IsBound(S0)) {
|
|
TDF_Label L1 = myShapeLabels.Find(S0);
|
|
TDF_LabelSequence Labels;
|
|
if(GetUsers(L1, Labels, Standard_True)) {
|
|
for(Standard_Integer i=1; i<=Labels.Length(); i++) {
|
|
TopoDS_Shape c = GetShape(Labels.Value(i));
|
|
if(c.IsSame(S)) {
|
|
L = Labels.Value(i);
|
|
return Standard_True;
|
|
}
|
|
}
|
|
}
|
|
if(findWithoutLoc) {
|
|
L = L1;
|
|
return Standard_True;
|
|
}
|
|
}
|
|
|
|
if(hasSimpleShapes) {
|
|
if(mySimpleShapes.IsBound(S)) {
|
|
L = mySimpleShapes.Find(S);
|
|
return Standard_True;
|
|
}
|
|
if(mySimpleShapes.IsBound(S0)) {
|
|
L = mySimpleShapes.Find(S0);
|
|
return Standard_True;
|
|
}
|
|
}
|
|
// search subshapes
|
|
if(!findSubShape) return Standard_False;
|
|
TDF_Label mainL = FindMainShapeUsingMap(S);
|
|
if(mainL.IsNull()) return Standard_False;
|
|
L = AddSubShape(mainL,S);
|
|
return !L.IsNull();//Standard_True;
|
|
}
|
|
|
|
|
|
//=======================================================================
|
|
//function : Search
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
Standard_Boolean XCAFDoc_ShapeTool::Search (const TopoDS_Shape &S,
|
|
TDF_Label &L,
|
|
const Standard_Boolean findInstance,
|
|
const Standard_Boolean findComponent,
|
|
const Standard_Boolean findSubShape) const
|
|
{
|
|
// search among shapes
|
|
Standard_Boolean isLocated = ! S.Location().IsIdentity();
|
|
|
|
if ( isLocated ) {
|
|
// try to find top-level instance
|
|
if ( findInstance && FindShape ( S, L, Standard_True ) )
|
|
return Standard_True;
|
|
// try to find component of assembly
|
|
if ( findComponent ) {
|
|
TDF_LabelSequence labels;
|
|
GetShapes ( labels );
|
|
for ( Standard_Integer i=1; i <= labels.Length(); i++ ) {
|
|
if ( ! IsAssembly ( labels.Value(i) ) ) continue;
|
|
TDF_LabelSequence comp;
|
|
GetComponents ( labels.Value(i), comp );
|
|
for ( Standard_Integer j=1; j <= comp.Length(); j++ ) {
|
|
TopoDS_Shape c = GetShape ( comp.Value(j) );
|
|
if ( c.IsSame ( S ) ) {
|
|
L = comp.Value(j);
|
|
return Standard_True;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
// try to find top-level simple shape
|
|
if ( FindShape ( S, L, Standard_False ) ) return Standard_True;
|
|
|
|
// search subshapes
|
|
if ( ! findSubShape ) return Standard_False;
|
|
TDF_Label mainL = FindMainShape ( S );
|
|
if ( mainL.IsNull() ) return Standard_False;
|
|
L = AddSubShape ( mainL, S );
|
|
return !L.IsNull();//Standard_True;
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : FindShape
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
Standard_Boolean XCAFDoc_ShapeTool::FindShape (const TopoDS_Shape& S,
|
|
TDF_Label& L,
|
|
const Standard_Boolean findInstance) const
|
|
{
|
|
// search for null-located shape
|
|
TopoDS_Shape S0 = S;
|
|
if ( ! findInstance ) {
|
|
TopLoc_Location loc;
|
|
S0.Location ( loc );
|
|
}
|
|
|
|
// this code is used instead of the following for performance reasons
|
|
if (TNaming_Tool::HasLabel(Label(), S0)) {
|
|
int TransDef = 0;
|
|
L = TNaming_Tool::Label(Label(), S0, TransDef);
|
|
return Standard_True;
|
|
}
|
|
/*
|
|
TDF_ChildIDIterator it(myLabel,TNaming_NamedShape::GetID());
|
|
for (; it.More(); it.Next()) {
|
|
TDF_Label aLabel = it.Value()->Label();
|
|
Handle(TNaming_NamedShape) NS;
|
|
if ( aLabel.FindAttribute(TNaming_NamedShape::GetID(), NS) &&
|
|
S0.IsSame ( TNaming_Tool::GetShape(NS) ) ) {
|
|
L = aLabel;
|
|
return Standard_True;
|
|
}
|
|
}
|
|
*/
|
|
return Standard_False;
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : FindShape
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
TDF_Label XCAFDoc_ShapeTool::FindShape (const TopoDS_Shape& S,
|
|
const Standard_Boolean findInstance) const
|
|
{
|
|
TDF_Label L;
|
|
FindShape ( S, L, findInstance );
|
|
return L;
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : GetShape
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
Standard_Boolean XCAFDoc_ShapeTool::GetShape (const TDF_Label& L, TopoDS_Shape& S)
|
|
{
|
|
Handle(XCAFDoc_Location) LocationAttribute;
|
|
|
|
if(IsExternRef(L)) {
|
|
TopoDS_Compound EmptyComp;
|
|
BRep_Builder B;
|
|
B.MakeCompound(EmptyComp);
|
|
S = EmptyComp;
|
|
}
|
|
|
|
// for instance, get referred shape
|
|
Handle(TDataStd_TreeNode) Node;
|
|
if ( L.FindAttribute(XCAFDoc::ShapeRefGUID(), Node) && Node->HasFather() &&
|
|
L.FindAttribute(XCAFDoc_Location::GetID(), LocationAttribute)) {
|
|
if ( ! GetShape(Node->Father()->Label(), S) ) return Standard_False;
|
|
S.Move ( LocationAttribute->Get() );
|
|
return Standard_True;
|
|
}
|
|
|
|
// else just return shape on this label
|
|
Handle(TNaming_NamedShape) NS;
|
|
if ( ! L.FindAttribute(TNaming_NamedShape::GetID(), NS) ) return Standard_False;
|
|
S = TNaming_Tool::GetShape(NS);
|
|
return Standard_True;
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : GetShape
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
TopoDS_Shape XCAFDoc_ShapeTool::GetShape(const TDF_Label& L)
|
|
{
|
|
TopoDS_Shape aShape;
|
|
GetShape(L,aShape);
|
|
return aShape;
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : NewShape
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
TDF_Label XCAFDoc_ShapeTool::NewShape() const
|
|
{
|
|
TopoDS_Compound aShape;
|
|
BRep_Builder tdsB;
|
|
tdsB.MakeCompound ( aShape );
|
|
|
|
TDF_TagSource aTag;
|
|
TDF_Label aLabel = aTag.NewChild(Label());
|
|
|
|
TNaming_Builder tnBuild(aLabel);
|
|
tnBuild.Generated(aShape);
|
|
|
|
return aLabel;
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : SetShape
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
void XCAFDoc_ShapeTool::SetShape (const TDF_Label& L, const TopoDS_Shape& S)
|
|
{
|
|
TNaming_Builder tnBuild(L);
|
|
tnBuild.Generated(S);
|
|
Handle(XCAFDoc_ShapeMapTool) A = XCAFDoc_ShapeMapTool::Set(L);
|
|
// if ( ! L.FindAttribute(XCAFDoc_ShapeMapTool::GetID(), A) ) {
|
|
// A = XCAFDoc_ShapeMapTool::Set(L);
|
|
// L.AddAttribute(A);
|
|
// }
|
|
A->SetShape(S);
|
|
|
|
if(!myShapeLabels.IsBound(S)) {
|
|
myShapeLabels.Bind(S,L);
|
|
}
|
|
|
|
//:abv 31.10.01: update assemblies that refer a shape
|
|
TDF_LabelSequence Labels;
|
|
if ( GetUsers ( L, Labels, Standard_True ) ) {
|
|
for ( Standard_Integer i=Labels.Length(); i >=1; i-- )
|
|
UpdateAssembly ( Labels(i) );
|
|
}
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : MakeReference
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
void XCAFDoc_ShapeTool::MakeReference (const TDF_Label &L,
|
|
const TDF_Label &refL,
|
|
const TopLoc_Location &loc)
|
|
{
|
|
// store location
|
|
XCAFDoc_Location::Set(L, loc);
|
|
|
|
// set reference
|
|
Handle(TDataStd_TreeNode) refNode, mainNode;
|
|
mainNode = TDataStd_TreeNode::Set ( refL, XCAFDoc::ShapeRefGUID() );
|
|
refNode = TDataStd_TreeNode::Set ( L, XCAFDoc::ShapeRefGUID() );
|
|
refNode->Remove(); // abv: fix against bug in TreeNode::Append()
|
|
mainNode->Append(refNode);
|
|
|
|
if (theAutoNaming)
|
|
SetLabelNameByLink(L);
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : addShape
|
|
//purpose : private
|
|
//=======================================================================
|
|
|
|
TDF_Label XCAFDoc_ShapeTool::addShape (const TopoDS_Shape& S, const Standard_Boolean makeAssembly)
|
|
{
|
|
TDF_Label ShapeLabel;
|
|
TDF_TagSource aTag;
|
|
|
|
// search if the shape already exists (with the same location)
|
|
if ( S.IsNull() || FindShape ( S, ShapeLabel, Standard_True ) ) return ShapeLabel;
|
|
|
|
// else add a new label
|
|
ShapeLabel = aTag.NewChild(Label());
|
|
|
|
// if shape has location, make a reference to the same shape without location
|
|
if ( ! S.Location().IsIdentity() /*&& FindShape ( S, L )*/ ) {
|
|
TopoDS_Shape S0 = S;
|
|
TopLoc_Location loc;
|
|
S0.Location ( loc );
|
|
TDF_Label L = addShape ( S0, makeAssembly );
|
|
MakeReference ( ShapeLabel, L, S.Location() );
|
|
return ShapeLabel;
|
|
}
|
|
|
|
// else add a shape to a label
|
|
TNaming_Builder tnBuild(ShapeLabel);
|
|
tnBuild.Generated(S);
|
|
|
|
Handle(XCAFDoc_ShapeMapTool) A = XCAFDoc_ShapeMapTool::Set(ShapeLabel);
|
|
// if ( ! ShapeLabel.FindAttribute(XCAFDoc_ShapeMapTool::GetID(), A) ) {
|
|
// A = XCAFDoc_ShapeMapTool::Set(ShapeLabel);
|
|
// ShapeLabel.AddAttribute(A);
|
|
// }
|
|
A->SetShape(S);
|
|
|
|
if (theAutoNaming)
|
|
SetLabelNameByShape(ShapeLabel);
|
|
|
|
// if shape is Compound and flag is set, create assembly
|
|
if ( makeAssembly && S.ShapeType() == TopAbs_COMPOUND ) {
|
|
// mark assembly by assigning UAttribute
|
|
Handle(TDataStd_UAttribute) Uattr;
|
|
Uattr = TDataStd_UAttribute::Set ( ShapeLabel, XCAFDoc::AssemblyGUID() );
|
|
if (theAutoNaming)
|
|
TDataStd_Name::Set(ShapeLabel, TCollection_ExtendedString("ASSEMBLY"));
|
|
|
|
// iterate on components
|
|
TopoDS_Iterator Iterator(S);
|
|
for (; Iterator.More(); Iterator.Next()) {
|
|
// get label for component`s shape
|
|
TopoDS_Shape Scomp = Iterator.Value(), S0 = Scomp;
|
|
TopLoc_Location loc;
|
|
S0.Location ( loc );
|
|
TDF_Label compL = addShape ( S0, makeAssembly );
|
|
|
|
// add a component as reference
|
|
TDF_Label RefLabel = aTag.NewChild(ShapeLabel);
|
|
MakeReference ( RefLabel, compL, Scomp.Location() );
|
|
}
|
|
}
|
|
|
|
if(!IsAssembly(ShapeLabel)) {
|
|
//const TopTools_IndexedMapOfShape tmpMap = A->GetMap();
|
|
//for(Standard_Integer i=1; i<=tmpMap.Extent(); i++)
|
|
//mySubShapes.Bind(tmpMap.FindKey(i),ShapeLabel);
|
|
for(Standard_Integer i=1; i<=A->GetMap().Extent(); i++)
|
|
{
|
|
TopoDS_Shape aSh = A->GetMap().FindKey(i);
|
|
mySubShapes.Bind(aSh,ShapeLabel);
|
|
}
|
|
//mySubShapes.Bind(ShapeLabel,A->GetMap());
|
|
}
|
|
|
|
return ShapeLabel;
|
|
}
|
|
|
|
|
|
//=======================================================================
|
|
//function : prepareAssembly
|
|
//purpose : auxilary
|
|
//=======================================================================
|
|
static Standard_Boolean prepareAssembly (const TopoDS_Shape& theShape,
|
|
TopoDS_Shape& theOUTShape)
|
|
{
|
|
// iterate on components
|
|
theOUTShape = theShape;
|
|
if (theShape.ShapeType() == TopAbs_COMPOUND) {
|
|
BRep_Builder B;
|
|
// check if shape if frosen
|
|
if (!theOUTShape.Free())
|
|
theOUTShape.Free(Standard_True);
|
|
|
|
TopTools_SequenceOfShape aSubShapeSeq;
|
|
TopoDS_Iterator Iterator(theShape);
|
|
for (; Iterator.More(); Iterator.Next())
|
|
aSubShapeSeq.Append(Iterator.Value());
|
|
for (Standard_Integer i = 1; i <= aSubShapeSeq.Length(); i++) {
|
|
TopoDS_Shape Scomp = aSubShapeSeq.Value(i);
|
|
TopoDS_Shape aNewScomp;
|
|
B.Remove(theOUTShape, Scomp);
|
|
prepareAssembly( Scomp, aNewScomp );
|
|
TopLoc_Location aLoc;
|
|
aLoc = aNewScomp.Location();
|
|
if ( aLoc.IsIdentity() ) {
|
|
// create an "empty" location
|
|
gp_Trsf aTrsf;
|
|
aTrsf.SetScale(gp_Pnt(0,0,0), 1);
|
|
aLoc = TopLoc_Location( aTrsf );
|
|
aNewScomp.Location( aLoc );
|
|
}
|
|
B.Add(theOUTShape, aNewScomp);
|
|
}
|
|
}
|
|
return Standard_True;
|
|
}
|
|
|
|
|
|
//=======================================================================
|
|
//function : AddShape
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
TDF_Label XCAFDoc_ShapeTool::AddShape (const TopoDS_Shape& theShape,
|
|
const Standard_Boolean makeAssembly,
|
|
const Standard_Boolean makePrepare)
|
|
{
|
|
// PTV 17.02.2003 to avoid components without location.
|
|
TopoDS_Shape S = theShape;
|
|
if ( makePrepare && makeAssembly && S.ShapeType() == TopAbs_COMPOUND )
|
|
prepareAssembly( theShape, S ); // OCC1669
|
|
|
|
TDF_Label L = addShape(S,makeAssembly);
|
|
|
|
if(!myShapeLabels.IsBound(S)) {
|
|
myShapeLabels.Bind(S,L);
|
|
}
|
|
|
|
return L;
|
|
|
|
//return addShape( S, makeAssembly );
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : RemoveShape
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
Standard_Boolean XCAFDoc_ShapeTool::RemoveShape (const TDF_Label& L,
|
|
const Standard_Boolean removeCompletely) const
|
|
{
|
|
if ( ! IsTopLevel ( L ) || ! IsFree ( L ) ) return Standard_False;
|
|
|
|
Handle(TDataStd_TreeNode) aNode;
|
|
TDF_Label aLabel;
|
|
if (removeCompletely &&
|
|
L.FindAttribute (XCAFDoc::ShapeRefGUID(), aNode) &&
|
|
aNode->HasFather() &&
|
|
L.IsAttribute (XCAFDoc_Location::GetID()))
|
|
{
|
|
aLabel = aNode->Father()->Label();
|
|
}
|
|
|
|
L.ForgetAllAttributes (Standard_True);
|
|
|
|
if (removeCompletely && !aLabel.IsNull())
|
|
{
|
|
return RemoveShape(aLabel);
|
|
}
|
|
return Standard_True;
|
|
}
|
|
|
|
|
|
//=======================================================================
|
|
//function : Init
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
void XCAFDoc_ShapeTool::Init()
|
|
{
|
|
hasSimpleShapes = Standard_False;
|
|
}
|
|
|
|
|
|
//=======================================================================
|
|
//function : SetAutoNaming
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
void XCAFDoc_ShapeTool::SetAutoNaming (const Standard_Boolean V)
|
|
{
|
|
theAutoNaming = V;
|
|
}
|
|
|
|
|
|
//=======================================================================
|
|
//function : AutoNaming
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
Standard_Boolean XCAFDoc_ShapeTool::AutoNaming()
|
|
{
|
|
return theAutoNaming;
|
|
}
|
|
|
|
|
|
//=======================================================================
|
|
//function : ComputeShapes
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
void XCAFDoc_ShapeTool::ComputeShapes(const TDF_Label& L)
|
|
{
|
|
TDF_ChildIterator it(L);
|
|
for(; it.More(); it.Next()) {
|
|
TDF_Label L1 = it.Value();
|
|
TopoDS_Shape S;
|
|
if(GetShape(L1,S)) {
|
|
if(!myShapeLabels.IsBound(S)) {
|
|
mySimpleShapes.Bind(S,L1);
|
|
}
|
|
}
|
|
ComputeShapes(L1);
|
|
}
|
|
}
|
|
|
|
|
|
//=======================================================================
|
|
//function : ComputeSimpleShapes
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
void XCAFDoc_ShapeTool::ComputeSimpleShapes()
|
|
{
|
|
ComputeShapes(Label());
|
|
hasSimpleShapes = Standard_True;
|
|
}
|
|
|
|
|
|
//=======================================================================
|
|
//function : GetShapes
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
void XCAFDoc_ShapeTool::GetShapes(TDF_LabelSequence& Labels) const
|
|
{
|
|
Labels.Clear();
|
|
|
|
TDF_ChildIterator it(Label());
|
|
for (; it.More(); it.Next()) {
|
|
TDF_Label L = it.Value();
|
|
TopoDS_Shape S;
|
|
if ( GetShape ( L, S ) ) Labels.Append ( L );
|
|
}
|
|
}
|
|
|
|
|
|
//=======================================================================
|
|
//function : GetFreeShapes
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
void XCAFDoc_ShapeTool::GetFreeShapes (TDF_LabelSequence& FreeLabels) const
|
|
{
|
|
FreeLabels.Clear();
|
|
|
|
TDF_ChildIterator it(Label());
|
|
for (; it.More(); it.Next()) {
|
|
TDF_Label L = it.Value();
|
|
TopoDS_Shape S;
|
|
if ( GetShape ( L, S ) && IsFree ( L ) ) FreeLabels.Append ( L );
|
|
}
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : IsTopLevel
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
Standard_Boolean XCAFDoc_ShapeTool::IsTopLevel (const TDF_Label& L) const
|
|
{
|
|
return L.Father() == Label();
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : IsShape
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
Standard_Boolean XCAFDoc_ShapeTool::IsShape (const TDF_Label& L)
|
|
{
|
|
return IsSimpleShape ( L ) || IsAssembly ( L ) || IsReference ( L );
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : IsSimpleShape
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
Standard_Boolean XCAFDoc_ShapeTool::IsSimpleShape (const TDF_Label& L)
|
|
{
|
|
Handle(TNaming_NamedShape) NS;
|
|
return L.FindAttribute ( TNaming_NamedShape::GetID(), NS ) &&
|
|
! IsAssembly ( L ) && ! IsReference ( L );
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : IsReference
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
Standard_Boolean XCAFDoc_ShapeTool::IsReference (const TDF_Label& L)
|
|
{
|
|
Handle(TDataStd_TreeNode) Node;
|
|
return L.FindAttribute(XCAFDoc::ShapeRefGUID(), Node) && Node->HasFather();
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : IsAssembly
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
Standard_Boolean XCAFDoc_ShapeTool::IsAssembly (const TDF_Label& L)
|
|
{
|
|
Handle(TDataStd_UAttribute) Uattr;
|
|
return L.FindAttribute(XCAFDoc::AssemblyGUID(), Uattr);
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : IsComponent
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
Standard_Boolean XCAFDoc_ShapeTool::IsComponent (const TDF_Label& L)
|
|
{
|
|
return IsReference ( L ) && IsAssembly ( L.Father() );
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : IsCompound
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
Standard_Boolean XCAFDoc_ShapeTool::IsCompound (const TDF_Label& L)
|
|
{
|
|
Handle(TDataStd_Name) Name;
|
|
if (L.FindAttribute(TDataStd_Name::GetID(),Name)) {
|
|
TCollection_ExtendedString estr1 = Name->Get();
|
|
TCollection_ExtendedString estr2("COMPOUND");
|
|
if(estr1==estr2) {
|
|
return Standard_True;
|
|
}
|
|
}
|
|
return Standard_False;
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : IsSubShape
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
Standard_Boolean XCAFDoc_ShapeTool::IsSubShape (const TDF_Label& L)
|
|
{
|
|
return IsSimpleShape ( L ) && IsShape ( L.Father() );
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : IsFree
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
Standard_Boolean XCAFDoc_ShapeTool::IsFree (const TDF_Label& L)
|
|
{
|
|
Handle(TDataStd_TreeNode) Node;
|
|
if ( ! L.FindAttribute(XCAFDoc::ShapeRefGUID(), Node) ||
|
|
! Node->HasFirst() ) return Standard_True;
|
|
|
|
return Standard_False;
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : GetUsers
|
|
//purpose : Returns number of users (0 if shape is free)
|
|
//=======================================================================
|
|
|
|
Standard_Integer XCAFDoc_ShapeTool::GetUsers (const TDF_Label& L,
|
|
TDF_LabelSequence& Labels,
|
|
const Standard_Boolean getsubchilds)
|
|
{
|
|
Standard_Integer NbUsers=0;
|
|
Handle(TDataStd_TreeNode) Node ;
|
|
|
|
if ( ! L.FindAttribute(XCAFDoc::ShapeRefGUID(), Node) ) return NbUsers;
|
|
|
|
Node = Node->First();
|
|
while ( ! Node.IsNull() ) {
|
|
|
|
if ( getsubchilds ) {
|
|
TDF_Label underL = Node->Label().Father();
|
|
NbUsers += GetUsers ( underL, Labels, getsubchilds );
|
|
}
|
|
|
|
Labels.Append(Node->Label());
|
|
Node = Node->Next();
|
|
NbUsers++;
|
|
}
|
|
return NbUsers;
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : NbComponents
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
Standard_Integer XCAFDoc_ShapeTool::NbComponents (const TDF_Label& L,
|
|
const Standard_Boolean getsubchilds)
|
|
{
|
|
TDF_LabelSequence subLabels;
|
|
GetComponents (L, subLabels, getsubchilds);
|
|
return subLabels.Length();
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : GetComponents
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
Standard_Boolean XCAFDoc_ShapeTool::GetComponents (const TDF_Label& L, TDF_LabelSequence& Labels,
|
|
const Standard_Boolean getsubchilds)
|
|
{
|
|
if ( ! IsAssembly(L) ) return Standard_False;
|
|
|
|
TDF_ChildIterator It(L);
|
|
for (; It.More(); It.Next() ) {
|
|
TDF_Label comp = It.Value();
|
|
if ( IsComponent ( comp ) ) {
|
|
if ( getsubchilds ) {
|
|
TDF_Label underL;
|
|
if ( GetReferredShape ( comp, underL ) )
|
|
GetComponents ( underL, Labels, getsubchilds);
|
|
}
|
|
Labels.Append ( comp );
|
|
}
|
|
}
|
|
return Standard_True;
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : GetLocation
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
TopLoc_Location XCAFDoc_ShapeTool::GetLocation (const TDF_Label& L)
|
|
{
|
|
Handle(XCAFDoc_Location) LocationAttribute;
|
|
if (L.FindAttribute(XCAFDoc_Location::GetID(), LocationAttribute))
|
|
return LocationAttribute->Get();
|
|
|
|
Handle(TNaming_NamedShape) NS;
|
|
TopoDS_Shape S;
|
|
if ( L.FindAttribute ( TNaming_NamedShape::GetID(), NS ) ) {
|
|
S = TNaming_Tool::GetShape(NS);
|
|
}
|
|
return S.Location();
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : GetReferredShape
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
Standard_Boolean XCAFDoc_ShapeTool::GetReferredShape (const TDF_Label& L,
|
|
TDF_Label& Label)
|
|
{
|
|
if ( ! IsReference(L) ) return Standard_False;
|
|
|
|
Handle (TDataStd_TreeNode) Node;
|
|
L.FindAttribute(XCAFDoc::ShapeRefGUID(), Node);
|
|
Label = Node->Father()->Label();
|
|
return Standard_True;
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : AddComponent
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
TDF_Label XCAFDoc_ShapeTool::AddComponent (const TDF_Label& assembly,
|
|
const TDF_Label& compL,
|
|
const TopLoc_Location &Loc) const
|
|
{
|
|
TDF_Label L;
|
|
|
|
// check that shape is assembly
|
|
if ( ! IsAssembly(assembly) ) {
|
|
// if it is simple shape, make it assembly
|
|
if ( IsSimpleShape(assembly) )
|
|
TDataStd_UAttribute::Set ( assembly, XCAFDoc::AssemblyGUID() );
|
|
else return L;
|
|
}
|
|
|
|
// add a component as reference
|
|
TDF_TagSource aTag;
|
|
L = aTag.NewChild(assembly);
|
|
MakeReference ( L, compL, Loc );
|
|
|
|
// update assembly`s TopoDS_Shape
|
|
UpdateAssembly ( assembly );
|
|
return L;
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : AddComponent
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
TDF_Label XCAFDoc_ShapeTool::AddComponent (const TDF_Label& assembly,
|
|
const TopoDS_Shape& comp,
|
|
const Standard_Boolean expand)
|
|
{
|
|
// get label for component`s shape
|
|
TopoDS_Shape S0 = comp;
|
|
TopLoc_Location loc;
|
|
S0.Location ( loc );
|
|
TDF_Label compL;
|
|
compL = AddShape ( S0, expand );
|
|
|
|
// add component by its label
|
|
return AddComponent ( assembly, compL, comp.Location() );
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : RemoveComponent
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
void XCAFDoc_ShapeTool::RemoveComponent (const TDF_Label& comp) const
|
|
{
|
|
if ( IsComponent(comp) ) {
|
|
comp.ForgetAllAttributes();
|
|
UpdateAssembly(comp.Father());
|
|
}
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : UpdateAssembly
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
void XCAFDoc_ShapeTool::UpdateAssembly (const TDF_Label& L) const
|
|
{
|
|
if ( ! IsAssembly(L) ) return;
|
|
|
|
TopoDS_Compound newassembly;
|
|
BRep_Builder b;
|
|
b.MakeCompound(newassembly);
|
|
|
|
TDF_ChildIterator chldLabIt(L);
|
|
for (; chldLabIt.More(); chldLabIt.Next() ) {
|
|
TDF_Label subLabel = chldLabIt.Value();
|
|
if ( IsComponent ( subLabel ) ) {
|
|
b.Add(newassembly, GetShape(subLabel));
|
|
}
|
|
}
|
|
TNaming_Builder tnBuild(L);
|
|
tnBuild.Generated(newassembly);
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : IsSubShape
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
//static Standard_Boolean CheckSubShape (const TopoDS_Shape &S, const TopoDS_Shape &sub)
|
|
//{
|
|
// if ( S.IsSame ( sub ) ) return Standard_True;
|
|
//
|
|
// if ( S.ShapeType() >= sub.ShapeType() ) return Standard_False;
|
|
//
|
|
// for ( TopoDS_Iterator it(S); it.More(); it.Next() ) {
|
|
// if ( CheckSubShape ( it.Value(), sub ) ) return Standard_True;
|
|
// }
|
|
// return Standard_False;
|
|
//}
|
|
|
|
//=======================================================================
|
|
//function : IsSubShape
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
Standard_Boolean XCAFDoc_ShapeTool::IsSubShape (const TDF_Label &shapeL,
|
|
const TopoDS_Shape &sub) const
|
|
{
|
|
Handle(XCAFDoc_ShapeMapTool) A;
|
|
if ( ! shapeL.FindAttribute(XCAFDoc_ShapeMapTool::GetID(), A) )
|
|
return Standard_False;
|
|
|
|
//TopoDS_Shape S = GetShape ( shapeL );
|
|
//return ! S.IsNull() && CheckSubShape ( S, sub );
|
|
|
|
return A->IsSubShape(sub);
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : FindSubShape
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
Standard_Boolean XCAFDoc_ShapeTool::FindSubShape (const TDF_Label &shapeL,
|
|
const TopoDS_Shape &sub,
|
|
TDF_Label &L) const
|
|
{
|
|
// this code is used instead of the following for performance reasons
|
|
if ( TNaming_Tool::HasLabel(Label(), sub) ) {
|
|
int TransDef = 0;
|
|
L = TNaming_Tool::Label(Label(), sub, TransDef);
|
|
return ( ! L.IsNull() && L.Father() == shapeL );
|
|
}
|
|
|
|
/*
|
|
TDF_ChildIterator chldLabIt(shapeL);
|
|
for (; chldLabIt.More(); chldLabIt.Next() ) {
|
|
TDF_Label subLabel = chldLabIt.Value();
|
|
TopoDS_Shape S;
|
|
if ( GetShape ( subLabel, S ) && S.IsSame ( sub ) ) {
|
|
L = subLabel;
|
|
return Standard_True;
|
|
}
|
|
}
|
|
*/
|
|
return Standard_False;
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : AddSubShape
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
TDF_Label XCAFDoc_ShapeTool::AddSubShape (const TDF_Label &shapeL,
|
|
const TopoDS_Shape &sub) const
|
|
{
|
|
TDF_Label L;
|
|
if ( FindSubShape ( shapeL, sub, L ) ) return L;
|
|
|
|
if ( ! IsSubShape ( shapeL, sub ) ) return L;
|
|
|
|
TDF_TagSource aTag;
|
|
L = aTag.NewChild(shapeL);
|
|
|
|
TNaming_Builder tnBuild(L);
|
|
tnBuild.Generated(sub);
|
|
|
|
// if(!mySubShapes.IsBound(sub))
|
|
// mySubShapes.Bind(sub,L);
|
|
|
|
return L;
|
|
}
|
|
|
|
|
|
//=======================================================================
|
|
//function : FindMainShapeUsingMap
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
TDF_Label XCAFDoc_ShapeTool::FindMainShapeUsingMap(const TopoDS_Shape &sub) const
|
|
{
|
|
//for(Standard_Integer i=1; i<=myNotAssemblies.Length(); i++) {
|
|
// TDF_Label L = myNotAssemblies.Value(i);
|
|
// if(IsSubShape(L,sub)) return L;
|
|
//}
|
|
if(mySubShapes.IsBound(sub))
|
|
return mySubShapes.Find(sub);
|
|
TDF_Label L0;
|
|
return L0;
|
|
}
|
|
|
|
|
|
//=======================================================================
|
|
//function : FindMainShape
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
TDF_Label XCAFDoc_ShapeTool::FindMainShape (const TopoDS_Shape &sub) const
|
|
{
|
|
TDF_ChildIterator it(Label());
|
|
for (; it.More(); it.Next()) {
|
|
TDF_Label L = it.Value();
|
|
if ( ! IsAssembly ( L ) && IsSubShape ( L, sub ) ) return L;
|
|
}
|
|
TDF_Label L0;
|
|
return L0;
|
|
}
|
|
|
|
|
|
//=======================================================================
|
|
//function : GetSubShapes
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
Standard_Boolean XCAFDoc_ShapeTool::GetSubShapes (const TDF_Label &L,
|
|
TDF_LabelSequence& Labels)
|
|
{
|
|
TDF_ChildIterator It(L);
|
|
for (; It.More(); It.Next() ) {
|
|
TDF_Label sub = It.Value();
|
|
if ( IsSubShape ( sub ) ) Labels.Append ( sub );
|
|
}
|
|
return Labels.Length() >0;
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : BaseLabel
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
TDF_Label XCAFDoc_ShapeTool::BaseLabel () const
|
|
{
|
|
return Label();
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : DumpAssembly
|
|
//purpose : recursive part of Dump()
|
|
//=======================================================================
|
|
|
|
static void DumpAssembly(const TDF_Label L,
|
|
const Standard_Integer level,
|
|
const Standard_Boolean deep)
|
|
{
|
|
for (Standard_Integer i=0; i<level; i++)
|
|
cout<<"\t";
|
|
|
|
TCollection_AsciiString Entry;
|
|
TDF_Tool::Entry(L, Entry);
|
|
|
|
cout<<"ASSEMBLY "<<Entry;
|
|
Handle(TDataStd_Name) Name;
|
|
if (L.FindAttribute(TDataStd_Name::GetID(), Name))
|
|
cout<<" "<<Name->Get();
|
|
|
|
if (deep) {
|
|
TopoDS_Shape S;
|
|
XCAFDoc_ShapeTool::GetShape(L, S);
|
|
cout<<"("<<*(void**)&S.TShape();
|
|
if (! S.Location().IsIdentity())
|
|
cout<<", "<< *(void**)&S.Location();
|
|
cout<<") ";
|
|
}
|
|
cout<<endl;
|
|
|
|
Handle(TDataStd_TreeNode) Node;
|
|
TDF_ChildIDIterator NodeIterator(L, XCAFDoc::ShapeRefGUID());
|
|
for (; NodeIterator.More(); NodeIterator.Next()) {
|
|
Node = Handle(TDataStd_TreeNode)::DownCast(NodeIterator.Value());
|
|
if (Node->HasFather()) {
|
|
if (Node->Father()->Label().HasChild())
|
|
DumpAssembly(Node->Father()->Label(), level+1, deep);
|
|
else {
|
|
XCAFDoc_ShapeTool::DumpShape(Node->Father()->Label(), level+1, deep);
|
|
cout<<endl;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : Dump
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
void XCAFDoc_ShapeTool::Dump(const Standard_Boolean deep) const
|
|
{
|
|
Standard_Integer level = 0;
|
|
// TopTools_SequenceOfShape SeqShapes;
|
|
TDF_LabelSequence SeqLabels;
|
|
GetShapes( SeqLabels);
|
|
|
|
if (SeqLabels.Length()>0) cout<<endl;
|
|
Standard_Integer i;
|
|
for (i=1; i<=SeqLabels.Length(); i++) {
|
|
DumpAssembly(SeqLabels.Value(i), level, deep);
|
|
}
|
|
|
|
SeqLabels.Clear();
|
|
GetFreeShapes(SeqLabels);
|
|
cout<<endl<<"Free Shapes: "<<SeqLabels.Length()<<endl;
|
|
for (i = 1; i<=SeqLabels.Length(); i++) {
|
|
DumpShape(SeqLabels.Value(i), level, deep);
|
|
cout<<endl;
|
|
}
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : DumpShape
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
void XCAFDoc_ShapeTool::DumpShape(const TDF_Label& L,const Standard_Integer level,const Standard_Boolean deep)
|
|
{
|
|
TopoDS_Shape S;
|
|
if(! XCAFDoc_ShapeTool::GetShape(L, S) ) return;
|
|
for (Standard_Integer i=0; i<level; i++)
|
|
cout<<"\t";
|
|
|
|
if (S.ShapeType() == TopAbs_COMPOUND) cout<<"ASSEMBLY";
|
|
else TopAbs::Print(S.ShapeType(), cout);
|
|
|
|
TCollection_AsciiString Entry;
|
|
TDF_Tool::Entry(L, Entry);
|
|
cout<<" "<<Entry;
|
|
//cout<<endl;
|
|
Handle(TDataStd_Name) Name;
|
|
if (L.FindAttribute(TDataStd_Name::GetID(),Name))
|
|
cout<<" "<<Name->Get();
|
|
|
|
if (deep) {
|
|
cout<<"("<<*(void**)&S.TShape();
|
|
if (! S.Location().IsIdentity())
|
|
cout<<", "<< *(void**)&S.Location();
|
|
cout<<") ";
|
|
}
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : IsExternRef
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
Standard_Boolean XCAFDoc_ShapeTool::IsExternRef(const TDF_Label& L)
|
|
{
|
|
Handle(TDataStd_UAttribute) Uattr;
|
|
return L.FindAttribute(XCAFDoc::ExternRefGUID(), Uattr);
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : SetExternRefs
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
void XCAFDoc_ShapeTool::SetExternRefs(const TDF_Label& L,
|
|
const TColStd_SequenceOfHAsciiString& SHAS) const
|
|
{
|
|
TDF_Label ShapeLabel = L.NewChild();
|
|
TDataStd_UAttribute::Set(ShapeLabel,XCAFDoc::ExternRefGUID());
|
|
for(Standard_Integer i=1; i<=SHAS.Length(); i++) {
|
|
TDF_Label tmplbl = ShapeLabel.FindChild(i,Standard_True);
|
|
Handle(TCollection_HAsciiString) str = SHAS(i);
|
|
TCollection_ExtendedString extstr(str->String());
|
|
TDataStd_Name::Set(tmplbl,extstr);
|
|
}
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : SetExternRefs
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
TDF_Label XCAFDoc_ShapeTool::SetExternRefs(const TColStd_SequenceOfHAsciiString& SHAS) const
|
|
{
|
|
TDF_Label ShapeLabel;
|
|
TDF_TagSource aTag;
|
|
// add a new label
|
|
ShapeLabel = aTag.NewChild(Label());
|
|
TDataStd_UAttribute::Set(ShapeLabel,XCAFDoc::ExternRefGUID());
|
|
for(Standard_Integer i=1; i<=SHAS.Length(); i++) {
|
|
TDF_Label tmplbl = ShapeLabel.FindChild(i,Standard_True);
|
|
Handle(TCollection_HAsciiString) str = SHAS(i);
|
|
TCollection_ExtendedString extstr(str->String());
|
|
TDataStd_Name::Set(tmplbl,extstr);
|
|
}
|
|
return ShapeLabel;
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : GetExternRefs
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
void XCAFDoc_ShapeTool::GetExternRefs(const TDF_Label& L,
|
|
TColStd_SequenceOfHAsciiString& SHAS)
|
|
{
|
|
Handle(TDataStd_Name) TDN;
|
|
TDF_Label tmplbl;
|
|
for(Standard_Integer i=1; i<=L.NbChildren(); i++) {
|
|
tmplbl = L.FindChild(i);
|
|
if(tmplbl.FindAttribute(TDataStd_Name::GetID(),TDN)) {
|
|
TCollection_ExtendedString extstr = TDN->Get();
|
|
Handle(TCollection_HAsciiString) str =
|
|
new TCollection_HAsciiString(TCollection_AsciiString(extstr, '?'));
|
|
SHAS.Append(str);
|
|
}
|
|
}
|
|
}
|
|
|
|
// API: API work with SHUO (Specified Higher Usage Occurrance) structure
|
|
|
|
//=======================================================================
|
|
//function : GetSHUO
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
Standard_Boolean XCAFDoc_ShapeTool::GetSHUO (const TDF_Label& SHUOLabel,
|
|
Handle(XCAFDoc_GraphNode)& aSHUOAttr)
|
|
{
|
|
if ( !SHUOLabel.FindAttribute( XCAFDoc::SHUORefGUID(), aSHUOAttr ) )
|
|
return Standard_False;
|
|
return Standard_True;
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : GetAllComponentSHUO
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
Standard_Boolean XCAFDoc_ShapeTool::GetAllComponentSHUO (const TDF_Label& theCompLabel,
|
|
TDF_AttributeSequence& theSHUOAttrs)
|
|
{
|
|
TDF_ChildIterator it(theCompLabel);
|
|
for (; it.More(); it.Next()) {
|
|
TDF_Label L = it.Value();
|
|
Handle(XCAFDoc_GraphNode) aSHUOAttr;
|
|
if ( GetSHUO( L, aSHUOAttr ) )
|
|
theSHUOAttrs.Append( aSHUOAttr );
|
|
}
|
|
return (theSHUOAttrs.Length() > 0);
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : SetSHUO
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
Standard_Boolean XCAFDoc_ShapeTool::SetSHUO (const TDF_LabelSequence& labels,
|
|
Handle(XCAFDoc_GraphNode)& MainSHUOAttr) const
|
|
{
|
|
MainSHUOAttr.Nullify();
|
|
// check number of labels
|
|
if (labels.Length() < 2)
|
|
return Standard_False;
|
|
// check is all labels contains components of any assemblyies
|
|
Standard_Integer i;
|
|
for (i = 1; i <= labels.Length(); i++)
|
|
if ( !IsComponent(labels.Value(i)) )
|
|
return Standard_False;
|
|
|
|
TDF_TagSource aTag;
|
|
TDF_Label UpperSubL = aTag.NewChild( labels( 1 ) );
|
|
if (theAutoNaming) {
|
|
TCollection_ExtendedString Entry("SHUO");
|
|
TDataStd_Name::Set(UpperSubL, TCollection_ExtendedString( Entry ));
|
|
}
|
|
Handle(XCAFDoc_GraphNode) aUpperSHUO;
|
|
aUpperSHUO = XCAFDoc_GraphNode::Set( UpperSubL, XCAFDoc::SHUORefGUID() );
|
|
// init out argument by main upper usage SHUO
|
|
MainSHUOAttr = aUpperSHUO;
|
|
// add other next_usage occurrences.
|
|
for (i = 2; i <= labels.Length(); i++) {
|
|
TDF_Label NextSubL = aTag.NewChild( labels( i ) );
|
|
if (theAutoNaming) {
|
|
TCollection_ExtendedString EntrySub("SHUO-");
|
|
EntrySub += i;
|
|
TDataStd_Name::Set(NextSubL, TCollection_ExtendedString( EntrySub ));
|
|
}
|
|
Handle(XCAFDoc_GraphNode) aNextSHUO;
|
|
aNextSHUO = XCAFDoc_GraphNode::Set( NextSubL, XCAFDoc::SHUORefGUID() );
|
|
// set references
|
|
aUpperSHUO->SetChild( aNextSHUO );
|
|
aNextSHUO->SetFather( aUpperSHUO );
|
|
// now lets next_usage become upper_usage for next level of SHUO
|
|
aUpperSHUO = aNextSHUO;
|
|
UpperSubL = NextSubL;
|
|
}
|
|
|
|
return Standard_True;
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : GetSHUOUpperUsage
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
Standard_Boolean XCAFDoc_ShapeTool::GetSHUOUpperUsage (const TDF_Label& NextUsageL,
|
|
TDF_LabelSequence& aLabels)
|
|
{
|
|
Handle(XCAFDoc_GraphNode) aNextSHUO;
|
|
if( !GetSHUO( NextUsageL, aNextSHUO ) || aNextSHUO->NbFathers()<1 )
|
|
return Standard_False;
|
|
|
|
// get upper_usage SHAO
|
|
for (Standard_Integer i = 1; i <= aNextSHUO->NbFathers(); i++)
|
|
aLabels.Append( aNextSHUO->GetFather(i)->Label() );
|
|
return Standard_True;
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : GetSHUONextUsage
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
Standard_Boolean XCAFDoc_ShapeTool::GetSHUONextUsage (const TDF_Label& UpperUsageL,
|
|
TDF_LabelSequence& aLabels)
|
|
{
|
|
Handle(XCAFDoc_GraphNode) aUpperSHUO;
|
|
if ( !GetSHUO( UpperUsageL, aUpperSHUO ) || aUpperSHUO->NbChildren()<1 )
|
|
return Standard_False;
|
|
// get upper_usage SHAO
|
|
for (Standard_Integer i = 1; i <= aUpperSHUO->NbChildren(); i++)
|
|
aLabels.Append( aUpperSHUO->GetChild(i)->Label() );
|
|
return Standard_True;
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : RemoveSHUO
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
Standard_Boolean XCAFDoc_ShapeTool::RemoveSHUO (const TDF_Label& L) const
|
|
{
|
|
L.ForgetAllAttributes (Standard_True);
|
|
return Standard_True;
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : checkForShape
|
|
//purpose : auxilary
|
|
//=======================================================================
|
|
|
|
static Standard_Boolean checkForShape (const TopoDS_Shape& theShape,
|
|
const TopoDS_Shape& theCurSh,
|
|
const TDF_Label& theUserL,
|
|
TDF_LabelSequence& theLabels)
|
|
{
|
|
// the label of an assembly which contains this component
|
|
TDF_Label aSuperUserL = theUserL.Father();
|
|
TopLoc_Location aSupLoc, aCompLoc;
|
|
aSupLoc = ::XCAFDoc_ShapeTool::GetLocation ( aSuperUserL );
|
|
aCompLoc = ::XCAFDoc_ShapeTool::GetLocation ( theUserL );
|
|
TopoDS_Shape aCopySh = theCurSh;
|
|
aCompLoc = aCompLoc.Multiplied( theCurSh.Location() );
|
|
aSupLoc = aSupLoc.Multiplied( aCompLoc );
|
|
aCopySh.Location( aSupLoc );
|
|
if ( aCopySh.IsSame( theShape ) ) {
|
|
theLabels.Prepend( theUserL );
|
|
return Standard_True;
|
|
}
|
|
// try to search deeply (upper by assmebly structure)
|
|
TDF_LabelSequence aNewLabels;
|
|
for (Standard_Integer j = 1; j <= theLabels.Length(); j++)
|
|
aNewLabels.Append( theLabels.Value( j ) );
|
|
aNewLabels.Prepend( theUserL );
|
|
TDF_LabelSequence aUsers;
|
|
::XCAFDoc_ShapeTool::GetUsers( aSuperUserL, aUsers );
|
|
for (Standard_Integer i = 1; i <= aUsers.Length(); i++)
|
|
if ( checkForShape( theShape, aCopySh, aUsers.Value( i ), aNewLabels ) ) {
|
|
// get solution
|
|
theLabels = aNewLabels;
|
|
return Standard_True;
|
|
}
|
|
return Standard_False;
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : FindComponent
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
Standard_Boolean XCAFDoc_ShapeTool::FindComponent (const TopoDS_Shape& theShape,
|
|
TDF_LabelSequence& theLabels) const
|
|
{
|
|
theLabels.Clear();
|
|
// search for a top-level shape that corresponds to this component
|
|
TopoDS_Shape S0 = theShape;
|
|
TopLoc_Location loc;
|
|
S0.Location ( loc );
|
|
TDF_Label aRefL = FindShape( S0 );
|
|
if (aRefL.IsNull())
|
|
return Standard_False; // cannot find top-level shape.
|
|
|
|
TDF_LabelSequence aUsers;
|
|
::XCAFDoc_ShapeTool::GetUsers( aRefL, aUsers );
|
|
for (Standard_Integer i = 1; i <= aUsers.Length(); i++)
|
|
if ( checkForShape( theShape, S0, aUsers.Value( i ), theLabels ) )
|
|
break;
|
|
|
|
return (theLabels.Length() > 0);
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : getShapesOfSHUO
|
|
//purpose : auxilary
|
|
//=======================================================================
|
|
|
|
static Standard_Boolean getShapesOfSHUO (TopLoc_IndexedMapOfLocation& theaPrevLocMap,
|
|
const Handle(XCAFDoc_ShapeTool)& theSTool,
|
|
const TDF_Label& theSHUOlab,
|
|
TopoDS_Shape& theShape)
|
|
{
|
|
Handle(XCAFDoc_GraphNode) SHUO;
|
|
TDF_LabelSequence aLabSeq;
|
|
theSTool->GetSHUONextUsage( theSHUOlab, aLabSeq );
|
|
if (aLabSeq.Length() >= 1)
|
|
for (Standard_Integer i = 1; i <= aLabSeq.Length(); i++) {
|
|
TDF_Label aSubCompL = aLabSeq.Value( i );
|
|
TopLoc_Location compLoc = XCAFDoc_ShapeTool::GetLocation ( aSubCompL.Father() );
|
|
// create new map of laocation (to not merge locations from different shapes)
|
|
TopLoc_IndexedMapOfLocation aNewPrevLocMap;
|
|
for (Standard_Integer m = 1; m <= theaPrevLocMap.Extent(); m++)
|
|
aNewPrevLocMap.Add( theaPrevLocMap.FindKey( m ) );
|
|
aNewPrevLocMap.Add( compLoc );
|
|
// got for the new sublocations and corresponding shape
|
|
getShapesOfSHUO( aNewPrevLocMap, theSTool, aSubCompL, theShape );
|
|
}
|
|
else {
|
|
TopoDS_Shape aSHUO_NUSh = theSTool->GetShape ( theSHUOlab.Father() );
|
|
if ( aSHUO_NUSh.IsNull() ) return Standard_False;
|
|
// cause got shape with location already.
|
|
TopLoc_Location nullLoc;
|
|
aSHUO_NUSh.Location ( nullLoc );
|
|
// multiply the locations
|
|
Standard_Integer intMapLenght = theaPrevLocMap.Extent();
|
|
if ( intMapLenght < 1 )
|
|
return Standard_False; // should not be, but to avoid exception...?
|
|
TopLoc_Location SupcompLoc;
|
|
SupcompLoc = theaPrevLocMap.FindKey( intMapLenght );
|
|
if (intMapLenght > 1) {
|
|
Standard_Integer l = intMapLenght - 1;
|
|
while (l >= 1) {
|
|
SupcompLoc = theaPrevLocMap.FindKey( l ).Multiplied( SupcompLoc );
|
|
l--;
|
|
}
|
|
}
|
|
aSHUO_NUSh.Location( SupcompLoc );
|
|
theShape = aSHUO_NUSh;
|
|
}
|
|
return (!theShape.IsNull());
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : GetSHUOInstance
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
TopoDS_Shape XCAFDoc_ShapeTool::GetSHUOInstance (const Handle(XCAFDoc_GraphNode)& theSHUO) const
|
|
{
|
|
TopoDS_Shape aShape;
|
|
if (theSHUO.IsNull())
|
|
return aShape;
|
|
|
|
TDF_Label aSHUOlab = theSHUO->Label();
|
|
// get location of the assembly
|
|
TopLoc_Location loc = XCAFDoc_ShapeTool::GetLocation ( aSHUOlab.Father().Father() );
|
|
// get location of the component
|
|
TopLoc_Location compLoc = XCAFDoc_ShapeTool::GetLocation ( aSHUOlab.Father() );
|
|
TopLoc_IndexedMapOfLocation aPrevLocMap;
|
|
// get previous setted location
|
|
if ( !loc.IsIdentity() )
|
|
aPrevLocMap.Add( loc );
|
|
aPrevLocMap.Add( compLoc );
|
|
// get shape by recurse method
|
|
const Handle(XCAFDoc_ShapeTool)& STool = this;
|
|
getShapesOfSHUO( aPrevLocMap, STool, aSHUOlab, aShape );
|
|
|
|
return aShape;
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : getUsersShapesOfSHUO
|
|
//purpose : auxilary
|
|
//=======================================================================
|
|
|
|
static Standard_Boolean getUsersShapesOfSHUO (TopLoc_IndexedMapOfLocation& aPrevLocMap,
|
|
const Handle(XCAFDoc_ShapeTool)& STool,
|
|
const TDF_Label& aSHUOlab,
|
|
const TDF_Label& theUserL,
|
|
TopTools_SequenceOfShape& theSHUOShapeSeq)
|
|
{
|
|
TopLoc_IndexedMapOfLocation aNewPrevLocMap;
|
|
// get location of the assembly
|
|
TopLoc_Location loc = XCAFDoc_ShapeTool::GetLocation ( theUserL.Father() );
|
|
// get location of the component
|
|
TopLoc_Location compLoc = XCAFDoc_ShapeTool::GetLocation ( theUserL );
|
|
// get previous setted location
|
|
aNewPrevLocMap.Add( loc );
|
|
aNewPrevLocMap.Add( compLoc );
|
|
Standard_Integer i;
|
|
for (i = 1; i <= aPrevLocMap.Extent(); i++)
|
|
aNewPrevLocMap.Add( aPrevLocMap.FindKey(i) );
|
|
TDF_Label L = theUserL.Father();
|
|
TDF_LabelSequence usersLab;
|
|
::XCAFDoc_ShapeTool::GetUsers( L, usersLab );
|
|
if (usersLab.Length() == 0) {
|
|
TopoDS_Shape aShape;
|
|
getShapesOfSHUO( aNewPrevLocMap, STool, aSHUOlab, aShape );
|
|
if (!aShape.IsNull()) {
|
|
theSHUOShapeSeq.Append(aShape);
|
|
return Standard_True;
|
|
}
|
|
}
|
|
// now iterates on users of this assembly as component
|
|
for ( i = 1; i <= usersLab.Length(); i++ ) {
|
|
TDF_Label aNewUserL = usersLab.Value(i);
|
|
getUsersShapesOfSHUO( aNewPrevLocMap, STool, aSHUOlab, aNewUserL, theSHUOShapeSeq );
|
|
}
|
|
|
|
return (theSHUOShapeSeq.Length() > 1);
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : GetAllSHUOInstances
|
|
//purpose : auxilary
|
|
//=======================================================================
|
|
|
|
Standard_Boolean XCAFDoc_ShapeTool::GetAllSHUOInstances (const Handle(XCAFDoc_GraphNode)& theSHUO,
|
|
TopTools_SequenceOfShape& theSHUOShapeSeq) const
|
|
{
|
|
if (theSHUO.IsNull())
|
|
return Standard_False;
|
|
|
|
TDF_Label aSHUOlab = theSHUO->Label();
|
|
TopLoc_IndexedMapOfLocation aPrevLocMap;
|
|
// get location of the assembly
|
|
TopLoc_Location loc = XCAFDoc_ShapeTool::GetLocation ( aSHUOlab.Father().Father() );
|
|
// get location of the component
|
|
TopLoc_Location compLoc = XCAFDoc_ShapeTool::GetLocation ( aSHUOlab.Father() );
|
|
// get previous setted location
|
|
if ( !loc.IsIdentity() )
|
|
aPrevLocMap.Add( loc );
|
|
aPrevLocMap.Add( compLoc );
|
|
// get label of assembly
|
|
TDF_Label L = aSHUOlab.Father().Father();
|
|
TDF_LabelSequence usersLab;
|
|
::XCAFDoc_ShapeTool::GetUsers( L, usersLab );
|
|
TopoDS_Shape aShape;
|
|
const Handle(XCAFDoc_ShapeTool)& STool = this;
|
|
if (usersLab.Length() == 0) {
|
|
getShapesOfSHUO( aPrevLocMap, STool, aSHUOlab, aShape );
|
|
if (!aShape.IsNull()) {
|
|
theSHUOShapeSeq.Append(aShape);
|
|
return Standard_True;
|
|
}
|
|
}
|
|
// now iterates on users of this assembly as component
|
|
for (Standard_Integer i = 1; i <= usersLab.Length(); i++) {
|
|
TDF_Label aUserL = usersLab.Value(i);
|
|
getUsersShapesOfSHUO( aPrevLocMap, STool, aSHUOlab, aUserL, theSHUOShapeSeq );
|
|
}
|
|
|
|
return (theSHUOShapeSeq.Length() > 1);
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : SetInstanceSHUO
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
Handle(XCAFDoc_GraphNode) XCAFDoc_ShapeTool::SetInstanceSHUO (const TopoDS_Shape& theShape) const
|
|
{
|
|
Handle(XCAFDoc_GraphNode) SHUO;
|
|
TDF_LabelSequence aLabels;
|
|
if ( FindComponent( theShape, aLabels ) )
|
|
// set shuo structure on labels of component-assembly structure
|
|
SetSHUO( aLabels, SHUO );
|
|
return SHUO;
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : FindSHUO
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
Standard_Boolean XCAFDoc_ShapeTool::FindSHUO (const TDF_LabelSequence& theLabels,
|
|
Handle(XCAFDoc_GraphNode)& theSHUOAttr)
|
|
{
|
|
TDF_AttributeSequence SHUOAttrs;
|
|
TDF_Label aCompLabel = theLabels.Value(1);
|
|
if (! ::XCAFDoc_ShapeTool::GetAllComponentSHUO( aCompLabel, SHUOAttrs ) )
|
|
return Standard_False;
|
|
// WARNING: manage that each SHUO upper_usage have only one SHUO next_usage
|
|
for (Standard_Integer i = 1; i <= SHUOAttrs.Length(); i++) {
|
|
TDF_LabelSequence aCondidate;
|
|
Handle(XCAFDoc_GraphNode) anSHUO = Handle(XCAFDoc_GraphNode)::DownCast(SHUOAttrs.Value(i));
|
|
aCondidate.Append( anSHUO->Label().Father() );
|
|
while (anSHUO->NbChildren()) {
|
|
anSHUO = anSHUO->GetChild( 1 );
|
|
aCondidate.Append( anSHUO->Label().Father() );
|
|
}
|
|
// check the label sequences
|
|
Standard_Boolean isEqual = Standard_True;
|
|
if (theLabels.Length() != aCondidate.Length())
|
|
isEqual = Standard_False;
|
|
else
|
|
for (Standard_Integer li = 1; li <= theLabels.Length(); li++)
|
|
if ( theLabels.Value(li) != aCondidate.Value(li) ) {
|
|
isEqual = Standard_False;
|
|
break;
|
|
}
|
|
if (!isEqual)
|
|
continue;
|
|
|
|
theSHUOAttr = Handle(XCAFDoc_GraphNode)::DownCast(SHUOAttrs.Value(i));
|
|
break;
|
|
}
|
|
return ( !theSHUOAttr.IsNull() );
|
|
}
|