us3DSceneWidget.h

/****************************************************************************
 *
 * This file is part of the ustk software.
 * Copyright (C) 2016 - 2017 by Inria. All rights reserved.
 *
 * This software is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * ("GPL") version 2 as published by the Free Software Foundation.
 * See the file LICENSE.txt at the root directory of this source
 * distribution for additional information about the GNU GPL.
 *
 * For using ustk with software that can not be combined with the GNU
 * GPL, please contact Inria about acquiring a ViSP Professional
 * Edition License.
 *
 * This software was developed at:
 * Inria Rennes - Bretagne Atlantique
 * Campus Universitaire de Beaulieu
 * 35042 Rennes Cedex
 * France
 *
 * If you have questions regarding the use of this file, please contact
 * Inria at ustk@inria.fr
 *
 * This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING THE
 * WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
 *
 * Authors:
 * Marc Pouliquen
 *
 *****************************************************************************/
 
#ifndef __us3DSceneWidget_h_
#define __us3DSceneWidget_h_
 
// VISP includes
#include <visp3/ustk_core/usConfig.h>
 
#ifdef USTK_HAVE_VTK_QT
 
#include <visp3/ustk_gui/usViewerWidget.h>
 
// VTK includes
#include <vtkAxesActor.h>
#include <vtkCubeSource.h>
#include <vtkCutter.h>
#include <vtkImageData.h>
#include <vtkImagePlaneWidget.h>
#include <vtkImageResliceMapper.h>
#include <vtkImageSlice.h>
#include <vtkPlane.h>
#include <vtkPolyDataMapper.h>
#include <vtkRenderWindow.h>
#include <vtkRenderer.h>
#include <vtkSmartPointer.h>
 
// Qt includes
#if defined(USTK_HAVE_VTK_QT4)
#include <QApplication>
#include <QtGui/QGridLayout>
#include <QtGui/QMainWindow>
#include <QtGui/QPushButton>
#elif defined USTK_HAVE_VTK_QT5
#include <QApplication>
#include <QtWidgets/QGridLayout>
#include <QtWidgets/QMainWindow>
#include <QtWidgets/QPushButton>
#endif
 
#if USTK_HAVE_VTK_VERSION < 0x090000
#include <QVTKWidget.h>
#else
#include <QVTKOpenGLStereoWidget.h>
#endif
 
class VISP_EXPORT us3DSceneWidget : public usViewerWidget
{
  Q_OBJECT
public:
  // Constructor/Destructor
  us3DSceneWidget(QWidget *parent = NULL, Qt::WindowFlags f = Qt::WindowFlags());
  ~us3DSceneWidget() {}
 
  vtkPolyData *getContour1();
  vtkPolyData *getContour2();
  vtkPolyData *getContour3();
 
  vtkImageData *getImageData();
 
  vtkPolyData *getMeshInPlane1();
  vtkPolyData *getMeshInPlane2();
  vtkPolyData *getMeshInPlane3();
 
  vtkPlane *getPlane1();
  vtkPlane *getPlane2();
  vtkPlane *getPlane3();
 
  void init();
 
  // Catch paint events, in case we want to display some informations (writing in this widget) over the vtk scene
  void paintEvent(QPaintEvent *event);
 
  // Set image to display
  void setImageData(vtkImageData *imageData);
 
  // Set planes
  void setPlane1(vtkPlane *plane);
  void setPlane2(vtkPlane *plane);
  void setPlane3(vtkPlane *plane);
  void setPlanes(vtkPlane *plane1, vtkPlane *plane2, vtkPlane *plane3);
 
public slots:
  //
  void updateImageData(vtkImageData *imageData);
  void updateMatrix1(vtkMatrix4x4 *matrix);
  void updateMatrix2(vtkMatrix4x4 *matrix);
  void updateMatrix3(vtkMatrix4x4 *matrix);
 
signals:
  void plane1Changed();
  void plane2Changed();
  void plane3Changed();
 
private:
  // image
  vtkImageData *imageData;
 
  // Cube for image bounds
  vtkSmartPointer<vtkCubeSource> imageBoundsCube;
 
  // planes (containing geometrical informations)
  vtkPlane *plane1;
  vtkPlane *plane2;
  vtkPlane *plane3;
 
  // mesh polydata
  vtkPolyData *meshPolyData;
 
  // vtk mappers
  vtkImageResliceMapper *imageResliceMapper1;
  vtkImageResliceMapper *imageResliceMapper2;
  vtkImageResliceMapper *imageResliceMapper3;
 
  // vtk actors
  vtkImageSlice *imageSlice1;
  vtkImageSlice *imageSlice2;
  vtkImageSlice *imageSlice3;
 
  // Cutters for intersections between plane and volume borders
  vtkSmartPointer<vtkCutter> cutter1;
  vtkSmartPointer<vtkCutter> cutter2;
  vtkSmartPointer<vtkCutter> cutter3;
 
  // Cutters for intersections between planes and mesh
  vtkSmartPointer<vtkCutter> cutterPolyDataPlane1;
  vtkSmartPointer<vtkCutter> cutterPolyDataPlane2;
  vtkSmartPointer<vtkCutter> cutterPolyDataPlane3;
 
  // axes representation
  vtkSmartPointer<vtkAxesActor> m_axesActor;
 
  // vtk renderer
  vtkRenderer *renderer;
};
#endif
#endif // __us3DSceneWidget_h_