from rsf.proj import*
def Grey(data,other):
Result(data,'window f2=3 |grey label2=Trace unit2="" clip=0.8 label1=Time unit1="s" title="" wherexlabel=b wanttitle=n screenratio=1.2 %s '%other)
def Greyplot(data,other):
Plot(data,'window f2=3 |grey label2=Trace unit2="" clip=0.8 label1=Time unit1="s" title="" wherexlabel=b wanttitle=n screenratio=1.2 %s '%other)
def Graph(data,other):
Result(data,'graph label1="Iter #no" label2="SNR" unit2=dB unit1="" title="" wherexlabel=b wheretitle=t %s' %other)
Flow('vrms',None,
'math d1=0.004 n1=1001 o1=0 output="4500" ')
Flow('synt',None,
'''
spike d1=0.004 n1=1001 |
noise rep=y seed=2006 |
cut n1=100 |
bandpass flo=4 fhi=20 |
spray axis=2 n=256 d=12.5 o=-1600 label=Offset unit=m
''')
Flow('hyper','synt vrms',
'inmo velocity=${SOURCES[1]} half=y | noise seed=2007 var=1e-10 | scale axis=2 | put o2=0 d2=1 | window n1=501')
Grey('hyper','')
Flow('hyper-zero','hyper','zerotrace j=2 l=1')
Grey('hyper-zero','')
Flow('mask-t',None,'math n1=1 n2=256 d2=1 output="1" | zerotrace j=2 l=1')
Flow('mask','mask-t','window |spray axis=1 n=501 d=0.004 o=0')
Grey('mask','color=j')
Flow('mask1','mask','math output="1-input"')
Flow('fk-hyper-zero','hyper-zero','rtoc | fft3 axis=1 pad=2 | fft3 axis=2 pad=2| cabs | window f1=512')
Flow('fk-hyper','hyper','rtoc | fft3 axis=1 pad=2 | fft3 axis=2 pad=2| cabs | window f1=512')
Grey('fk-hyper-zero','allpos=y color=j clip=255 label1=Frequency unit1=Hz label2=Wavenumber unit2= min2=-0.5 max2=0.5')
Grey('fk-hyper','allpos=y color=j clip=255 label1=Frequency unit1=Hz label2=Wavenumber unit2= min2=-0.5 max2=0.5')
ddip=3
fhi=30
r1=10
r2=10
padno=256
thr0=8
niter=200
mode='s'
forw = 'seislet dip=${SOURCES[1]} adj=y inv=y eps=0.1 type=b'
back = 'seislet dip=${SOURCES[1]} inv=y eps=0.1 type=b'
sig='hyper-zero'
data_pocs = sig
plots_pocs = [sig]
diffsa_pocs =[]
diffsb_pocs =[]
diffs_pocs =[]
snrs_pocs =[]
dips_pocs=[]
datas_pocs=[]
snrs_pocs=[]
Flow('dipmask','hyper','math output=1 | pad n2=%d'%(padno))
for iter in range(niter):
thr=thr0+((8.-thr0)*iter*iter/((niter-1)*(niter-1)))
if iter % ddip ==0 :
dip_pocs='dip-pocs%d'%int(iter/ddip)
Flow(dip_pocs,[data_pocs,'dipmask'],
'''
bandpass fhi=%d | pad n2=%d |
dip mask=${SOURCES[1]} rect1=%d rect2=%d liter=30
'''%(fhi,padno,r1,r2))
dips_pocs.append(dip_pocs)
Greyplot(dip_pocs,'clip=1 color=j')
old_pocs = data_pocs
data_pocs = 'data-pocs%d' % iter
diffa_pocs = 'diffa-pocs%d'%iter
diffb_pocs = 'diffb-pocs%d'%iter
diff_pocs = 'diff-pocs%d' %iter
snr_pocs = 'snr-pocs%d' %iter
Flow(data_pocs,[old_pocs,dip_pocs,'mask1',sig],
'''
%s | threshold1 ifperc=1 mode=%s thr=%g |
%s | mul ${SOURCES[2]} |
add ${SOURCES[3]}
''' % (forw,mode,thr,back))
Flow(diff_pocs,['hyper',data_pocs],'add scale=1,-1 ${SOURCES[1]}')
Flow(snr_pocs,['hyper',diff_pocs],'snr2 noise=${SOURCES[1]}')
Greyplot(data_pocs,'title="Iteration %d"' % (iter+1))
datas_pocs.append(data_pocs)
snrs_pocs.append(snr_pocs)
Flow('snrs-pocs',snrs_pocs,'cat axis=1 ${SOURCES[1:%d]}'%(len(snrs_pocs)))
Plot('dips-pocs',dips_pocs,'Movie')
Plot('datas-pocs',datas_pocs,'Movie')
Flow('hyper-seis',data_pocs,'cp')
Flow('hyper-seis-dif','hyper hyper-seis','add scale=1,-1 ${SOURCES[1]}')
Grey('hyper-seis','')
Grey('hyper-seis-dif','')
Flow('fk-hyper-seis','hyper-seis','rtoc | fft3 axis=1 pad=2 | fft3 axis=2 pad=2| cabs | window f1=512')
Grey('fk-hyper-seis','allpos=y color=j clip=255 label1=Frequency unit1=Hz label2=Wavenumber unit2= min2=-0.5 max2=0.5')
Graph('snrs-pocs','')
Result(dip_pocs,dip_pocs,'Overlay')
End()
|