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