diff options
Diffstat (limited to 'libxrdp/xrdp_jpeg_compress.c')
| -rw-r--r-- | libxrdp/xrdp_jpeg_compress.c | 225 |
1 files changed, 225 insertions, 0 deletions
diff --git a/libxrdp/xrdp_jpeg_compress.c b/libxrdp/xrdp_jpeg_compress.c new file mode 100644 index 00000000..48f12359 --- /dev/null +++ b/libxrdp/xrdp_jpeg_compress.c @@ -0,0 +1,225 @@ +/* + This program 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 program 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 program; if not, write to the Free Software + Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. + + xrdp: A Remote Desktop Protocol server. + Copyright (C) Jay Sorg 2012 + + jpeg compressor + +*/ + +#include "libxrdp.h" + +#if defined(XRDP_JPEG) + +#include <stdio.h> +#include <stdlib.h> +#include <string.h> +#include <jpeglib.h> + +#define JP_QUALITY 75 + +struct mydata_comp +{ + char* cb; + int cb_bytes; + int total_done; + int overwrite; +}; + +/*****************************************************************************/ +/* called at begining */ +static void DEFAULT_CC +my_init_destination(j_compress_ptr cinfo) +{ + struct mydata_comp* md; + + md = (struct mydata_comp*)(cinfo->client_data); + md->total_done = 0; + md->overwrite = 0; + cinfo->dest->next_output_byte = md->cb; + cinfo->dest->free_in_buffer = md->cb_bytes; +} + +/*****************************************************************************/ +/* called when buffer is full and we need more space */ +static int DEFAULT_CC +my_empty_output_buffer(j_compress_ptr cinfo) +{ + struct mydata_comp* md; + int chunk_bytes; + + md = (struct mydata_comp*)(cinfo->client_data); + chunk_bytes = md->cb_bytes; + md->total_done += chunk_bytes; + cinfo->dest->next_output_byte = md->cb; + cinfo->dest->free_in_buffer = md->cb_bytes; + md->overwrite = 1; + return 1; +} + +/*****************************************************************************/ +/* called at end */ +static void DEFAULT_CC +my_term_destination(j_compress_ptr cinfo) +{ + struct mydata_comp* md; + int chunk_bytes; + + md = (struct mydata_comp*)(cinfo->client_data); + chunk_bytes = md->cb_bytes - cinfo->dest->free_in_buffer; + md->total_done += chunk_bytes; +} + +/*****************************************************************************/ +static int APP_CC +jp_do_compress(char* data, int width, int height, int bpp, int quality, + char* comp_data, int* comp_data_bytes) +{ + struct jpeg_compress_struct cinfo; + struct jpeg_error_mgr jerr; + struct jpeg_destination_mgr dst_mgr; + struct mydata_comp md; + JSAMPROW row_pointer[4]; + int Bpp; + + Bpp = (bpp + 7) / 8; + memset(&cinfo, 0, sizeof(cinfo)); + cinfo.err = jpeg_std_error(&jerr); + jpeg_create_compress(&cinfo); + memset(&md, 0, sizeof(md)); + md.cb = comp_data, + md.cb_bytes = *comp_data_bytes; + cinfo.client_data = &md; + memset(&dst_mgr, 0, sizeof(dst_mgr)); + dst_mgr.init_destination = my_init_destination; + dst_mgr.empty_output_buffer = my_empty_output_buffer; + dst_mgr.term_destination = my_term_destination; + cinfo.dest = &dst_mgr; + cinfo.image_width = width; + cinfo.image_height = height; + cinfo.input_components = Bpp; + cinfo.in_color_space = JCS_RGB; + jpeg_set_defaults(&cinfo); + cinfo.num_components = 3; + cinfo.dct_method = JDCT_FLOAT; + jpeg_set_quality(&cinfo, quality, 1); + jpeg_start_compress(&cinfo, 1); + while (cinfo.next_scanline + 3 < cinfo.image_height) + { + row_pointer[0] = data; + data += width * Bpp; + row_pointer[1] = data; + data += width * Bpp; + row_pointer[2] = data; + data += width * Bpp; + row_pointer[3] = data; + data += width * Bpp; + jpeg_write_scanlines(&cinfo, row_pointer, 4); + } + while (cinfo.next_scanline < cinfo.image_height) + { + row_pointer[0] = data; + data += width * Bpp; + jpeg_write_scanlines(&cinfo, row_pointer, 1); + } + jpeg_finish_compress(&cinfo); + *comp_data_bytes = md.total_done; + jpeg_destroy_compress(&cinfo); + if (md.overwrite) + { + return 1; + } + return 0; +} + +/*****************************************************************************/ +static int APP_CC +jpeg_compress(char* in_data, int width, int height, + struct stream* s, int bpp, int byte_limit, + int start_line, struct stream* temp_s, + int e) +{ + char* data; + tui32* src32; + tui16* src16; + tui8* dst8; + tui32 pixel; + int red; + int blue; + int green; + int j; + int i; + int cdata_bytes; + + data = g_malloc((width + e) * height * 3, 0); + dst8 = data; + if (bpp == 24) + { + src32 = (tui32*)in_data; + for (j = 0; j < height; j++) + { + for (i = 0; i < width; i++) + { + pixel = src32[i + j * width]; + SPLITCOLOR32(red, green, blue, pixel); + *(dst8++)= blue; + *(dst8++)= green; + *(dst8++)= red; + } + for (i = 0; i < e; i++) + { + *(dst8++) = blue; + *(dst8++) = green; + *(dst8++) = red; + } + } + } + else + { + g_writeln("bpp wrong %d", bpp); + } + cdata_bytes = byte_limit; + jp_do_compress(data, width + e, height, 24, JP_QUALITY, s->p, &cdata_bytes); + s->p += cdata_bytes; + g_free(data); + return cdata_bytes; +} + +/*****************************************************************************/ +int APP_CC +xrdp_jpeg_compress(char* in_data, int width, int height, + struct stream* s, int bpp, int byte_limit, + int start_line, struct stream* temp_s, + int e) +{ + jpeg_compress(in_data, width, height, s, bpp, byte_limit, + start_line, temp_s, e); + return height; +} + +#else + +/*****************************************************************************/ +int APP_CC +xrdp_jpeg_compress(char* in_data, int width, int height, + struct stream* s, int bpp, int byte_limit, + int start_line, struct stream* temp_s, + int e) +{ + return height; +} + +#endif |