package com.hbm.animloader;

import java.nio.FloatBuffer;

import org.lwjgl.util.vector.Matrix4f;
import org.lwjgl.util.vector.Quaternion;

import com.hbm.util.BobMathUtil;

import net.minecraft.client.renderer.GLAllocation;
import net.minecraft.util.Vec3;

public class Transform {

	protected static FloatBuffer auxGLMatrix = GLAllocation.createDirectFloatBuffer(16);
	
	Vec3 scale;
	Vec3 translation;
	Quaternion rotation;
	
	boolean hidden = false;
	
	public Transform(float[] matrix){
		scale = getScaleFromMatrix(matrix);
		auxGLMatrix.put(matrix);
		auxGLMatrix.rewind();
		rotation = new Quaternion().setFromMatrix((Matrix4f) new Matrix4f().load(auxGLMatrix));
		translation = Vec3.createVectorHelper(matrix[0*4+3], matrix[1*4+3], matrix[2*4+3]);
		auxGLMatrix.rewind();
	}
	
	private Vec3 getScaleFromMatrix(float[] matrix){
		float scaleX = (float) Vec3.createVectorHelper(matrix[0], matrix[1], matrix[2]).lengthVector();
		float scaleY = (float) Vec3.createVectorHelper(matrix[4], matrix[5], matrix[6]).lengthVector();
		float scaleZ = (float) Vec3.createVectorHelper(matrix[8], matrix[9], matrix[10]).lengthVector();
		
		matrix[0] = matrix[0]/scaleX;
		matrix[1] = matrix[1]/scaleX;
		matrix[2] = matrix[2]/scaleX;
		
		matrix[4] = matrix[4]/scaleY;
		matrix[5] = matrix[5]/scaleY;
		matrix[6] = matrix[6]/scaleY;
		
		matrix[8] = matrix[8]/scaleZ;
		matrix[9] = matrix[9]/scaleZ;
		matrix[10] = matrix[10]/scaleZ;
		return Vec3.createVectorHelper(scaleX, scaleY, scaleZ);
	}
	
	public void interpolateAndApply(Transform other, float inter){
		Vec3 trans = BobMathUtil.interpVec(this.translation, other.translation, inter); //ORIGINAL: translation.interpolate(other.translation, inter);
		Vec3 scale = BobMathUtil.interpVec(this.scale, other.scale, inter); //ORIGINAL: this.scale.interpolate(other.scale, inter);
		Quaternion rot = slerp(rotation, other.rotation, inter);
		//GlStateManager.quatToGlMatrix(auxGLMatrix, rot); TODO: find implementation
		scale(auxGLMatrix, scale);
		auxGLMatrix.put(12, (float) trans.xCoord);
		auxGLMatrix.put(13, (float) trans.yCoord);
		auxGLMatrix.put(14, (float) trans.zCoord);
		
		//for(int i = 0; i < 16; i ++){
			//System.out.print(auxGLMatrix.get(i) + " ");
		//}
		//System.out.println();
		//GlStateManager.multMatrix(auxGLMatrix); TODO: find implementation
	}
	
	private void scale(FloatBuffer matrix, Vec3 scale){
		matrix.put(0, (float) (matrix.get(0)*scale.xCoord));
		matrix.put(4, (float) (matrix.get(4)*scale.xCoord));
		matrix.put(8, (float) (matrix.get(8)*scale.xCoord));
		matrix.put(12, (float) (matrix.get(12)*scale.xCoord));
		
		matrix.put(1, (float) (matrix.get(1)*scale.yCoord));
		matrix.put(5, (float) (matrix.get(5)*scale.yCoord));
		matrix.put(9, (float) (matrix.get(9)*scale.yCoord));
		matrix.put(13, (float) (matrix.get(13)*scale.yCoord));
		
		matrix.put(2, (float) (matrix.get(2)*scale.zCoord));
		matrix.put(6, (float) (matrix.get(6)*scale.zCoord));
		matrix.put(10, (float) (matrix.get(10)*scale.zCoord));
		matrix.put(14, (float) (matrix.get(14)*scale.zCoord));
	}
	
	//Thanks, wikipedia
	//God, I wish java had operator overloads. Those are one of my favorite things about c and glsl.
	protected Quaternion slerp(Quaternion v0, Quaternion v1, float t) {
		// Only unit quaternions are valid rotations.
	    // Normalize to avoid undefined behavior.
		//Drillgon200: Any quaternions loaded from blender should be normalized already
	    //v0.normalise();
	    //v1.normalise();

	    // Compute the cosine of the angle between the two vectors.
	    double dot = Quaternion.dot(v0, v1);

	    // If the dot product is negative, slerp won't take
	    // the shorter path. Note that v1 and -v1 are equivalent when
	    // the negation is applied to all four components. Fix by 
	    // reversing one quaternion.
	    if (dot < 0.0f) {
	        v1 = new Quaternion(-v1.x, -v1.y, -v1.z, -v1.w);
	        dot = -dot;
	    }

	    final double DOT_THRESHOLD = 0.9999999;
	    if (dot > DOT_THRESHOLD) {
	        // If the inputs are too close for comfort, linearly interpolate
	        // and normalize the result.
	        Quaternion result = new Quaternion(v0.x + t*v1.x, 
	        								v0.y + t*v1.y, 
	        								v0.z + t*v1.z, 
	        								v0.w + t*v1.w);
	        result.normalise();
	        return result;
	    }

	    // Since dot is in range [0, DOT_THRESHOLD], acos is safe
	    double theta_0 = Math.acos(dot);        // theta_0 = angle between input vectors
	    double theta = theta_0*t;          // theta = angle between v0 and result
	    double sin_theta = Math.sin(theta);     // compute this value only once
	    double sin_theta_0 = Math.sin(theta_0); // compute this value only once

	    float s0 = (float) (Math.cos(theta) - dot * sin_theta / sin_theta_0);  // == sin(theta_0 - theta) / sin(theta_0)
	    float s1 = (float) (sin_theta / sin_theta_0);

	    return new Quaternion(s0*v0.x + s1*v1.x, 
	    					s0*v0.y + s1*v1.y, 
	    					s0*v0.z + s1*v1.z, 
	    					s0*v0.w + s1*v1.w);
	}
	
}