src/SketchPlugin/SketchPlugin_SketchEntity.h

   1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D -->
   2
   3 // File:        SketchPlugin_SketchEntity.h
   4 // Created:     05 Mar 2015
   5 // Author:      Natalia ERMOLAEVA
   6
   7 #ifndef SketchPlugin_SketchEntity_H_
   8 #define SketchPlugin_SketchEntity_H_
   9
  10 #include "SketchPlugin.h"
  11 #include "SketchPlugin_Feature.h"
  12
  13 #include <ModelAPI_CompositeFeature.h>
  14 #include <GeomAPI_Shape.h>
  15 #include <GeomAPI_AISObject.h>
  16 #include <ModelAPI_Document.h>
  17 #include <ModelAPI_AttributeSelection.h>
  18 #include <ModelAPI_AttributeBoolean.h>
  19 #include <GeomAPI_ICustomPrs.h>
  20
  21 #include <Config_PropManager.h>
  22
  23 #define SKETCH_EDGE_COLOR "#ff0000"
  24 #define SKETCH_POINT_COLOR "#ff0000"
  25 #define SKETCH_EXTERNAL_EDGE_COLOR "#00ff00"
  26 #define SKETCH_CONSTRUCTION_COLOR "#000000"
  27
  28 /**\class SketchPlugin_SketchEntity
  29  * \ingroup Plugins
  30  * \brief Sketch Entity for creation of the new feature in PartSet. This is an abstract class to give
  31  * an interface to create the entity features such as line, circle, arc and point.
  32  */
  33 class SketchPlugin_SketchEntity : public SketchPlugin_Feature, public GeomAPI_ICustomPrs
  34 {
  35  public:
  36   /// Reference to the construction type of the feature
  37   inline static const std::string& CONSTRUCTION_ID()
  38   {
  39     static const std::string MY_CONSTRUCTION_ID("Construction");
  40     return MY_CONSTRUCTION_ID;
  41   }
  42
  43   /// Reference to the external edge or vertex as a AttributeSelection
  44   inline static const std::string& EXTERNAL_ID()
  45   {
  46     static const std::string MY_EXTERNAL_ID("External");
  47     return MY_EXTERNAL_ID;
  48   }
  49
  50   /// Request for initialization of data model of the feature: adding all attributes
  51   virtual void initAttributes();
  52
  53   /// Returns true of the feature is created basing on the external shape of not-this-sketch object
  54   virtual bool isExternal() const
  55   {
  56     AttributeSelectionPtr aAttr = data()->selection(EXTERNAL_ID());
  57     if (aAttr)
  58       return aAttr->context().get() != NULL;
  59     return false;
  60   }
  61
  62   /// Customize presentation of the feature
  63   virtual bool customisePresentation(ResultPtr theResult, AISObjectPtr thePrs,
  64                                      std::shared_ptr<GeomAPI_ICustomPrs> theDefaultPrs)
  65   {
  66     bool isCustomized = false;
  67     int aShapeType = thePrs->getShapeType();
  68     if (aShapeType != 6/*an edge*/ && aShapeType != 7/*a vertex*/)
  69       return false;
  70
  71     std::vector<int> aColor;
  72     std::shared_ptr<ModelAPI_AttributeBoolean> aConstructionAttr =
  73                                     data()->boolean(SketchPlugin_SketchEntity::CONSTRUCTION_ID());
  74     bool isConstruction = aConstructionAttr.get() != NULL && aConstructionAttr->value();
  75     if (aShapeType == 6) { // if this is an edge
  76       if (isConstruction) {
  77         isCustomized = thePrs->setWidth(1) || isCustomized;
  78         isCustomized = thePrs->setLineStyle(3) || isCustomized;
  79
  80         aColor = Config_PropManager::color("Visualization", "sketch_construction_color",
  81                                           SKETCH_CONSTRUCTION_COLOR);
  82       }
  83       else {
  84         isCustomized = thePrs->setWidth(3) || isCustomized;
  85         isCustomized = thePrs->setLineStyle(0) || isCustomized;
  86
  87         if (isExternal()) {
  88           // Set color from preferences
  89           aColor = Config_PropManager::color("Visualization", "sketch_external_color",
  90                                             SKETCH_EXTERNAL_EDGE_COLOR);
  91         }
  92         else {
  93           // Set color from preferences
  94           aColor = Config_PropManager::color("Visualization", "sketch_edge_color",
  95                                              SKETCH_EDGE_COLOR);
  96         }
  97       }
  98     }
  99     else if (aShapeType == 7) { // otherwise this is a vertex
 100       //  thePrs->setPointMarker(6, 2.);
 101       aColor = Config_PropManager::color("Visualization", "sketch_point_color",
 102                                         SKETCH_POINT_COLOR);
 103     }
 104
 105     if (!aColor.empty()) {
 106       isCustomized = thePrs->setColor(aColor[0], aColor[1], aColor[2]) || isCustomized;
 107     }
 108     return isCustomized;
 109   }
 110
 111 protected:
 112   /// initializes mySketch
 113   SketchPlugin_SketchEntity();
 114 };
 115
 116 #endif