ChartPipeline / docs /11_layout_optimizer.md
Ray1ee01's picture
Upload folder using huggingface_hub
1569336 verified
|
Raw
History Blame Contribute Delete
8.59 kB
# 布局优化模块 (layout_optimizer)
布局优化模块(`layout_optimizer`)是 ChartPipeline 框架的第九个也是最后一个处理环节,负责组合并优化图表和标题 SVG 元素,生成最终的可视化作品。
## 功能简介
本模块接收前面生成的图表、标题和主体图像 SVG 元素,根据布局配置将它们整合为一个统一的最终可视化作品。模块进行精确的空间布局、元素对齐和比例调整,确保最终作品视觉平衡、结构清晰,并适合目标显示环境。这一步是整个可视化生成流程的最后阶段,对最终输出的视觉质量和专业性至关重要。
## 输入与输出
### 输入
模块接收两个主要的 SVG 输入、主题图像和一个 JSON 数据对象:
- `chart_svg`: 图表引擎生成的图表 SVG
- `title_svg`: 标题样式化模块生成的标题 SVG
- `image_svg`: 主题图像 SVG
- 包含布局配置的数据对象(通过 JSON 文件)
### 输出
模块输出一个包含最终可视化的 SVG 文件:
```json
{
"final_svg": "<svg width=\"1000\" height=\"650\" xmlns=\"http://www.w3.org/2000/svg\">...</svg>"
}
```
最终 SVG 包含:
- 整合后的标题、图表和主体图像元素
- 调整后的尺寸和位置
## 布局优化功能
模块提供多种布局优化功能:
1. **元素组合**
- 根据布局配置组合图表和标题
- 处理元素间的相对位置关系
- 创建一致的视觉层次结构
2. **空间分配**
- 优化整体画布尺寸
- 分配合适的空间给各元素
- 调整元素间距和边距
3. **对齐和分布**
- 确保元素正确对齐
- 均匀分布相关元素组
- 创建视觉上的平衡感
## 元素表示与数学建模
为了实现精确的布局优化,本模块采用矩形表示方法对各元素进行建模:
### 元素形式化定义
- 元素集合 E={T,I,C},其中:
- T: 标题(Title)
- I: 主题图像(Image)
- C: 图表(Chart)
矩形表示 R<sub>e</sub> 可以写为:
- R<sub>e</sub> = [x<sub>e</sub>, y<sub>e</sub>, x<sub>e</sub> + w<sub>e</sub>, y<sub>e</sub> + h<sub>e</sub>],表示元素的矩形区域。
对于不同的元素:
- 标题和图表有固定尺寸 (w<sub>T</sub>, h<sub>T</sub>) 和 (w<sub>C</sub>, h<sub>C</sub>)。
- 主题图像的尺寸是可变的,范围为 w<sub>I</sub> ∈ [w<sub>I,min</sub>, w<sub>I,max</sub>] 和 h<sub>I</sub> ∈ [h<sub>I,min</sub>, h<sub>I,max</sub>]。
### 矩形表示可视化
矩形模型在二维空间中可以表示为:
```
(x,y)
┌───────────────────┐
│ │
│ Element │ h
│ │
└───────────────────┘
(x+w, y+h)
w
```
其中:
- (x,y) 是元素左上角的坐标
- w 是元素的宽度
- h 是元素的高度
- (x+w, y+h) 是元素右下角的坐标
当布局中包含多个元素时,它们的位置关系可以通过这些矩形的相对位置来表示。例如,当标题在图表上方时(title_to_chart: "T"),可表示为:
```
┌───────────────┐
│ Title │
└───────────────┘
┌───────────────┐
│ │
│ Chart │
│ │
└───────────────┘
```
### 元素间的位置关系
布局优化模块使用以下位置关系描述符来定义元素之间的空间组织:
1. **上下关系(Top/Bottom)**:元素在另一元素的上方或下方
- 标题相对于图表可以是上方(T)或下方(B)
- 图像相对于图表可以是上方或下方的一部分
2. **左右关系(Left/Right)**:元素在另一元素的左侧或右侧
- 标题可以在图表的左侧(L)或右侧(R)
- 图像可以在图表的左侧或右侧
3. **组合位置关系**:结合上下和左右的复合位置
- 左上(TL)、右上(TR)、左下(BL)或右下(BR)
- 这些位置关系通过坐标约束来表达
4. **内嵌关系(Center)**:元素嵌入另一元素内部
- 标题可以内嵌于图表(C)
- 图像也可以内嵌于图表中
5. **包含关系**:一个元素完全包含另一元素
- 图表是否包含标题(chart_contains_title)
- 图表是否包含图像(chart_contains_image)
### 位置约束格式
位置约束通过如下 JSON 格式指定:
```json
{
"title_to_chart": "T", // 标题相对图表的位置:T/B/L/R/TL/TR/BL/BR、内嵌(C)
"image_to_chart": "BL", // 图像相对图表的位置:T/B/L/R/TL/TR/BL/BR、内嵌(C)
"title_to_image": "L", // 标题相对图像的位置:T/B/L/R/TL/TR/BL/BR、内嵌(C)
"chart_contains_title": false, // 图表是否包含标题
"chart_contains_image": false // 图表是否包含图像
}
```
### 布局优化数学模型
基于矩形表示,布局优化问题可以抽象为一个矩形打包问题,目标是在满足元素间位置关系约束的条件下,找到一个优化的布局方案。布局优化的数学模型包括:
1. **目标函数**
- 最小化整个布局矩形的面积
- 最大化主题图像的显示面积(对于可变尺寸的元素)
- 维持视觉平衡度量指标
2. **约束条件**
- 位置关系约束(如上述位置描述符定义)
- 非重叠约束(确保元素不重叠)
- 包含关系约束(如嵌套结构)
- 空间边界约束(如最小/最大尺寸)
通过这种数学表示,可以将布局优化问题转化为一个可以通过算法求解的形式化问题。
### 布局优化算法
模块使用以下算法和方法来求解布局优化问题:
1. **约束满足算法**
- 处理位置关系约束,将抽象的位置描述(如"T"、"BL"等)转换为具体的几何约束
- 迭代求解过程中确保所有约束条件同时满足
- 解决约束冲突和优先级处理
2. **矩形打包优化**
- 使用启发式方法(如模拟退火)寻找优化布局方案
- 逐步调整元素位置和尺寸,最小化目标函数
- 采用多目标优化权衡布局紧凑性与视觉平衡
3. **布局流程**
- 初始化:为每个元素分配初始位置和尺寸
- 约束应用:应用位置关系约束调整元素坐标
- 优化迭代:通过目标函数评估并改进布局方案
- 后处理:微调元素间距、对齐和视觉平衡
#### 约束处理方法
在处理位置关系约束时,模块采用以下方法:
1. **相对位置转换**
-`title_to_chart: "T"` 时,确保 title.y + title.height ≤ chart.y
-`image_to_chart: "BL"` 时,确保 image.x + image.width ≤ chart.x 且 image.y ≥ chart.y + chart.height
- 当涉及内嵌关系(如 `title_to_chart: "I"`)时,确保 title.x ≥ chart.x 且 title.y ≥ chart.y 且 title.x + title.width ≤ chart.x + chart.width 且 title.y + title.height ≤ chart.y + chart.height
2. **包含关系处理**
-`chart_contains_title: true` 时,调整标题位置确保其完全位于图表区域内
-`chart_contains_image: true` 时,调整图像位置确保其完全位于图表区域内
3. **间距管理**
- 在元素间保持适当的间距(margin)以确保布局美观
- 根据设计指南自动计算元素间最佳间距
- 确保间距均匀一致,提升整体视觉效果
#### 优化策略
矩形打包优化采用多阶段策略:
1. **粗优化阶段**
- 初步定位各元素,满足基本约束
- 使用贪婪算法快速获得可行解
2. **细优化阶段**
- 微调元素位置和尺寸
- 最小化布局总面积
- 平衡各元素的视觉重要性
3. **特殊情况处理**
- 当元素过多或空间受限时,自动调整布局策略
- 处理不同纵横比的图表和标题组合
- 支持响应式调整,适应不同显示环境
## 使用示例
### 命令行调用
```bash
python -m modules.layout_optimizer.layout_optimizer --input_chart chart.svg --input_title title.svg --input_image image.svg --output final.svg
```
### 程序调用
```python
from modules.layout_optimizer.layout_optimizer import process
success = process(
input_chart='chart.svg',
input_title='title.svg',
input_image='image.svg',
output='final.svg'
)
if success:
print("布局优化完成")
else:
print("布局优化失败")
```