class="kw">package com.hypixel.hytale.codec.codecs.map;

class="kw">import com.hypixel.hytale.codec.Codec;
class="kw">import com.hypixel.hytale.codec.ExtraInfo;
class="kw">import com.hypixel.hytale.codec.WrappedCodec;
class="kw">import com.hypixel.hytale.codec.schema.SchemaContext;
class="kw">import com.hypixel.hytale.codec.schema.config.NumberSchema;
class="kw">import com.hypixel.hytale.codec.schema.config.ObjectSchema;
class="kw">import com.hypixel.hytale.codec.schema.config.Schema;
class="kw">import com.hypixel.hytale.codec.schema.config.StringSchema;
class="kw">import com.hypixel.hytale.codec.util.RawJsonReader;
class="kw">import it.unimi.dsi.fastutil.objects.Object2DoubleMap;
class="kw">import it.unimi.dsi.fastutil.objects.Object2DoubleMaps;
class="kw">import it.unimi.dsi.fastutil.objects.ObjectIterator;
class="kw">import java.io.IOException;
class="kw">import java.util.Map.Entry;
class="kw">import java.util.function.Supplier;
class="kw">import javax.annotation.Nonnull;
class="kw">import org.bson.BsonDocument;
class="kw">import org.bson.BsonDouble;
class="kw">import org.bson.BsonString;
class="kw">import org.bson.BsonValue;

class="kw">public class Object2DoubleMapCodec<T> class="kw">implements Codec<Object2DoubleMap<T>>, WrappedCodec<T> {
   class="kw">private class="kw">final Codec<T> keyCodec;
   class="kw">private class="kw">final Supplier<Object2DoubleMap<T>> supplier;
   class="kw">private class="kw">final boolean unmodifiable;

   class="kw">public Object2DoubleMapCodec(Codec<T> var1, Supplier<Object2DoubleMap<T>> var2, boolean var3) {
      this.keyCodec = var1;
      this.supplier = var2;
      this.unmodifiable = var3;
   }

   class="kw">public Object2DoubleMapCodec(Codec<T> var1, Supplier<Object2DoubleMap<T>> var2) {
      this(var1, var2, true);
   }

   @Override
   class="kw">public Codec<T> getChildCodec() {
      class="kw">return this.keyCodec;
   }

   class="kw">public Object2DoubleMap<T> decode(@Nonnull BsonValue var1, ExtraInfo var2) {
      BsonDocument var3 = var1.asDocument();
      Object2DoubleMap var4 = this.supplier.get();

      for (Entry var6 : var3.entrySet()) {
         Object var7 = this.keyCodec.decode(new BsonString(var6.getKey()), var2);
         var4.put(var7, var6.getValue().asNumber().doubleValue());
      }

      if (this.unmodifiable) {
         var4 = Object2DoubleMaps.unmodifiable(var4);
      }

      class="kw">return var4;
   }

   @Nonnull
   class="kw">public BsonValue encode(@Nonnull Object2DoubleMap<T> var1, ExtraInfo var2) {
      BsonDocument var3 = new BsonDocument();
      ObjectIterator var4 = var1.keySet().iterator();

      while (var4.hasNext()) {
         Object var5 = (T)var4.next();
         String var6 = this.keyCodec.encode(var5, var2).asString().getValue();
         var3.put(var6, new BsonDouble(var1.getDouble(var5)));
      }

      class="kw">return var3;
   }

   class="kw">public Object2DoubleMap<T> decodeJson(@Nonnull RawJsonReader var1, ExtraInfo var2) class="kw">throws IOException {
      var1.expect('{');
      var1.consumeWhiteSpace();
      Object2DoubleMap var3 = this.supplier.get();
      if (var1.tryConsume('}')) {
         if (this.unmodifiable) {
            var3 = Object2DoubleMaps.unmodifiable(var3);
         }

         class="kw">return var3;
      } else {
         while (true) {
            Object var4 = this.keyCodec.decodeJson(var1, var2);
            var1.consumeWhiteSpace();
            var1.expect(':');
            var1.consumeWhiteSpace();
            var3.put(var4, var1.readDoubleValue());
            var1.consumeWhiteSpace();
            if (var1.tryConsumeOrExpect('}', ',')) {
               if (this.unmodifiable) {
                  var3 = Object2DoubleMaps.unmodifiable(var3);
               }

               class="kw">return var3;
            }

            var1.consumeWhiteSpace();
         }
      }
   }

   @Nonnull
   @Override
   class="kw">public Schema toSchema(@Nonnull SchemaContext var1) {
      ObjectSchema var2 = new ObjectSchema();
      StringSchema var3 = (StringSchema)this.keyCodec.toSchema(var1);
      String var4 = var3.getTitle();
      if (var4 == null) {
         var4 = var3.getHytale().getType();
      }

      var2.setTitle("Map of " + var4 + " to double");
      var2.setPropertyNames(var3);
      var2.setAdditionalProperties(new NumberSchema());
      class="kw">return var2;
   }
}