Salome HOME
d27817df052e4edb27bb78cdad56b925a98c7f53
[modules/shaper.git] / src / FeaturesPlugin / FeaturesPlugin_ValidatorTransform.cpp
1 // Copyright (C) 2014-2023  CEA, EDF
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 "FeaturesPlugin_ValidatorTransform.h"
21
22 #include <Events_InfoMessage.h>
23
24 #include <ModelAPI_AttributeSelectionList.h>
25 #include <ModelAPI_ResultPart.h>
26 #include <ModelAPI_ResultBody.h>
27 #include <ModelAPI_Session.h>
28 #include <ModelAPI_Tools.h>
29
30 bool FeaturesPlugin_ValidatorTransform::isValid(const AttributePtr& theAttribute,
31                                                 const std::list<std::string>& /*theArguments*/,
32                                                 Events_InfoMessage& theError) const
33 {
34   bool aValid = true;
35   std::string anAttributeType = theAttribute->attributeType();
36   if (anAttributeType != ModelAPI_AttributeSelectionList::typeId()) {
37 // LCOV_EXCL_START
38     theError = "The attribute with the %1 type is not processed";
39     theError.arg(theAttribute->attributeType());
40     return false;
41 // LCOV_EXCL_STOP
42   }
43
44   std::shared_ptr<ModelAPI_AttributeSelectionList> aCurSelList =
45                            std::dynamic_pointer_cast<ModelAPI_AttributeSelectionList>(theAttribute);
46
47   DocumentPtr aDocument = theAttribute->owner()->document();
48   SessionPtr aMgr = ModelAPI_Session::get();
49   bool isPartSetDocument = aDocument == aMgr->moduleDocument();
50   bool isCheckCompsolid = !theAttribute->owner()->data()->version().empty();
51
52   std::string anErrorGroupName;
53   for(int i = 0; i < aCurSelList->size() && aValid; i++) {
54     std::shared_ptr<ModelAPI_AttributeSelection> aSelAttribute = aCurSelList->value(i);
55     ResultPtr aResult = aSelAttribute->context();
56     if (!aResult) {
57       // this could be a whole feature selected
58       FeaturePtr aFeature = aSelAttribute->contextFeature();
59       if (aFeature.get() && aFeature->results().size() > 0) {
60         if (aFeature->firstResult()->groupName() != ModelAPI_ResultBody::group()) {
61           anErrorGroupName = aFeature->firstResult()->groupName();
62           aValid = false;
63           break;
64         }
65         continue;
66       }
67       theError = "Invalid selection.";
68       return false;
69     }
70     if (isPartSetDocument) // PartSet document: Result Part is valid
71       aValid = aResult->groupName() == ModelAPI_ResultPart::group();
72     else { // Part document: Result CompSolid is valid, but the solid in compsolid is forbidden
73       aValid = aResult->groupName() == ModelAPI_ResultBody::group();
74       if (aValid && isCheckCompsolid) {
75         ResultBodyPtr aComp = ModelAPI_Tools::bodyOwner(aResult);
76         if (aComp && aComp->shape()->shapeType() == GeomAPI_Shape::COMPSOLID) {
77           theError = "Selecting a part of compsolid is forbidden.";
78           return false;
79         }
80       }
81     }
82     if (!aValid)
83       anErrorGroupName = aResult->groupName();
84   }
85   if (!aValid) {
86     std::string aResultGroupName = isPartSetDocument ? ModelAPI_ResultPart::group()
87                                                      : ModelAPI_ResultBody::group();
88     theError = "Objects from the %1 group can be selected in the %2 document, "
89                "but an objects from the %3 group is selected.";
90     theError.arg(aResultGroupName).arg(aDocument->kind()).arg(anErrorGroupName);
91   }
92   return aValid;
93 }