nmWTAI-Platform/Include/nmNum/nmRender/nmVTKImageExporter.h

#ifndef NMVTKIMAGEEXPORTER_H
#define NMVTKIMAGEEXPORTER_H

#include "nmRender_global.h"

#include <QImage>
#include <QObject>

class QPrinter;
class QWidget;
class vtkImageData;
class vtkRenderWindow;

/**
 * @brief VTK 静态图片、剪贴板和打印输出服务。
 *
 * RenderWindow 为非所有权指针。捕获结果会复制到 QImage 自有缓冲区，调用方
 * 不会获得 VTK 图像的悬空引用。所有接口只能在 Qt GUI 线程调用。
 */
class NM_RENDER_EXPORT nmVTKImageExporter : public QObject
{
    Q_OBJECT

public:
    /** @brief 创建无绑定 RenderWindow 的图片输出服务。 */
    explicit nmVTKImageExporter(QObject* pParent = NULL);
    /** @brief 只解除 RenderWindow 非所有权引用。 */
    virtual ~nmVTKImageExporter();

    /** @brief 切换截图来源；传入 NULL 表示解除。 */
    void setRenderWindow(vtkRenderWindow* pRenderWindow);
    /** @brief 返回当前截图来源的非所有权指针。 */
    vtkRenderWindow* getRenderWindow() const;

    /** @brief 捕获并垂直翻转当前画面；失败时返回空 QImage。 */
    QImage captureImage() const;
    /** @brief 根据文件扩展名保存 PNG、JPEG 或 BMP。 */
    bool saveImage(const QString& sFileName) const;
    /** @brief 捕获当前画面并复制到系统剪贴板。 */
    bool copyImageToClipboard() const;
    /** @brief 显示打印对话框，并按页面比例居中输出画面。 */
    bool printImage(QWidget* pParent);
    /** @brief 显示打印预览；每次预览重绘均重新捕获当前画面。 */
    bool printPreview(QWidget* pParent);

private slots:
    /** @brief 响应打印预览的同步重绘请求。 */
    void renderForPrint(QPrinter* pPrinter);

private:
    Q_DISABLE_COPY(nmVTKImageExporter)

    /** @brief 将 VTK 无符号字节标量复制为独立的 ARGB32 图像。 */
    QImage createQImage(vtkImageData* pImageData) const;
    /** @brief 保持宽高比，将当前画面绘制到打印机可打印区域中央。 */
    bool paintImage(QPrinter* pPrinter) const;

    vtkRenderWindow* m_pRenderWindow; ///< 非所有权指针，由渲染容器持有。
};

#endif // NMVTKIMAGEEXPORTER_H