#!/usr/bin/env python3
"""生成纯合成车牌（无场景背景）作为 CUT Domain A"""
import sys, os
sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', 'chinese_license_plate_generator'))
from generate_multi_plate import MultiPlateGenerator
import cv2, random
import numpy as np

random.seed(42)
np.random.seed(42)

g = MultiPlateGenerator(
    'chinese_license_plate_generator/plate_model',
    'chinese_license_plate_generator/font_model'
)

out = 'gan_data/domain_A_synthetic/police_pure'
os.makedirs(out, exist_ok=True)

for i in range(500):
    num, bg, _ = g.generate_plate_number()
    # 强制police类型
    if '警' not in num:
        num = num[:6] + '警'
    img = g.generate_plate_special(num, 'white', False)
    cv2.imwrite(f'{out}/synth_{i:04d}.jpg', img)

print(f'Done: {out} ({len(os.listdir(out))} images)')