Salome HOME
da64b0a5223111b692417666d218474da4ba0565
[modules/geom.git] / src / OCC2VTK / GEOM_EdgeSource.h
1 // Copyright (C) 2007-2014  CEA/DEN, EDF R&D, OPEN CASCADE
2 //
3 // This library is free software; you can redistribute it and/or
4 // modify it under the terms of the GNU Lesser General Public
5 // License as published by the Free Software Foundation; either
6 // version 2.1 of the License, or (at your option) any later version.
7 //
8 // This library is distributed in the hope that it will be useful,
9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
11 // Lesser General Public License for more details.
12 //
13 // You should have received a copy of the GNU Lesser General Public
14 // License along with this library; if not, write to the Free Software
15 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
16 //
17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
18 //
19
20 #ifndef GEOM_EDGESOURCE_H 
21 #define GEOM_EDGESOURCE_H 
22  
23 #include "OCC2VTK.h" 
24  
25 #include <TopoDS_Edge.hxx> 
26  
27 #include <vtkPoints.h> 
28 #include <vtkPolyDataAlgorithm.h> 
29
30 class vtkPolyData;
31 class EdgeSourceInternal;
32
33 class OCC2VTK_EXPORT GEOM_EdgeSource: public vtkPolyDataAlgorithm 
34
35 public: 
36   vtkTypeMacro(GEOM_EdgeSource, vtkPolyDataAlgorithm); 
37
38   static GEOM_EdgeSource* New(); 
39  
40   void AddEdge (const TopoDS_Edge& theEdge,
41                 bool theIsVector = false);
42   void Clear();
43   
44   bool IsEmpty();
45
46   void SetVectorMode(bool);
47
48   bool GetVectorMode();
49  
50   static  
51   void OCC2VTK(const TopoDS_Edge& theEdge,  
52                vtkPolyData* thePolyData, 
53                vtkPoints* thePts,
54                bool theIsVector = false); 
55  
56 protected: 
57   EdgeSourceInternal* myData;
58   // The <myIsVector> flag is common for all edges, because the shape,
59   // representing a vector, can have only one edge.
60   bool myIsVector, myIsVectorMode;
61  
62   virtual int RequestData(vtkInformation*, vtkInformationVector**, vtkInformationVector*);
63  
64   GEOM_EdgeSource(); 
65   ~GEOM_EdgeSource(); 
66  
67 private: 
68   // Not implememnted 
69   GEOM_EdgeSource(const GEOM_EdgeSource&); 
70   void operator=(const GEOM_EdgeSource&); 
71 }; 
72  
73 #endif //GEOM_EDGESOURCE_H