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 #include <NCollection_IndexedDataMap.hxx>
  25 #include <Quantity_Color.hxx>
  26 #include <TopTools_ShapeMapHasher.hxx>
  27
  28 class TopoDS_Shape;
  29 class TopoDS_Edge;
  30
  31 class HYDROGUI_Polyline : public AIS_Shape
  32 {
  33 public:
  34   HYDROGUI_Polyline( const TopoDS_Shape& shape );
  35   virtual ~HYDROGUI_Polyline();
  36
  37   virtual void Compute(const Handle(PrsMgr_PresentationManager3d)& aPresentationManager,
  38                                        const Handle(Prs3d_Presentation)& aPresentation,
  39                                        const Standard_Integer aMode = 0);
  40
  41   static QList<Handle(AIS_InteractiveObject)> createPresentations
  42     ( const TopoDS_Shape& theShape, int theType, int theSize );
  43   static void update( const QList<Handle(AIS_InteractiveObject)>&, int theType, int theSize );
  44
  45   NCollection_IndexedDataMap<TopoDS_Shape, Quantity_Color, TopTools_ShapeMapHasher> myShapeToColor;
  46
  47   bool GetColorOfSubShape(const TopoDS_Shape& SubShape, Quantity_Color& outColor);
  48
  49 public:
  50   DEFINE_STANDARD_RTTIEXT(HYDROGUI_Polyline, AIS_Shape);
  51
  52 };
  53
  54 class HYDROGUI_Arrow : public AIS_Shape
  55 {
  56 public:
  57   enum Type { None, Triangle, Cone };
  58
  59   HYDROGUI_Arrow( const TopoDS_Edge& edge, const Handle(HYDROGUI_Polyline)& polyline );
  60   virtual ~HYDROGUI_Arrow();
  61
  62   virtual void Compute( const Handle(PrsMgr_PresentationManager3d)& aPresentationManager,
  63                                                 const Handle(Prs3d_Presentation)& aPresentation,
  64                                                 const Standard_Integer aMode = 0 );
  65
  66   virtual void BoundingBox (Bnd_Box& theBndBox) override;
  67
  68   Type GetType() const;
  69   void SetType( Type );
  70
  71   int  GetSize() const;
  72   void SetSize( int );
  73
  74   const Handle(HYDROGUI_Polyline)& getParentPolyline() const;
  75
  76 public:
  77   DEFINE_STANDARD_RTTIEXT(HYDROGUI_Arrow, AIS_Shape);
  78
  79 private:
  80   Type myType;
  81   int  mySize;
  82   Handle(HYDROGUI_Polyline) myParentPoly;
  83 };
  84
  85 #endif