from rsf.proj import *
import math

Fetch('salt_slow_desp.HH','segeage')
Flow('salt','salt_slow_desp.HH','dd form=native')

Flow('seg','salt',
     'add mode=p $SOURCE | window j1=2 j2=2 j3=2 | eikonal vel=0 zshot=1500')

def traverse(n2,n3,d2,d3):
    return 'put n2=%d n3=1 d2=%g | window j2=%d' % \
           (n2*n3,math.hypot(d2,d3)/(n2+1),n2+1)

Flow('st','salt',traverse(676,676,20,20))
Flow('sgt','seg',traverse(338,338,10,10))

Flow('sxy','salt','window n1=1 f1=75')
Flow('sxy2','sxy',
     'remap1 d1=25 n1=512 | transp | remap1 d1=25 n1=512 | transp')

w = 10  # Hz
dx = 20 # m

Flow('kmax','sxy2','scale dscale=%g' % (w*dx))

Plot('sxy',
     '''
     grey pclip=100 gpow=1 bias=0.00034
     color=j scalebar=n
     title="SEG/EAGE salt model" screenratio=1 screenht=9.6571
     ''')
Plot('exy','seg',
     '''
     window n1=1 f1=75 |
     contour nc=100 plotcol=7 wantaxis=n wanttitle=n
     screenratio=1 screenht=9.6571
     ''')
Plot('cxy','sxy exy','Overlay')

Plot('sxz','st',
     '''
     grey pclip=100 gpow=1 bias=0.00034 
     color=j scalebar=n wanttitle=n screenratio=0.21966 screenht=3
     ''')
Plot('exz','sgt',
     '''
     contour nc=100 plotcol=7 wantaxis=n wanttitle=n
     screenratio=0.21966 screenht=3
     ''')
Plot('cxz','sxz exz','Overlay')

Result('salt','cxz cxy','OverUnderIso')

End()