src/BuildPlugin/BuildPlugin_Wire.cpp

   1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D
   2
   3 // File:        BuildPlugin_Wire.cpp
   4 // Created:     14 April 2016
   5 // Author:      Dmitry Bobylev
   6
   7 #include "BuildPlugin_Wire.h"
   8
   9 #include <ModelAPI_AttributeSelectionList.h>
  10 #include <ModelAPI_ResultBody.h>
  11 #include <ModelAPI_ResultConstruction.h>
  12
  13 #include <Events_Error.h>
  14
  15 #include <GeomAPI_DataMapOfShapeShape.h>
  16 #include <GeomAPI_PlanarEdges.h>
  17 #include <GeomAPI_ShapeExplorer.h>
  18
  19 #include <GeomAlgoAPI_ShapeTools.h>
  20 #include <GeomAlgoAPI_WireBuilder.h>
  21
  22 #include <algorithm>
  23
  24 //=================================================================================================
  25 BuildPlugin_Wire::BuildPlugin_Wire()
  26 {
  27 }
  28
  29 //=================================================================================================
  30 void BuildPlugin_Wire::initAttributes()
  31 {
  32   data()->addAttribute(BASE_OBJECTS_ID(), ModelAPI_AttributeSelectionList::typeId());
  33 }
  34
  35 //=================================================================================================
  36 void BuildPlugin_Wire::execute()
  37 {
  38   // Get base objects list.
  39   AttributeSelectionListPtr aSelectionList = selectionList(BASE_OBJECTS_ID());
  40   if(!aSelectionList.get()) {
  41     setError("Error: Could not get selection list.");
  42     return;
  43   }
  44   if(aSelectionList->size() == 0) {
  45     setError("Error: Empty selection list.");
  46     return;
  47   }
  48
  49   // Collect base shapes.
  50   ListOfShape aListOfShapes;
  51   for(int anIndex = 0; anIndex < aSelectionList->size(); ++anIndex) {
  52     AttributeSelectionPtr aSelection = aSelectionList->value(anIndex);
  53     GeomShapePtr aShape = aSelection->value();
  54     if(!aShape.get()) {
  55       setError("Error: Empty shape selected.");
  56       return;
  57     }
  58
  59     if(aShape->shapeType() != GeomAPI_Shape::EDGE && aShape->shapeType() != GeomAPI_Shape::WIRE) {
  60       setError("Error: Selected shape has wrong type. Only edges and wires acceptable.");
  61       return;
  62     }
  63
  64     aListOfShapes.push_back(aShape);
  65   }
  66
  67   // Create wire.
  68   GeomShapePtr aWire = GeomAlgoAPI_WireBuilder::wire(aListOfShapes);
  69   if(!aWire.get()) {
  70     setError("Error: Result wire empty. Probably it has disconnected edges or non-manifold.");
  71     return;
  72   }
  73
  74   // Store result.
  75   ResultBodyPtr aResultBody = document()->createBody(data());
  76   aResultBody->store(aWire);
  77   setResult(aResultBody);
  78 }
  79
  80 //=================================================================================================
  81 bool BuildPlugin_Wire::customAction(const std::string& theActionId)
  82 {
  83   if(theActionId == "add_contour") {
  84     return addContour();
  85   } else {
  86     Events_Error::send("Error: Feature \"" + getKind() + "\" does not support action \"" + theActionId + "\".");
  87   }
  88
  89   return false;
  90 }
  91
  92 //=================================================================================================
  93 bool BuildPlugin_Wire::addContour()
  94 {
  95   // Get base objects list.
  96   AttributeSelectionListPtr aSelectionList = selectionList(BASE_OBJECTS_ID());
  97   if(aSelectionList->size() == 0) {
  98     Events_Error::send("Error: Empty selection list.");
  99     return false;
 100   }
 101
 102   // Collect attributes to check.
 103   ListOfShape anAddedEdges;
 104   std::list<AttributeSelectionPtr> anAttributesToCheck;
 105   for(int anIndex = 0; anIndex < aSelectionList->size(); ++anIndex) {
 106     AttributeSelectionPtr aSelection = aSelectionList->value(anIndex);
 107     GeomShapePtr anEdgeInList = aSelection->value();
 108     if(!anEdgeInList.get()) {
 109       continue;
 110     }
 111
 112     // Check that it is edge.
 113     if(anEdgeInList->shapeType() != GeomAPI_Shape::EDGE) {
 114       continue;
 115     }
 116
 117     // Check that it is edge on sketch.
 118     ResultPtr aContext = aSelection->context();
 119     ResultConstructionPtr aConstruction = std::dynamic_pointer_cast<ModelAPI_ResultConstruction>(aContext);
 120     if(!aConstruction.get()) {
 121       continue;
 122     }
 123     GeomShapePtr aContextShape = aConstruction->shape();
 124     std::shared_ptr<GeomAPI_PlanarEdges> aPlanarEdges = std::dynamic_pointer_cast<GeomAPI_PlanarEdges>(aContextShape);
 125     if(!aPlanarEdges.get()) {
 126       continue;
 127     }
 128
 129     // Check that sketch have faces.
 130     if(aConstruction->facesNum() == 0) {
 131       continue;
 132     }
 133
 134     anAddedEdges.push_back(anEdgeInList);
 135     anAttributesToCheck.push_back(aSelection);
 136   }
 137
 138   // Check if edges have contours.
 139   bool isAnyContourAdded = false;
 140   for(std::list<AttributeSelectionPtr>::const_iterator aListIt = anAttributesToCheck.cbegin();
 141       aListIt != anAttributesToCheck.cend();
 142       ++aListIt) {
 143     AttributeSelectionPtr aSelection = *aListIt;
 144     std::shared_ptr<GeomAPI_Edge> anEdgeInList(new GeomAPI_Edge(aSelection->value()));
 145
 146     ListOfShape::const_iterator anEdgesIt = anAddedEdges.cbegin();
 147     for(; anEdgesIt != anAddedEdges.cend(); ++anEdgesIt) {
 148       if(anEdgeInList->isEqual(*anEdgesIt)) {
 149         break;
 150       }
 151     }
 152
 153     ResultConstructionPtr aConstruction = std::dynamic_pointer_cast<ModelAPI_ResultConstruction>(aSelection->context());
 154     std::shared_ptr<GeomAPI_PlanarEdges> aPlanarEdges = std::dynamic_pointer_cast<GeomAPI_PlanarEdges>(aConstruction->shape());
 155
 156     // Iterate on faces and add face with this edge.
 157     std::shared_ptr<GeomAPI_Face> aFoundFace;
 158     for(int anIndex = 0; anIndex < aConstruction->facesNum(); ++anIndex) {
 159       std::shared_ptr<GeomAPI_Face> aFace = aConstruction->face(anIndex);
 160       for(GeomAPI_ShapeExplorer anExp(aFace, GeomAPI_Shape::EDGE); anExp.more(); anExp.next()) {
 161         std::shared_ptr<GeomAPI_Edge> anEdgeOnFace(new GeomAPI_Edge(anExp.current()));
 162         if(anEdgeInList->isEqual(anEdgeOnFace)) {
 163           aFoundFace = aFace;
 164           break;
 165         }
 166       }
 167
 168       if(aFoundFace.get()) {
 169         break;
 170       }
 171     }
 172
 173     // If face with the same edge found. Add all other edges to list.
 174     if(aFoundFace.get()) {
 175       for(GeomAPI_ShapeExplorer anExp(aFoundFace, GeomAPI_Shape::EDGE); anExp.more(); anExp.next()) {
 176         std::shared_ptr<GeomAPI_Edge> anEdgeOnFace(new GeomAPI_Edge(anExp.current()));
 177         anEdgesIt = anAddedEdges.cbegin();
 178         for(; anEdgesIt != anAddedEdges.cend(); ++anEdgesIt) {
 179           if(anEdgeOnFace->isEqual(*anEdgesIt)) {
 180             break;
 181           }
 182         }
 183         if(anEdgesIt == anAddedEdges.cend()) {
 184           isAnyContourAdded = true;
 185           anAddedEdges.push_back(anEdgeOnFace);
 186           aSelectionList->append(aConstruction, anEdgeOnFace);
 187         }
 188       }
 189     }
 190   }
 191
 192   if(!isAnyContourAdded) {
 193     Events_Error::send("Error: Contours already closed or no contours found for selected edges.");
 194     return false;
 195   }
 196
 197   return true;
 198 }