package com.animationstudio.animation; import com.google.gson.*; import net.minecraft.nbt.NbtCompound; import net.minecraft.storage.ReadView; import net.minecraft.storage.WriteView; import net.minecraft.util.Identifier; import java.util.*; /** * Represents a complete animation with keyframes for multiple bones/model parts. * Supports serialization to/from JSON for export/import functionality. */ public class AnimationData { private final String id; private final String name; private final String description; private final float duration; // in seconds private final boolean looping; private final Map> boneKeyframes; private final long createdTimestamp; private final String author; public AnimationData(String id, String name, String description, float duration, boolean looping, Map> boneKeyframes, String author) { this.id = id; this.name = name; this.description = description; this.duration = duration; this.looping = looping; this.boneKeyframes = new HashMap<>(boneKeyframes); this.createdTimestamp = System.currentTimeMillis(); this.author = author; } public String getId() { return id; } public String getName() { return name; } public String getDescription() { return description; } public float getDuration() { return duration; } public boolean isLooping() { return looping; } public Map> getBoneKeyframes() { return boneKeyframes; } public long getCreatedTimestamp() { return createdTimestamp; } public String getAuthor() { return author; } /** * Get interpolated transform at a specific time. */ public BoneTransform getTransformAt(String boneName, float time) { List keyframes = boneKeyframes.get(boneName); if (keyframes == null || keyframes.isEmpty()) { return BoneTransform.IDENTITY; } float wrappedTime = looping ? time % duration : Math.min(time, duration); // Find surrounding keyframes AnimationKeyframe prev = keyframes.get(0); AnimationKeyframe next = keyframes.get(keyframes.size() - 1); for (int i = 0; i < keyframes.size(); i++) { AnimationKeyframe kf = keyframes.get(i); if (kf.getTime() <= wrappedTime) { prev = kf; } if (kf.getTime() >= wrappedTime) { next = kf; break; } } // If same keyframe or at exact time, return that transform if (prev == next || Math.abs(prev.getTime() - wrappedTime) < 0.001f) { return prev.getTransform(); } // Interpolate between keyframes float t = (wrappedTime - prev.getTime()) / (next.getTime() - prev.getTime()); return BoneTransform.lerp(prev.getTransform(), next.getTransform(), t); } /** * Serialize to JSON for export. */ public JsonObject toJson() { JsonObject json = new JsonObject(); json.addProperty("id", id); json.addProperty("name", name); json.addProperty("description", description); json.addProperty("duration", duration); json.addProperty("looping", looping); json.addProperty("created", createdTimestamp); json.addProperty("author", author); json.addProperty("format_version", "1.0"); JsonObject bones = new JsonObject(); for (Map.Entry> entry : boneKeyframes.entrySet()) { JsonArray keyframes = new JsonArray(); for (AnimationKeyframe kf : entry.getValue()) { keyframes.add(kf.toJson()); } bones.add(entry.getKey(), keyframes); } json.add("bones", bones); return json; } /** * Deserialize from JSON for import. */ public static AnimationData fromJson(JsonObject json) { String id = json.get("id").getAsString(); String name = json.get("name").getAsString(); String description = json.has("description") ? json.get("description").getAsString() : ""; float duration = json.get("duration").getAsFloat(); boolean looping = json.get("looping").getAsBoolean(); String author = json.has("author") ? json.get("author").getAsString() : "Unknown"; Map> boneKeyframes = new HashMap<>(); JsonObject bones = json.getAsJsonObject("bones"); for (Map.Entry entry : bones.entrySet()) { List keyframes = new ArrayList<>(); JsonArray array = entry.getValue().getAsJsonArray(); for (JsonElement elem : array) { keyframes.add(AnimationKeyframe.fromJson(elem.getAsJsonObject())); } boneKeyframes.put(entry.getKey(), keyframes); } return new AnimationData(id, name, description, duration, looping, boneKeyframes, author); } /** * Serialize to NBT for component storage. */ public void writeToView(WriteView view) { view.putString("id", id); view.putString("name", name); view.putString("description", description); view.putFloat("duration", duration); view.putBoolean("looping", looping); view.putLong("created", createdTimestamp); view.putString("author", author); } /** * Deserialize from NBT (reads minimal data - full animation loaded from library). */ public static AnimationData readFromView(ReadView view) { String id = view.getString("id", ""); String name = view.getString("name", "Unnamed"); String description = view.getString("description", ""); float duration = view.getFloat("duration", 1.0f); boolean looping = view.getBoolean("looping", false); String author = view.getString("author", "Unknown"); return new AnimationData(id, name, description, duration, looping, new HashMap<>(), author); } public static class Builder { private String id; private String name; private String description = ""; private float duration = 1.0f; private boolean looping = false; private Map> boneKeyframes = new HashMap<>(); private String author = "Unknown"; public Builder id(String id) { this.id = id; return this; } public Builder name(String name) { this.name = name; return this; } public Builder description(String desc) { this.description = desc; return this; } public Builder duration(float duration) { this.duration = duration; return this; } public Builder looping(boolean looping) { this.looping = looping; return this; } public Builder author(String author) { this.author = author; return this; } public Builder addKeyframe(String boneName, AnimationKeyframe keyframe) { boneKeyframes.computeIfAbsent(boneName, k -> new ArrayList<>()).add(keyframe); return this; } public AnimationData build() { // Sort keyframes by time boneKeyframes.values().forEach(list -> list.sort(Comparator.comparing(AnimationKeyframe::getTime))); return new AnimationData(id, name, description, duration, looping, boneKeyframes, author); } } }