summaryrefslogtreecommitdiff
path: root/src/main/java/com/encrox/instancedregions/chunkmap/ChunkWriter.java
blob: 398cb25f7cf76a59a338761d0bbd86428aa573ff (plain)
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
/**
 * @author Aleksey Terzi
 *
 */

package com.encrox.instancedregions.chunkmap;

import java.io.IOException;

public class ChunkWriter {
	private byte[] data;
	private int bitsPerBlock;
	private int byteIndex;
	private int bitIndex;
	private long buffer;
	
	public ChunkWriter(byte[] data) {
		this.data = data;
	}
	
	public int getByteIndex() {
		return this.byteIndex;
	}
	
	public void init() {
		this.byteIndex = 0;
		this.bitIndex = 0;
		this.buffer = 0;
	}
	
	public void setBitsPerBlock(int bitsPerBlock) {
		this.bitsPerBlock = bitsPerBlock;
	}
	
	public void save() throws IOException {
		writeLong();
	}
	
	public void skip(int count) {
		this.byteIndex += count;
		this.bitIndex = 0;
	}
	
	public void writeBytes(byte[] source, int index, int length) throws IOException {
		if(this.byteIndex + length > this.data.length) {
			throw new IOException("No space to write.");
		}
		
		System.arraycopy(source, index, this.data, this.byteIndex, length);
		
		this.byteIndex += length;
	}
	
	public void writeBlockBits(long bits) throws IOException {
		if(this.bitIndex >= 64) {
			writeLong();
			this.bitIndex = 0;
		}
		
		int leftBits = 64 - this.bitIndex;
		
		this.buffer |= bits << this.bitIndex; 
		
		if(leftBits >= this.bitsPerBlock) {
			this.bitIndex += this.bitsPerBlock;
		} else {
			writeLong();
			
			this.buffer = bits >>> leftBits;
			
			this.bitIndex = this.bitsPerBlock - leftBits;
		}
	}
	
	private void writeLong() throws IOException {
		if(this.byteIndex + 7 >= this.data.length) {
			throw new IOException("No space to write.");
		}
		
		this.data[this.byteIndex++] = (byte)(this.buffer >> 56);
		this.data[this.byteIndex++] = (byte)(this.buffer >> 48);
		this.data[this.byteIndex++] = (byte)(this.buffer >> 40);
		this.data[this.byteIndex++] = (byte)(this.buffer >> 32);
		this.data[this.byteIndex++] = (byte)(this.buffer >> 24);
		this.data[this.byteIndex++] = (byte)(this.buffer >> 16);
		this.data[this.byteIndex++] = (byte)(this.buffer >> 8);
		this.data[this.byteIndex++] = (byte)this.buffer;
		
		this.buffer = 0;
	}
	
	public void writeVarInt(int value) throws IOException {
		while((value & ~0x7F) != 0) {
            writeByte((value & 0x7F) | 0x80);
            value >>>= 7;
        }

        writeByte(value);
    }
	
	public void writeByte(int value) throws IOException {
		if(this.byteIndex >= this.data.length) {
			throw new IOException("No space to write.");
		}
		
		this.data[this.byteIndex++] = (byte)value;
	}
}