from tqdm.auto import tqdm
from dnafiber.data.utils import read_colormask
import numpy as np


def build_consensus_map(intergraders, root_img, list_img):
    all_masks = []
    for img_path in tqdm(list_img):
        path_from_root = img_path.relative_to(root_img)
        masks = []
        for intergrader in intergraders:
            intergrader_path = (intergrader / path_from_root).with_suffix(".png")
            if not intergrader_path.exists():
                print(f"Missing {intergrader_path}")
                continue
            mask = read_colormask(intergrader_path)
            masks.append(mask)
        masks = np.array(masks)

        all_masks.append(masks)
    return np.array(all_masks)