【深度学习驱动流体力学】VTK创建、处理和可视化流体数据

Visualization Toolkit(VTK)是一个强大的开源软件系统,用于处理和可视化三维数据。它提供了丰富的工具和算法,可以处理从简单的网格数据到复杂的流体动力学模拟数据等各种类型的数据。本文将完整介绍如何使用 VTK 创建、处理和可视化流体数据的基本步骤和相关概念。

  1. VTK 概述
    VTK 是由 Kitware 公司开发的开源软件,用于三维计算机图形学、图像处理和可视化。它提供了一组丰富的 C++ 类库和 Python 接口,支持多种数据类型的处理和可视化,包括结构化和非结构化网格、图像数据、多边形数据等。VTK 被广泛应用于科学研究、医学图像处理、工程仿真和可视化等领域。

  2. 创建流体数据
    在 VTK 中,可以通过多种方式创建流体数据,最常见的是创建结构化网格(Structured Grid)和非结构化网格(Unstructured Grid)。下面我们以创建结构化网格为例进行说明。

创建结构化网格
结构化网格是由规则排列的点组成的网格,适用于表示规则的三维空间区域。例如,我们可以创建一个简单的立方体结构化网格,并添加标量数据(如温度)。

import vtk# 创建一个结构化网格
def create_structured_grid():# 创建点数据points = vtk.vtkPoints()for i in range(10):for j in range(10):for k in range(10):points.InsertNextPoint(i, j, k)# 创建一个结构化网格grid = vtk.vtkStructuredGrid()grid.SetDimensions(10, 10, 10)grid.SetPoints(points)# 创建一个标量数据(假设为温度数据)temperature_array = vtk.vtkFloatArray()temperature_array.SetName("Temperature")temperature_array.SetNumberOfComponents(1)temperature_array.SetNumberOfTuples(grid.GetNumberOfPoints())# 设置温度数据for i in range(grid.GetNumberOfPoints()):temperature_array.SetValue(i, i)# 将温度数据添加到点数据grid.GetPointData().AddArray(temperature_array)grid.GetPointData().SetActiveScalars("Temperature")return grid

上述代码创建了一个立方体网格,网格包含 10x10x10 个点,每个点的坐标范围在 (0, 0, 0) 到 (9, 9, 9)。添加了名为 “Temperature” 的标量数据作为网格的一部分。

  1. 处理流体数据
    在处理流体数据时,VTK 提供了许多用于数据处理、过滤和分析的工具和算法。例如,可以对网格进行剖析、滤波、插值等操作,以准备数据进行后续的可视化或分析。

数据滤波和处理示例

import vtk# 加载 VTK 文件
def load_vtk_file(file_path):reader = vtk.vtkXMLStructuredGridReader()reader.SetFileName(file_path)reader.Update()return reader.GetOutput()# 数据滤波和处理
def process_data(grid):# 示例:创建一个高斯平滑滤波器smooth_filter = vtk.vtkImageGaussianSmooth()smooth_filter.SetInputData(grid)smooth_filter.SetStandardDeviation(2.0)smooth_filter.Update()processed_grid = smooth_filter.GetOutput()return processed_grid

在上面的示例中,使用了 VTK 的 vtkXMLStructuredGridReader 类来加载存储为 VTK 文件的结构化网格数据。然后,应用了一个高斯平滑滤波器 vtkImageGaussianSmooth 来平滑处理数据。

  1. 可视化流体数据
    VTK 提供了多种可视化技术和工具,可以将处理后的流体数据呈现为三维图像或动画。可以使用 VTK 自带的可视化工具,也可以结合其他库(如 PyVista)进行更高级的可视化操作。

可视化流体数据示例

import vtk
import pyvista as pv# 可视化结构化网格
def visualize_grid(grid):# 将 VTK 结构化网格转换为 PyVista 网格pv_grid = pv.wrap(grid)# 创建一个绘图窗口plotter = pv.Plotter()# 添加网格到绘图窗口plotter.add_mesh(pv_grid, scalars="Temperature", show_scalar_bar=True)# 显示绘图窗口plotter.show()# 主程序入口
if __name__ == "__main__":# 创建结构化网格grid = create_structured_grid()# 可视化网格数据visualize_grid(grid)

在上述示例中,使用了 PyVista 来创建一个绘图窗口,并将 VTK 结构化网格转换为 PyVista 网格对象进行可视化。通过 plotter.add_mesh 方法,可以将网格添加到绘图窗口中,并显示温度标量数据(假设已经将温度数据添加到网格中)。

  1. 保存流体数据
    VTK 允许将处理后的数据保存为 VTK 文件,以便后续加载和分析。这在数据处理和模拟结果保存中非常有用。

保存结构化网格示例

import vtk# 保存网格数据到 VTK 文件
def save_to_vtk(grid, file_path):writer = vtk.vtkXMLStructuredGridWriter()writer.SetFileName(file_path)writer.SetInputData(grid)writer.Write()# 主程序入口
if __name__ == "__main__":# 创建结构化网格grid = create_structured_grid()# 保存网格数据到 VTK 文件file_path = "structured_grid.vtk"save_to_vtk(grid, file_path)print(f"Grid saved to {file_path}")

在上述示例中,使用 vtkXMLStructuredGridWriter 将创建的结构化网格数据保存到名为 “structured_grid.vtk” 的文件中。

在这里插入图片描述

  1. 创建流体数据
    首先,我们将演示如何创建一个简单的流体数据模拟,然后将其用于动画效果的可视化。在 VTK 中,我们可以通过生成结构化网格来表示流体数据,然后将数据添加到网格中。
import vtk# 创建结构化网格表示流体数据
def create_structured_grid():# 创建点数据points = vtk.vtkPoints()dimensions = (20, 20, 20)  # 网格维度for i in range(dimensions[0]):for j in range(dimensions[1]):for k in range(dimensions[2]):points.InsertNextPoint(i, j, k)# 创建结构化网格grid = vtk.vtkStructuredGrid()grid.SetDimensions(dimensions)grid.SetPoints(points)# 添加速度向量数据velocity_array = vtk.vtkFloatArray()velocity_array.SetName("Velocity")velocity_array.SetNumberOfComponents(3)  # 三维向量velocity_array.SetNumberOfTuples(grid.GetNumberOfPoints())# 设置速度向量数据(示例中简单设置为随机向量)import randomfor i in range(grid.GetNumberOfPoints()):velocity_array.SetTuple(i, (random.uniform(-1, 1), random.uniform(-1, 1), random.uniform(-1, 1)))grid.GetPointData().AddArray(velocity_array)grid.GetPointData().SetActiveVectors("Velocity")return grid

在上述示例中,我们创建了一个简单的 20x20x20 的立方体结构化网格,并添加了名为 “Velocity” 的速度向量数据。实际应用中,速度数据可以根据实际模拟结果进行设置。

  1. 可视化流体数据并添加动画效果
    接下来,我们使用 VTK 来可视化上面创建的流体数据,并添加动画效果以展示流体的运动状态。
import vtk
import numpy as np
import pyvista as pv# 创建结构化网格
grid = create_structured_grid()# 将 VTK 结构化网格转换为 PyVista 网格
pv_grid = pv.wrap(grid)# 创建一个绘图窗口
plotter = pv.Plotter()# 添加网格到绘图窗口
plotter.add_mesh(pv_grid, scalars="Velocity", show_scalar_bar=True)# 设置相机位置和视角
plotter.camera_position = [(50, 50, 50), (10, 10, 10), (0, 0, 1)]# 添加动画效果
def update_scene():# 模拟动画效果,例如更新速度数据或网格位置displacement = np.random.uniform(-0.1, 0.1, size=pv_grid.points.shape)pv_grid.points += displacement# 添加回调函数来更新场景
plotter.add_callback(update_scene)# 显示绘图窗口
plotter.show()

在上述代码中,我们使用了 PyVista 来创建绘图窗口,并将 VTK 结构化网格转换为 PyVista 网格对象进行可视化。通过 plotter.add_callback 方法,我们添加了一个回调函数 update_scene,在每次渲染时更新网格的位置,以模拟流体的动态效果。

  1. 结果展示
    运行上述代码将打开一个 PyVista 的交互式窗口,显示具有动画效果的流体数据可视化。您可以通过修改回调函数 update_scene 中的更新逻辑来实现不同的动画效果,比如根据实际的流体模拟数据更新网格的位置或属性。

通过以上步骤,您可以在 VTK 中创建、处理和动态可视化流体数据,并根据需要定制动画效果,以便更好地展示流体的运动和行为。

import vtk
import pyvista as pv
import numpy as np
import random# 创建结构化网格表示流体数据
def create_structured_grid(dimensions=(20, 20, 20)):# 创建点数据points = vtk.vtkPoints()for i in range(dimensions[0]):for j in range(dimensions[1]):for k in range(dimensions[2]):points.InsertNextPoint(i, j, k)# 创建结构化网格grid = vtk.vtkStructuredGrid()grid.SetDimensions(dimensions)grid.SetPoints(points)# 添加速度向量数据velocity_array = vtk.vtkFloatArray()velocity_array.SetName("Velocity")velocity_array.SetNumberOfComponents(3)  # 三维向量velocity_array.SetNumberOfTuples(grid.GetNumberOfPoints())# 设置速度向量数据(示例中简单设置为随机向量)for i in range(grid.GetNumberOfPoints()):velocity_array.SetTuple(i, (random.uniform(-1, 1), random.uniform(-1, 1), random.uniform(-1, 1)))grid.GetPointData().AddArray(velocity_array)grid.GetPointData().SetActiveVectors("Velocity")return grid# 创建结构化网格
grid = create_structured_grid()# 将 VTK 结构化网格转换为 PyVista 网格
pv_grid = pv.wrap(grid)# 创建一个绘图窗口
plotter = pv.Plotter()# 添加网格到绘图窗口,确保指定正确的数据数组名称
plotter.add_mesh(pv_grid, scalars="Velocity", show_scalar_bar=True)# 显示绘图窗口
plotter.show()

在这里插入图片描述
代码说明:
创建结构化网格函数 (create_structured_grid):

使用 VTK 创建一个简单的结构化网格,其中包含了三维点数据。
添加了名为 “Velocity” 的速度向量数据数组,数据内容为随机生成的三维向量。
转换为 PyVista 网格:

使用 pv.wrap 将 VTK 结构化网格转换为 PyVista 网格,以便后续使用 PyVista 进行可视化。
创建绘图窗口 (plotter):
在这里插入图片描述

使用 pv.Plotter() 创建一个 PyVista 绘图窗口,用于显示网格和数据。
添加网格到绘图窗口:

使用 plotter.add_mesh() 将 PyVista 网格添加到绘图窗口中,并指定要显示的数据数组为 “Velocity”,同时显示标量条。
显示绘图窗口:

使用 plotter.show() 显示 PyVista 绘图窗口,以查看包含速度向量数据的结构化网格的可视化效果。

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.mzph.cn/pingmian/29577.shtml

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

Springboot + Mybatis 实现sql打印

参照这个视频:https://www.bilibili.com/video/BV1MS411N7mn/?vd_source90ebeef3261cec486646b6583e9f45f5 实现mybatis对外暴露的接口Interceptor 使用Intercepts接口,这里的写法参照mybatis-plus中的拦截器写法 Intercepts({Signature(type Executor.class, m…

CMake编译proto的方法(custom_target和custom_command)

最近在项目中涉及到在QNX平台上编译CyberRT,其中CyberRT使用到了protobuf,因此,仓库内部有许多proto文件,需要先行将这些proto文件生成对应的.cc和.h文件才能被其他文件使用。 之前一直使用protobuf_generate_cpp来编译proto文件&…

如何在C++中实现延迟删除功能

在软件开发中,缓存是一种常见的优化技术,它允许我们存储数据以供快速访问,从而减少对慢速存储或网络资源的依赖。然而,有时我们可能希望缓存中的某些数据在一段时间后自动过期并被删除,这就是所谓的“延迟删除”功能。…

HTB Freelancer

Freelancer user nmap ➜ htb nmap -A 10.129.221.155 -T 4 Starting Nmap 7.80 ( https://nmap.org ) at 2024-06-02 09:19 CST NSE Timing: About 97.92% done; ETC: 09:24 (0

PostgreSQL源码分析——COPY

导入数据的几种方式 在进行数据导入导出时常会用到copy命令,语法使用可参考下面这篇博文 [Postgres] Bulk Insert and Export Data with csv Files with Postgres copy Command。通常导入数据的方法,可以通过insert的方式(insert into t1 va…

BC153 [NOIP2010]数字统计

数字统计 一.题目描述二.输入描述:三.输出描述:四.数字范围五.题目思路六.代码实现 一.题目描述 请统计某个给定范围[L, R]的所有整数中,数字2出现的次数。 比如给定范围[2, 22],数字2在数2中出现了1次,在数12中出现1次…

如何恢复iPhone iCloud云盘资料删除?给出建议

🏆本文收录于「Bug调优」专栏,主要记录项目实战过程中的Bug之前因后果及提供真实有效的解决方案,希望能够助你一臂之力,帮你早日登顶实现财富自由🚀;同时,欢迎大家关注&&收藏&&…

【Java】已解决com.mysql.cj.jdbc.exceptions.CommunicationsException异常

文章目录 一、分析问题背景二、可能出错的原因三、错误代码示例四、正确代码示例五、注意事项 已解决com.mysql.cj.jdbc.exceptions.CommunicationsException异常 一、分析问题背景 com.mysql.cj.jdbc.exceptions.CommunicationsException是Java程序在使用MySQL Connector/J与…

目标检测—Fast RCNN

介绍Fast R-CNN之前先简单回顾一下R-CNN和SPP-net R-CNN(Regions with CNN) affine image wraping 解析 Bounding Box Regression(边界框回归) 如何回归? 问题1:为什么要使用相对坐标差? …

全面对比与选择指南:Milvus、PGVector、Zilliz及其他向量数据库

本文全面探讨了Milvus、PGVector、Zilliz等向量数据库的特性、性能、应用场景及选型建议,通过详细的对比分析,帮助开发者和架构师根据具体需求选择最合适的向量数据库解决方案。 文章目录 向量数据库概述向量数据库的关键功能向量数据库的扩展和选择向量…

立创开源学习篇(一)

1.机壳地 外面包围的一圈是机壳地,和金属外壳相连与电路板的GND不相连:(大疆很多产品有此设计) 屏蔽和接地:通过在电路板周围打孔,并连接到机壳地,可以形成有效的电磁屏蔽层(形成金…

004、KMeans和DBSCAN的比较

KMeans 聚类 工作原理 选择K个初始中心点(可以随机选择或使用其他方法)。迭代过程: 分配每个数据点到最近的中心点:计算每个数据点到所有中心点的距离,将数据点分配到最近的中心点所属的簇。更新中心点:计…

【C语言】回调函数 和 部分库函数的用法以及模拟实现

一、回调函数: 1、定义: 回调函数就是一个通过函数指针调用的函数。如果你把函数的指针(地址)作为参数传递给另一个函数,当这个指针被用来调用其所指向的函数时,我们就说这是回调函数。 2、qsort的模拟实现…

怎样打印微信文档文件?

在日常生活和工作中,我们经常需要打印微信中的文档文件,无论是工作资料、学习笔记还是其他重要信息。随着科技的发展,我们不再需要前往打印店进行繁琐的操作,而是可以通过一些便捷的在线打印平台轻松实现。今天,我们就…

微信小程序自定义组件(写一个点击查看大图有关闭按钮,复制即可使用)

今天碰到一个需求,本来使用wx.previewImage用的好好的,非要加一个关闭按钮,这可把不想写代码的我难住了,查看官方文档没有加这个按钮的属性,那就自己写了吧,自己写的途中发现胶囊隐藏不了,兴高采…

echarts学习:通过图例事件实现选中后控制多条折线的显隐

1.问题描述 我在工作中遇到了这样一个需求:我们都知道点击echarts折线图的图例,是可以控制折线的显隐的。我现在希望点击某一个图例可以改变多条折线的显隐。 例如在下面这张图中,我将“xxx水位”和“yyy水位”分为一组;将“xxx…

SGPT论文阅读笔记

这是篇想要用GPT来提取sentence embedding的工作,提出了两个框架,一个是SGPT-BE,一个是SGPT-CE,分别代表了Bi-Encoder setting和Cross-Encoder setting。CE的意思是在做阅读理解任务时,document和query是一起送进去&am…

IPv6改造是什么意思?网站IPv6改造怎么做?

随着互联网技术的飞速发展,IPv4地址资源短缺的问题日益凸显,已经难以满足物联网、人工智能等新基建的激增需求。为了解决这一困境,IPv6协议应运而生,其海量的地址资源和内置的安全机制为互联网的持续发展提供了广阔空间。目前各行…

kbadminv1版后台快速开发框架

探索高效开发的新境界!kbadminv1 版后台快速开发框架震撼登场! 基于强大的 thinkphp6 框架,kbadminv1 为开发者们带来了前所未有的便捷与高效。 它犹如一把智慧的钥匙,轻松开启后台开发的大门。简洁而直观的设计,让复…

2024-06-18 blue-VH-driver-订阅组态相关分析

摘要: 2024-06-18 blue-VH-driver-订阅组态相关分析 订阅组态相关: 1.1 创建订阅分组成功, 但是激活订阅分组失败 调用VH接口删除已经创建的订阅分组 无论否删除成功,都返回不做重试删除已经创建的订阅分组,避免无法结束调用1.2 向订阅分组添加的位号的数据量超过单个订阅分…