VectorSphereUtil.java

package com.hypixel.hytale.math.vector;

import com.hypixel.fastutil.FastCollection;
import com.hypixel.hytale.function.consumer.IntBiObjectConsumer;
import com.hypixel.hytale.function.consumer.IntObjectConsumer;
import com.hypixel.hytale.function.consumer.IntTriObjectConsumer;
import com.hypixel.hytale.function.consumer.TriConsumer;
import it.unimi.dsi.fastutil.ints.Int2ObjectMap;
import java.util.function.BiConsumer;
import java.util.function.Consumer;
import java.util.function.Function;
import javax.annotation.Nonnull;

public class VectorSphereUtil {
   public VectorSphereUtil() {
   }

   public static void forEachVector(Iterable<Vector3d> vectors, double originX, double originY, double originZ, double radius, Consumer<Vector3d> consumer) {
      forEachVector(vectors, originX, originY, originZ, radius, radius, radius, consumer);
   }

   public static void forEachVector(Iterable<Vector3d> vectors, double originX, double originY, double originZ, double radiusX, double radiusY, double radiusZ, Consumer<Vector3d> consumer) {
      forEachVector(vectors, Function.identity(), originX, originY, originZ, radiusX, radiusY, radiusZ, consumer);
   }

   public static <T> void forEachVector(Iterable<T> input, @Nonnull Function<T, Vector3d> func, double originX, double originY, double originZ, double radius, Consumer<T> consumer) {
      forEachVector(input, func, originX, originY, originZ, radius, radius, radius, consumer);
   }

   public static <T> void forEachVector(Iterable<T> input, @Nonnull Function<T, Vector3d> func, double originX, double originY, double originZ, double radiusX, double radiusY, double radiusZ, Consumer<T> consumer) {
      forEachVector(input, func, originX, originY, originZ, radiusX, radiusY, radiusZ, (TriConsumer)((t, c, n0) -> c.accept(t)), consumer, (Object)null);
   }

   public static <T, V> void forEachVector(Iterable<T> input, @Nonnull Function<T, Vector3d> func, double originX, double originY, double originZ, double radius, BiConsumer<T, V> consumer, V objV) {
      forEachVector(input, func, originX, originY, originZ, radius, radius, radius, consumer, objV);
   }

   public static <T, V> void forEachVector(Iterable<T> input, @Nonnull Function<T, Vector3d> func, double originX, double originY, double originZ, double radiusX, double radiusY, double radiusZ, BiConsumer<T, V> consumer, V objV) {
      forEachVector(input, func, originX, originY, originZ, radiusX, radiusY, radiusZ, (TriConsumer)((t, c, objV2) -> c.accept(t, objV2)), consumer, objV);
   }

   public static <T, V1, V2> void forEachVector(Iterable<T> input, @Nonnull Function<T, Vector3d> func, double originX, double originY, double originZ, double radius, @Nonnull TriConsumer<T, V1, V2> consumer, V1 objV1, V2 objV2) {
      forEachVector(input, func, originX, originY, originZ, radius, radius, radius, consumer, objV1, objV2);
   }

   public static <T, V1, V2> void forEachVector(Iterable<T> input, @Nonnull Function<T, Vector3d> func, double originX, double originY, double originZ, double radiusX, double radiusY, double radiusZ, @Nonnull TriConsumer<T, V1, V2> consumer, V1 objV1, V2 objV2) {
      if (input instanceof FastCollection<T> fastCollection) {
         fastCollection.forEach((objx, _func, _originX, _originY, _originZ, _radiusX, _radiusY, _radiusZ, _consumer, _objV1, _objV2) -> {
            Vector3d vector = (Vector3d)_func.apply(objx);
            if (isInside(_originX, _originY, _originZ, _radiusX, _radiusY, _radiusZ, vector)) {
               _consumer.accept(objx, _objV1, _objV2);
            }

         }, func, originX, originY, originZ, radiusX, radiusY, radiusZ, consumer, objV1, objV2);
      } else {
         for(T obj : input) {
            Vector3d vector = (Vector3d)func.apply(obj);
            if (isInside(originX, originY, originZ, radiusX, radiusY, radiusZ, vector)) {
               consumer.accept(obj, objV1, objV2);
            }
         }
      }

   }

   public static <T> void forEachVector(@Nonnull Int2ObjectMap<T> input, @Nonnull Function<T, Vector3d> func, double originX, double originY, double originZ, double radius, IntObjectConsumer<T> consumer) {
      forEachVector(input, func, originX, originY, originZ, radius, radius, radius, consumer);
   }

   public static <T> void forEachVector(@Nonnull Int2ObjectMap<T> input, @Nonnull Function<T, Vector3d> func, double originX, double originY, double originZ, double radiusX, double radiusY, double radiusZ, IntObjectConsumer<T> consumer) {
      forEachVector(input, func, originX, originY, originZ, radiusX, radiusY, radiusZ, (IntTriObjectConsumer)((i, t, c, n0) -> c.accept(i, t)), consumer, (Object)null);
   }

   public static <T, V> void forEachVector(@Nonnull Int2ObjectMap<T> input, @Nonnull Function<T, Vector3d> func, double originX, double originY, double originZ, double radius, IntBiObjectConsumer<T, V> consumer, V objV) {
      forEachVector(input, func, originX, originY, originZ, radius, radius, radius, consumer, objV);
   }

   public static <T, V> void forEachVector(@Nonnull Int2ObjectMap<T> input, @Nonnull Function<T, Vector3d> func, double originX, double originY, double originZ, double radiusX, double radiusY, double radiusZ, IntBiObjectConsumer<T, V> consumer, V objV) {
      forEachVector(input, func, originX, originY, originZ, radiusX, radiusY, radiusZ, (IntTriObjectConsumer)((i, t, objV1, c) -> c.accept(i, t, objV1)), objV, consumer);
   }

   public static <T, V1, V2> void forEachVector(@Nonnull Int2ObjectMap<T> input, @Nonnull Function<T, Vector3d> func, double originX, double originY, double originZ, double radius, @Nonnull IntTriObjectConsumer<T, V1, V2> consumer, V1 objV1, V2 objV2) {
      forEachVector(input, func, originX, originY, originZ, radius, radius, radius, consumer, objV1, objV2);
   }

   public static <T, V1, V2> void forEachVector(@Nonnull Int2ObjectMap<T> input, @Nonnull Function<T, Vector3d> func, double originX, double originY, double originZ, double radiusX, double radiusY, double radiusZ, @Nonnull IntTriObjectConsumer<T, V1, V2> consumer, V1 objV1, V2 objV2) {
      for(Int2ObjectMap.Entry<T> next : input.int2ObjectEntrySet()) {
         int key = next.getIntKey();
         T value = (T)next.getValue();
         Vector3d vector = (Vector3d)func.apply(value);
         if (isInside(originX, originY, originZ, radiusX, radiusY, radiusZ, vector)) {
            consumer.accept(key, value, objV1, objV2);
         }
      }

   }

   public static boolean isInside(double originX, double originY, double originZ, double radius, @Nonnull Vector3d vector) {
      return isInside(originX, originY, originZ, radius, radius, radius, vector);
   }

   public static boolean isInside(double originX, double originY, double originZ, double radiusX, double radiusY, double radiusZ, @Nonnull Vector3d vector) {
      double x = vector.getX() - originX;
      double y = vector.getY() - originY;
      double z = vector.getZ() - originZ;
      double xRatio = x / radiusX;
      double yRatio = y / radiusY;
      double zRatio = z / radiusZ;
      return xRatio * xRatio + yRatio * yRatio + zRatio * zRatio <= 1.0;
   }
}