"""Read verified recorded COGs: python read_result.py <result-directory>."""
import sys
from pathlib import Path
import numpy as np
import rasterio

folder = Path(sys.argv[1])
with rasterio.open(folder / "class_map.tif") as classes, \
        rasterio.open(folder / "valid_mask.tif") as mask:
    assert classes.crs == mask.crs
    assert classes.transform == mask.transform
    assert classes.shape == mask.shape
    prediction = classes.read(1)
    valid = (mask.read(1) > 0) & (classes.read_masks(1) > 0)
    valid &= prediction != 255
    assert np.all(prediction[valid] < 10)
    counts = np.bincount(prediction[valid], minlength=10)
    print("Grid:", classes.crs, classes.shape)
    print("Valid pixels:", int(valid.sum()))
    print("Excluded pixels:", int((~valid).sum()))
    for class_id, count in enumerate(counts):
        print(f"Class {class_id}: {int(count)}")