imager_scripts/flags.py at master · lmagnus/imager_scripts · GitHub

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
63
64
65
66
67
68
69
70
71
72
73
#! /usr/bin/python
# Heath checks of the system

# The *with* keyword is standard in Python 2.6, but has to be explicitly imported in Python 2.5
from __future__ import with_statement

import os,traceback,re
import time
import numpy as np
from scipy import signal
import optparse

import katpoint
import katfile
import katarchive
import matplotlib.pyplot as plt
from matplotlib.backends.backend_pdf import PdfPages

n = '1360832758.h5'
f = katarchive.get_archived_products(n)
h5 = katfile.open(f[0])
#pp = PdfPages('flags.pdf')


h5.select()
cor = h5.corr_products
ants = h5.ants
num_ants = len(ants)
#bp = np.array([t.tags.count('bpcal') for t in h5.catalogue.targets]) == 1
#bp = np.arange(len(bp))[bp]
#code to plot the cross phase ... fringes
#fig = plt.figure(figsize=(24,20))
fig = plt.figure()
try:
    j=0
#plt.figure()
    n_channels = len(h5.channels)
    for pol in ('h','v'):
        h5.select(corrprods = 'cross',pol=pol)
        crosscorr = [(h5.inputs.index(inpA), h5.inputs.index(inpB)) for inpA, inpB in h5.corr_products]
        #extract the fringes
	flags = h5.flags()[:]
        #For plotting the fringes
        fig.subplots_adjust(wspace=0., hspace=0.)
        #debug_here()
        for n, (indexA, indexB) in enumerate(crosscorr):
            subplot_index = (num_ants * indexA + indexB + 1) if pol == 'h' else (indexA + num_ants * indexB + 1)
            ax = fig.add_subplot(num_ants, num_ants, subplot_index)
            asp = np.trunc(flags.shape[1]/flags.shape[0])
	    ax.imshow(flags[:,:,n],cmap='binary')
	    ax.set_aspect('auto')
            ax.set_xticks([])
            ax.set_yticks([])
            if pol == 'h':
                if indexA == 0:
                    ax.xaxis.set_label_position('top')
                    ax.set_xlabel(h5.inputs[indexB][3:],size='xx-large')
                if indexB == len(h5.ants) - 1:
                    ax.yaxis.set_label_position('right')
                    ax.set_ylabel(h5.inputs[indexA][3:], rotation='horizontal',size = 'xx-large')
            else:
                if indexA == 0:
                    ax.set_ylabel(h5.inputs[indexB][3:], rotation='horizontal',size='xx-large')
                if indexB == len(h5.ants) - 1:
                    ax.set_xlabel(h5.inputs[indexA][3:],size='xx-large')
    #plt.savefig(pp,format='pdf')
except KeyError , error:
    print 'Failed to read scans from File: ',fn,' with Key Error:',error
except ValueError , error:
    print 'Failed to read scans from File: ',fn,' with Value Error:',error
#plt.savefig(pp,format='pdf')
#plt.close('all')
#pp.close()