SumCurveAsset.java

package com.hypixel.hytale.builtin.hytalegenerator.assets.curves;

import com.hypixel.hytale.codec.KeyedCodec;
import com.hypixel.hytale.codec.builder.BuilderCodec;
import com.hypixel.hytale.codec.codecs.array.ArrayCodec;
import it.unimi.dsi.fastutil.doubles.Double2DoubleFunction;
import java.util.Objects;
import javax.annotation.Nonnull;

public class SumCurveAsset extends CurveAsset {
   public static final BuilderCodec<SumCurveAsset> CODEC;
   private CurveAsset[] curveAssets = new CurveAsset[0];

   public SumCurveAsset() {
   }

   @Nonnull
   public Double2DoubleFunction build() {
      if (this.curveAssets.length == 0) {
         return (in) -> 0.0;
      } else {
         Double2DoubleFunction[] inputCurves = new Double2DoubleFunction[this.curveAssets.length];

         for(int i = 0; i < this.curveAssets.length; ++i) {
            inputCurves[i] = this.curveAssets[i].build();
         }

         if (inputCurves.length == 1) {
            Double2DoubleFunction curve = inputCurves[0];
            Objects.requireNonNull(curve);
            return curve::applyAsDouble;
         } else if (inputCurves.length == 2) {
            Double2DoubleFunction curveA = inputCurves[0];
            Double2DoubleFunction curveB = inputCurves[1];
            return (in) -> curveA.applyAsDouble(in) + curveB.applyAsDouble(in);
         } else if (inputCurves.length == 3) {
            Double2DoubleFunction curveA = inputCurves[0];
            Double2DoubleFunction curveB = inputCurves[1];
            Double2DoubleFunction curveC = inputCurves[2];
            return (in) -> curveA.applyAsDouble(in) + curveB.applyAsDouble(in) + curveC.applyAsDouble(in);
         } else {
            return (in) -> {
               double value = 0.0;

               for(Double2DoubleFunction curve : inputCurves) {
                  value += curve.applyAsDouble(in);
               }

               return value;
            };
         }
      }
   }

   public void cleanUp() {
      for(CurveAsset curveAsset : this.curveAssets) {
         curveAsset.cleanUp();
      }

   }

   static {
      CODEC = ((BuilderCodec.Builder)BuilderCodec.builder(SumCurveAsset.class, SumCurveAsset::new, CurveAsset.ABSTRACT_CODEC).append(new KeyedCodec("Curves", new ArrayCodec(CurveAsset.CODEC, (x$0) -> new CurveAsset[x$0]), true), (t, k) -> t.curveAssets = k, (k) -> k.curveAssets).add()).build();
   }
}