#!/usr/bin/env python3
#
#   Use star database from https://github.com/astronexus/HYG-Database
#   to generate constellation test data of the form:
#   ra dec symbol
#

import sys
import re

def Translate(inFileName, outFileName):

    if False:
        ignore = set()
    else:
        # Ignore disagreements with hygdata_v3.
        # See: https://github.com/astronexus/HYG-Database/issues/21
        ignore = set([
            3865, 7751, 10342, 19173, 22400, 27992, 29366,
            30112, 52886, 58221, 59234, 67599, 72017, 74021, 78313,
            84413, 85957, 87694, 89105, 91382, 92907, 93308,
            95468, 100777, 101541, 105623
        ])

    count = 0
    with open(inFileName, 'rt') as infile:
        with open(outFileName, 'wt') as outfile:
            lnum = 0
            for line in infile:
                lnum += 1
                token = line.strip().split(',')
                if lnum == 1:
                    columns = dict((key, col) for (col, key) in enumerate(token))
                else:
                    id = int(token[columns['id']])
                    ra = float(token[columns['ra']])
                    dec = float(token[columns['dec']])
                    sym = token[columns['con']]
                    if sym != '':
                        if not re.match(r'^[A-Z][a-zA-Z]{2}$', sym):
                            print('make_constellation_data: bad symbol "{}" in {} line {}'.format(sym, inFileName, lnum))
                            return 1

                        if id not in ignore:
                            count += 1
                            outfile.write('{:6d} {:10.6f} {:10.6f} {:3s}\n'.format(id, ra, dec, sym))
    print('make_constellation_data: Wrote {:d} test cases.'.format(count))
    return 0

if __name__ == '__main__':
    sys.exit(Translate('hygdata_v3.csv', 'constellation/test_input.txt'))