[occt.git] / src / XCAFDoc / XCAFDoc_Editor.cxx

// Created on: 2015-05-14
// Created by: data exchange team
// Copyright (c) 2000-2015 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.

#include <XCAFDoc_Editor.hxx>
#include <XCAFDoc.hxx>

#include <TDF_Label.hxx>
#include <TDF_LabelSequence.hxx>
#include <TDF_ChildIterator.hxx>
#include <TDataStd_Name.hxx>
#include <TDataStd_UAttribute.hxx>
#include <TopLoc_Location.hxx>
#include <TopoDS_Shape.hxx>

#include <XCAFDoc_DocumentTool.hxx>
#include <XCAFDoc_ColorTool.hxx>
#include <XCAFDoc_LayerTool.hxx>
#include <XCAFDoc_ShapeTool.hxx>
#include <XCAFDoc_ShapeMapTool.hxx>
#include <XCAFDoc_Location.hxx>
#include <TNaming_NamedShape.hxx>

//=======================================================================
//function : Expand
//purpose  : Convert Shape to assembly
//=======================================================================

Standard_Boolean XCAFDoc_Editor::Expand (const TDF_Label& Doc, const TDF_Label& Shape,
  const Standard_Boolean recursively)
{
  if(Doc.IsNull() || Shape.IsNull())
    return Standard_False;

  Handle(XCAFDoc_ColorTool) aColorTool = XCAFDoc_DocumentTool::ColorTool(Doc);
  Handle(XCAFDoc_LayerTool) aLayerTool = XCAFDoc_DocumentTool::LayerTool(Doc);
  Handle(XCAFDoc_ShapeTool) aShapeTool = XCAFDoc_DocumentTool::ShapeTool(Doc);
  Standard_Boolean isAutoNaming = aShapeTool->AutoNaming();
  aShapeTool->SetAutoNaming(Standard_False);

  TDF_Label aCompoundPartL = Shape;
  if (aShapeTool->IsReference(Shape))
    aShapeTool->GetReferredShape(aCompoundPartL, aCompoundPartL);

  TopoDS_Shape aS = aShapeTool->GetShape(aCompoundPartL);
  if (aShapeTool->Expand(aCompoundPartL))
  {
    //move attributes
    TDF_ChildIterator anIter(aCompoundPartL, Standard_True);
    for(; anIter.More(); anIter.Next())
    {
      TDF_Label aChild = anIter.Value();
      TDF_LabelSequence aLayers;
      TDF_LabelSequence aColors;
      Handle(TDataStd_Name) aName;
      getParams(Doc, aChild, aColors, aLayers, aName);
      //get part
      TDF_Label aPart;
      if(aShapeTool->GetReferredShape(aChild, aPart))
      {
        setParams(Doc, aPart, aColors, aLayers, aName);
        //remove unnecessary links
        TopoDS_Shape aShape = aShapeTool->GetShape(aChild);
        if(!aShapeTool->GetShape(aPart.Father()).IsNull())
        {
          TopLoc_Location nulloc;
          aPart.ForgetAttribute(XCAFDoc::ShapeRefGUID());
          if (aShapeTool->GetShape(aPart.Father()).ShapeType() == TopAbs_COMPOUND)
          {
            aShapeTool->SetShape(aPart, aShape);
          }
          aPart.ForgetAttribute(XCAFDoc_ShapeMapTool::GetID());
          aChild.ForgetAllAttributes(Standard_False);
        }
        aChild.ForgetAttribute(TNaming_NamedShape::GetID());
        aChild.ForgetAttribute(XCAFDoc_ShapeMapTool::GetID());
      }
      else
      {
        // If new original shape is not created, try to process this child
        // as subshape of new part
        TDF_LabelSequence aUsers;
        if (aShapeTool->GetUsers(aChild, aUsers) > 0)
        {
          for (Standard_Integer i = 1; i <= aUsers.Length(); i++)
          {
            TDF_Label aSubLabel = aUsers.Value(i);
            //remove unnecessary links
            aSubLabel.ForgetAttribute(XCAFDoc::ShapeRefGUID());
            aSubLabel.ForgetAttribute(XCAFDoc_ShapeMapTool::GetID());
            setParams(Doc, aSubLabel, aColors, aLayers, aName);
          }
          aChild.ForgetAllAttributes(Standard_False);
        }
      }
    }
    //if assembly contains compound, expand it recursively(if flag recursively is true)
    if(recursively)
    {
      anIter.Initialize(aCompoundPartL);
      for(; anIter.More(); anIter.Next())
      {
        TDF_Label aChild = anIter.Value();
        TDF_Label aPart;
        if(aShapeTool->GetReferredShape(aChild, aPart))
        {
          TopoDS_Shape aPartShape = aShapeTool->GetShape(aPart);
          if (!aPartShape.IsNull() && aPartShape.ShapeType() == TopAbs_COMPOUND)
            Expand(Doc, aPart, recursively);
        }
      }
    }
    aShapeTool->SetAutoNaming(isAutoNaming);
    return Standard_True;
  }
  aShapeTool->SetAutoNaming(isAutoNaming);
  return Standard_False;
}

//=======================================================================
//function : Expand
//purpose  : Convert all compounds in Doc to assembly
//=======================================================================

Standard_Boolean XCAFDoc_Editor::Expand (const TDF_Label& Doc, const Standard_Boolean recursively)
{
  Standard_Boolean result = Standard_False;
  TDF_LabelSequence aLabels;
  Handle(XCAFDoc_ShapeTool) aShapeTool = XCAFDoc_DocumentTool::ShapeTool(Doc);
  aShapeTool->GetFreeShapes(aLabels);
  for(Standard_Integer i = 1; i <= aLabels.Length(); i++)
  {
    TopoDS_Shape aS = aShapeTool->GetShape(aLabels(i));
    if (!aS.IsNull() && aS.ShapeType() == TopAbs_COMPOUND && !aShapeTool->IsAssembly(aLabels(i)))
      if (Expand(Doc, aLabels(i), recursively))
      {
        result = Standard_True;
      }
  }
  return result;
}

//=======================================================================
//function : getParams
//purpose  : Get colors layers and name
//=======================================================================

Standard_Boolean XCAFDoc_Editor::getParams (const TDF_Label& Doc, const TDF_Label& Label,
                                            TDF_LabelSequence& Colors, TDF_LabelSequence& Layers,
                                            Handle(TDataStd_Name)& Name)
{
  if(Doc.IsNull() || Label.IsNull())
    return Standard_False;

  Handle(XCAFDoc_ColorTool) aColorTool = XCAFDoc_DocumentTool::ColorTool(Doc);
  Handle(XCAFDoc_LayerTool) aLayerTool = XCAFDoc_DocumentTool::LayerTool(Doc);

  //get colors
  XCAFDoc_ColorType aTypeColor = XCAFDoc_ColorGen;
  for(;;)
  {
    TDF_Label aColor;
    aColorTool->GetColor(Label, aTypeColor, aColor);
    Colors.Append(aColor);
    if(aTypeColor == XCAFDoc_ColorCurv)
      break;
    aTypeColor = (XCAFDoc_ColorType)(aTypeColor + 1);
  }

  //get layers
  aLayerTool->GetLayers(Label, Layers);

  //get name
  Label.FindAttribute(TDataStd_Name::GetID(), Name);
  return Standard_True;
}

//=======================================================================
//function : setParams
//purpose  : set colors layers and name
//=======================================================================

Standard_Boolean XCAFDoc_Editor::setParams (const TDF_Label& Doc, const TDF_Label& Label,
                                            const TDF_LabelSequence& Colors, const TDF_LabelSequence& Layers,
                                            const Handle(TDataStd_Name)& Name)
{
  if(Doc.IsNull() || Label.IsNull())
    return Standard_False;

  Handle(XCAFDoc_ColorTool) aColorTool = XCAFDoc_DocumentTool::ColorTool(Doc);
  Handle(XCAFDoc_LayerTool) aLayerTool = XCAFDoc_DocumentTool::LayerTool(Doc);
  Handle(XCAFDoc_ShapeTool) aShapeTool = XCAFDoc_DocumentTool::ShapeTool(Doc);

  //set layers
  if(!Layers.IsEmpty())
  {
    for(Standard_Integer i = 1; i <= Layers.Length(); i++)
    {
      aLayerTool->SetLayer(Label, Layers.Value(i));
    }
  }
  //set colors
  if(!Colors.IsEmpty())
  {
    XCAFDoc_ColorType aTypeColor = XCAFDoc_ColorGen;
    for(Standard_Integer i = 1; i <= Colors.Length(); i++)
    {
      if(!Colors.Value(i).IsNull())
        aColorTool->SetColor(Label, Colors.Value(i), aTypeColor);
      aTypeColor = (XCAFDoc_ColorType)(aTypeColor + 1);
    }
  }
  //set name
  if(!Name.IsNull())
  {
    if(Name->Get().Search("=>") < 0)
      TDataStd_Name::Set(Label, Name->Get());
  }
  else
  {
    Standard_SStream Stream;
    TopoDS_Shape aShape = aShapeTool->GetShape(Label);
    TopAbs::Print(aShape.ShapeType(), Stream);
    TCollection_AsciiString aName (Stream.str().c_str());
    TDataStd_Name::Set(Label, TCollection_ExtendedString(aName));
  }
  return Standard_True;
}
Commit	Line	Data
5f07d05d	1	// Created on: 2015-05-14
	2	// Created by: data exchange team
	3	// Copyright (c) 2000-2015 OPEN CASCADE SAS
	4	//
	5	// This file is part of Open CASCADE Technology software library.
	6	//
	7	// This library is free software; you can redistribute it and/or modify it under
	8	// the terms of the GNU Lesser General Public License version 2.1 as published
	9	// by the Free Software Foundation, with special exception defined in the file
	10	// OCCT_LGPL_EXCEPTION.txt. Consult the file LICENSE_LGPL_21.txt included in OCCT
	11	// distribution for complete text of the license and disclaimer of any warranty.
	12	//
	13	// Alternatively, this file may be used under the terms of Open CASCADE
	14	// commercial license or contractual agreement.
	15
	16	#include <XCAFDoc_Editor.hxx>
	17	#include <XCAFDoc.hxx>
	18
	19	#include <TDF_Label.hxx>
	20	#include <TDF_LabelSequence.hxx>
	21	#include <TDF_ChildIterator.hxx>
	22	#include <TDataStd_Name.hxx>
	23	#include <TDataStd_UAttribute.hxx>
	24	#include <TopLoc_Location.hxx>
	25	#include <TopoDS_Shape.hxx>
	26
	27	#include <XCAFDoc_DocumentTool.hxx>
	28	#include <XCAFDoc_ColorTool.hxx>
	29	#include <XCAFDoc_LayerTool.hxx>
	30	#include <XCAFDoc_ShapeTool.hxx>
	31	#include <XCAFDoc_ShapeMapTool.hxx>
	32	#include <XCAFDoc_Location.hxx>
	33	#include <TNaming_NamedShape.hxx>
	34
	35	//=======================================================================
	36	//function : Expand
00dfcc76	37	//purpose : Convert Shape to assembly
5f07d05d	38	//=======================================================================
	39
	40	Standard_Boolean XCAFDoc_Editor::Expand (const TDF_Label& Doc, const TDF_Label& Shape,
	41	const Standard_Boolean recursively)
	42	{
	43	if(Doc.IsNull() \|\| Shape.IsNull())
	44	return Standard_False;
	45
	46	Handle(XCAFDoc_ColorTool) aColorTool = XCAFDoc_DocumentTool::ColorTool(Doc);
	47	Handle(XCAFDoc_LayerTool) aLayerTool = XCAFDoc_DocumentTool::LayerTool(Doc);
	48	Handle(XCAFDoc_ShapeTool) aShapeTool = XCAFDoc_DocumentTool::ShapeTool(Doc);
f277ba37	49	Standard_Boolean isAutoNaming = aShapeTool->AutoNaming();
bc728d26	50	aShapeTool->SetAutoNaming(Standard_False);
5f07d05d	51
8cbf1f8d	52	TDF_Label aCompoundPartL = Shape;
	53	if (aShapeTool->IsReference(Shape))
	54	aShapeTool->GetReferredShape(aCompoundPartL, aCompoundPartL);
	55
	56	TopoDS_Shape aS = aShapeTool->GetShape(aCompoundPartL);
	57	if (aShapeTool->Expand(aCompoundPartL))
5f07d05d	58	{
00dfcc76	59	//move attributes
8cbf1f8d	60	TDF_ChildIterator anIter(aCompoundPartL, Standard_True);
5f07d05d	61	for(; anIter.More(); anIter.Next())
	62	{
	63	TDF_Label aChild = anIter.Value();
5f07d05d	64	TDF_LabelSequence aLayers;
	65	TDF_LabelSequence aColors;
	66	Handle(TDataStd_Name) aName;
	67	getParams(Doc, aChild, aColors, aLayers, aName);
	68	//get part
	69	TDF_Label aPart;
	70	if(aShapeTool->GetReferredShape(aChild, aPart))
	71	{
	72	setParams(Doc, aPart, aColors, aLayers, aName);
	73	//remove unnecessary links
	74	TopoDS_Shape aShape = aShapeTool->GetShape(aChild);
	75	if(!aShapeTool->GetShape(aPart.Father()).IsNull())
	76	{
	77	TopLoc_Location nulloc;
bc728d26	78	aPart.ForgetAttribute(XCAFDoc::ShapeRefGUID());
bc728d26	79	if (aShapeTool->GetShape(aPart.Father()).ShapeType() == TopAbs_COMPOUND)
5f07d05d	80	{
bc728d26	81	aShapeTool->SetShape(aPart, aShape);
5f07d05d	82	}
bc728d26	83	aPart.ForgetAttribute(XCAFDoc_ShapeMapTool::GetID());
bc728d26	84	aChild.ForgetAllAttributes(Standard_False);
5f07d05d	85	}
	86	aChild.ForgetAttribute(TNaming_NamedShape::GetID());
	87	aChild.ForgetAttribute(XCAFDoc_ShapeMapTool::GetID());
	88	}
00dfcc76	89	else
	90	{
	91	// If new original shape is not created, try to process this child
	92	// as subshape of new part
	93	TDF_LabelSequence aUsers;
	94	if (aShapeTool->GetUsers(aChild, aUsers) > 0)
	95	{
	96	for (Standard_Integer i = 1; i <= aUsers.Length(); i++)
	97	{
	98	TDF_Label aSubLabel = aUsers.Value(i);
00dfcc76	99	//remove unnecessary links
	100	aSubLabel.ForgetAttribute(XCAFDoc::ShapeRefGUID());
	101	aSubLabel.ForgetAttribute(XCAFDoc_ShapeMapTool::GetID());
f277ba37	102	setParams(Doc, aSubLabel, aColors, aLayers, aName);
00dfcc76	103	}
	104	aChild.ForgetAllAttributes(Standard_False);
	105	}
	106	}
5f07d05d	107	}
	108	//if assembly contains compound, expand it recursively(if flag recursively is true)
	109	if(recursively)
	110	{
8cbf1f8d	111	anIter.Initialize(aCompoundPartL);
5f07d05d	112	for(; anIter.More(); anIter.Next())
	113	{
	114	TDF_Label aChild = anIter.Value();
	115	TDF_Label aPart;
	116	if(aShapeTool->GetReferredShape(aChild, aPart))
	117	{
00dfcc76	118	TopoDS_Shape aPartShape = aShapeTool->GetShape(aPart);
bc728d26	119	if (!aPartShape.IsNull() && aPartShape.ShapeType() == TopAbs_COMPOUND)
00dfcc76	120	Expand(Doc, aPart, recursively);
5f07d05d	121	}
	122	}
	123	}
f277ba37	124	aShapeTool->SetAutoNaming(isAutoNaming);
5f07d05d	125	return Standard_True;
5f07d05d	126	}
f277ba37	127	aShapeTool->SetAutoNaming(isAutoNaming);
5f07d05d	128	return Standard_False;
	129	}
	130
	131	//=======================================================================
	132	//function : Expand
00dfcc76	133	//purpose : Convert all compounds in Doc to assembly
5f07d05d	134	//=======================================================================
	135
	136	Standard_Boolean XCAFDoc_Editor::Expand (const TDF_Label& Doc, const Standard_Boolean recursively)
	137	{
	138	Standard_Boolean result = Standard_False;
	139	TDF_LabelSequence aLabels;
	140	Handle(XCAFDoc_ShapeTool) aShapeTool = XCAFDoc_DocumentTool::ShapeTool(Doc);
	141	aShapeTool->GetFreeShapes(aLabels);
	142	for(Standard_Integer i = 1; i <= aLabels.Length(); i++)
	143	{
	144	TopoDS_Shape aS = aShapeTool->GetShape(aLabels(i));
	145	if (!aS.IsNull() && aS.ShapeType() == TopAbs_COMPOUND && !aShapeTool->IsAssembly(aLabels(i)))
	146	if (Expand(Doc, aLabels(i), recursively))
	147	{
	148	result = Standard_True;
	149	}
	150	}
	151	return result;
	152	}
	153
	154	//=======================================================================
	155	//function : getParams
	156	//purpose : Get colors layers and name
	157	//=======================================================================
	158
	159	Standard_Boolean XCAFDoc_Editor::getParams (const TDF_Label& Doc, const TDF_Label& Label,
	160	TDF_LabelSequence& Colors, TDF_LabelSequence& Layers,
	161	Handle(TDataStd_Name)& Name)
	162	{
	163	if(Doc.IsNull() \|\| Label.IsNull())
	164	return Standard_False;
	165
	166	Handle(XCAFDoc_ColorTool) aColorTool = XCAFDoc_DocumentTool::ColorTool(Doc);
	167	Handle(XCAFDoc_LayerTool) aLayerTool = XCAFDoc_DocumentTool::LayerTool(Doc);
	168
	169	//get colors
	170	XCAFDoc_ColorType aTypeColor = XCAFDoc_ColorGen;
	171	for(;;)
	172	{
	173	TDF_Label aColor;
	174	aColorTool->GetColor(Label, aTypeColor, aColor);
	175	Colors.Append(aColor);
	176	if(aTypeColor == XCAFDoc_ColorCurv)
	177	break;
	178	aTypeColor = (XCAFDoc_ColorType)(aTypeColor + 1);
	179	}
	180
	181	//get layers
	182	aLayerTool->GetLayers(Label, Layers);
	183
	184	//get name
	185	Label.FindAttribute(TDataStd_Name::GetID(), Name);
	186	return Standard_True;
	187	}
	188
	189	//=======================================================================
	190	//function : setParams
	191	//purpose : set colors layers and name
	192	//=======================================================================
	193
	194	Standard_Boolean XCAFDoc_Editor::setParams (const TDF_Label& Doc, const TDF_Label& Label,
	195	const TDF_LabelSequence& Colors, const TDF_LabelSequence& Layers,
	196	const Handle(TDataStd_Name)& Name)
	197	{
198	if(Doc.IsNull() \|\| Label.IsNull())
199	return Standard_False;
200
201	Handle(XCAFDoc_ColorTool) aColorTool = XCAFDoc_DocumentTool::ColorTool(Doc);
202	Handle(XCAFDoc_LayerTool) aLayerTool = XCAFDoc_DocumentTool::LayerTool(Doc);
203	Handle(XCAFDoc_ShapeTool) aShapeTool = XCAFDoc_DocumentTool::ShapeTool(Doc);
204
205	//set layers
206	if(!Layers.IsEmpty())
207	{
208	for(Standard_Integer i = 1; i <= Layers.Length(); i++)
209	{
210	aLayerTool->SetLayer(Label, Layers.Value(i));
211	}
212	}
213	//set colors
214	if(!Colors.IsEmpty())
215	{
216	XCAFDoc_ColorType aTypeColor = XCAFDoc_ColorGen;
217	for(Standard_Integer i = 1; i <= Colors.Length(); i++)
218	{
219	if(!Colors.Value(i).IsNull())
220	aColorTool->SetColor(Label, Colors.Value(i), aTypeColor);
221	aTypeColor = (XCAFDoc_ColorType)(aTypeColor + 1);
222	}
223	}
224	//set name
225	if(!Name.IsNull())
226	{
227	if(Name->Get().Search("=>") < 0)
228	TDataStd_Name::Set(Label, Name->Get());
229	}
230	else
231	{
232	Standard_SStream Stream;
233	TopoDS_Shape aShape = aShapeTool->GetShape(Label);
234	TopAbs::Print(aShape.ShapeType(), Stream);
235	TCollection_AsciiString aName (Stream.str().c_str());
236	TDataStd_Name::Set(Label, TCollection_ExtendedString(aName));
237	}
238	return Standard_True;
239	}