from rsfproj import *

# percentage for thresholding
perc = {
    'cos':9,
    'dwt':6
    }

# Download data
Fetch('dunes3.HH','dunes')
Flow('dunes','dunes3.HH','dd form=native')

# Window size
n1=1024
n2=512

# Plotting macro
def plot(title):
    return '''
    grey color=H bias=-213 clip=150 title="%s"
    ''' % title

# Display data
Flow('dune','dunes','window n3=1 n1=%d n2=%d' % (n1,n2))
Result('dune',plot('Sand Dunes'))

# Transform dictionary
######################
transform = {
    'cos': ('Cosine Fourier',
            'cosft sign1=1 sign2=1',
            'cosft sign1=-1 sign2=-1'),
    'dwt': ('Digital Wavelet',
            '''
            dwt type=b inv=y unit=y | transp | 
            dwt type=b inv=y unit=y | transp
            ''',
            '''
            transp | dwt type=b inv=y unit=y adj=y | 
            transp | dwt type=b inv=y unit=y adj=y
            ''',3)
    }

for t in transform.keys():
    # Apply forward transform
    Flow(t,'dune',transform[t][1])
    Result(t,
           'grey title="%s Transform" ' % transform[t][0])

    # Sort coefficients
    sort = t+'-sort'
    Flow(sort,t,
         '''
         put n1=%d n2=1 d1=%g label1=Coefficient unit1=%% |
         math output="abs(input)" | sort | scale axis=1 |
         math output="10*log(input)/log(10)"
         ''' % (n1*n2,100.0/(n1*n2-1)))
    Result(sort,
           '''
           window max1=10 |
           graph title="%s Coefficient Decay" 
           label2=Magnitude unit2=DB
           ''' % transform[t][0])

    # Threshold and inverse transform
    thr = t+'-thr'
    inv = t+'-inv'
    Flow(thr,t,'threshold pclip=%g' % perc[t])
    Flow(inv,thr,transform[t][2])
    Plot(inv,plot('Decompressed Sand Dunes (%g%%)' % perc[t]))

    # Noise = Data - Signal
    diff = t+'-diff'
    Flow(diff,['dune',inv],'add scale=1,-1 ${SOURCES[1]}')
    Plot(diff,
         plot('Compression Noise (%g%%)' % perc[t]) + ' bias=0')

    Result(inv,[inv,diff],'SideBySideIso')

End()