#! /usr/bin/python3.6
# -*- coding: utf-8 -*-

""" Command line Python3 script to plot an Amplitude Matric of the XST
"""

import argparse
import os
import numpy as np
import matplotlib.pyplot as plt

from nenupy.read import XST

if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument('-x', '--xst', type=str, help="XST file", required=True)
    parser.add_argument('-p', '--pol', type=str, help="Polarisation (XX, XY, YX, YY)", default='XX', required=False)
    args = parser.parse_args()

    xst = XST( args.xst )
    xst.plotCorrMat( args.pol )