nmWATI/Src/nmNum/nmSubWxs/nmWxPostprocessingWidget.cpp

#include "nmWxPostprocessingWidget.h"

#include <vtkAutoInit.h>
VTK_MODULE_INIT(vtkRenderingFreeType)

#include <vtkPointData.h>
#include <vtkDataSetReader.h>
#include <vtkUnstructuredGrid.h>
#include <vtkUnstructuredGridReader.h>

#include "nmWxPostprocessingWidget.h"
#include <QVTKWidget.h>
#include <vtkSmartPointer.h>
#include <vtkDataSetReader.h>
#include <vtkUnstructuredGrid.h>
#include <vtkDataSetMapper.h>
#include <vtkActor.h>
#include <vtkRenderer.h>
#include <QPushButton>
#include <QVBoxLayout>
#include <QHBoxLayout>
#include <QDebug>
#include <vtkProperty.h>
#include <vtkRenderWindow.h>

#include <vtkScalarBarActor.h>
#include <vtkSmartPointer.h>
#include <vtkScalarBarActor.h>
#include <vtkLookupTable.h>
#include <vtkTextProperty.h>
#include <QFrame>
#include <vtkTextActor.h>
#include <QLabel>
nmWxPostprocessingWidget::nmWxPostprocessingWidget(QWidget *parent) : QWidget(parent)
{
    this->initLayout();
}

void nmWxPostprocessingWidget::initLayout()
{
    m_mainLayout = new QVBoxLayout;
    this->setLayout(m_mainLayout);
    QFrame* frame = new QFrame;
    m_frameLaoyt = new QVBoxLayout;
    frame->setLayout(m_frameLaoyt);
    m_mainLayout->addWidget(frame);
    qDebug() << "in initLayout";
}

void nmWxPostprocessingWidget::loadVTKFile(const QString &filePath)
{
    QVTKWidget* vtkWidget = new QVTKWidget;
    // 创建新的渲染器
    vtkSmartPointer<vtkRenderer> newRenderer = vtkSmartPointer<vtkRenderer>::New();
    // 设置通用的 VTK 数据集读取器
    vtkSmartPointer<vtkDataSetReader> reader = vtkSmartPointer<vtkDataSetReader>::New();
    reader->SetFileName(filePath.toStdString().c_str());
    reader->Update(); // 更新读取器，确保读取文件数据
    // 获取读取到的数据集
    vtkDataSet* dataSet = reader->GetOutput();
    // 处理 UnstructuredGrid 类型的数据
    vtkUnstructuredGrid* unstructuredGrid = vtkUnstructuredGrid::SafeDownCast(dataSet);
    if (!unstructuredGrid) {
        qDebug() << "Failed to load unstructured grid from file:" << filePath; // 添加失败信息
        return; // 如果加载失败，直接返回
    }
    double scalarRange[2];
    unstructuredGrid->GetPointData()->GetScalars()->GetRange(scalarRange); // 计算标量范围
    qDebug() << "Loading file:" << filePath << " Scalar Range:" << scalarRange[0] << " " << scalarRange[1];
    vtkSmartPointer<vtkDataSetMapper> unstructuredMapper = vtkSmartPointer<vtkDataSetMapper>::New();
    unstructuredMapper->SetInputData(unstructuredGrid);
    unstructuredMapper->SetScalarRange(scalarRange);
    unstructuredMapper->ScalarVisibilityOn();
    // 创建演员（actor）
    vtkSmartPointer<vtkActor> actor = vtkSmartPointer<vtkActor>::New();
    actor->SetMapper(unstructuredMapper);
    actor->GetProperty()->SetEdgeVisibility(true);
    actor->GetProperty()->SetLineWidth(1.0); // 设置边缘宽度
    // 添加新演员到新的渲染器
    newRenderer->AddActor(actor);
    newRenderer->SetBackground(0.1, 0.2, 0.3); // 设置背景颜色
    // 创建颜色条
    vtkSmartPointer<vtkScalarBarActor> scalarBar = vtkSmartPointer<vtkScalarBarActor>::New();
    vtkSmartPointer<vtkLookupTable> lut = vtkSmartPointer<vtkLookupTable>::New();
    lut->SetNumberOfColors(256);
    lut->SetRange(scalarRange); // 设置查找表的范围为标量范围
    lut->Build();
    scalarBar->SetLookupTable(lut);
    scalarBar->SetTitle("data");
    scalarBar->SetPosition(0.05, 0.1); // 设置位置（右侧）
    scalarBar->SetWidth(0.1); // 设置宽度
    scalarBar->SetHeight(0.8); // 设置高度
    newRenderer->AddActor(scalarBar);
    // 创建文本演员以显示文件名
    vtkSmartPointer<vtkTextActor> textActor = vtkSmartPointer<vtkTextActor>::New();
    textActor->SetInput(QString("File: %1").arg(filePath).toStdString().c_str()); // 设置文本内容
    textActor->GetPositionCoordinate()->SetCoordinateSystemToNormalizedDisplay();
    textActor->SetPosition(0.1, 0.95); // 设置位置为右上角（稍微向内偏移）
    textActor->GetTextProperty()->SetFontSize(24); // 设置字体大小
    textActor->GetTextProperty()->SetColor(1.0, 1.0, 1.0); // 设置字体颜色为白色
    textActor->GetTextProperty()->SetFontFamilyToArial(); // 设置字体
    textActor->GetTextProperty()->SetItalic(1); // 斜体
    textActor->GetTextProperty()->SetBold(1); // 加粗
    newRenderer->AddActor(textActor);
    // 将新的渲染器添加到渲染窗口
    vtkWidget->GetRenderWindow()->AddRenderer(newRenderer);
    vtkWidget->GetRenderWindow()->Render(); // 渲染窗口
    // 添加到frame上
    m_frameLaoyt->addWidget(vtkWidget);
}