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

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

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 = new Int2DoubleOpenHashMap();
   private int nextEventSlot;

   public EventSlotMapper(Class<EventType> classType, EventType[] types) {
      this.eventSets = new EnumMap<>(classType);
      this.eventSlotMappings = new EnumMap<>(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(EventType type, int set, double maxRange) {
      this.eventSets.computeIfAbsent(type, k -> new IntOpenHashSet()).add(set);
      Int2IntMap typeSlots = this.eventSlotMappings.computeIfAbsent(type, k -> {
         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);
      }

      double currentRange = this.eventSlotRanges.getOrDefault(slot, Double.MIN_VALUE);
      if (currentRange == Double.MIN_VALUE || currentRange < maxRange) {
         this.eventSlotRanges.put(slot, maxRange);
      }

      return slot;
   }
}