src/VTKViewer/VTKViewer_PolyDataMapper.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 VTKViewer_POLYDATAMAPPER_H
  21 #define VTKViewer_POLYDATAMAPPER_H
  22
  23 #include "VTKViewer.h"
  24 #include "VTKViewer_MarkerDef.h"
  25
  26 #ifdef WIN32
  27 #include <windows.h>
  28 #endif
  29
  30 #include <map>
  31
  32 #include <GL/gl.h>
  33
  34 #include <vtkSmartPointer.h>
  35
  36 class vtkImageData;
  37
  38 #ifndef VTK_IMPLEMENT_MESA_CXX
  39 #include <vtkOpenGLPolyDataMapper.h>
  40 #define MAPPER_SUPERCLASS vtkOpenGLPolyDataMapper
  41 #else
  42 #include <vtkMesaPolyDataMapper.h>
  43 #define MAPPER_SUPERCLASS vtkMesaPolyDataMapper
  44 #endif
  45
  46 #ifndef GL_ARB_shader_objects
  47 typedef GLuint GLhandleARB;
  48 #endif
  49
  50
  51 //----------------------------------------------------------------------------
  52 //! OpenGL Point Sprites PolyData Mapper.
  53 /*!
  54  * VTKViewer_PolyDataMapper is a class that maps polygonal data
  55  * (i.e., vtkPolyData) to graphics primitives. It is performing the mapping
  56  * to the rendering/graphics hardware/software. It is now possible to set a
  57  * memory limit for the pipeline in the mapper. If the total estimated memory
  58  * usage of the pipeline is larger than this limit, the mapper will divide
  59  * the data into pieces and render each in a for loop.
  60  */
  61 class VTKVIEWER_EXPORT VTKViewer_PolyDataMapper : public MAPPER_SUPERCLASS
  62 {
  63 public:
  64   enum ExtensionsState { ES_None = 0, ES_Error, ES_Ok };
  65
  66 public:
  67   static VTKViewer_PolyDataMapper* New();
  68   vtkTypeMacro( VTKViewer_PolyDataMapper, MAPPER_SUPERCLASS );
  69
  70   //! Set point marker enabled
  71   void SetMarkerEnabled( bool );
  72
  73   //! Set point ball enabled
  74   void SetBallEnabled( bool );
  75
  76   bool GetBallEnabled( );
  77
  78   //! Set ball scale factor
  79   void SetBallScale( double );
  80
  81   double GetBallScale( );
  82
  83   //! Set standard point marker
  84   void SetMarkerStd( VTK::MarkerType, VTK::MarkerScale );
  85
  86   //! Set custom point marker
  87   void SetMarkerTexture( int, VTK::MarkerTexture );
  88
  89   //! Get type of the point marker
  90   VTK::MarkerType GetMarkerType();
  91
  92   //! Get scale of the point marker
  93   VTK::MarkerScale GetMarkerScale();
  94
  95   //! Get texture identifier of the point marker
  96   int GetMarkerTexture();
  97
  98   //! Implement superclass render method.
  99   virtual void RenderPiece( vtkRenderer*, vtkActor* );
 100
 101   //! Draw method for OpenGL.
 102   virtual int Draw( vtkRenderer*, vtkActor* );
 103
 104 protected:
 105   VTKViewer_PolyDataMapper();
 106   ~VTKViewer_PolyDataMapper();
 107
 108   //! Initializing OpenGL extensions.
 109   int               InitExtensions();
 110
 111   //! Activate Point Sprites.
 112   void              InitPointSprites();
 113
 114   //! Deactivate Point Sprites.
 115   void              CleanupPointSprites();
 116
 117   //! Initializing textures for Point Sprites.
 118   void              InitTextures();
 119
 120   //! Initializing of the Vertex Shader.
 121   void              InitShader();
 122
 123 private:
 124   int               ExtensionsInitialized;
 125
 126   GLuint            PointSpriteTexture;
 127
 128   vtkSmartPointer<vtkImageData> ImageData;
 129
 130   GLhandleARB       VertexProgram;
 131
 132   bool              MarkerEnabled;
 133   bool              BallEnabled;
 134   double            BallScale;
 135   VTK::MarkerType   MarkerType;
 136   VTK::MarkerScale  MarkerScale;
 137   int               MarkerId;
 138
 139   typedef std::map< int, vtkSmartPointer<vtkImageData> > ImageDataMap;
 140   ImageDataMap      StandardTextures;
 141   ImageDataMap      CustomTextures;
 142   ImageDataMap      SpecialTextures;  //special predefined textures, used to draw point sprites.
 143 };
 144
 145 #endif