// 
// Decompiled by Procyon v0.6.0
// 

package com.hypixel.hytale.builtin.hytalegenerator.density.nodes;

import javax.annotation.Nonnull;
import com.hypixel.hytale.math.vector.Vector3d;
import javax.annotation.Nullable;
import com.hypixel.hytale.builtin.hytalegenerator.density.Density;

public class VectorWarpDensity extends Density
{
    @Nullable
    private Density input;
    @Nullable
    private Density warpInput;
    private final double warpFactor;
    @Nonnull
    private final Vector3d warpVector;
    
    public VectorWarpDensity(@Nonnull final Density input, @Nonnull final Density warpInput, final double warpFactor, @Nonnull final Vector3d warpVector) {
        this.input = input;
        this.warpInput = warpInput;
        this.warpFactor = warpFactor;
        this.warpVector = warpVector;
    }
    
    @Override
    public double process(@Nonnull final Context context) {
        if (this.input == null) {
            return 0.0;
        }
        if (this.warpInput == null) {
            return this.input.process(context);
        }
        double warp = this.warpInput.process(context);
        warp *= this.warpFactor;
        final Vector3d samplePoint = this.warpVector.clone();
        samplePoint.setLength(1.0);
        samplePoint.scale(warp);
        samplePoint.add(context.position);
        final Context childContext = new Context(context);
        childContext.position = samplePoint;
        return this.input.process(childContext);
    }
    
    @Override
    public void setInputs(@Nonnull final Density[] inputs) {
        if (inputs.length == 0) {
            this.input = null;
        }
        this.input = inputs[0];
        if (inputs.length < 2) {
            this.warpInput = null;
        }
        this.warpInput = inputs[1];
    }
}