Salome HOME
OCC functionality moving out from the widget
[modules/hydro.git] / src / HYDROCurveCreator / CurveCreator_Utils.cxx
index 30779616cf2d7830e0c04fa8af3256aa9d90f765..73b429b446db6a5a20047c7bf34a2f5c4d1163e4 100644 (file)
 
 #include "CurveCreator_Utils.h"
 
+#include <GEOMUtils.hxx>
+
 #include <gp_Pln.hxx>
+
+#include <TopoDS.hxx>
+#include <TopoDS_Vertex.hxx>
+#include <TopoDS_Wire.hxx>
+#include <TopoDS_Edge.hxx>
+
+#include <AIS_ListOfInteractive.hxx>
+#include <AIS_ListIteratorOfListOfInteractive.hxx>
+#include <AIS_Shape.hxx>
+#include <AIS_Point.hxx>
+#include <AIS_Line.hxx>
+#include <AIS_Trihedron.hxx>
+
+#include <Geom_Point.hxx>
+#include <Geom_BSplineCurve.hxx>
+#include <Geom_Line.hxx>
+
+#include <TopExp_Explorer.hxx>
+#include <BRep_Tool.hxx>
+#include <GeomAPI_ProjectPointOnCurve.hxx>
+#include <SelectMgr_EntityOwner.hxx>
+
 #include <ProjLib.hxx>
 #include <ElSLib.hxx>
 
+const double LOCAL_SELECTION_TOLERANCE = 0.0001;
+const int    SCENE_PIXEL_TOLERANCE = 10;
+
+//=======================================================================
+// function : ConvertClickToPoint()
+// purpose  : Returns the point clicked in 3D view
+//=======================================================================
+void CurveCreator_Utils::ConvertPointToClick( const gp_Pnt& thePoint,
+                                              Handle(V3d_View) theView,
+                                              int& x, int& y )
+{
+  theView->Convert(thePoint.X(), thePoint.Y(), thePoint.Z(), x, y );
+}
+
 
 //=======================================================================
 // function : ConvertClickToPoint()
 //=======================================================================
 gp_Pnt CurveCreator_Utils::ConvertClickToPoint( int x, int y, Handle(V3d_View) aView )
 {
-  V3d_Coordinate XEye, YEye, ZEye, XAt, YAt, ZAt;
-  aView->Eye( XEye, YEye, ZEye );
+  return GEOMUtils::ConvertClickToPoint( x, y, aView );
+}
+
+std::list<float> CurveCreator_Utils::getSelectedPoints( Handle(AIS_InteractiveContext) theContext )
+{
+  std::list<float> aSelectedPoints;
+
+  gp_Pnt aPnt;
+  for ( theContext->InitSelected(); theContext->MoreSelected(); theContext->NextSelected() ) {
+    TopoDS_Vertex aVertex;
+    TopoDS_Shape aShape = theContext->SelectedShape();
+    if ( !aShape.IsNull() && aShape.ShapeType() == TopAbs_VERTEX )
+      aVertex = TopoDS::Vertex( theContext->SelectedShape() );
+    else {
+      Handle(SelectMgr_EntityOwner) anOwner = theContext->SelectedOwner();
+      if ( !anOwner.IsNull() ) {
+        Handle(AIS_InteractiveObject) anAIS = Handle(AIS_InteractiveObject)::DownCast( anOwner->Selectable() );
+        if ( !anAIS.IsNull() ) {
+          Handle(AIS_Point) aPoint = Handle(AIS_Point)::DownCast( anAIS);
+          if ( !aPoint.IsNull() )
+            aVertex = TopoDS::Vertex( aPoint->Vertex() );
+        }
+        if ( aVertex.IsNull() ) {
+          // the following happens if there are no points in the current curve, there is only a shape
+          /*Handle(StdSelect_BRepOwner) aBrepOwner = Handle(StdSelect_BRepOwner)::DownCast(anOwner);
+          if ( aBrepOwner.IsNull() )
+            continue;
+          if ( aBrepOwner->HasShape() ) {
+            const TopoDS_Shape& aShape = aBrepOwner->Shape();
+            if ( !aShape.IsNull() && aShape.ShapeType() == TopAbs_VERTEX )
+            {
+              aVertex = TopoDS::Vertex( aShape );
+            }
+          }*/
+        }
+      }
+    }
+    if ( aVertex.IsNull() )
+      continue;
+    aPnt = BRep_Tool::Pnt( aVertex );
+    aSelectedPoints.push_back( aPnt.X() );
+    aSelectedPoints.push_back( aPnt.Y() );
+    aSelectedPoints.push_back( aPnt.Z() );
+  }
+
+  return aSelectedPoints;
+}
+//=======================================================================
+// function : setLocalPointContext
+// purpose  : Open/close the viewer local context
+//=======================================================================
+void CurveCreator_Utils::setLocalPointContext(
+                                              Handle(AIS_InteractiveContext) theContext,
+                                              const bool theOpen )
+{
+  if ( !theContext )
+    return;
+
+  if ( theOpen ) {
+    // Open local context if there is no one
+    if ( !theContext->HasOpenedContext() ) {
+      theContext->ClearCurrents( false );
+      theContext->OpenLocalContext( false/*use displayed objects*/, true/*allow shape decomposition*/ );
+    }
+    AIS_ListOfInteractive aList;
+    theContext->DisplayedObjects( aList );
+    int aLSize = 0;
+    for ( AIS_ListIteratorOfListOfInteractive it( aList ); it.More(); it.Next() )
+      aLSize++;
+
+    for ( AIS_ListIteratorOfListOfInteractive it( aList ); it.More(); it.Next() )
+    {
+      Handle(AIS_InteractiveObject) anAIS = it.Value();
+      if ( !anAIS.IsNull() )
+      {
+        if ( anAIS->IsKind( STANDARD_TYPE( AIS_Shape ) ) )
+        {
+          theContext->Load( anAIS, -1/*selection mode*/, true/*allow decomposition*/ );
+          theContext->Activate( anAIS, AIS_Shape::SelectionMode( (TopAbs_ShapeEnum)TopAbs_WIRE ) );
+          //theContext->Activate( anAIS, AIS_Shape::SelectionMode( (TopAbs_ShapeEnum)TopAbs_VERTEX ) );
+        }
+        else if ( anAIS->DynamicType() != STANDARD_TYPE(AIS_Trihedron) )
+        {
+          theContext->Load( anAIS, -1/*selection mode*/, false/*allow decomposition*/ );
+          theContext->Activate( anAIS, TopAbs_VERTEX );
+        }
+      }
+      continue;
+    }
+  }
+  else {
+    if ( theContext->HasOpenedContext() )
+      theContext->CloseAllContexts();
+  }
+}
+
+bool CurveCreator_Utils::getNeighbourPoints( Handle(AIS_InteractiveContext) theContext,
+                                             Handle(V3d_View) theView,
+                                             const int theX, const int theY,
+                                             gp_Pnt& thePoint, gp_Pnt& thePoint1,
+                                             gp_Pnt& thePoint2 )
+{
+  bool isFoundPoint = false;
+  if ( theContext.IsNull() )
+    return isFoundPoint;
+
+  for ( theContext->InitSelected(); theContext->MoreSelected() && !isFoundPoint;
+        theContext->NextSelected() ) {
+    TopoDS_Shape aTShape = theContext->SelectedShape();
+    if ( !aTShape.IsNull() && aTShape.ShapeType() == TopAbs_VERTEX )
+      continue;
+    else {
+      Handle(SelectMgr_EntityOwner) anOwner = theContext->SelectedOwner();
+      if ( anOwner.IsNull() )
+        continue;
+      const TopLoc_Location& aLocation = anOwner->Location();
+      Handle(AIS_InteractiveObject) anAIS =
+                        Handle(AIS_InteractiveObject)::DownCast( anOwner->Selectable() );
+      isFoundPoint = CurveCreator_Utils::pointOnObject( theView, anAIS, theX, theY, thePoint,
+                                                        thePoint1, thePoint2 );
+    }
+  }
+  return isFoundPoint;
+}
+
+bool CurveCreator_Utils::pointOnObject( Handle(V3d_View) theView,
+                                        Handle(AIS_InteractiveObject) theObject,
+                                        const int theX, const int theY,
+                                        gp_Pnt& thePoint,
+                                        gp_Pnt& thePoint1, gp_Pnt& thePoint2 )
+{
+  bool isFound = false;
+
+  if ( theObject.IsNull() || theView.IsNull() )
+    return isFound;
+
+  gp_Pnt aPoint;
+  Standard_Real aParameter;
+  gp_Pnt aPnt1, aPnt2;
+  Handle(AIS_Line) aLine = Handle(AIS_Line)::DownCast( theObject );
+  if ( !aLine.IsNull() ) {
+    const Handle(Geom_Line) aGLine = aLine->Line();
+    isFound = hasProjectPointOnCurve( theView, theX, theY, aGLine, aParameter );
+    if ( isFound ) {
+      aPoint = aGLine->Value( aParameter );
+
+      Handle(Geom_Point) aPStart;
+      Handle(Geom_Point) aPEnd;
+      aLine->Points( aPStart, aPEnd );
+      aPnt1 = aPStart->Pnt();
+      aPnt2 = aPEnd->Pnt();
+
+      // in case of Geom line a projection is performed to the infinite line,
+      // so it is necessary to bound it by the line size
+      Bnd_Box aLineBox;
+      aLineBox.Set( aPnt1, gp_Vec( aPnt1, aPnt2 ) );
+      isFound = !aLineBox.IsOut( aPoint );
+    }
+  }
+  else {
+    Handle(AIS_Shape) aShape = Handle(AIS_Shape)::DownCast( theObject );
+    if ( !aShape.IsNull() ) {
+      const TopoDS_Wire& aWire = TopoDS::Wire( aShape->Shape() );
+      if ( !aWire.IsNull() ) {
+        TopExp_Explorer anExp( aWire, TopAbs_EDGE );
+        for ( ; anExp.More(); anExp.Next())
+        {
+          const TopoDS_Edge& anEdge = TopoDS::Edge(anExp.Current());
+          if ( !anEdge.IsNull() ) {
+            Standard_Real aFirst, aLast;
+            Handle(Geom_Curve) aCurve = BRep_Tool::Curve( anEdge, aFirst, aLast );
+
+            if ( aCurve->IsKind( STANDARD_TYPE(Geom_BSplineCurve) ) ) {
+              Handle(Geom_BSplineCurve) aBSplineCurve = Handle(Geom_BSplineCurve)::DownCast( aCurve );
+              if ( !aBSplineCurve.IsNull() ) {
+                isFound = hasProjectPointOnCurve( theView, theX, theY, aCurve, aParameter );
+                if ( isFound ) {
+                  aPoint = aBSplineCurve->Value( aParameter );
+                  Standard_Integer anI1, anI2;
+                  aBSplineCurve->LocateU( aParameter, LOCAL_SELECTION_TOLERANCE, anI1, anI2 );
+
+                  aPnt1 = aBSplineCurve->Value( aBSplineCurve->Knot( anI1 ) );
+                  aPnt2 = aBSplineCurve->Value( aBSplineCurve->Knot( anI2 ) );
+                }
+              }
+            }
+          }
+        }
+      }
+    }
+  }
+  if ( isFound ) {
+    thePoint = aPoint;
+    thePoint1 = aPnt1;
+    thePoint2 = aPnt2;
+  }
+  return isFound;
+}
+
+bool CurveCreator_Utils::hasProjectPointOnCurve( Handle(V3d_View) theView,
+                                                 const int theX, const int theY,
+                                                 const Handle(Geom_Curve)& theCurve,
+                                                 Standard_Real& theParameter )
+{
+  bool isFound = false;
+  if ( theView.IsNull() )
+    return isFound;
 
-  aView->At( XAt, YAt, ZAt );
-  gp_Pnt EyePoint( XEye, YEye, ZEye );
-  gp_Pnt AtPoint( XAt, YAt, ZAt );
+  gp_Pnt aPoint = CurveCreator_Utils::ConvertClickToPoint( theX, theY, theView );
 
-  gp_Vec EyeVector( EyePoint, AtPoint );
-  gp_Dir EyeDir( EyeVector );
+  GeomAPI_ProjectPointOnCurve aProj( aPoint, theCurve );
+  Standard_Integer aNbPoint = aProj.NbPoints();
+  if (aNbPoint > 0) {
+    for (Standard_Integer j = 1; j <= aNbPoint && !isFound; j++) {
+      gp_Pnt aNewPoint = aProj.Point( j );
+      theParameter = aProj.Parameter( j );
 
-  gp_Pln PlaneOfTheView = gp_Pln( AtPoint, EyeDir );
-  Standard_Real X, Y, Z;
-  aView->Convert( x, y, X, Y, Z );
-  gp_Pnt ConvertedPoint( X, Y, Z );
+      int aX, anY;
+      CurveCreator_Utils::ConvertPointToClick( aNewPoint, theView, aX, anY );
 
-  gp_Pnt2d ConvertedPointOnPlane = ProjLib::Project( PlaneOfTheView, ConvertedPoint );
-  gp_Pnt ResultPoint = ElSLib::Value( ConvertedPointOnPlane.X(), ConvertedPointOnPlane.Y(), PlaneOfTheView );
-  return ResultPoint;
+      int aXDelta = abs( aX - theX );
+      int anYDelta = abs( anY - theY );
+      isFound = aXDelta < SCENE_PIXEL_TOLERANCE && anYDelta < SCENE_PIXEL_TOLERANCE;
+    }
+  }
+  return isFound;
 }