Salome HOME
Unit tests fixes related to python swigged lists iteration
[modules/shaper.git] / src / Model / Model_Validator.cpp
index 771ea6c34bd461fdcf5899b88d8b9aece6736d89..7314b48aaadd89a409aca0ab582003e22add29ce 100644 (file)
+// Copyright (C) 2014-20xx CEA/DEN, EDF R&D
+
 // File:        Model_Validator.cpp
 // Created:     2 Jul 2014
 // Author:      Mikhail PONIKAROV
 
-#include <Model_Validator.h>
+#include "Model_Validator.h"
+
+#include "Model_AttributeValidator.h"
+#include "Model_FeatureValidator.h"
+
+#include <ModelAPI_Attribute.h>
+#include <ModelAPI_AttributeString.h>
+#include <ModelAPI_AttributeValidator.h>
 #include <ModelAPI_Feature.h>
-#include <Events_Error.h>
+#include <ModelAPI_Result.h>
+#include <Model_Data.h>
 
-using namespace std;
+#include <Config_Translator.h>
 
-void Model_ValidatorsFactory::registerValidator(
-  const string& theID, ModelAPI_Validator* theValidator)
+#include <Events_InfoMessage.h>
+
+void Model_ValidatorsFactory::registerValidator(const std::string& theID,
+  ModelAPI_Validator* theValidator)
 {
   if (myIDs.find(theID) != myIDs.end()) {
-    Events_Error::send(string("Validator ") + theID + " is already registered");
+    Events_InfoMessage("Model_Validator", "Validator %1 is already registered").arg(theID).send();
   } else {
     myIDs[theID] = theValidator;
   }
 }
 
-void Model_ValidatorsFactory::assignValidator(const string& theID, const string& theFeatureID)
+void Model_ValidatorsFactory::assignValidator(const std::string& theID,
+  const std::string& theFeatureID)
 {
-  bool isError = false;
-  map<string, ModelAPI_Validator*>::iterator aVal = myIDs.find(theID);
-  if (aVal == myIDs.end()) {
-    Events_Error::send(
-      string("Validator ") + theID + " for feature " + theFeatureID + " was not registered");
-    isError = true;
+  if (myFeatures.find(theFeatureID) == myFeatures.end()) {
+    myFeatures[theFeatureID] = AttrValidators();
   }
-  if (myFeatures.find(theFeatureID) != myFeatures.end()) {
-    // TODO: it is called many times now because of not-optimized XML reader
-      //Events_Error::send(
-      //  string("Validator for feature ") + theFeatureID + " is already registered");
-    isError = true;
+  if (myFeatures[theFeatureID].find(theID) != myFeatures[theFeatureID].end()) {
+    //Events_Error::send(std::string("Validator ") + theID +
+    //  " for feature " + theFeatureID + "is already registered");
+  } else {
+    myFeatures[theFeatureID][theID] = std::list<std::string>();
   }
-  if (!isError)
-    myFeatures[theFeatureID] = aVal->second;
 }
 
-void Model_ValidatorsFactory::assignValidator(const string& theID, 
-  const string& theFeatureID, const string& theAttrID, const list<string>& theArguments)
+void Model_ValidatorsFactory::assignValidator(const std::string& theID,
+  const std::string& theFeatureID,
+  const std::list<std::string>& theArguments)
 {
-  bool isError = false;
-  map<string, ModelAPI_Validator*>::iterator aVal = myIDs.find(theID);
-  if (aVal == myIDs.end()) {
-    Events_Error::send(
-      string("Validator ") + theID + " for feature " + theFeatureID + " was not registered");
-    isError = true;
+  if (myFeatures.find(theFeatureID) == myFeatures.end()) {
+    myFeatures[theFeatureID] = AttrValidators();
   }
+
+  if (myFeatures[theFeatureID].find(theID) != myFeatures[theFeatureID].end()) {
+    //Events_Error::send(std::string("Validator ") + theID +
+    //  " for feature " + theFeatureID + "is already registered");
+  } else {
+    myFeatures[theFeatureID][theID] = theArguments;
+  }
+}
+
+void Model_ValidatorsFactory::assignValidator(const std::string& theID,
+  const std::string& theFeatureID,
+  const std::string& theAttrID,
+  const std::list<std::string>& theArguments)
+{
   // create feature-structures if not exist
-  map<string, map<string, pair<ModelAPI_Validator*, list<string> > > >::iterator
-    aFeature = myAttrs.find(theFeatureID);
+  std::map<std::string, std::map<std::string, AttrValidators> >::iterator aFeature = myAttrs.find(
+    theFeatureID);
   if (aFeature == myAttrs.end()) {
-    myAttrs[theFeatureID] = map<string, pair<ModelAPI_Validator*, list<string> > >();
+    myAttrs[theFeatureID] = std::map<std::string, AttrValidators>();
     aFeature = myAttrs.find(theFeatureID);
   }
   // add attr-structure if not exist, or generate error if already exist
-  map<string, pair<ModelAPI_Validator*, list<string> > >::iterator 
-    anAttr = aFeature->second.find(theAttrID);
+  std::map<std::string, AttrValidators>::iterator anAttr = aFeature->second.find(theAttrID);
   if (anAttr == aFeature->second.end()) {
-    if (!isError) {
-      aFeature->second[theAttrID] = 
-        pair<ModelAPI_Validator*, list<string> >(aVal->second, theArguments);
+    aFeature->second[theAttrID] = AttrValidators();
+  }
+  aFeature->second[theAttrID][theID] = theArguments;
+}
+
+void Model_ValidatorsFactory::validators(const std::string& theFeatureID,
+                                         Validators& theValidators) const
+{
+  std::map<std::string, AttrValidators>::const_iterator aFeatureIt =
+      myFeatures.find(theFeatureID);
+  if (aFeatureIt != myFeatures.cend()) {
+    AttrValidators::const_iterator aValidatorsIt = aFeatureIt->second.cbegin();
+    for (; aValidatorsIt != aFeatureIt->second.cend(); aValidatorsIt++) {
+      if (!validator(aValidatorsIt->first)) {
+        Events_InfoMessage("Model_Validator", "Validator %1 was not registered")
+          .arg(aValidatorsIt->first).send();
+      } else {
+        theValidators.push_back(std::make_pair(aValidatorsIt->first, aValidatorsIt->second));
+      }
+    }
+  }
+  addDefaultValidators(theValidators);
+}
+
+void Model_ValidatorsFactory::validators(const std::string& theFeatureID,
+                                         const std::string& theAttrID,
+                                         Validators& theValidators) const
+{
+  std::map<std::string, std::map<std::string, AttrValidators> >::const_iterator aFeatureIt =
+      myAttrs.find(theFeatureID);
+  if (aFeatureIt != myAttrs.cend()) {
+    std::map<std::string, AttrValidators>::const_iterator anAttrIt =
+      aFeatureIt->second.find(theAttrID);
+    if (anAttrIt != aFeatureIt->second.end()) {
+      AttrValidators::const_iterator aValidatorsIt = anAttrIt->second.cbegin();
+      for (; aValidatorsIt != anAttrIt->second.cend(); aValidatorsIt++) {
+        if (!validator(aValidatorsIt->first)) {
+          Events_InfoMessage("Model_Validator", "Validator %1 was not registered")
+            .arg(aValidatorsIt->first).send();
+        } else {
+          theValidators.push_back(std::make_pair(aValidatorsIt->first, aValidatorsIt->second));
+        }
+      }
+    }
+  }
+  addDefaultAttributeValidators(theValidators);
+}
+
+Model_ValidatorsFactory::Model_ValidatorsFactory()
+  : ModelAPI_ValidatorsFactory()
+{
+  registerValidator("Model_FeatureValidator", new Model_FeatureValidator);
+  registerValidator("Model_AttributeValidator", new Model_AttributeValidator);
+}
+
+const ModelAPI_Validator* Model_ValidatorsFactory::validator(const std::string& theID) const
+{
+  std::map<std::string, ModelAPI_Validator*>::const_iterator aIt = myIDs.find(theID);
+  if (aIt != myIDs.end()) {
+    return aIt->second;
+  }
+  return NULL;
+}
+
+void Model_ValidatorsFactory::addDefaultValidators(Validators& theValidators) const
+{
+  const static std::string kDefaultId = "Model_FeatureValidator";
+  if (!validator(kDefaultId))
+    return;
+  theValidators.push_back(std::make_pair(kDefaultId, std::list<std::string>()));
+}
+
+void Model_ValidatorsFactory::addDefaultAttributeValidators(Validators& theValidators) const
+{
+  const static std::string kDefaultId = "Model_AttributeValidator";
+  if (!validator(kDefaultId))
+    return;
+  theValidators.push_back(std::make_pair(kDefaultId, std::list<std::string>()));
+}
+
+bool Model_ValidatorsFactory::validate(const std::shared_ptr<ModelAPI_Feature>& theFeature) const
+{
+  std::shared_ptr<Model_Data> aData = std::dynamic_pointer_cast<Model_Data>(theFeature->data());
+  if (aData.get() && aData->isValid()) {
+    if (aData->execState() == ModelAPI_StateDone)
+      aData->eraseErrorString(); // no error => erase the information string
+  } else {
+    return false; // feature is broken, already not presented in the data structure
+  }
+
+  // check feature validators first
+  Validators aValidators;
+  validators(theFeature->getKind(), aValidators);
+
+  if (!aValidators.empty()) {
+    Validators::const_iterator aValidatorIt = aValidators.cbegin();
+    for(; aValidatorIt != aValidators.cend(); aValidatorIt++) {
+      const std::string& aValidatorID = aValidatorIt->first;
+      const std::list<std::string>& anArguments = aValidatorIt->second;
+      const ModelAPI_FeatureValidator* aFValidator =
+        dynamic_cast<const ModelAPI_FeatureValidator*>(validator(aValidatorID));
+      if (aFValidator) {
+        Events_InfoMessage anError;
+        if (!aFValidator->isValid(theFeature, anArguments, anError)) {
+          if (anError.empty())
+            anError = "Unknown error.";
+          if (anError.context().empty()) {
+            anError.setContext(theFeature->getKind() + ":" + aValidatorID);
+          }
+          theFeature->setError(Config_Translator::translate(anError), false, false);
+          theFeature->data()->execState(ModelAPI_StateInvalidArgument);
+          return false;
+        }
+      }
+    }
+  }
+
+  // check all attributes for validity
+  static const std::string kAllTypes = "";
+  std::list<std::string> aLtAttributes = aData->attributesIDs(kAllTypes);
+  std::list<std::string>::const_iterator anAttrIt = aLtAttributes.cbegin();
+  for (; anAttrIt != aLtAttributes.cend(); anAttrIt++) {
+    const std::string& anAttributeID = *anAttrIt;
+    AttributePtr anAttribute = theFeature->data()->attribute(anAttributeID);
+
+    std::string aValidatorID;
+    Events_InfoMessage anError;
+    if (!validate(anAttribute, aValidatorID, anError)) {
+      if (anError.empty())
+        anError = "Unknown error.";
+      if (anError.context().empty()) {
+        anError.setContext(theFeature->getKind() + ":" + anAttributeID + ":" + aValidatorID);
+      }
+      theFeature->setError(Config_Translator::translate(anError), false, false);
+      theFeature->data()->execState(ModelAPI_StateInvalidArgument);
+      return false;
+    }
+  }
+
+  return true;
+}
+
+bool Model_ValidatorsFactory::validate(const std::shared_ptr<ModelAPI_Attribute>& theAttribute,
+                                       std::string& theValidator,
+                                       Events_InfoMessage& theError) const
+{
+  FeaturePtr aFeature = ModelAPI_Feature::feature(theAttribute->owner());
+  if (!aFeature.get()) {
+    theValidator = "Model_ValidatorsFactory";
+    theError = "Attribute has no feature.";
+    return false;
+  }
+
+  // skip not-case attributes, that really may be invalid (issue 671)
+  if (!const_cast<Model_ValidatorsFactory*>(this)->isCase(aFeature, theAttribute->id()))
+    return true;
+
+  Validators aValidators;
+  validators(aFeature->getKind(), theAttribute->id(), aValidators);
+
+  Validators::iterator aValidatorIt = aValidators.begin();
+  for (; aValidatorIt != aValidators.end(); ++aValidatorIt) {
+    const std::string& aValidatorID = aValidatorIt->first;
+    const std::list<std::string>& anArguments = aValidatorIt->second;
+    const ModelAPI_AttributeValidator* anAttrValidator =
+        dynamic_cast<const ModelAPI_AttributeValidator*>(validator(aValidatorID));
+    if (!anAttrValidator)
+      continue;
+    if (!anAttrValidator->isValid(theAttribute, anArguments, theError)) {
+      theValidator = aValidatorID;
+      return false;
+    }
+  }
+
+  return true;
+}
+
+void Model_ValidatorsFactory::registerNotObligatory(std::string theFeature,
+                                                    std::string theAttribute)
+{
+  const static std::string kDefaultId = "Model_FeatureValidator";
+  std::map<std::string, ModelAPI_Validator*>::const_iterator it = myIDs.find(kDefaultId);
+  if (it != myIDs.end()) {
+    Model_FeatureValidator* aValidator = dynamic_cast<Model_FeatureValidator*>(it->second);
+    if (aValidator) {
+      aValidator->registerNotObligatory(theFeature, theAttribute);
+    }
+  }
+}
+
+bool Model_ValidatorsFactory::isNotObligatory(std::string theFeature, std::string theAttribute)
+{
+  const static std::string kDefaultId = "Model_FeatureValidator";
+  std::map<std::string, ModelAPI_Validator*>::const_iterator it = myIDs.find(kDefaultId);
+  if (it != myIDs.end()) {
+    Model_FeatureValidator* aValidator = dynamic_cast<Model_FeatureValidator*>(it->second);
+    if (aValidator) {
+      return aValidator->isNotObligatory(theFeature, theAttribute);
     }
+  }
+  return false; // default
+}
+
+void Model_ValidatorsFactory::registerConcealment(std::string theFeature, std::string theAttribute)
+{
+  std::map<std::string, std::set<std::string> >::iterator aFind = myConcealed.find(theFeature);
+  if (aFind == myConcealed.end()) {
+    std::set<std::string> aNewSet;
+    aNewSet.insert(theAttribute);
+    myConcealed[theFeature] = aNewSet;
   } else {
-    // TODO: it is called many times now because of not-optimized XML reader
-    //Events_Error::send(
-    //  string("Validator ") + theID + " for feature " + theFeatureID +
-    //          "attribute " + theAttrID + " is already registered");
-    isError = true;
+    aFind->second.insert(theAttribute);
   }
 }
 
-const ModelAPI_Validator* Model_ValidatorsFactory::validator(const string& theFeatureID) const
+bool Model_ValidatorsFactory::isConcealed(std::string theFeature, std::string theAttribute)
 {
-  map<string, ModelAPI_Validator*>::const_iterator aFeature = myFeatures.find(theFeatureID);
-  if (aFeature != myFeatures.cend())
-    return aFeature->second;
-  return NULL; // not found
+  std::map<std::string, std::set<std::string> >::iterator aFind = myConcealed.find(theFeature);
+  return aFind != myConcealed.end() && aFind->second.find(theAttribute) != aFind->second.end();
 }
 
-/*bool Model_ValidatorsFactory::validate(
-  const boost::shared_ptr<ModelAPI_Feature>& theFeature, const string& theAttrID ) const
+void Model_ValidatorsFactory::registerCase(std::string theFeature, std::string theAttribute,
+                            const std::list<std::pair<std::string, std::string> >& theCases)
 {
-  map<string, map<string, pair<ModelAPI_Validator*, list<string> > > >::const_iterator
-    aFeature = myAttrs.find(theFeature->getKind());
-  if (aFeature == myAttrs.cend()) return true; // feature is not found
-  map<string, pair<ModelAPI_Validator*, list<string> > >::const_iterator 
-    anAttr = aFeature->second.find(theAttrID);
-  if (anAttr == aFeature->second.cend()) return true; // attribute is not found
-  return anAttr->second.first->validate(theFeature, theAttrID, anAttr->second.second);
-}*/
+  std::map<std::string, std::map<std::string, std::map<std::string, std::set<std::string> > > >
+    ::iterator aFindFeature = myCases.find(theFeature);
+  if (aFindFeature == myCases.end()) {
+    myCases[theFeature] = std::map<std::string, std::map<std::string, std::set<std::string> > >();
+    aFindFeature = myCases.find(theFeature);
+  }
+  std::map<std::string, std::map<std::string, std::set<std::string> > >::iterator aFindAttrID =
+    aFindFeature->second.find(theAttribute);
+
+  if (aFindAttrID == aFindFeature->second.end()) {
+    aFindFeature->second[theAttribute] =
+      std::map<std::string, std::set<std::string> >();
+    aFindAttrID = aFindFeature->second.find(theAttribute);
+  }
+  std::list<std::pair<std::string, std::string> >::const_iterator aCasesIt = theCases.begin(),
+                                                                       aCasesLast = theCases.end();
+  std::map<std::string, std::set<std::string> > aFindCases = aFindAttrID->second;
+  for (; aCasesIt != aCasesLast; aCasesIt++) {
+    std::pair<std::string, std::string> aCasePair = *aCasesIt;
+    std::string aSwitch = aCasePair.first;
+    if (aFindAttrID->second.find(aSwitch) == aFindAttrID->second.end()) {
+      aFindAttrID->second[aSwitch] = std::set<std::string>();
+    }
+    aFindAttrID->second[aSwitch].insert(aCasePair.second);
+  }
+}
 
-Model_ValidatorsFactory::Model_ValidatorsFactory() : ModelAPI_ValidatorsFactory()
+bool Model_ValidatorsFactory::isCase(FeaturePtr theFeature, std::string theAttribute)
 {
+  bool anInCase = true;
+  std::map<std::string, std::map<std::string, std::map<std::string, std::set<std::string> > > >
+    ::iterator aFindFeature = myCases.find(theFeature->getKind());
+  if (aFindFeature != myCases.end()) {
+    std::map<std::string, std::map<std::string, std::set<std::string> > >::iterator
+      aFindAttrID = aFindFeature->second.find(theAttribute);
+    if (aFindAttrID != aFindFeature->second.end()) {
+      std::map<std::string, std::set<std::string> >::iterator
+              aCasesIt = aFindAttrID->second.begin(), aCasesLast = aFindAttrID->second.end();
+      for (; aCasesIt != aCasesLast && anInCase; aCasesIt++) {
+        // the the switch-attribute that contains the case value
+        AttributeStringPtr aSwitch = theFeature->string(aCasesIt->first);
+        if (aSwitch.get()) {
+          // the second has the case identifier
+          anInCase =  aCasesIt->second.find(aSwitch->value()) != aCasesIt->second.end();
+        }
+      }
+    }
+  }
+  return anInCase; // if no additional conditions, this attribute is the case to be validated
 }