game-cards-poker-design/backend/apps/exports/utils.py

"""
扑克牌渲染工具 - 使用 Pillow 在服务器端生成牌面 PNG。

渲染图层结构（从下到上）：
  1. 背景层（颜色或图片）
  2. 边框层
  3. 主体层（数字牌的花色阵列 / JQK 的人物对称 / 大小王的图案 + JOKER 文字）
  4. 角标层（左上 + 右下 旋转 180°）
"""
from PIL import Image, ImageDraw, ImageFont
import io
import os

# 标准扑克牌花色符号（用作默认渲染）
SUIT_TEXT = {
    'spade':   '♠',
    'heart':   '♥',
    'club':    '♣',
    'diamond': '♦',
}
RED_SUITS = {'heart', 'diamond'}
BLACK_SUITS = {'spade', 'club'}

# 数字牌点数 1-10 的花色位置（相对坐标 0~1）
LAYOUT_POSITIONS = {
    1:  [(0.50, 0.50)],
    2:  [(0.50, 0.25), (0.50, 0.75)],
    3:  [(0.50, 0.20), (0.50, 0.50), (0.50, 0.80)],
    4:  [(0.30, 0.25), (0.70, 0.25), (0.30, 0.75), (0.70, 0.75)],
    5:  [(0.30, 0.20), (0.70, 0.20), (0.50, 0.50), (0.30, 0.80), (0.70, 0.80)],
    6:  [(0.30, 0.20), (0.70, 0.20), (0.30, 0.50), (0.70, 0.50), (0.30, 0.80), (0.70, 0.80)],
    7:  [(0.30, 0.15), (0.70, 0.15), (0.50, 0.35), (0.30, 0.55), (0.70, 0.55), (0.30, 0.85), (0.70, 0.85)],
    8:  [(0.30, 0.15), (0.70, 0.15), (0.50, 0.32), (0.30, 0.50), (0.70, 0.50), (0.50, 0.68), (0.30, 0.85), (0.70, 0.85)],
    9:  [(0.30, 0.15), (0.70, 0.15), (0.50, 0.30), (0.22, 0.50), (0.50, 0.50), (0.78, 0.50), (0.50, 0.70), (0.30, 0.85), (0.70, 0.85)],
    10: [(0.30, 0.15), (0.70, 0.15), (0.30, 0.35), (0.70, 0.35), (0.50, 0.50), (0.30, 0.65), (0.70, 0.65), (0.30, 0.85), (0.70, 0.85)],
}


def hex_to_rgba(hex_color, alpha=255):
    """将 #RRGGBB 转为 (r, g, b, a)"""
    if not hex_color:
        return (255, 255, 255, alpha)
    h = hex_color.lstrip('#')
    if len(h) == 3:
        h = ''.join(c * 2 for c in h)
    try:
        return (int(h[0:2], 16), int(h[2:4], 16), int(h[4:6], 16), alpha)
    except Exception:
        return (255, 255, 255, alpha)


def is_red(suit):
    return suit in RED_SUITS


def is_black(suit):
    return suit in BLACK_SUITS


def get_effective_design(project, card_key):
    """合并项目级设计与单牌覆盖，返回最终 design dict"""
    base = dict(project.design or {})
    overrides = (project.card_overrides or {}).get(card_key, {})
    if overrides:
        base.update(overrides)
    return base


def load_image_safe(file_path):
    """加载图片，找不到时返回 None"""
    if not file_path:
        return None
    if not os.path.exists(file_path):
        return None
    try:
        return Image.open(file_path).convert('RGBA')
    except Exception:
        return None


def draw_background(canvas, design, project, card_key):
    """绘制背景层"""
    w, h = canvas.size
    bg_color = design.get('background_color', '#FFFFFF') or '#FFFFFF'
    canvas.paste(hex_to_rgba(bg_color, 255), (0, 0, w, h))

    bg_image = design.get('background_image')
    if bg_image:
        bg_path = os.path.join(project._meta.get_field('design').model._meta.app_config and
                                os.path.dirname(__file__) or '',
                               bg_image)
        # 优先尝试绝对路径 / media 根
        candidates = [bg_image,
                      os.path.join(os.path.dirname(os.path.dirname(os.path.dirname(__file__))), 'media', bg_image)]
        for path in candidates:
            img = load_image_safe(path)
            if img:
                img = img.resize((w, h), Image.LANCZOS)
                canvas.alpha_composite(img)
                break


def draw_border(canvas, design):
    """绘制边框"""
    w, h = canvas.size
    color = design.get('border_color', '#333333') or '#333333'
    width = int(design.get('border_width', 0) or 0)
    if width <= 0:
        return
    draw = ImageDraw.Draw(canvas)
    half = max(1, width // 2)
    draw.rectangle(((half, half), (w - half, h - half)),
                   outline=hex_to_rgba(color, 255), width=width)


def make_text_font(family, size, bold=False):
    """根据字体族名尝试加载字体；找不到时退回默认"""
    if not family:
        return ImageFont.load_default()
    # 仅在 Windows 上尝试用注册字体
    if os.name == 'nt':
        candidates = [
            family,
            family.replace(' ', '') + (' Bold' if bold else ''),
            family + (' Bold' if bold else ''),
        ]
        for name in candidates:
            try:
                # 走 Windows 字体名
                return ImageFont.truetype(name + '.ttf', size)
            except Exception:
                pass
        # 一些常见回退
        try:
            return ImageFont.truetype('arial.ttf', size)
        except Exception:
            pass
    return ImageFont.load_default()


def draw_corner_index(canvas, design, suit, rank, project, card_key):
    """绘制左上角 + 右下角（旋转 180°）的牌面角标
    角标由 上方的 rank + 小花色 组成
    """
    w, h = canvas.size
    color = hex_to_rgba(design.get('suit_symbols', {}).get(suit, {}).get('color')
                        or ('#E53935' if is_red(suit) else '#000000'), 255)

    # 角标和中心花色大小（占牌面宽度比例）
    corner_ratio = float(design.get('corner_size_ratio', 0.13) or 0.13)
    pip_ratio = float(design.get('pip_size_ratio', 0.16) or 0.16)

    corner_size = max(20, int(w * corner_ratio))
    font = make_text_font(design.get('font_family', 'Times New Roman'), corner_size)
    suit_font = make_text_font('Arial', int(corner_size * 0.9))

    # 左上角
    pad = max(8, int(w * 0.04))
    rank_str = str(rank)
    draw = ImageDraw.Draw(canvas)
    rank_bbox = draw.textbbox((0, 0), rank_str, font=font)
    rank_h = rank_bbox[3] - rank_bbox[1]
    rank_w = rank_bbox[2] - rank_bbox[0]
    suit_bbox = draw.textbbox((0, 0), SUIT_TEXT[suit], font=suit_font)
    suit_h = suit_bbox[3] - suit_bbox[1]
    suit_w = suit_bbox[2] - suit_bbox[0]

    # rank 在上、suit 在下
    top_y = pad
    canvas_rgba = canvas
    draw.text((pad, top_y), rank_str, font=font, fill=color)
    draw.text((pad, top_y + rank_h + 2), SUIT_TEXT[suit], font=suit_font, fill=color)

    # 右下角：旋转 180° 整体贴到 (w-pad, h-pad)
    block_w = max(rank_w, suit_w)
    block_h = rank_h + 2 + suit_h
    # 透明小图旋转
    corner_img = Image.new('RGBA', (block_w + 4, block_h + 4), (0, 0, 0, 0))
    cdraw = ImageDraw.Draw(corner_img)
    cdraw.text((0, 0), rank_str, font=font, fill=color)
    cdraw.text((0, rank_h + 2), SUIT_TEXT[suit], font=suit_font, fill=color)
    corner_img = corner_img.rotate(180)
    canvas_rgba.alpha_composite(corner_img, (w - block_w - 4 - pad, h - block_h - 4 - pad))


def draw_number_card(canvas, design, suit, rank, project, card_key):
    """绘制数字牌：中心花色阵列"""
    w, h = canvas.size
    pip_ratio = float(design.get('pip_size_ratio', 0.16) or 0.16)
    pip_size = max(40, int(w * pip_ratio))
    suit_color = hex_to_rgba(design.get('suit_symbols', {}).get(suit, {}).get('color')
                             or ('#E53935' if is_red(suit) else '#000000'), 255)
    font = make_text_font('Arial', pip_size)

    # 决定位置：优先用项目里的 number_layout 覆盖默认
    rank_int = int(rank) if str(rank).isdigit() else 1
    default_positions = LAYOUT_POSITIONS.get(rank_int, LAYOUT_POSITIONS[1])
    layout_key = str(rank_int)
    user_overrides = (project.number_layout or {}).get(layout_key, [])

    positions = []
    for i, (fx, fy) in enumerate(default_positions):
        if i < len(user_overrides) and isinstance(user_overrides[i], dict):
            dx = float(user_overrides[i].get('dx', 0))
            dy = float(user_overrides[i].get('dy', 0))
            scale = float(user_overrides[i].get('scale', 1))
        else:
            dx = dy = 0
            scale = 1.0
        positions.append((fx + dx, fy + dy, scale))

    draw = ImageDraw.Draw(canvas)
    symbol = SUIT_TEXT[suit]
    for fx, fy, scale in positions:
        sz = max(20, int(pip_size * scale))
        fnt = make_text_font('Arial', sz)
        bbox = draw.textbbox((0, 0), symbol, font=fnt)
        tw = bbox[2] - bbox[0]
        th = bbox[3] - bbox[1]
        cx = int(fx * w)
        cy = int(fy * h)
        draw.text((cx - tw // 2, cy - th // 2), symbol, font=fnt, fill=suit_color)


def draw_face_card(canvas, design, suit, rank, project, card_key, asset):
    """绘制 JQK 人物图：上半 + 上下翻转的下半，形成中心对称
    资源找不到时退化为一个大花色符号 + 字母
    """
    w, h = canvas.size
    # 主体区域：留出上下 25% 边距给角标
    body_pad_x = int(w * 0.15)
    body_pad_y_top = int(h * 0.13)
    body_pad_y_bot = int(h * 0.13)
    body_w = w - 2 * body_pad_x
    body_h = h - body_pad_y_top - body_pad_y_bot

    if asset:
        try:
            img = asset.copy()
            # 等比缩放 fill body_w x body_h
            img_ratio = img.width / img.height
            target_ratio = body_w / body_h
            if img_ratio > target_ratio:
                new_w = body_w
                new_h = int(body_w / img_ratio)
            else:
                new_h = body_h
                new_w = int(body_h * img_ratio)
            img = img.resize((new_w, new_h), Image.LANCZOS)

            # 取上半部分 + 上下翻转的下半部分（取上半 = 下半 = 整个图按上下中线翻转）
            top = img.crop((0, 0, img.width, img.height // 2))
            bot = img.crop((0, img.height // 2, img.width, img.height)).transpose(Image.FLIP_TOP_BOTTOM)

            # 拼接成 body_h 高的新图
            full = Image.new('RGBA', (img.width, body_h), (0, 0, 0, 0))
            top = top.resize((img.width, body_h // 2), Image.LANCZOS)
            bot = bot.resize((img.width, body_h - body_h // 2), Image.LANCZOS)
            full.paste(top, (0, 0), top)
            full.paste(bot, (0, body_h // 2), bot)

            full = full.resize((body_w, body_h), Image.LANCZOS)
            canvas.alpha_composite(full, (body_pad_x, body_pad_y_top))
            return
        except Exception:
            pass

    # 退化：绘制大字 + 花色
    suit_color = hex_to_rgba(design.get('suit_symbols', {}).get(suit, {}).get('color')
                             or ('#E53935' if is_red(suit) else '#000000'), 255)
    big = max(80, int(h * 0.35))
    fnt_rank = make_text_font('Times New Roman', big, bold=True)
    fnt_suit = make_text_font('Arial', big)
    draw = ImageDraw.Draw(canvas)
    rb = draw.textbbox((0, 0), rank, font=fnt_rank)
    rw = rb[2] - rb[0]
    rh = rb[3] - rb[1]
    draw.text(((w - rw) // 2, body_pad_y_top + (body_h - rh) // 2 - big // 2),
              rank, font=fnt_rank, fill=suit_color)
    sb = draw.textbbox((0, 0), SUIT_TEXT[suit], font=fnt_suit)
    sw = sb[2] - sb[0]
    sh = sb[3] - sb[1]
    draw.text(((w - sw) // 2, body_pad_y_top + (body_h - sh) // 2 + big // 4),
              SUIT_TEXT[suit], font=fnt_suit, fill=suit_color)


def draw_joker(canvas, design, which, project, card_key, asset):
    """绘制大小王：独立背景 + 主体图 + 文字"""
    w, h = canvas.size

    # 优先用单牌覆盖的背景色
    bg_color = design.get('background_color', '#1B5E20' if which == 'big' else '#B71C1C')
    canvas.paste(hex_to_rgba(bg_color, 255), (0, 0, w, h))

    # 主体图
    body_pad_x = int(w * 0.15)
    body_pad_y_top = int(h * 0.15)
    body_pad_y_bot = int(h * 0.20)
    body_w = w - 2 * body_pad_x
    body_h = h - body_pad_y_top - body_pad_y_bot
    if asset:
        try:
            img = asset.copy()
            img.thumbnail((body_w, body_h), Image.LANCZOS)
            canvas.alpha_composite(img, (body_pad_x + (body_w - img.width) // 2,
                                         body_pad_y_top + (body_h - img.height) // 2))
        except Exception:
            pass
    else:
        # 退化
        fnt = make_text_font('Times New Roman', max(80, int(h * 0.25)), bold=True)
        draw = ImageDraw.Draw(canvas)
        text = 'JOKER'
        color = (255, 255, 255, 255) if which == 'big' else (255, 255, 255, 255)
        bb = draw.textbbox((0, 0), text, font=fnt)
        tw, th = bb[2] - bb[0], bb[3] - bb[1]
        draw.text(((w - tw) // 2, (h - th) // 2), text, font=fnt, fill=color)

    # 文字标识
    label = 'BIG' if which == 'big' else 'SMALL'
    text = 'JOKER'
    fnt_label = make_text_font('Times New Roman', max(20, int(w * 0.06)), bold=True)
    fnt_text = make_text_font('Times New Roman', max(16, int(w * 0.045)))
    color = (255, 255, 255, 255)
    draw = ImageDraw.Draw(canvas)
    bb = draw.textbbox((0, 0), text, font=fnt_text)
    tw = bb[2] - bb[0]
    th = bb[3] - bb[1]
    pad = max(8, int(w * 0.04))
    # 左上
    draw.text((pad, pad), text, font=fnt_text, fill=color)
    draw.text((pad, pad + th + 2), label, font=fnt_label, fill=color)
    # 右下
    block = Image.new('RGBA', (tw + 4, 2 * th + 6), (0, 0, 0, 0))
    bdraw = ImageDraw.Draw(block)
    bdraw.text((0, 0), text, font=fnt_text, fill=color)
    bdraw.text((0, th + 2), label, font=fnt_label, fill=color)
    block = block.rotate(180)
    canvas.alpha_composite(block, (w - tw - 4 - pad, h - 2 * th - 6 - pad))


def generate_card_png(project, card_key, resolution='standard'):
    """根据项目配置生成单张牌 PNG"""
    scale_map = {'standard': 1, 'hd': 2, 'ultra-hd': 4}
    scale = scale_map.get(resolution, 1)

    w = int(project.card_width * scale)
    h = int(project.card_height * scale)
    canvas = Image.new('RGBA', (w, h), (255, 255, 255, 255))

    design = get_effective_design(project, card_key)

    # 1. 背景
    draw_background(canvas, design, project, card_key)

    # 2. 边框
    draw_border(canvas, design)

    # 3. 主体内容
    if card_key.startswith('joker-'):
        which = card_key.split('-', 1)[1]  # big / small
        asset = None
        for a in project.assets.filter(asset_type='joker', asset_key=which):
            p = os.path.join('media', a.file_path) if a.file_path else None
            asset = load_image_safe(p) if p else None
            break
        draw_joker(canvas, design, which, project, card_key, asset)
    elif card_key in ('back', 'card-back'):
        # 简化：背面同整体背景 + 一行文字
        draw = ImageDraw.Draw(canvas)
        fnt = make_text_font('Times New Roman', max(40, int(h * 0.08)), bold=True)
        text = 'CARD BACK'
        color = hex_to_rgba(design.get('border_color', '#333333'), 255)
        bb = draw.textbbox((0, 0), text, font=fnt)
        tw, th = bb[2] - bb[0], bb[3] - bb[1]
        draw.text(((w - tw) // 2, (h - th) // 2), text, font=fnt, fill=color)
    else:
        # 'suit-rank'
        parts = card_key.split('-')
        suit = parts[0]
        rank = parts[1]
        is_face = rank in ('J', 'Q', 'K')
        if is_face:
            asset = None
            for a in project.assets.filter(asset_type='face_card', asset_key=card_key):
                p = os.path.join('media', a.file_path) if a.file_path else None
                asset = load_image_safe(p) if p else None
                break
            draw_face_card(canvas, design, suit, rank, project, card_key, asset)
        else:
            # A 写成 '1'，但角标用 'A'
            rk = 'A' if rank == '1' else rank
            draw_number_card(canvas, design, suit, rank, project, card_key)
            # 角标
            draw_corner_index(canvas, design, suit, rk, project, card_key)
            return canvas

        # JQK 角标：花色小，rank 字母 J/Q/K
        draw_corner_index(canvas, design, suit, rank, project, card_key)

    return canvas