// 
// Decompiled by Procyon v0.6.0
// 

package com.hypixel.hytale.server.npc.asset.builder;

import it.unimi.dsi.fastutil.ints.Int2IntOpenHashMap;
import it.unimi.dsi.fastutil.ints.IntOpenHashSet;
import java.util.EnumMap;
import it.unimi.dsi.fastutil.ints.Int2DoubleOpenHashMap;
import it.unimi.dsi.fastutil.ints.Int2DoubleMap;
import it.unimi.dsi.fastutil.ints.Int2IntMap;
import javax.annotation.Nonnull;
import it.unimi.dsi.fastutil.ints.IntSet;
import java.util.Map;

public class EventSlotMapper<EventType extends Enum<EventType>>
{
    @Nonnull
    private final Map<EventType, IntSet> eventSets;
    @Nonnull
    private final Map<EventType, Int2IntMap> eventSlotMappings;
    private final Int2DoubleMap eventSlotRanges;
    private int nextEventSlot;
    
    public EventSlotMapper(final Class<EventType> classType, final EventType[] types) {
        this.eventSlotRanges = new Int2DoubleOpenHashMap();
        this.eventSets = new EnumMap<EventType, IntSet>(classType);
        this.eventSlotMappings = new EnumMap<EventType, Int2IntMap>(classType);
    }
    
    @Nonnull
    public Map<EventType, IntSet> getEventSets() {
        return this.eventSets;
    }
    
    @Nonnull
    public Map<EventType, Int2IntMap> getEventSlotMappings() {
        return this.eventSlotMappings;
    }
    
    @Nonnull
    public Int2DoubleMap getEventSlotRanges() {
        return this.eventSlotRanges;
    }
    
    public int getEventSlotCount() {
        return this.nextEventSlot;
    }
    
    public int getEventSlot(final EventType type, final int set, final double maxRange) {
        this.eventSets.computeIfAbsent(type, k -> new IntOpenHashSet()).add(set);
        final Int2IntMap typeSlots = this.eventSlotMappings.computeIfAbsent(type, k -> {
            final Int2IntOpenHashMap m = new Int2IntOpenHashMap();
            m.defaultReturnValue(Integer.MIN_VALUE);
            return m;
        });
        int slot = typeSlots.get(set);
        if (slot == Integer.MIN_VALUE) {
            slot = this.nextEventSlot++;
            typeSlots.put(set, slot);
        }
        final double currentRange = this.eventSlotRanges.getOrDefault(slot, Double.MIN_VALUE);
        if (currentRange == Double.MIN_VALUE || currentRange < maxRange) {
            this.eventSlotRanges.put(slot, maxRange);
        }
        return slot;
    }
}