src/HYDROGUI/HYDROGUI_VTKPrsShape.h

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