Better management of errors.

[modules/shaper.git] / src / ModuleBase / ModuleBase_ViewerPrs.cpp

diff --git a/src/ModuleBase/ModuleBase_ViewerPrs.cpp b/src/ModuleBase/ModuleBase_ViewerPrs.cpp
index deb08fd53e92d2ac1d750b7214a2935cacb89ad7..0cf6d0f81de9e824e7b1b5f77e5da7585d17b5ea 100644 (file)
--- a/src/ModuleBase/ModuleBase_ViewerPrs.cpp
+++ b/src/ModuleBase/ModuleBase_ViewerPrs.cpp
@@ -8,14 +8,9 @@
 
 #include <ModuleBase_ResultPrs.h>
 
-ModuleBase_ViewerPrs::ModuleBase_ViewerPrs()
-{
-
-}
-
-ModuleBase_ViewerPrs::ModuleBase_ViewerPrs(ObjectPtr theResult, 
-                                           const TopoDS_Shape& theShape, 
-                                           Handle_SelectMgr_EntityOwner theOwner) 
+ModuleBase_ViewerPrs::ModuleBase_ViewerPrs(ObjectPtr theResult,
+                                           const GeomShapePtr& theShape,
+                                           Handle_SelectMgr_EntityOwner theOwner)
 : myResult(theResult),
   myShape(theShape),
   myOwner(theOwner)
@@ -26,13 +21,59 @@ ModuleBase_ViewerPrs::~ModuleBase_ViewerPrs()
 {
 }
 
+bool ModuleBase_ViewerPrs::isEqual(ModuleBase_ViewerPrs* thePrs) const
+{
+  if (!thePrs || thePrs->isEmpty())
+    return false;
+
+  bool isEqualResult = (myResult.get() == thePrs->object().get());
+  bool isEqualShape = (!myShape.get() && !thePrs->shape().get()) ||
+                       (myShape.get() && myShape->isEqual(thePrs->shape()));
+  bool isEqualIO = (myInteractive == thePrs->interactive()) == Standard_True;
+
+  bool isEqualOwner = myOwner.get() == thePrs->owner().get();
+  if (isEqualResult && isEqualShape && isEqualIO &&
+      !isEqualOwner) { /// owners are different
+    // as we might loading object with the same shape in different modes like
+    // "objects" and "other", it is possible that two owners are created linked
+    // to one shape. We should accept such ViewerPrs as equal to current in order
+    // to do not use the same <result, shape> twice
+    Handle(StdSelect_BRepOwner) anOwner1 = Handle(StdSelect_BRepOwner)::DownCast(myOwner);
+    Handle(StdSelect_BRepOwner) anOwner2 = Handle(StdSelect_BRepOwner)::DownCast(thePrs->owner());
+    if (!anOwner1.IsNull() && !anOwner2.IsNull())
+      isEqualOwner = (anOwner1->Shape().IsNull() && anOwner2->Shape().IsNull()) ||
+                      anOwner1->Shape().IsEqual(anOwner2->Shape());
+  }
+
+  if (isEqualResult && isEqualShape &&
+      !isEqualIO) { /// AIS are different
+    // check that the owner is a fictive owner for compsolid object, created in the
+    // ComputeSelection of ModuleBase_ResultPrs. A new owner is created there for each subsolid
+    // and set in the sub-solid AIS. ViewerPrs of these fictive owners are accepted as equal
+    // as they use the same shape and result(of compsolid)
+    Handle(StdSelect_BRepOwner) aCSolidOwner1 = Handle(StdSelect_BRepOwner)::DownCast(myOwner);
+    Handle(StdSelect_BRepOwner) aCSolidOwner2 =
+      Handle(StdSelect_BRepOwner)::DownCast(thePrs->owner());
+    isEqualIO = !aCSolidOwner1.IsNull() && !aCSolidOwner2.IsNull();
+    if (!aCSolidOwner1.IsNull() && !aCSolidOwner2.IsNull())
+      isEqualOwner = (aCSolidOwner1->Shape().IsNull() && aCSolidOwner2->Shape().IsNull()) ||
+                      aCSolidOwner1->Shape().IsEqual(aCSolidOwner2->Shape());
+  }
+
+  return isEqualResult && isEqualShape && isEqualOwner && isEqualIO;
+}
+
 bool ModuleBase_ViewerPrs::operator==(const ModuleBase_ViewerPrs& thePrs)
 {
+  if (thePrs.isEmpty())
+    return false;
+
   bool isEqualResult = (myResult.get() == thePrs.object().get());
-  bool isEqualShape = myShape.IsEqual(thePrs.shape()) == Standard_True;
+  bool isEqualShape = (!myShape.get() && !thePrs.shape().get()) ||
+                       (myShape.get() && myShape->isEqual(thePrs.shape()));
   bool isEqualIO = (myInteractive == thePrs.interactive()) == Standard_True;
 
-  bool isEqualOwner = (myOwner.Access() == thePrs.owner().Access());
+  bool isEqualOwner = myOwner.get() == thePrs.owner().get();
   if (isEqualResult && isEqualShape && isEqualIO &&
       !isEqualOwner) { /// owners are different
     // as we might loading object with the same shape in different modes like
@@ -42,7 +83,8 @@ bool ModuleBase_ViewerPrs::operator==(const ModuleBase_ViewerPrs& thePrs)
     Handle(StdSelect_BRepOwner) anOwner1 = Handle(StdSelect_BRepOwner)::DownCast(myOwner);
     Handle(StdSelect_BRepOwner) anOwner2 = Handle(StdSelect_BRepOwner)::DownCast(thePrs.owner());
     if (!anOwner1.IsNull() && !anOwner2.IsNull())
-      isEqualOwner = anOwner1->Shape() == anOwner2->Shape();
+      isEqualOwner = (anOwner1->Shape().IsNull() && anOwner2->Shape().IsNull()) ||
+                      anOwner1->Shape().IsEqual(anOwner2->Shape());
   }
 
   if (isEqualResult && isEqualShape &&
@@ -51,11 +93,13 @@ bool ModuleBase_ViewerPrs::operator==(const ModuleBase_ViewerPrs& thePrs)
     // ComputeSelection of ModuleBase_ResultPrs. A new owner is created there for each subsolid
     // and set in the sub-solid AIS. ViewerPrs of these fictive owners are accepted as equal
     // as they use the same shape and result(of compsolid)
-    Handle(ModuleBase_BRepOwner) aCSolidOwner1 = Handle(ModuleBase_BRepOwner)::DownCast(myOwner);
-    Handle(ModuleBase_BRepOwner) aCSolidOwner2 = Handle(ModuleBase_BRepOwner)::DownCast(thePrs.owner());
+    Handle(StdSelect_BRepOwner) aCSolidOwner1 = Handle(StdSelect_BRepOwner)::DownCast(myOwner);
+    Handle(StdSelect_BRepOwner) aCSolidOwner2 =
+      Handle(StdSelect_BRepOwner)::DownCast(thePrs.owner());
     isEqualIO = !aCSolidOwner1.IsNull() && !aCSolidOwner2.IsNull();
     if (!aCSolidOwner1.IsNull() && !aCSolidOwner1.IsNull())
-      isEqualOwner = aCSolidOwner1->Shape() == aCSolidOwner1->Shape();
+      isEqualOwner = (aCSolidOwner1->Shape().IsNull() && aCSolidOwner2->Shape().IsNull()) ||
+                      aCSolidOwner1->Shape().IsEqual(aCSolidOwner2->Shape());
   }
 
   return isEqualResult && isEqualShape && isEqualOwner && isEqualIO;