CustomNPC-Plus/src/main/java/noppes/npcs/controllers/data/Frame.java

package noppes.npcs.controllers.data;

import net.minecraft.nbt.NBTTagCompound;
import net.minecraft.nbt.NBTTagList;
import noppes.npcs.api.handler.data.IFrame;
import noppes.npcs.api.handler.data.IFramePart;
import noppes.npcs.constants.EnumAnimationPart;

import java.util.HashMap;
import java.util.Map;


public class Frame implements IFrame {
	public Animation parent;
	public HashMap<EnumAnimationPart,FramePart> frameParts = new HashMap<>();
	public int duration = 0;
	boolean customized = false;
	public float speed = 1.0F;
	public byte smooth = 0;
	public boolean renderTicks = false; // If true, MC ticks are used. If false, render ticks are used.

	private int colorMarker = 0xFFFFFF;

	public Frame(){}

	public Frame(int duration) {
		this.duration = duration;
	}

	public Frame(int duration, float speed, byte smooth) {
		this.duration = duration;
		this.speed = speed;
		this.smooth = smooth;
		this.customized = true;
	}

	public HashMap<EnumAnimationPart,FramePart> getFrameMap() {
		return frameParts;
	}

	public IFramePart[] getParts() {
		return frameParts.values().toArray(new IFramePart[0]);
	}

	public IFrame addPart(IFramePart partConfig) {
		this.frameParts.put(((FramePart)partConfig).getPart(),(FramePart) partConfig);
		return this;
	}

	public IFrame removePart(String partName) {
		try {
			this.frameParts.remove(EnumAnimationPart.valueOf(partName));
		} catch (IllegalArgumentException ignored) {}
		return this;
	}

	public IFrame removePart(int partId) {
		for (EnumAnimationPart part : EnumAnimationPart.values()) {
			this.frameParts.remove(part);
		}
		return this;
	}

	public IFrame clearParts() {
		frameParts.clear();
		return this;
	}

	public int getDuration() {
		return duration;
	}

	public IFrame setDuration(int duration) {
		this.duration = duration;
		return this;
	}

	public boolean isCustomized() {
		return customized;
	}

	public IFrame setCustomized(boolean customized) {
		this.customized = customized;
		return this;
	}

	public float getSpeed() {
		return speed;
	}

	public IFrame setSpeed(float speed) {
		this.speed = speed;
		return this;
	}

	public byte smoothType() {
		return smooth;
	}

	public IFrame setSmooth(byte smooth) {
		this.smooth = smooth;
		return this;
	}

	public IFrame useRenderTicks(boolean renderTicks) {
		this.renderTicks = renderTicks;
		return this;
	}

	public boolean useRenderTicks() {
		return this.renderTicks;
	}

	public int getColorMarker() {
		return this.colorMarker;
	}

	public void setColorMarker(int color) {
		this.colorMarker = color;
	}

	public void readFromNBT(NBTTagCompound compound){
		duration = compound.getInteger("Duration");
		if (compound.hasKey("ColorMarker")) {
			this.setColorMarker(compound.getInteger("ColorMarker"));
		}

		// Customized = TRUE if Speed or Smooth Exist
		if(compound.hasKey("Speed")){
			customized = true;
			speed = compound.getFloat("Speed");
		}
		if(compound.hasKey("Smooth")){
			customized = true;
			smooth = compound.getByte("Smooth");
		}
		if(compound.hasKey("RenderTicks")){
			customized = true;
			renderTicks = compound.getBoolean("RenderTicks");
		}

		if (!customized) {
			this.speed = parent.speed;
			this.smooth = parent.smooth;
			this.renderTicks = parent.renderTicks;
		}

		HashMap<EnumAnimationPart,FramePart> frameParts = new HashMap<>();
		NBTTagList list = compound.getTagList("FrameParts", 10);
		for (int i = 0; i < list.tagCount(); i++) {
			NBTTagCompound item = list.getCompoundTagAt(i);
			FramePart framePart = new FramePart();
			framePart.readFromNBT(item);
			if (!framePart.customized) {
				framePart.smooth = this.smooth;
				framePart.speed = this.speed;
			}
			frameParts.put(framePart.part,framePart);
		}
		this.frameParts = frameParts;
	}

	public NBTTagCompound writeToNBT(){
		NBTTagCompound compound = new NBTTagCompound();
		compound.setInteger("Duration", duration);
		compound.setInteger("ColorMarker", this.colorMarker);

		if(customized){
			compound.setFloat("Speed", speed);
			compound.setByte("Smooth", smooth);
			compound.setBoolean("RenderTicks", renderTicks);
		}

		NBTTagList list = new NBTTagList();
		for(FramePart framePart : frameParts.values()){
			NBTTagCompound item = framePart.writeToNBT();
			list.appendTag(item);
		}
		compound.setTag("FrameParts", list);
		return compound;
	}

	public Frame copy() {
		Frame frame = new Frame(this.duration);
		HashMap<EnumAnimationPart,FramePart> frameParts = this.frameParts;
		for (Map.Entry<EnumAnimationPart,FramePart> entry : frameParts.entrySet()) {
			frame.frameParts.put(entry.getKey(),entry.getValue().copy());
		}
		frame.parent = this.parent;
		frame.duration = this.duration;
		frame.customized = this.customized;
		frame.speed = this.speed;
		frame.smooth = this.smooth;
		frame.renderTicks = this.renderTicks;
		frame.colorMarker = this.colorMarker;
		return frame;
	}
}