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

class="kw">import com.hypixel.hytale.component.Ref;
class="kw">import com.hypixel.hytale.component.Store;
class="kw">import com.hypixel.hytale.server.core.modules.blockset.BlockSetModule;
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.SensorBase;
class="kw">import com.hypixel.hytale.server.npc.corecomponents.combat.builders.BuilderSensorChargeBlockCollisions;
class="kw">import com.hypixel.hytale.server.npc.instructions.ExecutionSupport;
class="kw">import com.hypixel.hytale.server.npc.instructions.Instruction;
class="kw">import com.hypixel.hytale.server.npc.movement.controllers.BlockHit;
class="kw">import com.hypixel.hytale.server.npc.sensorinfo.BlockCollisionProvider;
class="kw">import com.hypixel.hytale.server.npc.sensorinfo.InfoProvider;
class="kw">import it.unimi.dsi.fastutil.objects.ObjectArrayList;
class="kw">import java.util.List;
class="kw">import javax.annotation.Nonnull;
class="kw">import javax.annotation.Nullable;

class="kw">public class SensorChargeBlockCollisions class="kw">extends SensorBase {
   class="kw">private class="kw">final BlockCollisionProvider blockCollisionProvider = new BlockCollisionProvider();
   class="kw">private class="kw">final List<BlockHit> hits = new ObjectArrayList();
   class="kw">private class="kw">final int blockFilterSet;
   @Nullable
   class="kw">private BodyMotionCharge matchingChargeBodyMotion;

   class="kw">public SensorChargeBlockCollisions(@Nonnull BuilderSensorChargeBlockCollisions var1, @Nonnull BuilderSupport var2) {
      super(var1);
      this.blockFilterSet = var1.getBlockFilterSet(var2);
   }

   @Override
   class="kw">public void loaded(ExecutionSupport var1) {
      if (this.parent class="kw">instanceof Instruction var2) {
         this.matchingChargeBodyMotion = var2.findNearestPrecedingBodyMotion(BodyMotionCharge.class);
      }
   }

   @Override
   class="kw">public boolean matches(@Nonnull Ref<EntityStore> var1, @Nonnull ExecutionSupport var2, double var3, @Nonnull Store<EntityStore> var5) {
      this.hits.clear();
      this.blockCollisionProvider.clear();
      if (super.matches(var1, var2, var3, var5) && this.matchingChargeBodyMotion != null && this.matchingChargeBodyMotion.getBlockHitCount() != 0) {
         int var6 = this.matchingChargeBodyMotion.getBlockHitCount();

         for (int var7 = 0; var7 < var6; var7++) {
            BlockHit var8 = this.matchingChargeBodyMotion.getBlockHit(var7);
            if (this.blockFilterSet == Integer.MIN_VALUE || BlockSetModule.getInstance().blockInSet(this.blockFilterSet, var8.blockTypeId)) {
               this.hits.add(var8);
            }
         }

         class="kw">return this.blockCollisionProvider.populate(this.hits, this.hits.size());
      } else {
         class="kw">return false;
      }
   }

   @Override
   class="kw">public InfoProvider getSensorInfo() {
      class="kw">return this.blockCollisionProvider;
   }

   @Override
   class="kw">public void done() {
      super.done();
      this.blockCollisionProvider.clear();
      this.hits.clear();
   }
}