lib/cpPlugins/Interface/BaseMPRWidget.h

   1 #ifndef __CPPLUGINS__INTERFACE__BASEMPRWIDGET__H__
   2 #define __CPPLUGINS__INTERFACE__BASEMPRWIDGET__H__
   3
   4 #include <cpPlugins/Interface/cpPlugins_Interface_Export.h>
   5 #include <cpPlugins/Interface/Config.h>
   6
   7 #ifdef cpPlugins_Interface_QT4
   8
   9 #include <map>
  10 #include <string>
  11
  12 #include <QWidget>
  13
  14 #include <vtkSmartPointer.h>
  15 #include <vtkImageData.h>
  16 #include <vtkPolyDataMapper.h>
  17 #include <vtkPolyDataNormals.h>
  18 #include <vtkQuadricLODActor.h>
  19 #include <vtkStripper.h>
  20
  21 #include <cpExtensions/Visualization/MPRObjects.h>
  22
  23 // -------------------------------------------------------------------------
  24 class QTreeWidgetItem;
  25 class QVTKWidget;
  26
  27 namespace Ui
  28 {
  29   class BaseMPRWidget;
  30 }
  31
  32 // -------------------------------------------------------------------------
  33 namespace cpPlugins
  34 {
  35   namespace Interface
  36   {
  37     // Some forward declarations
  38     class DataObject;
  39
  40     /**
  41      */
  42     class cpPlugins_Interface_EXPORT BaseMPRWidget
  43       : public QWidget
  44     {
  45       Q_OBJECT;
  46
  47     public:
  48       typedef BaseMPRWidget Self;
  49       typedef cpExtensions::Visualization::MPRObjects TMPRObjects;
  50
  51       typedef TMPRObjects::TStyle                   TStyle;
  52       typedef TMPRObjects::TMouseCommand            TMouseCommand;
  53       typedef TMPRObjects::TMouseWheelCommand       TMouseWheelCommand;
  54       typedef TMPRObjects::TKeyCommand              TKeyCommand;
  55       typedef TMPRObjects::TVoidCommand             TVoidCommand;
  56       typedef TMPRObjects::TMouseMoveCommand        TMouseMoveCommand;
  57       typedef TMPRObjects::TMouseClickCommand       TMouseClickCommand;
  58       typedef TMPRObjects::TMouseDoubleClickCommand TMouseDoubleClickCommand;
  59       typedef TMPRObjects::TExposeCommand           TExposeCommand;
  60       typedef TMPRObjects::TConfigureCommand        TConfigureCommand;
  61       typedef TMPRObjects::TEnterCommand            TEnterCommand;
  62       typedef TMPRObjects::TLeaveCommand            TLeaveCommand;
  63
  64     public:
  65       explicit BaseMPRWidget( QWidget* parent = 0 );
  66       virtual ~BaseMPRWidget( );
  67
  68       // Data management
  69       bool AddData(
  70         DataObject* data, const std::string& name,
  71         const std::string& parent
  72         );
  73       const std::string& GetMainImage( ) const;
  74       bool SetMainImage( const std::string& name );
  75       void DeleteData( const std::string& name );
  76       void DeleteAllData( );
  77
  78       // Some visualization accessors
  79       void SetDataColor(
  80         const std::string& name,
  81         const double& r, const double& g, const double& b
  82         );
  83       void ShowData( const std::string& name );
  84       void HideData( const std::string& name );
  85
  86       vtkRenderWindowInteractor* GetInteractor( unsigned int i );
  87
  88       // Visual objects accessors
  89       std::string GetSelectedData( ) const;
  90
  91     protected:
  92       QTreeWidgetItem* _FindItemInTree( const std::string& name ) const;
  93       QTreeWidgetItem* _UpdateTreeItem(
  94         const std::string& name,
  95         const std::string& parent
  96         );
  97
  98     private slots:
  99       void _SyncBottom( int a, int b );
 100       void _SyncTop( int a, int b );
 101
 102     protected:
 103       static double cm_Colors[ 8 ][ 3 ];
 104
 105       Ui::BaseMPRWidget*             m_UI;
 106       vtkSmartPointer< TMPRObjects > m_MPRObjects;
 107       QVTKWidget*                    m_VTK[ 4 ];
 108
 109       struct PolyDataActor
 110       {
 111         vtkPolyData*        Mesh;
 112         vtkPolyDataNormals* Normals;
 113         vtkStripper*        Stripper;
 114         vtkPolyDataMapper*  Mapper;
 115         vtkQuadricLODActor* Actor;
 116
 117         PolyDataActor( );
 118         virtual ~PolyDataActor( );
 119         void Configure( vtkPolyData* pd );
 120       };
 121
 122       struct Data
 123       {
 124         enum { IMAGE, MESH } Tag;
 125         DataObject*          Source;
 126         union
 127         {
 128           vtkImageData* Image;
 129           PolyDataActor Mesh;
 130         };
 131
 132         Data( );
 133         virtual ~Data( );
 134
 135         Data& operator=( const Data& data );
 136
 137         DataObject* GetSourceDataObject( );
 138         const DataObject* GetSourceDataObject( ) const;
 139         void SetSourceDataObject( DataObject* dobj );
 140
 141         vtkImageData* GetImage( );
 142         vtkPolyData* GetMesh( );
 143         vtkProp* GetMeshActor( );
 144       };
 145
 146       std::string m_MainImage;
 147       std::map< std::string, Data >        m_Data;
 148       std::map< std::string, std::string > m_Tree;
 149     };
 150
 151   } // ecapseman
 152
 153 } // ecapseman
 154
 155 #endif // cpPlugins_Interface_QT4
 156
 157 #endif // __CPPLUGINS__INTERFACE__BASEMPRWIDGET__H__
 158
 159 // eof - $RCSfile$