1 // Copyright (C) 2014-2017 CEA/DEN, EDF R&D
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.
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.
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
17 // See http://www.salome-platform.org/ or
18 // email : webmaster.salome@opencascade.com<mailto:webmaster.salome@opencascade.com>
21 #include "ModelHighAPI_Selection.h"
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_ResultBody.h>
31 #include <GeomAPI_Pnt.h>
32 //--------------------------------------------------------------------------------------
34 //--------------------------------------------------------------------------------------
35 ModelHighAPI_Selection::ModelHighAPI_Selection()
36 : myVariantType(VT_Empty)
40 ModelHighAPI_Selection::ModelHighAPI_Selection(const std::shared_ptr<ModelAPI_Result>& theContext,
41 const std::shared_ptr<GeomAPI_Shape>& theSubShape)
42 : myVariantType(VT_ResultSubShapePair)
43 , myResultSubShapePair(theContext, theSubShape)
47 ModelHighAPI_Selection::ModelHighAPI_Selection(const std::string& theType,
48 const std::string& theSubShapeName)
49 : myVariantType(VT_TypeSubShapeNamePair)
50 , myTypeSubShapeNamePair(theType, theSubShapeName)
54 ModelHighAPI_Selection::ModelHighAPI_Selection(const std::string& theType,
55 const GeomPointPtr& theSubShapeInnerPoint)
56 : myVariantType(VT_TypeInnerPointPair)
57 , myTypeInnerPointPair(theType, theSubShapeInnerPoint)
61 ModelHighAPI_Selection::ModelHighAPI_Selection(const std::string& theType,
62 const std::string& theContextName, const int theIndex)
63 : myVariantType(VT_WeakNamingPair)
64 , myWeakNamingPair(theType, std::pair<std::string, int>(theContextName, theIndex))
69 ModelHighAPI_Selection::~ModelHighAPI_Selection()
73 //--------------------------------------------------------------------------------------
74 void ModelHighAPI_Selection::fillAttribute(
75 const std::shared_ptr<ModelAPI_AttributeSelection> & theAttribute) const
77 switch(myVariantType) {
78 case VT_Empty: return;
79 case VT_ResultSubShapePair:
80 theAttribute->setValue(myResultSubShapePair.first, myResultSubShapePair.second);
82 case VT_TypeSubShapeNamePair:
83 theAttribute->selectSubShape(myTypeSubShapeNamePair.first, myTypeSubShapeNamePair.second);
85 case VT_TypeInnerPointPair:
86 theAttribute->selectSubShape(myTypeInnerPointPair.first, myTypeInnerPointPair.second);
88 case VT_WeakNamingPair:
89 theAttribute->selectSubShape(
90 myWeakNamingPair.first, myWeakNamingPair.second.first, myWeakNamingPair.second.second);
94 if (theAttribute->isInvalid()) {
95 FeaturePtr aFeature = ModelAPI_Feature::feature(theAttribute->owner());
97 std::string("Error: attribute \"") + theAttribute->id() + std::string("\" is invalid."));
101 //--------------------------------------------------------------------------------------
102 void ModelHighAPI_Selection::appendToList(
103 const std::shared_ptr<ModelAPI_AttributeSelectionList> & theAttribute) const
105 switch(myVariantType) {
106 case VT_Empty: return;
107 case VT_ResultSubShapePair:
108 theAttribute->append(myResultSubShapePair.first, myResultSubShapePair.second);
110 case VT_TypeSubShapeNamePair:
111 // Note: the reverse order (first - type, second - sub-shape name)
112 theAttribute->append(myTypeSubShapeNamePair.second, myTypeSubShapeNamePair.first);
114 case VT_TypeInnerPointPair:
115 // Note: the reverse order (first - type, second - selected point)
116 theAttribute->append(myTypeInnerPointPair.second, myTypeInnerPointPair.first);
118 case VT_WeakNamingPair:
119 // Note: the reverse order (first - type, second - selected point)
120 theAttribute->append(
121 myWeakNamingPair.first, myWeakNamingPair.second.first, myWeakNamingPair.second.second);
126 //==================================================================================================
127 ModelHighAPI_Selection::VariantType ModelHighAPI_Selection::variantType() const
129 return myVariantType;
132 //==================================================================================================
133 ResultSubShapePair ModelHighAPI_Selection::resultSubShapePair() const
135 return myResultSubShapePair;
138 //==================================================================================================
139 TypeSubShapeNamePair ModelHighAPI_Selection::typeSubShapeNamePair() const
141 return myTypeSubShapeNamePair;
144 //==================================================================================================
145 TypeInnerPointPair ModelHighAPI_Selection::typeInnerPointPair() const
147 return myTypeInnerPointPair;
150 //==================================================================================================
151 std::string ModelHighAPI_Selection::shapeType() const
153 switch(myVariantType) {
154 case VT_ResultSubShapePair:
155 return myResultSubShapePair.second.get() ? myResultSubShapePair.second->shapeTypeStr() :
156 myResultSubShapePair.first->shape()->shapeTypeStr();
157 case VT_TypeSubShapeNamePair: return myTypeSubShapeNamePair.first;
158 case VT_TypeInnerPointPair: return myTypeInnerPointPair.first;
164 //==================================================================================================
165 void ModelHighAPI_Selection::setName(const std::string& theName)
167 if (myVariantType == VT_ResultSubShapePair) {
168 std::shared_ptr<ModelAPI_Result> aResult = myResultSubShapePair.first;
172 aResult->data()->setName(theName);
176 std::string ModelHighAPI_Selection::name() const
178 if (myVariantType == VT_ResultSubShapePair) {
179 std::shared_ptr<ModelAPI_Result> aResult = myResultSubShapePair.first;
181 return aResult->data()->name();
183 return std::string();
186 void ModelHighAPI_Selection::setColor(int theRed, int theGreen, int theBlue)
188 if (myVariantType != VT_ResultSubShapePair || !myResultSubShapePair.first.get())
191 AttributeIntArrayPtr aColor =
192 myResultSubShapePair.first->data()->intArray(ModelAPI_Result::COLOR_ID());
194 aColor->setValue(0, theRed);
195 aColor->setValue(1, theGreen);
196 aColor->setValue(2, theBlue);
199 void ModelHighAPI_Selection::setDeflection(double theValue)
201 if (myVariantType != VT_ResultSubShapePair)
204 AttributeDoublePtr aDeflectionAttr =
205 myResultSubShapePair.first->data()->real(ModelAPI_Result::DEFLECTION_ID());
207 aDeflectionAttr->setValue(theValue);
210 void ModelHighAPI_Selection::setTransparency(double theValue)
212 if (myVariantType != VT_ResultSubShapePair)
215 AttributeDoublePtr aTransparencyAttr =
216 myResultSubShapePair.first->data()->real(ModelAPI_Result::TRANSPARENCY_ID());
218 aTransparencyAttr->setValue(theValue);
221 int ModelHighAPI_Selection::numberOfSubs() const
223 if (myVariantType != VT_ResultSubShapePair)
226 ResultBodyPtr aBody = std::dynamic_pointer_cast<ModelAPI_ResultBody>(myResultSubShapePair.first);
230 return aBody->numberOfSubs();
233 ModelHighAPI_Selection ModelHighAPI_Selection::subResult(int theIndex) const
235 if (myVariantType != VT_ResultSubShapePair)
236 return ModelHighAPI_Selection();
238 ResultBodyPtr aBody = std::dynamic_pointer_cast<ModelAPI_ResultBody>(myResultSubShapePair.first);
240 return ModelHighAPI_Selection();
241 if (theIndex >= aBody->numberOfSubs())
242 return ModelHighAPI_Selection();
244 ResultBodyPtr aResult = aBody->subResult(theIndex);
245 return ModelHighAPI_Selection(aResult, aResult->shape());