PAL9166. Add possibility to make sketcher on an existing plane

diff --git a/src/GEOMImpl/GEOMImpl_ICurvesOperations.cxx b/src/GEOMImpl/GEOMImpl_ICurvesOperations.cxx
index 2bf1dfc6ebf483c967cf91c626217d941af0d47d..e80e61c792114fec1a28233627dbb0e6943b55ee 100644 (file)
--- a/src/GEOMImpl/GEOMImpl_ICurvesOperations.cxx
+++ b/src/GEOMImpl/GEOMImpl_ICurvesOperations.cxx
@@ -543,7 +543,7 @@ Handle(GEOM_Object) GEOMImpl_ICurvesOperations::MakeSketcher (const char* theCom
 
   //Add a new Sketcher function
   Handle(GEOM_Function) aFunction =
-    aSketcher->AddFunction(GEOMImpl_SketcherDriver::GetID(), SKETCHER_COMMAND);
+    aSketcher->AddFunction(GEOMImpl_SketcherDriver::GetID(), SKETCHER_NINE_DOUBLS);
   if (aFunction.IsNull()) return NULL;
 
   //Check if the function is set correctly
@@ -594,3 +594,65 @@ Handle(GEOM_Object) GEOMImpl_ICurvesOperations::MakeSketcher (const char* theCom
   SetErrorCode(OK);
   return aSketcher;
 }
+
+
+//=============================================================================
+/*!
+ *  MakeSketcherOnPlane
+ */
+//=============================================================================
+Handle(GEOM_Object) GEOMImpl_ICurvesOperations::MakeSketcherOnPlane (const char* theCommand,
+                                                                     Handle(GEOM_Object) theWorkingPlane)
+{
+  SetErrorCode(KO);
+
+  if (!theCommand) return NULL;
+  if (strcmp(theCommand, "") == 0) return NULL;
+
+  //Add a new Sketcher object
+  Handle(GEOM_Object) aSketcher = GetEngine()->AddObject(GetDocID(), GEOM_SKETCHER);
+
+  //Add a new Sketcher function
+  Handle(GEOM_Function) aFunction =
+    aSketcher->AddFunction(GEOMImpl_SketcherDriver::GetID(), SKETCHER_PLANE);
+  if (aFunction.IsNull()) return NULL;
+
+  //Check if the function is set correctly
+  if (aFunction->GetDriverGUID() != GEOMImpl_SketcherDriver::GetID()) return NULL;
+
+  GEOMImpl_ISketcher aCI (aFunction);
+
+  TCollection_AsciiString aCommand ((char*) theCommand);
+  aCI.SetCommand(aCommand);
+
+  Handle(GEOM_Function) aRefPlane = theWorkingPlane->GetLastFunction();
+  if (aRefPlane.IsNull()) return NULL;
+  aCI.SetWorkingPlane( aRefPlane );
+
+  //Compute the Sketcher value
+  try {
+    if (!GetSolver()->ComputeFunction(aFunction)) {
+      SetErrorCode("Sketcher driver failed");
+      return NULL;
+    }
+  }
+  catch (Standard_Failure) {
+    Handle(Standard_Failure) aFail = Standard_Failure::Caught();
+    SetErrorCode(aFail->GetMessageString());
+    return NULL;
+  }
+
+  //Make a Python command
+  TCollection_AsciiString anEntry, aDescr;
+  TDF_Tool::Entry(aSketcher->GetEntry(), anEntry);
+  aDescr += anEntry;
+  aDescr += " = ICurvesOperations.MakeSketcherOnPlane(";
+  aDescr += aCommand+", ";
+  TDF_Tool::Entry(theWorkingPlane->GetEntry(), anEntry);
+  aDescr += anEntry + " )";
+
+  aFunction->SetDescription(aDescr);
+
+  SetErrorCode(OK);
+  return aSketcher;
+}

diff --git a/src/GEOMImpl/GEOMImpl_ICurvesOperations.hxx b/src/GEOMImpl/GEOMImpl_ICurvesOperations.hxx
index 0c52cc7ecf56d696994dfa27a56559fc41b9506d..e421f642ea7eef8015f7febef0a1863875e4bfdb 100644 (file)
--- a/src/GEOMImpl/GEOMImpl_ICurvesOperations.hxx
+++ b/src/GEOMImpl/GEOMImpl_ICurvesOperations.hxx
@@ -35,6 +35,8 @@ class GEOMImpl_ICurvesOperations : public GEOM_IOperations {
   Handle(GEOM_Object) MakeSplineInterpolation (list<Handle(GEOM_Object)> thePoints);
 
   Handle(GEOM_Object) MakeSketcher (const char* theCommand, list<double> theWorkingPlane);
+  Handle(GEOM_Object) MakeSketcherOnPlane (const char*         theCommand,
+                                           Handle(GEOM_Object) theWorkingPlane);
 };
 
 #endif

diff --git a/src/GEOMImpl/GEOMImpl_ISketcher.hxx b/src/GEOMImpl/GEOMImpl_ISketcher.hxx
index 74912ff38b27c216382b8fb3e8734ba26d99f3f0..b32dab4d60eaef410411790d53901f311133ddb5 100644 (file)
--- a/src/GEOMImpl/GEOMImpl_ISketcher.hxx
+++ b/src/GEOMImpl/GEOMImpl_ISketcher.hxx
@@ -4,7 +4,8 @@
 #include "GEOM_Function.hxx"
 
 #define SKETCH_ARG_CMD 1
-#define SKETCH_ARG_WPLANE 2
+#define SKETCH_ARG_DOUBLE 2
+#define SKETCH_ARG_WPLANE 12
 
 class GEOMImpl_ISketcher
 {
@@ -18,9 +19,14 @@ class GEOMImpl_ISketcher
   TCollection_AsciiString GetCommand() { return _func->GetString(SKETCH_ARG_CMD); }
 
   void SetWorkingPlane(int theInd, double theValue)
-              { _func->SetReal(SKETCH_ARG_WPLANE + theInd, theValue); }
+              { _func->SetReal(SKETCH_ARG_DOUBLE + theInd, theValue); }
 
-  double GetWorkingPlane(int theInd) { return _func->GetReal(SKETCH_ARG_WPLANE + theInd); }
+  double GetWorkingPlane(int theInd) { return _func->GetReal(SKETCH_ARG_DOUBLE + theInd); }
+
+  void SetWorkingPlane(Handle(GEOM_Function) thePlane)
+              { _func->SetReference(SKETCH_ARG_WPLANE, thePlane); }
+
+  Handle(GEOM_Function) GetWorkingPlane() { return _func->GetReference(SKETCH_ARG_WPLANE); }
 
  private:
 

diff --git a/src/GEOMImpl/GEOMImpl_SketcherDriver.cxx b/src/GEOMImpl/GEOMImpl_SketcherDriver.cxx
index 0b79130b20c136899c482e88de5ee306eee6bab3..b0822626abac35137f7e45b8a1939e982d475df8 100644 (file)
--- a/src/GEOMImpl/GEOMImpl_SketcherDriver.cxx
+++ b/src/GEOMImpl/GEOMImpl_SketcherDriver.cxx
@@ -5,10 +5,12 @@ using namespace std;
 #include "GEOMImpl_Types.hxx"
 #include "GEOM_Function.hxx"
 
-#include <TopoDS_Shape.hxx>
-#include <gp_Dir.hxx>
-#include <gp_Ax3.hxx>
 #include <BRepBuilderAPI_Transform.hxx>
+#include <BRep_Tool.hxx>
+#include <Geom_Plane.hxx>
+#include <TopoDS.hxx>
+#include <TopoDS_Shape.hxx>
+#include <gp_Pln.hxx>
 
 #include <Sketcher_Profile.hxx>
 
@@ -63,12 +65,30 @@ Standard_Integer GEOMImpl_SketcherDriver::Execute(TFunction_Logbook& log) const
   if (aShape.IsNull())
     return 0;
 
+  gp_Ax3 aWPlane;
+  if ( aFunction->GetType() == SKETCHER_NINE_DOUBLS )
+  {
+    gp_Pnt aOrigin =
+      gp_Pnt(aCI.GetWorkingPlane(1), aCI.GetWorkingPlane(2), aCI.GetWorkingPlane(3));
+    gp_Dir aDirZ =
+      gp_Dir(aCI.GetWorkingPlane(4), aCI.GetWorkingPlane(5), aCI.GetWorkingPlane(6));
+    gp_Dir aDirX =
+      gp_Dir(aCI.GetWorkingPlane(7), aCI.GetWorkingPlane(8), aCI.GetWorkingPlane(9));
+    aWPlane = gp_Ax3(aOrigin, aDirZ, aDirX);
+  }
+  else
+  {
+    Handle(GEOM_Function) aRefFace = aCI.GetWorkingPlane();
+    TopoDS_Shape aShape = aRefFace->GetValue();
+    if ( aShape.IsNull() || aShape.ShapeType() != TopAbs_FACE )
+      return 0;
+    Handle(Geom_Surface) aGS = BRep_Tool::Surface( TopoDS::Face( aShape ));
+    if ( aGS.IsNull() || !aGS->IsKind( STANDARD_TYPE( Geom_Plane )))
+      return 0;
+    Handle(Geom_Plane) aGPlane = Handle(Geom_Plane)::DownCast( aGS );
+    aWPlane = aGPlane->Pln().Position();
+  }
   gp_Trsf aTrans;
-  gp_Pnt aOrigin = gp_Pnt(aCI.GetWorkingPlane(1), aCI.GetWorkingPlane(2), aCI.GetWorkingPlane(3));
-  gp_Dir aDirZ = gp_Dir(aCI.GetWorkingPlane(4), aCI.GetWorkingPlane(5), aCI.GetWorkingPlane(6));
-  gp_Dir aDirX = gp_Dir(aCI.GetWorkingPlane(7), aCI.GetWorkingPlane(8), aCI.GetWorkingPlane(9));
-  gp_Ax3 aWPlane = gp_Ax3(aOrigin, aDirZ, aDirX);
-
   aTrans.SetTransformation(aWPlane);
   aTrans.Invert();
   BRepBuilderAPI_Transform aTransformation (aShape, aTrans, Standard_False);

diff --git a/src/GEOMImpl/GEOMImpl_Types.hxx b/src/GEOMImpl/GEOMImpl_Types.hxx
index 982ef59ae597379631c9ebbe2df3f8585152e15c..d3ede4bd558064ade97ed097de4bce7b7868f6b9 100755 (executable)
--- a/src/GEOMImpl/GEOMImpl_Types.hxx
+++ b/src/GEOMImpl/GEOMImpl_Types.hxx
@@ -188,7 +188,8 @@
 
 #define GLUE_FACES 1
 
-#define SKETCHER_COMMAND 1
+#define SKETCHER_NINE_DOUBLS 1
+#define SKETCHER_PLANE 2
 
 #define CDG_MEASURE 1