package net.minecraft.world.entity.schedule;

import com.google.common.collect.ImmutableList;
import com.google.common.collect.Lists;
import it.unimi.dsi.fastutil.ints.Int2ObjectAVLTreeMap;
import it.unimi.dsi.fastutil.ints.Int2ObjectSortedMap;
import java.util.Collection;
import java.util.List;

public class Timeline {
    private final List<Keyframe> keyframes = Lists.newArrayList();
    private int previousIndex;

    public ImmutableList<Keyframe> getKeyframes() {
        return ImmutableList.copyOf(this.keyframes);
    }

    public Timeline addKeyframe(int p_38061_, float p_38062_) {
        this.keyframes.add(new Keyframe(p_38061_, p_38062_));
        this.sortAndDeduplicateKeyframes();
        return this;
    }

    public Timeline addKeyframes(Collection<Keyframe> p_150248_) {
        this.keyframes.addAll(p_150248_);
        this.sortAndDeduplicateKeyframes();
        return this;
    }

    private void sortAndDeduplicateKeyframes() {
        Int2ObjectSortedMap<Keyframe> int2objectsortedmap = new Int2ObjectAVLTreeMap<>();
        this.keyframes.forEach(p_38065_ -> int2objectsortedmap.put(p_38065_.getTimeStamp(), p_38065_));
        this.keyframes.clear();
        this.keyframes.addAll(int2objectsortedmap.values());
        this.previousIndex = 0;
    }

    public float getValueAt(int p_38059_) {
        if (this.keyframes.size() <= 0) {
            return 0.0F;
        } else {
            Keyframe keyframe = this.keyframes.get(this.previousIndex);
            Keyframe keyframe1 = this.keyframes.get(this.keyframes.size() - 1);
            boolean flag = p_38059_ < keyframe.getTimeStamp();
            int i = flag ? 0 : this.previousIndex;
            float f = flag ? keyframe1.getValue() : keyframe.getValue();

            for (int j = i; j < this.keyframes.size(); j++) {
                Keyframe keyframe2 = this.keyframes.get(j);
                if (keyframe2.getTimeStamp() > p_38059_) {
                    break;
                }

                this.previousIndex = j;
                f = keyframe2.getValue();
            }

            return f;
        }
    }
}