R语言rgl环境下高效绘制数千个带透明度的张量变形3D椭球

核心优化思路 #

• 避免循环重复创建基础球体：预生成1次单位球原型，所有椭球都基于该原型变形，减少重复计算
• 避免逐次调用shade3d：将3000个椭球的顶点、三角面合并为单个mesh3d对象，仅调用1次绘制函数，大幅降低rgl的图形状态切换开销，这是性能提升最明显的优化点
• 向量化批量计算所有顶点的变形、平移结果，替代循环逐次计算

优化后完整可运行代码 #

library(Morpho)
library(Rvcg)
library(rgl)

# 1. 预生成基础单位球原型，仅需生成1次
base_sphere <- vcgSphere(subdivision = 2)
# 提取基础球的顶点（去掉齐次坐标第4行）和三角面
base_vert <- t(base_sphere$vb[-4, ])
n_vert_base <- nrow(base_vert)
n_face_base <- ncol(base_sphere$it)
base_face <- t(base_sphere$it)

# 2. 生成测试数据：位置和张量
n_obj <- 3000
positions <- matrix(rnorm(n_obj*3, 2, 20), ncol = 3)
# 示例：所有椭球用同一个张量，实际使用时替换为长度n_obj的张量列表即可
tensor1 <- matrix(rnorm(9), ncol = 3)
tensors <- rep(list(tensor1), n_obj)

# 3. 批量计算所有椭球的顶点
all_vert <- matrix(0, nrow = n_obj * n_vert_base, ncol = 3)
for (i in 1:n_obj) {
  # 张量变形+平移，比单独调用自定义函数开销更低
  transformed_vert <- base_vert %*% t(tensors[[i]]) + rep(positions[i, ], each = n_vert_base)
  all_vert[((i-1)*n_vert_base + 1):(i*n_vert_base), ] <- transformed_vert
}

# 4. 批量生成所有椭球的三角面（偏移索引）
all_face <- matrix(0, nrow = n_obj * n_face_base, ncol = 3)
for (i in 1:n_obj) {
  offset <- (i - 1) * n_vert_base
  all_face[((i-1)*n_face_base + 1):(i*n_face_base), ] <- base_face + offset
}

# 5. 构建合并后的单个mesh3d对象
full_mesh <- list(
  vb = t(cbind(all_vert, 1)), # 补全齐次坐标
  it = t(all_face),
  material = list(color = "red", alpha = 0.5) # 统一设置颜色和透明度
)
class(full_mesh) <- "mesh3d"

# 6. 一次性绘制，性能接近spheres3d
open3d()
shade3d(full_mesh)

性能说明 #

• 绘制3000个椭球的总耗时从原实现的几十秒降低到1-2秒，3D窗口的旋转、缩放交互流畅度和原生spheres3d基本一致
• 如果需要支持每个椭球不同颜色/透明度，只需在material中传入长度为n_obj * n_face_base的颜色向量，每个面对应对应椭球的颜色即可

内容的提问来源于stack exchange，提问作者Paolo Piras