VisPy中的跳转泛洪算法(JFA)实现解析-CSDN博客

本文链接：https://blog.csdn.net/gitblog_00412/article/details/148577909

VisPy中的跳转泛洪算法(JFA)实现解析

vispy Main repository for Vispy 项目地址: https://gitcode.com/gh_mirrors/vi/vispy

跳转泛洪算法(Jump Flooding Algorithm, JFA)是一种高效的图形处理算法，常用于计算欧几里得距离变换(Euclidean Distance Transform, EDT)。本文将深入分析VisPy项目中基于GLSL实现的JFA算法，帮助读者理解其核心原理和实现细节。

算法概述

跳转泛洪算法是一种并行算法，通过多轮迭代逐步传播距离信息。每轮迭代中，算法以当前步长的一半作为新的步长，直到步长降为1。这种策略使得算法的时间复杂度为O(log n)，远优于传统的O(n²)算法。

核心实现解析

1. 着色器程序管理

VisPy实现中使用了三个着色器程序：

programObj0 = createShader("vertex.glsl", "fragment_seed.glsl")  # 种子阶段
programObj1 = createShader("vertex.glsl", "fragment_flood.glsl")  # 泛洪阶段
programObj2 = createShader("vertex.glsl", "fragment_display.glsl")  # 显示阶段

每个着色器程序由顶点着色器和片段着色器组成，分别处理不同阶段的算法逻辑。

2. 纹理处理

算法使用多张纹理进行数据交换：

textureID = gl.glGenTextures(3)  # 生成3个纹理
texw, texh = loadShapeTexture("shape1.tga", textureID[0])  # 加载原始图像
createBufferTexture(textureID[1], texw, texh)  # 创建缓冲纹理1
createBufferTexture(textureID[2], texw, texh)  # 创建缓冲纹理2

纹理参数设置确保了正确的采样行为，这对于算法精度至关重要。

3. 帧缓冲对象(FBO)使用

fboID = gl.glGenFramebuffers(1)  # 生成帧缓冲对象
gl.glFramebufferTexture2D(gl.GL_DRAW_FRAMEBUFFER,
                         gl.GL_COLOR_ATTACHMENT0,
                         gl.GL_TEXTURE_2D,
                         textureID[lastRendered], 0)

FBO用于将渲染结果输出到纹理，而不是屏幕，这是实现多轮纹理处理的关键。

4. 算法主循环

算法主循环包含三个主要阶段：

种子阶段：将原始图像转换为距离场初始状态
泛洪阶段：多轮迭代传播距离信息
显示阶段：将最终结果渲染到屏幕

# 种子阶段
setUniformVariables(programObj0, 0, texw, texh, 0)
renderScene(programObj0, texw, texh)

# 泛洪阶段
stepsize = texw//2 if texw > texh else texh//2
while stepsize > 0:
    setUniformVariables(programObj1, 0, texw, texh, stepsize)
    # ...渲染到交替纹理...
    stepsize = stepsize // 2

# 显示阶段
setUniformVariables(programObj2, 0, texw, texh, 0)
renderScene(programObj2, width, height)

关键技术点

1. 纹理交替使用

算法使用两个纹理交替作为输入和输出，避免数据竞争：

lastRendered = 1 if lastRendered == 2 else 2

这种乒乓缓冲技术是GPU计算中常见的模式。

2. 步长递减策略

步长从最大尺寸开始，每轮减半：

stepsize = texw//2 if texw > texh else texh//2
while stepsize > 0:
    # ...处理...
    stepsize = stepsize // 2

这种策略确保了距离信息能够高效传播到整个图像。

3. OpenGL ES 2.0兼容性

实现中做了两处重要修改以确保兼容性：

将GL_CLAMP_TO_BORDER改为GL_CLAMP_TO_EDGE
将GL_RG16改为GL_RGBA，并硬编码"texlevels"为65536

性能优化

代码中包含了FPS计算和显示功能，便于性能评估：

def showFPS(texw, texh):
    global frames, t0
    t = time.time()
    if (t - t0) > 1.:
        fps = frames / (t - t0)
        titlestr = "%sx%s texture, %.1f FPS" % (texw, texh, fps)
        glfw.glfwSetWindowTitle(window, titlestr)
        t0 = t
        frames = 0
    frames += 1