"""Share of saturated red pixels (painted runes) per image, for domain analysis."""
import json
import sys

import numpy as np
from PIL import Image


def red_share(path):
    im = Image.open(path).convert("RGB")
    im.draft("RGB", (256, 256))
    im.thumbnail((256, 256))
    a = np.asarray(im).astype(np.int16)
    r, g, b = a[..., 0], a[..., 1], a[..., 2]
    return float(((r > 110) & (r - g > 45) & (r - b > 40)).mean())


if __name__ == "__main__":
    out = {}
    for split in sys.argv[2:]:
        for l in open(split):
            rec = json.loads(l)
            if rec["image"] not in out:
                out[rec["image"]] = red_share(f"{sys.argv[1]}/{rec['image']}")
    json.dump(out, open(f"{sys.argv[1]}/data/red_share.json", "w"))
    print(len(out))