from rsf.proj import *

Fetch('avo.HH','avo')
Flow('avo','avo.HH','dd form=native')

Flow('b1','avo','window f1=50 n1=250 j2=2')
Flow('n','b1','window n2=12')
Flow('ms','b1','window f2=12 n2=12 | scale dscale=0.80')
Flow('f','b1','window f2=24')
Flow('bsr','n ms f','cat axis=2 ${SOURCES[1:3]} | put label1=Time unit1=s label2=Offset unit2=km')

Plot('seafloor',None,'box x0=8.168333 y0=6.953333 label="seafloor" xt=1 yt=-1')
Plot('bsr-ann',None, 'box x0=5.533333 y0=3.100000 label="BSR" xt=-1 yt=1')

Plot('bsr','wiggle wanttitle=n transp=y yreverse=y poly=y wheretitle=t')

Result('bsr-ann','bsr seafloor bsr-ann','Overlay')

Flow('nb1','avo','window f1=200 n1=100 j2=2')
Flow('nn','nb1','window n2=12')
Flow('nms','nb1','window f2=12 n2=12 | scale dscale=0.80')
Flow('nf','nb1','window f2=24')
Flow('bsr-nmo','nn nms nf','cat axis=2 ${SOURCES[1:3]} | put label1=Time unit1=s label2=Offset unit2=km')

tp0=4.975
twidth=0.025
tp1=tp0+twidth

Flow('bsr-picks','bsr-nmo','scale dscale=0.375 | window min1=%g max1=%g | sfmin axis=1' % (tp0,tp1))
Plot('bsr-picks',
     '''
     graph  wanttitle=n wantaxis=n min2=-0.5 max2=0 symbol=+ symbolsz=5
     ''')

vp1=3.00
vs1=1.2
rho1=1.7

vp2=1.65
vs2=0.66
rho2=1.6

nth=30
th0=0
dth=1

vp1 = dict(thin=3.0, brine=2.5, gas=2.5)
vs1 = dict(thin=1.2, brine=1.0, gas=0.5)
vs2 = dict(thin=0.66,brine=0.66,gas=1.1)

for case in vp1.keys():
    Flow(case,None,
         '''
         zoeppritz na=30 a0=0 da=1 vp1=%g vp2=1.65 vs1=%g vs2=%g rho1=1.7 rho2=1.6
         ''' % (vp1[case],vs1[case],vs2[case]))
    Plot(case,
         '''
         graph wanttitle=n label1=angle unit1="degree" label2="reflection amplitude" unit2=
         min2=-0.5 max2=0 max1=30 plotcol=5
         ''')
    Result(case,[case,'bsr-picks'],'Overlay')

End()