tut/augmented_matrix/AugmentTools.py

Go to the documentation of this file.

"""
simple example of how to create a vector and run 
commands
"""

from slimpy_base import Execute , vector, fft1, env
from slimpy_base import AugVector, AugOperator

from slimpy_base.setup import check_required, parse_env, parse_args, Parameters


def Augment_bldr(target,source,env):
    """
    Example builder showing how to use a simple aumented matrix
    Applys an fft transorn on two vectors then adds them
    """
    parse_env( env )
    
    x = vector( source[0] )
    y = vector( source[1] )
    
#    #Creating a vector of zeros from the vector space
    V = AugVector( [[x], 
                    [y]] )
    
    A = fft1( x.space )
    B = fft1( y.space )
    
    L = AugOperator( [[ A, B ], 
                      [ A, B ]] )
    
    # Apply the operator to the vector
    tmp = L*V
    
    RES = L.adj() * tmp 
    print map(str,target)
    # sets the name of the vectors to target[0] target[1]
    RES.setName( target[:2] )  #IGNORE:E1101 *pylint error
    
    Execute()
    

Parameters( 't1','t2','s1','s2' )

if __name__ == '__main__':
    
    # parse the command line arguments
    check_required('t1'); check_required('t2')
    check_required('s1'); check_required('s2')
    
    env = parse_args()
    
    target = [ env.pop('t1'), env.pop('t2') ]
    source = [ env.pop('s1'), env.pop('s2') ]
    
    Augment_bldr(target, source, env)

else:
#    __all__ = ['Augment']    
    from slimproj_core.builders.CreateBuilders import add_to_slim_env
    
    Augment = add_to_slim_env( "Augment", Augment_bldr )
    

---