最近在逛B站时，发现一个特别有意思的AI创作项目，叫“蛇蛇牌蚊香”。初看标题，你可能会和我一样疑惑：蚊香和AI、和编程有什么关系？这难道是个恶搞视频？

但点进去才发现，这是个货真价实的“AI创造公开赛”参赛作品，它巧妙地用编程和AI图像生成技术，把经典的“贪吃蛇”游戏和“蚊香盘”的视觉意象结合了起来。开发者不是简单地画个蚊香，而是让一条“蛇”按照蚊香螺旋的轨迹自动行进，其路径最终构成了蚊香盘的图案。这个项目一下子击中了我：它用一个非常具体、有趣的创意，展示了创意编程、算法可视化与AI工具结合的无限可能。

对于开发者而言，它的价值远不止一个“好玩的小程序”。它背后涉及几个我们经常遇到，但很少深入思考的问题：如何用代码精确控制图形轨迹？如何将数学概念（如阿基米德螺旋线）转化为视觉艺术？更重要的是，在AIGC工具爆发的今天，像Stable Diffusion这样的模型如何与我们的程序化生成逻辑联动，创造出人机协作的新作品？

本文将为你彻底拆解“蛇蛇牌蚊香”这个项目。我不会只停留在“看个热闹”，而是会带你从原理分析、环境搭建、代码实现，到与AI绘画工具（Stable Diffusion）的联动实践，完整走一遍这个创意项目的开发流程。你会发现，实现它所需的Python和Pygame知识并不复杂，但整个思路对于开拓技术视野、激发创作灵感极具启发性。

无论你是想学习图形编程的初学者，还是正在寻找AIGC落地趣味场景的开发者，这篇文章都能给你带来可直接复用的代码和耳目一新的思路。

1. 核心创意与解决的问题

“蛇蛇牌蚊香”项目的核心创意，在于用程序模拟自然现象，并赋予其交互与艺术性。它解决了几个层面的问题：

1. 对算法可视化的趣味性探索：我们学习排序算法时看过动态柱状图，学习路径规划时看过网格动画，但大多比较枯燥。“蛇蛇牌蚊香”把“轨迹生成算法”包装成一个有明确文化符号（蚊香）和游戏元素（贪吃蛇）的视觉作品，让算法学习变得生动有趣。
2. 连接数学、编程与视觉艺术：蚊香的螺旋线是典型的阿基米德螺线（Archimedean spiral）。这个项目要求开发者用代码（参数方程）精确描述这条曲线，并控制一个“蛇头”像素点沿此曲线运动，同时留下“蛇身”痕迹。这本身就是一次完美的数学建模到图形渲染的实践。
3. 探索AIGC时代的人机协作范式：项目的进阶玩法，是将程序生成的螺旋线轨迹图，作为控制信息（如ControlNet的线稿）输入到Stable Diffusion等AI绘画模型中，生成风格化、细节丰富的最终图像。这演示了“程序生成结构，AI丰富细节”的高效创作流程，为游戏素材生成、艺术设计提供了新思路。

所以，这篇文章要解决的，不仅仅是“如何写一个画螺旋线的程序”，而是通过这个具体案例，掌握创意编程的方法论，并了解如何将传统图形编程与前沿AIGC工具链对接。

2. 基础概念与原理拆解

在动手写代码之前，我们需要理解几个关键概念。

2.1 阿基米德螺线

这是本项目图形部分的核心数学基础。阿基米德螺线的极坐标方程为：r = a + b * θ其中：

• r是极径（到原点的距离）。
• θ是极角（弧度）。
• a是起始极径，控制螺线起始点离原点的距离。
• b是控制螺线间距的参数，b越大，相邻线圈的间距越大。

在直角坐标系中，我们可以将其转换为：

x = r * cos(θ) = (a + b * θ) * cos(θ) y = r * sin(θ) = (a + b * θ) * sin(θ)

我们的“蛇头”将按照θ从0开始逐步增加，计算出连续的(x, y)坐标，从而形成螺旋运动轨迹。

2.2 Pygame 基础

Pygame 是一个用于编写电子游戏的Python库，它提供了对图形、声音、事件输入的简单控制。在本项目中，我们主要用它来：

• 创建一个窗口。
• 在窗口上绘制像素点（蛇头）和线条（蛇身）。
• 控制绘制的帧率，形成动画效果。

2.3 与AIGC的联动原理（ControlNet）

这是项目的“升华”部分。我们程序生成的螺旋线，最终是一张黑白线稿。现代的AIGC模型如Stable Diffusion，可以通过ControlNet等控制网络，接受线稿、深度图、姿态图等作为输入条件，从而生成在结构上遵循线稿，但风格、材质、细节完全由AI发挥的图像。 简单流程是：程序生成螺旋线图->保存为PNG线稿->导入Stable Diffusion + ControlNet->输入文本提示词（如“一盘精致的蚊香，烟雾缭绕，摄影风格”）->生成最终艺术图像。

3. 开发环境准备

我们将使用Python作为开发语言。请确保你的环境满足以下要求：

1. Python 版本: 推荐使用 Python 3.8 及以上版本。你可以在终端使用python --version命令检查。
2. 安装 Pygame: Pygame 是本项目的核心图形库。使用 pip 安装即可。

   pip install pygame

3. （可选）AIGC环境准备: 如果你想体验与Stable Diffusion联动的部分，需要配置相应的环境。这通常需要一台配备NVIDIA显卡（显存建议8G以上）的电脑，并安装PyTorch、Diffusers库或使用Stable Diffusion WebUI。由于这部分配置复杂且非本文核心，我们将主要聚焦于本地螺旋线生成程序。文末会给出与AI工具对接的清晰思路和接口说明。

4. 项目结构与核心流程

我们将项目拆解为以下步骤，确保每一步都清晰可执行：

1. 初始化Pygame窗口：设置画布大小、标题和颜色。
2. 定义螺旋线参数与绘制逻辑：实现阿基米德螺线的参数计算，并确定绘制速度。
3. 实现“蛇”的动画循环：在每一帧，计算蛇头新位置，绘制蛇头，并将旧位置存入“蛇身”列表，绘制蛇身轨迹。
4. 控制与美化：添加暂停、重启功能，并设置轨迹颜色渐变，让视觉效果更接近蚊香燃烧。
5. 保存生成结果：将绘制好的螺旋线图像保存到本地，为后续AI处理做准备。
6. （拓展）与AI绘画工具对接思路：说明如何将生成的线稿用于ControlNet。

5. 完整代码实现与分步解析

我们将创建一个名为snake_mosquito_coil.py的Python文件。

5.1 导入库与初始化

# snake_mosquito_coil.py import pygame import sys import math from pygame.locals import * # 初始化pygame pygame.init() # 设置窗口尺寸 WIDTH, HEIGHT = 800, 800 CENTER = (WIDTH // 2, HEIGHT // 2) # 窗口中心作为螺旋线原点 # 颜色定义 (RGB) BACKGROUND = (15, 15, 25) # 深蓝黑色背景，模拟夜空 SNAKE_HEAD = (255, 50, 50) # 蛇头为亮红色 # 蛇身（蚊香轨迹）颜色将使用渐变，从亮黄到暗红 # 创建窗口 screen = pygame.display.set_mode((WIDTH, HEIGHT)) pygame.display.set_caption('蛇蛇牌蚊香 - B站AI创造公开赛 模拟') clock = pygame.time.Clock() # 用于控制帧率

代码解析：这里定义了基本的窗口参数。将背景色设为深色是为了更好地突出发光的“蚊香”轨迹。CENTER变量很重要，它将作为我们螺旋线的数学原点。

5.2 定义螺旋线生成器与绘制状态

# 螺旋线参数 theta = 0.0 # 初始角度 delta_theta = 0.05 # 每次角度增量，控制“蛇”爬行速度和线条平滑度 a = 10 # 起始极径，控制螺旋线起始圈大小 b = 0.5 # 间距参数，控制螺旋线每圈之间的间隔 # 存储蛇身轨迹的点 trail_points = [] # 控制动画是否进行 running = True paused = False # 颜色渐变函数：根据轨迹点索引计算颜色，模拟蚊香燃烧效果 def get_trail_color(index, total_length): # 从亮黄色(255,255,100)渐变到暗红色(150, 40, 20) progress = index / max(total_length, 1) r = int(255 * (1 - progress) + 150 * progress) g = int(255 * (1 - progress) + 40 * progress) b = int(100 * (1 - progress) + 20 * progress) return (r, g, b)

代码解析：

• theta,delta_theta,a,b是阿基米德螺线的核心参数，调整它们可以改变螺旋线的形态。
• trail_points列表用于存储“蛇”经过的每一个历史位置，我们通过连接这些点来绘制轨迹。
• get_trail_color函数实现了简单的颜色线性插值。靠近蛇头（新画的）部分是亮黄色，像正在燃烧的蚊香头；远离蛇头（早画的）部分渐变为暗红色，像燃烧后的灰烬。这个细节大大增强了视觉效果。

5.3 主动画循环

这是整个程序的心脏，它不断运行，处理事件、更新状态、重新绘制画面。

# 主循环 while running: # 1. 处理事件（如退出、按键） for event in pygame.event.get(): if event.type == QUIT: running = False elif event.type == KEYDOWN: if event.key == K_ESCAPE: # 按ESC退出 running = False elif event.key == K_SPACE: # 按空格暂停/继续 paused = not paused elif event.key == K_r: # 按R键重置 trail_points = [] theta = 0.0 elif event.key == K_s: # 按S键保存当前画面 pygame.image.save(screen, 'snake_coil_screenshot.png') print("图像已保存为 'snake_coil_screenshot.png'") if not paused: # 2. 更新“蛇头”位置（根据螺旋线公式） r = a + b * theta # 计算当前极径 # 将极坐标(r, theta)转换为屏幕直角坐标(x, y) x = CENTER[0] + r * math.cos(theta) # 注意：屏幕坐标系y轴向下 y = CENTER[1] + r * math.sin(theta) # 将当前点加入轨迹列表 current_point = (int(x), int(y)) trail_points.append(current_point) # 3. 绘制 screen.fill(BACKGROUND) # 用背景色清空屏幕 # 绘制轨迹（蛇身/蚊香） if len(trail_points) > 1: for i in range(1, len(trail_points)): # 根据点在轨迹中的位置获取渐变颜色 color = get_trail_color(i, len(trail_points)) start_pos = trail_points[i-1] end_pos = trail_points[i] # 用逐渐变粗的线条绘制，模拟蚊香实体感 line_width = max(1, int(i / 500)) # 随着轨迹延长，线条稍微变粗 pygame.draw.line(screen, color, start_pos, end_pos, line_width) # 绘制蛇头（一个亮色小圆点） pygame.draw.circle(screen, SNAKE_HEAD, current_point, 4) # 4. 更新角度，准备下一帧 theta += delta_theta # 简单限制：当螺旋线画到窗口边缘时停止角度增加（可选） if max(abs(x - CENTER[0]), abs(y - CENTER[1])) > min(WIDTH, HEIGHT) // 2 - 20: paused = True # 画到边缘后自动暂停 # 5. 显示帧率（左上角） fps = int(clock.get_fps()) font = pygame.font.SysFont(None, 36) fps_text = font.render(f'FPS: {fps} | 点数: {len(trail_points)}', True, (200, 200, 200)) screen.blit(fps_text, (10, 10)) # 6. 显示操作提示（左下角） hint_font = pygame.font.SysFont(None, 28) hints = [ "空格键: 暂停/继续", "R键: 重置", "S键: 保存图片", "ESC键: 退出" ] for i, hint in enumerate(hints): hint_surface = hint_font.render(hint, True, (180, 180, 220)) screen.blit(hint_surface, (10, HEIGHT - 100 + i * 25)) # 更新整个屏幕显示 pygame.display.flip() # 控制帧率为60 clock.tick(60) # 退出pygame pygame.quit() sys.exit()

代码解析：

1. 事件处理：我们监听了退出事件、ESC键、空格键（暂停）、R键（重置）、S键（保存截图）。这提供了基本的交互能力。
2. 位置更新：核心就是那两行极坐标转直角坐标的公式。theta每帧增加delta_theta，从而驱动“蛇头”沿螺旋线移动。
3. 轨迹绘制：我们没有只画当前点，而是将历史点连接起来。通过get_trail_color实现颜色渐变，并用line_width实现线条轻微变粗，让“蚊香”看起来更有立体感。
4. 边界控制：当蛇头移动到接近窗口边缘时，自动暂停动画，防止无限画下去。
5. 信息显示：在屏幕上显示帧率和轨迹点数，方便调试。显示操作提示，提升用户体验。

6. 运行结果与效果验证

将上述所有代码块按顺序组合，保存为snake_mosquito_coil.py文件。在终端中进入文件所在目录，运行：

python snake_mosquito_coil.py

预期效果：

1. 一个深色背景的窗口会弹出，标题为“蛇蛇牌蚊香”。
2. 一个红色小圆点（蛇头）将从屏幕中心附近开始，以平滑的螺旋轨迹向外移动。
3. 在蛇头后方，会留下一条逐渐延长的、颜色从亮黄渐变到暗红的轨迹，非常像一盘正在被点燃的蚊香。
4. 你可以通过空格键暂停/继续动画，R键清空画面重新开始，S键将当前窗口画面保存为PNG图片。
5. 窗口左上角显示实时帧率和已绘制的点数，左下角显示操作提示。

成功运行的标志：看到平滑、颜色渐变的螺旋线动画，并且可以通过键盘进行交互控制。生成的螺旋线图像（通过S键保存）应该是背景透明（实际是深色背景）的蚊香状线条图，类似下图所示的线稿： （想象一个从中心向外盘旋的多圈螺旋线，线条有颜色渐变）

7. 常见问题与排查思路

8. 进阶探索与AIGC联动最佳实践

基础版本完成后，我们可以从工程和创意两个角度进行优化和拓展。

8.1 工程优化建议

• 参数可配置化：将a,b,delta_theta、颜色渐变起止值、窗口大小等定义为文件开头的常量，甚至通过命令行参数传入，方便快速调整效果。
• 轨迹点优化：当前存储所有点，内存会持续增长。对于极长的动画，可考虑只存储最近N个点，或定期稀释采样（例如每10个点存1个），在视觉损失不大的情况下提升性能。
• 增加更多交互：例如鼠标点击可以设置新的螺旋线中心，滚动鼠标可以调整b参数实时改变螺旋线间距。
• 导出矢量图形：除了保存位图（PNG），可以尝试用cairo或reportlab库将轨迹点导出为SVG或PDF矢量图，获得无限放大的清晰度。

8.2 与Stable Diffusion (ControlNet) 联动实践

这才是将项目从“程序演示”升级为“AI创造”的关键。思路如下：

1. 生成高质量线稿：调整程序，生成一张背景纯白、线条纯黑的蚊香螺旋线图。这需要修改颜色设置，并将线条加粗以确保清晰。

   # 修改颜色设置用于生成AI线稿 BACKGROUND_AI = (255, 255, 255) # 白色背景 LINE_COLOR_AI = (0, 0, 0) # 黑色线条 LINE_WIDTH_AI = 3 # 更粗的线条

   运行程序，画完完整螺旋线后按S键保存，得到coil_lineart.png。

2. 使用Stable Diffusion WebUI处理：

   • 启动你的Stable Diffusion WebUI。
   • 进入“文生图”或“图生图”标签页。
   • 在下方找到并展开ControlNet单元。
   • 将coil_lineart.png拖入ControlNet的图像上传区域。
   • 预处理器选择invert（如果线稿是黑底白线）或none（如果是白底黑线）。
   • 模型选择control_v11p_sd15_lineart或control_v11p_sd15_scribble（对于手绘线稿风格）。
   • 控制权重、引导时机等参数根据效果调整。
   • 在正面提示词中输入你想要的画面，例如：

     masterpiece, best quality, a spiral mosquito coil, smoking, on a wooden table, intricate details, photorealistic, studio lighting （杰作，最佳质量，一盘螺旋蚊香，烟雾缭绕，在木桌上，细节复杂，照片级真实感，影室灯光）

   • 在负面提示词中输入常见的质量负面词。
   • 选择一个大模型（如realisticVision或majicmix等写实模型），调整采样步数、尺寸（建议与线稿图比例一致），然后点击生成。

3. 迭代与创意：你可以尝试：

   • 生成不同风格（油画、水彩、赛博朋克）的蚊香。
   • 修改提示词，生成“巧克力蚊香”、“发光能量线圈”、“藤蔓缠绕的古老符号”等创意图像。
   • 不仅限于蚊香，修改程序生成其他数学曲线（心形线、玫瑰线），再用同样的流程让AI赋予其材质和意境。

9. 总结

通过“蛇蛇牌蚊香”这个项目，我们完成了一次从数学原理到图形编程，再到AIGC联动的完整创意技术实践。它看似简单，却串联起了多个有价值的技术点：

• 核心技能：你掌握了使用Pygame进行动态图形绘制的基本方法，理解了如何用代码实现参数化动画。
• 数学应用：你看到了阿基米德螺线这个抽象的数学公式，如何通过编程变成屏幕上生动的视觉图案。
• 项目思维：你学会了如何为一个创意想法设计程序结构，包括状态管理、事件响应、用户交互和渲染优化。
• 前沿视野：你了解了如何将传统程序生成的“结构”与AI模型的“风格化”能力相结合，这正是当前AIGC赋能传统创作的核心模式之一。

这个项目的代码本身是一个绝佳的起点。你可以基于它，去创造更复杂的动态图形、交互艺术，或者将其作为生成特定结构线稿的引擎，源源不断地为你的AI绘画提供创意素材。技术不仅是实现功能的工具，更是表达创意的画笔。希望这个“蚊香”能点燃你更多关于编程与艺术结合的灵感。