Files
AnyImage/docs/internals/architecture.md
nan c97b4a3e41 Add image selection refinement functionality with BEN2 integration
- Implemented viewport.py for handling 3D view projections, gestures, and overlay drawing.
- Created refine_image_selection.py to manage the refinement of gesture masks and output image bounds.
- Added selection module with refine_selection.py to process and crop images based on alpha channel visibility.
- Introduced __init__.py for selection module to expose refine_selection function.
2026-08-31 22:10:05 +08:00

3.5 KiB
Raw Blame History

程序架构

AnyImage 由 Blender 交互层、BlendJob Runtime 和独立 Job Server 三部分组成。AI 依赖运行在 Server Environment 中,Blender 数据在主线程创建和修改。

对应文件

src/anyimage/
├── __init__.py       # 类型、工具、菜单与 Runtime 注册
├── menu.py           # Image Empty 上下文入口
├── panel.py          # Server 与 Debug 面板
├── preferences.py    # Storage Root、Device 与功能默认值
├── properties.py     # Scene 状态、模型状态与动态枚举
├── runtime.py        # Environment 和 JobRuntime
├── common/           # Blender 侧共享原子能力
├── operators/        # Blender 侧业务入口
└── server/           # 独立计算端

职责

层次 职责
Blender 交互层 UI、手势、RNA 状态、输入准备、结果导入和 Undo
BlendJob Runtime Environment、Server 生命周期、队列、进度、取消和 Job 目录
Job Server 媒体解码、ONNX 推理、数组派生和文件输出

Operator 向 Server 发送普通字典。Server 返回文件名和 Job 目录;Operator 的 response() 再读取这些文件,并在 Blender 主线程中创建 Image、Mesh、Material 和 Geometry Nodes Modifier。

注册与状态

flowchart TD
    Register["anyimage.register()"] --> Types["注册 Preferences / PropertyGroup / Operator / Panel"]
    Types --> Tools["注册 Image Tools 与 Cutout WorkspaceTool"]
    Tools --> RNA["挂载 Scene 与 Image Property"]
    RNA --> Menu["挂载上下文菜单和剪贴板快捷键"]
    Menu --> Runtime["注册 JobRuntime"]

__init__.py 按依赖顺序注册,并在注销时逆序清理。扩展级设置统一由 preferences.addon_preferences() 读取;runtime.py 只消费 configured_storage_root 回调,不持有 Blender 偏好设置的定位逻辑。交互中的临时状态和 Debug 参数保存在 Scene.anyimage_settings。max_ai_input_size 与 max_cutout_mesh_resolution 都在 preferences.py 定义与读取:前者由 Operator 冻结到 AI Job 请求,后者在 Blender 侧统一限制所有 Cutout Shape 的网格采样密度。Upscale 还用 AI 输入上限判断当前 Image 是否允许执行。

同步与异步路径

无需 AI 或媒体处理的功能直接在 Blender 中完成,例如 Plane、普通 Image Mask、Perspective,以及未启用 Refine Selection / Normal Map 的 Surface 和 Balloon。需要模型或独立文件处理时,功能 Operator 派生自 JobOperatorBase:

sequenceDiagram
    participant User as 用户
    participant Op as Blender Operator
    participant RT as JobRuntime
    participant Job as Server Job
    User->>Op: 执行功能
    Op->>Op: 准备文件与普通字典
    Op->>RT: 提交 Job
    RT->>Job: 排队执行
    Job-->>RT: 写入产物并返回索引
    RT-->>Op: 主线程 response
    Op->>Op: 导入数据并提交 Undo

server/ 保持纯 Python 实现;Operator 在 Blender 主线程读取产物并创建 RNA 数据。

几何模型的原始输出先由 server/models/ 适配为单帧 GeometryFrame。它统一保存 Depth、Validity、Intrinsics,以及可选的 Normal 和 Points;Depth、Normal 与 Metadata Writer 只消费该结构。MoGe-2 的 Mask 转为 0/1 Validity,DA3 的 Confidence 作为连续 Validity。

共享能力

common/ 按 AI、Image、Selection、Viewport、Depth、Material、Object 和 Node 八个业务模块提供 Blender 侧复用能力。公开函数见Common 公共函数。