; NAME: ; ; stplot_rotate ; ; PURPOSE: ; ; Plot space-time maps for several cuts by first rotating the image to let the cuts vertical or horizontal ; ; CATEGORY: ; ; Data analysis ; ; CALLING SEQUENCES: ; ; stplot_rotate,data ; ; stplot_rotate,data,dir_sav=dir_sav,cutwid=cutwid,xtitle=xtitle,ytitle=ytitle,log=log,sm_scl=sm_scl,plotfrm=plotfrm ; ; INPUTS: ; ; data: an image sequence [nx,ny,nt] ; ; KEYWORDS: ; ; dir_sav: the directory where the resulting images and data will be saved, e.g., './sampledata/SOT_CaII_20070114/' ; ; cutwid: width of each cut ; ; xtitle: lable of the horizontal axis (time) of the space-time plot, e.g., 'Time (s)' ; ; ytitle: lable of the vertical axis (distance) of the space-time plot, e.g., 'Distance (arcsec)' ; ; log: take common logarithm of the intensity if set ; ; sm_scl: the smooth scale used to generate the trend which will be removed from the original time series ; ; plotfrm: the index of the image shown, default is 0 (the first image in the sequence) ; ; panel_len: the size of one panel in the horizontal dimension (time) ; ; dpercent: the dynamic range of the image to be shown, in unit of percentage, e.g., [0.01,0.99] ; ; tr: range of time frames to be plotted, default is [0,nt-1] ; ; OUTPUTS: ; ; the following images and data files will be created: ; ; stplot_cuts.png: the locations of the selected cuts ploted on the first image in the image sequence ; ; stplot_cutslocation.sav: the locations of the selected cuts (xc, yc) and number of cuts (ncut) ; ; stplot_cut=i.sav: the space-time data array for cut i ; ; stplot.png: the original and de-trended space-time plots for all cuts ; ; HISTORY: ; ; Written by Hui Tian at CfA, 27 March 2013 pro stplot_rotate,data,dir_sav=dir_sav,cutwid=cutwid,xtitle=xtitle,ytitle=ytitle,log=log,sm_scl=sm_scl,$ plotfrm=plotfrm,panel_len=panel_len,dpercent=dpercent,tr=tr if not keyword_set(dir_sav) then dir_sav='' if not keyword_set(cutwid) then cutwid=5 if not keyword_set(xtitle) then xtitle='Time step' if not keyword_set(ytitle) then ytitle='Spatial pixel' ytitle=['',ytitle] if not keyword_set(sm_scl) then sm_scl=5 if not keyword_set(plotfrm) then plotfrm=0 if not keyword_set(panel_len) then panel_len=400 if not keyword_set(dpercent) then dpercent=[0.01,0.99] nt=(size(data))[3] if not keyword_set(tr) then tr=[0,nt-1] ;select a cut by clicking two positions on the image, more than one cuts can be selected ncut=1 ;the number of cuts read,'how many cuts do you want to select (e.g., 2):',ncut dum2='' xi=fltarr(ncut,2) & yi=fltarr(ncut,2) xc=fltarr(ncut,4) & yc=fltarr(ncut,4) loadct,0 & window,0,xs=700,ys=700, retain = 2 & !p.multi=0 repeat begin if keyword_set(log) then img=alog10(reform(data[*,*,plotfrm])) else img=reform(data[*,*,plotfrm]) img_sort=img(sort(img)) & np=n_elements(img[where(finite(img))]) dminmax = [img_sort[np*dpercent[0]], img_sort[np*dpercent[1]]] plot_image,imgdminmax[0],title='select a cut by clicking two positions on the image, more than one cuts can be selected' for j=0,ncut-1 do begin cursor,xx,yy,/data & wait,.3 plots,xx,yy,/data,psym=5,symsize=0.6 print,'x/y:',xx,yy xi(j,0)=xx & yi(j,0)=yy cursor,xx,yy,/data & wait,.3 plots,xx,yy,/data,psym=5,symsize=0.6 print,'x/y:',xx,yy xi(j,1)=xx & yi(j,1)=yy xc(j,*)=[min(xi(j,*)),min(xi(j,*)),max(xi(j,*)),max(xi(j,*))] yc(j,*)=[min(yi(j,*)),max(yi(j,*)),min(yi(j,*)),max(yi(j,*))] oplot,xi(j,*),yi(j,*),thick=2,linestyle=2 ;plot the cut xyouts,xi(j,0),yi(j,0),'cut '+strtrim(string(j),2),alignment=1.0 endfor read,'Positions OK (hit any key rather than n if OK)?',dum2 endrep until dum2 ne 'n' write_png,dir_sav+'stplot_cuts.png', tvrd(/true), r, g, b save,filename=dir_sav+'stplot_cutslocation.sav',xi,yi,ncut loadct,0 & window,1,xs=panel_len*ncut,ys=400,xpos=0,ypos=0, retain = 2 & !p.multi=0 posi=[0.07,0.60,0.07+0.9/ncut,0.95] cxy=intarr(ncut,4) for j=0,ncut-1 do begin wlow=(where(xc(j,*) eq min(xc(j,*)) and yc(j,*) eq min(yc(j,*)),nlow))(0) delx=max(xc(j,*)-xc(j,wlow)) & dely=max(yc(j,*)-yc(j,wlow)) cxy(j,*)=[xc(j,wlow),yc(j,wlow),delx,dely] data_sub=data[cxy(j,0):cxy(j,0)+delx,cxy(j,1):cxy(j,1)+dely,*] rot_ang=atan((yi[j,1]-yi[j,0])/(xi[j,1]-xi[j,0]))/3.14*180 ;rotation angle in the case of delx > dely ;first rotate the image to let the cut vertical or horizontal, then average intensities over several pixels centered around the cut if delx ge dely then begin if dely gt cutwid then begin for im=0,nt-1 do data_sub[*,*,im]=ROT(data_sub[*,*,im],rot_ang,/interp,/missing) int_slice=average(data_sub[*,dely/2-cutwid/2:dely/2+cutwid/2,*],2) endif else begin int_slice=average(data_sub,2) endelse endif else begin if delx gt cutwid then begin for im=0,nt-1 do data_sub[*,*,im]=ROT(data_sub[*,*,im],(abs(rot_ang)-90)*(rot_ang/abs(rot_ang)),/interp,/missing) int_slice=average(data_sub[delx/2-cutwid/2:delx/2+cutwid/2,*,*],1) endif else begin int_slice=average(data_sub,1) endelse endelse if keyword_set(log) then int_slice=alog10(int_slice) ;original intensity int1=int_slice[*,tr[0]:tr[1]] int1_sort=int1(sort(int1)) & np=n_elements(int1[where(finite(int1))]) dminmax = [int1_sort[np*dpercent[0]], int1_sort[np*dpercent[1]]] plot_image,transpose(int1)dminmax[0],xtitle=xtitle,ytitle=ytitle[j eq 0],/noerase,xticklen=-0.020,xminor=5,$ title='cut '+strtrim(string(j),2),charsize=1.0,position=posi+[0.9/ncut,0,0.9/ncut,0]*j ;detrended intensity int2=(int1-smooth(int1,[1,sm_scl],/EDGE_TRUNCATE,/NAN))/smooth(int1,[1,sm_scl],/EDGE_TRUNCATE,/NAN) ;detrend and normalize int2_sort=int2(sort(int2)) & np=n_elements(int2[where(finite(int2))]) dminmax = [int2_sort[np*dpercent[0]], int2_sort[np*dpercent[1]]] plot_image,transpose(int2)dminmax[0],xtitle=xtitle,ytitle=ytitle[j eq 0],/noerase,xticklen=-0.020,xminor=5,$ title='cut '+strtrim(string(j),2)+' - detrended',charsize=1.0,position=posi-[0,0.5,0,0.5]+[0.9/ncut,0,0.9/ncut,0]*j save,filename=dir_sav+'stplot_cut='+strtrim(string(j),2)+'.sav',int_slice endfor write_png,dir_sav+'stplot.png', tvrd(/true), r, g, b end