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 VectorWarpDensity class="kw">extends Density {
   @Nullable
   class="kw">private Density input;
   @Nullable
   class="kw">private Density warpInput;
   class="kw">private class="kw">final double warpFactor;
   @Nonnull
   class="kw">private class="kw">final Vector3d warpVector;
   @Nonnull
   class="kw">private class="kw">final Vector3d rSamplePoint;
   @Nonnull
   class="kw">private class="kw">final Density.Context rChildContext;

   class="kw">public VectorWarpDensity(@Nonnull Density var1, @Nonnull Density var2, double var3, @Nonnull Vector3d var5) {
      this.input = var1;
      this.warpInput = var2;
      this.warpFactor = var3;
      this.warpVector = var5;
      this.rSamplePoint = 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;
      }

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

      double var2 = this.warpInput.process(var1);
      var2 *= this.warpFactor;
      this.rSamplePoint.set(this.warpVector);
      this.rSamplePoint.normalize(1.0);
      this.rSamplePoint.mul(var2);
      this.rSamplePoint.add(var1.position);
      this.rChildContext.assign(var1);
      this.rChildContext.position = this.rSamplePoint;
      double var4 = this.input.process(this.rChildContext);
      this.rChildContext.clearReferences();
      class="kw">return var4;
   }

   @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];
   }
}