1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
|
#!/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()
|