def ravel_multi_index(multi_index, dims, mode='raise'):
'''Converts a tuple of coordinate arrays to an array of flat indexes
'''
if isinstance(mode, tuple):
mode = [_prep_mode.get(m, -1) for m in mode]
else:
mode = _prep_mode.get(mode, -1)
if isinstance(multi_index, _ds): # split single array
multi_index = [ _getslice(multi_index, i) for i in range(multi_index.shape[0]) ]
single = False
if isinstance(multi_index[0], int):
single = True
multi_index = [ array(m)._jdataset() for m in multi_index ]
pos = _dsutils.calcIndexesFromPositions(multi_index, dims, mode)
if single:
return pos.getObject([])
return pos
