/*
 * Copyright 2011-2013 Branimir Karadzic. All rights reserved.
 * License: http://www.opensource.org/licenses/BSD-2-Clause
 */

// FPU math lib

#ifndef __FPU_MATH_H__
#define __FPU_MATH_H__

#define _USE_MATH_DEFINES
#include <math.h>
#include <string.h>

inline float fmin(float _a, float _b)
{
	return _a < _b ? _a : _b;
}

inline float fmax(float _a, float _b)
{
	return _a > _b ? _a : _b;
}

inline float fclamp(float _a, float _min, float _max)
{
	return fmin(fmax(_a, _min), _max);
}

inline float fsaturate(float _a)
{
	return fclamp(_a, 0.0f, 1.0f);
}

inline float flerp(float _a, float _b, float _t)
{
	return _a + (_b - _a) * _t;
}

inline float fsign(float _a)
{
	return _a < 0.0f ? -1.0f : 1.0f;
}

inline void vec3Add(float* __restrict _result, const float* __restrict _a, const float* __restrict _b)
{
	_result[0] = _a[0] + _b[0];
	_result[1] = _a[1] + _b[1];
	_result[2] = _a[2] + _b[2];
}

inline void vec3Sub(float* __restrict _result, const float* __restrict _a, const float* __restrict _b)
{
	_result[0] = _a[0] - _b[0];
	_result[1] = _a[1] - _b[1];
	_result[2] = _a[2] - _b[2];
}

inline void vec3Mul(float* __restrict _result, const float* __restrict _a, const float* __restrict _b)
{
	_result[0] = _a[0] * _b[0];
	_result[1] = _a[1] * _b[1];
	_result[2] = _a[2] * _b[2];
}

inline void vec3Mul(float* __restrict _result, const float* __restrict _a, float _b)
{
	_result[0] = _a[0] * _b;
	_result[1] = _a[1] * _b;
	_result[2] = _a[2] * _b;
}

inline float vec3Dot(const float* __restrict _a, const float* __restrict _b)
{
	return _a[0]*_b[0] + _a[1]*_b[1] + _a[2]*_b[2];
}

inline void vec3Cross(float* __restrict _result, const float* __restrict _a, const float* __restrict _b)
{
	_result[0] = _a[1]*_b[2] - _a[2]*_b[1];
	_result[1] = _a[2]*_b[0] - _a[0]*_b[2];
	_result[2] = _a[0]*_b[1] - _a[1]*_b[0];
}

inline void vec3Norm(float* __restrict _result, const float* __restrict _a)
{
	float scale = 1.0f/sqrtf(vec3Dot(_a, _a) );
	_result[0] = _a[0] * scale;
	_result[1] = _a[1] * scale;
	_result[2] = _a[2] * scale;
}

inline void mtxIdentity(float* _result)
{
	memset(_result, 0, sizeof(float)*16);
	_result[0] = _result[5] = _result[10] = _result[15] = 1.0f;
}

inline void mtxTranslate(float* _result, float _x, float _y, float _z)
{
	mtxIdentity(_result);
	_result[12] = _x;
	_result[13] = _y;
	_result[14] = _z;
}

inline void mtxScale(float* _result, float _x, float _y, float _z)
{
	memset(_result, 0, sizeof(float) * 16);
	_result[0] = _x;
	_result[5] = _y;
	_result[10] = _z;
	_result[15] = 1.0f;
}

inline void mtxLookAt(float* __restrict _result, const float* __restrict _eye, const float* __restrict _at)
{
	float tmp[4];
	vec3Sub(tmp, _at, _eye);

	float view[4];
	vec3Norm(view, tmp);

	float up[3] = { 0.0f, 1.0f, 0.0f };
	vec3Cross(tmp, up, view);

	float right[4];
	vec3Norm(right, tmp);

	vec3Cross(up, view, right);

	memset(_result, 0, sizeof(float)*16);
	_result[ 0] = right[0];
	_result[ 1] = up[0];
	_result[ 2] = view[0];

	_result[ 4] = right[1];
	_result[ 5] = up[1];
	_result[ 6] = view[1];

	_result[ 8] = right[2];
	_result[ 9] = up[2];
	_result[10] = view[2];

	_result[12] = -vec3Dot(right, _eye);
	_result[13] = -vec3Dot(up, _eye);
	_result[14] = -vec3Dot(view, _eye);
	_result[15] = 1.0f;
}

inline void mtxProj(float* _result, float _fovy, float _aspect, float _near, float _far)
{
	float height = 1.0f/tanf(_fovy*( (float)M_PI/180.0f)*0.5f);
	float width = height * 1.0f/_aspect;
	float aa = _far/(_far-_near);
	float bb = -_near * aa;

	memset(_result, 0, sizeof(float)*16);
	_result[0] = width;
	_result[5] = height;
	_result[10] = aa;
	_result[11] = 1.0f;
	_result[14] = bb;
}

inline void mtxOrtho(float* _result, float _left, float _right, float _bottom, float _top, float _near, float _far)
{
	const float aa = 2.0f/(_right - _left);
	const float bb = 2.0f/(_top - _bottom);
	const float cc = 1.0f/(_far - _near);
	const float dd = (_left + _right)/(_left - _right);
	const float ee = (_top + _bottom)/(_bottom - _top);
	const float ff = _near / (_near - _far);

	memset(_result, 0, sizeof(float)*16);
	_result[0] = aa;
	_result[5] = bb;
	_result[10] = cc;
	_result[12] = dd;
	_result[13] = ee;
	_result[14] = ff;
	_result[15] = 1.0f;
}

inline void mtxRotateX(float* _result, float _ax)
{
	float sx = sinf(_ax);
	float cx = cosf(_ax);

	memset(_result, 0, sizeof(float)*16);
	_result[ 0] = 1.0f;
	_result[ 5] = cx;
	_result[ 6] = -sx;
	_result[ 9] = sx;
	_result[10] = cx;
	_result[15] = 1.0f;
}

inline void mtxRotateY(float* _result, float _ay)
{
	float sy = sinf(_ay);
	float cy = cosf(_ay);

	memset(_result, 0, sizeof(float)*16);
	_result[ 0] = cy;
	_result[ 2] = sy;
	_result[ 5] = 1.0f;
	_result[ 8] = -sy;
	_result[10] = cy;
	_result[15] = 1.0f;
}

inline void mtxRotateZ(float* _result, float _az)
{
	float sz = sinf(_az);
	float cz = cosf(_az);

	memset(_result, 0, sizeof(float)*16);
	_result[ 0] = cz;
	_result[ 1] = -sz;
	_result[ 4] = sz;
	_result[ 5] = cz;
	_result[10] = 1.0f;
	_result[15] = 1.0f;
}

inline void mtxRotateXY(float* _result, float _ax, float _ay)
{
	float sx = sinf(_ax);
	float cx = cosf(_ax);
	float sy = sinf(_ay);
	float cy = cosf(_ay);

	memset(_result, 0, sizeof(float)*16);
	_result[ 0] = cy;
	_result[ 2] = -sy;
	_result[ 4] = -sx*sy;
	_result[ 5] = cx;
	_result[ 6] = -sx*cy;
	_result[ 8] = cx*sy;
	_result[ 9] = sx;
	_result[10] = cx*cy;
	_result[15] = 1.0f;
}

inline void mtxRotateXYZ(float* _result, float _ax, float _ay, float _az)
{
	float sx = sinf(_ax);
	float cx = cosf(_ax);
	float sy = sinf(_ay);
	float cy = cosf(_ay);
	float sz = sinf(_az);
	float cz = cosf(_az);

	memset(_result, 0, sizeof(float)*16);
	_result[ 0] = cy*cz;
	_result[ 1] = -cy*sz;
	_result[ 2] = sy;
	_result[ 4] = cz*sx*sy + cx*sz;
	_result[ 5] = cx*cz - sx*sy*sz;
	_result[ 6] = -cy*sx;
	_result[ 8] = -cx*cz*sy + sx*sz;
	_result[ 9] = cz*sx + cx*sy*sz;
	_result[10] = cx*cy;
	_result[15] = 1.0f;
}

inline void mtxRotateZYX(float* _result, float _ax, float _ay, float _az)
{
	float sx = sinf(_ax);
	float cx = cosf(_ax);
	float sy = sinf(_ay);
	float cy = cosf(_ay);
	float sz = sinf(_az);
	float cz = cosf(_az);

	memset(_result, 0, sizeof(float)*16);
	_result[ 0] = cy*cz;
	_result[ 1] = cz*sx*sy-cx*sz;
	_result[ 2] = cx*cz*sy+sx*sz;
	_result[ 4] = cy*sz;
	_result[ 5] = cx*cz + sx*sy*sz;
	_result[ 6] = -cz*sx + cx*sy*sz;
	_result[ 8] = -sy;
	_result[ 9] = cy*sx;
	_result[10] = cx*cy;
	_result[15] = 1.0f;
};

inline void vec3MulMtx(float* __restrict _result, const float* __restrict _vec, const float* __restrict _mat)
{
	_result[0] = _vec[0] * _mat[ 0] + _vec[1] * _mat[4] + _vec[2] * _mat[ 8] + _mat[12];
	_result[1] = _vec[0] * _mat[ 1] + _vec[1] * _mat[5] + _vec[2] * _mat[ 9] + _mat[13];
	_result[2] = _vec[0] * _mat[ 2] + _vec[1] * _mat[6] + _vec[2] * _mat[10] + _mat[14];
}

inline void vec3MulMtxH(float* __restrict _result, const float* __restrict _vec, const float* __restrict _mat)
{
	float xx = _vec[0] * _mat[ 0] + _vec[1] * _mat[4] + _vec[2] * _mat[ 8] + _mat[12];
	float yy = _vec[0] * _mat[ 1] + _vec[1] * _mat[5] + _vec[2] * _mat[ 9] + _mat[13];
	float zz = _vec[0] * _mat[ 2] + _vec[1] * _mat[6] + _vec[2] * _mat[10] + _mat[14];
	float ww = _vec[0] * _mat[ 3] + _vec[1] * _mat[7] + _vec[2] * _mat[11] + _mat[15];
	float invW = fsign(ww)/ww;
	_result[0] = xx*invW;
	_result[1] = yy*invW;
	_result[2] = zz*invW;
}

inline void vec4MulMtx(float* __restrict _result, const float* __restrict _vec, const float* __restrict _mat)
{
	_result[0] = _vec[0] * _mat[ 0] + _vec[1] * _mat[4] + _vec[2] * _mat[ 8] + _vec[3] * _mat[12];
	_result[1] = _vec[0] * _mat[ 1] + _vec[1] * _mat[5] + _vec[2] * _mat[ 9] + _vec[3] * _mat[13];
	_result[2] = _vec[0] * _mat[ 2] + _vec[1] * _mat[6] + _vec[2] * _mat[10] + _vec[3] * _mat[14];
	_result[3] = _vec[0] * _mat[ 3] + _vec[1] * _mat[7] + _vec[2] * _mat[11] + _vec[3] * _mat[15];
}

inline void mtxMul(float* __restrict _result, const float* __restrict _a, const float* __restrict _b)
{
	vec4MulMtx(&_result[ 0], &_a[ 0], _b);
	vec4MulMtx(&_result[ 4], &_a[ 4], _b);
	vec4MulMtx(&_result[ 8], &_a[ 8], _b);
	vec4MulMtx(&_result[12], &_a[12], _b);
}

inline void mtxTranspose(float* __restrict _result, const float* __restrict _a)
{
	_result[ 0] = _a[ 0];
	_result[ 4] = _a[ 1];
	_result[ 8] = _a[ 2];
	_result[12] = _a[ 3];
	_result[ 1] = _a[ 4];
	_result[ 5] = _a[ 5];
	_result[ 9] = _a[ 6];
	_result[13] = _a[ 7];
	_result[ 2] = _a[ 8];
	_result[ 6] = _a[ 9];
	_result[10] = _a[10];
	_result[14] = _a[11];
	_result[ 3] = _a[12];
	_result[ 7] = _a[13];
	_result[11] = _a[14];
	_result[15] = _a[15];
}

inline void mtxInverse(float* __restrict _result, const float* __restrict _a)
{
	float xx = _a[ 0];
	float xy = _a[ 1];
	float xz = _a[ 2];
	float xw = _a[ 3];
	float yx = _a[ 4];
	float yy = _a[ 5];
	float yz = _a[ 6];
	float yw = _a[ 7];
	float zx = _a[ 8];
	float zy = _a[ 9];
	float zz = _a[10];
	float zw = _a[11];
	float wx = _a[12];
	float wy = _a[13];
	float wz = _a[14];
	float ww = _a[15];

	float det = 0.0f;
	det += xx * (yy*(zz*ww - zw*wz) - yz*(zy*ww - zw*wy) + yw*(zy*wz - zz*wy) );
	det -= xy * (yx*(zz*ww - zw*wz) - yz*(zx*ww - zw*wx) + yw*(zx*wz - zz*wx) );
	det += xz * (yx*(zy*ww - zw*wy) - yy*(zx*ww - zw*wx) + yw*(zx*wy - zy*wx) );
	det -= xw * (yx*(zy*wz - zz*wy) - yy*(zx*wz - zz*wx) + yz*(zx*wy - zy*wx) );

	float invDet = 1.0f/det;

	_result[ 0] = +(yy*(zz*ww - wz*zw) - yz*(zy*ww - wy*zw) + yw*(zy*wz - wy*zz) ) * invDet;
	_result[ 1] = -(xy*(zz*ww - wz*zw) - xz*(zy*ww - wy*zw) + xw*(zy*wz - wy*zz) ) * invDet;
	_result[ 2] = +(xy*(yz*ww - wz*yw) - xz*(yy*ww - wy*yw) + xw*(yy*wz - wy*yz) ) * invDet;
	_result[ 3] = -(xy*(yz*zw - zz*yw) - xz*(yy*zw - zy*yw) + xw*(yy*zz - zy*yz) ) * invDet;
				  
	_result[ 4] = -(yx*(zz*ww - wz*zw) - yz*(zx*ww - wx*zw) + yw*(zx*wz - wx*zz) ) * invDet;
	_result[ 5] = +(xx*(zz*ww - wz*zw) - xz*(zx*ww - wx*zw) + xw*(zx*wz - wx*zz) ) * invDet;
	_result[ 6] = -(xx*(yz*ww - wz*yw) - xz*(yx*ww - wx*yw) + xw*(yx*wz - wx*yz) ) * invDet;
	_result[ 7] = +(xx*(yz*zw - zz*yw) - xz*(yx*zw - zx*yw) + xw*(yx*zz - zx*yz) ) * invDet;
				  
	_result[ 8] = +(yx*(zy*ww - wy*zw) - yy*(zx*ww - wx*zw) + yw*(zx*wy - wx*zy) ) * invDet;
	_result[ 9] = -(xx*(zy*ww - wy*zw) - xy*(zx*ww - wx*zw) + xw*(zx*wy - wx*zy) ) * invDet;
	_result[10] = +(xx*(yy*ww - wy*yw) - xy*(yx*ww - wx*yw) + xw*(yx*wy - wx*yy) ) * invDet;
	_result[11] = -(xx*(yy*zw - zy*yw) - xy*(yx*zw - zx*yw) + xw*(yx*zy - zx*yy) ) * invDet;
				  
	_result[12] = -(yx*(zy*wz - wy*zz) - yy*(zx*wz - wx*zz) + yz*(zx*wy - wx*zy) ) * invDet;
	_result[13] = +(xx*(zy*wz - wy*zz) - xy*(zx*wz - wx*zz) + xz*(zx*wy - wx*zy) ) * invDet;
	_result[14] = -(xx*(yy*wz - wy*yz) - xy*(yx*wz - wx*yz) + xz*(yx*wy - wx*yy) ) * invDet;
	_result[15] = +(xx*(yy*zz - zy*yz) - xy*(yx*zz - zx*yz) + xz*(yx*zy - zx*yy) ) * invDet;
}

/// Convert LH to RH projection matrix and vice versa.
inline void mtxProjFlipHandedness(float* __restrict _dst, const float* __restrict _src)
{
	_dst[ 0] = -_src[ 0];
	_dst[ 1] = -_src[ 1];
	_dst[ 2] = -_src[ 2];
	_dst[ 3] = -_src[ 3];
	_dst[ 4] =  _src[ 4];
	_dst[ 5] =  _src[ 5];
	_dst[ 6] =  _src[ 6];
	_dst[ 7] =  _src[ 7];
	_dst[ 8] = -_src[ 8];
	_dst[ 9] = -_src[ 9];
	_dst[10] = -_src[10];
	_dst[11] = -_src[11];
	_dst[12] =  _src[12];
	_dst[13] =  _src[13];
	_dst[14] =  _src[14];
	_dst[15] =  _src[15];
}

/// Convert LH to RH view matrix and vice versa.
inline void mtxViewFlipHandedness(float* __restrict _dst, const float* __restrict _src)
{
	_dst[ 0] = -_src[ 0];
	_dst[ 1] =  _src[ 1];
	_dst[ 2] = -_src[ 2];
	_dst[ 3] =  _src[ 3];
	_dst[ 4] = -_src[ 4];
	_dst[ 5] =  _src[ 5];
	_dst[ 6] = -_src[ 6];
	_dst[ 7] =  _src[ 7];
	_dst[ 8] = -_src[ 8];
	_dst[ 9] =  _src[ 9];
	_dst[10] = -_src[10];
	_dst[11] =  _src[11];
	_dst[12] = -_src[12];
	_dst[13] =  _src[13];
	_dst[14] = -_src[14];
	_dst[15] =  _src[15];
}

#endif // __FPU_MATH_H__