from rsf.proj import *

Flow('plane',None,
     '''
     spike n1=64 n2=32 d2=1 o2=1 label2=Trace unit2=
     nsp=3 k1=8,20,32 k2=4 l2=28 p2=2,1,0 
     ''')

Flow('timevel.asc',None,
     '''
     echo 
     0 0.15 0.2
     1500 2000 3000 
     n1=3 n2=2 in=$TARGET data_format=ascii_float
     ''')

Flow('velt','timevel.asc',
     'dd form=native | linear n1=64 d1=0.004 o1=0')
Plot('velt',
     '''
     graph wanttitle=n transp=y yreverse=y 
     label1=Time unit1=s label2=Velocity unit2=m/s
     ''')

Flow('velz','velt','time2depth velocity=$SOURCE intime=y nz=128 dz=2')
Plot('velz',
     '''
     graph wanttitle=n transp=y yreverse=y 
     label1=Depth unit1=m label2=Velocity unit2=m/s
     ''')
Result('vel','velt velz','SideBySideAniso')

Flow('velt2','velt','spray axis=2 n=32')
Flow('depth','plane velt2',
     '''
     time2depth velocity=${SOURCES[1]} intime=y nz=128 dz=2 |
     put label1=Depth unit1=m
     ''')

Flow('velz2','velz','spray axis=2 n=32')
Flow('plane2','depth velz2',
     'depth2time velocity=${SOURCES[1]} nt=64 dt=0.004')

for plot in ('plane','depth','plane2'):
    Plot(plot,
         '''
         wiggle clip=1 transp=y yreverse=y poly=y
         wanttitle=n wheretitle=b wherexlabel=t
         ''')

Result('depth','plane depth', 'SideBySideAniso')
Result('plane','plane plane2','SideBySideAniso')

End()