package com.jsoniter.extra;
import com.jsoniter.CodegenAccess;
import com.jsoniter.JsonIterator;
import com.jsoniter.Slice;
import com.jsoniter.any.Any;
import com.jsoniter.output.JsonStream;
import com.jsoniter.spi.Decoder;
import com.jsoniter.spi.Encoder;
import com.jsoniter.spi.JsonException;
import com.jsoniter.spi.JsoniterSpi;
import java.io.IOException;
/**
* encode float/double as base64, faster than PreciseFloatSupport
*/
public class Base64FloatSupport {
final static int[] DIGITS = new int[256];
final static int[] HEX = new int[]{'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', 'a', 'b', 'c', 'd', 'e', 'f'};
final static int[] DEC = new int[127];
static {
for (int i = 0; i < 256; i++) {
DIGITS[i] = HEX[i >> 4] << 8 | HEX[i & 0xf];
}
DEC['0'] = 0;
DEC['1'] = 1;
DEC['2'] = 2;
DEC['3'] = 3;
DEC['4'] = 4;
DEC['5'] = 5;
DEC['6'] = 6;
DEC['7'] = 7;
DEC['8'] = 8;
DEC['9'] = 9;
DEC['a'] = 10;
DEC['b'] = 11;
DEC['c'] = 12;
DEC['d'] = 13;
DEC['e'] = 14;
DEC['f'] = 15;
}
private static boolean enabled;
public static synchronized void enableEncodersAndDecoders() {
if (enabled) {
throw new JsonException("BinaryFloatSupport.enable can only be called once");
}
enabled = true;
enableDecoders();
JsoniterSpi.registerTypeEncoder(Double.class, new Encoder() {
@Override
public void encode(Object obj, JsonStream stream) throws IOException {
Double number = (Double) obj;
long bits = Double.doubleToRawLongBits(number.doubleValue());
Base64.encodeLongBits(bits, stream);
}
@Override
public Any wrap(Object obj) {
Double number = (Double) obj;
return Any.wrap(number.doubleValue());
}
});
JsoniterSpi.registerTypeEncoder(double.class, new Encoder.DoubleEncoder() {
@Override
public void encodeDouble(double obj, JsonStream stream) throws IOException {
long bits = Double.doubleToRawLongBits(obj);
Base64.encodeLongBits(bits, stream);
}
});
JsoniterSpi.registerTypeEncoder(Float.class, new Encoder() {
@Override
public void encode(Object obj, JsonStream stream) throws IOException {
Float number = (Float) obj;
long bits = Double.doubleToRawLongBits(number.doubleValue());
Base64.encodeLongBits(bits, stream);
}
@Override
public Any wrap(Object obj) {
Float number = (Float) obj;
return Any.wrap(number.floatValue());
}
});
JsoniterSpi.registerTypeEncoder(float.class, new Encoder.FloatEncoder() {
@Override
public void encodeFloat(float obj, JsonStream stream) throws IOException {
long bits = Double.doubleToRawLongBits(obj);
Base64.encodeLongBits(bits, stream);
}
});
}
public static void enableDecoders() {
JsoniterSpi.registerTypeDecoder(Double.class, new Decoder() {
@Override
public Object decode(JsonIterator iter) throws IOException {
byte token = CodegenAccess.nextToken(iter);
CodegenAccess.unreadByte(iter);
if (token == '"') {
return Double.longBitsToDouble(Base64.decodeLongBits(iter));
} else {
return iter.readDouble();
}
}
});
JsoniterSpi.registerTypeDecoder(double.class, new Decoder.DoubleDecoder() {
@Override
public double decodeDouble(JsonIterator iter) throws IOException {
byte token = CodegenAccess.nextToken(iter);
CodegenAccess.unreadByte(iter);
if (token == '"') {
return Double.longBitsToDouble(Base64.decodeLongBits(iter));
}else {
return iter.readDouble();
}
}
});
JsoniterSpi.registerTypeDecoder(Float.class, new Decoder() {
@Override
public Object decode(JsonIterator iter) throws IOException {
byte token = CodegenAccess.nextToken(iter);
CodegenAccess.unreadByte(iter);
if (token == '"') {
return (float)Double.longBitsToDouble(Base64.decodeLongBits(iter));
}else {
return (float)iter.readDouble();
}
}
});
JsoniterSpi.registerTypeDecoder(float.class, new Decoder.FloatDecoder() {
@Override
public float decodeFloat(JsonIterator iter) throws IOException {
byte token = CodegenAccess.nextToken(iter);
CodegenAccess.unreadByte(iter);
if (token == '"') {
return (float)Double.longBitsToDouble(Base64.decodeLongBits(iter));
}else {
return (float)iter.readDouble();
}
}
});
}
private static long readLongBits(JsonIterator iter) throws IOException {
Slice slice = iter.readStringAsSlice();
byte[] data = slice.data();
long val = 0;
for (int i = slice.head(); i < slice.tail(); i++) {
byte b = data[i];
val = val << 4 | DEC[b];
}
return val;
}
private static void writeLongBits(long bits, JsonStream stream) throws IOException {
int digit = DIGITS[(int) (bits & 0xff)];
byte b2 = (byte) (digit >> 8);
byte b1 = (byte) digit;
bits = bits >> 8;
if (bits == 0) {
stream.write((byte) '"', b2, b1, (byte) '"');
}
digit = DIGITS[(int) (bits & 0xff)];
byte b4 = (byte) (digit >> 8);
byte b3 = (byte) digit;
bits = bits >> 8;
if (bits == 0) {
stream.write((byte) '"', b4, b3, b2, b1, (byte) '"');
}
digit = DIGITS[(int) (bits & 0xff)];
byte b6 = (byte) (digit >> 8);
byte b5 = (byte) digit;
bits = bits >> 8;
if (bits == 0) {
stream.write((byte) '"', b6, b5, b4, b3);
stream.write(b2, b1, (byte) '"');
}
digit = DIGITS[(int) (bits & 0xff)];
byte b8 = (byte) (digit >> 8);
byte b7 = (byte) digit;
bits = bits >> 8;
if (bits == 0) {
stream.write((byte) '"', b8, b7, b6, b5, b4);
stream.write(b3, b2, b1, (byte) '"');
}
digit = DIGITS[(int) (bits & 0xff)];
byte b10 = (byte) (digit >> 8);
byte b9 = (byte) digit;
bits = bits >> 8;
if (bits == 0) {
stream.write((byte) '"', b10, b9, b8, b7, b6);
stream.write(b5, b4, b3, b2, b1, (byte) '"');
}
digit = DIGITS[(int) (bits & 0xff)];
byte b12 = (byte) (digit >> 8);
byte b11 = (byte) digit;
bits = bits >> 8;
if (bits == 0) {
stream.write((byte) '"', b12, b11, b10, b9, b8);
stream.write(b7, b6, b5, b4, b3, b2);
stream.write(b1, (byte) '"');
}
digit = DIGITS[(int) (bits & 0xff)];
byte b14 = (byte) (digit >> 8);
byte b13 = (byte) digit;
bits = bits >> 8;
if (bits == 0) {
stream.write((byte) '"', b14, b13, b12, b11, b10);
stream.write(b9, b8, b7, b6, b5, b4);
stream.write(b3, b2, b1, (byte) '"');
}
digit = DIGITS[(int) (bits & 0xff)];
byte b16 = (byte) (digit >> 8);
byte b15 = (byte) digit;
stream.write((byte) '"', b16, b15, b14, b13, b12);
stream.write(b11, b10, b9, b8, b7, b6);
stream.write(b5, b4, b3, b2, b1, (byte) '"');
}
}