Fix for Bug IPAL8945

[modules/visu.git] / src / PIPELINE / VISU_VectorsPL.hxx

diff --git a/src/PIPELINE/VISU_VectorsPL.hxx b/src/PIPELINE/VISU_VectorsPL.hxx
index 81a1f33b5d3cacf2b45418c0021bd00ae3dd5db4..312d56fc989a28362980c66a4054484da6ad7f5b 100644 (file)
--- a/src/PIPELINE/VISU_VectorsPL.hxx
+++ b/src/PIPELINE/VISU_VectorsPL.hxx
@@ -29,8 +29,14 @@
 
 #include "VISU_DeformedShapePL.hxx"
 
-class SALOME_TransformFilter;
-class SALOME_Transform;
+class VTKViewer_TransformFilter;
+class VTKViewer_Transform;
+class vtkCellCenters;
+
+class vtkGlyphSource2D;
+class vtkConeSource;
+class vtkLineSource;
+
 class vtkGlyph3D;
 
 class VISU_VectorsPL : public VISU_DeformedShapePL{
@@ -55,20 +61,31 @@ public:
   virtual GlyphPos GetGlyphPos() { return myPosGlyph;}
 
 public:
-  virtual THook* DoHook();
   virtual void Init();
   virtual void Update();
 
-  virtual void SetTransform(SALOME_Transform* theTransform); 
-  virtual SALOME_Transform* GetTransform();
+  vtkDataSet* 
+  GetOutput();
+
+  virtual void SetTransform(VTKViewer_Transform* theTransform); 
+  virtual VTKViewer_Transform* GetTransform();
   
   virtual void SetMapScale(float theMapScale = 1.0);
 
 protected:
+  virtual THook* DoHook();
+
   GlyphType myTypeGlyph;
   GlyphPos myPosGlyph;
-  vtkGlyph3D *myGlyph;
-  SALOME_TransformFilter *myTransformFilter;
+  vtkGlyph3D *myBaseGlyph;
+  vtkGlyph3D *myTransformedGlyph;
+
+  vtkGlyphSource2D *myGlyphSource;
+  vtkLineSource *myLineSource;
+  vtkConeSource *myConeSource;
+
+  vtkCellCenters* myCenters;
+  VTKViewer_TransformFilter *myTransformFilter;
 };