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_ENTITY_COLOR "225,0,0"
  24 #define SKETCH_EXTERNAL_COLOR "170,0,225"
  25 #define SKETCH_AUXILIARY_COLOR "0,85,0"
  26
  27 /**\class SketchPlugin_SketchEntity
  28  * \ingroup Plugins
  29  * \brief Sketch Entity for creation of the new feature in PartSet. This is an abstract class to give
  30  * an interface to create the entity features such as line, circle, arc and point.
  31  */
  32 class SketchPlugin_SketchEntity : public SketchPlugin_Feature, public GeomAPI_ICustomPrs
  33 {
  34  public:
  35   /// Reference to the construction type of the feature
  36   inline static const std::string& AUXILIARY_ID()
  37   {
  38     static const std::string MY_AUXILIARY_ID("Auxiliary");
  39     return MY_AUXILIARY_ID;
  40   }
  41
  42   /// Reference to the external edge or vertex as a AttributeSelection
  43   inline static const std::string& EXTERNAL_ID()
  44   {
  45     static const std::string MY_EXTERNAL_ID("External");
  46     return MY_EXTERNAL_ID;
  47   }
  48
  49   /// Request for initialization of data model of the feature: adding all attributes
  50   virtual void initAttributes();
  51
  52   /// Returns true of the feature is created basing on the external shape of not-this-sketch object
  53   virtual bool isExternal() const
  54   {
  55     AttributeSelectionPtr aAttr = data()->selection(EXTERNAL_ID());
  56     if (aAttr)
  57       return aAttr->context().get() != NULL;
  58     return false;
  59   }
  60
  61   /// Customize presentation of the feature
  62   virtual bool customisePresentation(ResultPtr theResult, AISObjectPtr thePrs,
  63                                      std::shared_ptr<GeomAPI_ICustomPrs> theDefaultPrs)
  64   {
  65     bool isCustomized = false;
  66     int aShapeType = thePrs->getShapeType();
  67     // a compound is processed like the edge because the arc feature uses the compound for presentable AIS
  68     if (aShapeType != 6/*an edge*/ && aShapeType != 7/*a vertex*/ && aShapeType != 0/*compound*/)
  69       return false;
  70
  71     // set color from preferences
  72     std::vector<int> aColor;
  73     std::shared_ptr<ModelAPI_AttributeBoolean> anAuxiliaryAttr =
  74                                     data()->boolean(SketchPlugin_SketchEntity::AUXILIARY_ID());
  75     bool isConstruction = anAuxiliaryAttr.get() != NULL && anAuxiliaryAttr->value();
  76     if (isConstruction) {
  77       aColor = Config_PropManager::color("Visualization", "sketch_auxiliary_color",
  78                                          SKETCH_AUXILIARY_COLOR);
  79     }
  80     else if (isExternal()) {
  81       aColor = Config_PropManager::color("Visualization", "sketch_external_color",
  82                                         SKETCH_EXTERNAL_COLOR);
  83     }
  84     else {
  85       aColor = Config_PropManager::color("Visualization", "sketch_entity_color",
  86                                           SKETCH_ENTITY_COLOR);
  87     }
  88     if (!aColor.empty())
  89       isCustomized = thePrs->setColor(aColor[0], aColor[1], aColor[2]) || isCustomized;
  90
  91     if (aShapeType == 6 || aShapeType == 0) { // if this is an edge or a compound
  92       if (isConstruction) {
  93         isCustomized = thePrs->setWidth(1) || isCustomized;
  94         isCustomized = thePrs->setLineStyle(3) || isCustomized;
  95       }
  96       else {
  97         isCustomized = thePrs->setWidth(3) || isCustomized;
  98         isCustomized = thePrs->setLineStyle(0) || isCustomized;
  99       }
 100     }
 101     else if (aShapeType == 7) { // otherwise this is a vertex
 102       //  thePrs->setPointMarker(6, 2.);
 103     }
 104     return isCustomized;
 105   }
 106
 107 protected:
 108   /// initializes mySketch
 109   SketchPlugin_SketchEntity();
 110 };
 111
 112 #endif