tut/integrate_new_operator/IntegrateLinOp.py

Go to the documentation of this file.

"""
Simple example on how to Integrate a linear
Operator into SLIMpy.
"""


from NewFFT import fft1X
from slimpy_base.setup import *
from slimpy_base import Execute
from slimpy_base import vector



def new_fft_bldr( target, source, env ):
    """
    use our new fft operator
    """    
    z = vector( source[0] )

    
    F = fft1X( z.space, sym=True, opt=False )
    
    x = F*z
    
    scal_max = x.norm( 1 )
    
    tmp = x - scal_max
    
    res = F.adj( ) * tmp
    
    res.setName( 'fftRes' ) #IGNORE:E1101
    
    Execute( )
    

if __name__ == '__main__':
    
    Parameters( 'target', 'source' )
    
    # add standalone functionality
    # parse the command line arguments
    check_required( 'target' )
    check_required( 'source' )
    
    env = parse_args( )
#    defaults.update(env)
    
    target = [env.pop( 'target' )]
    source = [env.pop( 'source' )]
    
    new_fft_bldr( target, source, env  )
    

else:
    
    # Add integration with scons 
    __all__ = ['New_FFT']
    
    # add Simple function to SCons environment
    from slimproj_core.builders.CreateBuilders import add_to_slim_env
    
    New_FFT = add_to_slim_env("New_FFT", new_fft_bldr )
    

---