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

class="kw">import com.hypixel.hytale.builtin.hytalegenerator.VectorUtil;
class="kw">import javax.annotation.Nonnull;
class="kw">import javax.annotation.Nullable;
class="kw">import org.joml.Vector3d;

class="kw">public class RotatorDensity class="kw">extends Density {
   @Nonnull
   class="kw">private class="kw">static class="kw">final Vector3d Y_AXIS = new Vector3d(0.0, 1.0, 0.0);
   @Nullable
   class="kw">private Density input;
   class="kw">private Vector3d rotationAxis;
   class="kw">private Vector3d tiltAxis;
   class="kw">private double tiltAngle;
   class="kw">private class="kw">final double spinAngle;
   @Nonnull
   class="kw">private class="kw">final RotatorDensity.SpecialCase axisSpecialCase;
   @Nonnull
   class="kw">private class="kw">final Vector3d rChildPosition;
   @Nonnull
   class="kw">private class="kw">final Density.Context rChildContext;

   class="kw">public RotatorDensity(@Nonnull Density var1, @Nonnull Vector3d var2, double var3) {
      this.input = var1;
      this.spinAngle = var3 * Math.PI / 180.0;
      Vector3d var5 = new Vector3d(0.0, 1.0, 0.0);
      this.rotationAxis = var2.cross(var5, new Vector3d());
      if (this.rotationAxis.length() < 1.0E-8) {
         this.rotationAxis = var5;
         if (var2.dot(var5) < 0.0) {
            this.axisSpecialCase = RotatorDensity.SpecialCase.INVERTED_Y_AXIS;
         } else {
            this.axisSpecialCase = RotatorDensity.SpecialCase.Y_AXIS;
         }
      } else {
         this.axisSpecialCase = RotatorDensity.SpecialCase.NONE;
      }

      this.rotationAxis.normalize();
      if (this.axisSpecialCase == RotatorDensity.SpecialCase.INVERTED_Y_AXIS || this.axisSpecialCase == RotatorDensity.SpecialCase.Y_AXIS) {
         this.tiltAxis = new Vector3d();
         this.tiltAngle = 0.0;
      }

      this.tiltAxis = var5.cross(var2, new Vector3d());
      this.tiltAngle = Math.acos(var2.dot(var5) / (var2.length() * var5.length()));
      this.rChildPosition = new Vector3d();
      this.rChildContext = new Density.Context();
   }

   @Override
   class="kw">public double process(@Nonnull Density.Context var1) {
      if (this.input == null) {
         class="kw">return 0.0;
      }

      this.rChildPosition.set(var1.position);
      class="kw">switch (this.axisSpecialCase) {
         case INVERTED_Y_AXIS:
            this.rChildPosition.mul(-1.0);
         case NONE:
            VectorUtil.rotateAroundAxis(this.rChildPosition, this.tiltAxis, this.tiltAngle);
         case Y_AXIS:
         class="kw">default:
            VectorUtil.rotateAroundAxis(this.rChildPosition, Y_AXIS, this.spinAngle);
            this.rChildContext.assign(var1);
            this.rChildContext.position = this.rChildPosition;
            double var2 = this.input.process(this.rChildContext);
            this.rChildContext.clearReferences();
            class="kw">return var2;
      }
   }

   @Override
   class="kw">public void setInputs(@Nonnull Density[] var1) {
      if (var1.length == 0) {
         this.input = null;
      }

      this.input = var1[0];
   }

   class="kw">private enum SpecialCase {
      NONE,
      Y_AXIS,
      INVERTED_Y_AXIS;

      SpecialCase() {
      }
   }
}