47 lines
1.6 KiB
Python
Executable File
47 lines
1.6 KiB
Python
Executable File
"""绿幕抠像引擎 — 基于 FFmpeg colorkey / chromakey 滤镜.
|
||
|
||
支持将指定颜色（默认绿色）变为透明，可用于虚拟背景、画中画背景替换等场景。
|
||
|
||
注：核心领域模型已抽离到 packages/domain/chroma_key_config.py，
|
||
本模块保留薄包装层，确保向后兼容。
|
||
"""
|
||
|
||
from __future__ import annotations
|
||
|
||
import logging
|
||
from typing import Optional
|
||
|
||
from packages.domain.chroma_key_config import ( # noqa: F401 — 向后兼容
|
||
CHROMA_KEY_PRESETS,
|
||
ChromaKeyConfig,
|
||
apply_chroma_key_if_needed,
|
||
build_chromakey_filter as _build_chromakey_filter_base,
|
||
build_colorkey_filter as _build_colorkey_filter_base,
|
||
normalize_color as _normalize_color_base,
|
||
)
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
|
||
class ChromaKeyEngine:
|
||
"""绿幕抠像引擎.
|
||
|
||
薄包装层，实际逻辑委托给 packages.domain.chroma_key_config。
|
||
"""
|
||
|
||
def __init__(self, config: ChromaKeyConfig):
|
||
self.config = config
|
||
|
||
@staticmethod
|
||
def _normalize_color(color_str: str) -> str:
|
||
"""将颜色字符串转为 FFmpeg colorkey 接受的格式."""
|
||
return _normalize_color_base(color_str)
|
||
|
||
def build_filter(self, input_label: str, output_label: str) -> str:
|
||
"""构建 colorkey 滤镜字符串."""
|
||
return _build_colorkey_filter_base(self.config, input_label, output_label)
|
||
|
||
def build_filter_chromakey(self, input_label: str, output_label: str) -> str:
|
||
"""使用 chromakey 滤镜（更高级的版本，支持更多参数）."""
|
||
return _build_chromakey_filter_base(self.config, input_label, output_label)
|