class="kw">package com.hypixel.hytale.builtin.hytalegenerator.assets.graph.contentsuppliers;

class="kw">import com.hypixel.hytale.assetstore.AssetExtraInfo;
class="kw">import com.hypixel.hytale.assetstore.codec.AssetBuilderCodec;
class="kw">import com.hypixel.hytale.assetstore.map.DefaultAssetMap;
class="kw">import com.hypixel.hytale.assetstore.map.JsonAssetWithMap;
class="kw">import com.hypixel.hytale.builtin.hytalegenerator.WeightedMap;
class="kw">import com.hypixel.hytale.builtin.hytalegenerator.assets.Cleanable;
class="kw">import com.hypixel.hytale.builtin.hytalegenerator.assets.graph.GraphGeneratorAsset;
class="kw">import com.hypixel.hytale.builtin.hytalegenerator.graph.contentsuppliers.ConstantContentSupplier;
class="kw">import com.hypixel.hytale.builtin.hytalegenerator.graph.contentsuppliers.ContentSupplier;
class="kw">import com.hypixel.hytale.builtin.hytalegenerator.graph.contentsuppliers.WeightedContentSupplier;
class="kw">import com.hypixel.hytale.codec.Codec;
class="kw">import com.hypixel.hytale.codec.KeyedCodec;
class="kw">import com.hypixel.hytale.codec.builder.BuilderCodec;
class="kw">import com.hypixel.hytale.codec.codecs.array.ArrayCodec;
class="kw">import com.hypixel.hytale.codec.validation.Validators;
class="kw">import javax.annotation.Nonnull;

class="kw">public class WeightedContentSupplierAsset class="kw">extends ContentSupplierAsset {
   @Nonnull
   class="kw">public class="kw">static class="kw">final BuilderCodec<WeightedContentSupplierAsset> CODEC = BuilderCodec.builder(
         WeightedContentSupplierAsset.class, WeightedContentSupplierAsset::new, ContentSupplierAsset.ABSTRACT_CODEC
      )
      .append(
         new KeyedCodec<>("Entries", new ArrayCodec<>(WeightedContentSupplierAsset.EntryAsset.CODEC, WeightedContentSupplierAsset.EntryAsset[]::new), true),
         (var0, var1) -> var0.entryAssets = var1,
         var0 -> var0.entryAssets
      )
      .add()
      .append(new KeyedCodec<>("Seed", Codec.STRING, true), (var0, var1) -> var0.seed = var1, var0 -> var0.seed)
      .add()
      .build();
   @Nonnull
   class="kw">private WeightedContentSupplierAsset.EntryAsset[] entryAssets = new WeightedContentSupplierAsset.EntryAsset[0];
   @Nonnull
   class="kw">private String seed = "";

   class="kw">public WeightedContentSupplierAsset() {
   }

   @Nonnull
   @Override
   class="kw">public ContentSupplier build(@Nonnull GraphGeneratorAsset.Argument var1) {
      if (!super.skip() && this.entryAssets.length != 0) {
         WeightedMap var2 = new WeightedMap<>(this.entryAssets.length);

         for (WeightedContentSupplierAsset.EntryAsset var6 : this.entryAssets) {
            var2.add(var6.contentSupplierAsset.build(var1), var6.weight);
         }

         class="kw">return new WeightedContentSupplier(var2, var1.parentSeed.child(this.seed).createSupplier().get());
      } else {
         class="kw">return ConstantContentSupplier.INSTANCE;
      }
   }

   @Override
   class="kw">public void cleanUp() {
      for (WeightedContentSupplierAsset.EntryAsset var4 : this.entryAssets) {
         var4.cleanUp();
      }
   }

   class="kw">public class="kw">static class EntryAsset class="kw">implements Cleanable, JsonAssetWithMap<String, DefaultAssetMap<String, WeightedContentSupplierAsset.EntryAsset>> {
      @Nonnull
      class="kw">public class="kw">static class="kw">final AssetBuilderCodec<String, WeightedContentSupplierAsset.EntryAsset> CODEC = AssetBuilderCodec.builder(
            WeightedContentSupplierAsset.EntryAsset.class,
            WeightedContentSupplierAsset.EntryAsset::new,
            Codec.STRING,
            (var0, var1) -> var0.id = var1,
            var0 -> var0.id,
            (var0, var1) -> var0.data = var1,
            var0 -> var0.data
         )
         .append(new KeyedCodec<>("Weight", Codec.DOUBLE, true), (var0, var1) -> var0.weight = var1, var0 -> var0.weight)
         .addValidator(Validators.greaterThan(0.0))
         .add()
         .append(
            new KeyedCodec<>("ContentSupplier", ContentSupplierAsset.CODEC, true),
            (var0, var1) -> var0.contentSupplierAsset = var1,
            var0 -> var0.contentSupplierAsset
         )
         .add()
         .build();
      class="kw">private String id;
      class="kw">private AssetExtraInfo.Data data;
      class="kw">private double weight = 1.0;
      @Nonnull
      class="kw">private ContentSupplierAsset contentSupplierAsset = ConstantContentSupplierAsset.INSTANCE;

      class="kw">public EntryAsset() {
      }

      class="kw">public String getId() {
         class="kw">return this.id;
      }

      @Override
      class="kw">public void cleanUp() {
         this.contentSupplierAsset.cleanUp();
      }
   }
}