Extract Voyager 1 signal from GUPPI files

extract_voyager.py

#!/usr/bin/env python3

import argparse

from blimpy import GuppiRaw


def parse_args():
    parser = argparse.ArgumentParser(
        description='Extract IQ channel for Voyager-1 from GUPPI file')
    parser.add_argument('input_file', type=str,
                        help='input file')
    parser.add_argument('output_file', type=str,
                        help='output file')
    return parser.parse_args()


def main():
    vgr_chan = 6  # PFB channel for Voyager-1
    args = parse_args()
    f_in = GuppiRaw(args.input_file)
    with open(args.output_file, 'wb') as f_out:
        try:
            for _, data_x, _ in f_in.get_data():
                x = data_x[vgr_chan]
                x.tofile(f_out)
        except StopIteration:
            pass


if __name__ == '__main__':
    main()