#!/usr/bin/env python

import logging
from argparse import ArgumentParser
from glob import glob

from BinaryBufferReader import BinaryBufferReader


logging.basicConfig()
_logger = logging.getLogger("BinaryBufferReader")


BUFFER_DIR = "/gpfs/photonics/swissfel/buffer"


def get_root_folder(detector):
    pattern = f"{BUFFER_DIR}/{detector}*"
    dns = glob(pattern)
    nfound = len(dns)
    if nfound == 1:
        return dns[0]
    if nfound == 0:
        raise SystemExit(f"no root folder matching {detector}")
    if nfound > 1:
        raise SystemExit(f"cannot unambiguously match {detector} to root folder: {dns}")

def get_pulse_id(root_folder):
    latest = f"{root_folder}/M00/LATEST"
    with open(latest) as f:
        data = f.read()
    #/gpfs/photonics/swissfel/buffer/JF02T09V03/M00/21125800000/21125820000.bin
    pid = data.strip().split("/")[-1].split(".")[0]
    return int(pid)

def get_n_modules(root_folder):
    detname = get_detname(root_folder)
    nmod = detname[5:7]
    return int(nmod)

def get_detname(root_folder):
    return root_folder.split("/")[-1]





def main():
    parser = ArgumentParser(description="Read DAQ Binary Buffer for JFs")

    parser.add_argument("detector", help="detector name (JF01...)")
    parser.add_argument("--n_modules", type=int, help="number of modules to read from this device (if not given, all modules are read)")
    parser.add_argument("--pulse_id", type=int, help="pulse ID to retrieve (if not given, the last pulse ID is retrieved -- negative values count backwards from the last)")
    parser.add_argument("--log_level", default="WARNING", choices=["CRITICAL", "ERROR", "WARNING", "INFO", "DEBUG"], help="log level")

    clargs = parser.parse_args()

    detector = clargs.detector

    root_folder = get_root_folder(detector)

    if clargs.pulse_id is None:
        pulse_id = get_pulse_id(root_folder)
    elif clargs.pulse_id < 0:
        pulse_id = get_pulse_id(root_folder) + clargs.pulse_id
    else:
        pulse_id = clargs.pulse_id

    n_modules = clargs.n_modules or get_n_modules(root_folder)

    _logger.setLevel(clargs.log_level)

    reader = BinaryBufferReader(root_folder, n_modules)

    metadata, _data = reader.read_pulse_id(pulse_id)
    print(metadata)



if __name__ == "__main__":
    main()