DoubleCodec.java

package com.hypixel.hytale.codec.codecs.simple;

import com.hypixel.hytale.codec.Codec;
import com.hypixel.hytale.codec.ExtraInfo;
import com.hypixel.hytale.codec.PrimitiveCodec;
import com.hypixel.hytale.codec.RawJsonCodec;
import com.hypixel.hytale.codec.schema.SchemaContext;
import com.hypixel.hytale.codec.schema.config.NumberSchema;
import com.hypixel.hytale.codec.schema.config.Schema;
import com.hypixel.hytale.codec.util.RawJsonReader;
import java.io.IOException;
import javax.annotation.Nonnull;
import javax.annotation.Nullable;
import org.bson.BsonDouble;
import org.bson.BsonValue;

public class DoubleCodec implements Codec<Double>, RawJsonCodec<Double>, PrimitiveCodec {
   public DoubleCodec() {
   }

   @Nonnull
   public Double decode(@Nonnull BsonValue bsonValue, ExtraInfo extraInfo) {
      if (bsonValue.isString()) {
         switch (bsonValue.asString().getValue()) {
            case "NaN" -> {
               return 0.0 / 0.0;
            }
            case "Infinity" -> {
               return 1.0 / 0.0;
            }
            case "-Infinity" -> {
               return -1.0 / 0.0;
            }
         }
      }

      return bsonValue.asNumber().doubleValue();
   }

   @Nonnull
   public BsonValue encode(Double t, ExtraInfo extraInfo) {
      return new BsonDouble(t);
   }

   @Nonnull
   public Double decodeJson(@Nonnull RawJsonReader reader, ExtraInfo extraInfo) throws IOException {
      return reader.readDoubleValue();
   }

   @Nonnull
   public Schema toSchema(@Nonnull SchemaContext context) {
      return new NumberSchema();
   }

   @Nonnull
   public Schema toSchema(@Nonnull SchemaContext context, @Nullable Double def) {
      NumberSchema s = new NumberSchema();
      if (def != null && !def.isNaN() && !def.isInfinite()) {
         s.setDefault(def);
      }

      return s;
   }
}