from rsf.proj import *

nx=64
nz=64
dx=4
dz=4

lengthx = (nx-1)*dx
lengthz = (nz-1)*dz

par = dict(alpha=20,
           x0=lengthx*0.5,z0=lengthz*0.3,
           x1=lengthx*0.5,z1=lengthz*0.5)

Flow('vel',None,
    '''
    math n1=61 n2=61 d1=4 d2=4 o1=0 o2=0
    label1=Depth   unit1=m
    label2=Surface unit2=m
    label=Velocity unit=m/s
    output="-2000*exp(-((x1-%(z0)g)*(x1-%(z0)g)+(x2-%(x0)g)*(x2-%(x0)g))/%(alpha)g/%(alpha)g)+5000+2000*exp(-((x1-%(z1)g)*(x1-%(z1)g)+(x2-%(x1)g)*(x2-%(x1)g))/%(alpha)g/%(alpha)g)-2000"
    ''' % par)

Plot('vel',
     'grey allpos=y scalebar=y bartype=h wanttitle=n wherexlabel=t')

Plot('gauss-wv time.rsf ampl.rsf','vel',
     '''
     norsar time=${TARGETS[1]} ampl=${TARGETS[2]}
     model=4
     gox=-120
     nang=21 rays=1 wfront=1 gap=1 inter=1
     DSmax=4. dt=0.0005 nt=2 nrmax=2000
     first=1 nou=1 ns=1 ds=1. os=120. depth=175.
     freq=80. N=3
     ''')

Result('gauss-wv','vel gauss-wv','Overlay')

End()