import os
from argparse import ArgumentParser
import json

script_dir = os.path.dirname(os.path.realpath(__file__))


def main(config_path):
    with open(config_path) as json_file:
        config = json.load(json_file)
    tile_path = config["output_path"]
    
    total_tumor = 0
    total_tiles = 0
    slides = sorted([f for f in os.listdir(tile_path) if os.path.isdir(os.path.join(tile_path, f))])
    for i in range(len(slides)):
        slide = slides[i]
        slide_path = os.path.join(tile_path, slide)
        n_tumor = len(os.listdir(os.path.join(slide_path, 'tumor')))
        n_other = len(os.listdir(os.path.join(slide_path, 'non_tumor')))
        n_total = n_tumor + n_other
        frac = n_tumor / n_total * 100
        print("{}: {}/{}, {:.4f}% tumor tiles of total tiles, name: {}".format(i, n_tumor, n_total, frac, slide))
        total_tumor += n_tumor
        total_tiles += n_total
    frac = total_tumor / total_tiles * 100
    print("Total: {}/{}, {:.4f}% tumor tiles of total tiles".format(total_tumor, total_tiles, frac))


if __name__ == "__main__":
    parser = ArgumentParser()
    parser.add_argument("--config_path", default=script_dir + "/resources/config.json")
    args = parser.parse_args()

    main(args.config_path)