fullstack/algebricks/algebricks-data/src/main/java/edu/uci/ics/hyracks/algebricks/data/utils/WriteValueTools.java

package edu.uci.ics.hyracks.algebricks.data.utils;

import java.io.IOException;
import java.io.OutputStream;

import edu.uci.ics.hyracks.data.std.primitive.UTF8StringPointable;

public final class WriteValueTools {

    private final static int[] INT_INTERVALS = { 9, 99, 999, 9999, 99999, 999999, 9999999, 99999999, 999999999,
            Integer.MAX_VALUE };
    private final static int[] INT_DIVIDERS = { 10, 100, 1000, 10000, 100000, 1000000, 10000000, 100000000, 1000000000 };
    private final static int[] DIGITS = { '0', '1', '2', '3', '4', '5', '6', '7', '8', '9' };

    public static void writeInt(int i, OutputStream os) throws IOException {
        if (i < 0) {
            if (i == Integer.MIN_VALUE) {
                os.write("-2147483648".getBytes());
                return;
            }
            os.write('-');
            i = -i;
        }
        int k = 0;
        for (; k < INT_INTERVALS.length; k++) {
            if (i <= INT_INTERVALS[k]) {
                break;
            }
        }
        while (k > 0) {
            int q = i / INT_DIVIDERS[k - 1];
            os.write(DIGITS[q % 10]);
            k--;
        }
        // now, print the units
        os.write(DIGITS[i % 10]);
    }

    public static void writeLong(long d, OutputStream os) throws IOException {
        // now, print the units
        if (d < 0) {
            if (d == Long.MIN_VALUE) {
                os.write("-9223372036854775808".getBytes());
                return;
            }
            os.write('-');
            d = -d;
        }
        long divisor = 1000000000000000000L;
        while (divisor > d) {
            divisor = divisor / 10;
        }
        while (divisor > 1) {
            os.write(DIGITS[(int) ((d / divisor) % 10)]);
            divisor = divisor / 10;
        }
        os.write(DIGITS[(int) (d % 10)]);
    }

    public static void writeUTF8String(byte[] b, int s, int l, OutputStream os) throws IOException {
        int stringLength = UTF8StringPointable.getUTFLength(b, s);
        int position = s + 2;
        int maxPosition = position + stringLength;
        os.write('\"');
        while (position < maxPosition) {
            char c = UTF8StringPointable.charAt(b, position);
            switch (c) {
            // escape
                case '\\':
                case '"':
                    os.write('\\');
                    break;
            }
            int sz = UTF8StringPointable.charSize(b, position);
            while (sz > 0) {
                os.write(b[position]);
                position++;
                sz--;
            }
        }
        os.write('\"');
    }

    public static void writeUTF8StringNoQuotes(byte[] b, int s, int l, OutputStream os) throws IOException {
        int stringLength = UTF8StringPointable.getUTFLength(b, s);
        int position = s + 2;
        int maxPosition = position + stringLength;
        while (position < maxPosition) {
            char c = UTF8StringPointable.charAt(b, position);
            switch (c) {
            // escape
                case '\\':
                case '"':
                    os.write('\\');
                    break;
            }
            int sz = UTF8StringPointable.charSize(b, position);
            while (sz > 0) {
                os.write(b[position]);
                position++;
                sz--;
            }
        }
    }

}