VisualizationWidget.cpp

Go to the documentation of this file.

 
// ** INCLUDES **
#include "VisualizationWidget.h"
 
#include <vtkAxesActor.h>
#include <vtkCamera.h>
#include <vtkCellPicker.h>
#include <vtkCommand.h>
#include <vtkGenericOpenGLRenderWindow.h>
#include <vtkInteractorStyleRubberBandZoom.h>
#include <vtkInteractorStyleSwitch.h>
#include <vtkMath.h>
#include <vtkNew.h>
#include <vtkOrientationMarkerWidget.h>
#include <vtkPNGWriter.h>
#include <vtkRenderWindow.h>
#include <vtkRenderWindowInteractor.h>
#include <vtkRenderer.h>
#include <vtkSmartPointer.h>
#include <vtkWindowToImageFilter.h>
 
#include <QCursor>
#include <QDir>
#include <QFileDialog>
#include <QInputDialog>
#include <QLineEdit>
#include <QSettings>
#include <QString>
#include <cmath>
 
#include "GeoLib/Point.h"
#include "VtkCustomInteractorStyle.h"
#include "VtkPickCallback.h"
 
VisualizationWidget::VisualizationWidget(QWidget* parent /*= 0*/)
    : QWidget(parent)
{
    this->setupUi(this);
 
    vtkNew<vtkRenderer> ren;
    _vtkRender = ren;
 
    vtkNew<vtkGenericOpenGLRenderWindow> renderWindow;
    vtkWidget->setRenderWindow(renderWindow);
    vtkWidget->renderWindow()->AddRenderer(ren);
 
    _interactorStyle = VtkCustomInteractorStyle::New();
    renderWindow->GetInteractor()->SetInteractorStyle(_interactorStyle);
    _interactorStyle->SetDefaultRenderer(ren);
 
    _vtkPickCallback = VtkPickCallback::New();
    vtkSmartPointer<vtkCellPicker> picker =
        vtkSmartPointer<vtkCellPicker>::New();
    picker->AddObserver(vtkCommand::EndPickEvent, _vtkPickCallback);
    renderWindow->GetInteractor()->SetPicker(picker);
 
    QSettings settings;
 
    ren->SetBackground(0.0, 0.0, 0.0);
 
    // Create an orientation marker using vtkAxesActor
    vtkSmartPointer<vtkAxesActor> axesActor =
        vtkSmartPointer<vtkAxesActor>::New();
    _markerWidget = vtkOrientationMarkerWidget::New();
    _markerWidget->SetOrientationMarker(axesActor);
    _markerWidget->SetInteractor(renderWindow->GetInteractor());
    _markerWidget->EnabledOn();
    _markerWidget->InteractiveOff();
 
    _isShowAllOnLoad = settings.value("resetViewOnLoad", true).toBool();
 
    // Set alternate cursor shapes
    connect(_interactorStyle, SIGNAL(cursorChanged(Qt::CursorShape)), this,
            SLOT(setCursorShape(Qt::CursorShape)));
 
    connect((QObject*)_interactorStyle, SIGNAL(requestViewUpdate()), this,
            SLOT(updateView()));
}
 
VisualizationWidget::~VisualizationWidget()
{
    _vtkPickCallback->Delete();
    _interactorStyle->Delete();
    _markerWidget->Delete();
}
 
VtkCustomInteractorStyle* VisualizationWidget::interactorStyle() const
{
    return _interactorStyle;
}
 
VtkPickCallback* VisualizationWidget::vtkPickCallback() const
{
    return _vtkPickCallback;
}
 
void VisualizationWidget::updateView()
{
    vtkWidget->renderWindow()->Render();
}
 
void VisualizationWidget::showAll(int x, int y, int z)
{
    _vtkRender->ResetCamera();
    vtkCamera* cam = _vtkRender->GetActiveCamera();
    double* fp = cam->GetFocalPoint();
    double* p = cam->GetPosition();
    double dist = std::sqrt(vtkMath::Distance2BetweenPoints(p, fp));
    cam->SetPosition(fp[0] + (x * dist), fp[1] + (y * dist),
                     fp[2] + (z * dist));
 
    if (x != 0 || y != 0)
    {
        cam->SetViewUp(0.0, 0.0, 1.0);
    }
    else
    {
        cam->SetViewUp(0.0, 1.0, 0.0);
    }
    this->updateView();
}
 
void VisualizationWidget::updateViewOnLoad()
{
    if (_isShowAllOnLoad)
    {
        this->showAll(0, 0, 1);
    }
    else
    {
        updateView();
    }
}
 
void VisualizationWidget::on_zoomToolButton_toggled(bool checked)
{
    if (checked)
    {
        vtkSmartPointer<vtkInteractorStyleRubberBandZoom> interactorStyle =
            vtkSmartPointer<vtkInteractorStyleRubberBandZoom>::New();
        vtkWidget->renderWindow()->GetInteractor()->SetInteractorStyle(
            interactorStyle);
        QCursor cursor;
        cursor.setShape(Qt::CrossCursor);
        vtkWidget->setCursor(cursor);
    }
    else
    {
        vtkWidget->renderWindow()->GetInteractor()->SetInteractorStyle(
            _interactorStyle);
        QCursor cursor;
        cursor.setShape(Qt::ArrowCursor);
        vtkWidget->setCursor(cursor);
    }
}
 
void VisualizationWidget::on_highlightToolButton_toggled(bool checked)
{
    _interactorStyle->setHighlightActor(checked);
}
 
void VisualizationWidget::on_orthogonalProjectionToolButton_toggled(
    bool checked)
{
    _vtkRender->GetActiveCamera()->SetParallelProjection(checked);
    this->updateView();
}
 
void VisualizationWidget::on_screenshotPushButton_pressed()
{
    QSettings settings;
    QString filename = QFileDialog::getSaveFileName(
        this, tr("Save screenshot"),
        settings.value("lastScreenshotDir").toString(), "PNG file (*.png)");
    if (filename.count() > 4)
    {
        bool ok;
        int magnification = QInputDialog::getInt(
            this, tr("Screenshot magnification"),
            tr("Enter a magnification factor for the resulting image."), 2, 1,
            10, 1, &ok);
        if (ok)
        {
            QDir dir(filename);
            settings.setValue("lastScreenshotDir", dir.absolutePath());
            this->screenshot(filename, magnification);
        }
    }
}
 
void VisualizationWidget::screenshot(QString filename, int magnification)
{
    vtkSmartPointer<vtkWindowToImageFilter> windowToImageFilter =
        vtkSmartPointer<vtkWindowToImageFilter>::New();
    windowToImageFilter->SetInput(vtkWidget->renderWindow());
    // Set the resolution of the output image
    // magnification times the current resolution of vtk render window
    windowToImageFilter->SetScale(magnification);
    // Also record the alpha (transparency) channel
    windowToImageFilter->SetInputBufferTypeToRGBA();
    windowToImageFilter->Update();
 
    vtkSmartPointer<vtkPNGWriter> writer = vtkSmartPointer<vtkPNGWriter>::New();
    writer->SetFileName(filename.toStdString().c_str());
    writer->SetInputData(windowToImageFilter->GetOutput());
    writer->Write();
 
    this->updateView();
}
 
void VisualizationWidget::setCursorShape(Qt::CursorShape shape)
{
    this->setCursor(QCursor(shape));
}