Index: chrome/common/jpeg_codec.cc |
=================================================================== |
--- chrome/common/jpeg_codec.cc (revision 8694) |
+++ chrome/common/jpeg_codec.cc (working copy) |
@@ -1,521 +0,0 @@ |
-// Copyright (c) 2006-2008 The Chromium Authors. All rights reserved. |
-// Use of this source code is governed by a BSD-style license that can be |
-// found in the LICENSE file. |
- |
-#include <setjmp.h> |
- |
-#include "chrome/common/jpeg_codec.h" |
-#include "base/logging.h" |
-#include "base/scoped_ptr.h" |
-#include "SkBitmap.h" |
- |
-extern "C" { |
-#include "jpeglib.h" |
-} |
- |
-// Encoder/decoder shared stuff ------------------------------------------------ |
- |
-namespace { |
- |
-// used to pass error info through the JPEG library |
-struct CoderErrorMgr { |
- jpeg_error_mgr pub; |
- jmp_buf setjmp_buffer; |
-}; |
- |
-void ErrorExit(jpeg_common_struct* cinfo) { |
- CoderErrorMgr *err = reinterpret_cast<CoderErrorMgr*>(cinfo->err); |
- |
- // Return control to the setjmp point. |
- longjmp(err->setjmp_buffer, false); |
-} |
- |
-} // namespace |
- |
-// Encoder --------------------------------------------------------------------- |
-// |
-// This code is based on nsJPEGEncoder from Mozilla. |
-// Copyright 2005 Google Inc. (Brett Wilson, contributor) |
- |
-namespace { |
- |
-// Initial size for the output buffer in the JpegEncoderState below. |
-const static int initial_output_buffer_size = 8192; |
- |
-struct JpegEncoderState { |
- JpegEncoderState(std::vector<unsigned char>* o) : out(o), image_buffer_used(0) { |
- } |
- |
- // Output buffer, of which 'image_buffer_used' bytes are actually used (this |
- // will often be less than the actual size of the vector because we size it |
- // so that libjpeg can write directly into it. |
- std::vector<unsigned char>* out; |
- |
- // Number of bytes in the 'out' buffer that are actually used (see above). |
- size_t image_buffer_used; |
-}; |
- |
-// Initializes the JpegEncoderState for encoding, and tells libjpeg about where |
-// the output buffer is. |
-// |
-// From the JPEG library: |
-// "Initialize destination. This is called by jpeg_start_compress() before |
-// any data is actually written. It must initialize next_output_byte and |
-// free_in_buffer. free_in_buffer must be initialized to a positive value." |
-void InitDestination(jpeg_compress_struct* cinfo) { |
- JpegEncoderState* state = static_cast<JpegEncoderState*>(cinfo->client_data); |
- DCHECK(state->image_buffer_used == 0) << "initializing after use"; |
- |
- state->out->resize(initial_output_buffer_size); |
- state->image_buffer_used = 0; |
- |
- cinfo->dest->next_output_byte = &(*state->out)[0]; |
- cinfo->dest->free_in_buffer = initial_output_buffer_size; |
-} |
- |
-// Resize the buffer that we give to libjpeg and update our and its state. |
-// |
-// From the JPEG library: |
-// "Callback used by libjpeg whenever the buffer has filled (free_in_buffer |
-// reaches zero). In typical applications, it should write out the *entire* |
-// buffer (use the saved start address and buffer length; ignore the current |
-// state of next_output_byte and free_in_buffer). Then reset the pointer & |
-// count to the start of the buffer, and return TRUE indicating that the |
-// buffer has been dumped. free_in_buffer must be set to a positive value |
-// when TRUE is returned. A FALSE return should only be used when I/O |
-// suspension is desired (this operating mode is discussed in the next |
-// section)." |
-boolean EmptyOutputBuffer(jpeg_compress_struct* cinfo) { |
- JpegEncoderState* state = static_cast<JpegEncoderState*>(cinfo->client_data); |
- |
- // note the new size, the buffer is full |
- state->image_buffer_used = state->out->size(); |
- |
- // expand buffer, just double size each time |
- state->out->resize(state->out->size() * 2); |
- |
- // tell libjpeg where to write the next data |
- cinfo->dest->next_output_byte = &(*state->out)[state->image_buffer_used]; |
- cinfo->dest->free_in_buffer = state->out->size() - state->image_buffer_used; |
- return 1; |
-} |
- |
-// Cleans up the JpegEncoderState to prepare for returning in the final form. |
-// |
-// From the JPEG library: |
-// "Terminate destination --- called by jpeg_finish_compress() after all data |
-// has been written. In most applications, this must flush any data |
-// remaining in the buffer. Use either next_output_byte or free_in_buffer to |
-// determine how much data is in the buffer." |
-void TermDestination(jpeg_compress_struct* cinfo) { |
- JpegEncoderState* state = static_cast<JpegEncoderState*>(cinfo->client_data); |
- DCHECK(state->out->size() >= state->image_buffer_used); |
- |
- // update the used byte based on the next byte libjpeg would write to |
- state->image_buffer_used = cinfo->dest->next_output_byte - &(*state->out)[0]; |
- DCHECK(state->image_buffer_used < state->out->size()) << |
- "JPEG library busted, got a bad image buffer size"; |
- |
- // update our buffer so that it exactly encompases the desired data |
- state->out->resize(state->image_buffer_used); |
-} |
- |
-// Converts RGBA to RGB (removing the alpha values) to prepare to send data to |
-// libjpeg. This converts one row of data in rgba with the given width in |
-// pixels the the given rgb destination buffer (which should have enough space |
-// reserved for the final data). |
-void StripAlpha(const unsigned char* rgba, int pixel_width, unsigned char* rgb) |
-{ |
- for (int x = 0; x < pixel_width; x++) { |
- const unsigned char* pixel_in = &rgba[x * 4]; |
- unsigned char* pixel_out = &rgb[x * 3]; |
- pixel_out[0] = pixel_in[0]; |
- pixel_out[1] = pixel_in[1]; |
- pixel_out[2] = pixel_in[2]; |
- } |
-} |
- |
-// Converts BGRA to RGB by reordering the color components and dropping the |
-// alpha. This converts one row of data in rgba with the given width in |
-// pixels the the given rgb destination buffer (which should have enough space |
-// reserved for the final data). |
-void BGRAtoRGB(const unsigned char* bgra, int pixel_width, unsigned char* rgb) |
-{ |
- for (int x = 0; x < pixel_width; x++) { |
- const unsigned char* pixel_in = &bgra[x * 4]; |
- unsigned char* pixel_out = &rgb[x * 3]; |
- pixel_out[0] = pixel_in[2]; |
- pixel_out[1] = pixel_in[1]; |
- pixel_out[2] = pixel_in[0]; |
- } |
-} |
- |
-// This class destroys the given jpeg_compress object when it goes out of |
-// scope. It simplifies the error handling in Encode (and even applies to the |
-// success case). |
-class CompressDestroyer { |
- public: |
- CompressDestroyer() : cinfo_(NULL) { |
- } |
- ~CompressDestroyer() { |
- DestroyManagedObject(); |
- } |
- void SetManagedObject(jpeg_compress_struct* ci) { |
- DestroyManagedObject(); |
- cinfo_ = ci; |
- } |
- void DestroyManagedObject() { |
- if (cinfo_) { |
- jpeg_destroy_compress(cinfo_); |
- cinfo_ = NULL; |
- } |
- } |
- private: |
- jpeg_compress_struct* cinfo_; |
-}; |
- |
-} // namespace |
- |
-bool JPEGCodec::Encode(const unsigned char* input, ColorFormat format, |
- int w, int h, int row_byte_width, |
- int quality, std::vector<unsigned char>* output) { |
- jpeg_compress_struct cinfo; |
- CompressDestroyer destroyer; |
- output->clear(); |
- |
- // We set up the normal JPEG error routines, then override error_exit. |
- // This must be done before the call to create_compress. |
- CoderErrorMgr errmgr; |
- cinfo.err = jpeg_std_error(&errmgr.pub); |
- errmgr.pub.error_exit = ErrorExit; |
- // Establish the setjmp return context for ErrorExit to use. |
- if (setjmp(errmgr.setjmp_buffer)) { |
- // If we get here, the JPEG code has signaled an error. |
- // MSDN notes: "if you intend your code to be portable, do not rely on |
- // correct destruction of frame-based objects when executing a nonlocal |
- // goto using a call to longjmp." So we delete the CompressDestroyer's |
- // object manually instead. |
- destroyer.DestroyManagedObject(); |
- return false; |
- } |
- |
- // The destroyer will destroy() cinfo on exit. |
- jpeg_create_compress(&cinfo); |
- destroyer.SetManagedObject(&cinfo); |
- |
- cinfo.image_width = w; |
- cinfo.image_height = h; |
- cinfo.input_components = 3; |
- cinfo.in_color_space = JCS_RGB; |
- cinfo.data_precision = 8; |
- |
- jpeg_set_defaults(&cinfo); |
- jpeg_set_quality(&cinfo, quality, 1); // quality here is 0-100 |
- |
- // set up the destination manager |
- jpeg_destination_mgr destmgr; |
- destmgr.init_destination = InitDestination; |
- destmgr.empty_output_buffer = EmptyOutputBuffer; |
- destmgr.term_destination = TermDestination; |
- cinfo.dest = &destmgr; |
- |
- JpegEncoderState state(output); |
- cinfo.client_data = &state; |
- |
- jpeg_start_compress(&cinfo, 1); |
- |
- // feed it the rows, doing necessary conversions for the color format |
- if (format == FORMAT_RGB) { |
- // no conversion necessary |
- while (cinfo.next_scanline < cinfo.image_height) { |
- const unsigned char* row = &input[cinfo.next_scanline * row_byte_width]; |
- jpeg_write_scanlines(&cinfo, const_cast<unsigned char**>(&row), 1); |
- } |
- } else { |
- // get the correct format converter |
- void (*converter)(const unsigned char* in, int w, unsigned char* rgb); |
- if (format == FORMAT_RGBA) { |
- converter = StripAlpha; |
- } else if (format == FORMAT_BGRA) { |
- converter = BGRAtoRGB; |
- } else { |
- NOTREACHED() << "Invalid pixel format"; |
- return false; |
- } |
- |
- // output row after converting |
- unsigned char* row = new unsigned char[w * 3]; |
- |
- while (cinfo.next_scanline < cinfo.image_height) { |
- converter(&input[cinfo.next_scanline * row_byte_width], w, row); |
- jpeg_write_scanlines(&cinfo, &row, 1); |
- } |
- delete[] row; |
- } |
- |
- jpeg_finish_compress(&cinfo); |
- return true; |
-} |
- |
-// Decoder -------------------------------------------------------------------- |
- |
-namespace { |
- |
-struct JpegDecoderState { |
- JpegDecoderState(const unsigned char* in, size_t len) |
- : input_buffer(in), input_buffer_length(len) { |
- } |
- |
- const unsigned char* input_buffer; |
- size_t input_buffer_length; |
-}; |
- |
-// Callback to initialize the source. |
-// |
-// From the JPEG library: |
-// "Initialize source. This is called by jpeg_read_header() before any data is |
-// actually read. May leave bytes_in_buffer set to 0 (in which case a |
-// fill_input_buffer() call will occur immediately)." |
-void InitSource(j_decompress_ptr cinfo) { |
- JpegDecoderState* state = static_cast<JpegDecoderState*>(cinfo->client_data); |
- cinfo->src->next_input_byte = state->input_buffer; |
- cinfo->src->bytes_in_buffer = state->input_buffer_length; |
-} |
- |
-// Callback to fill the buffer. Since our buffer already contains all the data, |
-// we should never need to provide more data. If libjpeg thinks it needs more |
-// data, our input is probably corrupt. |
-// |
-// From the JPEG library: |
-// "This is called whenever bytes_in_buffer has reached zero and more data is |
-// wanted. In typical applications, it should read fresh data into the buffer |
-// (ignoring the current state of next_input_byte and bytes_in_buffer), reset |
-// the pointer & count to the start of the buffer, and return TRUE indicating |
-// that the buffer has been reloaded. It is not necessary to fill the buffer |
-// entirely, only to obtain at least one more byte. bytes_in_buffer MUST be |
-// set to a positive value if TRUE is returned. A FALSE return should only |
-// be used when I/O suspension is desired." |
-boolean FillInputBuffer(j_decompress_ptr cinfo) { |
- return false; |
-} |
- |
-// Skip data in the buffer. Since we have all the data at once, this operation |
-// is easy. It is not clear if this ever gets called because the JPEG library |
-// should be able to do the skip itself (it has all the data). |
-// |
-// From the JPEG library: |
-// "Skip num_bytes worth of data. The buffer pointer and count should be |
-// advanced over num_bytes input bytes, refilling the buffer as needed. This |
-// is used to skip over a potentially large amount of uninteresting data |
-// (such as an APPn marker). In some applications it may be possible to |
-// optimize away the reading of the skipped data, but it's not clear that |
-// being smart is worth much trouble; large skips are uncommon. |
-// bytes_in_buffer may be zero on return. A zero or negative skip count |
-// should be treated as a no-op." |
-void SkipInputData(j_decompress_ptr cinfo, long num_bytes) { |
- if (num_bytes > static_cast<long>(cinfo->src->bytes_in_buffer)) { |
- // Since all our data should be in the buffer, trying to skip beyond it |
- // means that there is some kind of error or corrupt input data. A 0 for |
- // bytes left means it will call FillInputBuffer which will then fail. |
- cinfo->src->next_input_byte += cinfo->src->bytes_in_buffer; |
- cinfo->src->bytes_in_buffer = 0; |
- } else if (num_bytes > 0) { |
- cinfo->src->bytes_in_buffer -= static_cast<size_t>(num_bytes); |
- cinfo->src->next_input_byte += num_bytes; |
- } |
-} |
- |
-// Our source doesn't need any cleanup, so this is a NOP. |
-// |
-// From the JPEG library: |
-// "Terminate source --- called by jpeg_finish_decompress() after all data has |
-// been read to clean up JPEG source manager. NOT called by jpeg_abort() or |
-// jpeg_destroy()." |
-void TermSource(j_decompress_ptr cinfo) { |
-} |
- |
-// Converts one row of rgb data to rgba data by adding a fully-opaque alpha |
-// value. |
-void AddAlpha(const unsigned char* rgb, int pixel_width, unsigned char* rgba) { |
- for (int x = 0; x < pixel_width; x++) { |
- const unsigned char* pixel_in = &rgb[x * 3]; |
- unsigned char* pixel_out = &rgba[x * 4]; |
- pixel_out[0] = pixel_in[0]; |
- pixel_out[1] = pixel_in[1]; |
- pixel_out[2] = pixel_in[2]; |
- pixel_out[3] = 0xff; |
- } |
-} |
- |
-// Converts one row of RGB data to BGRA by reordering the color components and |
-// adding alpha values of 0xff. |
-void RGBtoBGRA(const unsigned char* bgra, int pixel_width, unsigned char* rgb) |
-{ |
- for (int x = 0; x < pixel_width; x++) { |
- const unsigned char* pixel_in = &bgra[x * 3]; |
- unsigned char* pixel_out = &rgb[x * 4]; |
- pixel_out[0] = pixel_in[2]; |
- pixel_out[1] = pixel_in[1]; |
- pixel_out[2] = pixel_in[0]; |
- pixel_out[3] = 0xff; |
- } |
-} |
- |
-// This class destroys the given jpeg_decompress object when it goes out of |
-// scope. It simplifies the error handling in Decode (and even applies to the |
-// success case). |
-class DecompressDestroyer { |
- public: |
- DecompressDestroyer() : cinfo_(NULL) { |
- } |
- ~DecompressDestroyer() { |
- DestroyManagedObject(); |
- } |
- void SetManagedObject(jpeg_decompress_struct* ci) { |
- DestroyManagedObject(); |
- cinfo_ = ci; |
- } |
- void DestroyManagedObject() { |
- if (cinfo_) { |
- jpeg_destroy_decompress(cinfo_); |
- cinfo_ = NULL; |
- } |
- } |
- private: |
- jpeg_decompress_struct* cinfo_; |
-}; |
- |
-} // namespace |
- |
-bool JPEGCodec::Decode(const unsigned char* input, size_t input_size, |
- ColorFormat format, std::vector<unsigned char>* output, |
- int* w, int* h) { |
- jpeg_decompress_struct cinfo; |
- DecompressDestroyer destroyer; |
- output->clear(); |
- |
- // We set up the normal JPEG error routines, then override error_exit. |
- // This must be done before the call to create_decompress. |
- CoderErrorMgr errmgr; |
- cinfo.err = jpeg_std_error(&errmgr.pub); |
- errmgr.pub.error_exit = ErrorExit; |
- // Establish the setjmp return context for ErrorExit to use. |
- if (setjmp(errmgr.setjmp_buffer)) { |
- // If we get here, the JPEG code has signaled an error. |
- // See note in JPEGCodec::Encode() for why we need to destroy the cinfo |
- // manually here. |
- destroyer.DestroyManagedObject(); |
- return false; |
- } |
- |
- // The destroyer will destroy() cinfo on exit. We don't want to set the |
- // destroyer's object until cinfo is initialized. |
- jpeg_create_decompress(&cinfo); |
- destroyer.SetManagedObject(&cinfo); |
- |
- // set up the source manager |
- jpeg_source_mgr srcmgr; |
- srcmgr.init_source = InitSource; |
- srcmgr.fill_input_buffer = FillInputBuffer; |
- srcmgr.skip_input_data = SkipInputData; |
- srcmgr.resync_to_restart = jpeg_resync_to_restart; // use default routine |
- srcmgr.term_source = TermSource; |
- cinfo.src = &srcmgr; |
- |
- JpegDecoderState state(input, input_size); |
- cinfo.client_data = &state; |
- |
- // fill the file metadata into our buffer |
- if (jpeg_read_header(&cinfo, true) != JPEG_HEADER_OK) |
- return false; |
- |
- // we want to always get RGB data out |
- switch (cinfo.jpeg_color_space) { |
- case JCS_GRAYSCALE: |
- case JCS_RGB: |
- case JCS_YCbCr: |
- cinfo.out_color_space = JCS_RGB; |
- break; |
- case JCS_CMYK: |
- case JCS_YCCK: |
- default: |
- // Mozilla errors out on these color spaces, so I presume that the jpeg |
- // library can't do automatic color space conversion for them. We don't |
- // care about these anyway. |
- return false; |
- } |
- cinfo.output_components = 3; |
- |
- jpeg_calc_output_dimensions(&cinfo); |
- *w = cinfo.output_width; |
- *h = cinfo.output_height; |
- |
- jpeg_start_decompress(&cinfo); |
- |
- // FIXME(brettw) we may want to allow the capability for callers to request |
- // how to align row lengths as we do for the compressor. |
- int row_read_stride = cinfo.output_width * cinfo.output_components; |
- |
- if (format == FORMAT_RGB) { |
- // easy case, row needs no conversion |
- int row_write_stride = row_read_stride; |
- output->resize(row_write_stride * cinfo.output_height); |
- |
- for (int row = 0; row < static_cast<int>(cinfo.output_height); row++) { |
- unsigned char* rowptr = &(*output)[row * row_write_stride]; |
- if (!jpeg_read_scanlines(&cinfo, &rowptr, 1)) |
- return false; |
- } |
- } else { |
- // Rows need conversion to output format: read into a temporary buffer and |
- // expand to the final one. Performance: we could avoid the extra |
- // allocation by doing the expansion in-place. |
- int row_write_stride; |
- void (*converter)(const unsigned char* rgb, int w, unsigned char* out); |
- if (format == FORMAT_RGBA) { |
- row_write_stride = cinfo.output_width * 4; |
- converter = AddAlpha; |
- } else if (format == FORMAT_BGRA) { |
- row_write_stride = cinfo.output_width * 4; |
- converter = RGBtoBGRA; |
- } else { |
- NOTREACHED() << "Invalid pixel format"; |
- jpeg_destroy_decompress(&cinfo); |
- return false; |
- } |
- |
- output->resize(row_write_stride * cinfo.output_height); |
- |
- scoped_array<unsigned char> row_data(new unsigned char[row_read_stride]); |
- unsigned char* rowptr = row_data.get(); |
- for (int row = 0; row < static_cast<int>(cinfo.output_height); row++) { |
- if (!jpeg_read_scanlines(&cinfo, &rowptr, 1)) |
- return false; |
- converter(rowptr, *w, &(*output)[row * row_write_stride]); |
- } |
- } |
- |
- jpeg_finish_decompress(&cinfo); |
- jpeg_destroy_decompress(&cinfo); |
- return true; |
-} |
- |
-// static |
-SkBitmap* JPEGCodec::Decode(const unsigned char* input, size_t input_size) { |
- int w, h; |
- std::vector<unsigned char> data_vector; |
- // Use FORMAT_BGRA as that maps to Skia's 32 bit (kARGB_8888_Config) format. |
- if (!Decode(input, input_size, FORMAT_BGRA, &data_vector, &w, &h)) |
- return NULL; |
- |
- // Skia only handles 32 bit images. |
- int data_length = w * h * 4; |
- |
- SkBitmap* bitmap = new SkBitmap(); |
- bitmap->setConfig(SkBitmap::kARGB_8888_Config, w, h); |
- bitmap->allocPixels(); |
- memcpy(bitmap->getAddr32(0, 0), &data_vector[0], data_length); |
- |
- return bitmap; |
-} |
- |