Salome HOME
refs #1330: type/size support for polyline arrow
[modules/hydro.git] / src / HYDROGUI / HYDROGUI_Polyline.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_Polyline_HeaderFile
20 #define _HYDROGUI_Polyline_HeaderFile
21
22 #include <AIS_Shape.hxx>
23 #include <QList>
24
25 class TopoDS_Shape;
26 class TopoDS_Edge;
27
28 class HYDROGUI_Polyline : public AIS_Shape
29 {
30 public:
31   HYDROGUI_Polyline( const TopoDS_Shape& shape );
32   virtual ~HYDROGUI_Polyline();
33
34   virtual void Compute(const Handle(PrsMgr_PresentationManager3d)& aPresentationManager,
35                                        const Handle(Prs3d_Presentation)& aPresentation,
36                                        const Standard_Integer aMode = 0);
37   
38   static QList<Handle(AIS_InteractiveObject)> createPresentations( const TopoDS_Shape& shape );
39
40 public:
41   DEFINE_STANDARD_RTTIEXT(HYDROGUI_Polyline, AIS_Shape);
42 };
43
44 class HYDROGUI_Arrow : public AIS_Shape
45 {
46 public:
47   enum Type { None, Triangle, Cone };
48
49   HYDROGUI_Arrow( const TopoDS_Edge& edge );
50   virtual ~HYDROGUI_Arrow();
51
52   virtual void Compute( const Handle(PrsMgr_PresentationManager3d)& aPresentationManager,
53                                                 const Handle(Prs3d_Presentation)& aPresentation,
54                                                 const Standard_Integer aMode = 0 );
55   
56   Type GetType() const;
57   void SetType( Type );
58
59   int  GetSize() const;
60   void SetSize( int );
61
62 public:
63   DEFINE_STANDARD_RTTIEXT(HYDROGUI_Arrow, AIS_Shape);
64
65 private:
66   Type myType;
67   int  mySize;
68 };
69
70 #endif