from rsf.proj import *

#####Smooth-vary Velocty 
Flow('vel7',None,
     '''
     spike n1=513 d1=5 k1=98 n2=513 d2=5 k2=98 |
     math output="550+0.00015*(x1-800)*(x1-800)+0.0001*(x2-500)*(x2-500)"
     ''')
Plot('vel7',
       '''
       grey transp=n pclip=100 screenratio=1 color=i
       label1="X" label2="Z" scalebar=y poly=y bias=1200
       unit1=m unit2=m  title=a barlabel=Velocity barunit=m/s
       ''')
Result('vel7',
       '''
       grey transp=n pclip=100 screenratio=1 color=j
       label1="X" label2="Z" scalebar=y poly=y mean=y
       unit1=m unit2=m  title=Velocity barlabel=Velocity barunit=m/s
       ''')
nt=1250
dt=0.002
nv=2
Flow('source7',None,
     '''
     spike n1=%d d1=%g k1=100 | 
     ricker1 frequency=20 
     '''%(nt,dt))
Result('source7','graph  title="Source Wavelet" ')
Flow('wavp7',['source7','vel7'],
     '''
     pspifcos2bs vel=${SOURCES[1]}  dt=%g nt=%d nv=%d isx=256 isz=256 nb=0 c=0.01 
     '''%(dt,nt,nv) )

Plot('wavp7',
     ''' 
     window j3=20 |
     grey poly=y label2="Z" label1="X" title="pspif2" unit1="m" unit2="m"
     yreverse=y transp=n gainpanel=each pclip=100
     ''',view=1)
#Flow('wavp_dehf7',['source7','vel7'],
#     '''
#     pspifcos2bs_dehf vel=${SOURCES[1]}  dt=%g nt=%d nv=%d isx=256 isz=256 nb=0 c=0.01 ax=1.0 az=1.0 factor=0.9 
#     '''%(dt,nt,nv) )

#Result('wavp_dehf7',
#       ''' 
#       window j3=20 |
#       grey poly=y label2="Z" label1="X" title="pspif2"
#       yreverse=y transp=n gainpanel=each pclip=100 unit1="m" unit2="m"
#       ''' )
#Flow('wavf7',['source7','vel7'],
#     '''
#     ffdcos2bs vel=${SOURCES[1]}  dt=%g nt=%d isx=256 isz=256 nb=0 c=0.01 
#     '''%(dt,nt) )
Flow('wavf7',['source7','vel7'],
     '''
     ffd2dehf vel=${SOURCES[1]}  dt=%g nt=%d isx=256 isz=256 nbb=0 nbl=0 nbt=0 nbr=0 ax=1.0 az=1.0 factor=0.9 
     '''%(dt,nt) )

Plot('wavf7',
     ''' 
     window j3=20 |
     grey poly=y label2="Z" label1="X" title="ffd2"
     yreverse=y transp=n gainpanel=each pclip=100 unit1="m" unit2="m"
     ''',view=0)
Flow('wavfd7',['source7','vel7'],
     '''
     fd2bs vel=${SOURCES[1]}  dt=%g nt=%d isx=256 isz=256 nb=0 c=0.01 
     '''%(dt,nt) )

Plot('wavfd7',
     ''' 
     window j3=20 |
     grey poly=y label2="Z" label1="X" title="fd2" unit1="m" unit2="m"
     yreverse=y transp=n gainpanel=each pclip=100
     ''',view=0)
Flow('wavsp7',['source7','vel7'],
     '''
     sp vel=${SOURCES[1]}  dt=%g nt=%d isx=256 isz=256 nb=0 c=0.01 ax=1.0 az=1.0 factor=0.7 
     '''%(dt,nt) )

Plot('wavfd7',
     ''' 
     window j3=20 |
     grey poly=y label2="Z" label1="X" title="fd2" unit1="m" unit2="m"
     yreverse=y transp=n gainpanel=each pclip=100
     ''',view=0)
Flow('wavfd_7','wavfd7','window n3=1 f3=800')
Flow('wavffd_7','wavf7','window n3=1 f3=800')
Flow('wavpspi_7','wavp7','window n3=1 f3=800')
#Flow('wavpspi_dehf7','wavp_dehf7','window n3=1 f3=800')
Flow('wavsp_7','wavsp7','window n3=1 f3=800')
Flow('compare_7',['wavfd_7','wavffd_7','wavpspi_7'],
     '''
     cat ${SOURCES[1]} ${SOURCES[2]} axis=3
     ''')
Plot('wavfd_7',
       '''
       grey poly=y label2="Z" label1="X" title=b screenratio=1
       yreverse=y transp=n gainpanel=all pclip=99 unit1=m unit2=m
       ''' )
Plot('wavffd_7',
       '''
       grey poly=y label2="Z" label1="X" title=c screenratio=1
       yreverse=y transp=n gainpanel=all pclip=99 unit1=m unit2=m
       ''' )
Plot('wavpspi_7',
       '''
       grey poly=y label2="Z" label1="X" title=d screenratio=1
       yreverse=y transp=n gainpanel=all pclip=99 unit1=m unit2=m
       ''' )
#Plot('wavpspi_dehf7',
#       '''
#       grey poly=y label2="Z" label1="X" title=d screenratio=1
#       yreverse=y transp=n gainpanel=all pclip=99 unit1=m unit2=m labelsz=6
#       ''' )
Plot('wavsp_7',
       '''
       grey poly=y label2="Z" label1="X" title=d screenratio=1
       yreverse=y transp=n gainpanel=all pclip=99 unit1=m unit2=m labelsz=6
       ''' )
Result('wavfd','wavfd_7',
       '''
       grey poly=y label2="Depth" label1="Distance" title= screenratio=1
       yreverse=y transp=n gainpanel=all pclip=99 unit1=m unit2=m labelsz=6 
       ''' )
Result('wavffd','wavffd_7',
       '''
       grey poly=y label2="Depth" label1="Distance" title= screenratio=1
       yreverse=y transp=n gainpanel=all pclip=99 unit1="m" unit2="m" labelsz=6
       ''' )
Result('wavpspi','wavpspi_7',
       '''
       grey poly=y label2="Depth" label1="Distance" title= screenratio=1
       yreverse=y transp=n gainpanel=all pclip=99 unit1="m" unit2="m" labelsz=6
       ''' )
Result('wavsp','wavsp_7',
       '''
       grey poly=y label2="Depth" label1="Distance" title= screenratio=1
       yreverse=y transp=n gainpanel=all pclip=99 unit1="m" unit2="m" labelsz=6
       ''' )
End()