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)
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)