qemu

FORK: QEMU emulator
git clone https://git.neptards.moe/neptards/qemu.git
Log | Files | Refs | Submodules | LICENSE

vnc-enc-zrle.c.inc (7132B)

      1 /*
      2  * QEMU VNC display driver: Zlib Run-length Encoding (ZRLE)
      3  *
      4  * From libvncserver/libvncserver/zrleencodetemplate.c
      5  * Copyright (C) 2002 RealVNC Ltd.  All Rights Reserved.
      6  * Copyright (C) 2003 Sun Microsystems, Inc.
      7  *
      8  * Copyright (C) 2010 Corentin Chary <corentin.chary@gmail.com>
      9  *
     10  * This work is licensed under the terms of the GNU GPL, version 2 or later.
     11  * See the COPYING file in the top-level directory.
     12  */
     13 
     14 /*
     15  * Before including this file, you must define a number of CPP macros.
     16  *
     17  * ZRLE_BPP should be 8, 16 or 32 depending on the bits per pixel.
     18  *
     19  * Note that the buf argument to ZRLE_ENCODE needs to be at least one pixel
     20  * bigger than the largest tile of pixel data, since the ZRLE encoding
     21  * algorithm writes to the position one past the end of the pixel data.
     22  */
     23 
     24 
     25 #include "qemu/osdep.h"
     26 
     27 #undef ZRLE_ENDIAN_SUFFIX
     28 
     29 #if ZYWRLE_ENDIAN == ENDIAN_LITTLE
     30 #define ZRLE_ENDIAN_SUFFIX le
     31 #elif ZYWRLE_ENDIAN == ENDIAN_BIG
     32 #define ZRLE_ENDIAN_SUFFIX be
     33 #else
     34 #define ZRLE_ENDIAN_SUFFIX ne
     35 #endif
     36 
     37 #ifndef ZRLE_CONCAT
     38 #define ZRLE_CONCAT_I(a, b)    a##b
     39 #define ZRLE_CONCAT2(a, b)     ZRLE_CONCAT_I(a, b)
     40 #define ZRLE_CONCAT3(a, b, c)  ZRLE_CONCAT2(a, ZRLE_CONCAT2(b, c))
     41 #endif
     42 
     43 #ifdef ZRLE_COMPACT_PIXEL
     44 #define ZRLE_ENCODE_SUFFIX   ZRLE_CONCAT2(ZRLE_COMPACT_PIXEL,ZRLE_ENDIAN_SUFFIX)
     45 #define ZRLE_WRITE_SUFFIX    ZRLE_COMPACT_PIXEL
     46 #define ZRLE_PIXEL           ZRLE_CONCAT3(uint,ZRLE_BPP,_t)
     47 #define ZRLE_BPP_OUT         24
     48 #elif ZRLE_BPP == 15
     49 #define ZRLE_ENCODE_SUFFIX   ZRLE_CONCAT2(ZRLE_BPP,ZRLE_ENDIAN_SUFFIX)
     50 #define ZRLE_WRITE_SUFFIX    16
     51 #define ZRLE_PIXEL           uint16_t
     52 #define ZRLE_BPP_OUT         16
     53 #else
     54 #define ZRLE_ENCODE_SUFFIX   ZRLE_CONCAT2(ZRLE_BPP,ZRLE_ENDIAN_SUFFIX)
     55 #define ZRLE_WRITE_SUFFIX    ZRLE_BPP
     56 #define ZRLE_BPP_OUT         ZRLE_BPP
     57 #define ZRLE_PIXEL           ZRLE_CONCAT3(uint,ZRLE_BPP,_t)
     58 #endif
     59 
     60 #define ZRLE_WRITE_PIXEL     ZRLE_CONCAT2(zrle_write_u,       ZRLE_WRITE_SUFFIX)
     61 #define ZRLE_ENCODE          ZRLE_CONCAT2(zrle_encode_,      ZRLE_ENCODE_SUFFIX)
     62 #define ZRLE_ENCODE_TILE     ZRLE_CONCAT2(zrle_encode_tile,  ZRLE_ENCODE_SUFFIX)
     63 #define ZRLE_WRITE_PALETTE   ZRLE_CONCAT2(zrle_write_palette,ZRLE_ENCODE_SUFFIX)
     64 
     65 static void ZRLE_ENCODE_TILE(VncState *vs, ZRLE_PIXEL *data, int w, int h,
     66                              int zywrle_level);
     67 
     68 #if ZRLE_BPP != 8
     69 #include "vnc-enc-zywrle-template.c"
     70 #endif
     71 
     72 
     73 static void ZRLE_ENCODE(VncState *vs, int x, int y, int w, int h,
     74                         int zywrle_level)
     75 {
     76     int ty;
     77 
     78     for (ty = y; ty < y + h; ty += VNC_ZRLE_TILE_HEIGHT) {
     79 
     80         int tx, th;
     81 
     82         th = MIN(VNC_ZRLE_TILE_HEIGHT, y + h - ty);
     83 
     84         for (tx = x; tx < x + w; tx += VNC_ZRLE_TILE_WIDTH) {
     85             int tw;
     86             ZRLE_PIXEL *buf;
     87 
     88             tw = MIN(VNC_ZRLE_TILE_WIDTH, x + w - tx);
     89 
     90             buf = zrle_convert_fb(vs, tx, ty, tw, th, ZRLE_BPP);
     91             ZRLE_ENCODE_TILE(vs, buf, tw, th, zywrle_level);
     92         }
     93     }
     94 }
     95 
     96 static void ZRLE_ENCODE_TILE(VncState *vs, ZRLE_PIXEL *data, int w, int h,
     97                              int zywrle_level)
     98 {
     99     VncPalette *palette = &vs->zrle->palette;
    100 
    101     int runs = 0;
    102     int single_pixels = 0;
    103 
    104     bool use_rle;
    105     bool use_palette;
    106 
    107     int i;
    108 
    109     ZRLE_PIXEL *ptr = data;
    110     ZRLE_PIXEL *end = ptr + h * w;
    111     *end = ~*(end-1); /* one past the end is different so the while loop ends */
    112 
    113     /* Real limit is 127 but we wan't a way to know if there is more than 127 */
    114     palette_init(palette, 256, ZRLE_BPP);
    115 
    116     while (ptr < end) {
    117         ZRLE_PIXEL pix = *ptr;
    118         if (*++ptr != pix) { /* FIXME */
    119             single_pixels++;
    120         } else {
    121             while (*++ptr == pix) ;
    122             runs++;
    123         }
    124         palette_put(palette, pix);
    125     }
    126 
    127     /* Solid tile is a special case */
    128 
    129     if (palette_size(palette) == 1) {
    130         bool found;
    131 
    132         vnc_write_u8(vs, 1);
    133         ZRLE_WRITE_PIXEL(vs, palette_color(palette, 0, &found));
    134         return;
    135     }
    136 
    137     zrle_choose_palette_rle(vs, w, h, palette, ZRLE_BPP_OUT,
    138                             runs, single_pixels, zywrle_level,
    139                             &use_rle, &use_palette);
    140 
    141     if (!use_palette) {
    142         vnc_write_u8(vs, (use_rle ? 128 : 0));
    143     } else {
    144         uint32_t colors[VNC_PALETTE_MAX_SIZE];
    145         size_t size = palette_size(palette);
    146 
    147         vnc_write_u8(vs, (use_rle ? 128 : 0) | size);
    148         palette_fill(palette, colors);
    149 
    150         for (i = 0; i < size; i++) {
    151             ZRLE_WRITE_PIXEL(vs, colors[i]);
    152         }
    153     }
    154 
    155     if (use_rle) {
    156         ZRLE_PIXEL *ptr = data;
    157         ZRLE_PIXEL *end = ptr + w * h;
    158         ZRLE_PIXEL *run_start;
    159         ZRLE_PIXEL pix;
    160 
    161         while (ptr < end) {
    162             int len;
    163             int index = 0;
    164 
    165             run_start = ptr;
    166             pix = *ptr++;
    167 
    168             while (*ptr == pix && ptr < end) {
    169                 ptr++;
    170             }
    171 
    172             len = ptr - run_start;
    173 
    174             if (use_palette)
    175                 index = palette_idx(palette, pix);
    176 
    177             if (len <= 2 && use_palette) {
    178                 if (len == 2) {
    179                     vnc_write_u8(vs, index);
    180                 }
    181                 vnc_write_u8(vs, index);
    182                 continue;
    183             }
    184             if (use_palette) {
    185                 vnc_write_u8(vs, index | 128);
    186             } else {
    187                 ZRLE_WRITE_PIXEL(vs, pix);
    188             }
    189 
    190             len -= 1;
    191 
    192             while (len >= 255) {
    193                 vnc_write_u8(vs, 255);
    194                 len -= 255;
    195             }
    196 
    197             vnc_write_u8(vs, len);
    198         }
    199     } else if (use_palette) { /* no RLE */
    200         int bppp;
    201         ZRLE_PIXEL *ptr = data;
    202 
    203         /* packed pixels */
    204 
    205         assert (palette_size(palette) < 17);
    206 
    207         bppp = bits_per_packed_pixel[palette_size(palette)-1];
    208 
    209         for (i = 0; i < h; i++) {
    210             uint8_t nbits = 0;
    211             uint8_t byte = 0;
    212 
    213             ZRLE_PIXEL *eol = ptr + w;
    214 
    215             while (ptr < eol) {
    216                 ZRLE_PIXEL pix = *ptr++;
    217                 uint8_t index = palette_idx(palette, pix);
    218 
    219                 byte = (byte << bppp) | index;
    220                 nbits += bppp;
    221                 if (nbits >= 8) {
    222                     vnc_write_u8(vs, byte);
    223                     nbits = 0;
    224                 }
    225             }
    226             if (nbits > 0) {
    227                 byte <<= 8 - nbits;
    228                 vnc_write_u8(vs, byte);
    229             }
    230         }
    231     } else {
    232 
    233         /* raw */
    234 
    235 #if ZRLE_BPP != 8
    236         if (zywrle_level > 0 && !(zywrle_level & 0x80)) {
    237             ZYWRLE_ANALYZE(data, data, w, h, w, zywrle_level, vs->zywrle.buf);
    238             ZRLE_ENCODE_TILE(vs, data, w, h, zywrle_level | 0x80);
    239         }
    240         else
    241 #endif
    242         {
    243 #ifdef ZRLE_COMPACT_PIXEL
    244             ZRLE_PIXEL *ptr;
    245 
    246             for (ptr = data; ptr < data + w * h; ptr++) {
    247                 ZRLE_WRITE_PIXEL(vs, *ptr);
    248             }
    249 #else
    250             vnc_write(vs, data, w * h * (ZRLE_BPP / 8));
    251 #endif
    252         }
    253     }
    254 }
    255 
    256 #undef ZRLE_PIXEL
    257 #undef ZRLE_WRITE_PIXEL
    258 #undef ZRLE_ENCODE
    259 #undef ZRLE_ENCODE_TILE
    260 #undef ZYWRLE_ENCODE_TILE
    261 #undef ZRLE_BPP_OUT
    262 #undef ZRLE_WRITE_SUFFIX
    263 #undef ZRLE_ENCODE_SUFFIX