src/FeaturesPlugin/FeaturesPlugin_ValidatorTransform.cpp

   1 // Copyright (C) 2014-2019  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 "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
  29 bool FeaturesPlugin_ValidatorTransform::isValid(const AttributePtr& theAttribute,
  30                                                 const std::list<std::string>& theArguments,
  31                                                 Events_InfoMessage& theError) const
  32 {
  33   bool aValid = true;
  34   std::string anAttributeType = theAttribute->attributeType();
  35   if (anAttributeType != ModelAPI_AttributeSelectionList::typeId()) {
  36 // LCOV_EXCL_START
  37     theError = "The attribute with the %1 type is not processed";
  38     theError.arg(theAttribute->attributeType());
  39     return false;
  40 // LCOV_EXCL_STOP
  41   }
  42
  43   std::shared_ptr<ModelAPI_AttributeSelectionList> aCurSelList =
  44                            std::dynamic_pointer_cast<ModelAPI_AttributeSelectionList>(theAttribute);
  45
  46   DocumentPtr aDocument = theAttribute->owner()->document();
  47   SessionPtr aMgr = ModelAPI_Session::get();
  48   bool isPartSetDocument = aDocument == aMgr->moduleDocument();
  49
  50   std::string anErrorGroupName;
  51   for(int i = 0; i < aCurSelList->size() && aValid; i++) {
  52     std::shared_ptr<ModelAPI_AttributeSelection> aSelAttribute = aCurSelList->value(i);
  53     ResultPtr aResult = aSelAttribute->context();
  54     if (!aResult) {
  55       // this could be a whole feature selected
  56       FeaturePtr aFeature = aSelAttribute->contextFeature();
  57       if (aFeature.get() && aFeature->results().size() > 0) {
  58         if (aFeature->firstResult()->groupName() != ModelAPI_ResultBody::group()) {
  59           anErrorGroupName = aFeature->firstResult()->groupName();
  60           aValid = false;
  61           break;
  62         }
  63         continue;
  64       }
  65       theError = "Invalid selection.";
  66       return false;
  67     }
  68     if (isPartSetDocument) // PartSet document: Result Part is valid
  69       aValid = aResult->groupName() == ModelAPI_ResultPart::group();
  70     else { // Part document: Result CompSolid is valid
  71       aValid = aResult->groupName() == ModelAPI_ResultBody::group();
  72       if (aValid) {
  73         ResultBodyPtr aComp = std::dynamic_pointer_cast<ModelAPI_ResultBody>(aResult);
  74         aValid = aComp.get() != NULL;
  75       }
  76     }
  77     if (!aValid)
  78       anErrorGroupName = aResult->groupName();
  79   }
  80   if (!aValid) {
  81     std::string aResultGroupName = isPartSetDocument ? ModelAPI_ResultPart::group()
  82                                                      : ModelAPI_ResultBody::group();
  83     theError = "Objects from the %1 group can be selected in the %2 document, "
  84                "but an objects from the %3 group is selected.";
  85     theError.arg(aResultGroupName).arg(aDocument->kind()).arg(anErrorGroupName);
  86   }
  87   return aValid;
  88 }