0
|
1 // Copyright (C) 2010, 2011, 2012, 2013 GlavSoft LLC.
|
|
2 // All rights reserved.
|
|
3 //
|
|
4 //-------------------------------------------------------------------------
|
|
5 // This file is part of the TightVNC software. Please visit our Web site:
|
|
6 //
|
|
7 // http://www.tightvnc.com/
|
|
8 //
|
|
9 // This program is free software; you can redistribute it and/or modify
|
|
10 // it under the terms of the GNU General Public License as published by
|
|
11 // the Free Software Foundation; either version 2 of the License, or
|
|
12 // (at your option) any later version.
|
|
13 //
|
|
14 // This program is distributed in the hope that it will be useful,
|
|
15 // but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
16 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
17 // GNU General Public License for more details.
|
|
18 //
|
|
19 // You should have received a copy of the GNU General Public License along
|
|
20 // with this program; if not, write to the Free Software Foundation, Inc.,
|
|
21 // 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
22 //-------------------------------------------------------------------------
|
|
23 //
|
|
24
|
|
25 package com.glavsoft.rfb.encoding.decoder;
|
|
26
|
20
|
27 import java.util.zip.DataFormatException;
|
|
28 import java.util.zip.Inflater;
|
|
29
|
5
|
30 import alice.datasegment.DataSegment;
|
18
|
31 import alice.datasegment.ReceiveData;
|
5
|
32
|
0
|
33 import com.glavsoft.drawing.Renderer;
|
|
34 import com.glavsoft.exceptions.TransportException;
|
20
|
35 import com.glavsoft.rfb.encoding.EncodingType;
|
0
|
36 import com.glavsoft.transport.Reader;
|
|
37
|
|
38 public class ZRLEDecoder extends ZlibDecoder {
|
5
|
39 private static final int MAX_TILE_SIZE = 64;
|
0
|
40 private int[] decodedBitmap;
|
|
41 private int[] palette;
|
|
42
|
|
43 @Override
|
5
|
44 public void decode(Reader reader, Renderer renderer,
|
|
45 FramebufferUpdateRectangle rect) throws TransportException {
|
|
46 int zippedLength = (int) reader.readUInt32();
|
|
47 if (0 == zippedLength) return;
|
|
48 int length = rect.width * rect.height * renderer.getBytesPerPixel();
|
|
49 byte[] bytes = unzip(reader, zippedLength, length);
|
17
|
50
|
7
|
51 AliceVNCMessage message = new AliceVNCMessage();
|
|
52 message.setRectangle(rect);
|
20
|
53 message.encodingTypeId = EncodingType.ZRLEE.getId();
|
18
|
54
|
19
|
55 ReceiveData rData = new ReceiveData(bytes.clone(), false, false);
|
|
56 DataSegment.getLocal().put("pixelByteArray", rData, null);
|
|
57 rData = new ReceiveData(message, false, false);
|
18
|
58 DataSegment.getLocal().put("aliceVNCMessage", rData, null);
|
20
|
59 decode(renderer, rect, bytes, 0);
|
7
|
60 }
|
|
61
|
17
|
62 public void decode(Renderer renderer,
|
7
|
63 FramebufferUpdateRectangle rect, byte[] bytes, int offset)
|
|
64 throws TransportException {
|
5
|
65 int maxX = rect.x + rect.width;
|
|
66 int maxY = rect.y + rect.height;
|
0
|
67 if (null == palette) {
|
|
68 palette = new int [128];
|
|
69 }
|
|
70 if (null == decodedBitmap) {
|
|
71 decodedBitmap = new int[MAX_TILE_SIZE * MAX_TILE_SIZE];
|
|
72 }
|
5
|
73 for (int tileY = rect.y; tileY < maxY; tileY += MAX_TILE_SIZE) {
|
|
74 int tileHeight = Math.min(maxY - tileY, MAX_TILE_SIZE);
|
|
75
|
|
76 for (int tileX = rect.x; tileX < maxX; tileX += MAX_TILE_SIZE) {
|
|
77 int tileWidth = Math.min(maxX - tileX, MAX_TILE_SIZE);
|
|
78 int subencoding = bytes[offset++] & 0x0ff;
|
|
79 // 128 -plain RLE, 130-255 - Palette RLE
|
|
80 boolean isRle = (subencoding & 128) != 0;
|
|
81 // 2 to 16 for raw packed palette data, 130 to 255 for Palette RLE (subencoding - 128)
|
|
82 int paletteSize = subencoding & 127;
|
|
83 offset += readPalette(bytes, offset, renderer, paletteSize);
|
|
84 if (1 == subencoding) { // A solid tile consisting of a single colour
|
|
85 renderer.fillRect(palette[0], tileX, tileY, tileWidth, tileHeight);
|
|
86 continue;
|
|
87 }
|
|
88 if (isRle) {
|
|
89 if (0 == paletteSize) { // subencoding == 128 (or paletteSize == 0) - Plain RLE
|
|
90 offset += decodePlainRle(bytes, offset, renderer, tileX, tileY, tileWidth, tileHeight);
|
|
91 } else {
|
|
92 offset += decodePaletteRle(bytes, offset, renderer, tileX, tileY, tileWidth, tileHeight);
|
|
93 }
|
|
94 } else {
|
|
95 if (0 == paletteSize) { // subencoding == 0 (or paletteSize == 0) - raw CPIXEL data
|
|
96 offset += decodeRaw(bytes, offset, renderer, tileX, tileY, tileWidth, tileHeight);
|
|
97 } else {
|
|
98 offset += decodePacked(bytes, offset, renderer, paletteSize, tileX, tileY, tileWidth, tileHeight);
|
|
99 }
|
|
100 }
|
|
101 }
|
|
102 }
|
|
103 }
|
0
|
104
|
5
|
105 private int decodePlainRle(byte[] bytes, int offset, Renderer renderer,
|
|
106 int tileX, int tileY, int tileWidth, int tileHeight) {
|
|
107 int bytesPerCPixel = renderer.getBytesPerCPixel();
|
|
108 int decodedOffset = 0;
|
|
109 int decodedEnd = tileWidth * tileHeight;
|
|
110 int index = offset;
|
|
111 while (decodedOffset < decodedEnd) {
|
|
112 int color = renderer.getCompactPixelColor(bytes, index);
|
|
113 index += bytesPerCPixel;
|
|
114 int rlength = 1;
|
|
115 do {
|
|
116 rlength += bytes[index] & 0x0ff;
|
|
117 } while ((bytes[index++] & 0x0ff) == 255);
|
|
118 assert rlength <= decodedEnd - decodedOffset;
|
|
119 renderer.fillColorBitmapWithColor(decodedBitmap, decodedOffset, rlength, color);
|
|
120 decodedOffset += rlength;
|
|
121 }
|
|
122 renderer.drawColoredBitmap(decodedBitmap, tileX, tileY, tileWidth, tileHeight);
|
|
123 return index - offset;
|
|
124 }
|
0
|
125
|
5
|
126 private int decodePaletteRle(byte[] bytes, int offset, Renderer renderer,
|
|
127 int tileX, int tileY, int tileWidth, int tileHeight) {
|
|
128 int decodedOffset = 0;
|
|
129 int decodedEnd = tileWidth * tileHeight;
|
|
130 int index = offset;
|
|
131 while (decodedOffset < decodedEnd) {
|
|
132 int colorIndex = bytes[index++];
|
|
133 int color = palette[colorIndex & 127];
|
|
134 int rlength = 1;
|
|
135 if ((colorIndex & 128) != 0) {
|
|
136 do {
|
|
137 rlength += bytes[index] & 0x0ff;
|
|
138 } while (bytes[index++] == (byte) 255);
|
|
139 }
|
|
140 assert rlength <= decodedEnd - decodedOffset;
|
|
141 renderer.fillColorBitmapWithColor(decodedBitmap, decodedOffset, rlength, color);
|
|
142 decodedOffset += rlength;
|
|
143 }
|
|
144 renderer.drawColoredBitmap(decodedBitmap, tileX, tileY, tileWidth, tileHeight);
|
|
145 return index - offset;
|
|
146 }
|
0
|
147
|
5
|
148 private int decodePacked(byte[] bytes, int offset, Renderer renderer,
|
|
149 int paletteSize, int tileX, int tileY, int tileWidth, int tileHeight) {
|
|
150 int bitsPerPalletedPixel = paletteSize > 16 ? 8 : paletteSize > 4 ? 4
|
|
151 : paletteSize > 2 ? 2 : 1;
|
|
152 int packedOffset = offset;
|
|
153 int decodedOffset = 0;
|
|
154 for (int i = 0; i < tileHeight; ++i) {
|
|
155 int decodedRowEnd = decodedOffset + tileWidth;
|
|
156 int byteProcessed = 0;
|
|
157 int bitsRemain = 0;
|
|
158
|
|
159 while (decodedOffset < decodedRowEnd) {
|
|
160 if (bitsRemain == 0) {
|
|
161 byteProcessed = bytes[packedOffset++];
|
|
162 bitsRemain = 8;
|
|
163 }
|
|
164 bitsRemain -= bitsPerPalletedPixel;
|
|
165 int index = byteProcessed >> bitsRemain & (1 << bitsPerPalletedPixel) - 1 & 127;
|
|
166 int color = palette[index];
|
|
167 renderer.fillColorBitmapWithColor(decodedBitmap, decodedOffset, 1, color);
|
|
168 ++decodedOffset;
|
|
169 }
|
|
170 }
|
|
171 renderer.drawColoredBitmap(decodedBitmap, tileX, tileY, tileWidth, tileHeight);
|
|
172 return packedOffset - offset;
|
|
173 }
|
|
174
|
|
175 private int decodeRaw(byte[] bytes, int offset, Renderer renderer,
|
|
176 int tileX, int tileY, int tileWidth, int tileHeight) throws TransportException {
|
|
177 return renderer.drawCompactBytes(bytes, offset, tileX, tileY, tileWidth, tileHeight);
|
|
178 }
|
0
|
179
|
5
|
180 private int readPalette(byte[] bytes, int offset, Renderer renderer, int paletteSize) {
|
|
181 final int bytesPerCPixel = renderer.getBytesPerCPixel();
|
|
182 for (int i=0; i<paletteSize; ++i) {
|
|
183 palette[i] = renderer.getCompactPixelColor(bytes, offset + i* bytesPerCPixel);
|
|
184 }
|
|
185 return paletteSize * bytesPerCPixel;
|
|
186 }
|
20
|
187
|
|
188 @Override
|
|
189 protected byte[] unzip(Reader reader, int zippedLength, int length)
|
|
190 throws TransportException {
|
|
191 byte[] input = ByteBuffer.getInstance().getBuffer(zippedLength);
|
|
192 byte[] bytes = new byte[length];
|
|
193 reader.readBytes(input, 0, zippedLength);
|
|
194 if (null == decoder) {
|
|
195 decoder = new Inflater();
|
|
196 }
|
|
197 decoder.setInput(input, 0, zippedLength);
|
|
198 int position = 0;
|
|
199 try {
|
|
200 do {
|
|
201 int len = decoder.inflate(bytes, position, bytes.length - position);
|
|
202 if (len > 0) {
|
|
203 position += len;
|
|
204 }
|
|
205 } while (!decoder.needsInput());
|
|
206 } catch (DataFormatException e) {
|
|
207 throw new TransportException("cannot inflate Zlib data", e);
|
|
208 }
|
|
209 return bytes;
|
|
210 }
|
0
|
211 }
|