"""Editing inputs: independent noise and unchanged RGBA pixels."""
import math
from pathlib import Path

from PIL import Image, ImageOps

DEFAULT_SEEDS = (1000042, 1000123)


def validate_seeds(cases, seeds):
    if (not seeds or len(set(seeds)) != len(seeds)
            or any(type(s) is not int or not 0 <= s < 2**63 for s in seeds)):
        raise ValueError('Use distinct integer seeds in [0, 2**63)')
    source_seeds = {c.get('source_seed') for c in cases} - {None}
    if source_seeds.intersection(seeds):
        raise ValueError('Editing seeds must differ from every known source seed (noise replay)')
    return list(seeds)


def load_input(path):
    with Image.open(path) as image:
        image = ImageOps.exif_transpose(image)
        has_alpha = 'A' in image.getbands() or 'transparency' in image.info
        return image.convert('RGBA' if has_alpha else 'RGB')


def edit_dimensions(size, resolution):
    if resolution < 32 or resolution % 32 or min(size) <= 0:
        raise ValueError('Positive image size and resolution divisible by 32 required')
    ratio = size[0] / size[1]
    width = math.sqrt(resolution**2 * ratio)
    return max(32, round(width / 32) * 32), max(32, round(width / ratio / 32) * 32)