コード例 #1
0
ファイル: RotMeasTest.py プロジェクト: nkern/capo
fq = np.linspace(0.1, 0.2, 512)
RMsim = [10., np.sqrt(200.)]
RMlabel = ['$10\ m^{-2}$', '10$\sqrt{2}\ m^{-2}$', 'Sum']
QiUsim = []
for r in RMsim:
    QiUsim.append(RMT.gen_rm_spec(fq, r))
QiUsim.append(np.sum(QiUsim, axis=0) / 2.)

plot3(fq, QiUsim, RMlabel, 'Frequency (GHz)',
      'Simulated Spectra $\Re\{Q + iU\}$')

##TEST THE REBIN/FFT METHOD

l2, QiUsim_l2 = [], []
for i, spec in enumerate(QiUsim):
    l2i, spec_l2 = RMT.rebin_nu2lam2(fq, spec)
    l2.append(l2i)
    QiUsim_l2.append(spec_l2)
#plot3(l2,QiUsim_l2,RMlabel,'$\lambda^2\ (m^2)$ ','Rebinned into $\lambda^2$')

RM_samp, QiUsim_rm = [], []
for i, spec in enumerate(QiUsim_l2):
    QiUsim_rm.append(np.fft.fft(QiUsim_l2[i]))
    RM_samp.append(np.fft.fftfreq(len(spec), l2[i][1] - l2[i][0]) * RMT.twopi)
#plot3(RM_samp,np.abs(QiUsim_rm),RMlabel,'RM $(m^{-2})$','FFTed')

#TEST THE DFT METHOD

RM_samp, W = RMT.RMTmat(fq)
for i, spec in enumerate(QiUsim):
    QiUsim_rm[i] = RMT.RMT(spec, W)
コード例 #2
0
ファイル: RotMeasTest.py プロジェクト: SaulAryehKohn/capo
fq = np.linspace(0.1,0.2,512)
RMsim = [10.,np.sqrt(200.)]
RMlabel = ['$10\ m^{-2}$','10$\sqrt{2}\ m^{-2}$','Sum']
QiUsim = []
for r in RMsim:
    QiUsim.append(RMT.gen_rm_spec(fq,r))
QiUsim.append(np.sum(QiUsim,axis=0)/2.)

plot3(fq,QiUsim,RMlabel,'Frequency (GHz)','Simulated Spectra $\Re\{Q + iU\}$')

##TEST THE REBIN/FFT METHOD

l2,QiUsim_l2 = [],[]
for i,spec in enumerate(QiUsim):
    l2i,spec_l2 = RMT.rebin_nu2lam2(fq,spec)
    l2.append(l2i)
    QiUsim_l2.append(spec_l2)
#plot3(l2,QiUsim_l2,RMlabel,'$\lambda^2\ (m^2)$ ','Rebinned into $\lambda^2$')

RM_samp,QiUsim_rm = [],[]
for i,spec in enumerate(QiUsim_l2): 
    QiUsim_rm.append(np.fft.fft(QiUsim_l2[i]))
    RM_samp.append(np.fft.fftfreq(len(spec),l2[i][1]-l2[i][0])*RMT.twopi)
#plot3(RM_samp,np.abs(QiUsim_rm),RMlabel,'RM $(m^{-2})$','FFTed')

#TEST THE DFT METHOD

RM_samp,W = RMT.RMTmat(fq)
for i,spec in enumerate(QiUsim):
    QiUsim_rm[i] = RMT.RMT(spec,W)