Salome HOME
Draft of transparency
[modules/shaper.git] / src / ModelHighAPI / ModelHighAPI_Selection.cpp
1 // Copyright (C) 2014-2017  CEA/DEN, EDF R&D
2 //
3 // This library is free software; you can redistribute it and/or
4 // modify it under the terms of the GNU Lesser General Public
5 // License as published by the Free Software Foundation; either
6 // version 2.1 of the License, or (at your option) any later version.
7 //
8 // This library is distributed in the hope that it will be useful,
9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
11 // Lesser General Public License for more details.
12 //
13 // You should have received a copy of the GNU Lesser General Public
14 // License along with this library; if not, write to the Free Software
15 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
16 //
17 // See http://www.salome-platform.org/ or
18 // email : webmaster.salome@opencascade.com<mailto:webmaster.salome@opencascade.com>
19 //
20
21 #include "ModelHighAPI_Selection.h"
22
23 #include <ModelAPI_AttributeDouble.h>
24 #include <ModelAPI_AttributeIntArray.h>
25 #include <ModelAPI_AttributeSelection.h>
26 #include <ModelAPI_AttributeSelectionList.h>
27 #include <ModelAPI_Feature.h>
28 #include <ModelAPI_ResultCompSolid.h>
29 //--------------------------------------------------------------------------------------
30
31 //--------------------------------------------------------------------------------------
32 ModelHighAPI_Selection::ModelHighAPI_Selection()
33 : myVariantType(VT_Empty)
34 {
35 }
36
37 ModelHighAPI_Selection::ModelHighAPI_Selection(const std::shared_ptr<ModelAPI_Result>& theContext,
38                                                const std::shared_ptr<GeomAPI_Shape>& theSubShape)
39 : myVariantType(VT_ResultSubShapePair)
40 , myResultSubShapePair(theContext, theSubShape)
41 {
42 }
43
44 ModelHighAPI_Selection::ModelHighAPI_Selection(const std::string& theType,
45                                                const std::string& theSubShapeName)
46 : myVariantType(VT_TypeSubShapeNamePair)
47 , myTypeSubShapeNamePair(theType, theSubShapeName)
48 {
49 }
50
51 ModelHighAPI_Selection::~ModelHighAPI_Selection()
52 {
53 }
54
55 //--------------------------------------------------------------------------------------
56 void ModelHighAPI_Selection::fillAttribute(
57     const std::shared_ptr<ModelAPI_AttributeSelection> & theAttribute) const
58 {
59   switch(myVariantType) {
60     case VT_Empty: return;
61     case VT_ResultSubShapePair:
62       theAttribute->setValue(myResultSubShapePair.first, myResultSubShapePair.second);
63       return;
64     case VT_TypeSubShapeNamePair:
65       theAttribute->selectSubShape(myTypeSubShapeNamePair.first, myTypeSubShapeNamePair.second);
66       if(theAttribute->isInvalid()) {
67         FeaturePtr aFeature = ModelAPI_Feature::feature(theAttribute->owner());
68         aFeature->setError(
69           std::string("Error: attribute \"") + theAttribute->id() + std::string("\" is invalid."));
70       }
71       return;
72   }
73 }
74
75 //--------------------------------------------------------------------------------------
76 void ModelHighAPI_Selection::appendToList(
77     const std::shared_ptr<ModelAPI_AttributeSelectionList> & theAttribute) const
78 {
79   switch(myVariantType) {
80     case VT_Empty: return;
81     case VT_ResultSubShapePair:
82       theAttribute->append(myResultSubShapePair.first, myResultSubShapePair.second);
83       return;
84     case VT_TypeSubShapeNamePair:
85       // Note: the reverse order (first - type, second - sub-shape name)
86       theAttribute->append(myTypeSubShapeNamePair.second, myTypeSubShapeNamePair.first);
87       return;
88   }
89 }
90
91 //==================================================================================================
92 ModelHighAPI_Selection::VariantType ModelHighAPI_Selection::variantType() const
93 {
94   return myVariantType;
95 }
96
97 //==================================================================================================
98 ResultSubShapePair ModelHighAPI_Selection::resultSubShapePair() const
99 {
100   return myResultSubShapePair;
101 }
102
103 //==================================================================================================
104 TypeSubShapeNamePair ModelHighAPI_Selection::typeSubShapeNamePair() const
105 {
106   return myTypeSubShapeNamePair;
107 }
108
109 //==================================================================================================
110 std::string ModelHighAPI_Selection::shapeType() const
111 {
112   switch(myVariantType) {
113   case VT_ResultSubShapePair:
114     return myResultSubShapePair.second.get() ? myResultSubShapePair.second->shapeTypeStr() :
115                                                myResultSubShapePair.first->shape()->shapeTypeStr();
116   case VT_TypeSubShapeNamePair: return myTypeSubShapeNamePair.first;
117   }
118
119   return "SHAPE";
120 }
121
122 //==================================================================================================
123 void ModelHighAPI_Selection::setName(const std::string& theName)
124 {
125   if (myVariantType == VT_ResultSubShapePair) {
126     std::shared_ptr<ModelAPI_Result> aResult = myResultSubShapePair.first;
127     if(!aResult.get()) {
128       return;
129     }
130     aResult->data()->setName(theName);
131   }
132 }
133
134 void ModelHighAPI_Selection::setColor(int theRed, int theGreen, int theBlue)
135 {
136   if (myVariantType != VT_ResultSubShapePair || !myResultSubShapePair.first.get())
137     return;
138
139   AttributeIntArrayPtr aColor =
140       myResultSubShapePair.first->data()->intArray(ModelAPI_Result::COLOR_ID());
141   aColor->setSize(3);
142   aColor->setValue(0, theRed);
143   aColor->setValue(1, theGreen);
144   aColor->setValue(2, theBlue);
145 }
146
147 void ModelHighAPI_Selection::setDeflection(double theValue)
148 {
149   if (myVariantType != VT_ResultSubShapePair)
150     return;
151
152   AttributeDoublePtr aDeflectionAttr =
153     myResultSubShapePair.first->data()->real(ModelAPI_Result::DEFLECTION_ID());
154
155   aDeflectionAttr->setValue(theValue);
156 }
157
158 void ModelHighAPI_Selection::setTransparency(double theValue)
159 {
160   if (myVariantType != VT_ResultSubShapePair)
161     return;
162
163   AttributeDoublePtr aTransparencyAttr =
164     myResultSubShapePair.first->data()->real(ModelAPI_Result::TRANSPARENCY_ID());
165
166   aTransparencyAttr->setValue(theValue);
167 }
168
169 int ModelHighAPI_Selection::numberOfSubs() const
170 {
171   if (myVariantType != VT_ResultSubShapePair)
172     return 0;
173
174   ResultCompSolidPtr aCompSolid =
175       std::dynamic_pointer_cast<ModelAPI_ResultCompSolid>(myResultSubShapePair.first);
176   if (!aCompSolid)
177     return 0;
178
179   return aCompSolid->numberOfSubs();
180 }
181
182 ModelHighAPI_Selection ModelHighAPI_Selection::subResult(int theIndex) const
183 {
184   if (myVariantType != VT_ResultSubShapePair)
185     return ModelHighAPI_Selection();
186
187   ResultCompSolidPtr aCompSolid =
188       std::dynamic_pointer_cast<ModelAPI_ResultCompSolid>(myResultSubShapePair.first);
189   if (!aCompSolid)
190     return ModelHighAPI_Selection();
191
192   ResultBodyPtr aResult = aCompSolid->subResult(theIndex);
193   return ModelHighAPI_Selection(aResult, aResult->shape());
194 }