from rsf.proj import *
from rsf.recipes.beg import server as private
from rsf.prog import RSFROOT

def Grey(data,other):
        Result(data,'grey labelsz=10 titlesz=10 clip=0.2 color=g label1=Depth unit1=m %s'%other)


#Fetch('elf7_win.HH','total',private)
Flow('post-data0','elf7_win.HH','dd form=native |window f3=18 n3=1 n2=150 | put unit1=s unit2= label2=Trace d2=1 o2=1 | scale axis=2 ')

Flow('post-gath','post-data0','cp')
Plot('post-gath','grey labelsz=12 titlesz=12.5 title="(a) Input" ')


Flow('post-mask','post-gath','math output=1 | pad n1=300')
Flow('post-gpad','post-gath','pad n1=300')

Flow('post-dip','post-gpad post-mask','dip rect1=10 rect2=10 mask=${SOURCES[1]} order=2')
Plot('post-dip','window n1=194 | grey labelsz=12 titlesz=12.5 color=j title="(b) Slope" allpos=y')

Flow('post-trace','post-dip','window n2=1 | math output=x1')
Flow('post-pick','post-dip post-trace','pwpaint seed=${SOURCES[1]} eps=0.05 order=2')

Plot('post-pick0','post-pick',
     '''
     window n1=194 |
     grey labelsz=12 titlesz=12.5 color=j allpos=y
     title="(d) Relative Age" clip=3.2
     ''')
Plot('post-pick1','post-pick',
     '''
     window n1=194 |
     contour c0=0 dc=0.05 nc=64
     transp=y yreverse=y plotcol=7 plotfat=5
     wanttitle=n wantaxis=n
     ''')
Plot('post-pick','post-pick0 post-pick1','Overlay')

Flow('post-flat','post-gpad post-pick','iwarp warp=${SOURCES[1]} n1=194')

Flow('post-gath-rec','post-flat post-pick','iwarp warp=${SOURCES[1]} inv=n | window n1=194')

Plot('post-flat','grey labelsz=12 titlesz=12.5 title="(d) Flattened" ')
Plot('post-gath-rec','grey labelsz=12 titlesz=12.5 title="Mapped back" ')


########################################################################
# INITIALIZATION
########################################################################
matlab         = WhereIs('matlab')
matROOT = '../matfun'
matfun = 'FX_EMD'
matlabpath = os.environ.get('MATLABPATH',os.path.join(RSFROOT,'lib'))

if not matlab:
    sys.stderr.write('\nCannot find Matlab.\n')
    sys.exit(1)
n1=194
n2=150
dt=0.004
lf=5
hf=120
N=1
verb=0
############################################################
## with parameter
############################################################
Flow('post-flat-emd0',[os.path.join(matROOT,matfun+'.m'),'post-flat'],
     '''MATLABPATH=%(matlabpath)s %(matlab)s 
     -nosplash -nojvm -r "addpath %(matROOT)s;%(matfun)s('${SOURCES[1]}','${TARGETS[0]}',%(n1)d,%(n2)d,%(dt)g,%(lf)g,%(hf)g,%(N)d,%(verb)d);quit"
     '''%vars(),stdin=0,stdout=-1)

Flow('post-flat-emd','post-flat-emd0','put d2=1 d1=12.5 o2=1 o1=925 label2=Trace unit2= ')
Flow('post-dif0','post-gath post-gath-rec','add scale=1,-1 ${SOURCES[1]}')
Flow('post-emd-rec','post-flat-emd post-pick post-dif0','iwarp warp=${SOURCES[1]} inv=n | window n1=194 | add scale=1,1 ${SOURCES[2]}')

Flow('post-dif','post-gath post-emd-rec','add scale=1,-1 ${SOURCES[1]}')

Grey('post-gath','title="Original"')
Grey('post-flat','title="Flattened"')
Grey('post-flat-emd','title="EMD"')
Grey('post-emd-rec','title="Mapped back"')
Grey('post-dif','title="Difference"')



Flow('post-zooma-1','post-gath','window f1=50 n1=50 f2=120 n2=20')
Flow('post-zooma-2','post-emd-rec','window f1=50 n1=50 f2=120 n2=20')

Flow('post-zoomb-1','post-gath','window f1=140 n1=50 f2=60 n2=40')
Flow('post-zoomb-2','post-emd-rec','window f1=140 n1=50 f2=60 n2=40')

Grey('post-zooma-1','title="Zoomed C (original)"')
Grey('post-zooma-2','title="Zoomed C (Filtered)"')
Grey('post-zoomb-1','title="Zoomed D (original)"')
Grey('post-zoomb-2','title="Zoomed D (Filtered)"')


## Creating framebox1
x=380
y=0.54
w=60
w1=0.50

Flow('frame1.asc',None,'echo %s n1=10 data_format=ascii_float in=$TARGET'% \
        ' '.join(map(str,(x,y,x+w,y,x+w,y+w1,x,y+w1,x,y))))
Plot('frame1','frame1.asc',
        '''
        dd type=complex form=native |
        graph min1=0 max1=471 min2=0 max2=2.048 pad=n plotfat=15 plotcol=4 
        wantaxis=n wanttitle=n yreverse=y 
        ''')

## Creating framebox2
x=185
y=1.50
w=60
w1=0.50

Flow('frame2.asc',None,'echo %s n1=10 data_format=ascii_float in=$TARGET'% \
        ' '.join(map(str,(x,y,x+w,y,x+w,y+w1,x,y+w1,x,y))))
Plot('frame2','frame2.asc',
        '''
        dd type=complex form=native |
        graph min1=0 max1=471 min2=0 max2=2.048 pad=n plotfat=15 plotcol=2 
        wantaxis=n wanttitle=n yreverse=y 
        ''')

## Create label A
Plot('labela',None,
        '''
        box x0=11.2 y0=7.0 label="C" xt=-0.5 yt=0.5 length=0.75 
        ''')

## Create label B
Plot('labelb',None,
        '''
        box x0=6.4 y0=3.3 label="D" xt=-0.5 yt=0.5 length=0.75
        ''')

Result('post-gath0','Fig/post-gath.vpl frame1 frame2 labela labelb','Overlay')
Result('post-emd-rec0','Fig/post-emd-rec.vpl frame1 frame2 labela labelb','Overlay')



End()