]> SALOME platform Git repositories - modules/shaper.git/blob - src/SketchPlugin/SketchPlugin_SketchEntity.h
Salome HOME
Fixed validators for Boolean and Intersection
[modules/shaper.git] / 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 #define SKETCH_OVERCONSTRAINT_COLOR "0,0,0"
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& AUXILIARY_ID()
38   {
39     static const std::string MY_AUXILIARY_ID("Auxiliary");
40     return MY_AUXILIARY_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   /// Reference to the copy type of the feature
51   inline static const std::string& COPY_ID()
52   {
53     static const std::string MY_COPY_ID("Copy");
54     return MY_COPY_ID;
55   }
56
57   /// Width of the auxiliary line
58   inline static const double SKETCH_LINE_WIDTH_AUXILIARY()
59   {
60     return 2;
61   }
62
63   /// Width of the line
64   inline static const double SKETCH_LINE_WIDTH()
65   {
66     return 3;
67   }
68
69   /// Style of the auxiliary line
70   inline static const int SKETCH_LINE_STYLE_AUXILIARY()
71   {
72     return  3;
73   }
74
75   /// Style of the line
76   inline static const int SKETCH_LINE_STYLE()
77   {
78     return  0;
79   }
80
81   /// Request for initialization of data model of the feature: adding all attributes
82   virtual void initAttributes();
83
84   /// Returns true of the feature is created basing on the external shape of not-this-sketch object
85   virtual bool isExternal() const
86   {
87     AttributeSelectionPtr aAttr = data()->selection(EXTERNAL_ID());
88     if (aAttr)
89       return aAttr->context().get() != NULL && !aAttr->isInvalid();
90     return false;
91   }
92
93   /// Returns true of the feature is a copy of other feature
94   virtual bool isCopy() const
95   {
96     AttributeBooleanPtr anAttr = data()->boolean(COPY_ID());
97     if(anAttr.get()) {
98       return anAttr->value();
99     }
100     return false;
101   }
102
103   /// Customize presentation of the feature
104   virtual bool customisePresentation(ResultPtr theResult, AISObjectPtr thePrs,
105                                      std::shared_ptr<GeomAPI_ICustomPrs> theDefaultPrs)
106   {
107     //bool isCustomized = false;
108     bool isCustomized = theDefaultPrs.get() != NULL &&
109                         theDefaultPrs->customisePresentation(theResult, thePrs, theDefaultPrs);
110     int aShapeType = thePrs->getShapeType();
111     // a compound is processed like the edge because the arc feature uses the compound for presentable AIS
112     if (aShapeType != 6/*an edge*/ && aShapeType != 7/*a vertex*/ && aShapeType != 0/*compound*/)
113       return false;
114
115     // set color from preferences
116     std::vector<int> aColor;
117     std::shared_ptr<ModelAPI_AttributeBoolean> anAuxiliaryAttr =
118                                     data()->boolean(SketchPlugin_SketchEntity::AUXILIARY_ID());
119     bool isConstruction = anAuxiliaryAttr.get() != NULL && anAuxiliaryAttr->value();
120     if (isConstruction) {
121       aColor = Config_PropManager::color("Visualization", "sketch_auxiliary_color",
122                                          SKETCH_AUXILIARY_COLOR);
123     }
124     else if (isExternal()) {
125       aColor = Config_PropManager::color("Visualization", "sketch_external_color",
126                                         SKETCH_EXTERNAL_COLOR);
127     }
128     else {
129       aColor = Config_PropManager::color("Visualization", "sketch_entity_color",
130                                           SKETCH_ENTITY_COLOR);
131     }
132     if (!aColor.empty())
133       isCustomized = thePrs->setColor(aColor[0], aColor[1], aColor[2]) || isCustomized;
134
135     if (aShapeType == 6 || aShapeType == 0) { // if this is an edge or a compound
136       if (isConstruction) {
137         isCustomized = thePrs->setWidth(SKETCH_LINE_WIDTH_AUXILIARY()) || isCustomized;
138         isCustomized = thePrs->setLineStyle(SKETCH_LINE_STYLE_AUXILIARY()) || isCustomized;
139       }
140       else {
141         isCustomized = thePrs->setWidth(SKETCH_LINE_WIDTH()) || isCustomized;
142         isCustomized = thePrs->setLineStyle(SKETCH_LINE_STYLE()) || isCustomized;
143       }
144     }
145     else if (aShapeType == 7) { // otherwise this is a vertex
146       // The width value do not have effect on the point presentation.
147       // It is defined in order to extend selection area of the object.
148       thePrs->setWidth(17);
149     //  thePrs->setPointMarker(1, 1.); // Set point as a '+' symbol
150     }
151     if(isCopy()) {
152       double aWidth = thePrs->width();
153       isCustomized = thePrs->setWidth(aWidth / 2.5) || isCustomized;
154     }
155     return isCustomized;
156   }
157
158 protected:
159   /// initializes mySketch
160   SketchPlugin_SketchEntity();
161
162   /// \brief Initializes attributes of derived class.
163   virtual void initDerivedClassAttributes(){};
164
165 };
166
167 #endif