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

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

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

public class CacheDensity extends Density
{
    private final WorkerIndexer.Data<Cache> threadData;
    @Nonnull
    private Density input;
    
    public CacheDensity(@Nonnull final Density input, final int threadCount) {
        this.input = input;
        this.threadData = new WorkerIndexer.Data<Cache>(threadCount, Cache::new);
    }
    
    @Override
    public double process(@Nonnull final Context context) {
        final Cache cache = this.threadData.get(context.workerId);
        if (cache.position != null && cache.position.x == context.position.x && cache.position.y == context.position.y && cache.position.z == context.position.z) {
            return cache.value;
        }
        if (cache.position == null) {
            cache.position = new Vector3d();
        }
        cache.position.assign(context.position);
        return cache.value = this.input.process(context);
    }
    
    @Override
    public void setInputs(@Nonnull final Density[] inputs) {
        assert inputs.length != 0;
        assert inputs[0] != null;
        this.input = inputs[0];
    }
    
    private static class Cache
    {
        Vector3d position;
        double value;
    }
}