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