src/GeomAlgoAPI/GeomAlgoAPI_CompoundBuilder.cpp

   1 // Copyright (C) 2014-2023  CEA, EDF
   2 //
   3 // This library is free software; you can redistribute it and/or
   4 // modify it under the terms of the GNU Lesser General Public
   5 // License as published by the Free Software Foundation; either
   6 // version 2.1 of the License, or (at your option) any later version.
   7 //
   8 // This library is distributed in the hope that it will be useful,
   9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  11 // Lesser General Public License for more details.
  12 //
  13 // You should have received a copy of the GNU Lesser General Public
  14 // License along with this library; if not, write to the Free Software
  15 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  16 //
  17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  18 //
  19
  20 #include <GeomAlgoAPI_CompoundBuilder.h>
  21 #include <BRep_Builder.hxx>
  22 #include <TopoDS_Compound.hxx>
  23 #include <TopTools_IndexedMapOfShape.hxx>
  24 #include <TopExp.hxx>
  25 #include <TopExp_Explorer.hxx>
  26
  27 std::shared_ptr<GeomAPI_Shape> GeomAlgoAPI_CompoundBuilder::compound(
  28     std::list<std::shared_ptr<GeomAPI_Shape> > theShapes)
  29 {
  30   BRep_Builder aBuilder;
  31   TopoDS_Compound aComp;
  32   aBuilder.MakeCompound(aComp);
  33
  34   std::list<std::shared_ptr<GeomAPI_Shape> >::const_iterator anIt = theShapes.begin(), aLast =
  35       theShapes.end();
  36   for (; anIt != aLast; anIt++) {
  37     aBuilder.Add(aComp, (*anIt)->impl<TopoDS_Shape>());
  38   }
  39
  40   std::shared_ptr<GeomAPI_Shape> aRes(new GeomAPI_Shape);
  41   aRes->setImpl(new TopoDS_Shape(aComp));
  42   return aRes;
  43 }
  44
  45 // Returns true if transformations are equal with the given precision
  46 static bool isEqual(const gp_Trsf& theT1, const gp_Trsf& theT2, const double thePrecision)
  47 {
  48   for(int aRow = 1; aRow < 4; aRow++) {
  49     for(int aCol = 1; aCol < 5; aCol++) {
  50       double aDiff = theT1.Value(aRow, aCol) - theT2.Value(aRow, aCol);
  51       if (aDiff < 0) aDiff = -aDiff;
  52       if (aDiff > thePrecision)
  53         return false;
  54     }
  55   }
  56   return true;
  57 }
  58
  59 int GeomAlgoAPI_CompoundBuilder::id(
  60       std::shared_ptr<GeomAPI_Shape> theContext, std::shared_ptr<GeomAPI_Shape> theSub)
  61 {
  62   int anID = 0;
  63   TopoDS_Shape aMainShape = theContext->impl<TopoDS_Shape>();
  64   const TopoDS_Shape& aSubShape = theSub->impl<TopoDS_Shape>();
  65   if (!aMainShape.IsNull() && !aSubShape.IsNull()) {
  66     TopTools_IndexedMapOfShape aSubShapesMap;
  67     TopExp::MapShapes(aMainShape, aSubShapesMap);
  68     anID = aSubShapesMap.FindIndex(aSubShape);
  69     if (anID == 0) { // try to search shape with the same location if TopLoc_Location is different
  70       TopExp_Explorer anExp(aMainShape, aSubShape.ShapeType());
  71       for(; anExp.More(); anExp.Next()) {
  72         if (anExp.Current().TShape() == aSubShape.TShape()) {
  73           const TopLoc_Location aLoc1 = anExp.Current().Location();
  74           if (isEqual(aLoc1.Transformation(), aSubShape.Location().Transformation(), 1.e-7)) {
  75             anID = aSubShapesMap.FindIndex(anExp.Current());
  76             break;
  77           }
  78         }
  79       }
  80     }
  81   }
  82
  83   return anID;
  84 }