ClusterPropAsset.java

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

import com.hypixel.hytale.assetstore.AssetExtraInfo;
import com.hypixel.hytale.assetstore.codec.AssetBuilderCodec;
import com.hypixel.hytale.assetstore.map.DefaultAssetMap;
import com.hypixel.hytale.assetstore.map.JsonAssetWithMap;
import com.hypixel.hytale.builtin.hytalegenerator.assets.Cleanable;
import com.hypixel.hytale.builtin.hytalegenerator.assets.curves.ConstantCurveAsset;
import com.hypixel.hytale.builtin.hytalegenerator.assets.curves.CurveAsset;
import com.hypixel.hytale.builtin.hytalegenerator.assets.patterns.ConstantPatternAsset;
import com.hypixel.hytale.builtin.hytalegenerator.assets.patterns.PatternAsset;
import com.hypixel.hytale.builtin.hytalegenerator.assets.scanners.OriginScannerAsset;
import com.hypixel.hytale.builtin.hytalegenerator.assets.scanners.ScannerAsset;
import com.hypixel.hytale.builtin.hytalegenerator.datastructures.WeightedMap;
import com.hypixel.hytale.builtin.hytalegenerator.patterns.Pattern;
import com.hypixel.hytale.builtin.hytalegenerator.props.ClusterProp;
import com.hypixel.hytale.builtin.hytalegenerator.props.Prop;
import com.hypixel.hytale.builtin.hytalegenerator.scanners.OriginScanner;
import com.hypixel.hytale.builtin.hytalegenerator.scanners.Scanner;
import com.hypixel.hytale.codec.Codec;
import com.hypixel.hytale.codec.KeyedCodec;
import com.hypixel.hytale.codec.builder.BuilderCodec;
import com.hypixel.hytale.codec.codecs.array.ArrayCodec;
import com.hypixel.hytale.codec.validation.Validators;
import javax.annotation.Nonnull;

public class ClusterPropAsset extends PropAsset {
   public static final BuilderCodec<ClusterPropAsset> CODEC;
   private int range = 0;
   private CurveAsset distanceCurve = new ConstantCurveAsset();
   private String seed = "A";
   private WeightedPropAsset[] weightedPropAssets = new WeightedPropAsset[0];
   private PatternAsset patternAsset = new ConstantPatternAsset();
   private ScannerAsset scannerAsset = new OriginScannerAsset();

   public ClusterPropAsset() {
   }

   @Nonnull
   public Prop build(@Nonnull PropAsset.Argument argument) {
      if (super.skip()) {
         return Prop.noProp();
      } else {
         WeightedMap<Prop> weightedMap = new WeightedMap<Prop>();

         for(WeightedPropAsset entry : this.weightedPropAssets) {
            weightedMap.add(entry.propAsset.build(argument), entry.weight);
         }

         Pattern pattern = this.patternAsset == null ? Pattern.yesPattern() : this.patternAsset.build(PatternAsset.argumentFrom(argument));
         Scanner scanner = (Scanner)(this.scannerAsset == null ? OriginScanner.getInstance() : this.scannerAsset.build(ScannerAsset.argumentFrom(argument)));
         int intSeed = (Integer)argument.parentSeed.child(this.seed).createSupplier().get();
         return new ClusterProp(this.range, this.distanceCurve.build(), intSeed, weightedMap, pattern, scanner);
      }
   }

   public void cleanUp() {
      this.distanceCurve.cleanUp();

      for(WeightedPropAsset weightedPropAsset : this.weightedPropAssets) {
         weightedPropAsset.cleanUp();
      }

      this.patternAsset.cleanUp();
      this.scannerAsset.cleanUp();
   }

   static {
      CODEC = ((BuilderCodec.Builder)((BuilderCodec.Builder)((BuilderCodec.Builder)((BuilderCodec.Builder)((BuilderCodec.Builder)((BuilderCodec.Builder)BuilderCodec.builder(ClusterPropAsset.class, ClusterPropAsset::new, PropAsset.ABSTRACT_CODEC).append(new KeyedCodec("Range", Codec.INTEGER, false), (asset, v) -> asset.range = v, (asset) -> asset.range).addValidator(Validators.greaterThanOrEqual(0)).add()).append(new KeyedCodec("DistanceCurve", CurveAsset.CODEC, true), (asset, v) -> asset.distanceCurve = v, (asset) -> asset.distanceCurve).add()).append(new KeyedCodec("Seed", Codec.STRING, false), (asset, v) -> asset.seed = v, (asset) -> asset.seed).add()).append(new KeyedCodec("WeightedProps", new ArrayCodec(ClusterPropAsset.WeightedPropAsset.CODEC, (x$0) -> new WeightedPropAsset[x$0]), true), (asset, v) -> asset.weightedPropAssets = v, (asset) -> asset.weightedPropAssets).add()).append(new KeyedCodec("Pattern", PatternAsset.CODEC, false), (asset, v) -> asset.patternAsset = v, (asset) -> asset.patternAsset).add()).append(new KeyedCodec("Scanner", ScannerAsset.CODEC, false), (asset, v) -> asset.scannerAsset = v, (asset) -> asset.scannerAsset).add()).build();
   }

   public static class WeightedPropAsset implements Cleanable, JsonAssetWithMap<String, DefaultAssetMap<String, WeightedPropAsset>> {
      public static final AssetBuilderCodec<String, WeightedPropAsset> CODEC;
      private String id;
      private AssetExtraInfo.Data data;
      private double weight = 1.0;
      private PropAsset propAsset;

      public WeightedPropAsset() {
      }

      public String getId() {
         return this.id;
      }

      public void cleanUp() {
         this.propAsset.cleanUp();
      }

      static {
         CODEC = ((AssetBuilderCodec.Builder)((AssetBuilderCodec.Builder)AssetBuilderCodec.builder(WeightedPropAsset.class, WeightedPropAsset::new, Codec.STRING, (asset, id) -> asset.id = id, (config) -> config.id, (config, data) -> config.data = data, (config) -> config.data).append(new KeyedCodec("Weight", Codec.DOUBLE, true), (t, w) -> t.weight = w, (t) -> t.weight).addValidator(Validators.greaterThan(0.0)).add()).append(new KeyedCodec("ColumnProp", PropAsset.CODEC, true), (t, out) -> t.propAsset = out, (t) -> t.propAsset).add()).build();
      }
   }
}