src/HYDROGUI/HYDROGUI_VTKPrsShape.h

   1 // Copyright (C) 2007-2013  CEA/DEN, EDF R&D, OPEN CASCADE
   2 //
   3 // Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
   4 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
   5 //
   6 // This library is free software; you can redistribute it and/or
   7 // modify it under the terms of the GNU Lesser General Public
   8 // License as published by the Free Software Foundation; either
   9 // version 2.1 of the License.
  10 //
  11 // This library is distributed in the hope that it will be useful,
  12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14 // Lesser General Public License for more details.
  15 //
  16 // You should have received a copy of the GNU Lesser General Public
  17 // License along with this library; if not, write to the Free Software
  18 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  19 //
  20 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  21 //
  22
  23 #ifndef HYDROGUI_VTK_PRS_SHAPE_H
  24 #define HYDROGUI_VTK_PRS_SHAPE_H
  25
  26 #include "HYDROGUI_VTKPrs.h"
  27
  28 #include <HYDROData_Entity.h>
  29
  30 #include <vtkScalarsToColors.h>
  31 #include <vtkWeakPointer.h>
  32 #include <vtkNew.h>
  33 #include <vtkPolyDataMapper.h>
  34
  35 #include <TopoDS_Face.hxx>
  36 #include <TopoDS_Wire.hxx>
  37 #include <TopoDS_Compound.hxx>
  38
  39 #include <QColor>
  40 #include <QString>
  41
  42 class TopoDS_Shape;
  43
  44 /*
  45   Class       : HYDROGUI_VTKPrsShape
  46   Description : Presentation for HYDRO object on the base of OCC shape
  47 */
  48 class HYDROGUI_VTKPrsShape : public HYDROGUI_VTKPrs
  49 {
  50 public:
  51   HYDROGUI_VTKPrsShape( const Handle(HYDROData_Entity)& theObject );
  52   virtual ~HYDROGUI_VTKPrsShape();
  53
  54   virtual void compute();
  55
  56   //! Get the range of colored
  57   void setLookupTable( vtkScalarsToColors* theTable ) { myLookupTable = theTable; }
  58
  59 protected:
  60   virtual vtkMapper*               mapper() { return myMapper.GetPointer(); }
  61
  62   void buildShape();
  63
  64   virtual void               setWire( const TopoDS_Wire& theWire,
  65                                       const bool         theToDisplay = true,
  66                                       const bool         theIsUpdateViewer = true );
  67
  68   virtual void               setFaces( const TopoDS_Compound& theWires,
  69                                        const bool             theToDisplay = true,
  70                                        const bool             theIsUpdateViewer = true );
  71
  72   virtual void               setFace( const TopoDS_Wire& theWire,
  73                                       const bool         theToDisplay = true,
  74                                       const bool         theIsUpdateViewer = true );
  75
  76   virtual void               setFace( const TopoDS_Face& theFace,
  77                                       const bool         theToDisplay = true,
  78                                       const bool         theIsUpdateViewer = true );
  79
  80   virtual void               setShape( const TopoDS_Shape& theShape,
  81                                        const bool          theToDisplay = true,
  82                                        const bool          theIsUpdateViewer = true );
  83 private:
  84   vtkWeakPointer< vtkScalarsToColors > myLookupTable;
  85   vtkNew< vtkPolyDataMapper >          myMapper;
  86   TopoDS_Shape                         myTopoShape;
  87   int                                  myDisplayMode;
  88 };
  89
  90 #endif