Python DataIO.read方法代码示例

# 需要导入模块: from cc.tools.io import DataIO [as 别名]
# 或者: from cc.tools.io.DataIO import read [as 别名]
    def __init__(self,x,y,func,*args,**kwargs):
    
        '''
        Create an instance of the Profiler2D() class. Requires 2 coordinate 
        arrays and a function object for profile.
       
        The function can also be given as an interpolation object.
        
        The optional args and kwargs give the additional arguments for the 
        function, which are ignored in case func is an interpolation object.
        
        The default coordinate grids are both evaluated for the function. 
        They are saved in self.z, as an array of dimensions (x.size,y.size). 
        Alternatively, new evaluations can be attained through eval and diff.
        
        @param x: The default coordinates of the primary independent variable. 
                  Minimum three points.
        @type x: array
        @param y: The default coordinates of the secondary independent variable. 
                  Minimum three points.
        @type y: array
        @param func: The function that describes the profile with respect to x 
                     and y. Can be given as a 2D interpolation object.
        @type func: function/interpolation object
        
        @keyword args: Additional parameters passed to the functions when eval
                       or diff are called. 
                       
                       (default: [])
        @type args: tuple
        @keyword kwargs: Additional keywords passed to the functions when eval
                         or diff are called. 
                       
                         (default: {})
        @type kwargs: dict
        
        '''

        #-- If the function is given as a string, retrieve it from the local 
        #   child module, or from the Profiler module (as parent). If the latter
        #   check if the function comes from one of Profiler's attributes, such
        #   as the loaded DataIO module.
        if isinstance(func,str):
            try:
                #-- Note that self.module refers to the child, not Profiler
                func = getattr(sys.modules[self.__module__],func)
            except AttributeError:
                #-- Recursively find the function of loaded modules in Profiler
                #   or a function of the Profiler module itself if no '.' 
                func = DataIO.read(sys.modules[__name__],return_func=1)
                
        #-- set functional args, remember spline order. args/kwargs for 
        #   func are saved separately. They are removed if either are
        #   interpolation objects. They are always accessible, the _** variables
        #   are passed to the evaluation.
        self.args = args
        self.kwargs = kwargs
        self._args = args
        self._kwargs = kwargs
        self.func = func

        if isinstance(self.func,interp2d) \
                or isinstance(self.func,BivariateSpline):
            self.interp_func = 1
            self._args = []
            self._kwargs = {}
        else:
            self.interp_func = 0
        
        #-- Evaluate the default grid with function. Set x as None first so it
        #   can actually evaluate.
        self.x = None
        self.y = None
        self.z = self.func(x,y,*self._args,**self._kwargs)
        
        #-- Now set x and y.
        self.x = x
        self.y = y