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

class="kw">import javax.annotation.Nonnull;
class="kw">import javax.annotation.Nullable;
class="kw">import org.joml.Vector3d;

class="kw">public class GradientWarpDensity class="kw">extends Density {
   class="kw">private class="kw">static class="kw">final double HALF_PI = Math.PI / 2;
   @Nullable
   class="kw">private Density input;
   @Nullable
   class="kw">private Density warpInput;
   class="kw">private class="kw">final double slopeRange;
   class="kw">private class="kw">final double warpFactor;
   @Nonnull
   class="kw">private class="kw">final Density.Context rChildContext;
   @Nonnull
   class="kw">private class="kw">final Vector3d rGradient;
   @Nonnull
   class="kw">public class="kw">final Vector3d rPosition;

   class="kw">public GradientWarpDensity(@Nonnull Density var1, @Nonnull Density var2, double var3, double var5) {
      if (var3 <= 0.0) {
         throw new IllegalArgumentException();
      }

      this.slopeRange = var3;
      this.warpFactor = var5;
      this.input = var1;
      this.warpInput = var2;
      this.rChildContext = new Density.Context();
      this.rGradient = new Vector3d();
      this.rPosition = new Vector3d();
   }

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

      if (this.warpInput == null) {
         class="kw">return this.input.process(var1);
      }

      double var2 = this.warpInput.process(var1);
      double var4 = var1.position.x + this.slopeRange;
      double var6 = var1.position.y + this.slopeRange;
      double var8 = var1.position.z + this.slopeRange;
      this.rPosition.set(var1.position);
      this.rChildContext.assign(var1);
      this.rChildContext.position = this.rPosition;
      this.rChildContext.position.set(var4, var1.position.y, var1.position.z);
      double var10 = this.warpInput.process(this.rChildContext) - var2;
      this.rChildContext.position.set(var1.position.x, var6, var1.position.z);
      double var12 = this.warpInput.process(this.rChildContext) - var2;
      this.rChildContext.position.set(var1.position.x, var1.position.y, var8);
      double var14 = this.warpInput.process(this.rChildContext) - var2;
      this.rGradient.set(var10, var12, var14);
      this.rGradient.mul(1.0 / this.slopeRange);
      this.rGradient.mul(this.warpFactor);
      this.rGradient.add(var1.position.x, var1.position.y, var1.position.z);
      this.rChildContext.position = this.rGradient;
      double var16 = this.input.process(this.rChildContext);
      this.rChildContext.clearReferences();
      class="kw">return var16;
   }

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

      this.input = var1[0];
      if (var1.length < 2) {
         this.warpInput = null;
      }

      this.warpInput = var1[1];
   }
}