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 c515e2b4bf0acca26805d51b1063f2a116212c2e..0cf6d0f81de9e824e7b1b5f77e5da7585d17b5ea 100644 (file)
--- a/src/ModuleBase/ModuleBase_ViewerPrs.cpp
+++ b/src/ModuleBase/ModuleBase_ViewerPrs.cpp
@@ -6,22 +6,101 @@
 
 #include "ModuleBase_ViewerPrs.h"
 
-ModuleBase_ViewerPrs::ModuleBase_ViewerPrs()
-{
-
-}
+#include <ModuleBase_ResultPrs.h>
 
-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)
 {
-
 }
 
 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.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() && !aCSolidOwner1.IsNull())
+      isEqualOwner = (aCSolidOwner1->Shape().IsNull() && aCSolidOwner2->Shape().IsNull()) ||
+                      aCSolidOwner1->Shape().IsEqual(aCSolidOwner2->Shape());
+  }
 
+  return isEqualResult && isEqualShape && isEqualOwner && isEqualIO;
 }