from PIL import Image, ImageOps
from typing import Any
VIEWS = ('ego', 'left_wrist', 'right_wrist')
CAPTION_TOP_SIZE = (768, 432)
CAPTION_WRIST_SIZE = (384, 216)
CAPTION_MOSAIC_WIDTH = 768
CAPTION_MOSAIC_HEIGHT = 648

def compose_frame(frames, views):
    canvas = Image.new('RGB', (640, 720))
    for view, (x, y, w, h) in zip(VIEWS, [(0, 0, 640, 480), (0, 480, 320, 240), (320, 480, 320, 240)]):
        if view in views:
            panel = ImageOps.contain(frames[view].convert('RGB'), (w, h), Image.Resampling.BICUBIC)
            canvas.paste(panel, (x + (w - panel.width) // 2, y + (h - panel.height) // 2))
    return canvas

def compose_top_dominant_mosaic(top: Any, left: Any, right: Any) -> Any:
    from PIL import Image
    top_panel = top.convert('RGB').resize(CAPTION_TOP_SIZE, Image.Resampling.BICUBIC)
    left_panel = left.convert('RGB').resize(CAPTION_WRIST_SIZE, Image.Resampling.BICUBIC)
    right_panel = right.convert('RGB').resize(CAPTION_WRIST_SIZE, Image.Resampling.BICUBIC)
    output = Image.new('RGB', (CAPTION_MOSAIC_WIDTH, CAPTION_MOSAIC_HEIGHT))
    output.paste(top_panel, (0, 0))
    output.paste(left_panel, (0, CAPTION_TOP_SIZE[1]))
    output.paste(right_panel, (CAPTION_WRIST_SIZE[0], CAPTION_TOP_SIZE[1]))
    return output