1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
|
package com.seaofnodes.simple.util;
import java.util.HashMap;
import java.util.Arrays;
// "ByteArrayOutputStream", and Input Stream and not synchronized and has
// matching pack/unpack strategies.
public class BAOS {
private byte[] _buf;
private int _len;
public BAOS() { _buf = new byte[1]; }
public BAOS(byte[] buf) { _buf = buf; }
public BAOS(byte[] buf, int x) { _buf = buf; _len = x; }
public byte[] toByteArray() {
return Arrays.copyOf(_buf,_len);
}
public byte[] buf() { return _buf; }
public int size() { return _len; }
public void set( byte[] buf, int len ) { _buf=buf; _len=len; }
public void clear() { _len=0; }
// -----------------------------------------------------------------------
// Basic read, write (expand)
private void grow(int len) {
while( _len+len >= _buf.length )
_buf = Arrays.copyOf(_buf,_buf.length*2);
}
// Write a byte, grow as needed; silent chop
public BAOS write(int b) {
grow(1);
_buf[_len++] = (byte)b; // Silent chop
return this;
}
// Read an UNSIGNED bytes at INDEX, AIOOBE if off end
public int read1(int idx) { return _buf[idx] & 0xFF; }
// Read an UNSIGNED byte, AIOOBE if off end
public int read() { return read1(_len++); }
// write 2 bytes
public BAOS write2( int x ) {
grow(2);
_buf[_len++] = (byte)(x>> 0);
_buf[_len++] = (byte)(x>> 8);
return this;
}
// Read a u16 in 2 bytes
public int read2(int idx) { return read1(idx) | (read1(idx+1)<<8); }
public int read2() { int rez = read2(_len); _len+=2; return rez; }
// write 4 bytes
public BAOS write4( int x ) {
grow(4);
_buf[_len++] = (byte)(x>> 0);
_buf[_len++] = (byte)(x>> 8);
_buf[_len++] = (byte)(x>>16);
_buf[_len++] = (byte)(x>>24);
return this;
}
// Read a i32 in 4 bytes
public int read4(int idx) { return read2(idx) | (read2(idx+2)<<16); }
public int read4() { int rez = read4(_len); _len+=4; return rez; }
// write 8 bytes
public BAOS write8( long x ) {
write4((int) x );
write4((int)(x>>32));
return this;
}
// Read a i64 in 8 bytes
public long read8(int idx) { return ((long)read4(idx) & 0xFFFFFFFFL) | ((long)read4(idx+4)<<32); }
public long read8() { long rez = read8(_len); _len+=8; return rez; }
// Write byte array
public void write( byte[] bs ) {
grow(bs.length);
System.arraycopy(bs,0,_buf,_len,bs.length);
_len += bs.length;
}
// Read/fill byte array
public byte[] read( byte[] bs ) {
System.arraycopy(_buf,_len,bs,0,bs.length);
_len += bs.length;
return bs;
}
// -----------------------------------------------------------------------
// Write a packed integer to the BAOS. Track a histogram to make more
// intelligent choices in the future.
private static final HashMap<Long,Integer> PACKED_HISTOGRAM = new HashMap<>();
private BAOS histo(long x) {
Integer cnt = PACKED_HISTOGRAM.get(x);
PACKED_HISTOGRAM.put(x,cnt==null ? 1 : cnt+1);
return this;
}
// Write a u8 (checked)
public void packed1(int x) {
assert 0 <= x && x <= 255;
histo(x).write(x);
}
public int packed1() { return read(); }
// Write a u16 (checked)
public void packed2(int x) {
assert 0 <= x && x <= 65535;
(( x <= 254 ) ? write(x) : write(0xFF).write2(x)).histo(x);
}
// Read a packed u16
public int packed2() {
int x = read();
return x==0xFF ? read2() : x;
}
// Write a packed i32
public void packed4(int x) {
((-127 <= x && x <= 127 ) ? write(x) : write(-128).write4(x)).histo(x);
}
// Read a packed i32
public int packed4() {
int x = _buf[_len++]; // Signed byte read
return x== -128 ? read4() : x;
}
// Write a packed i64
public void packed8(long x) {
((-127 <= x && x <= 127 ) ? write((int)x) : write(-128).write8(x)).histo(x);
}
// Read a packed i64
public long packed8() {
int x = _buf[_len++]; // Signed byte read
return x== -128 ? read8() : x;
}
public void packedS(String s) {
packed4(s.length());
for( int i=0; i<s.length(); i++ )
packed4(s.charAt(i));
}
public String packedS() {
int len = packed4();
String s = new String(_buf,_len,len).intern();
_len += len;
return s;
}
}
|
