_util.py

import numpy as np
import os
from datetime import datetime
import argparse
from astropy.time import Time

from observation_sim.config import Pointing


def parse_args():
    '''
    Parse command line arguments. Many of the following
    can be set in the .yaml config file as well.
    '''
    parser = argparse.ArgumentParser()
    parser.add_argument('--config_file', type=str, required=True,
                        help='.yaml config file for simulation settings.')
    parser.add_argument('--catalog', type=str,
                        help='name of the catalog interface class to be loaded.')
    parser.add_argument('-c', '--config_dir', type=str,
                        help='Directory that houses the .yaml config file.')
    parser.add_argument('-d', '--data_dir', type=str,
                        help='Directory that houses the input data.')
    parser.add_argument('-w', '--work_dir', type=str,
                        help='The path for output.')
    return parser.parse_args()


def generate_pointing_list(config, pointing_filename=None, data_dir=None):
    pointing_list = []

    # Only valid when the pointing list does not contain time stamp column
    t0 = datetime(2021, 5, 25, 12, 0, 0)
    delta_t = 10.  # Time elapsed between exposures (minutes)

    # Calculate starting time(s) for CAL exposures
    # NOTE: temporary implementation
    t = datetime.timestamp(t0)
    ipoint = 0

    run_pointings = config['obs_setting']['run_pointings']
    if "obs_config_file" in config['obs_setting']:
        obs_config_file = config['obs_setting']["obs_config_file"]
    else:
        obs_config_file = None

    # if pointing_filename and data_dir:
    if pointing_filename:
        if data_dir:
            pointing_file = os.path.join(data_dir, pointing_filename)
        else:
            pointing_file = pointing_filename
        f = open(pointing_file, 'r')
        # for _ in range(1):
        #     header = f.readline()
        iline = 0
        for line in f:
            if len(line.strip()) == 0 or line[0] == '#':
                continue
            if run_pointings and iline not in run_pointings:
                iline += 1
                ipoint += 1
                continue
            line = line.strip()
            columns = line.split()
            pointing = Pointing(obs_config_file=obs_config_file)
            pointing.read_pointing_columns(columns=columns, id=ipoint)
            t += delta_t * 60.
            pointing_list.append(pointing)
            iline += 1