refs #1330: draft preferences implementation for polyline arrow

[modules/hydro.git] / src / HYDROGUI / HYDROGUI_Polyline.h

diff --git a/src/HYDROGUI/HYDROGUI_Polyline.h b/src/HYDROGUI/HYDROGUI_Polyline.h
index 130dc8791408562bc5206de720a65e8bfc0f6c5a..ca41ae7cb827eadb74199a2b48d07e96a52f48bb 100644 (file)
--- a/src/HYDROGUI/HYDROGUI_Polyline.h
+++ b/src/HYDROGUI/HYDROGUI_Polyline.h
@@ -35,7 +35,9 @@ public:
                                       const Handle(Prs3d_Presentation)& aPresentation,
                                       const Standard_Integer aMode = 0);
   
-  static QList<Handle(AIS_InteractiveObject)> createPresentations( const TopoDS_Shape& shape );
+  static QList<Handle(AIS_InteractiveObject)> createPresentations
+    ( const TopoDS_Shape& theShape, int theType, int theSize );
+  static void update( const QList<Handle(AIS_InteractiveObject)>&, int theType, int theSize );
 
 public:
   DEFINE_STANDARD_RTTIEXT(HYDROGUI_Polyline, AIS_Shape);
@@ -44,6 +46,8 @@ public:
 class HYDROGUI_Arrow : public AIS_Shape
 {
 public:
+  enum Type { None, Triangle, Cone };
+
   HYDROGUI_Arrow( const TopoDS_Edge& edge );
   virtual ~HYDROGUI_Arrow();
 
@@ -51,8 +55,18 @@ public:
                                                const Handle(Prs3d_Presentation)& aPresentation,
                                                const Standard_Integer aMode = 0 );
   
+  Type GetType() const;
+  void SetType( Type );
+
+  int  GetSize() const;
+  void SetSize( int );
+
 public:
   DEFINE_STANDARD_RTTIEXT(HYDROGUI_Arrow, AIS_Shape);
+
+private:
+  Type myType;
+  int  mySize;
 };
 
 #endif