1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D -->
3 // File: SketchPlugin_SketchEntity.h
4 // Created: 05 Mar 2015
5 // Author: Natalia ERMOLAEVA
7 #ifndef SketchPlugin_SketchEntity_H_
8 #define SketchPlugin_SketchEntity_H_
10 #include "SketchPlugin.h"
11 #include "SketchPlugin_Feature.h"
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>
21 #include <Config_PropManager.h>
23 #define SKETCH_ENTITY_COLOR "225,0,0"
24 #define SKETCH_EXTERNAL_COLOR "170,0,225"
25 #define SKETCH_AUXILIARY_COLOR "0,85,0"
27 /**\class SketchPlugin_SketchEntity
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.
32 class SketchPlugin_SketchEntity : public SketchPlugin_Feature, public GeomAPI_ICustomPrs
35 /// Reference to the construction type of the feature
36 inline static const std::string& AUXILIARY_ID()
38 static const std::string MY_AUXILIARY_ID("Auxiliary");
39 return MY_AUXILIARY_ID;
42 /// Reference to the external edge or vertex as a AttributeSelection
43 inline static const std::string& EXTERNAL_ID()
45 static const std::string MY_EXTERNAL_ID("External");
46 return MY_EXTERNAL_ID;
49 /// Reference to the copy type of the feature
50 inline static const std::string& COPY_ID()
52 static const std::string MY_COPY_ID("Copy");
56 /// Request for initialization of data model of the feature: adding all attributes
57 virtual void initAttributes();
59 /// Returns true of the feature is created basing on the external shape of not-this-sketch object
60 virtual bool isExternal() const
62 AttributeSelectionPtr aAttr = data()->selection(EXTERNAL_ID());
64 return aAttr->context().get() != NULL;
68 /// Returns true of the feature is a copy of other feature
69 virtual bool isCopy() const
71 AttributeBooleanPtr anAttr = data()->boolean(COPY_ID());
73 return anAttr->value();
78 /// Customize presentation of the feature
79 virtual bool customisePresentation(ResultPtr theResult, AISObjectPtr thePrs,
80 std::shared_ptr<GeomAPI_ICustomPrs> theDefaultPrs)
82 bool isCustomized = false;
83 int aShapeType = thePrs->getShapeType();
84 // a compound is processed like the edge because the arc feature uses the compound for presentable AIS
85 if (aShapeType != 6/*an edge*/ && aShapeType != 7/*a vertex*/ && aShapeType != 0/*compound*/)
88 // set color from preferences
89 std::vector<int> aColor;
90 std::shared_ptr<ModelAPI_AttributeBoolean> anAuxiliaryAttr =
91 data()->boolean(SketchPlugin_SketchEntity::AUXILIARY_ID());
92 bool isConstruction = anAuxiliaryAttr.get() != NULL && anAuxiliaryAttr->value();
94 aColor = Config_PropManager::color("Visualization", "sketch_auxiliary_color",
95 SKETCH_AUXILIARY_COLOR);
97 else if (isExternal()) {
98 aColor = Config_PropManager::color("Visualization", "sketch_external_color",
99 SKETCH_EXTERNAL_COLOR);
102 aColor = Config_PropManager::color("Visualization", "sketch_entity_color",
103 SKETCH_ENTITY_COLOR);
106 isCustomized = thePrs->setColor(aColor[0], aColor[1], aColor[2]) || isCustomized;
108 if (aShapeType == 6 || aShapeType == 0) { // if this is an edge or a compound
109 if (isConstruction) {
110 isCustomized = thePrs->setWidth(1) || isCustomized;
111 isCustomized = thePrs->setLineStyle(3) || isCustomized;
114 isCustomized = thePrs->setWidth(3) || isCustomized;
115 isCustomized = thePrs->setLineStyle(0) || isCustomized;
118 else if (aShapeType == 7) { // otherwise this is a vertex
119 // The width value do not have effect on the point presentation.
120 // It is defined in order to extend selection area of the object.
121 thePrs->setWidth(17);
122 // thePrs->setPointMarker(1, 1.); // Set point as a '+' symbol
125 double aWidth = thePrs->width();
126 isCustomized = thePrs->setWidth(aWidth / 2.5) || isCustomized;
132 /// initializes mySketch
133 SketchPlugin_SketchEntity();
135 /// \brief Initializes attributes of derived class.
136 virtual void initDerivedClassAttributes(){};