#!/usr/bin/env python3
#
# Copyright (c) 2017 Weitian LI <liweitianux@live.com>
# MIT license
#
# Weitian LI
# 2017-02-06

"""
Determine the Chandra ACIS type for the given observation.
"""

import argparse
import subprocess
import re

from setup_pfiles import setup_pfiles


def get_acis_type(infile):
    """
    Determine the Chandra ACIS type (``I`` or ``S``) according the
    active ACIS chips.

    Parameters
    ----------
    infile : str
        Filename of the FITS file

    Returns
    -------
    acis_type : str
        ``I`` if ACIS-I, ``S`` if ACIS-S, otherwise, ``ValueError`` raised.
    """
    subprocess.check_call(["punlearn", "dmkeypar"])
    detnam = subprocess.check_output([
        "dmkeypar", "infile=%s" % infile, "keyword=DETNAM", "echo=yes"
    ]).decode("utf-8").strip()
    if re.match(r"^ACIS-0123", detnam):
        return "I"
    elif re.match(r"^ACIS-[0-6]*7", detnam):
        return "S"
    else:
        raise ValueError("unknown chip combination: %s" % detnam)


def main():
    parser = argparse.ArgumentParser(description="Determine Chandra ACIS type")
    parser.add_argument("-b", "--brief", dest="brief",
                        action="store_true", help="Be brief")
    parser.add_argument("infile", help="Input FITS file")
    args = parser.parse_args()

    setup_pfiles(["dmkeypar"])
    acis_type = get_acis_type(args.infile)
    if not args.brief:
        print("ACIS-type:", end=" ")
    print(acis_type)


if __name__ == "__main__":
    main()