Hbm-s-Nuclear-Tech-GIT/src/main/java/com/hbm/render/anim/BusAnimationSequence.java

package com.hbm.render.anim;

import java.util.ArrayList;
import java.util.List;

import com.hbm.render.anim.BusAnimationKeyframe.InterpolationType;

//the actual bus, a sequence of keyframes with their own behavior and such
public class BusAnimationSequence {

	public static enum Dimension {
		TX,
		TY,
		TZ,
		RX,
		RY,
		RZ,
		SX,
		SY,
		SZ
	}


	// Storing a matrix of keyframe data, each keyframe stores a SINGLE dimension, so we can stagger frames over each parameter
	private List<List<BusAnimationKeyframe>> transformKeyframes = new ArrayList<List<BusAnimationKeyframe>>(9);

	public double[] offset = new double[3];


	public BusAnimationSequence() {
		// Initialise our keyframe storage, since it's multidimensional
		for (int i = 0; i < 9; i++) {
			transformKeyframes.add(new ArrayList<BusAnimationKeyframe>());
		}
	}


	
	// Adds a keyframe to the given dimension
	public BusAnimationSequence addKeyframe(Dimension dimension, BusAnimationKeyframe keyframe) {
		transformKeyframes.get(dimension.ordinal()).add(keyframe);
		
		return this;
	}

	public BusAnimationSequence addKeyframe(Dimension dimension, double value, int duration) {
		return addKeyframe(dimension, new BusAnimationKeyframe(value, duration));
	}


	// Two helper methods for the old hard-coded animations
	public BusAnimationSequence addKeyframePosition(double x, double y, double z, int duration) {
		addKeyframe(Dimension.TX, new BusAnimationKeyframe(x, duration));
		addKeyframe(Dimension.TY, new BusAnimationKeyframe(y, duration));
		addKeyframe(Dimension.TZ, new BusAnimationKeyframe(z, duration));

		return this;
	}

	public BusAnimationSequence addKeyframeRotation(double x, double y, double z, int duration) {
		addKeyframe(Dimension.RX, new BusAnimationKeyframe(x, duration));
		addKeyframe(Dimension.RY, new BusAnimationKeyframe(y, duration));
		addKeyframe(Dimension.RZ, new BusAnimationKeyframe(z, duration));

		return this;
	}
	
	//all transformation data is absolute, additive transformations have not yet been implemented
	public double[] getTransformation(int millis) {
		double[] transform = new double[12];

		for (int i = 0; i < 9; i++) {
			List<BusAnimationKeyframe> keyframes = transformKeyframes.get(i);

			BusAnimationKeyframe currentFrame = null;
			BusAnimationKeyframe previousFrame = null;

			int startTime = 0;
			int endTime = 0;
			for (BusAnimationKeyframe keyframe: keyframes) {
				startTime = endTime;
				endTime += keyframe.duration;
				previousFrame = currentFrame;
				currentFrame = keyframe;
				if (millis < endTime) break;
			}

			if (currentFrame == null) {
				// Scale defaults to 1, others are 0
				transform[i] = i >= 6 ? 1 : 0;
				continue;
			}

			if (currentFrame.interpolationType == InterpolationType.NONE || millis >= endTime) {
				transform[i] = currentFrame.value;
				continue;
			}

			double a = currentFrame.value;
			double b = previousFrame != null ? previousFrame.value : 0;
			double t = (double)(millis - startTime) / (double)currentFrame.duration;

			transform[i] = (a - b) * t + b;
		}

		transform[9] = offset[0];
		transform[10] = offset[1];
		transform[11] = offset[2];

		return transform;
	}
	
	public int getTotalTime() {
		int highestTime = 0;
		
		for (List<BusAnimationKeyframe> keyframes: transformKeyframes) {
			int time = 0;
			for (BusAnimationKeyframe frame: keyframes) {
				time += frame.duration;
			}

			highestTime = Math.max(time, highestTime);
		}
		
		return highestTime;
	}
}