def extract_vis(src, pol, quiet, fl, usePredict=False, raoffset=0.0): # fill antennas and baseline lengths aname = pf.open(fl, ignore_missing_end=True)[1].header['ANTENNAS'].split('-') nchan = pf.open(fl, ignore_missing_end=True)[1].header['NCHAN'] tp = np.loadtxt('antpos_ITRF.txt') fqs = pf.open(fl, ignore_missing_end=True)[1].header['FCH1'] * 1e6 - ( np.arange(nchan) + 0.5) * 2. * 2.5e8 / 2048. blen = [] bname = [] for i in np.arange(9) + 1: for j in np.arange(i): a1 = int(aname[i]) - 1 a2 = int(aname[j]) - 1 bname.append([a1, a2]) blen.append(tp[a1, 1:] - tp[a2, 1:]) print "Antennas: " + str(aname) print "Baseline: " + str(bname) print "Source " + str(src) src_model = read_src_model(src) stmid = src_model[0][0] + raoffset data = [] # READ DATA data, st, mjd, Imid = extract_segment(fname=fl, stmid=stmid, seg_len=5. / 60.0 / 24.0 * PI2, pol=pol, quiet=False) nsamps = data[0].shape[0] if usePredict: model = predict_vis(blen, mjd, fqs, src_model=src_model, pointing=POINTING, quiet=quiet) # Predict the visibilities return model, mjd, bname return data, mjd, bname
print nchan blen = [] bname = [] for i in np.arange(9) + 1: for j in np.arange(i): a1 = int(aname[i]) - 1 a2 = int(aname[j]) - 1 bname.append([a1, a2]) blen.append(tp[a1, 1:] - tp[a2, 1:]) print "Antennas: " + str(aname) print "Baseline: " + str(bname) print "Source " + str(src) src_model = read_src_model(src) stmid = src_model[0][0] data = [] # READ DATA data, st, mjd, Imid = extract_segment(fname=fl, stmid=stmid, seg_len=10.0 / 60.0 / 24.0 * PI2, pol=pol, quiet=False) nsamps = data[0].shape[0] model = predict_vis(blen, mjd, fqs, src_model=src_model,