mirror of
https://github.com/TigerVNC/tigervnc.git
synced 2024-08-21 17:54:34 +02:00
47ed8d321c
git-svn-id: svn://svn.code.sf.net/p/tigervnc/code/trunk@2 3789f03b-4d11-0410-bbf8-ca57d06f2519
123 lines
3.4 KiB
C++
123 lines
3.4 KiB
C++
/* Copyright (C) 2002-2004 RealVNC Ltd. All Rights Reserved.
|
|
*
|
|
* This is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation; either version 2 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This software is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this software; if not, write to the Free Software
|
|
* Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307,
|
|
* USA.
|
|
*/
|
|
#include <rdr/OutStream.h>
|
|
#include <rfb/Exception.h>
|
|
#include <rfb/ImageGetter.h>
|
|
#include <rfb/encodings.h>
|
|
#include <rfb/ConnParams.h>
|
|
#include <rfb/SMsgWriter.h>
|
|
#include <rfb/ZRLEEncoder.h>
|
|
#include <rfb/Configuration.h>
|
|
|
|
using namespace rfb;
|
|
|
|
rdr::MemOutStream* ZRLEEncoder::sharedMos = 0;
|
|
int ZRLEEncoder::maxLen = 513 * 1024; // enough for width 2048 32-bit pixels
|
|
|
|
IntParameter zlibLevel("ZlibLevel","Zlib compression level",-1);
|
|
|
|
#define EXTRA_ARGS ImageGetter* ig
|
|
#define GET_IMAGE_INTO_BUF(r,buf) ig->getImage(buf, r);
|
|
#define BPP 8
|
|
#include <rfb/zrleEncode.h>
|
|
#undef BPP
|
|
#define BPP 16
|
|
#include <rfb/zrleEncode.h>
|
|
#undef BPP
|
|
#define BPP 32
|
|
#include <rfb/zrleEncode.h>
|
|
#define CPIXEL 24A
|
|
#include <rfb/zrleEncode.h>
|
|
#undef CPIXEL
|
|
#define CPIXEL 24B
|
|
#include <rfb/zrleEncode.h>
|
|
#undef CPIXEL
|
|
#undef BPP
|
|
|
|
Encoder* ZRLEEncoder::create(SMsgWriter* writer)
|
|
{
|
|
return new ZRLEEncoder(writer);
|
|
}
|
|
|
|
ZRLEEncoder::ZRLEEncoder(SMsgWriter* writer_)
|
|
: writer(writer_), zos(0,0,zlibLevel)
|
|
{
|
|
if (sharedMos)
|
|
mos = sharedMos;
|
|
else
|
|
mos = new rdr::MemOutStream(129*1024);
|
|
}
|
|
|
|
ZRLEEncoder::~ZRLEEncoder()
|
|
{
|
|
if (!sharedMos)
|
|
delete mos;
|
|
}
|
|
|
|
bool ZRLEEncoder::writeRect(const Rect& r, ImageGetter* ig, Rect* actual)
|
|
{
|
|
rdr::U8* imageBuf = writer->getImageBuf(64 * 64 * 4 + 4);
|
|
mos->clear();
|
|
bool wroteAll = true;
|
|
*actual = r;
|
|
|
|
switch (writer->bpp()) {
|
|
case 8:
|
|
wroteAll = zrleEncode8(r, mos, &zos, imageBuf, maxLen, actual, ig);
|
|
break;
|
|
case 16:
|
|
wroteAll = zrleEncode16(r, mos, &zos, imageBuf, maxLen, actual, ig);
|
|
break;
|
|
case 32:
|
|
{
|
|
const PixelFormat& pf = writer->getConnParams()->pf();
|
|
|
|
bool fitsInLS3Bytes = ((pf.redMax << pf.redShift) < (1<<24) &&
|
|
(pf.greenMax << pf.greenShift) < (1<<24) &&
|
|
(pf.blueMax << pf.blueShift) < (1<<24));
|
|
|
|
bool fitsInMS3Bytes = (pf.redShift > 7 &&
|
|
pf.greenShift > 7 &&
|
|
pf.blueShift > 7);
|
|
|
|
if ((fitsInLS3Bytes && !pf.bigEndian) ||
|
|
(fitsInMS3Bytes && pf.bigEndian))
|
|
{
|
|
wroteAll = zrleEncode24A(r, mos, &zos, imageBuf, maxLen, actual, ig);
|
|
}
|
|
else if ((fitsInLS3Bytes && pf.bigEndian) ||
|
|
(fitsInMS3Bytes && !pf.bigEndian))
|
|
{
|
|
wroteAll = zrleEncode24B(r, mos, &zos, imageBuf, maxLen, actual, ig);
|
|
}
|
|
else
|
|
{
|
|
wroteAll = zrleEncode32(r, mos, &zos, imageBuf, maxLen, actual, ig);
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
|
|
writer->startRect(*actual, encodingZRLE);
|
|
rdr::OutStream* os = writer->getOutStream();
|
|
os->writeU32(mos->length());
|
|
os->writeBytes(mos->data(), mos->length());
|
|
writer->endRect();
|
|
return wroteAll;
|
|
}
|