Initial project baseline
This commit is contained in:
@@ -0,0 +1,396 @@
|
||||
import argparse
|
||||
import json
|
||||
import logging
|
||||
import os
|
||||
import random
|
||||
import sys
|
||||
from pathlib import Path
|
||||
|
||||
import numpy as np
|
||||
from PIL import ImageFont
|
||||
|
||||
from . import paths
|
||||
|
||||
BASE_DIR = paths.BASE_DIR
|
||||
RUNTIME_DIR = paths.RUNTIME_DIR
|
||||
ASSETS_DIR = paths.ASSETS_DIR
|
||||
FONTS_DIR = paths.FONTS_DIR
|
||||
PROJECT_DEFAULT_FONT = paths.PROJECT_DEFAULT_FONT
|
||||
|
||||
# 配置日志：只写入文件，不干扰控制台输出
|
||||
logging.basicConfig(
|
||||
filename=str(RUNTIME_DIR / "ewc_concurrency.log"),
|
||||
level=logging.DEBUG,
|
||||
format='%(asctime)s - %(levelname)s - %(message)s',
|
||||
filemode='w'
|
||||
)
|
||||
|
||||
# ==================== 0. 配置区（默认值） ====================
|
||||
MODE = "IMAGE"
|
||||
|
||||
# --- Image Mode ---
|
||||
MASK_IMAGE_PATH = "7887.png"
|
||||
IMAGE_CANVAS_MODE = "WIDTH"
|
||||
EXPAND_FOR_SPIRAL = True # 放大画布使螺旋填充覆盖边角
|
||||
EXPAND_RATIO = 2.5 # 更大倍率确保覆盖边缘
|
||||
FILL_CORNERS = False
|
||||
CORNER_FILL_RATIO = 0.15
|
||||
|
||||
# --- Text Mode ---
|
||||
MASK_TEXT = "A"
|
||||
MASK_FONT_PATH = str(PROJECT_DEFAULT_FONT)
|
||||
MASK_FONT_SIZE = 3000
|
||||
|
||||
# --- 自动画幅与清晰度 ---
|
||||
AUTO_EXPAND_CANVAS = True
|
||||
BASE_HD_WIDTH = 8000
|
||||
BASE_HD_HEIGHT = 4000
|
||||
MIN_READABLE_HEIGHT_PX = 25
|
||||
WORK_SCALE = 0.25
|
||||
|
||||
# --- 阴阳刻 ---
|
||||
FILL_ON = "BLACK"
|
||||
|
||||
# --- 数据与字体 ---
|
||||
EXCEL_PATH = "四个方向汇总录取名单.xlsx"
|
||||
DATA_COL_INDEX = 1
|
||||
WEIGHT_COL_INDEX = None
|
||||
WEIGHT_COL_NAME = None
|
||||
REMOVE_DUPLICATES = False
|
||||
ENABLE_STROKE_WEIGHTS = True
|
||||
|
||||
WC_FONT_PATH = str(PROJECT_DEFAULT_FONT)
|
||||
FONT_FALLBACK_PATHS = (
|
||||
"/System/Library/Fonts/STHeiti Medium.ttc",
|
||||
"/System/Library/Fonts/Hiragino Sans GB.ttc",
|
||||
"/Library/Fonts/Arial Unicode.ttf",
|
||||
)
|
||||
|
||||
# --- 填充策略 ---
|
||||
N_REPETITIONS = 1
|
||||
TARGET_FILL_RATIO = 0.0 # 关闭填充率检测
|
||||
SIZE_RATIO = 2.0
|
||||
PACKING_EFFICIENCY = 0.85
|
||||
|
||||
# --- 分层采样（边缘覆盖） ---
|
||||
ENABLE_STRATIFIED_SAMPLING = True
|
||||
STRATIFIED_BANDS = 3 # Mix Center, Middle, and Edge
|
||||
|
||||
# --- 填充率补偿（低填充时略增字号） ---
|
||||
GROW_FONT_ON_LOW_FILL = False # 关闭
|
||||
GROW_FONT_STEP = 1.05
|
||||
|
||||
# --- 填充率检测 ---
|
||||
MIN_ACCEPT_FILL_RATIO = 0.75
|
||||
FILL_RETRY_RELAX_LARGE_CAP = True
|
||||
FILL_RETRY_MAX_ROUNDS = 3
|
||||
FILL_RETRY_MAX_SCALE = 1.5
|
||||
|
||||
# --- 智能字号搜索 ---
|
||||
REQUIRE_ALL_WORDS = True
|
||||
MIN_FONT_SIZE = int(MIN_READABLE_HEIGHT_PX * WORK_SCALE)
|
||||
USER_MIN_FONT_SIZE = None
|
||||
USER_MAX_FONT_SIZE = None
|
||||
MIN_FONT_FLOOR = 2
|
||||
FONT_SCALE_MIN = 0.5
|
||||
FONT_SCALE_MAX = 1.2
|
||||
SCALE_SEARCH_STEPS = 7
|
||||
SCALE_SEARCH_ROUNDS = 5
|
||||
SCALE_DECAY = 0.85
|
||||
SCALE_FLOOR = 0.25
|
||||
AUTO_SHRINK_ROUNDS = 4
|
||||
LOG_WEIGHT_RATIO = 0.72
|
||||
RANK_WEIGHT_RATIO = 0.28
|
||||
|
||||
# --- 大字号智能降级 ---
|
||||
# 开启后，如果填不满，会自动尝试减少大字号的数量，给小词腾空间
|
||||
ENABLE_SMART_LARGE_FONT_REDUCTION = True
|
||||
LIMIT_LARGE_FONTS = True
|
||||
LARGE_FONT_LIMIT_RATIO = 0.2 # 初始允许 20% 的词是大字
|
||||
LARGE_FONT_THRESHOLD_RATIO = 0.8 # 超过最大字号 80% 算大字
|
||||
LARGE_FONT_CAP_RATIO = 0.6 # 被限制时，缩小到阈值的 60%
|
||||
|
||||
# --- 点阵补偿 ---
|
||||
ENABLE_DOT_MATRIX = False
|
||||
DOT_SPACING = 15
|
||||
DOT_RADIUS = 0
|
||||
DOT_SAFETY_BUFFER = 12
|
||||
|
||||
# --- 画布重试 ---
|
||||
CANVAS_RETRY_MAX_ROUNDS = 1
|
||||
CANVAS_RETRY_GROWTH = 1.12
|
||||
|
||||
# --- 配色 ---
|
||||
DARK_COLOR_PALETTE = (
|
||||
"#102A43",
|
||||
"#1F4E5F",
|
||||
"#206A5D",
|
||||
"#7B341E",
|
||||
"#5D1F45",
|
||||
)
|
||||
LIGHT_COLOR_PALETTE = (
|
||||
"#EAF2FF",
|
||||
"#CDECF6",
|
||||
"#CFF7E6",
|
||||
"#FFD8C2",
|
||||
"#F6D1EB",
|
||||
)
|
||||
FONT_COLOR = "#000000" # 统一字体颜色，None 则使用调色板
|
||||
|
||||
# --- 输出 ---
|
||||
MAX_ATTEMPTS = 5
|
||||
OUTPUT_DIR = "."
|
||||
OUTPUT_PREFIX = ""
|
||||
OUTPUT_PNG = "Efficient_Result_HD_AutoResize.png"
|
||||
OUTPUT_SVG = "Efficient_Result_HD_AutoResize.svg"
|
||||
DB_PATH = "wordcloud_hd.db"
|
||||
METRICS_FILE = "metrics.json"
|
||||
SAVE_DEBUG_IMAGES = True
|
||||
DEBUG_OUTPUT_DIR = "output"
|
||||
|
||||
# --- 可复现性 ---
|
||||
SEED = None
|
||||
LAYOUT_ORDER_MODE_SORTED = "SORTED"
|
||||
LAYOUT_ORDER_MODE_INTERLEAVED_RANDOM = "INTERLEAVED_RANDOM"
|
||||
VALID_LAYOUT_ORDER_MODES = (
|
||||
LAYOUT_ORDER_MODE_SORTED,
|
||||
LAYOUT_ORDER_MODE_INTERLEAVED_RANDOM,
|
||||
)
|
||||
LAYOUT_ORDER_MODE = LAYOUT_ORDER_MODE_SORTED
|
||||
LAYOUT_SEED = None
|
||||
|
||||
KNOWN_CONFIG_KEYS = {
|
||||
'MODE', 'MASK_IMAGE_PATH', 'IMAGE_CANVAS_MODE', 'EXPAND_FOR_SPIRAL', 'EXPAND_RATIO', 'FILL_CORNERS',
|
||||
'CORNER_FILL_RATIO', 'MASK_TEXT', 'MASK_FONT_PATH', 'MASK_FONT_SIZE', 'AUTO_EXPAND_CANVAS',
|
||||
'BASE_HD_WIDTH', 'BASE_HD_HEIGHT', 'MIN_READABLE_HEIGHT_PX', 'WORK_SCALE', 'FILL_ON', 'EXCEL_PATH',
|
||||
'DATA_COL_INDEX', 'WEIGHT_COL_INDEX', 'WEIGHT_COL_NAME', 'REMOVE_DUPLICATES', 'ENABLE_STROKE_WEIGHTS',
|
||||
'WC_FONT_PATH',
|
||||
'FONT_FALLBACK_PATHS',
|
||||
'N_REPETITIONS', 'TARGET_FILL_RATIO', 'SIZE_RATIO', 'PACKING_EFFICIENCY', 'ENABLE_STRATIFIED_SAMPLING',
|
||||
'STRATIFIED_BANDS', 'GROW_FONT_ON_LOW_FILL', 'GROW_FONT_STEP', 'MIN_ACCEPT_FILL_RATIO',
|
||||
'FILL_RETRY_RELAX_LARGE_CAP', 'FILL_RETRY_MAX_ROUNDS', 'FILL_RETRY_MAX_SCALE', 'REQUIRE_ALL_WORDS',
|
||||
'MIN_FONT_SIZE', 'USER_MIN_FONT_SIZE', 'USER_MAX_FONT_SIZE', 'MIN_FONT_FLOOR', 'FONT_SCALE_MIN',
|
||||
'FONT_SCALE_MAX', 'SCALE_SEARCH_STEPS', 'SCALE_SEARCH_ROUNDS', 'SCALE_DECAY', 'SCALE_FLOOR',
|
||||
'LOG_WEIGHT_RATIO', 'RANK_WEIGHT_RATIO',
|
||||
'AUTO_SHRINK_ROUNDS', 'ENABLE_SMART_LARGE_FONT_REDUCTION', 'LIMIT_LARGE_FONTS',
|
||||
'LARGE_FONT_LIMIT_RATIO', 'LARGE_FONT_THRESHOLD_RATIO', 'LARGE_FONT_CAP_RATIO', 'ENABLE_DOT_MATRIX',
|
||||
'DOT_SPACING', 'DOT_RADIUS', 'DOT_SAFETY_BUFFER', 'CANVAS_RETRY_MAX_ROUNDS', 'CANVAS_RETRY_GROWTH',
|
||||
'DARK_COLOR_PALETTE', 'LIGHT_COLOR_PALETTE', 'FONT_COLOR', 'MAX_ATTEMPTS', 'OUTPUT_DIR', 'OUTPUT_PREFIX',
|
||||
'OUTPUT_PNG', 'OUTPUT_SVG', 'DB_PATH', 'METRICS_FILE', 'SAVE_DEBUG_IMAGES', 'DEBUG_OUTPUT_DIR', 'SEED',
|
||||
'LAYOUT_ORDER_MODE', 'LAYOUT_SEED'
|
||||
}
|
||||
|
||||
CONFIG_ALIASES = {
|
||||
'seed': 'SEED',
|
||||
'layout_order_mode': 'LAYOUT_ORDER_MODE',
|
||||
'layout_seed': 'LAYOUT_SEED',
|
||||
'excel_path': 'EXCEL_PATH',
|
||||
'mask_image_path': 'MASK_IMAGE_PATH',
|
||||
'output_dir': 'OUTPUT_DIR',
|
||||
'output_prefix': 'OUTPUT_PREFIX',
|
||||
'mode': 'MODE',
|
||||
'work_scale': 'WORK_SCALE',
|
||||
'weight_col_index': 'WEIGHT_COL_INDEX',
|
||||
'weight_col_name': 'WEIGHT_COL_NAME',
|
||||
'min_font_size': 'USER_MIN_FONT_SIZE',
|
||||
'max_font_size': 'USER_MAX_FONT_SIZE',
|
||||
'font_color': 'FONT_COLOR',
|
||||
'stroke_weights': 'ENABLE_STROKE_WEIGHTS',
|
||||
}
|
||||
|
||||
CRITICAL_TYPE_CHECKS = {
|
||||
'MODE': str,
|
||||
'WORK_SCALE': (int, float),
|
||||
'DATA_COL_INDEX': int,
|
||||
'WEIGHT_COL_INDEX': (int, type(None)),
|
||||
'WEIGHT_COL_NAME': (str, type(None)),
|
||||
'FONT_FALLBACK_PATHS': (list, tuple),
|
||||
'USER_MIN_FONT_SIZE': (int, float, type(None)),
|
||||
'USER_MAX_FONT_SIZE': (int, float, type(None)),
|
||||
'MAX_ATTEMPTS': int,
|
||||
'SAVE_DEBUG_IMAGES': bool,
|
||||
'REMOVE_DUPLICATES': bool,
|
||||
'ENABLE_STROKE_WEIGHTS': bool,
|
||||
'CANVAS_RETRY_MAX_ROUNDS': int,
|
||||
'CANVAS_RETRY_GROWTH': (int, float),
|
||||
'LOG_WEIGHT_RATIO': (int, float),
|
||||
'RANK_WEIGHT_RATIO': (int, float),
|
||||
'SEED': (int, type(None)),
|
||||
'LAYOUT_ORDER_MODE': str,
|
||||
'LAYOUT_SEED': (int, type(None)),
|
||||
}
|
||||
|
||||
DEFAULT_CONFIG = {k: v for k, v in globals().items() if k in KNOWN_CONFIG_KEYS}
|
||||
|
||||
|
||||
def parse_args():
|
||||
parser = argparse.ArgumentParser(description="Efficient WordCloud generator")
|
||||
parser.add_argument("--config", type=str, help="JSON 配置文件路径")
|
||||
parser.add_argument("--seed", type=int, help="随机种子（可复现）")
|
||||
parser.add_argument("--layout-order-mode", type=lambda s: s.upper(), choices=VALID_LAYOUT_ORDER_MODES, help="布局顺序模式")
|
||||
parser.add_argument("--layout-seed", type=int, help="布局顺序随机种子")
|
||||
parser.add_argument("--excel-path", type=str, help="Excel 输入路径")
|
||||
parser.add_argument("--mask-image-path", type=str, help="掩膜图片路径（IMAGE 模式）")
|
||||
parser.add_argument("--output-dir", type=str, help="输出目录")
|
||||
parser.add_argument("--output-prefix", type=str, help="输出文件前缀")
|
||||
parser.add_argument("--mode", type=str, choices=["TEXT", "IMAGE"], help="掩膜模式")
|
||||
parser.add_argument("--work-scale", type=float, help="运算缩放比例")
|
||||
parser.add_argument("--weight-col-index", type=int, help="Excel 权重列索引")
|
||||
parser.add_argument("--weight-col-name", type=str, help="Excel 权重列名（优先于索引）")
|
||||
parser.add_argument("--min-font-size", type=float, help="覆盖最小字号")
|
||||
parser.add_argument("--max-font-size", type=float, help="覆盖最大字号")
|
||||
return parser.parse_args()
|
||||
|
||||
|
||||
def _warn(msg):
|
||||
print(f"[WARN] {msg}")
|
||||
|
||||
|
||||
def _resolve_path(path_str):
|
||||
p = Path(path_str)
|
||||
if p.is_absolute():
|
||||
return p
|
||||
return BASE_DIR / p
|
||||
|
||||
|
||||
def _with_prefix(filename, prefix):
|
||||
if not prefix:
|
||||
return filename
|
||||
return f"{prefix}_{filename}"
|
||||
|
||||
|
||||
def apply_json_config(config_path):
|
||||
cfg_path = _resolve_path(config_path)
|
||||
if not cfg_path.exists():
|
||||
print(f"错误: 配置文件不存在: {cfg_path}")
|
||||
sys.exit(1)
|
||||
|
||||
try:
|
||||
with cfg_path.open("r", encoding="utf-8") as f:
|
||||
data = json.load(f)
|
||||
except Exception as e:
|
||||
print(f"错误: 读取配置文件失败: {e}")
|
||||
sys.exit(1)
|
||||
|
||||
if not isinstance(data, dict):
|
||||
print("错误: 配置文件顶层必须是 JSON 对象")
|
||||
sys.exit(1)
|
||||
|
||||
normalized_data = {}
|
||||
for key, value in data.items():
|
||||
key_upper = CONFIG_ALIASES.get(key, key)
|
||||
normalized_data[key_upper] = value
|
||||
|
||||
for key in normalized_data.keys():
|
||||
if key not in KNOWN_CONFIG_KEYS:
|
||||
_warn(f"未知配置键: {key}")
|
||||
|
||||
for key, expected in CRITICAL_TYPE_CHECKS.items():
|
||||
if key in normalized_data and not isinstance(normalized_data[key], expected):
|
||||
print(f"错误: 配置键 {key} 类型错误，期望 {expected}，实际 {type(normalized_data[key])}")
|
||||
sys.exit(1)
|
||||
|
||||
for key, value in normalized_data.items():
|
||||
if key in KNOWN_CONFIG_KEYS:
|
||||
globals()[key] = value
|
||||
|
||||
|
||||
def apply_cli_overrides(args):
|
||||
mapping = {
|
||||
'seed': 'SEED',
|
||||
'layout_order_mode': 'LAYOUT_ORDER_MODE',
|
||||
'layout_seed': 'LAYOUT_SEED',
|
||||
'excel_path': 'EXCEL_PATH',
|
||||
'mask_image_path': 'MASK_IMAGE_PATH',
|
||||
'output_dir': 'OUTPUT_DIR',
|
||||
'output_prefix': 'OUTPUT_PREFIX',
|
||||
'mode': 'MODE',
|
||||
'work_scale': 'WORK_SCALE',
|
||||
'weight_col_index': 'WEIGHT_COL_INDEX',
|
||||
'weight_col_name': 'WEIGHT_COL_NAME',
|
||||
'min_font_size': 'USER_MIN_FONT_SIZE',
|
||||
'max_font_size': 'USER_MAX_FONT_SIZE',
|
||||
}
|
||||
for arg_key, cfg_key in mapping.items():
|
||||
value = getattr(args, arg_key)
|
||||
if value is not None:
|
||||
globals()[cfg_key] = value
|
||||
|
||||
|
||||
def _resolve_font_path(configured_path, fallback_paths, *, role):
|
||||
candidates = [_resolve_path(configured_path), *[Path(path) for path in fallback_paths]]
|
||||
errors = []
|
||||
|
||||
for index, candidate in enumerate(candidates):
|
||||
if not candidate.exists():
|
||||
errors.append(f"{candidate}: missing")
|
||||
continue
|
||||
try:
|
||||
ImageFont.truetype(str(candidate), 32)
|
||||
if index == 0:
|
||||
print(f"[Font] {role} 使用项目字体: {candidate}")
|
||||
else:
|
||||
_warn(f"{role} 字体未命中项目内资源，回退到系统字体: {candidate}")
|
||||
return str(candidate)
|
||||
except OSError as exc:
|
||||
errors.append(f"{candidate}: {exc}")
|
||||
|
||||
print(f"错误: {role} 字体初始化失败。候选路径: {errors}")
|
||||
sys.exit(1)
|
||||
|
||||
|
||||
def finalize_runtime_config():
|
||||
global EXCEL_PATH, MASK_IMAGE_PATH, MASK_FONT_PATH, WC_FONT_PATH
|
||||
global OUTPUT_DIR, OUTPUT_PNG, OUTPUT_SVG, DB_PATH, METRICS_FILE, DEBUG_OUTPUT_DIR, MIN_FONT_SIZE
|
||||
global LAYOUT_ORDER_MODE, LAYOUT_SEED
|
||||
|
||||
# 运行时派生字段
|
||||
MIN_FONT_SIZE = int(MIN_READABLE_HEIGHT_PX * WORK_SCALE)
|
||||
|
||||
if LAYOUT_SEED is None:
|
||||
LAYOUT_SEED = SEED
|
||||
|
||||
LAYOUT_ORDER_MODE = str(LAYOUT_ORDER_MODE).upper()
|
||||
if LAYOUT_ORDER_MODE not in VALID_LAYOUT_ORDER_MODES:
|
||||
print(f"错误: 不支持的 LAYOUT_ORDER_MODE: {LAYOUT_ORDER_MODE}")
|
||||
sys.exit(1)
|
||||
|
||||
EXCEL_PATH = str(_resolve_path(EXCEL_PATH))
|
||||
MASK_IMAGE_PATH = str(_resolve_path(MASK_IMAGE_PATH))
|
||||
|
||||
MASK_FONT_PATH = _resolve_font_path(MASK_FONT_PATH, FONT_FALLBACK_PATHS, role="mask")
|
||||
WC_FONT_PATH = _resolve_font_path(WC_FONT_PATH, FONT_FALLBACK_PATHS, role="layout")
|
||||
|
||||
OUTPUT_DIR = str(_resolve_path(OUTPUT_DIR))
|
||||
Path(OUTPUT_DIR).mkdir(parents=True, exist_ok=True)
|
||||
|
||||
OUTPUT_PNG = str(Path(OUTPUT_DIR) / _with_prefix(Path(OUTPUT_PNG).name, OUTPUT_PREFIX))
|
||||
OUTPUT_SVG = str(Path(OUTPUT_DIR) / _with_prefix(Path(OUTPUT_SVG).name, OUTPUT_PREFIX))
|
||||
DB_PATH = str(Path(OUTPUT_DIR) / _with_prefix(Path(DB_PATH).name, OUTPUT_PREFIX))
|
||||
METRICS_FILE = str(Path(OUTPUT_DIR) / _with_prefix(Path(METRICS_FILE).name, OUTPUT_PREFIX))
|
||||
DEBUG_OUTPUT_DIR = str(Path(OUTPUT_DIR) / Path(DEBUG_OUTPUT_DIR).name)
|
||||
|
||||
|
||||
def set_random_seed():
|
||||
if SEED is None:
|
||||
return
|
||||
np.random.seed(SEED)
|
||||
random.seed(SEED)
|
||||
print(f"[Seed] 使用固定随机种子: {SEED}")
|
||||
|
||||
|
||||
def get_output_background():
|
||||
return "black" if FILL_ON == "WHITE" else "white"
|
||||
|
||||
|
||||
def get_output_palette():
|
||||
return LIGHT_COLOR_PALETTE if FILL_ON == "WHITE" else DARK_COLOR_PALETTE
|
||||
|
||||
|
||||
def write_metrics(metrics):
|
||||
try:
|
||||
with Path(METRICS_FILE).open("w", encoding="utf-8") as f:
|
||||
json.dump(metrics, f, ensure_ascii=False, indent=2)
|
||||
print(f"已保存: {METRICS_FILE}")
|
||||
except Exception as e:
|
||||
_warn(f"写入 metrics 失败（不影响主产物）: {e}")
|
||||
Reference in New Issue
Block a user