Salome HOME
Copyright update 2020
[modules/shaper.git] / src / SketcherPrs / SketcherPrs_Middle.cpp
index 67326f1561c3dc0d9a09686acac4b8f5aa0b4af7..9233c364c6b234bcf4dcf8f89258fb008f637518 100644 (file)
@@ -1,4 +1,4 @@
-// Copyright (C) 2014-2019  CEA/DEN, EDF R&D
+// Copyright (C) 2014-2020  CEA/DEN, EDF R&D
 //
 // This library is free software; you can redistribute it and/or
 // modify it under the terms of the GNU Lesser General Public
@@ -38,20 +38,20 @@ SketcherPrs_Middle::SketcherPrs_Middle(ModelAPI_Feature* theConstraint,
 }
 
 bool SketcherPrs_Middle::IsReadyToDisplay(ModelAPI_Feature* theConstraint,
-                                         const std::shared_ptr<GeomAPI_Ax3>&/* thePlane*/)
+                                         const std::shared_ptr<GeomAPI_Ax3>& thePlane)
 {
   bool aReadyToDisplay = false;
-
-  ObjectPtr aObj1 =
-    SketcherPrs_Tools::getResult(theConstraint, SketchPlugin_Constraint::ENTITY_A());
-  ObjectPtr aObj2 =
-    SketcherPrs_Tools::getResult(theConstraint, SketchPlugin_Constraint::ENTITY_B());
-
-  // one object is a feature Line, other object is a point result. We check shape of point result
-  aReadyToDisplay = aObj1.get() && aObj2.get() &&
-                    (SketcherPrs_Tools::getShape(aObj1).get() != NULL ||
-                     SketcherPrs_Tools::getShape(aObj2).get() != NULL);
-
+  if (thePlane) {
+    ObjectPtr aObj1 =
+      SketcherPrs_Tools::getResult(theConstraint, SketchPlugin_Constraint::ENTITY_A());
+    ObjectPtr aObj2 =
+      SketcherPrs_Tools::getResult(theConstraint, SketchPlugin_Constraint::ENTITY_B());
+
+    // one object is a feature Line, other object is a point result. We check shape of point result
+    aReadyToDisplay = aObj1.get() && aObj2.get() &&
+                      (SketcherPrs_Tools::getShape(aObj1).get() != NULL ||
+                       SketcherPrs_Tools::getShape(aObj2).get() != NULL);
+  }
   return aReadyToDisplay;
 }