class="kw">package com.hypixel.hytale.builtin.encountermanager;

class="kw">import com.hypixel.hytale.builtin.audio.components.AudioStateComponent;
class="kw">import com.hypixel.hytale.builtin.audio.components.ForcedMusicTracker;
class="kw">import com.hypixel.hytale.component.Component;
class="kw">import com.hypixel.hytale.component.ComponentAccessor;
class="kw">import com.hypixel.hytale.component.ComponentType;
class="kw">import com.hypixel.hytale.component.Ref;
class="kw">import com.hypixel.hytale.protocol.StateTransition;
class="kw">import com.hypixel.hytale.server.core.asset.type.audiostate.config.AudioState;
class="kw">import com.hypixel.hytale.server.core.universe.world.storage.EntityStore;
class="kw">import it.unimi.dsi.fastutil.ints.Int2IntOpenHashMap;
class="kw">import it.unimi.dsi.fastutil.ints.Int2ObjectOpenHashMap;
class="kw">import it.unimi.dsi.fastutil.ints.Int2IntMap.Entry;
class="kw">import it.unimi.dsi.fastutil.objects.ObjectIterator;
class="kw">import javax.annotation.Nonnull;
class="kw">import javax.annotation.Nullable;

class="kw">public class EncounterAudioState class="kw">implements Component<EntityStore>, EncounterMemberConsumer {
   class="kw">private int desiredMusicIndex;
   @Nonnull
   class="kw">private class="kw">final Int2IntOpenHashMap desiredAxisValues = new Int2IntOpenHashMap();
   @Nonnull
   class="kw">private class="kw">final Int2ObjectOpenHashMap<StateTransition> axisTransitions = new Int2ObjectOpenHashMap();

   class="kw">public EncounterAudioState() {
   }

   @Nonnull
   class="kw">public class="kw">static ComponentType<EntityStore, EncounterAudioState> getComponentType() {
      class="kw">return EncounterManagerPlugin.get().getEncounterAudioStateComponentType();
   }

   class="kw">public void setDesiredMusicIndex(int var1) {
      this.desiredMusicIndex = var1;
   }

   class="kw">public int getDesiredMusicIndex() {
      class="kw">return this.desiredMusicIndex;
   }

   class="kw">public void setDesiredAxisValue(int var1, int var2, @Nullable StateTransition var3) {
      this.desiredAxisValues.put(var1, var2);
      if (var3 != null) {
         this.axisTransitions.put(var1, var3);
      } else {
         this.axisTransitions.remove(var1);
      }
   }

   class="kw">public void resetDesiredToNeutral() {
      this.desiredMusicIndex = 0;
      ObjectIterator var1 = this.desiredAxisValues.int2IntEntrySet().iterator();

      while (var1.hasNext()) {
         Entry var2 = (Entry)var1.next();
         var2.setValue(defaultValueFor(var2.getIntKey()));
      }

      this.axisTransitions.clear();
   }

   @Override
   class="kw">public void applyToPlayer(@Nonnull ComponentAccessor<EntityStore> var1, @Nonnull Ref<EntityStore> var2) {
      ForcedMusicTracker var3 = var1.getComponent(var2, ForcedMusicTracker.getComponentType());
      if (var3 != null) {
         var3.setCurrentContainerIndex(this.desiredMusicIndex);
      }

      if (!this.desiredAxisValues.isEmpty()) {
         AudioStateComponent var4 = var1.getComponent(var2, AudioStateComponent.getComponentType());
         if (var4 != null) {
            ObjectIterator var5 = this.desiredAxisValues.int2IntEntrySet().iterator();

            while (var5.hasNext()) {
               Entry var6 = (Entry)var5.next();
               var4.setValue(var6.getIntKey(), var6.getIntValue(), (StateTransition)this.axisTransitions.get(var6.getIntKey()));
            }
         }
      }
   }

   @Override
   class="kw">public void revertPlayer(@Nonnull ComponentAccessor<EntityStore> var1, @Nonnull Ref<EntityStore> var2) {
      ForcedMusicTracker var3 = var1.getComponent(var2, ForcedMusicTracker.getComponentType());
      if (var3 != null) {
         var3.setCurrentContainerIndex(0);
      }

      if (!this.desiredAxisValues.isEmpty()) {
         AudioStateComponent var4 = var1.getComponent(var2, AudioStateComponent.getComponentType());
         if (var4 != null) {
            ObjectIterator var5 = this.desiredAxisValues.int2IntEntrySet().iterator();

            while (var5.hasNext()) {
               Entry var6 = (Entry)var5.next();
               var4.setValue(var6.getIntKey(), defaultValueFor(var6.getIntKey()), null);
            }
         }
      }
   }

   class="kw">private class="kw">static int defaultValueFor(int var0) {
      AudioState var1 = AudioState.getAssetMap().getAsset(var0);
      class="kw">return var1 != null ? var1.getDefaultValueIndex() : 0;
   }

   @Nonnull
   @Override
   class="kw">public Component<EntityStore> clone() {
      class="kw">return new EncounterAudioState();
   }
}