forked from 170010011/fr
57 lines
1.6 KiB
Python
57 lines
1.6 KiB
Python
#
|
|
# Author: Pearu Peterson, March 2002
|
|
#
|
|
|
|
__all__ = ['get_flinalg_funcs']
|
|
|
|
# The following ensures that possibly missing flavor (C or Fortran) is
|
|
# replaced with the available one. If none is available, exception
|
|
# is raised at the first attempt to use the resources.
|
|
try:
|
|
from . import _flinalg
|
|
except ImportError:
|
|
_flinalg = None
|
|
# from numpy.distutils.misc_util import PostponedException
|
|
# _flinalg = PostponedException()
|
|
# print _flinalg.__doc__
|
|
has_column_major_storage = lambda a:0
|
|
|
|
|
|
def has_column_major_storage(arr):
|
|
return arr.flags['FORTRAN']
|
|
|
|
|
|
_type_conv = {'f':'s', 'd':'d', 'F':'c', 'D':'z'} # 'd' will be default for 'i',..
|
|
|
|
|
|
def get_flinalg_funcs(names,arrays=(),debug=0):
|
|
"""Return optimal available _flinalg function objects with
|
|
names. Arrays are used to determine optimal prefix."""
|
|
ordering = []
|
|
for i in range(len(arrays)):
|
|
t = arrays[i].dtype.char
|
|
if t not in _type_conv:
|
|
t = 'd'
|
|
ordering.append((t,i))
|
|
if ordering:
|
|
ordering.sort()
|
|
required_prefix = _type_conv[ordering[0][0]]
|
|
else:
|
|
required_prefix = 'd'
|
|
# Some routines may require special treatment.
|
|
# Handle them here before the default lookup.
|
|
|
|
# Default lookup:
|
|
if ordering and has_column_major_storage(arrays[ordering[0][1]]):
|
|
suffix1,suffix2 = '_c','_r'
|
|
else:
|
|
suffix1,suffix2 = '_r','_c'
|
|
|
|
funcs = []
|
|
for name in names:
|
|
func_name = required_prefix + name
|
|
func = getattr(_flinalg,func_name+suffix1,
|
|
getattr(_flinalg,func_name+suffix2,None))
|
|
funcs.append(func)
|
|
return tuple(funcs)
|