tut/wavelet_denoise/Set2/step1/op_imospray.py

Go to the documentation of this file.

from slimpy_base.Core.User.linop.rclinOp import linearop_r as LinearOperatorStruct
from numpy import ceil
from slimpy_base.api.Plugins.slim2rsf.sfCommandFactory import rsfCommandFactory
from slimpy_base.api.Plugins.slim2rsf.sfcommands.sfConverter import sfConverter

########################################################################
# Define how fftx behaves
########################################################################

# create a Converter class
class imospray_Converter( sfConverter ):
    """
    This is a mapping instance that maps a SLIMpy command into an object that 
    can be run 
    This also maps the agruments and keyword arguments to be pugin specific
    """
    # alwas use classmethod or static method
    @classmethod
    def map( cls, source, command ):
        """
        map a SLIMpy command to a rsf command
        """
        # the RSF bin and 'sf' will be automaticaly prepended to fft1
        command = cls.default_function( command, "imospray" ) 
        #change all python True/False objects to 'y'/'n' strings
        command = cls.truefalseHelper( command )
        #alwas must return a CommandPack instance
        return cls.pack( command )

    @classmethod
    def trans( cls, command, space, *spaces ):
        'define how this operator affect the space'
        
        n2 = command['n2']
        space["n2"] = n2
        return space
    
    @classmethod
    def trans_adj( cls, command, space, *spaces ):
        """
        trans_adj will automatically be called in the case
        where the command has an adj keyword that is true
        """
        space['n2']= 1
        return space
    
    @classmethod
    def constr( cls, command, space ):
        'make sure the data on the forward command is float'
        cls.match( space, data_type='float' )
        
    @classmethod
    def constr_adj( cls, command, space ):
        'make sure the data on the adjoint command is float'
        cls.match( space, data_type='float' )


#===============================================================================
# Linear operator class itself 
#===============================================================================
class IMOSpray( LinearOperatorStruct ):
    """
    Inversion of constant-velocity nearest-neighbor inverse NMO
    """
    name = "imospray"
    
    def __init__( self, domain, n2, d2=200,o2=0, velocity=1000, adj=False ):
        
        kparams = dict( n2=n2, d2=d2,o2=o2, v=velocity, adj=adj )

        LinearOperatorStruct.__init__( self, domain, **kparams )

#===============================================================================
# Add it to the list of existing rsf commands  
#===============================================================================

factory = rsfCommandFactory()
# add the converter class to the factory
# now converter will be invoked when 
# a linear operator with the 'name' attribute
# of fftX
factory['imospray'] = imospray_Converter

---