]> SALOME platform Git repositories - modules/hydro.git/blob - src/HYDROGUI/HYDROGUI_VTKPrsShape.h
Salome HOME
Merge branch 'BR_H2018_2' of https://codev-tuleap.cea.fr/plugins/git/salome/hydro...
[modules/hydro.git] / 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