###
# Copyright 2011 Diamond Light Source Ltd.
# 
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
# 
#   http://www.apache.org/licenses/LICENSE-2.0
# 
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
###

'''
Maths package
'''

import numpy as _np #@UnresolvedImport

from math import pi as _ppi
from math import e as _e

pi = _ppi
e = _e
inf = _np.inf
nan = _np.nan

isnan = _np.isnan

import sys
floatmax = sys.float_info.max # maximum float value #@UndefinedVariable

prod = _np.prod
sum = _np.sum #@ReservedAssignment

mean = _np.mean
std = _np.std
var = _np.var
ptp = _np.ptp

amax = _np.amax
amin = _np.amin

real = _np.real
imag = _np.imag

abs = _np.abs #@ReservedAssignment

absolute = _np.absolute

fabs = _np.fabs

angle = _np.angle

conjugate = _np.conjugate

conj = _np.conj

add = _np.add

subtract = _np.subtract

multiply = _np.multiply

divide = _np.divide

floor_divide = _np.floor_divide

remainder = _np.remainder

modf = _np.modf

fmod = _np.fmod

mod = _np.mod

reciprocal = _np.reciprocal

bitwise_and = _np.bitwise_and
bitwise_or = _np.bitwise_or
bitwise_xor = _np.bitwise_xor
invert = _np.invert
left_shift = _np.left_shift
right_shift = _np.right_shift

sin = _np.sin
cos = _np.cos
tan = _np.tan

arcsin = _np.arcsin
arccos = _np.arccos
arctan = _np.arctan
arctan2 = _np.arctan2
hypot = _np.hypot

sinh = _np.sinh
cosh = _np.cosh
tanh = _np.tanh

arcsinh = _np.arcsinh
arccosh = _np.arccosh
arctanh = _np.arctanh

log = _np.log
log2 = _np.log2
log10 = _np.log10
log1p = _np.log1p

exp = _np.exp
expm1 = _np.expm1

sqrt = _np.sqrt

square = _np.square
power = _np.power

floor = _np.floor
ceil = _np.ceil
rint = _np.rint
trunc = _np.trunc
fix = _np.fix

rad2deg = _np.rad2deg
deg2rad = _np.deg2rad

degrees = rad2deg
radians = deg2rad

sign = _np.sign
negative = _np.negative

clip = _np.clip
maximum = _np.maximum
minimum = _np.minimum

median = _np.median
cumprod = _np.cumprod
cumsum = _np.cumsum

diff = _np.diff

histogram = _np.histogram

dot = _np.dot
vdot = _np.vdot
inner = _np.inner
outer = _np.outer
matmul = _np.matmul
tensordot = _np.tensordot
kron = _np.kron
trace = _np.trace
cross = _np.cross

def gradient(f, *arg):
    '''Extended version of gradient that allows arrays for arg
    '''
    r = f.ndim
    al = len(arg)
    if al > 0:
        narg = [ a if not isinstance(a, _np.ndarray) else _np.gradient(a) for a in arg ]
        if al == 1:
            narg = [narg[0]]*r
        elif al != r:
            raise SyntaxError, "Invalid number of arguments"

        for i in range(al):
            a = narg[i]
            if isinstance(a, _np.ndarray):
                if a.ndim != r:
                    ashape = [1]*r
                    ashape[i] = a.size
                    a.shape = ashape

        result = _np.gradient(f)

        if r == 1:
            result /= narg[0]
        else:
            for i in range(r):
                result[i] /= narg[i]
        return result

    return _np.gradient(f, *arg)

roots = _np.roots

interp = _np.interp