mirror of
https://github.com/LostArtefacts/TRX.git
synced 2025-04-28 12:47:58 +03:00
79 lines
2 KiB
Python
79 lines
2 KiB
Python
|
import tempfile
|
||
|
|
from dataclasses import dataclass
|
||
|
|
from pathlib import Path
|
||
|
|
from subprocess import check_call
|
||
|
|
|
||
|
|
|
||
|
|
@dataclass
|
||
|
|
class IconSpec:
|
||
|
|
size: int
|
||
|
|
type: str
|
||
|
|
|
||
|
|
|
||
|
|
SPECS = [
|
||
|
|
IconSpec(size=32, type="bmp"),
|
||
|
|
IconSpec(size=16, type="bmp"),
|
||
|
|
IconSpec(size=256, type="png"),
|
||
|
|
IconSpec(size=128, type="png"),
|
||
|
|
IconSpec(size=64, type="png"),
|
||
|
|
IconSpec(size=48, type="png"),
|
||
|
|
IconSpec(size=32, type="png"),
|
||
|
|
IconSpec(size=16, type="png"),
|
||
|
|
]
|
||
|
|
|
||
|
|
|
||
|
|
def resize_transformer(path: Path, spec: IconSpec) -> None:
|
||
|
|
check_call(
|
||
|
|
[
|
||
|
|
"convert",
|
||
|
|
f"{path}[0]",
|
||
|
|
"-filter",
|
||
|
|
"lanczos",
|
||
|
|
"-resize",
|
||
|
|
f"{spec.size}x{spec.size}",
|
||
|
|
f"PNG:{path}",
|
||
|
|
]
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def quantize_transformer(path: Path, spec: IconSpec) -> None:
|
||
|
|
quantized_path = path.with_stem(f"{path.stem}-quantized")
|
||
|
|
check_call(["pngquant", path, "--output", quantized_path])
|
||
|
|
path.write_bytes(quantized_path.read_bytes())
|
||
|
|
quantized_path.unlink()
|
||
|
|
|
||
|
|
|
||
|
|
def optimize_transformer(path: Path, spec: IconSpec) -> None:
|
||
|
|
check_call(["zopflipng", "-y", path, path])
|
||
|
|
|
||
|
|
|
||
|
|
def convert_transformer(path: Path, spec: IconSpec) -> None:
|
||
|
|
if spec.type != "png":
|
||
|
|
check_call(["convert", path, f"{spec.type.upper()}:{path}"])
|
||
|
|
|
||
|
|
|
||
|
|
TRANSFORMERS = [
|
||
|
|
resize_transformer,
|
||
|
|
quantize_transformer,
|
||
|
|
optimize_transformer,
|
||
|
|
convert_transformer,
|
||
|
|
]
|
||
|
|
|
||
|
|
|
||
|
|
def generate_icon(source_path: Path, target_path: Path) -> None:
|
||
|
|
aux_paths = []
|
||
|
|
|
||
|
|
with tempfile.TemporaryDirectory() as tmpdir:
|
||
|
|
tmp_path = Path(tmpdir)
|
||
|
|
for spec in SPECS:
|
||
|
|
aux_path = tmp_path / f"{spec.size}-{spec.type}.tmp"
|
||
|
|
aux_path.write_bytes(source_path.read_bytes())
|
||
|
|
for transform in TRANSFORMERS:
|
||
|
|
transform(aux_path, spec)
|
||
|
|
|
||
|
|
aux_paths.append(aux_path)
|
||
|
|
|
||
|
|
# NOTE: image order is important for certain software.
|
||
|
|
check_call(["identify", *aux_paths])
|
||
|
|
check_call(["convert", *aux_paths, target_path])
|