Function _attribute

numexpr3/ne3compiler.py:1158–1205

AST factory function for attributes, such as :code:`numpy.pi`. Other than the attribute handle these are always treated similar to :code:`_name()`. An attribute has a .value node which is a Name, and .value.id is the module/class reference. Then .attr is the attribute refere

(self: NumExpr, node: ast.AST)

        

def _attribute(self: NumExpr, node: ast.AST) -> NumReg:
    '''
    AST factory function for attributes, such as :code:`numpy.pi`. Other 
    than the attribute handle these are always treated similar to 
    :code:`_name()`.

    An attribute has a .value node which is a Name, and .value.id is the 
    module/class reference. Then .attr is the attribute reference that 
    we need to resolve.

    **Only a single deference level is supported** To go deeper would require
    a recursive solution.

    :code:`.real` and :code:`.imag` need special handling because they are actually 
    mapped to function calls.
    '''
    if node.attr == 'imag' or node.attr == 'real':
        return _real_imag(self, node)

    className = node.value.id
    attrName = ''.join( [className, '.', node.attr] )
    
    if attrName in self.registers:
        register = self.registers[attrName]
        regToken = register.token
    else:
        regToken =  next(self._regCount)
        # Get address
        arr = None
        
        if className in self.local_dict:
            classRef = self.local_dict[className]
            if node.attr in classRef.__dict__:
                arr = self.local_dict[className].__dict__[node.attr]
        # Globals is, as usual, slower than the locals, so we prefer not to 
        # search it.  
        elif className in self._global_dict:
            classRef = self._global_dict[className]
            if node.attr in classRef.__dict__:
               arr = self._global_dict[className].__dict__[node.attr]
        
        if np.isscalar(arr):
            # Build tuple and add to the namesReg
            dchar = np.asarray(arr).dtype.char
            self.registers[attrName] = register = NumReg(regToken, attrName, arr, dchar, _KIND_ARRAY)
        else:
            self.registers[attrName] = register = NumReg(regToken, attrName, weakref.ref(arr), arr.dtype.char, _KIND_ARRAY )
    return register
    
def _real_imag(self: NumExpr, node: ast.AST) -> NumReg:
    '''