import pspec c = 0.3 #m/ns twopi = 2.*np.pi N = 512 fq = np.linspace(0.1,0.2,N) L2 = RMT.better_guess_l2(fq) Rms,W = RMT.RMTmat(fq) plot_me = np.zeros((N,N),dtype=np.complex) delays = np.fft.fftshift(np.fft.fftfreq(N,fq[1]-fq[0])) wgt = dsp.gen_window(N,window='blackman-harris') Rm0 = 5. for i,Rm in enumerate(Rms): QiUi = RMT.gen_rm_spec(fq,Rm+Rm0) plot_me[:,i] = np.fft.fftshift(np.fft.ifft(QiUi*wgt)) plot_me1 = plot_me*np.conjugate(plot_me) plot_me1 = plot_me1.real plot_me1 /= np.max(plot_me1) figure(0) imshow(10.*np.log10(plot_me1),aspect='auto',interpolation='nearest', extent=[delays[0],delays[-1],Rms[-1],Rms[0]], vmax=0,vmin=-100) ylabel('Rotation Measure [$m^{-2}$]') xlabel('Delay [ns]') title('$|W|^2$ [dB]') colorbar() draw()
#yticks([]) xlabel(xlabels) suptitle(title) #subplots_adjust(hspace=0) draw() nfig += 1 #GENERATE A FEW SIMULATED SPECTRA 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 = [], []
#yticks([]) xlabel(xlabels) suptitle(title) #subplots_adjust(hspace=0) draw() nfig += 1 #GENERATE A FEW SIMULATED SPECTRA 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):