src/Model/Model_BodyBuilder.cpp

   1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D
   2
   3 // File:        Model_ResultBody.cpp
   4 // Created:     08 Jul 2014
   5 // Author:      Mikhail PONIKAROV
   6
   7 #include <Model_BodyBuilder.h>
   8
   9 #include <Model_Data.h>
  10 #include <Model_Document.h>
  11 #include <TNaming_Builder.hxx>
  12 #include <TNaming_NamedShape.hxx>
  13 #include <TNaming_Iterator.hxx>
  14 #include <TNaming_Tool.hxx>
  15 #include <TDataStd_Name.hxx>
  16 #include <TDataStd_Integer.hxx>
  17 #include <TopoDS.hxx>
  18 #include <TopoDS_Face.hxx>
  19 #include <TDF_ChildIterator.hxx>
  20 #include <TDF_Reference.hxx>
  21 #include <TopTools_MapOfShape.hxx>
  22 #include <TopExp_Explorer.hxx>
  23 #include <TopTools_ListOfShape.hxx>
  24 #include <TopTools_ListIteratorOfListOfShape.hxx>
  25 #include <TopTools_DataMapOfShapeListOfShape.hxx>
  26 #include <TopTools_DataMapIteratorOfDataMapOfShapeListOfShape.hxx>
  27 #include <TopTools_DataMapIteratorOfDataMapOfShapeShape.hxx>
  28 #include <TopTools_MapIteratorOfMapOfShape.hxx>
  29 #include <TopTools_IndexedDataMapOfShapeListOfShape.hxx>
  30 #include <TopTools_IndexedMapOfShape.hxx>
  31 #include <TopTools_DataMapOfShapeShape.hxx>
  32 #include <TopExp.hxx>
  33 #include <BRepTools.hxx>
  34 #include <BRep_Tool.hxx>
  35 #include <GeomAPI_Shape.h>
  36 #include <GeomAlgoAPI_MakeShape.h>
  37 #include <Config_PropManager.h>
  38 // DEB
  39 //#include <TCollection_AsciiString.hxx>
  40 //#include <TDF_Tool.hxx>
  41 //#define DEB_IMPORT 1
  42
  43 Model_BodyBuilder::Model_BodyBuilder(ModelAPI_Object* theOwner)
  44 : ModelAPI_BodyBuilder(theOwner)
  45 {
  46 }
  47
  48 // Converts evolution of naming shape to selection evelution and back to avoid
  49 // naming support on the disabled results. Deeply in the labels tree, recursively.
  50 static void evolutionToSelectionRec(TDF_Label theLab, const bool theFlag) {
  51   std::list<std::pair<TopoDS_Shape, TopoDS_Shape> > aShapePairs; // to store old and new shapes
  52   Handle(TNaming_NamedShape) aName;
  53   int anEvolution = -1;
  54   if (theLab.FindAttribute(TNaming_NamedShape::GetID(), aName)) {
  55     TNaming_Evolution aNSEvol = aName->Evolution();
  56     if ((aNSEvol == TNaming_SELECTED && theFlag) ||
  57         (aNSEvol != TNaming_SELECTED && !theFlag)) { // nothing to do, it is already correct
  58       return;
  59     }
  60     anEvolution = (int)(aNSEvol);
  61     if (!theFlag) {
  62       Handle(TDataStd_Integer) anAttrEvol;
  63       if (theLab.FindAttribute(TDataStd_Integer::GetID(), anAttrEvol)) {
  64         anEvolution = anAttrEvol->Get();
  65       }
  66     } else {
  67       TDataStd_Integer::Set(theLab, anEvolution);
  68     }
  69
  70     for(TNaming_Iterator anIter(aName); anIter.More(); anIter.Next()) {
  71       aShapePairs.push_back(std::pair<TopoDS_Shape, TopoDS_Shape>
  72         (anIter.OldShape(), anIter.NewShape()));
  73     }
  74   }
  75   // create new
  76   TNaming_Builder aBuilder(theLab);
  77   TNaming_Evolution anEvol = (TNaming_Evolution)(anEvolution);
  78   std::list<std::pair<TopoDS_Shape, TopoDS_Shape> >::iterator aPairsIter = aShapePairs.begin();
  79   for(; aPairsIter != aShapePairs.end(); aPairsIter++) {
  80     if (theFlag) { // disabled => make selection
  81       aBuilder.Select(aPairsIter->second, aPairsIter->first);
  82     } else if (anEvol == TNaming_GENERATED) {
  83       aBuilder.Generated(aPairsIter->first, aPairsIter->second);
  84     } else if (anEvol == TNaming_MODIFY) {
  85       aBuilder.Modify(aPairsIter->first, aPairsIter->second);
  86     } else if (anEvol == TNaming_DELETE) {
  87       aBuilder.Delete(aPairsIter->first);
  88     } else if (anEvol == TNaming_PRIMITIVE) {
  89       aBuilder.Generated(aPairsIter->second);
  90     } else if (anEvol == TNaming_SELECTED) {
  91       aBuilder.Select(aPairsIter->second, aPairsIter->first);
  92     }
  93   }
  94   // recursive call for all sub-labels
  95   TDF_ChildIterator anIter(theLab, Standard_False);
  96   for(; anIter.More(); anIter.Next()) {
  97     evolutionToSelectionRec(anIter.Value(), theFlag);
  98   }
  99 }
 100
 101 void Model_BodyBuilder::evolutionToSelection(const bool theFlag)
 102 {
 103   std::shared_ptr<Model_Data> aData = std::dynamic_pointer_cast<Model_Data>(data());
 104   if (!aData) // unknown case
 105     return;
 106   TDF_Label& aShapeLab = aData->shapeLab();
 107   evolutionToSelectionRec(aShapeLab, theFlag);
 108 }
 109
 110 void Model_BodyBuilder::store(const std::shared_ptr<GeomAPI_Shape>& theShape,
 111                               const bool theIsStoreSameShapes)
 112 {
 113   std::shared_ptr<Model_Data> aData = std::dynamic_pointer_cast<Model_Data>(data());
 114   if (aData) {
 115     TDF_Label& aShapeLab = aData->shapeLab();
 116     // clean builders
 117     clean();
 118     // store the new shape as primitive
 119     TNaming_Builder aBuilder(aShapeLab);
 120     if (!theShape)
 121       return;  // bad shape
 122     TopoDS_Shape aShape = theShape->impl<TopoDS_Shape>();
 123     if (aShape.IsNull())
 124       return;  // null shape inside
 125
 126     if(!theIsStoreSameShapes) {
 127       Handle(TNaming_NamedShape) aNS = TNaming_Tool::NamedShape(aShape, aShapeLab);
 128       if(!aNS.IsNull() && !aNS->IsEmpty()) {
 129         // This shape is already in document, store reference instead of shape;
 130         const TDF_Label aFoundLabel = aNS->Label();
 131         TDF_Reference::Set(aShapeLab, aFoundLabel);
 132         aShapeLab.ForgetAttribute(TNaming_NamedShape::GetID());
 133         return;
 134       }
 135     }
 136
 137     aBuilder.Generated(aShape);
 138     // register name
 139     aShapeLab.ForgetAttribute(TDF_Reference::GetID());
 140     if(!aBuilder.NamedShape()->IsEmpty()) {
 141       Handle(TDataStd_Name) anAttr;
 142       if(aBuilder.NamedShape()->Label().FindAttribute(TDataStd_Name::GetID(),anAttr)) {
 143         std::string aName (TCollection_AsciiString(anAttr->Get()).ToCString());
 144         if(!aName.empty()) {
 145           std::shared_ptr<Model_Document> aDoc = std::dynamic_pointer_cast<Model_Document>(document());
 146           aDoc->addNamingName(aBuilder.NamedShape()->Label(), aName);
 147         }
 148       }
 149     }
 150   }
 151 }
 152
 153 void Model_BodyBuilder::storeGenerated(const std::shared_ptr<GeomAPI_Shape>& theFromShape,
 154   const std::shared_ptr<GeomAPI_Shape>& theToShape)
 155 {
 156   std::shared_ptr<Model_Data> aData = std::dynamic_pointer_cast<Model_Data>(data());
 157   if (aData) {
 158     TDF_Label& aShapeLab = aData->shapeLab();
 159     // clean builders
 160     clean();
 161     // store the new shape as primitive
 162     TNaming_Builder aBuilder(aShapeLab);
 163     if (!theFromShape || !theToShape)
 164       return;  // bad shape
 165     TopoDS_Shape aShapeBasis = theFromShape->impl<TopoDS_Shape>();
 166     if (aShapeBasis.IsNull())
 167       return;  // null shape inside
 168     TopoDS_Shape aShapeNew = theToShape->impl<TopoDS_Shape>();
 169     if (aShapeNew.IsNull())
 170       return;  // null shape inside
 171     aBuilder.Generated(aShapeBasis, aShapeNew);
 172     // register name
 173     if(!aBuilder.NamedShape()->IsEmpty()) {
 174       Handle(TDataStd_Name) anAttr;
 175       if(aBuilder.NamedShape()->Label().FindAttribute(TDataStd_Name::GetID(),anAttr)) {
 176         std::string aName (TCollection_AsciiString(anAttr->Get()).ToCString());
 177         if(!aName.empty()) {
 178           std::shared_ptr<Model_Document> aDoc = std::dynamic_pointer_cast<Model_Document>(document());
 179           aDoc->addNamingName(aBuilder.NamedShape()->Label(), aName);
 180         }
 181       }
 182     }
 183   }
 184 }
 185
 186 void Model_BodyBuilder::storeModified(const std::shared_ptr<GeomAPI_Shape>& theOldShape,
 187   const std::shared_ptr<GeomAPI_Shape>& theNewShape, const int theDecomposeSolidsTag)
 188 {
 189   std::shared_ptr<Model_Data> aData = std::dynamic_pointer_cast<Model_Data>(data());
 190   if (aData) {
 191     TDF_Label& aShapeLab = aData->shapeLab();
 192     // clean builders
 193     clean();
 194     // store the new shape as primitive
 195     TNaming_Builder aBuilder(aShapeLab);
 196     if (!theOldShape || !theNewShape)
 197       return;  // bad shape
 198     TopoDS_Shape aShapeOld = theOldShape->impl<TopoDS_Shape>();
 199     if (aShapeOld.IsNull())
 200       return;  // null shape inside
 201     TopoDS_Shape aShapeNew = theNewShape->impl<TopoDS_Shape>();
 202     if (aShapeNew.IsNull())
 203       return;  // null shape inside
 204     aBuilder.Modify(aShapeOld, aShapeNew);
 205     if (theDecomposeSolidsTag && aShapeNew.ShapeType() == TopAbs_COMPOUND) { // make sub elements as subs
 206
 207       // register name if it is possible
 208       TCollection_AsciiString aName;
 209       if(!aBuilder.NamedShape()->IsEmpty()) {
 210         Handle(TDataStd_Name) anAttr;
 211         if(aBuilder.NamedShape()->Label().FindAttribute(TDataStd_Name::GetID(),anAttr)) {
 212           aName = TCollection_AsciiString(anAttr->Get()).ToCString();
 213         }
 214       }
 215
 216       TopoDS_Iterator aSubIter(aShapeNew);
 217       for(int aTag = theDecomposeSolidsTag; aSubIter.More(); aSubIter.Next()) {
 218         const TopoDS_Shape& aShape = aSubIter.Value();
 219         Handle(TNaming_NamedShape) aNS = TNaming_Tool::NamedShape(aShape, aShapeLab);
 220         if(!aNS.IsNull() && !aNS->IsEmpty()) {
 221           // This shape is already in document, don't add it.
 222           continue;
 223         }
 224         TNaming_Builder aSubBuilder(aShapeLab.FindChild(aTag++));
 225         aSubBuilder.Generated(aSubIter.Value());
 226         if(!aName.IsEmpty()) {
 227           TCollection_AsciiString aShapeType = aShape.ShapeType() == TopAbs_EDGE ? "_Edge_" :
 228                                                aShape.ShapeType() == TopAbs_FACE ? "_Face_" :
 229                                                aShape.ShapeType() == TopAbs_SOLID ? "_Solid_" : "_Shape_";
 230           std::string aSolidName =
 231             (aName + aShapeType + TCollection_AsciiString(aTag - theDecomposeSolidsTag)).ToCString();
 232           std::shared_ptr<Model_Document> aDoc =
 233             std::dynamic_pointer_cast<Model_Document>(document());
 234           aDoc->addNamingName(aSubBuilder.NamedShape()->Label(), aSolidName);
 235           TDataStd_Name::Set(aSubBuilder.NamedShape()->Label(), aSolidName.c_str());
 236         }
 237       }
 238     }
 239   }
 240 }
 241
 242 void  Model_BodyBuilder::storeWithoutNaming(const std::shared_ptr<GeomAPI_Shape>& theShape)
 243 {
 244   std::shared_ptr<Model_Data> aData = std::dynamic_pointer_cast<Model_Data>(data());
 245   if (aData) {
 246     clean();
 247     if (!theShape.get())
 248       return; // bad shape
 249     TopoDS_Shape aShape = theShape->impl<TopoDS_Shape>();
 250     if (aShape.IsNull())
 251       return;  // null shape inside
 252     TNaming_Builder aBuilder(aData->shapeLab());
 253     aBuilder.Select(aShape, aShape);
 254   }
 255 }
 256
 257 void Model_BodyBuilder::clean()
 258 {
 259   std::vector<TNaming_Builder*>::iterator aBuilder = myBuilders.begin();
 260   for(; aBuilder != myBuilders.end(); aBuilder++)
 261     delete *aBuilder;
 262   myBuilders.clear();
 263 }
 264
 265 Model_BodyBuilder::~Model_BodyBuilder()
 266 {
 267   clean();
 268 }
 269
 270 TNaming_Builder* Model_BodyBuilder::builder(const int theTag)
 271 {
 272   if (myBuilders.size() <= (unsigned int)theTag) {
 273     myBuilders.insert(myBuilders.end(), theTag - myBuilders.size() + 1, NULL);
 274   }
 275   if (!myBuilders[theTag]) {
 276     std::shared_ptr<Model_Data> aData = std::dynamic_pointer_cast<Model_Data>(data());
 277     myBuilders[theTag] = new TNaming_Builder(aData->shapeLab().FindChild(theTag));
 278     //TCollection_AsciiString entry;//
 279     //TDF_Tool::Entry(aData->shapeLab().FindChild(theTag), entry);
 280     //cout << "Label = " <<entry.ToCString() <<endl;
 281   }
 282   return myBuilders[theTag];
 283 }
 284
 285 void Model_BodyBuilder::buildName(const int theTag, const std::string& theName)
 286 {
 287   std::string aName = data()->name() + "/" + theName;
 288   std::shared_ptr<Model_Document> aDoc = std::dynamic_pointer_cast<Model_Document>(document());
 289   aDoc->addNamingName(builder(theTag)->NamedShape()->Label(), aName);
 290   TDataStd_Name::Set(builder(theTag)->NamedShape()->Label(),aName.c_str());
 291 }
 292 void Model_BodyBuilder::generated(
 293   const std::shared_ptr<GeomAPI_Shape>& theNewShape, const std::string& theName, const int theTag)
 294 {
 295   TopoDS_Shape aShape = theNewShape->impl<TopoDS_Shape>();
 296   builder(theTag)->Generated(aShape);
 297   if(!theName.empty())
 298     buildName(theTag, theName);
 299 }
 300
 301 void Model_BodyBuilder::generated(const std::shared_ptr<GeomAPI_Shape>& theOldShape,
 302   const std::shared_ptr<GeomAPI_Shape>& theNewShape, const std::string& theName, const int theTag)
 303 {
 304   TopoDS_Shape anOldShape = theOldShape->impl<TopoDS_Shape>();
 305   TopoDS_Shape aNewShape = theNewShape->impl<TopoDS_Shape>();
 306   builder(theTag)->Generated(anOldShape, aNewShape);
 307   if(!theName.empty())
 308     buildName(theTag, theName);
 309   TopAbs_ShapeEnum aGenShapeType = aNewShape.ShapeType();
 310   if(aGenShapeType == TopAbs_WIRE || aGenShapeType == TopAbs_SHELL) {
 311     TopAbs_ShapeEnum anExplodeShapeType = aGenShapeType == TopAbs_WIRE ? TopAbs_EDGE : TopAbs_FACE;
 312     const TDF_Label aLabel = builder(theTag)->NamedShape()->Label();
 313     int aTag = 1;
 314     std::shared_ptr<Model_Document> aDoc = std::dynamic_pointer_cast<Model_Document>(document());
 315     for(TopExp_Explorer anExp(aNewShape, anExplodeShapeType); anExp.More(); anExp.Next()) {
 316       TDF_Label aChildLabel = aLabel.FindChild(aTag);
 317       TNaming_Builder aBuilder(aChildLabel);
 318       aBuilder.Generated(anOldShape, anExp.Current());
 319       TCollection_AsciiString aChildName = TCollection_AsciiString((data()->name() + "/" + theName + "_").c_str()) + aTag;
 320       aDoc->addNamingName(aChildLabel, aChildName.ToCString());
 321       TDataStd_Name::Set(aChildLabel, aChildName.ToCString());
 322       aTag++;
 323     }
 324   }
 325 }
 326
 327
 328 void Model_BodyBuilder::modified(const std::shared_ptr<GeomAPI_Shape>& theOldShape,
 329   const std::shared_ptr<GeomAPI_Shape>& theNewShape, const std::string& theName, const int theTag)
 330 {
 331   TopoDS_Shape anOldShape = theOldShape->impl<TopoDS_Shape>();
 332   TopoDS_Shape aNewShape = theNewShape->impl<TopoDS_Shape>();
 333   builder(theTag)->Modify(anOldShape, aNewShape);
 334   if(!theName.empty())
 335     buildName(theTag, theName);
 336 }
 337
 338 void Model_BodyBuilder::deleted(const std::shared_ptr<GeomAPI_Shape>& theOldShape,
 339   const int theTag)
 340 {
 341   TopoDS_Shape aShape = theOldShape->impl<TopoDS_Shape>();
 342   builder(theTag)->Delete(aShape);
 343 }
 344
 345 void Model_BodyBuilder::loadDeletedShapes (GeomAlgoAPI_MakeShape* theMS,
 346   std::shared_ptr<GeomAPI_Shape>  theShapeIn,
 347   const int  theKindOfShape,
 348   const int  theTag)
 349 {
 350   TopoDS_Shape aShapeIn = theShapeIn->impl<TopoDS_Shape>();
 351   TopTools_MapOfShape aView;
 352   TopExp_Explorer ShapeExplorer (aShapeIn, (TopAbs_ShapeEnum)theKindOfShape);
 353   for (; ShapeExplorer.More(); ShapeExplorer.Next ()) {
 354     const TopoDS_Shape& aRoot = ShapeExplorer.Current ();
 355     if (!aView.Add(aRoot)) continue;
 356     std::shared_ptr<GeomAPI_Shape> aRShape(new GeomAPI_Shape());
 357     aRShape->setImpl((new TopoDS_Shape(aRoot)));
 358     if (theMS->isDeleted (aRShape)) {
 359       builder(theTag)->Delete(aRoot);
 360     }
 361   }
 362 }
 363
 364 void Model_BodyBuilder::loadAndOrientModifiedShapes (
 365   GeomAlgoAPI_MakeShape* theMS,
 366   std::shared_ptr<GeomAPI_Shape>  theShapeIn,
 367   const int  theKindOfShape,
 368   const int  theTag,
 369   const std::string& theName,
 370   GeomAPI_DataMapOfShapeShape& theSubShapes,
 371   const bool theIsStoreSeparate)
 372 {
 373   int anIndex = 1;
 374   int aTag = theTag;
 375   bool isBuilt = !theName.empty();
 376   std::string aName = theName;
 377   std::ostringstream aStream;
 378   GeomShapePtr aResultShape = shape();
 379   TopoDS_Shape aShapeIn = theShapeIn->impl<TopoDS_Shape>();
 380   TopTools_MapOfShape aView;
 381   TopExp_Explorer aShapeExplorer (aShapeIn, (TopAbs_ShapeEnum)theKindOfShape);
 382   for (; aShapeExplorer.More(); aShapeExplorer.Next ()) {
 383     const TopoDS_Shape& aRoot = aShapeExplorer.Current ();
 384     if (!aView.Add(aRoot)) continue;
 385     ListOfShape aList;
 386     std::shared_ptr<GeomAPI_Shape> aRShape(new GeomAPI_Shape());
 387     aRShape->setImpl((new TopoDS_Shape(aRoot)));
 388     theMS->modified(aRShape, aList);
 389     std::list<std::shared_ptr<GeomAPI_Shape> >::const_iterator anIt = aList.begin(), aLast = aList.end();
 390     for (; anIt != aLast; anIt++) {
 391       TopoDS_Shape aNewShape = (*anIt)->impl<TopoDS_Shape>();
 392       if (theSubShapes.isBound(*anIt)) {
 393         std::shared_ptr<GeomAPI_Shape> aMapShape(theSubShapes.find(*anIt));
 394         aNewShape.Orientation(aMapShape->impl<TopoDS_Shape>().Orientation());
 395       }
 396       if(!aRoot.IsSame(aNewShape) && aResultShape->isSubShape(*anIt)) {
 397         builder(aTag)->Modify(aRoot,aNewShape);
 398         if(isBuilt) {
 399           if(theIsStoreSeparate) {
 400             aStream.str(std::string());
 401             aStream.clear();
 402             aStream << theName << "_" << anIndex++;
 403             aName = aStream.str();
 404           }
 405           buildName(aTag, aName);
 406         }
 407         if(theIsStoreSeparate) {
 408           aTag++;
 409         }
 410       }
 411     }
 412   }
 413 }
 414
 415 void Model_BodyBuilder::loadAndOrientGeneratedShapes (
 416   GeomAlgoAPI_MakeShape* theMS,
 417   std::shared_ptr<GeomAPI_Shape>  theShapeIn,
 418   const int  theKindOfShape,
 419   const int  theTag,
 420   const std::string& theName,
 421   GeomAPI_DataMapOfShapeShape& theSubShapes)
 422 {
 423   TopoDS_Shape aShapeIn = theShapeIn->impl<TopoDS_Shape>();
 424   TopTools_MapOfShape aView;
 425   bool isBuilt = !theName.empty();
 426   TopExp_Explorer aShapeExplorer (aShapeIn, (TopAbs_ShapeEnum)theKindOfShape);
 427   for (; aShapeExplorer.More(); aShapeExplorer.Next ()) {
 428     const TopoDS_Shape& aRoot = aShapeExplorer.Current ();
 429     if (!aView.Add(aRoot)) continue;
 430     ListOfShape aList;
 431     std::shared_ptr<GeomAPI_Shape> aRShape(new GeomAPI_Shape());
 432     aRShape->setImpl((new TopoDS_Shape(aRoot)));
 433     theMS->generated(aRShape, aList);
 434     std::list<std::shared_ptr<GeomAPI_Shape> >::const_iterator anIt = aList.begin(), aLast = aList.end();
 435     for (; anIt != aLast; anIt++) {
 436       TopoDS_Shape aNewShape = (*anIt)->impl<TopoDS_Shape>();
 437       if (theSubShapes.isBound(*anIt)) {
 438         std::shared_ptr<GeomAPI_Shape> aMapShape(theSubShapes.find(*anIt));
 439         aNewShape.Orientation(aMapShape->impl<TopoDS_Shape>().Orientation());
 440       }
 441       if (!aRoot.IsSame (aNewShape)) {
 442         builder(theTag)->Generated(aRoot,aNewShape);
 443         if(isBuilt)
 444           buildName(theTag, theName);
 445       }
 446       TopAbs_ShapeEnum aGenShapeType = aNewShape.ShapeType();
 447       if(aGenShapeType == TopAbs_WIRE || aGenShapeType == TopAbs_SHELL) {
 448         TopAbs_ShapeEnum anExplodeShapeType = aGenShapeType == TopAbs_WIRE ? TopAbs_EDGE : TopAbs_FACE;
 449         const TDF_Label aLabel = builder(theTag)->NamedShape()->Label();
 450         int aTag = 1;
 451         std::shared_ptr<Model_Document> aDoc = std::dynamic_pointer_cast<Model_Document>(document());
 452         for(TopExp_Explorer anExp(aNewShape, anExplodeShapeType); anExp.More(); anExp.Next()) {
 453           TDF_Label aChildLabel = aLabel.FindChild(aTag);
 454           TNaming_Builder aBuilder(aChildLabel);
 455           aBuilder.Generated(aRoot, anExp.Current());
 456           TCollection_AsciiString aChildName = TCollection_AsciiString((data()->name() + "/" + theName + "_").c_str()) + aTag;
 457           aDoc->addNamingName(aChildLabel, aChildName.ToCString());
 458           TDataStd_Name::Set(aChildLabel, aChildName.ToCString());
 459           aTag++;
 460         }
 461       }
 462     }
 463   }
 464 }
 465
 466 //=======================================================================
 467 int getDangleShapes(const TopoDS_Shape&           theShapeIn,
 468   const TopAbs_ShapeEnum        theGeneratedFrom,
 469   TopTools_DataMapOfShapeShape& theDangles)
 470 {
 471   theDangles.Clear();
 472   TopTools_IndexedDataMapOfShapeListOfShape subShapeAndAncestors;
 473   TopAbs_ShapeEnum GeneratedTo;
 474   if (theGeneratedFrom == TopAbs_FACE) GeneratedTo = TopAbs_EDGE;
 475   else if (theGeneratedFrom == TopAbs_EDGE) GeneratedTo = TopAbs_VERTEX;
 476   else return Standard_False;
 477   TopExp::MapShapesAndAncestors(theShapeIn, GeneratedTo, theGeneratedFrom, subShapeAndAncestors);
 478   for (Standard_Integer i = 1; i <= subShapeAndAncestors.Extent(); i++) {
 479     const TopoDS_Shape& mayBeDangle = subShapeAndAncestors.FindKey(i);
 480     const TopTools_ListOfShape& ancestors = subShapeAndAncestors.FindFromIndex(i);
 481     if (ancestors.Extent() == 1) theDangles.Bind(ancestors.First(), mayBeDangle);
 482   }
 483   return theDangles.Extent();
 484 }
 485
 486 //=======================================================================
 487 void loadGeneratedDangleShapes(
 488   const TopoDS_Shape&      theShapeIn,
 489   const TopAbs_ShapeEnum   theGeneratedFrom,
 490   TNaming_Builder *        theBuilder)
 491 {
 492   TopTools_DataMapOfShapeShape dangles;
 493   if (!getDangleShapes(theShapeIn, theGeneratedFrom, dangles)) return;
 494   TopTools_DataMapIteratorOfDataMapOfShapeShape itr(dangles);
 495   for (; itr.More(); itr.Next())
 496     theBuilder->Generated(itr.Key(), itr.Value());
 497 }
 498
 499 //=======================================================================
 500 void Model_BodyBuilder::loadNextLevels(std::shared_ptr<GeomAPI_Shape> theShape,
 501   const std::string& theName, int&  theTag)
 502 {
 503   if(theShape->isNull()) return;
 504   TopoDS_Shape aShape = theShape->impl<TopoDS_Shape>();
 505   std::string aName;
 506   if (aShape.ShapeType() == TopAbs_SOLID) {
 507     TopExp_Explorer expl(aShape, TopAbs_FACE);
 508     for (; expl.More(); expl.Next()) {
 509       builder(theTag)->Generated(expl.Current());
 510       TCollection_AsciiString aStr(theTag);
 511       aName = theName + aStr.ToCString();
 512       buildName(theTag, aName);
 513       theTag++;
 514     }
 515   }
 516   else if (aShape.ShapeType() == TopAbs_SHELL || aShape.ShapeType() == TopAbs_FACE) {
 517     // load faces and all the free edges
 518     TopTools_IndexedMapOfShape Faces;
 519     TopExp::MapShapes(aShape, TopAbs_FACE, Faces);
 520     if (Faces.Extent() > 1 || (aShape.ShapeType() == TopAbs_SHELL && Faces.Extent() == 1)) {
 521       TopExp_Explorer expl(aShape, TopAbs_FACE);
 522       for (; expl.More(); expl.Next()) {
 523         builder(theTag)->Generated(expl.Current());
 524         TCollection_AsciiString aStr(theTag);
 525         aName = theName + aStr.ToCString();
 526         buildName(theTag, aName);
 527         theTag++;
 528       }
 529     }
 530     TopTools_IndexedDataMapOfShapeListOfShape anEdgeAndNeighbourFaces;
 531     TopExp::MapShapesAndAncestors(aShape, TopAbs_EDGE, TopAbs_FACE, anEdgeAndNeighbourFaces);
 532     for (Standard_Integer i = 1; i <= anEdgeAndNeighbourFaces.Extent(); i++)
 533     {
 534       const TopTools_ListOfShape& aLL = anEdgeAndNeighbourFaces.FindFromIndex(i);
 535       if (aLL.Extent() < 2) {
 536         if (BRep_Tool::Degenerated(TopoDS::Edge(anEdgeAndNeighbourFaces.FindKey(i))))
 537           continue;
 538         builder(theTag)->Generated(anEdgeAndNeighbourFaces.FindKey(i));
 539         TCollection_AsciiString aStr(theTag);
 540         aName = theName + aStr.ToCString();
 541         buildName(theTag, aName);
 542         theTag++;
 543       } else {
 544         TopTools_ListIteratorOfListOfShape anIter(aLL);
 545         const TopoDS_Face& aFace = TopoDS::Face(anIter.Value());
 546         anIter.Next();
 547         if(aFace.IsEqual(anIter.Value())) {
 548           builder(theTag)->Generated(anEdgeAndNeighbourFaces.FindKey(i));
 549           TCollection_AsciiString aStr(theTag);
 550           aName = theName + aStr.ToCString();
 551           buildName(theTag, aName);
 552           theTag++;
 553         }
 554       }
 555     }
 556   } else if (aShape.ShapeType() == TopAbs_WIRE) {
 557     TopTools_IndexedMapOfShape Edges;
 558     BRepTools::Map3DEdges(aShape, Edges);
 559     if (Edges.Extent() == 1) {
 560       builder(++theTag)->Generated(Edges.FindKey(1));
 561       TopExp_Explorer expl(aShape, TopAbs_VERTEX);
 562       for (; expl.More(); expl.Next()) {
 563         builder(theTag)->Generated(expl.Current());
 564         TCollection_AsciiString aStr(theTag);
 565         aName = theName + aStr.ToCString();
 566         buildName(theTag, aName);
 567         theTag++;
 568       }
 569     } else {
 570       TopExp_Explorer expl(aShape, TopAbs_EDGE);
 571       for (; expl.More(); expl.Next()) {
 572         builder(theTag)->Generated(expl.Current());
 573         TCollection_AsciiString aStr(theTag);
 574         aName = theName + aStr.ToCString();
 575         buildName(theTag, aName);
 576         theTag++;
 577       }
 578       // and load generated vertices.
 579       TopTools_DataMapOfShapeShape generated;
 580       if (getDangleShapes(aShape, TopAbs_EDGE, generated))
 581       {
 582         TNaming_Builder* pBuilder = builder(theTag++);
 583         loadGeneratedDangleShapes(aShape, TopAbs_EDGE, pBuilder);
 584       }
 585     }
 586   } else if (aShape.ShapeType() == TopAbs_EDGE) {
 587     TopExp_Explorer expl(aShape, TopAbs_VERTEX);
 588     for (; expl.More(); expl.Next()) {
 589       builder(theTag)->Generated(expl.Current());
 590       TCollection_AsciiString aStr(theTag);
 591       aName = theName + aStr.ToCString();
 592       buildName(theTag, aName);
 593       theTag++;
 594     }
 595   }
 596 }
 597
 598 //=======================================================================
 599 int findAmbiguities(const TopoDS_Shape&           theShapeIn,
 600   TopTools_ListOfShape&   theList)
 601 {
 602   int aNumEdges(0);
 603   theList.Clear();
 604   TopTools_IndexedDataMapOfShapeListOfShape subShapeAndAncestors;
 605   TopAbs_ShapeEnum aTS(TopAbs_EDGE);
 606   TopAbs_ShapeEnum aTA(TopAbs_FACE);
 607   TopTools_MapOfShape aMap1, aMap2; // map1 - for edge ancestors; map2 - for keys => edges
 608   TopTools_ListOfShape aKeyList;
 609   TopExp::MapShapesAndAncestors(theShapeIn, aTS, aTA, subShapeAndAncestors);
 610   for (Standard_Integer i = 1; i <= subShapeAndAncestors.Extent(); i++) {
 611     const TopoDS_Shape& aKeyEdge1 = subShapeAndAncestors.FindKey(i);
 612     const TopTools_ListOfShape& ancestors1 = subShapeAndAncestors.FindFromIndex(i);
 613     aMap1.Clear();
 614     TopTools_ListIteratorOfListOfShape it(ancestors1);
 615     for(;it.More();it.Next()) aMap1.Add(it.Value()); // fill map with key ancestors => aKey1
 616     for (Standard_Integer j = 1; j <= subShapeAndAncestors.Extent(); j++) {
 617       if (i == j) continue;
 618       const TopoDS_Shape& aKeyEdge2 = subShapeAndAncestors.FindKey(j);
 619       const TopTools_ListOfShape& ancestors2 = subShapeAndAncestors.FindFromIndex(j);
 620       if(ancestors1.Extent() == ancestors2.Extent() && ancestors1.Extent() > 1) {
 621         int aNum (ancestors2.Extent());
 622         TopTools_ListIteratorOfListOfShape it(ancestors2);
 623         for(;it.More();it.Next())
 624           if(aMap1.Contains(it.Value())) aNum--;
 625         if(aNum == 0) {
 626           if(aMap2.Add(aKeyEdge1))
 627             aKeyList.Append(aKeyEdge1);
 628           if(aMap2.Add(aKeyEdge2))
 629             aKeyList.Append(aKeyEdge2);
 630         }
 631       }
 632     } // at the end ==> List of edges to be named in addition
 633   }
 634   aNumEdges = aKeyList.Extent();
 635   if(aNumEdges)
 636     theList.Assign(aKeyList);
 637   return aNumEdges;
 638 }
 639
 640 //=======================================================================
 641 void Model_BodyBuilder::loadFirstLevel(
 642   std::shared_ptr<GeomAPI_Shape> theShape, const std::string& theName, int&  theTag)
 643 {
 644   if(theShape->isNull()) return;
 645   TopoDS_Shape aShape = theShape->impl<TopoDS_Shape>();
 646   std::string aName;
 647   if (aShape.ShapeType() == TopAbs_COMPOUND || aShape.ShapeType() == TopAbs_COMPSOLID) {
 648     TopoDS_Iterator itr(aShape);
 649     for (; itr.More(); itr.Next(),theTag++) {
 650       builder(theTag)->Generated(itr.Value());
 651       TCollection_AsciiString aStr(theTag);
 652       aName = theName + aStr.ToCString();
 653       buildName(theTag, aName);
 654       if(!theName.empty()) buildName(theTag, aName);
 655       if (itr.Value().ShapeType() == TopAbs_COMPOUND ||
 656         itr.Value().ShapeType() == TopAbs_COMPSOLID)
 657       {
 658         std::shared_ptr<GeomAPI_Shape> itrShape(new GeomAPI_Shape());
 659         itrShape->setImpl(new TopoDS_Shape(itr.Value()));
 660         loadFirstLevel(itrShape, theName, theTag);
 661       } else {
 662         std::shared_ptr<GeomAPI_Shape> itrShape(new GeomAPI_Shape());
 663         itrShape->setImpl(new TopoDS_Shape(itr.Value()));
 664         loadNextLevels(itrShape, theName, theTag);
 665       }
 666     }
 667   } else {
 668     std::shared_ptr<GeomAPI_Shape> itrShape(new GeomAPI_Shape());
 669     itrShape->setImpl(new TopoDS_Shape(aShape));
 670     loadNextLevels(itrShape, theName, theTag);
 671   }
 672   TopTools_ListOfShape   aList;
 673   if(findAmbiguities(aShape, aList)) {
 674     TopTools_ListIteratorOfListOfShape it(aList);
 675     for (; it.More(); it.Next(),theTag++) {
 676       builder(theTag)->Generated(it.Value());
 677       TCollection_AsciiString aStr(theTag);
 678       aName = theName + aStr.ToCString();
 679       buildName(theTag, aName);
 680     }
 681   }
 682 }
 683
 684 //=======================================================================
 685 void Model_BodyBuilder::loadDisconnectedEdges(
 686   std::shared_ptr<GeomAPI_Shape> theShape, const std::string& theName, int&  theTag)
 687 {
 688   if(theShape->isNull()) return;
 689   TopoDS_Shape aShape = theShape->impl<TopoDS_Shape>();
 690   TopTools_DataMapOfShapeListOfShape edgeNaborFaces;
 691   TopTools_ListOfShape empty;
 692   TopExp_Explorer explF(aShape, TopAbs_FACE);
 693   for (; explF.More(); explF.Next()) {
 694     const TopoDS_Shape& aFace = explF.Current();
 695     TopExp_Explorer explV(aFace, TopAbs_EDGE);
 696     for (; explV.More(); explV.Next()) {
 697       const TopoDS_Shape& anEdge = explV.Current();
 698       if (!edgeNaborFaces.IsBound(anEdge)) edgeNaborFaces.Bind(anEdge, empty);
 699       Standard_Boolean faceIsNew = Standard_True;
 700       TopTools_ListIteratorOfListOfShape itrF(edgeNaborFaces.Find(anEdge));
 701       for (; itrF.More(); itrF.Next()) {
 702         if (itrF.Value().IsSame(aFace)) {
 703           faceIsNew = Standard_False;
 704           break;
 705         }
 706       }
 707       if (faceIsNew)
 708         edgeNaborFaces.ChangeFind(anEdge).Append(aFace);
 709     }
 710   }
 711
 712   /*  TopTools_IndexedDataMapOfShapeListOfShape aDM;
 713   TopExp::MapShapesAndAncestors(aShape, TopAbs_EDGE, TopAbs_FACE, aDM);
 714   for(int i=1; i <= aDM.Extent(); i++) {
 715   if(aDM.FindFromIndex(i).Extent() > 1) continue;
 716   if (BRep_Tool::Degenerated(TopoDS::Edge(aDM.FindKey(i))))
 717   continue;
 718   builder(theTag)->Generated(aDM.FindKey(i));
 719   TCollection_AsciiString aStr(theTag);
 720   std::string aName = theName + aStr.ToCString();
 721   buildName(theTag, aName);
 722   #ifdef DEB_IMPORT
 723   aName +=  + ".brep";
 724   BRepTools::Write(aDM.FindKey(i), aName.c_str());
 725   #endif
 726   theTag++;
 727   }
 728   */
 729   TopTools_MapOfShape anEdgesToDelete;
 730   TopExp_Explorer anEx(aShape,TopAbs_EDGE);
 731   std::string aName;
 732   for(;anEx.More();anEx.Next()) {
 733     Standard_Boolean aC0 = Standard_False;
 734     TopoDS_Shape anEdge1 = anEx.Current();
 735     if (edgeNaborFaces.IsBound(anEdge1)) {
 736       const TopTools_ListOfShape& aList1 = edgeNaborFaces.Find(anEdge1);
 737       if (aList1.Extent()<2) continue;
 738       TopTools_DataMapIteratorOfDataMapOfShapeListOfShape itr(edgeNaborFaces);
 739       for (; itr.More(); itr.Next()) {
 740         TopoDS_Shape anEdge2 = itr.Key();
 741         if(anEdgesToDelete.Contains(anEdge2)) continue;
 742         if (anEdge1.IsSame(anEdge2)) continue;
 743         const TopTools_ListOfShape& aList2 = itr.Value();
 744         // compare lists of the neighbour faces of edge1 and edge2
 745         if (aList1.Extent() == aList2.Extent()) {
 746           Standard_Integer aMatches = 0;
 747           for(TopTools_ListIteratorOfListOfShape aLIter1(aList1);aLIter1.More();aLIter1.Next())
 748             for(TopTools_ListIteratorOfListOfShape aLIter2(aList2);aLIter2.More();aLIter2.Next())
 749               if (aLIter1.Value().IsSame(aLIter2.Value())) aMatches++;
 750           if (aMatches == aList1.Extent()) {
 751             aC0=Standard_True;
 752             builder(theTag)->Generated(anEdge2);
 753             anEdgesToDelete.Add(anEdge2);
 754             TCollection_AsciiString aStr(theTag);
 755             aName = theName + aStr.ToCString();
 756             buildName(theTag, aName);
 757             theTag++;
 758           }
 759         }
 760       }
 761       TopTools_MapIteratorOfMapOfShape itDelete(anEdgesToDelete);
 762       for(;itDelete.More();itDelete.Next())
 763         edgeNaborFaces.UnBind(itDelete.Key());
 764       edgeNaborFaces.UnBind(anEdge1);
 765     }
 766     if (aC0) {
 767       builder(theTag)->Generated(anEdge1);
 768       TCollection_AsciiString aStr(theTag);
 769       aName = theName + aStr.ToCString();
 770       buildName(theTag, aName);
 771       theTag++;
 772     }
 773   }
 774 }
 775
 776 void Model_BodyBuilder::loadDisconnectedVertexes(std::shared_ptr<GeomAPI_Shape> theShape, const std::string& theName, int&  theTag)
 777 {
 778   if(theShape->isNull()) return;
 779   TopoDS_Shape aShape = theShape->impl<TopoDS_Shape>();
 780   TopTools_DataMapOfShapeListOfShape vertexNaborEdges;
 781   TopTools_ListOfShape empty;
 782   TopExp_Explorer explF(aShape, TopAbs_EDGE);
 783   for (; explF.More(); explF.Next()) {
 784     const TopoDS_Shape& anEdge = explF.Current();
 785     TopExp_Explorer explV(anEdge, TopAbs_VERTEX);
 786     for (; explV.More(); explV.Next()) {
 787       const TopoDS_Shape& aVertex = explV.Current();
 788       if (!vertexNaborEdges.IsBound(aVertex)) vertexNaborEdges.Bind(aVertex, empty);
 789       Standard_Boolean faceIsNew = Standard_True;
 790       TopTools_ListIteratorOfListOfShape itrF(vertexNaborEdges.Find(aVertex));
 791       for (; itrF.More(); itrF.Next()) {
 792         if (itrF.Value().IsSame(anEdge)) {
 793           faceIsNew = Standard_False;
 794           break;
 795         }
 796       }
 797       if (faceIsNew) {
 798         vertexNaborEdges.ChangeFind(aVertex).Append(anEdge);
 799       }
 800     }
 801   }
 802   std::string aName;
 803   TopTools_DataMapIteratorOfDataMapOfShapeListOfShape itr(vertexNaborEdges);
 804   for (; itr.More(); itr.Next()) {
 805     const TopTools_ListOfShape& naborEdges = itr.Value();
 806     if (naborEdges.Extent() < 2) {
 807       builder(theTag)->Generated(itr.Key());
 808       TCollection_AsciiString aStr(theTag);
 809       aName = theName + aStr.ToCString();
 810       buildName(theTag, aName);
 811       theTag++;
 812     }
 813   }
 814 }
 815
 816 std::shared_ptr<GeomAPI_Shape> Model_BodyBuilder::shape()
 817 {
 818   std::shared_ptr<Model_Data> aData = std::dynamic_pointer_cast<Model_Data>(data());
 819   if (aData) {
 820     TDF_Label aShapeLab = aData->shapeLab();
 821     Handle(TDF_Reference) aRef;
 822     if (aShapeLab.FindAttribute(TDF_Reference::GetID(), aRef)) {
 823       aShapeLab = aRef->Get();
 824     }
 825     Handle(TNaming_NamedShape) aName;
 826     if (aShapeLab.FindAttribute(TNaming_NamedShape::GetID(), aName)) {
 827       TopoDS_Shape aShape = aName->Get();
 828       if (!aShape.IsNull()) {
 829         std::shared_ptr<GeomAPI_Shape> aRes(new GeomAPI_Shape);
 830         aRes->setImpl(new TopoDS_Shape(aShape));
 831         return aRes;
 832       }
 833     }
 834   }
 835   return std::shared_ptr<GeomAPI_Shape>();
 836 }
 837
 838 bool Model_BodyBuilder::isLatestEqual(const std::shared_ptr<GeomAPI_Shape>& theShape)
 839 {
 840   if (theShape.get()) {
 841     TopoDS_Shape aShape = theShape->impl<TopoDS_Shape>();
 842     std::shared_ptr<Model_Data> aData = std::dynamic_pointer_cast<Model_Data>(data());
 843     if (aData) {
 844       TDF_Label& aShapeLab = aData->shapeLab();
 845       Handle(TNaming_NamedShape) aName;
 846       if (aShapeLab.FindAttribute(TNaming_NamedShape::GetID(), aName)) {
 847         TopoDS_Shape aLatest = TNaming_Tool::CurrentShape(aName);
 848         if (aLatest.IsNull())
 849           return false;
 850         if (aLatest.IsEqual(aShape))
 851           return true;
 852         // check sub-shapes for comp-solids:
 853         for (TopExp_Explorer anExp(aShape, aLatest.ShapeType()); anExp.More(); anExp.Next()) {
 854           if (aLatest.IsEqual(anExp.Current()))
 855             return true;
 856         }
 857       }
 858     }
 859   }
 860   return false;
 861 }