From 2b9a0ebe5b0b562ead8c4516fea9ebdc331e0963 Mon Sep 17 00:00:00 2001
From: nds <natalia.donis@opencascade.com>
Date: Mon, 2 Feb 2015 12:22:58 +0300
Subject: [PATCH] Issue #392 Can not create axis on two points of one sketch

---
 src/ConstructionPlugin/axis_widget.xml        |  2 +-
 src/ModelAPI/CMakeLists.txt                   |  2 ++
 src/ModelAPI/ModelAPI_ShapeValidator.cpp      | 36 +++++++++++++++++++
 src/ModelAPI/ModelAPI_ShapeValidator.h        | 33 +++++++++++++++++
 .../ModuleBase_WidgetShapeSelector.cpp        | 12 +++++++
 src/PartSet/PartSet_Module.cpp                |  2 ++
 6 files changed, 86 insertions(+), 1 deletion(-)
 create mode 100644 src/ModelAPI/ModelAPI_ShapeValidator.cpp
 create mode 100644 src/ModelAPI/ModelAPI_ShapeValidator.h

diff --git a/src/ConstructionPlugin/axis_widget.xml b/src/ConstructionPlugin/axis_widget.xml
index f612a83d0..a3424e3f1 100644
--- a/src/ConstructionPlugin/axis_widget.xml
+++ b/src/ConstructionPlugin/axis_widget.xml
@@ -14,6 +14,6 @@
     tooltip="Select a second point"
     shape_types="vertex">
     <selection_filter id="NoConstructionSubShapesFilter"/>
-    <validator id="PartSet_DifferentObjects"/>
+    <validator id="PartSet_DifferentShapes"/>
   </shape_selector>
 </source>
diff --git a/src/ModelAPI/CMakeLists.txt b/src/ModelAPI/CMakeLists.txt
index 738664c17..e8ef2bad2 100644
--- a/src/ModelAPI/CMakeLists.txt
+++ b/src/ModelAPI/CMakeLists.txt
@@ -37,6 +37,7 @@ SET(PROJECT_HEADERS
     ModelAPI_ResultValidator.h
     ModelAPI_Session.h
     ModelAPI_Tools.h
+    ModelAPI_ShapeValidator.h
     ModelAPI_Validator.h
 )
 
@@ -65,6 +66,7 @@ SET(PROJECT_SOURCES
     ModelAPI_ResultGroup.cpp
     ModelAPI_ResultPart.cpp
     ModelAPI_Session.cpp
+    ModelAPI_ShapeValidator.cpp
     ModelAPI_Tools.cpp
 )
 
diff --git a/src/ModelAPI/ModelAPI_ShapeValidator.cpp b/src/ModelAPI/ModelAPI_ShapeValidator.cpp
new file mode 100644
index 000000000..b451ed1bb
--- /dev/null
+++ b/src/ModelAPI/ModelAPI_ShapeValidator.cpp
@@ -0,0 +1,36 @@
+// Copyright (C) 2014-20xx CEA/DEN, EDF R&D
+
+// File:        ModelAPI_ShapeValidator.cpp
+// Created:     2 Feb 2015
+// Author:      Natalia ERMOLAEVA
+
+#include "ModelAPI_ShapeValidator.h"
+
+#include <ModelAPI_AttributeSelection.h>
+#include "ModelAPI_Object.h"
+
+bool ModelAPI_ShapeValidator::isValid(const FeaturePtr& theFeature,
+                                      const AttributePtr& theAttribute,
+                                      const GeomShapePtr& theShape) const
+{
+  std::string aCurrentAttributeId = theAttribute->id();
+  // get all feature attributes
+  std::list<AttributePtr> anAttrs = 
+                   theFeature->data()->attributes(ModelAPI_AttributeSelection::type());
+  if (anAttrs.size() > 0 && theShape.get() != NULL) {
+    std::list<AttributePtr>::iterator anAttr = anAttrs.begin();
+    for(; anAttr != anAttrs.end(); anAttr++) {
+      AttributePtr anAttribute = *anAttr;
+      // take into concideration only other attributes
+      if (anAttribute.get() != NULL && anAttribute->id() != aCurrentAttributeId) {
+        AttributeSelectionPtr aSelectionAttribute = 
+          std::dynamic_pointer_cast<ModelAPI_AttributeSelection>(anAttribute);
+        // the shape of the attribute should be not the same
+        if (aSelectionAttribute.get() != NULL && theShape->isEqual(aSelectionAttribute->value())) {
+          return false;
+        }
+      }
+    }
+  }
+  return true;
+}
diff --git a/src/ModelAPI/ModelAPI_ShapeValidator.h b/src/ModelAPI/ModelAPI_ShapeValidator.h
new file mode 100644
index 000000000..596e1c08e
--- /dev/null
+++ b/src/ModelAPI/ModelAPI_ShapeValidator.h
@@ -0,0 +1,33 @@
+// Copyright (C) 2014-20xx CEA/DEN, EDF R&D
+
+// File:        ModelAPI_ShapeValidator.h
+// Created:     2 Feb 2015
+// Author:      Natalia ERMOLAEVA
+
+#ifndef ModelAPI_ShapeValidator_H
+#define ModelAPI_ShapeValidator_H
+
+#include <ModelAPI.h>
+
+#include <ModelAPI_Validator.h>
+#include <ModelAPI_Feature.h>
+#include <ModelAPI_Attribute.h>
+
+#include <GeomAPI_Shape.h>
+
+/**
+ * Generic validator for any attribute of a feature.
+ */
+class ModelAPI_ShapeValidator : public ModelAPI_Validator
+{
+public:
+  /// returns True if the attribute is valid. It checks whether the feature of the attribute
+  /// does not contain a selection attribute filled with the same shape
+  /// \param theFeature a feature to check
+  /// \param theAttribute an attribute to check
+  /// \param theShape a shape
+  MODELAPI_EXPORT bool isValid(const FeaturePtr& theFeature, const AttributePtr& theAttribute,
+                               const GeomShapePtr& theShape) const;
+};
+
+#endif
diff --git a/src/ModuleBase/ModuleBase_WidgetShapeSelector.cpp b/src/ModuleBase/ModuleBase_WidgetShapeSelector.cpp
index 070cec76c..81e45ef0a 100644
--- a/src/ModuleBase/ModuleBase_WidgetShapeSelector.cpp
+++ b/src/ModuleBase/ModuleBase_WidgetShapeSelector.cpp
@@ -35,6 +35,7 @@
 #include <ModelAPI_Validator.h>
 #include <ModelAPI_ResultValidator.h>
 #include <ModelAPI_RefAttrValidator.h>
+#include <ModelAPI_ShapeValidator.h>
 
 #include <Config_WidgetAPI.h>
 #include <Events_Error.h>
@@ -502,6 +503,17 @@ bool ModuleBase_WidgetShapeSelector::isValid(ObjectPtr theObj, std::shared_ptr<G
         return false;
       }
     }
+    else {
+      const ModelAPI_ShapeValidator* aShapeValidator = 
+                               dynamic_cast<const ModelAPI_ShapeValidator*>(*aValidator);
+      if (aShapeValidator) {
+        DataPtr aData = myFeature->data();
+        AttributeSelectionPtr aSelectAttr = aData->selection(attributeID());
+        if (!aShapeValidator->isValid(myFeature, aSelectAttr, theShape)) {
+          return false;
+        }
+      }
+    }
   }
   return true;
 }
\ No newline at end of file
diff --git a/src/PartSet/PartSet_Module.cpp b/src/PartSet/PartSet_Module.cpp
index 253514b35..d94e396a6 100644
--- a/src/PartSet/PartSet_Module.cpp
+++ b/src/PartSet/PartSet_Module.cpp
@@ -26,6 +26,7 @@
 #include <ModelAPI_Validator.h>
 #include <ModelAPI_Data.h>
 #include <ModelAPI_Session.h>
+#include <ModelAPI_ShapeValidator.h>
 
 #include <GeomDataAPI_Point2D.h>
 #include <GeomDataAPI_Point.h>
@@ -115,6 +116,7 @@ void PartSet_Module::registerValidators()
   aFactory->registerValidator("PartSet_RadiusValidator", new PartSet_RadiusValidator);
   aFactory->registerValidator("PartSet_RigidValidator", new PartSet_RigidValidator);
   aFactory->registerValidator("PartSet_DifferentObjects", new PartSet_DifferentObjectsValidator);
+  aFactory->registerValidator("PartSet_DifferentShapes", new ModelAPI_ShapeValidator);
   aFactory->registerValidator("PartSet_SketchValidator", new PartSet_SketchValidator);
 }
 
-- 
2.39.2