class="kw">package com.hypixel.hytale.server.npc.corecomponents.entity.filters;

class="kw">import com.hypixel.hytale.component.Ref;
class="kw">import com.hypixel.hytale.component.Store;
class="kw">import com.hypixel.hytale.server.core.asset.type.attitude.Attitude;
class="kw">import com.hypixel.hytale.server.core.universe.world.storage.EntityStore;
class="kw">import com.hypixel.hytale.server.npc.asset.builder.BuilderSupport;
class="kw">import com.hypixel.hytale.server.npc.corecomponents.EntityFilterBase;
class="kw">import com.hypixel.hytale.server.npc.corecomponents.entity.filters.builders.BuilderEntityFilterAttitude;
class="kw">import com.hypixel.hytale.server.npc.instructions.ExecutionSupport;
class="kw">import java.util.Arrays;
class="kw">import java.util.EnumSet;
class="kw">import javax.annotation.Nonnull;

class="kw">public class EntityFilterAttitude class="kw">extends EntityFilterBase {
   class="kw">public class="kw">static class="kw">final String TYPE = "Attitude";
   class="kw">public class="kw">static class="kw">final int COST = 0;
   class="kw">protected class="kw">final EnumSet<Attitude> attitudes;

   class="kw">public EntityFilterAttitude(@Nonnull BuilderEntityFilterAttitude var1, @Nonnull BuilderSupport var2) {
      this.attitudes = var1.getAttitudes(var2);
   }

   class="kw">public EntityFilterAttitude(Attitude[] var1) {
      this.attitudes = EnumSet.noneOf(Attitude.class);
      this.attitudes.addAll(Arrays.asList(var1));
   }

   @Override
   class="kw">public boolean matchesEntity(
      @Nonnull Ref<EntityStore> var1, @Nonnull Ref<EntityStore> var2, @Nonnull ExecutionSupport var3, @Nonnull Store<EntityStore> var4
   ) {
      Attitude var5 = var3.getWorldSupport().getAttitude(var1, var2, var4);
      class="kw">return this.attitudes.contains(var5);
   }

   @Override
   class="kw">public int cost() {
      class="kw">return 0;
   }

   @Override
   class="kw">public void registerWithSupport(@Nonnull ExecutionSupport var1) {
      var1.getWorldSupport().requireAttitudeCache();
   }
}