Issue 0020904: [CEA 411] export VTK in GEOM

[modules/geom.git] / src / GEOMImpl / GEOMImpl_PlaneDriver.cxx

diff --git a/src/GEOMImpl/GEOMImpl_PlaneDriver.cxx b/src/GEOMImpl/GEOMImpl_PlaneDriver.cxx
index 534fc8565e498115de8cd1b098f1b51112c7069a..54dd73686ec40b67d00e5cc31fa338d5526b2791 100644 (file)
--- a/src/GEOMImpl/GEOMImpl_PlaneDriver.cxx
+++ b/src/GEOMImpl/GEOMImpl_PlaneDriver.cxx
@@ -1,4 +1,4 @@
-//  Copyright (C) 2007-2008  CEA/DEN, EDF R&D, OPEN CASCADE
+//  Copyright (C) 2007-2010  CEA/DEN, EDF R&D, OPEN CASCADE
 //
 //  Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
 //  CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
@@ -19,6 +19,7 @@
 //
 //  See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
 //
+
 #include <Standard_Stream.hxx>
 
 #include <GEOMImpl_PlaneDriver.hxx>
@@ -174,6 +175,58 @@ Standard_Integer GEOMImpl_PlaneDriver::Execute(TFunction_Logbook& log) const
     if(aTool.IsDone())
       aShape = aTool.Shape();
   }
+  else if (aType == PLANE_2_VEC) {
+    Handle(GEOM_Function) aRefVec1 = aPI.GetVector1();
+    Handle(GEOM_Function) aRefVec2 = aPI.GetVector2();
+    TopoDS_Shape aShape1 = aRefVec1->GetValue();
+    TopoDS_Shape aShape2 = aRefVec2->GetValue();
+    if (aShape1.ShapeType() != TopAbs_EDGE ||
+        aShape2.ShapeType() != TopAbs_EDGE) return 0;
+    TopoDS_Edge aVectX = TopoDS::Edge(aShape1);
+    TopoDS_Edge aVectZ = TopoDS::Edge(aShape2);
+
+    TopoDS_Vertex VX1, VX2, VZ1, VZ2;
+    TopExp::Vertices( aVectX, VX1, VX2, Standard_True );
+    TopExp::Vertices( aVectZ, VZ1, VZ2, Standard_True );
+
+    gp_Vec aVX = gp_Vec( BRep_Tool::Pnt( VX1 ), BRep_Tool::Pnt( VX2 ) );
+    gp_Vec aVZ = gp_Vec( BRep_Tool::Pnt( VZ1 ), BRep_Tool::Pnt( VZ2 ) );
+
+    if ( aVX.Magnitude() < Precision::Confusion() || aVZ.Magnitude() < Precision::Confusion())
+      Standard_TypeMismatch::Raise("Invalid vector selected");
+
+    gp_Dir aDirX = gp_Dir( aVX.X(), aVX.Y(), aVX.Z() );
+    gp_Dir aDirZ = gp_Dir( aVZ.X(), aVZ.Y(), aVZ.Z() );
+
+    if ( aDirX.IsParallel( aDirZ, Precision::Angular() ) )
+      Standard_TypeMismatch::Raise("Parallel vectors selected");
+
+    gp_Ax3 aPlane = gp_Ax3( BRep_Tool::Pnt( VX1 ), aDirZ, aDirX );
+    BRepBuilderAPI_MakeFace aTool(aPlane, -aSize, +aSize, -aSize, +aSize);
+    if(aTool.IsDone())
+      aShape = aTool.Shape();
+  }   else if (aType == PLANE_LCS) {
+    Handle(GEOM_Function) aRef = aPI.GetLCS();
+    double anOrientation = aPI.GetOrientation();    
+    gp_Ax3 anAx3;
+    if (aRef.IsNull()) {
+      gp_Ax2 anAx2 = gp::XOY();
+      anAx3 = gp_Ax3( anAx2 );
+    } else {
+      TopoDS_Shape aRefShape = aRef->GetValue();
+      if (aRefShape.ShapeType() != TopAbs_FACE)
+        return 0;
+      anAx3 = GEOMImpl_IMeasureOperations::GetPosition(aRefShape);
+    }
+
+    if ( anOrientation == 2)
+      anAx3 = gp_Ax3(anAx3.Location(), anAx3.XDirection(), anAx3.YDirection() );
+    else if ( anOrientation == 3 )
+      anAx3 = gp_Ax3(anAx3.Location(), anAx3.YDirection(), anAx3.XDirection() );
+
+    gp_Pln aPln(anAx3);
+    aShape = BRepBuilderAPI_MakeFace(aPln, -aSize, +aSize, -aSize, +aSize).Shape();
+  }
   else {
   }
 
@@ -204,10 +257,10 @@ Standard_EXPORT Handle_Standard_Type& GEOMImpl_PlaneDriver_Type_()
 
   static Handle_Standard_Transient _Ancestors[]= {aType1,aType2,aType3,NULL};
   static Handle_Standard_Type _aType = new Standard_Type("GEOMImpl_PlaneDriver",
-                                                        sizeof(GEOMImpl_PlaneDriver),
-                                                        1,
-                                                        (Standard_Address)_Ancestors,
-                                                        (Standard_Address)NULL);
+                                                         sizeof(GEOMImpl_PlaneDriver),
+                                                         1,
+                                                         (Standard_Address)_Ancestors,
+                                                         (Standard_Address)NULL);
 
   return _aType;
 }