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
|
5
|
27 import alice.datasegment.DataSegment;
|
|
28
|
0
|
29 import com.glavsoft.drawing.Renderer;
|
|
30 import com.glavsoft.exceptions.TransportException;
|
|
31 import com.glavsoft.transport.Reader;
|
|
32
|
|
33 public class ZRLEDecoder extends ZlibDecoder {
|
5
|
34 private static final int MAX_TILE_SIZE = 64;
|
0
|
35 private int[] decodedBitmap;
|
|
36 private int[] palette;
|
|
37
|
|
38 @Override
|
5
|
39 public void decode(Reader reader, Renderer renderer,
|
|
40 FramebufferUpdateRectangle rect) throws TransportException {
|
|
41 int zippedLength = (int) reader.readUInt32();
|
|
42 if (0 == zippedLength) return;
|
|
43 int length = rect.width * rect.height * renderer.getBytesPerPixel();
|
|
44 byte[] bytes = unzip(reader, zippedLength, length);
|
|
45 int offset = zippedLength;
|
17
|
46
|
7
|
47 AliceVNCMessage message = new AliceVNCMessage();
|
|
48 message.setRectangle(rect);
|
12
|
49 message.buf = bytes.clone();
|
7
|
50 message.offset = offset;
|
17
|
51 DataSegment.getLocal().put("aliceVNCMessage", message, null);
|
|
52 decode(renderer, rect, bytes, offset);
|
7
|
53 }
|
|
54
|
17
|
55 public void decode(Renderer renderer,
|
7
|
56 FramebufferUpdateRectangle rect, byte[] bytes, int offset)
|
|
57 throws TransportException {
|
5
|
58 int maxX = rect.x + rect.width;
|
|
59 int maxY = rect.y + rect.height;
|
0
|
60 if (null == palette) {
|
|
61 palette = new int [128];
|
|
62 }
|
|
63 if (null == decodedBitmap) {
|
|
64 decodedBitmap = new int[MAX_TILE_SIZE * MAX_TILE_SIZE];
|
|
65 }
|
5
|
66 for (int tileY = rect.y; tileY < maxY; tileY += MAX_TILE_SIZE) {
|
|
67 int tileHeight = Math.min(maxY - tileY, MAX_TILE_SIZE);
|
|
68
|
|
69 for (int tileX = rect.x; tileX < maxX; tileX += MAX_TILE_SIZE) {
|
|
70 int tileWidth = Math.min(maxX - tileX, MAX_TILE_SIZE);
|
|
71 int subencoding = bytes[offset++] & 0x0ff;
|
|
72 // 128 -plain RLE, 130-255 - Palette RLE
|
|
73 boolean isRle = (subencoding & 128) != 0;
|
|
74 // 2 to 16 for raw packed palette data, 130 to 255 for Palette RLE (subencoding - 128)
|
|
75 int paletteSize = subencoding & 127;
|
|
76 offset += readPalette(bytes, offset, renderer, paletteSize);
|
|
77 if (1 == subencoding) { // A solid tile consisting of a single colour
|
|
78 renderer.fillRect(palette[0], tileX, tileY, tileWidth, tileHeight);
|
|
79 continue;
|
|
80 }
|
|
81 if (isRle) {
|
|
82 if (0 == paletteSize) { // subencoding == 128 (or paletteSize == 0) - Plain RLE
|
|
83 offset += decodePlainRle(bytes, offset, renderer, tileX, tileY, tileWidth, tileHeight);
|
|
84 } else {
|
|
85 offset += decodePaletteRle(bytes, offset, renderer, tileX, tileY, tileWidth, tileHeight);
|
|
86 }
|
|
87 } else {
|
|
88 if (0 == paletteSize) { // subencoding == 0 (or paletteSize == 0) - raw CPIXEL data
|
|
89 offset += decodeRaw(bytes, offset, renderer, tileX, tileY, tileWidth, tileHeight);
|
|
90 } else {
|
|
91 offset += decodePacked(bytes, offset, renderer, paletteSize, tileX, tileY, tileWidth, tileHeight);
|
|
92 }
|
|
93 }
|
|
94 }
|
|
95 }
|
|
96 }
|
0
|
97
|
5
|
98 private int decodePlainRle(byte[] bytes, int offset, Renderer renderer,
|
|
99 int tileX, int tileY, int tileWidth, int tileHeight) {
|
|
100 int bytesPerCPixel = renderer.getBytesPerCPixel();
|
|
101 int decodedOffset = 0;
|
|
102 int decodedEnd = tileWidth * tileHeight;
|
|
103 int index = offset;
|
|
104 while (decodedOffset < decodedEnd) {
|
|
105 int color = renderer.getCompactPixelColor(bytes, index);
|
|
106 index += bytesPerCPixel;
|
|
107 int rlength = 1;
|
|
108 do {
|
|
109 rlength += bytes[index] & 0x0ff;
|
|
110 } while ((bytes[index++] & 0x0ff) == 255);
|
|
111 assert rlength <= decodedEnd - decodedOffset;
|
|
112 renderer.fillColorBitmapWithColor(decodedBitmap, decodedOffset, rlength, color);
|
|
113 decodedOffset += rlength;
|
|
114 }
|
|
115 renderer.drawColoredBitmap(decodedBitmap, tileX, tileY, tileWidth, tileHeight);
|
|
116 return index - offset;
|
|
117 }
|
0
|
118
|
5
|
119 private int decodePaletteRle(byte[] bytes, int offset, Renderer renderer,
|
|
120 int tileX, int tileY, int tileWidth, int tileHeight) {
|
|
121 int decodedOffset = 0;
|
|
122 int decodedEnd = tileWidth * tileHeight;
|
|
123 int index = offset;
|
|
124 while (decodedOffset < decodedEnd) {
|
|
125 int colorIndex = bytes[index++];
|
|
126 int color = palette[colorIndex & 127];
|
|
127 int rlength = 1;
|
|
128 if ((colorIndex & 128) != 0) {
|
|
129 do {
|
|
130 rlength += bytes[index] & 0x0ff;
|
|
131 } while (bytes[index++] == (byte) 255);
|
|
132 }
|
|
133 assert rlength <= decodedEnd - decodedOffset;
|
|
134 renderer.fillColorBitmapWithColor(decodedBitmap, decodedOffset, rlength, color);
|
|
135 decodedOffset += rlength;
|
|
136 }
|
|
137 renderer.drawColoredBitmap(decodedBitmap, tileX, tileY, tileWidth, tileHeight);
|
|
138 return index - offset;
|
|
139 }
|
0
|
140
|
5
|
141 private int decodePacked(byte[] bytes, int offset, Renderer renderer,
|
|
142 int paletteSize, int tileX, int tileY, int tileWidth, int tileHeight) {
|
|
143 int bitsPerPalletedPixel = paletteSize > 16 ? 8 : paletteSize > 4 ? 4
|
|
144 : paletteSize > 2 ? 2 : 1;
|
|
145 int packedOffset = offset;
|
|
146 int decodedOffset = 0;
|
|
147 for (int i = 0; i < tileHeight; ++i) {
|
|
148 int decodedRowEnd = decodedOffset + tileWidth;
|
|
149 int byteProcessed = 0;
|
|
150 int bitsRemain = 0;
|
|
151
|
|
152 while (decodedOffset < decodedRowEnd) {
|
|
153 if (bitsRemain == 0) {
|
|
154 byteProcessed = bytes[packedOffset++];
|
|
155 bitsRemain = 8;
|
|
156 }
|
|
157 bitsRemain -= bitsPerPalletedPixel;
|
|
158 int index = byteProcessed >> bitsRemain & (1 << bitsPerPalletedPixel) - 1 & 127;
|
|
159 int color = palette[index];
|
|
160 renderer.fillColorBitmapWithColor(decodedBitmap, decodedOffset, 1, color);
|
|
161 ++decodedOffset;
|
|
162 }
|
|
163 }
|
|
164 renderer.drawColoredBitmap(decodedBitmap, tileX, tileY, tileWidth, tileHeight);
|
|
165 return packedOffset - offset;
|
|
166 }
|
|
167
|
|
168 private int decodeRaw(byte[] bytes, int offset, Renderer renderer,
|
|
169 int tileX, int tileY, int tileWidth, int tileHeight) throws TransportException {
|
|
170 return renderer.drawCompactBytes(bytes, offset, tileX, tileY, tileWidth, tileHeight);
|
|
171 }
|
0
|
172
|
5
|
173 private int readPalette(byte[] bytes, int offset, Renderer renderer, int paletteSize) {
|
|
174 final int bytesPerCPixel = renderer.getBytesPerCPixel();
|
|
175 for (int i=0; i<paletteSize; ++i) {
|
|
176 palette[i] = renderer.getCompactPixelColor(bytes, offset + i* bytesPerCPixel);
|
|
177 }
|
|
178 return paletteSize * bytesPerCPixel;
|
|
179 }
|
0
|
180 }
|