--- /dev/null
+/*
+ Copyright (C) 2002 Mathieu Olivier
+
+ 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:
+
+ Free Software Foundation, Inc.
+ 59 Temple Place - Suite 330
+ Boston, MA 02111-1307, USA
+
+*/
+
+
+#include "quakedef.h"
+#include "image.h"
+#include "jpeg.h"
+#include "image_png.h"
+
+cvar_t sv_writepicture_quality = {CVAR_SAVE, "sv_writepicture_quality", "10", "WritePicture quality offset (higher means better quality, but slower)"};
+cvar_t r_texture_jpeg_fastpicmip = {CVAR_SAVE, "r_texture_jpeg_fastpicmip", "1", "perform gl_picmip during decompression for JPEG files (faster)"};
+
+// jboolean is unsigned char instead of int on Win32
+#ifdef WIN32
+typedef unsigned char jboolean;
+#else
+typedef int jboolean;
+#endif
+
+#ifdef LINK_TO_LIBJPEG
+#include <jpeglib.h>
+#define qjpeg_create_compress jpeg_create_compress
+#define qjpeg_create_decompress jpeg_create_decompress
+#define qjpeg_destroy_compress jpeg_destroy_compress
+#define qjpeg_destroy_decompress jpeg_destroy_decompress
+#define qjpeg_finish_compress jpeg_finish_compress
+#define qjpeg_finish_decompress jpeg_finish_decompress
+#define qjpeg_resync_to_restart jpeg_resync_to_restart
+#define qjpeg_read_header jpeg_read_header
+#define qjpeg_read_scanlines jpeg_read_scanlines
+#define qjpeg_set_defaults jpeg_set_defaults
+#define qjpeg_set_quality jpeg_set_quality
+#define qjpeg_start_compress jpeg_start_compress
+#define qjpeg_start_decompress jpeg_start_decompress
+#define qjpeg_std_error jpeg_std_error
+#define qjpeg_write_scanlines jpeg_write_scanlines
+#define qjpeg_simple_progression jpeg_simple_progression
+#define jpeg_dll true
+#else
+/*
+=================================================================
+
+ Minimal set of definitions from the JPEG lib
+
+ WARNING: for a matter of simplicity, several pointer types are
+ casted to "void*", and most enumerated values are not included
+
+=================================================================
+*/
+
+typedef void *j_common_ptr;
+typedef struct jpeg_compress_struct *j_compress_ptr;
+typedef struct jpeg_decompress_struct *j_decompress_ptr;
+
+#define JPEG_LIB_VERSION 62 // Version 6b
+
+typedef enum
+{
+ JCS_UNKNOWN,
+ JCS_GRAYSCALE,
+ JCS_RGB,
+ JCS_YCbCr,
+ JCS_CMYK,
+ JCS_YCCK
+} J_COLOR_SPACE;
+typedef enum {JPEG_DUMMY1} J_DCT_METHOD;
+typedef enum {JPEG_DUMMY2} J_DITHER_MODE;
+typedef unsigned int JDIMENSION;
+
+#define JPOOL_PERMANENT 0 // lasts until master record is destroyed
+#define JPOOL_IMAGE 1 // lasts until done with image/datastream
+
+#define JPEG_EOI 0xD9 // EOI marker code
+
+#define JMSG_STR_PARM_MAX 80
+
+#define DCTSIZE2 64
+#define NUM_QUANT_TBLS 4
+#define NUM_HUFF_TBLS 4
+#define NUM_ARITH_TBLS 16
+#define MAX_COMPS_IN_SCAN 4
+#define C_MAX_BLOCKS_IN_MCU 10
+#define D_MAX_BLOCKS_IN_MCU 10
+
+struct jpeg_memory_mgr
+{
+ void* (*alloc_small) (j_common_ptr cinfo, int pool_id, size_t sizeofobject);
+ void (*_reserve_space_for_alloc_large) (void *dummy, ...);
+ void (*_reserve_space_for_alloc_sarray) (void *dummy, ...);
+ void (*_reserve_space_for_alloc_barray) (void *dummy, ...);
+ void (*_reserve_space_for_request_virt_sarray) (void *dummy, ...);
+ void (*_reserve_space_for_request_virt_barray) (void *dummy, ...);
+ void (*_reserve_space_for_realize_virt_arrays) (void *dummy, ...);
+ void (*_reserve_space_for_access_virt_sarray) (void *dummy, ...);
+ void (*_reserve_space_for_access_virt_barray) (void *dummy, ...);
+ void (*_reserve_space_for_free_pool) (void *dummy, ...);
+ void (*_reserve_space_for_self_destruct) (void *dummy, ...);
+
+ long max_memory_to_use;
+ long max_alloc_chunk;
+};
+
+struct jpeg_error_mgr
+{
+ void (*error_exit) (j_common_ptr cinfo);
+ void (*emit_message) (j_common_ptr cinfo, int msg_level);
+ void (*output_message) (j_common_ptr cinfo);
+ void (*format_message) (j_common_ptr cinfo, char * buffer);
+ void (*reset_error_mgr) (j_common_ptr cinfo);
+ int msg_code;
+ union {
+ int i[8];
+ char s[JMSG_STR_PARM_MAX];
+ } msg_parm;
+ int trace_level;
+ long num_warnings;
+ const char * const * jpeg_message_table;
+ int last_jpeg_message;
+ const char * const * addon_message_table;
+ int first_addon_message;
+ int last_addon_message;
+};
+
+struct jpeg_source_mgr
+{
+ const unsigned char *next_input_byte;
+ size_t bytes_in_buffer;
+
+ void (*init_source) (j_decompress_ptr cinfo);
+ jboolean (*fill_input_buffer) (j_decompress_ptr cinfo);
+ void (*skip_input_data) (j_decompress_ptr cinfo, long num_bytes);
+ jboolean (*resync_to_restart) (j_decompress_ptr cinfo, int desired);
+ void (*term_source) (j_decompress_ptr cinfo);
+};
+
+typedef struct {
+ /* These values are fixed over the whole image. */
+ /* For compression, they must be supplied by parameter setup; */
+ /* for decompression, they are read from the SOF marker. */
+ int component_id; /* identifier for this component (0..255) */
+ int component_index; /* its index in SOF or cinfo->comp_info[] */
+ int h_samp_factor; /* horizontal sampling factor (1..4) */
+ int v_samp_factor; /* vertical sampling factor (1..4) */
+ int quant_tbl_no; /* quantization table selector (0..3) */
+ /* These values may vary between scans. */
+ /* For compression, they must be supplied by parameter setup; */
+ /* for decompression, they are read from the SOS marker. */
+ /* The decompressor output side may not use these variables. */
+ int dc_tbl_no; /* DC entropy table selector (0..3) */
+ int ac_tbl_no; /* AC entropy table selector (0..3) */
+
+ /* Remaining fields should be treated as private by applications. */
+
+ /* These values are computed during compression or decompression startup: */
+ /* Component's size in DCT blocks.
+ * Any dummy blocks added to complete an MCU are not counted; therefore
+ * these values do not depend on whether a scan is interleaved or not.
+ */
+ JDIMENSION width_in_blocks;
+ JDIMENSION height_in_blocks;
+ /* Size of a DCT block in samples. Always DCTSIZE for compression.
+ * For decompression this is the size of the output from one DCT block,
+ * reflecting any scaling we choose to apply during the IDCT step.
+ * Values of 1,2,4,8 are likely to be supported. Note that different
+ * components may receive different IDCT scalings.
+ */
+ int DCT_scaled_size;
+ /* The downsampled dimensions are the component's actual, unpadded number
+ * of samples at the main buffer (preprocessing/compression interface), thus
+ * downsampled_width = ceil(image_width * Hi/Hmax)
+ * and similarly for height. For decompression, IDCT scaling is included, so
+ * downsampled_width = ceil(image_width * Hi/Hmax * DCT_scaled_size/DCTSIZE)
+ */
+ JDIMENSION downsampled_width; /* actual width in samples */
+ JDIMENSION downsampled_height; /* actual height in samples */
+ /* This flag is used only for decompression. In cases where some of the
+ * components will be ignored (eg grayscale output from YCbCr image),
+ * we can skip most computations for the unused components.
+ */
+ jboolean component_needed; /* do we need the value of this component? */
+
+ /* These values are computed before starting a scan of the component. */
+ /* The decompressor output side may not use these variables. */
+ int MCU_width; /* number of blocks per MCU, horizontally */
+ int MCU_height; /* number of blocks per MCU, vertically */
+ int MCU_blocks; /* MCU_width * MCU_height */
+ int MCU_sample_width; /* MCU width in samples, MCU_width*DCT_scaled_size */
+ int last_col_width; /* # of non-dummy blocks across in last MCU */
+ int last_row_height; /* # of non-dummy blocks down in last MCU */
+
+ /* Saved quantization table for component; NULL if none yet saved.
+ * See jdinput.c comments about the need for this information.
+ * This field is currently used only for decompression.
+ */
+ void *quant_table;
+
+ /* Private per-component storage for DCT or IDCT subsystem. */
+ void * dct_table;
+} jpeg_component_info;
+
+struct jpeg_decompress_struct
+{
+ struct jpeg_error_mgr *err; // USED
+ struct jpeg_memory_mgr *mem; // USED
+
+ void *progress;
+ void *client_data;
+ jboolean is_decompressor;
+ int global_state;
+
+ struct jpeg_source_mgr *src; // USED
+ JDIMENSION image_width; // USED
+ JDIMENSION image_height; // USED
+
+ int num_components;
+ J_COLOR_SPACE jpeg_color_space;
+ J_COLOR_SPACE out_color_space;
+ unsigned int scale_num, scale_denom;
+ double output_gamma;
+ jboolean buffered_image;
+ jboolean raw_data_out;
+ J_DCT_METHOD dct_method;
+ jboolean do_fancy_upsampling;
+ jboolean do_block_smoothing;
+ jboolean quantize_colors;
+ J_DITHER_MODE dither_mode;
+ jboolean two_pass_quantize;
+ int desired_number_of_colors;
+ jboolean enable_1pass_quant;
+ jboolean enable_external_quant;
+ jboolean enable_2pass_quant;
+ JDIMENSION output_width;
+
+ JDIMENSION output_height; // USED
+
+ int out_color_components;
+
+ int output_components; // USED
+
+ int rec_outbuf_height;
+ int actual_number_of_colors;
+ void *colormap;
+
+ JDIMENSION output_scanline; // USED
+
+ int input_scan_number;
+ JDIMENSION input_iMCU_row;
+ int output_scan_number;
+ JDIMENSION output_iMCU_row;
+ int (*coef_bits)[DCTSIZE2];
+ void *quant_tbl_ptrs[NUM_QUANT_TBLS];
+ void *dc_huff_tbl_ptrs[NUM_HUFF_TBLS];
+ void *ac_huff_tbl_ptrs[NUM_HUFF_TBLS];
+ int data_precision;
+ jpeg_component_info *comp_info;
+ jboolean progressive_mode;
+ jboolean arith_code;
+ unsigned char arith_dc_L[NUM_ARITH_TBLS];
+ unsigned char arith_dc_U[NUM_ARITH_TBLS];
+ unsigned char arith_ac_K[NUM_ARITH_TBLS];
+ unsigned int restart_interval;
+ jboolean saw_JFIF_marker;
+ unsigned char JFIF_major_version;
+ unsigned char JFIF_minor_version;
+ unsigned char density_unit;
+ unsigned short X_density;
+ unsigned short Y_density;
+ jboolean saw_Adobe_marker;
+ unsigned char Adobe_transform;
+ jboolean CCIR601_sampling;
+ void *marker_list;
+ int max_h_samp_factor;
+ int max_v_samp_factor;
+ int min_DCT_scaled_size;
+ JDIMENSION total_iMCU_rows;
+ void *sample_range_limit;
+ int comps_in_scan;
+ jpeg_component_info *cur_comp_info[MAX_COMPS_IN_SCAN];
+ JDIMENSION MCUs_per_row;
+ JDIMENSION MCU_rows_in_scan;
+ int blocks_in_MCU;
+ int MCU_membership[D_MAX_BLOCKS_IN_MCU];
+ int Ss, Se, Ah, Al;
+ int unread_marker;
+ void *master;
+ void *main;
+ void *coef;
+ void *post;
+ void *inputctl;
+ void *marker;
+ void *entropy;
+ void *idct;
+ void *upsample;
+ void *cconvert;
+ void *cquantize;
+};
+
+
+struct jpeg_compress_struct
+{
+ struct jpeg_error_mgr *err;
+ struct jpeg_memory_mgr *mem;
+ void *progress;
+ void *client_data;
+ jboolean is_decompressor;
+ int global_state;
+
+ void *dest;
+ JDIMENSION image_width;
+ JDIMENSION image_height;
+ int input_components;
+ J_COLOR_SPACE in_color_space;
+ double input_gamma;
+ int data_precision;
+
+ int num_components;
+ J_COLOR_SPACE jpeg_color_space;
+ jpeg_component_info *comp_info;
+ void *quant_tbl_ptrs[NUM_QUANT_TBLS];
+ void *dc_huff_tbl_ptrs[NUM_HUFF_TBLS];
+ void *ac_huff_tbl_ptrs[NUM_HUFF_TBLS];
+ unsigned char arith_dc_L[NUM_ARITH_TBLS];
+ unsigned char arith_dc_U[NUM_ARITH_TBLS];
+ unsigned char arith_ac_K[NUM_ARITH_TBLS];
+
+ int num_scans;
+ const void *scan_info;
+ jboolean raw_data_in;
+ jboolean arith_code;
+ jboolean optimize_coding;
+ jboolean CCIR601_sampling;
+ int smoothing_factor;
+ J_DCT_METHOD dct_method;
+
+ unsigned int restart_interval;
+ int restart_in_rows;
+
+ jboolean write_JFIF_header;
+ unsigned char JFIF_major_version;
+ unsigned char JFIF_minor_version;
+ unsigned char density_unit;
+ unsigned short X_density;
+ unsigned short Y_density;
+ jboolean write_Adobe_marker;
+ JDIMENSION next_scanline;
+
+ jboolean progressive_mode;
+ int max_h_samp_factor;
+ int max_v_samp_factor;
+ JDIMENSION total_iMCU_rows;
+ int comps_in_scan;
+ jpeg_component_info *cur_comp_info[MAX_COMPS_IN_SCAN];
+ JDIMENSION MCUs_per_row;
+ JDIMENSION MCU_rows_in_scan;
+ int blocks_in_MCU;
+ int MCU_membership[C_MAX_BLOCKS_IN_MCU];
+ int Ss, Se, Ah, Al;
+
+ void *master;
+ void *main;
+ void *prep;
+ void *coef;
+ void *marker;
+ void *cconvert;
+ void *downsample;
+ void *fdct;
+ void *entropy;
+ void *script_space;
+ int script_space_size;
+};
+
+struct jpeg_destination_mgr
+{
+ unsigned char* next_output_byte;
+ size_t free_in_buffer;
+
+ void (*init_destination) (j_compress_ptr cinfo);
+ jboolean (*empty_output_buffer) (j_compress_ptr cinfo);
+ void (*term_destination) (j_compress_ptr cinfo);
+};
+
+
+/*
+=================================================================
+
+ DarkPlaces definitions
+
+=================================================================
+*/
+
+// Functions exported from libjpeg
+#define qjpeg_create_compress(cinfo) \
+ qjpeg_CreateCompress((cinfo), JPEG_LIB_VERSION, (size_t) sizeof(struct jpeg_compress_struct))
+#define qjpeg_create_decompress(cinfo) \
+ qjpeg_CreateDecompress((cinfo), JPEG_LIB_VERSION, (size_t) sizeof(struct jpeg_decompress_struct))
+
+static void (*qjpeg_CreateCompress) (j_compress_ptr cinfo, int version, size_t structsize);
+static void (*qjpeg_CreateDecompress) (j_decompress_ptr cinfo, int version, size_t structsize);
+static void (*qjpeg_destroy_compress) (j_compress_ptr cinfo);
+static void (*qjpeg_destroy_decompress) (j_decompress_ptr cinfo);
+static void (*qjpeg_finish_compress) (j_compress_ptr cinfo);
+static jboolean (*qjpeg_finish_decompress) (j_decompress_ptr cinfo);
+static jboolean (*qjpeg_resync_to_restart) (j_decompress_ptr cinfo, int desired);
+static int (*qjpeg_read_header) (j_decompress_ptr cinfo, jboolean require_image);
+static JDIMENSION (*qjpeg_read_scanlines) (j_decompress_ptr cinfo, unsigned char** scanlines, JDIMENSION max_lines);
+static void (*qjpeg_set_defaults) (j_compress_ptr cinfo);
+static void (*qjpeg_set_quality) (j_compress_ptr cinfo, int quality, jboolean force_baseline);
+static jboolean (*qjpeg_start_compress) (j_compress_ptr cinfo, jboolean write_all_tables);
+static jboolean (*qjpeg_start_decompress) (j_decompress_ptr cinfo);
+static struct jpeg_error_mgr* (*qjpeg_std_error) (struct jpeg_error_mgr *err);
+static JDIMENSION (*qjpeg_write_scanlines) (j_compress_ptr cinfo, unsigned char** scanlines, JDIMENSION num_lines);
+static void (*qjpeg_simple_progression) (j_compress_ptr cinfo);
+
+static dllfunction_t jpegfuncs[] =
+{
+ {"jpeg_CreateCompress", (void **) &qjpeg_CreateCompress},
+ {"jpeg_CreateDecompress", (void **) &qjpeg_CreateDecompress},
+ {"jpeg_destroy_compress", (void **) &qjpeg_destroy_compress},
+ {"jpeg_destroy_decompress", (void **) &qjpeg_destroy_decompress},
+ {"jpeg_finish_compress", (void **) &qjpeg_finish_compress},
+ {"jpeg_finish_decompress", (void **) &qjpeg_finish_decompress},
+ {"jpeg_resync_to_restart", (void **) &qjpeg_resync_to_restart},
+ {"jpeg_read_header", (void **) &qjpeg_read_header},
+ {"jpeg_read_scanlines", (void **) &qjpeg_read_scanlines},
+ {"jpeg_set_defaults", (void **) &qjpeg_set_defaults},
+ {"jpeg_set_quality", (void **) &qjpeg_set_quality},
+ {"jpeg_start_compress", (void **) &qjpeg_start_compress},
+ {"jpeg_start_decompress", (void **) &qjpeg_start_decompress},
+ {"jpeg_std_error", (void **) &qjpeg_std_error},
+ {"jpeg_write_scanlines", (void **) &qjpeg_write_scanlines},
+ {"jpeg_simple_progression", (void **) &qjpeg_simple_progression},
+ {NULL, NULL}
+};
+
+// Handle for JPEG DLL
+dllhandle_t jpeg_dll = NULL;
+qboolean jpeg_tried_loading = 0;
+#endif
+
+static unsigned char jpeg_eoi_marker [2] = {0xFF, JPEG_EOI};
+static jmp_buf error_in_jpeg;
+static qboolean jpeg_toolarge;
+
+// Our own output manager for JPEG compression
+typedef struct
+{
+ struct jpeg_destination_mgr pub;
+
+ qfile_t* outfile;
+ unsigned char* buffer;
+ size_t bufsize; // used if outfile is NULL
+} my_destination_mgr;
+typedef my_destination_mgr* my_dest_ptr;
+
+
+/*
+=================================================================
+
+ DLL load & unload
+
+=================================================================
+*/
+
+/*
+====================
+JPEG_OpenLibrary
+
+Try to load the JPEG DLL
+====================
+*/
+qboolean JPEG_OpenLibrary (void)
+{
+#ifdef LINK_TO_LIBJPEG
+ return true;
+#else
+ const char* dllnames [] =
+ {
+#if defined(WIN32)
+ "libjpeg.dll",
+#elif defined(MACOSX)
+ "libjpeg.62.dylib",
+#else
+ "libjpeg.so.62",
+ "libjpeg.so",
+#endif
+ NULL
+ };
+
+ // Already loaded?
+ if (jpeg_dll)
+ return true;
+
+ if (jpeg_tried_loading) // only try once
+ return false;
+
+ jpeg_tried_loading = true;
+
+ // Load the DLL
+ return Sys_LoadLibrary (dllnames, &jpeg_dll, jpegfuncs);
+#endif
+}
+
+
+/*
+====================
+JPEG_CloseLibrary
+
+Unload the JPEG DLL
+====================
+*/
+void JPEG_CloseLibrary (void)
+{
+#ifndef LINK_TO_LIBJPEG
+ Sys_UnloadLibrary (&jpeg_dll);
+ jpeg_tried_loading = false; // allow retry
+#endif
+}
+
+
+/*
+=================================================================
+
+ JPEG decompression
+
+=================================================================
+*/
+
+static void JPEG_Noop (j_decompress_ptr cinfo) {}
+
+static jboolean JPEG_FillInputBuffer (j_decompress_ptr cinfo)
+{
+ // Insert a fake EOI marker
+ cinfo->src->next_input_byte = jpeg_eoi_marker;
+ cinfo->src->bytes_in_buffer = 2;
+
+ return TRUE;
+}
+
+static void JPEG_SkipInputData (j_decompress_ptr cinfo, long num_bytes)
+{
+ if (cinfo->src->bytes_in_buffer <= (unsigned long)num_bytes)
+ {
+ cinfo->src->bytes_in_buffer = 0;
+ return;
+ }
+
+ cinfo->src->next_input_byte += num_bytes;
+ cinfo->src->bytes_in_buffer -= num_bytes;
+}
+
+static void JPEG_MemSrc (j_decompress_ptr cinfo, const unsigned char *buffer, size_t filesize)
+{
+ cinfo->src = (struct jpeg_source_mgr *)cinfo->mem->alloc_small ((j_common_ptr) cinfo, JPOOL_PERMANENT, sizeof (struct jpeg_source_mgr));
+
+ cinfo->src->next_input_byte = buffer;
+ cinfo->src->bytes_in_buffer = filesize;
+
+ cinfo->src->init_source = JPEG_Noop;
+ cinfo->src->fill_input_buffer = JPEG_FillInputBuffer;
+ cinfo->src->skip_input_data = JPEG_SkipInputData;
+ cinfo->src->resync_to_restart = qjpeg_resync_to_restart; // use the default method
+ cinfo->src->term_source = JPEG_Noop;
+}
+
+static void JPEG_ErrorExit (j_common_ptr cinfo)
+{
+ ((struct jpeg_decompress_struct*)cinfo)->err->output_message (cinfo);
+ longjmp(error_in_jpeg, 1);
+}
+
+
+/*
+====================
+JPEG_LoadImage
+
+Load a JPEG image into a BGRA buffer
+====================
+*/
+unsigned char* JPEG_LoadImage_BGRA (const unsigned char *f, int filesize, int *miplevel)
+{
+ struct jpeg_decompress_struct cinfo;
+ struct jpeg_error_mgr jerr;
+ unsigned char *image_buffer = NULL, *scanline = NULL;
+ unsigned int line;
+ int submip = 0;
+
+ // No DLL = no JPEGs
+ if (!jpeg_dll)
+ return NULL;
+
+ if(miplevel && r_texture_jpeg_fastpicmip.integer)
+ submip = bound(0, *miplevel, 3);
+
+ cinfo.err = qjpeg_std_error (&jerr);
+ qjpeg_create_decompress (&cinfo);
+ if(setjmp(error_in_jpeg))
+ goto error_caught;
+ cinfo.err = qjpeg_std_error (&jerr);
+ cinfo.err->error_exit = JPEG_ErrorExit;
+ JPEG_MemSrc (&cinfo, f, filesize);
+ qjpeg_read_header (&cinfo, TRUE);
+ cinfo.scale_num = 1;
+ cinfo.scale_denom = (1 << submip);
+ qjpeg_start_decompress (&cinfo);
+
+ image_width = cinfo.output_width;
+ image_height = cinfo.output_height;
+
+ if (image_width > 32768 || image_height > 32768 || image_width <= 0 || image_height <= 0)
+ {
+ Con_Printf("JPEG_LoadImage: invalid image size %ix%i\n", image_width, image_height);
+ return NULL;
+ }
+
+ image_buffer = (unsigned char *)Mem_Alloc(tempmempool, image_width * image_height * 4);
+ scanline = (unsigned char *)Mem_Alloc(tempmempool, image_width * cinfo.output_components);
+ if (!image_buffer || !scanline)
+ {
+ if (image_buffer)
+ Mem_Free (image_buffer);
+ if (scanline)
+ Mem_Free (scanline);
+
+ Con_Printf("JPEG_LoadImage: not enough memory for %i by %i image\n", image_width, image_height);
+ qjpeg_finish_decompress (&cinfo);
+ qjpeg_destroy_decompress (&cinfo);
+ return NULL;
+ }
+
+ // Decompress the image, line by line
+ line = 0;
+ while (cinfo.output_scanline < cinfo.output_height)
+ {
+ unsigned char *buffer_ptr;
+ int ind;
+
+ qjpeg_read_scanlines (&cinfo, &scanline, 1);
+
+ // Convert the image to BGRA
+ switch (cinfo.output_components)
+ {
+ // RGB images
+ case 3:
+ buffer_ptr = &image_buffer[image_width * line * 4];
+ for (ind = 0; ind < image_width * 3; ind += 3, buffer_ptr += 4)
+ {
+ buffer_ptr[2] = scanline[ind];
+ buffer_ptr[1] = scanline[ind + 1];
+ buffer_ptr[0] = scanline[ind + 2];
+ buffer_ptr[3] = 255;
+ }
+ break;
+
+ // Greyscale images (default to it, just in case)
+ case 1:
+ default:
+ buffer_ptr = &image_buffer[image_width * line * 4];
+ for (ind = 0; ind < image_width; ind++, buffer_ptr += 4)
+ {
+ buffer_ptr[0] = scanline[ind];
+ buffer_ptr[1] = scanline[ind];
+ buffer_ptr[2] = scanline[ind];
+ buffer_ptr[3] = 255;
+ }
+ }
+
+ line++;
+ }
+ Mem_Free (scanline); scanline = NULL;
+
+ qjpeg_finish_decompress (&cinfo);
+ qjpeg_destroy_decompress (&cinfo);
+
+ if(miplevel)
+ *miplevel -= submip;
+
+ return image_buffer;
+
+error_caught:
+ if(scanline)
+ Mem_Free (scanline);
+ if(image_buffer)
+ Mem_Free (image_buffer);
+ qjpeg_destroy_decompress (&cinfo);
+ return NULL;
+}
+
+
+/*
+=================================================================
+
+ JPEG compression
+
+=================================================================
+*/
+
+#define JPEG_OUTPUT_BUF_SIZE 4096
+static void JPEG_InitDestination (j_compress_ptr cinfo)
+{
+ my_dest_ptr dest = (my_dest_ptr)cinfo->dest;
+ dest->buffer = (unsigned char*)cinfo->mem->alloc_small ((j_common_ptr) cinfo, JPOOL_IMAGE, JPEG_OUTPUT_BUF_SIZE * sizeof(unsigned char));
+ dest->pub.next_output_byte = dest->buffer;
+ dest->pub.free_in_buffer = JPEG_OUTPUT_BUF_SIZE;
+}
+
+static jboolean JPEG_EmptyOutputBuffer (j_compress_ptr cinfo)
+{
+ my_dest_ptr dest = (my_dest_ptr)cinfo->dest;
+
+ if (FS_Write (dest->outfile, dest->buffer, JPEG_OUTPUT_BUF_SIZE) != (size_t) JPEG_OUTPUT_BUF_SIZE)
+ longjmp(error_in_jpeg, 1);
+
+ dest->pub.next_output_byte = dest->buffer;
+ dest->pub.free_in_buffer = JPEG_OUTPUT_BUF_SIZE;
+ return true;
+}
+
+static void JPEG_TermDestination (j_compress_ptr cinfo)
+{
+ my_dest_ptr dest = (my_dest_ptr)cinfo->dest;
+ size_t datacount = JPEG_OUTPUT_BUF_SIZE - dest->pub.free_in_buffer;
+
+ // Write any data remaining in the buffer
+ if (datacount > 0)
+ if (FS_Write (dest->outfile, dest->buffer, datacount) != (fs_offset_t)datacount)
+ longjmp(error_in_jpeg, 1);
+}
+
+static void JPEG_FileDest (j_compress_ptr cinfo, qfile_t* outfile)
+{
+ my_dest_ptr dest;
+
+ // First time for this JPEG object?
+ if (cinfo->dest == NULL)
+ cinfo->dest = (struct jpeg_destination_mgr *)(*cinfo->mem->alloc_small) ((j_common_ptr) cinfo, JPOOL_PERMANENT, sizeof(my_destination_mgr));
+
+ dest = (my_dest_ptr)cinfo->dest;
+ dest->pub.init_destination = JPEG_InitDestination;
+ dest->pub.empty_output_buffer = JPEG_EmptyOutputBuffer;
+ dest->pub.term_destination = JPEG_TermDestination;
+ dest->outfile = outfile;
+}
+
+static void JPEG_Mem_InitDestination (j_compress_ptr cinfo)
+{
+ my_dest_ptr dest = (my_dest_ptr)cinfo->dest;
+ dest->pub.next_output_byte = dest->buffer;
+ dest->pub.free_in_buffer = dest->bufsize;
+}
+
+static jboolean JPEG_Mem_EmptyOutputBuffer (j_compress_ptr cinfo)
+{
+ my_dest_ptr dest = (my_dest_ptr)cinfo->dest;
+ jpeg_toolarge = true;
+ dest->pub.next_output_byte = dest->buffer;
+ dest->pub.free_in_buffer = dest->bufsize;
+ return true;
+}
+
+static void JPEG_Mem_TermDestination (j_compress_ptr cinfo)
+{
+ my_dest_ptr dest = (my_dest_ptr)cinfo->dest;
+ dest->bufsize = dest->pub.next_output_byte - dest->buffer;
+}
+static void JPEG_MemDest (j_compress_ptr cinfo, void* buf, size_t bufsize)
+{
+ my_dest_ptr dest;
+
+ // First time for this JPEG object?
+ if (cinfo->dest == NULL)
+ cinfo->dest = (struct jpeg_destination_mgr *)(*cinfo->mem->alloc_small) ((j_common_ptr) cinfo, JPOOL_PERMANENT, sizeof(my_destination_mgr));
+
+ dest = (my_dest_ptr)cinfo->dest;
+ dest->pub.init_destination = JPEG_Mem_InitDestination;
+ dest->pub.empty_output_buffer = JPEG_Mem_EmptyOutputBuffer;
+ dest->pub.term_destination = JPEG_Mem_TermDestination;
+ dest->outfile = NULL;
+
+ dest->buffer = (unsigned char *) buf;
+ dest->bufsize = bufsize;
+}
+
+
+/*
+====================
+JPEG_SaveImage_preflipped
+
+Save a preflipped JPEG image to a file
+====================
+*/
+qboolean JPEG_SaveImage_preflipped (const char *filename, int width, int height, unsigned char *data)
+{
+ struct jpeg_compress_struct cinfo;
+ struct jpeg_error_mgr jerr;
+ unsigned char *scanline;
+ unsigned int offset, linesize;
+ qfile_t* file;
+
+ // No DLL = no JPEGs
+ if (!jpeg_dll)
+ {
+ Con_Print("You need the libjpeg library to save JPEG images\n");
+ return false;
+ }
+
+ // Open the file
+ file = FS_OpenRealFile(filename, "wb", true);
+ if (!file)
+ return false;
+
+ if(setjmp(error_in_jpeg))
+ goto error_caught;
+ cinfo.err = qjpeg_std_error (&jerr);
+ cinfo.err->error_exit = JPEG_ErrorExit;
+
+ qjpeg_create_compress (&cinfo);
+ JPEG_FileDest (&cinfo, file);
+
+ // Set the parameters for compression
+ cinfo.image_width = width;
+ cinfo.image_height = height;
+ cinfo.in_color_space = JCS_RGB;
+ cinfo.input_components = 3;
+ qjpeg_set_defaults (&cinfo);
+ qjpeg_set_quality (&cinfo, (int)(scr_screenshot_jpeg_quality.value * 100), TRUE);
+ qjpeg_simple_progression (&cinfo);
+
+ // turn off subsampling (to make text look better)
+ cinfo.optimize_coding = 1;
+ cinfo.comp_info[0].h_samp_factor = 1;
+ cinfo.comp_info[0].v_samp_factor = 1;
+ cinfo.comp_info[1].h_samp_factor = 1;
+ cinfo.comp_info[1].v_samp_factor = 1;
+ cinfo.comp_info[2].h_samp_factor = 1;
+ cinfo.comp_info[2].v_samp_factor = 1;
+
+ qjpeg_start_compress (&cinfo, true);
+
+ // Compress each scanline
+ linesize = cinfo.image_width * 3;
+ offset = linesize * (cinfo.image_height - 1);
+ while (cinfo.next_scanline < cinfo.image_height)
+ {
+ scanline = &data[offset - cinfo.next_scanline * linesize];
+
+ qjpeg_write_scanlines (&cinfo, &scanline, 1);
+ }
+
+ qjpeg_finish_compress (&cinfo);
+ qjpeg_destroy_compress (&cinfo);
+
+ FS_Close (file);
+ return true;
+
+error_caught:
+ qjpeg_destroy_compress (&cinfo);
+ FS_Close (file);
+ return false;
+}
+
+static size_t JPEG_try_SaveImage_to_Buffer (struct jpeg_compress_struct *cinfo, char *jpegbuf, size_t jpegsize, int quality, int width, int height, unsigned char *data)
+{
+ unsigned char *scanline;
+ unsigned int linesize;
+
+ jpeg_toolarge = false;
+ JPEG_MemDest (cinfo, jpegbuf, jpegsize);
+
+ // Set the parameters for compression
+ cinfo->image_width = width;
+ cinfo->image_height = height;
+ cinfo->in_color_space = JCS_RGB;
+ cinfo->input_components = 3;
+ qjpeg_set_defaults (cinfo);
+ qjpeg_set_quality (cinfo, quality, FALSE);
+
+ cinfo->comp_info[0].h_samp_factor = 2;
+ cinfo->comp_info[0].v_samp_factor = 2;
+ cinfo->comp_info[1].h_samp_factor = 1;
+ cinfo->comp_info[1].v_samp_factor = 1;
+ cinfo->comp_info[2].h_samp_factor = 1;
+ cinfo->comp_info[2].v_samp_factor = 1;
+ cinfo->optimize_coding = 1;
+
+ qjpeg_start_compress (cinfo, true);
+
+ // Compress each scanline
+ linesize = width * 3;
+ while (cinfo->next_scanline < cinfo->image_height)
+ {
+ scanline = &data[cinfo->next_scanline * linesize];
+
+ qjpeg_write_scanlines (cinfo, &scanline, 1);
+ }
+
+ qjpeg_finish_compress (cinfo);
+
+ if(jpeg_toolarge)
+ return 0;
+
+ return ((my_dest_ptr) cinfo->dest)->bufsize;
+}
+
+size_t JPEG_SaveImage_to_Buffer (char *jpegbuf, size_t jpegsize, int width, int height, unsigned char *data)
+{
+ struct jpeg_compress_struct cinfo;
+ struct jpeg_error_mgr jerr;
+
+ int quality;
+ int quality_guess;
+ size_t result;
+
+ // No DLL = no JPEGs
+ if (!jpeg_dll)
+ {
+ Con_Print("You need the libjpeg library to save JPEG images\n");
+ return false;
+ }
+
+ if(setjmp(error_in_jpeg))
+ goto error_caught;
+ cinfo.err = qjpeg_std_error (&jerr);
+ cinfo.err->error_exit = JPEG_ErrorExit;
+
+ qjpeg_create_compress (&cinfo);
+
+#if 0
+ // used to get the formula below
+ {
+ char buf[1048576];
+ unsigned char *img;
+ int i;
+
+ img = Mem_Alloc(tempmempool, width * height * 3);
+ for(i = 0; i < width * height * 3; ++i)
+ img[i] = rand() & 0xFF;
+
+ for(i = 0; i <= 100; ++i)
+ {
+ Con_Printf("! %d %d %d %d\n", width, height, i, (int) JPEG_try_SaveImage_to_Buffer(&cinfo, buf, sizeof(buf), i, width, height, img));
+ }
+
+ Mem_Free(img);
+ }
+#endif
+
+ //quality_guess = (100 * jpegsize - 41000) / (width*height) + 2; // fits random data
+ quality_guess = (256 * jpegsize - 81920) / (width*height) - 8; // fits Nexuiz's/Xonotic's map pictures
+
+ quality_guess = bound(0, quality_guess, 100);
+ quality = bound(0, quality_guess + sv_writepicture_quality.integer, 100); // assume it can do 10 failed attempts
+
+ while(!(result = JPEG_try_SaveImage_to_Buffer(&cinfo, jpegbuf, jpegsize, quality, width, height, data)))
+ {
+ --quality;
+ if(quality < 0)
+ {
+ Con_Printf("couldn't write image at all, probably too big\n");
+ return 0;
+ }
+ }
+ qjpeg_destroy_compress (&cinfo);
+ Con_DPrintf("JPEG_SaveImage_to_Buffer: guessed quality/size %d/%d, actually got %d/%d\n", quality_guess, (int)jpegsize, quality, (int)result);
+
+ return result;
+
+error_caught:
+ qjpeg_destroy_compress (&cinfo);
+ return 0;
+}
+
+typedef struct CompressedImageCacheItem
+{
+ char imagename[MAX_QPATH];
+ size_t maxsize;
+ void *compressed;
+ size_t compressed_size;
+ struct CompressedImageCacheItem *next;
+}
+CompressedImageCacheItem;
+#define COMPRESSEDIMAGECACHE_SIZE 4096
+static CompressedImageCacheItem *CompressedImageCache[COMPRESSEDIMAGECACHE_SIZE];
+
+static void CompressedImageCache_Add(const char *imagename, size_t maxsize, void *compressed, size_t compressed_size)
+{
+ const char *hashkey = va("%s:%d", imagename, (int) maxsize);
+ int hashindex = CRC_Block((unsigned char *) hashkey, strlen(hashkey)) % COMPRESSEDIMAGECACHE_SIZE;
+ CompressedImageCacheItem *i;
+
+ if(strlen(imagename) >= MAX_QPATH)
+ return; // can't add this
+
+ i = (CompressedImageCacheItem*) Z_Malloc(sizeof(CompressedImageCacheItem));
+ strlcpy(i->imagename, imagename, sizeof(i->imagename));
+ i->maxsize = maxsize;
+ i->compressed = compressed;
+ i->compressed_size = compressed_size;
+ i->next = CompressedImageCache[hashindex];
+ CompressedImageCache[hashindex] = i;
+}
+
+static CompressedImageCacheItem *CompressedImageCache_Find(const char *imagename, size_t maxsize)
+{
+ const char *hashkey = va("%s:%d", imagename, (int) maxsize);
+ int hashindex = CRC_Block((unsigned char *) hashkey, strlen(hashkey)) % COMPRESSEDIMAGECACHE_SIZE;
+ CompressedImageCacheItem *i = CompressedImageCache[hashindex];
+
+ while(i)
+ {
+ if(i->maxsize == maxsize)
+ if(!strcmp(i->imagename, imagename))
+ return i;
+ i = i->next;
+ }
+ return NULL;
+}
+
+qboolean Image_Compress(const char *imagename, size_t maxsize, void **buf, size_t *size)
+{
+ unsigned char *imagedata, *newimagedata;
+ int maxPixelCount;
+ int components[3] = {2, 1, 0};
+ CompressedImageCacheItem *i;
+
+ JPEG_OpenLibrary (); // for now; LH had the idea of replacing this by a better format
+ PNG_OpenLibrary (); // for loading
+
+ // No DLL = no JPEGs
+ if (!jpeg_dll)
+ {
+ Con_Print("You need the libjpeg library to save JPEG images\n");
+ return false;
+ }
+
+ i = CompressedImageCache_Find(imagename, maxsize);
+ if(i)
+ {
+ *size = i->compressed_size;
+ *buf = i->compressed;
+ }
+
+ // load the image
+ imagedata = loadimagepixelsbgra(imagename, true, false, false, NULL);
+ if(!imagedata)
+ return false;
+
+ // find an appropriate size for somewhat okay compression
+ if(maxsize <= 768)
+ maxPixelCount = 32 * 32;
+ else if(maxsize <= 1024)
+ maxPixelCount = 64 * 64;
+ else if(maxsize <= 4096)
+ maxPixelCount = 128 * 128;
+ else
+ maxPixelCount = 256 * 256;
+
+ while(image_width * image_height > maxPixelCount)
+ {
+ int one = 1;
+ Image_MipReduce32(imagedata, imagedata, &image_width, &image_height, &one, image_width/2, image_height/2, 1);
+ }
+
+ newimagedata = (unsigned char *) Mem_Alloc(tempmempool, image_width * image_height * 3);
+
+ // convert the image from BGRA to RGB
+ Image_CopyMux(newimagedata, imagedata, image_width, image_height, false, false, false, 3, 4, components);
+ Mem_Free(imagedata);
+
+ // try to compress it to JPEG
+ *buf = Z_Malloc(maxsize);
+ *size = JPEG_SaveImage_to_Buffer((char *) *buf, maxsize, image_width, image_height, newimagedata);
+ Mem_Free(newimagedata);
+
+ if(!*size)
+ {
+ Z_Free(*buf);
+ *buf = NULL;
+ Con_Printf("could not compress image %s to %d bytes\n", imagename, (int)maxsize);
+ // return false;
+ // also cache failures!
+ }
+
+ // store it in the cache
+ CompressedImageCache_Add(imagename, maxsize, *buf, *size);
+ return (*buf != NULL);
+}