src/HYDROGUI/HYDROGUI_PolylineExtractionOp.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_POLYLINEEXTRACTIONOP_H
  20 #define HYDROGUI_POLYLINEEXTRACTIONOP_H
  21
  22 #include "HYDROGUI_Operation.h"
  23
  24 #include <HYDROData_Stream.h>
  25 #include <HYDROData_Channel.h>
  26 #include <HYDROData_Obstacle.h>
  27
  28 #include <NCollection_DataMap.hxx>
  29
  30 class HYDROGUI_PolylineExtractionOp : public HYDROGUI_Operation
  31 {
  32   Q_OBJECT
  33
  34 public:
  35
  36   HYDROGUI_PolylineExtractionOp( HYDROGUI_Module* theModule );
  37   virtual ~HYDROGUI_PolylineExtractionOp();
  38
  39 protected:
  40   virtual void               startOperation();
  41
  42 private:
  43   class ShapeInfo
  44   {
  45   public:
  46       ShapeInfo( const TopoDS_Shape s, const QString& n ) : myShape( s ), myName( n ) {}
  47       TopoDS_Shape shape() const { return myShape; }
  48       QString      name() const { return myName; }
  49
  50   private:
  51       TopoDS_Shape myShape;
  52       QString      myName;
  53   };
  54
  55   typedef QList<ShapeInfo> ShapeInfoList;
  56
  57 private:
  58   ShapeInfoList              extract( const Handle(HYDROData_Stream)& ) const;
  59   ShapeInfoList              extract( const Handle(HYDROData_Channel)& ) const;
  60   ShapeInfoList              extract( const Handle(HYDROData_Obstacle)& ) const;
  61 };
  62
  63 #endif