Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(2703)

Unified Diff: chrome/common/jpeg_codec.cc

Issue 18838: Try2 of issue 18772. (move jpeg_codec* to base/gfx) (Closed) Base URL: svn://chrome-svn/chrome/trunk/src/
Patch Set: Created 11 years, 11 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
« no previous file with comments | « chrome/common/jpeg_codec.h ('k') | chrome/common/jpeg_codec_unittest.cc » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
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;
-}
-
« no previous file with comments | « chrome/common/jpeg_codec.h ('k') | chrome/common/jpeg_codec_unittest.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698