src/PartSet/PartSet_ResultSketchPrs.cpp

   1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D
   2
   3 // File:        PartSet_ResultSketchPrs.h
   4 // Created:     25 March 2015
   5 // Author:      Natalia ERMOLAEVA
   6
   7 #include "PartSet_ResultSketchPrs.h"
   8 #include "PartSet_Tools.h"
   9 #include "ModuleBase_Tools.h"
  10
  11 #include <ModelAPI_Events.h>
  12 #include <ModelAPI_Tools.h>
  13 #include <ModelAPI_ResultConstruction.h>
  14 #include <GeomAPI_PlanarEdges.h>
  15
  16 #include <Events_Error.h>
  17 #include <Events_Loop.h>
  18
  19 #include <SketchPlugin_SketchEntity.h>
  20
  21 #include <Config_PropManager.h>
  22
  23 #include <BRep_Builder.hxx>
  24 #include <Prs3d_Drawer.hxx>
  25 #include <Prs3d.hxx>
  26 #include <Prs3d_PointAspect.hxx>
  27 #include <TopoDS_Builder.hxx>
  28 #include <SelectMgr_SelectionManager.hxx>
  29 #include <StdPrs_WFDeflectionShape.hxx>
  30 #include <StdSelect_BRepSelectionTool.hxx>
  31 #include <Graphic3d_AspectLine3d.hxx>
  32 #include <Prs3d_LineAspect.hxx>
  33
  34 #define DEBUG_WIRE
  35
  36 #ifdef DEBUG_WIRE
  37 #include <TopTools_IndexedMapOfShape.hxx>
  38 #include <TopExp.hxx>
  39 #endif
  40
  41 //*******************************************************************************************
  42
  43 IMPLEMENT_STANDARD_HANDLE(PartSet_ResultSketchPrs, ViewerData_AISShape);
  44 IMPLEMENT_STANDARD_RTTIEXT(PartSet_ResultSketchPrs, ViewerData_AISShape);
  45
  46 PartSet_ResultSketchPrs::PartSet_ResultSketchPrs(ResultPtr theResult)
  47   : ViewerData_AISShape(TopoDS_Shape()), myResult(theResult)
  48 {
  49   std::shared_ptr<GeomAPI_Shape> aShapePtr = ModelAPI_Tools::shape(theResult);
  50
  51   TopoDS_Shape aShape = aShapePtr->impl<TopoDS_Shape>();
  52   Set(aShape);
  53   Handle(Prs3d_Drawer) aDrawer = Attributes();
  54   if (aDrawer->HasOwnPointAspect())
  55     aDrawer->PointAspect()->SetTypeOfMarker(Aspect_TOM_PLUS);
  56   else
  57     aDrawer->SetPointAspect(new Prs3d_PointAspect(Aspect_TOM_PLUS, Quantity_NOC_YELLOW, 1.));
  58
  59   // Activate individual repaintng if this is a part of compsolid
  60   ResultCompSolidPtr aCompSolid = ModelAPI_Tools::compSolidOwner(myResult);
  61   SetAutoHilight(aCompSolid.get() == NULL);
  62 }
  63
  64 bool PartSet_ResultSketchPrs::isValidShapeType(const TopAbs_ShapeEnum& theBaseType,
  65                                             const TopAbs_ShapeEnum& theCheckedType)
  66 {
  67   bool aValid = theBaseType == theCheckedType;
  68   if (!aValid) {
  69     // currently this functionality is for all, as we have no separate wire selection mode
  70     // lately it should be corrected to have the following check only for sketch presentations
  71     aValid = theBaseType == TopAbs_FACE && theCheckedType == TopAbs_WIRE;
  72   }
  73   return aValid;
  74 }
  75
  76 void PartSet_ResultSketchPrs::Compute(const Handle(PrsMgr_PresentationManager3d)& thePresentationManager,
  77                                       const Handle(Prs3d_Presentation)& thePresentation,
  78                                       const Standard_Integer theMode)
  79 {
  80   thePresentation->Clear();
  81
  82   TopoDS_Shape aResultShape;
  83   TopoDS_Compound anAuxiliaryCompound;
  84   NCollection_List<TopoDS_Shape> aFaceList;
  85   fillShapes(aResultShape, anAuxiliaryCompound, mySketchFaceList);
  86
  87   bool aReadyToDisplay = !aResultShape.IsNull() || !anAuxiliaryCompound.IsNull();
  88
  89   if (aReadyToDisplay) {
  90     if (!aResultShape.IsNull()) {
  91       myOriginalShape = aResultShape;
  92       if (!myOriginalShape.IsNull())
  93         Set(myOriginalShape);
  94     }
  95     myAuxiliaryCompound = anAuxiliaryCompound;
  96   }
  97
  98   setAuxiliaryPresentationStyle(false);
  99
 100   // change deviation coefficient to provide more precise circle
 101   ModuleBase_Tools::setDefaultDeviationCoefficient(Shape(), Attributes());
 102   AIS_Shape::Compute(thePresentationManager, thePresentation, theMode);
 103
 104   if (!myAuxiliaryCompound.IsNull()) {
 105     setAuxiliaryPresentationStyle(true);
 106
 107     Handle(Prs3d_Drawer) aDrawer = Attributes();
 108     StdPrs_WFDeflectionShape::Add(thePresentation, myAuxiliaryCompound, aDrawer);
 109   }
 110
 111   if (!aReadyToDisplay) {
 112     Events_Error::throwException("An empty AIS presentation: PartSet_ResultSketchPrs");
 113     static const Events_ID anEvent = Events_Loop::eventByName(EVENT_EMPTY_AIS_PRESENTATION);
 114     ModelAPI_EventCreator::get()->sendUpdated(myResult, anEvent);
 115   }
 116 }
 117
 118 #ifdef DEBUG_WIRE
 119 void debugInfo(const TopoDS_Shape& theShape, const TopAbs_ShapeEnum theType)
 120 {
 121   TopTools_IndexedMapOfShape aSubShapes;
 122   TopExp::MapShapes (theShape, theType, aSubShapes);
 123
 124   Standard_Boolean isComesFromDecomposition = !((aSubShapes.Extent() == 1) && (theShape == aSubShapes (1)));
 125   int anExtent = aSubShapes.Extent();
 126   for (Standard_Integer aShIndex = 1; aShIndex <= aSubShapes.Extent(); ++aShIndex)
 127   {
 128     const TopoDS_Shape& aSubShape = aSubShapes (aShIndex);
 129     int aValue = 0;
 130   }
 131 }
 132 #endif
 133
 134 void PartSet_ResultSketchPrs::ComputeSelection(const Handle(SelectMgr_Selection)& aSelection,
 135                                             const Standard_Integer aMode)
 136 {
 137   if (aMode > 8)
 138     // In order to avoid using custom selection modes
 139     return;
 140
 141   if (aMode == AIS_Shape::SelectionMode(TopAbs_FACE)) {
 142 #ifdef DEBUG_WIRE
 143     const TopoDS_Shape& aShape = Shape();
 144     debugInfo(aShape, TopAbs_VERTEX); // 24
 145     debugInfo(aShape, TopAbs_EDGE); // 12
 146     debugInfo(aShape, TopAbs_WIRE); // 0
 147     debugInfo(aShape, TopAbs_FACE); // 0
 148 #endif
 149     BRep_Builder aBuilder;
 150     TopoDS_Compound aComp;
 151     aBuilder.MakeCompound(aComp);
 152     aBuilder.Add(aComp, myOriginalShape);
 153
 154     for(NCollection_List<TopoDS_Shape>::Iterator anIt(mySketchFaceList); anIt.More(); anIt.Next()) {
 155       const TopoDS_Shape& aFace = anIt.Value();
 156       aBuilder.Add(aComp, aFace);
 157       // for sketch presentation in the face mode wires should be selectable also
 158       // accoring to #1343 Improvement of Extrusion and Revolution operations
 159       appendShapeSelection(aSelection, aFace, TopAbs_WIRE);
 160     }
 161 #ifdef DEBUG_WIRE
 162     debugInfo(aComp, TopAbs_VERTEX); // 24
 163     debugInfo(aComp, TopAbs_EDGE); // 12
 164     debugInfo(aComp, TopAbs_WIRE); // 4
 165     debugInfo(aComp, TopAbs_FACE); // 2
 166 #endif
 167     Set(aComp);
 168   } else
 169     Set(myOriginalShape);
 170
 171   // append auxiliary compound to selection of edges/vertices
 172   if (aMode == AIS_Shape::SelectionMode(TopAbs_EDGE) ||
 173       aMode == AIS_Shape::SelectionMode(TopAbs_VERTEX)) {
 174
 175         bool isVertex = aMode == AIS_Shape::SelectionMode(TopAbs_VERTEX);
 176     appendShapeSelection(aSelection, myAuxiliaryCompound, isVertex ? TopAbs_VERTEX : TopAbs_EDGE);
 177   }
 178
 179   AIS_Shape::ComputeSelection(aSelection, aMode);
 180 }
 181
 182 void PartSet_ResultSketchPrs::appendShapeSelection(const Handle(SelectMgr_Selection)& theSelection,
 183                                                    const TopoDS_Shape& theShape,
 184                                                    const TopAbs_ShapeEnum& theTypeOfSelection)
 185 {
 186   // POP protection against crash in low layers
 187   Standard_Real aDeflection = Prs3d::GetDeflection(theShape, myDrawer);
 188   try {
 189     StdSelect_BRepSelectionTool::Load(theSelection,
 190                                       this,
 191                                       theShape,
 192                                       theTypeOfSelection,
 193                                       aDeflection,
 194                                       myDrawer->HLRAngle(),
 195                                       myDrawer->IsAutoTriangulation());
 196   } catch ( Standard_Failure ) {
 197   }
 198 }
 199
 200 void PartSet_ResultSketchPrs::setAuxiliaryPresentationStyle(const bool isAuxiliary)
 201 {
 202   std::vector<int> aColor;
 203   Standard_Real aWidth;
 204   Standard_Integer aLineStyle;
 205
 206   if (!isAuxiliary) {
 207     aColor = Config_PropManager::color("Visualization", "result_construction_color",
 208                                         ModelAPI_ResultConstruction::DEFAULT_COLOR());
 209     aWidth = PartSet_Tools::getAISDefaultWidth();// default width value
 210     aLineStyle = SketchPlugin_SketchEntity::SKETCH_LINE_STYLE();
 211   }
 212   else {
 213     aColor = Config_PropManager::color("Visualization", "sketch_auxiliary_color",
 214                                          SKETCH_AUXILIARY_COLOR);
 215     aWidth = SketchPlugin_SketchEntity::SKETCH_LINE_WIDTH_AUXILIARY();
 216     aLineStyle = SketchPlugin_SketchEntity::SKETCH_LINE_STYLE_AUXILIARY();
 217   }
 218
 219   Handle(Prs3d_Drawer) aDrawer = Attributes();
 220   setColor(aDrawer, Quantity_Color(aColor[0] / 255., aColor[1] / 255., aColor[2] / 255.,
 221            Quantity_TOC_RGB));
 222
 223   setWidth(aDrawer, aWidth);
 224   // set line style
 225   Handle(Prs3d_LineAspect) aLineAspect;
 226
 227   Aspect_TypeOfLine aType = (Aspect_TypeOfLine)aLineStyle;
 228   if (aDrawer->HasOwnLineAspect()) {
 229     aLineAspect = aDrawer->LineAspect();
 230   }
 231   if (aDrawer->HasOwnWireAspect()) {
 232     aLineAspect = aDrawer->WireAspect();
 233   }
 234   Quantity_Color aCurrentColor;
 235   Aspect_TypeOfLine aPrevLineType;
 236   Standard_Real aCurrentWidth;
 237   aLineAspect->Aspect()->Values(aCurrentColor, aPrevLineType, aCurrentWidth);
 238   bool isChangedLineType = aType != aPrevLineType;
 239   if (isChangedLineType) {
 240     aLineAspect->SetTypeOfLine(aType);
 241   }
 242 }
 243
 244 void PartSet_ResultSketchPrs::fillShapes(TopoDS_Shape& theResultShape,
 245                                          TopoDS_Compound& theAuxiliaryCompound,
 246                                          NCollection_List<TopoDS_Shape>& theFaceList)
 247 {
 248   //if (!aResultShape.IsNull() || !anAuxiliaryCompound.IsNull())
 249   std::shared_ptr<GeomAPI_Shape> aShapePtr = ModelAPI_Tools::shape(myResult);
 250   if (!aShapePtr)
 251     return;
 252
 253   theFaceList.Clear();
 254   ResultConstructionPtr aConstruction =
 255     std::dynamic_pointer_cast<ModelAPI_ResultConstruction>(myResult);
 256   if (aConstruction.get()) {
 257     int aFacesNum = aConstruction->facesNum();
 258     for(int aFaceIndex = 0; aFaceIndex < aFacesNum; aFaceIndex++) {
 259       std::shared_ptr<GeomAPI_Face> aFaceShape = aConstruction->face(aFaceIndex);
 260       if (aFaceShape.get()) {
 261         TopoDS_Shape aFace = (aFaceShape)->impl<TopoDS_Shape>();
 262         theFaceList.Append(aFace);
 263       }
 264     }
 265   }
 266   theResultShape = aShapePtr->impl<TopoDS_Shape>();
 267
 268   /// find auxiliary shapes
 269   FeaturePtr aResultFeature = ModelAPI_Feature::feature(myResult);
 270   CompositeFeaturePtr aSketchFeature = std::dynamic_pointer_cast<ModelAPI_CompositeFeature>
 271                                                                           (aResultFeature);
 272   std::list<ResultPtr> anAuxiliaryResults;
 273   for (int i = 0; i < aSketchFeature->numberOfSubs(); i++) {
 274     FeaturePtr aFeature = aSketchFeature->subFeature(i);
 275     if (PartSet_Tools::isAuxiliarySketchEntity(aFeature)) {
 276       std::list<ResultPtr> aResults = aFeature->results();
 277       std::list<ResultPtr>::const_iterator aIt;
 278       for (aIt = aResults.cbegin(); aIt != aResults.cend(); ++aIt) {
 279         ResultPtr aResult = *aIt;
 280         if (aResult.get() && aResult->shape().get())
 281           anAuxiliaryResults.push_back(aResult);
 282       }
 283     }
 284     else {
 285     /// append not-edges shapes, e.g. center of a circle, an arc, a point feature
 286       const std::list<std::shared_ptr<ModelAPI_Result> >& aRes = aFeature->results();
 287       std::list<std::shared_ptr<ModelAPI_Result> >::const_iterator aResIter = aRes.cbegin();
 288       for (; aResIter != aRes.cend(); aResIter++) {
 289         std::shared_ptr<ModelAPI_ResultConstruction> aConstr = std::dynamic_pointer_cast<
 290             ModelAPI_ResultConstruction>(*aResIter);
 291         if (aConstr) {
 292           std::shared_ptr<GeomAPI_Shape> aGeomShape = aConstr->shape();
 293           if (aGeomShape.get()) {
 294             const TopoDS_Shape& aShape = aGeomShape->impl<TopoDS_Shape>();
 295             if (aShape.ShapeType() != TopAbs_EDGE)
 296               anAuxiliaryResults.push_back(aConstr);
 297           }
 298         }
 299       }
 300     }
 301   }
 302
 303   if (anAuxiliaryResults.size() > 0) {
 304     BRep_Builder aBuilder;
 305     //TopoDS_Compound aComp;
 306     aBuilder.MakeCompound(theAuxiliaryCompound);
 307     std::list<ResultPtr>::const_iterator anIt = anAuxiliaryResults.begin(),
 308                                          aLast = anAuxiliaryResults.end();
 309     for (; anIt != aLast; anIt++) {
 310       ResultPtr aResult = *anIt;
 311       if (aResult.get()) {
 312         GeomShapePtr aGeomShape = aResult->shape();
 313         if (aGeomShape.get()) {
 314           const TopoDS_Shape& aShape = aGeomShape->impl<TopoDS_Shape>();
 315           if (!aShape.IsNull())
 316             aBuilder.Add(theAuxiliaryCompound, aShape);
 317         }
 318       }
 319     }
 320   }
 321 }