Salome HOME
Issue 0020904: [CEA 411] export VTK in GEOM
[modules/geom.git] / src / GEOMImpl / GEOMImpl_PlaneDriver.cxx
index 1f3e28c77ee873ecd2c5b7f8f9b34b57b6d3c67c..54dd73686ec40b67d00e5cc31fa338d5526b2791 100644 (file)
@@ -1,21 +1,23 @@
-// Copyright (C) 2005  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
-// CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
+//  Copyright (C) 2007-2010  CEA/DEN, EDF R&D, OPEN CASCADE
 //
-// This library is free software; you can redistribute it and/or
-// modify it under the terms of the GNU Lesser General Public
-// License as published by the Free Software Foundation; either
-// version 2.1 of the License.
+//  Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
+//  CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
 //
-// This library is distributed in the hope that it will be useful
-// but WITHOUT ANY WARRANTY; without even the implied warranty of
-// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
-// Lesser General Public License for more details.
+//  This library is free software; you can redistribute it and/or
+//  modify it under the terms of the GNU Lesser General Public
+//  License as published by the Free Software Foundation; either
+//  version 2.1 of the License.
 //
-// You should have received a copy of the GNU Lesser General Public
-// License along with this library; if not, write to the Free Software
-// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
+//  This library is distributed in the hope that it will be useful,
+//  but WITHOUT ANY WARRANTY; without even the implied warranty of
+//  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+//  Lesser General Public License for more details.
 //
-// See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
+//  You should have received a copy of the GNU Lesser General Public
+//  License along with this library; if not, write to the Free Software
+//  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
+//
+//  See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
 //
 
 #include <Standard_Stream.hxx>
@@ -173,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 {
   }
 
@@ -203,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;
 }