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_ACTION_ID()) {
  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
 154     // Iterate on wires and add wire with this edge.
 155     std::shared_ptr<GeomAPI_Shape> aFoundWire;
 156     for(int anIndex = 0; anIndex < aConstruction->facesNum(); ++anIndex) {
 157       std::shared_ptr<GeomAPI_Face> aFace = aConstruction->face(anIndex);
 158       for(GeomAPI_ShapeExplorer aWireExp(aFace, GeomAPI_Shape::WIRE); aWireExp.more(); aWireExp.next()) {
 159         GeomShapePtr aWireOnFace = aWireExp.current();
 160         for(GeomAPI_ShapeExplorer anExp(aWireOnFace, 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             aFoundWire = aWireOnFace;
 164             break;
 165           }
 166         }
 167       }
 168       if(aFoundWire.get()) {
 169         break;
 170       }
 171     }
 172
 173     // If wire with the same edge found. Add all other edges to list.
 174     if(aFoundWire.get()) {
 175       for(GeomAPI_ShapeExplorer anExp(aFoundWire, GeomAPI_Shape::EDGE); anExp.more(); anExp.next()) {
 176         std::shared_ptr<GeomAPI_Edge> anEdgeOnFace(new GeomAPI_Edge(anExp.current()));
 177         ListOfShape::const_iterator 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_InfoMessage("BuildPlugin_Wire",
 194       "Error: Contours already closed or no contours found for selected edges.").send();
 195     return false;
 196   }
 197
 198   return true;
 199 }