Salome HOME
Use Feature as a value for selection attribute
[modules/shaper.git] / src / GeomValidators / GeomValidators_BodyShapes.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 "GeomValidators_BodyShapes.h"
22
23 #include <Events_InfoMessage.h>
24
25 #include <ModelAPI_AttributeSelectionList.h>
26 #include <ModelAPI_Object.h>
27 #include <ModelAPI_ResultConstruction.h>
28 #include <ModelAPI_Tools.h>
29 #include <ModelAPI_ResultBody.h>
30
31 bool GeomValidators_BodyShapes::isValid(const AttributePtr& theAttribute,
32                                         const std::list<std::string>& theArguments,
33                                         Events_InfoMessage& theError) const
34 {
35   std::string anAttributeType = theAttribute->attributeType();
36   if(anAttributeType == ModelAPI_AttributeSelection::typeId()) {
37     AttributeSelectionPtr anAttrSelection =
38       std::dynamic_pointer_cast<ModelAPI_AttributeSelection>(theAttribute);
39     ResultPtr aContext = anAttrSelection->context();
40     FeaturePtr aContextFeature = anAttrSelection->contextFeature();
41     if (!(aContext.get() || aContextFeature.get())) {
42       theError = "Error: Context is empty.";
43       return false;
44     }
45     if (aContext.get()) {
46       ResultConstructionPtr aResultConstruction =
47         std::dynamic_pointer_cast<ModelAPI_ResultConstruction>(aContext);
48       if (aResultConstruction.get()) {
49         theError = "Error: Result construction selected.";
50         return false;
51       }
52       // additional check that the selected object is top-level result
53       if (theArguments.size() > 0 && *(theArguments.rbegin()) == "toplevel") {
54         if (ModelAPI_Tools::bodyOwner(aContext).get()) {
55           theError = "Error: Only higher level shape allowed.";
56           return false;
57         }
58       }
59     } else {
60       std::list<ResultPtr> aResList = aContextFeature->results();
61       std::list<ResultPtr>::const_iterator aIt;
62       for (aIt = aResList.cbegin(); aIt != aResList.cend(); aIt++) {
63         ResultPtr aRes = (*aIt);
64         ResultBodyPtr aBody = std::dynamic_pointer_cast<ModelAPI_ResultBody>(aRes);
65         if (aBody.get())
66           break;
67       }
68       if (aIt == aResList.cend()) {
69         theError = "Error: Feature doesn't creates body.";
70         return false;
71       }
72     }
73   } else if(anAttributeType == ModelAPI_AttributeSelectionList::typeId()) {
74     AttributeSelectionListPtr anAttrSelectionList =
75       std::dynamic_pointer_cast<ModelAPI_AttributeSelectionList>(theAttribute);
76
77     // All objects should not be result constructions.
78     for(int anIndex = 0, aSize = anAttrSelectionList->size(); anIndex < aSize; ++anIndex) {
79       AttributeSelectionPtr anAttrSelection = anAttrSelectionList->value(anIndex);
80       if(!isValid(anAttrSelection, theArguments, theError)) {
81         return false;
82       }
83     }
84   } else {
85     theError = "Error: Attribute \"%1\" does not supported by this validator.";
86     theError.arg(anAttributeType);
87     return false;
88   }
89
90   return true;
91 }