DRC | cd2c5d4 | 2011-08-11 11:18:34 +0000 | [diff] [blame] | 1 | /* Copyright (C) 2002-2005 RealVNC Ltd. All Rights Reserved. |
| 2 | * Copyright (C) 2011 D. R. Commander. All Rights Reserved. |
Pierre Ossman | 4d0bc6e | 2014-02-12 13:12:31 +0100 | [diff] [blame] | 3 | * Copyright 2014 Pierre Ossman for Cendio AB |
DRC | cd2c5d4 | 2011-08-11 11:18:34 +0000 | [diff] [blame] | 4 | * |
| 5 | * This is free software; you can redistribute it and/or modify |
| 6 | * it under the terms of the GNU General Public License as published by |
| 7 | * the Free Software Foundation; either version 2 of the License, or |
| 8 | * (at your option) any later version. |
| 9 | * |
| 10 | * This software is distributed in the hope that it will be useful, |
| 11 | * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| 12 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
| 13 | * GNU General Public License for more details. |
| 14 | * |
| 15 | * You should have received a copy of the GNU General Public License |
| 16 | * along with this software; if not, write to the Free Software |
| 17 | * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, |
| 18 | * USA. |
| 19 | */ |
| 20 | |
| 21 | #include <rfb/JpegCompressor.h> |
| 22 | #include <rdr/Exception.h> |
| 23 | #include <rfb/Rect.h> |
| 24 | #include <rfb/PixelFormat.h> |
Pierre Ossman | 0d3ce87 | 2018-06-18 15:59:00 +0200 | [diff] [blame] | 25 | #include <rfb/ClientParams.h> |
DRC | cd2c5d4 | 2011-08-11 11:18:34 +0000 | [diff] [blame] | 26 | |
Pierre Ossman | 9144ae0 | 2011-09-07 11:35:04 +0000 | [diff] [blame] | 27 | #include <stdio.h> |
| 28 | extern "C" { |
| 29 | #include <jpeglib.h> |
| 30 | } |
| 31 | #include <setjmp.h> |
| 32 | |
DRC | cd2c5d4 | 2011-08-11 11:18:34 +0000 | [diff] [blame] | 33 | using namespace rfb; |
| 34 | |
| 35 | // |
Pierre Ossman | 4d0bc6e | 2014-02-12 13:12:31 +0100 | [diff] [blame] | 36 | // Special formats that libjpeg can have optimised code paths for |
| 37 | // |
| 38 | |
| 39 | static const PixelFormat pfRGBX(32, 24, false, true, 255, 255, 255, 0, 8, 16); |
| 40 | static const PixelFormat pfBGRX(32, 24, false, true, 255, 255, 255, 16, 8, 0); |
| 41 | static const PixelFormat pfXRGB(32, 24, false, true, 255, 255, 255, 8, 16, 24); |
| 42 | static const PixelFormat pfXBGR(32, 24, false, true, 255, 255, 255, 24, 16, 8); |
| 43 | |
| 44 | // |
klemens | 0536d09 | 2017-01-28 20:56:56 +0100 | [diff] [blame] | 45 | // Error manager implementation for the JPEG library |
DRC | cd2c5d4 | 2011-08-11 11:18:34 +0000 | [diff] [blame] | 46 | // |
| 47 | |
Pierre Ossman | 9144ae0 | 2011-09-07 11:35:04 +0000 | [diff] [blame] | 48 | struct JPEG_ERROR_MGR { |
| 49 | struct jpeg_error_mgr pub; |
| 50 | jmp_buf jmpBuffer; |
| 51 | char lastError[JMSG_LENGTH_MAX]; |
| 52 | }; |
| 53 | |
DRC | cd2c5d4 | 2011-08-11 11:18:34 +0000 | [diff] [blame] | 54 | static void |
| 55 | JpegErrorExit(j_common_ptr cinfo) |
| 56 | { |
| 57 | JPEG_ERROR_MGR *err = (JPEG_ERROR_MGR *)cinfo->err; |
| 58 | |
| 59 | (*cinfo->err->output_message)(cinfo); |
| 60 | longjmp(err->jmpBuffer, 1); |
| 61 | } |
| 62 | |
| 63 | static void |
| 64 | JpegOutputMessage(j_common_ptr cinfo) |
| 65 | { |
| 66 | JPEG_ERROR_MGR *err = (JPEG_ERROR_MGR *)cinfo->err; |
| 67 | |
| 68 | (*cinfo->err->format_message)(cinfo, err->lastError); |
| 69 | } |
| 70 | |
| 71 | // |
| 72 | // Destination manager implementation for the JPEG library. |
| 73 | // |
| 74 | |
Pierre Ossman | 9144ae0 | 2011-09-07 11:35:04 +0000 | [diff] [blame] | 75 | struct JPEG_DEST_MGR { |
| 76 | struct jpeg_destination_mgr pub; |
| 77 | JpegCompressor *instance; |
| 78 | }; |
| 79 | |
DRC | cd2c5d4 | 2011-08-11 11:18:34 +0000 | [diff] [blame] | 80 | static void |
| 81 | JpegInitDestination(j_compress_ptr cinfo) |
| 82 | { |
| 83 | JPEG_DEST_MGR *dest = (JPEG_DEST_MGR *)cinfo->dest; |
| 84 | JpegCompressor *jc = dest->instance; |
| 85 | |
| 86 | jc->clear(); |
| 87 | dest->pub.next_output_byte = jc->getptr(); |
| 88 | dest->pub.free_in_buffer = jc->getend() - jc->getptr(); |
| 89 | } |
| 90 | |
| 91 | static boolean |
| 92 | JpegEmptyOutputBuffer(j_compress_ptr cinfo) |
| 93 | { |
| 94 | JPEG_DEST_MGR *dest = (JPEG_DEST_MGR *)cinfo->dest; |
| 95 | JpegCompressor *jc = dest->instance; |
| 96 | |
Pierre Ossman | 1e61325 | 2015-10-16 13:03:14 +0200 | [diff] [blame] | 97 | jc->setptr(jc->getend()); |
DRC | cd2c5d4 | 2011-08-11 11:18:34 +0000 | [diff] [blame] | 98 | jc->overrun(jc->getend() - jc->getstart(), 1); |
| 99 | dest->pub.next_output_byte = jc->getptr(); |
| 100 | dest->pub.free_in_buffer = jc->getend() - jc->getptr(); |
| 101 | |
| 102 | return TRUE; |
| 103 | } |
| 104 | |
| 105 | static void |
| 106 | JpegTermDestination(j_compress_ptr cinfo) |
| 107 | { |
| 108 | JPEG_DEST_MGR *dest = (JPEG_DEST_MGR *)cinfo->dest; |
| 109 | JpegCompressor *jc = dest->instance; |
| 110 | |
| 111 | jc->setptr(dest->pub.next_output_byte); |
| 112 | } |
| 113 | |
| 114 | JpegCompressor::JpegCompressor(int bufferLen) : MemOutStream(bufferLen) |
| 115 | { |
Pierre Ossman | 9144ae0 | 2011-09-07 11:35:04 +0000 | [diff] [blame] | 116 | cinfo = new jpeg_compress_struct; |
DRC | cd2c5d4 | 2011-08-11 11:18:34 +0000 | [diff] [blame] | 117 | |
Pierre Ossman | 9144ae0 | 2011-09-07 11:35:04 +0000 | [diff] [blame] | 118 | err = new struct JPEG_ERROR_MGR; |
| 119 | cinfo->err = jpeg_std_error(&err->pub); |
| 120 | snprintf(err->lastError, JMSG_LENGTH_MAX, "No error"); |
| 121 | err->pub.error_exit = JpegErrorExit; |
| 122 | err->pub.output_message = JpegOutputMessage; |
| 123 | |
| 124 | if(setjmp(err->jmpBuffer)) { |
DRC | cd2c5d4 | 2011-08-11 11:18:34 +0000 | [diff] [blame] | 125 | // this will execute if libjpeg has an error |
Pierre Ossman | a7bbe9c | 2015-03-03 16:17:51 +0100 | [diff] [blame] | 126 | throw rdr::Exception("%s", err->lastError); |
DRC | cd2c5d4 | 2011-08-11 11:18:34 +0000 | [diff] [blame] | 127 | } |
| 128 | |
Pierre Ossman | 9144ae0 | 2011-09-07 11:35:04 +0000 | [diff] [blame] | 129 | jpeg_create_compress(cinfo); |
DRC | cd2c5d4 | 2011-08-11 11:18:34 +0000 | [diff] [blame] | 130 | |
Pierre Ossman | 9144ae0 | 2011-09-07 11:35:04 +0000 | [diff] [blame] | 131 | dest = new struct JPEG_DEST_MGR; |
| 132 | dest->pub.init_destination = JpegInitDestination; |
| 133 | dest->pub.empty_output_buffer = JpegEmptyOutputBuffer; |
| 134 | dest->pub.term_destination = JpegTermDestination; |
| 135 | dest->instance = this; |
| 136 | cinfo->dest = (struct jpeg_destination_mgr *)dest; |
DRC | cd2c5d4 | 2011-08-11 11:18:34 +0000 | [diff] [blame] | 137 | } |
| 138 | |
| 139 | JpegCompressor::~JpegCompressor(void) |
| 140 | { |
Pierre Ossman | 9144ae0 | 2011-09-07 11:35:04 +0000 | [diff] [blame] | 141 | if(setjmp(err->jmpBuffer)) { |
DRC | cd2c5d4 | 2011-08-11 11:18:34 +0000 | [diff] [blame] | 142 | // this will execute if libjpeg has an error |
| 143 | return; |
| 144 | } |
| 145 | |
Pierre Ossman | 9144ae0 | 2011-09-07 11:35:04 +0000 | [diff] [blame] | 146 | jpeg_destroy_compress(cinfo); |
| 147 | |
| 148 | delete err; |
| 149 | delete dest; |
| 150 | |
| 151 | delete cinfo; |
DRC | cd2c5d4 | 2011-08-11 11:18:34 +0000 | [diff] [blame] | 152 | } |
| 153 | |
Pierre Ossman | a10d8fe | 2014-01-22 11:28:05 +0100 | [diff] [blame] | 154 | void JpegCompressor::compress(const rdr::U8 *buf, int stride, const Rect& r, |
Pierre Ossman | b948a91 | 2014-01-15 13:23:43 +0100 | [diff] [blame] | 155 | const PixelFormat& pf, int quality, int subsamp) |
DRC | cd2c5d4 | 2011-08-11 11:18:34 +0000 | [diff] [blame] | 156 | { |
| 157 | int w = r.width(); |
| 158 | int h = r.height(); |
| 159 | int pixelsize; |
| 160 | rdr::U8 *srcBuf = NULL; |
| 161 | bool srcBufIsTemp = false; |
| 162 | JSAMPROW *rowPointer = NULL; |
| 163 | |
Pierre Ossman | 9144ae0 | 2011-09-07 11:35:04 +0000 | [diff] [blame] | 164 | if(setjmp(err->jmpBuffer)) { |
DRC | cd2c5d4 | 2011-08-11 11:18:34 +0000 | [diff] [blame] | 165 | // this will execute if libjpeg has an error |
Pierre Ossman | 9144ae0 | 2011-09-07 11:35:04 +0000 | [diff] [blame] | 166 | jpeg_abort_compress(cinfo); |
DRC | cd2c5d4 | 2011-08-11 11:18:34 +0000 | [diff] [blame] | 167 | if (srcBufIsTemp && srcBuf) delete[] srcBuf; |
| 168 | if (rowPointer) delete[] rowPointer; |
Pierre Ossman | a7bbe9c | 2015-03-03 16:17:51 +0100 | [diff] [blame] | 169 | throw rdr::Exception("%s", err->lastError); |
DRC | cd2c5d4 | 2011-08-11 11:18:34 +0000 | [diff] [blame] | 170 | } |
| 171 | |
Pierre Ossman | 9144ae0 | 2011-09-07 11:35:04 +0000 | [diff] [blame] | 172 | cinfo->image_width = w; |
| 173 | cinfo->image_height = h; |
| 174 | cinfo->in_color_space = JCS_RGB; |
DRC | cd2c5d4 | 2011-08-11 11:18:34 +0000 | [diff] [blame] | 175 | pixelsize = 3; |
| 176 | |
| 177 | #ifdef JCS_EXTENSIONS |
Pierre Ossman | 4d0bc6e | 2014-02-12 13:12:31 +0100 | [diff] [blame] | 178 | // Try to have libjpeg output directly to our native format |
| 179 | // libjpeg can only handle some "standard" formats |
| 180 | if (pfRGBX.equal(pf)) |
| 181 | cinfo->in_color_space = JCS_EXT_RGBX; |
| 182 | else if (pfBGRX.equal(pf)) |
| 183 | cinfo->in_color_space = JCS_EXT_BGRX; |
| 184 | else if (pfXRGB.equal(pf)) |
| 185 | cinfo->in_color_space = JCS_EXT_XRGB; |
| 186 | else if (pfXBGR.equal(pf)) |
| 187 | cinfo->in_color_space = JCS_EXT_XBGR; |
DRC | cd2c5d4 | 2011-08-11 11:18:34 +0000 | [diff] [blame] | 188 | |
Pierre Ossman | 4d0bc6e | 2014-02-12 13:12:31 +0100 | [diff] [blame] | 189 | if (cinfo->in_color_space != JCS_RGB) { |
| 190 | srcBuf = (rdr::U8 *)buf; |
| 191 | pixelsize = 4; |
DRC | cd2c5d4 | 2011-08-11 11:18:34 +0000 | [diff] [blame] | 192 | } |
| 193 | #endif |
| 194 | |
Pierre Ossman | a10d8fe | 2014-01-22 11:28:05 +0100 | [diff] [blame] | 195 | if (stride == 0) |
| 196 | stride = w; |
DRC | ffe09d6 | 2011-08-17 02:27:59 +0000 | [diff] [blame] | 197 | |
Pierre Ossman | 9144ae0 | 2011-09-07 11:35:04 +0000 | [diff] [blame] | 198 | if (cinfo->in_color_space == JCS_RGB) { |
DRC | cd2c5d4 | 2011-08-11 11:18:34 +0000 | [diff] [blame] | 199 | srcBuf = new rdr::U8[w * h * pixelsize]; |
| 200 | srcBufIsTemp = true; |
Pierre Ossman | a10d8fe | 2014-01-22 11:28:05 +0100 | [diff] [blame] | 201 | pf.rgbFromBuffer(srcBuf, (const rdr::U8 *)buf, w, stride, h); |
| 202 | stride = w; |
DRC | cd2c5d4 | 2011-08-11 11:18:34 +0000 | [diff] [blame] | 203 | } |
| 204 | |
Pierre Ossman | 9144ae0 | 2011-09-07 11:35:04 +0000 | [diff] [blame] | 205 | cinfo->input_components = pixelsize; |
DRC | cd2c5d4 | 2011-08-11 11:18:34 +0000 | [diff] [blame] | 206 | |
Pierre Ossman | 9144ae0 | 2011-09-07 11:35:04 +0000 | [diff] [blame] | 207 | jpeg_set_defaults(cinfo); |
Pierre Ossman | ca51953 | 2014-03-21 12:38:59 +0100 | [diff] [blame] | 208 | |
| 209 | if (quality >= 1 && quality <= 100) { |
| 210 | jpeg_set_quality(cinfo, quality, TRUE); |
| 211 | if (quality >= 96) |
| 212 | cinfo->dct_method = JDCT_ISLOW; |
| 213 | else |
| 214 | cinfo->dct_method = JDCT_FASTEST; |
| 215 | } |
DRC | cd2c5d4 | 2011-08-11 11:18:34 +0000 | [diff] [blame] | 216 | |
| 217 | switch (subsamp) { |
Pierre Ossman | b948a91 | 2014-01-15 13:23:43 +0100 | [diff] [blame] | 218 | case subsample16X: |
| 219 | case subsample8X: |
| 220 | // FIXME (fall through) |
| 221 | case subsample4X: |
Pierre Ossman | 9144ae0 | 2011-09-07 11:35:04 +0000 | [diff] [blame] | 222 | cinfo->comp_info[0].h_samp_factor = 2; |
| 223 | cinfo->comp_info[0].v_samp_factor = 2; |
DRC | cd2c5d4 | 2011-08-11 11:18:34 +0000 | [diff] [blame] | 224 | break; |
Pierre Ossman | b948a91 | 2014-01-15 13:23:43 +0100 | [diff] [blame] | 225 | case subsample2X: |
Pierre Ossman | 9144ae0 | 2011-09-07 11:35:04 +0000 | [diff] [blame] | 226 | cinfo->comp_info[0].h_samp_factor = 2; |
| 227 | cinfo->comp_info[0].v_samp_factor = 1; |
DRC | cd2c5d4 | 2011-08-11 11:18:34 +0000 | [diff] [blame] | 228 | break; |
Pierre Ossman | b948a91 | 2014-01-15 13:23:43 +0100 | [diff] [blame] | 229 | case subsampleGray: |
Pierre Ossman | 9144ae0 | 2011-09-07 11:35:04 +0000 | [diff] [blame] | 230 | jpeg_set_colorspace(cinfo, JCS_GRAYSCALE); |
DRC | cd2c5d4 | 2011-08-11 11:18:34 +0000 | [diff] [blame] | 231 | default: |
Pierre Ossman | 9144ae0 | 2011-09-07 11:35:04 +0000 | [diff] [blame] | 232 | cinfo->comp_info[0].h_samp_factor = 1; |
| 233 | cinfo->comp_info[0].v_samp_factor = 1; |
DRC | cd2c5d4 | 2011-08-11 11:18:34 +0000 | [diff] [blame] | 234 | } |
| 235 | |
| 236 | rowPointer = new JSAMPROW[h]; |
| 237 | for (int dy = 0; dy < h; dy++) |
Pierre Ossman | a10d8fe | 2014-01-22 11:28:05 +0100 | [diff] [blame] | 238 | rowPointer[dy] = (JSAMPROW)(&srcBuf[dy * stride * pixelsize]); |
DRC | cd2c5d4 | 2011-08-11 11:18:34 +0000 | [diff] [blame] | 239 | |
Pierre Ossman | 9144ae0 | 2011-09-07 11:35:04 +0000 | [diff] [blame] | 240 | jpeg_start_compress(cinfo, TRUE); |
| 241 | while (cinfo->next_scanline < cinfo->image_height) |
| 242 | jpeg_write_scanlines(cinfo, &rowPointer[cinfo->next_scanline], |
| 243 | cinfo->image_height - cinfo->next_scanline); |
DRC | cd2c5d4 | 2011-08-11 11:18:34 +0000 | [diff] [blame] | 244 | |
Pierre Ossman | 9144ae0 | 2011-09-07 11:35:04 +0000 | [diff] [blame] | 245 | jpeg_finish_compress(cinfo); |
DRC | cd2c5d4 | 2011-08-11 11:18:34 +0000 | [diff] [blame] | 246 | |
| 247 | if (srcBufIsTemp) delete[] srcBuf; |
| 248 | delete[] rowPointer; |
| 249 | } |
| 250 | |
| 251 | void JpegCompressor::writeBytes(const void* data, int length) |
| 252 | { |
| 253 | throw rdr::Exception("writeBytes() is not valid with a JpegCompressor instance. Use compress() instead."); |
| 254 | } |