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

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.material.Material;
class="kw">import com.hypixel.hytale.builtin.hytalegenerator.materialproviders.MaterialProvider;
class="kw">import com.hypixel.hytale.builtin.hytalegenerator.materialproviders.WeightedMaterialProvider;
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 WeightedMaterialProviderAsset class="kw">extends MaterialProviderAsset {
   @Nonnull
   class="kw">public class="kw">static class="kw">final BuilderCodec<WeightedMaterialProviderAsset> CODEC = BuilderCodec.builder(
         WeightedMaterialProviderAsset.class, WeightedMaterialProviderAsset::new, MaterialProviderAsset.ABSTRACT_CODEC
      )
      .append(
         new KeyedCodec<>(
            "WeightedMaterials",
            new ArrayCodec<>(WeightedMaterialProviderAsset.WeightedMaterialAsset.CODEC, WeightedMaterialProviderAsset.WeightedMaterialAsset[]::new),
            true
         ),
         (var0, var1) -> var0.weighedMapEntries = var1,
         var0 -> var0.weighedMapEntries
      )
      .add()
      .append(new KeyedCodec<>("SkipChance", Codec.DOUBLE, true), (var0, var1) -> var0.skipChance = var1, var0 -> var0.skipChance)
      .add()
      .append(new KeyedCodec<>("Seed", Codec.STRING, true), (var0, var1) -> var0.seed = var1, var0 -> var0.seed)
      .add()
      .build();
   class="kw">private WeightedMaterialProviderAsset.WeightedMaterialAsset[] weighedMapEntries = new WeightedMaterialProviderAsset.WeightedMaterialAsset[0];
   class="kw">private double skipChance;
   class="kw">private String seed = "";

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

   @Nonnull
   @Override
   class="kw">public MaterialProvider<Material> build(@Nonnull MaterialProviderAsset.Argument var1) {
      if (super.skip()) {
         class="kw">return MaterialProvider.noMaterialProvider();
      }

      WeightedMap var2 = new WeightedMap<>();

      for (WeightedMaterialProviderAsset.WeightedMaterialAsset var6 : this.weighedMapEntries) {
         var2.add(var6.materialProviderAsset.build(var1), var6.weight);
      }

      class="kw">return new WeightedMaterialProvider<>(var2, var1.parentSeed.child(this.seed), this.skipChance);
   }

   @Override
   class="kw">public void cleanUp() {
      for (WeightedMaterialProviderAsset.WeightedMaterialAsset var4 : this.weighedMapEntries) {
         var4.cleanUp();
      }
   }

   class="kw">public class="kw">static class WeightedMaterialAsset
      class="kw">implements Cleanable,
      JsonAssetWithMap<String, DefaultAssetMap<String, WeightedMaterialProviderAsset.WeightedMaterialAsset>> {
      @Nonnull
      class="kw">public class="kw">static class="kw">final AssetBuilderCodec<String, WeightedMaterialProviderAsset.WeightedMaterialAsset> CODEC = AssetBuilderCodec.builder(
            WeightedMaterialProviderAsset.WeightedMaterialAsset.class,
            WeightedMaterialProviderAsset.WeightedMaterialAsset::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.greaterThanOrEqual(0.0))
         .add()
         .append(
            new KeyedCodec<>("Material", MaterialProviderAsset.CODEC, true),
            (var0, var1) -> var0.materialProviderAsset = var1,
            var0 -> var0.materialProviderAsset
         )
         .add()
         .build();
      class="kw">private String id;
      class="kw">private AssetExtraInfo.Data data;
      class="kw">private double weight = 1.0;
      class="kw">private MaterialProviderAsset materialProviderAsset = new ConstantMaterialProviderAsset();

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

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

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