vtkutils.cpp

Go to the documentation of this file.

// ----------------------------------------------------------------------------
// -                        CloudViewer: www.cloudViewer.org                  -
// ----------------------------------------------------------------------------
// Copyright (c) 2018-2024 www.cloudViewer.org
// SPDX-License-Identifier: MIT
// ----------------------------------------------------------------------------
 
#include "vtkutils.h"
 
#include <QVTKOpenGLNativeWidget.h>
#include <vtkActor.h>
#include <vtkDelaunay2D.h>
#include <vtkDiskSource.h>
#include <vtkDoubleArray.h>
#include <vtkImageData.h>
#include <vtkIntArray.h>
#include <vtkPlaneSource.h>
#include <vtkPointData.h>
#include <vtkPoints.h>
#include <vtkPolyData.h>
#include <vtkPolyDataMapper.h>
#include <vtkQImageToImageSource.h>
#include <vtkRenderWindow.h>
#include <vtkRenderer.h>
#include <vtkSphereSource.h>
#include <vtkVRMLExporter.h>
 
#include <QDebug>
#include <QThreadPool>
 
#include "actorexporter.h"
 
namespace VtkUtils {
 
QImage vtkImageDataToQImage(vtkImageData* imageData) {
    if (!imageData) return QImage();
 
    int width = imageData->GetDimensions()[0];
    int height = imageData->GetDimensions()[1];
    QImage image(width, height, QImage::Format_RGB32);
    QRgb* rgbPtr = reinterpret_cast<QRgb*>(image.bits()) + width * (height - 1);
    unsigned char* colorsPtr =
            reinterpret_cast<unsigned char*>(imageData->GetScalarPointer());
    // mirror vertically
    for (int row = 0; row < height; ++row) {
        for (int col = 0; col < width; ++col) {
            // Swap rgb
            *(rgbPtr++) =
                    QColor(colorsPtr[0], colorsPtr[1], colorsPtr[2]).rgb();
            colorsPtr += 3;
        }
        rgbPtr -= width * 2;
    }
    return image;
}
 
void qImageToVtkImage(QImage& img, vtkImageData* imageData) {
    if (!imageData) {
        qDebug() << "null image data.";
        return;
    }
 
    vtkSmartPointer<vtkQImageToImageSource> qimageToImageSource =
            vtkSmartPointer<vtkQImageToImageSource>::New();
    qimageToImageSource->SetQImage(&img);
    qimageToImageSource->Update();
    imageData->DeepCopy(qimageToImageSource->GetOutput());
}
 
QImage vtkWidgetSnapshot(QVTKOpenGLNativeWidget* widget) {
    if (!widget) return QImage();
    // return vtkImageDataToQImage(widget->cachedImage());
    return vtkImageDataToQImage(nullptr);
}
 
void exportActorToFile(vtkActor* actor, const QString& outfile) {
    VtkUtils::ActorExporter* exporter =
            new VtkUtils::ActorExporter(actor, outfile);
    QThreadPool::globalInstance()->start(exporter);
}
 
}  // namespace VtkUtils