Index: chrome/browser/image_decoder.h |
diff --git a/chrome/browser/image_decoder.h b/chrome/browser/image_decoder.h |
index 2206dd9f454ebcbfc3a29674cf41e1d42eb594ee..fcb83ccfc52182d1f5f4c519f62ad6a03659ea44 100644 |
--- a/chrome/browser/image_decoder.h |
+++ b/chrome/browser/image_decoder.h |
@@ -5,30 +5,37 @@ |
#ifndef CHROME_BROWSER_IMAGE_DECODER_H_ |
#define CHROME_BROWSER_IMAGE_DECODER_H_ |
+#include <map> |
#include <string> |
#include <vector> |
#include "base/compiler_specific.h" |
+#include "base/id_map.h" |
+#include "base/lazy_instance.h" |
#include "base/memory/ref_counted.h" |
#include "base/threading/sequenced_worker_pool.h" |
+#include "base/timer/timer.h" |
+#include "content/public/browser/utility_process_host.h" |
#include "content/public/browser/utility_process_host_client.h" |
class SkBitmap; |
-// Decodes an image in a sandboxed process. |
-class ImageDecoder : public content::UtilityProcessHostClient { |
+// This is a helper class for decoding images safely in a utility process. To |
+// use this, call ImageDecoder::GetInstance()->Start(...) on any thread. |
+// |
+// Internally, all of the work happens on the IO thread, and then |
+// the result (Delegate::OnImageDecoded or Delegate::OnDecodeImageFailed) is |
+// posted back to the task runner specified when Start(...) was called. |
+class ImageDecoder { |
public: |
class Delegate { |
public: |
// Called when image is decoded. |
- // |decoder| is used to identify the image in case of decoding several |
- // images simultaneously. |
- virtual void OnImageDecoded(const ImageDecoder* decoder, |
- const SkBitmap& decoded_image) = 0; |
+ virtual void OnImageDecoded(const SkBitmap& decoded_image) = 0; |
// Called when decoding image failed. Delegate can do some cleanup in |
// this handler. |
- virtual void OnDecodeImageFailed(const ImageDecoder* decoder) {} |
+ virtual void OnDecodeImageFailed() {} |
protected: |
virtual ~Delegate() {} |
@@ -39,44 +46,93 @@ class ImageDecoder : public content::UtilityProcessHostClient { |
ROBUST_JPEG_CODEC, // Restrict decoding to robust jpeg codec. |
}; |
- ImageDecoder(Delegate* delegate, |
- const std::string& image_data, |
- ImageCodec image_codec); |
- |
- ImageDecoder(Delegate* delegate, |
- const std::vector<char>& image_data, |
- ImageCodec image_codec); |
+ static ImageDecoder* GetInstance(); |
// Starts asynchronous image decoding. Once finished, the callback will be |
// posted back to |task_runner|. |
- void Start(scoped_refptr<base::SequencedTaskRunner> task_runner); |
- |
- const std::vector<unsigned char>& get_image_data() const { |
- return image_data_; |
- } |
+ void Start(Delegate* delegate, |
+ const std::string& image_data, |
+ ImageCodec image_codec, |
+ scoped_refptr<base::SequencedTaskRunner> task_runner, |
+ bool shrink_to_fit); |
- void set_delegate(Delegate* delegate) { delegate_ = delegate; } |
- void set_shrink_to_fit(bool shrink_to_fit) { shrink_to_fit_ = shrink_to_fit; } |
+ // Removes all instances of delegate from the IDMap, ensuring callbacks |
+ // are not made to the delegate after it is destroyed. |
+ void RemoveDelegate(Delegate* delegate); |
private: |
- // It's a reference counted object, so destructor is private. |
- ~ImageDecoder() override; |
+ friend struct base::DefaultLazyInstanceTraits<ImageDecoder>; |
+ ImageDecoder(); |
+ ~ImageDecoder(); |
+ |
+ class ImageDecoderImpl : public content::UtilityProcessHostClient { |
+ public: |
+ ImageDecoderImpl(); |
+ |
+ // Sends a request to the sandboxed process to decode the image. Starts |
+ // batch mode if necessary. |
+ void DecodeImageInSandbox( |
+ Delegate* delegate, |
+ const std::vector<unsigned char>& image_data, |
+ ImageCodec image_codec, |
+ scoped_refptr<base::SequencedTaskRunner> task_runner, |
+ bool shrink_to_fit); |
+ |
+ void RemoveDelegate(Delegate* delegate); |
+ |
+ private: |
+ // It's a reference counted object, so destructor is private. |
+ ~ImageDecoderImpl() override; |
+ |
+ // Contains attributes we need to know about each image decode |
+ // request we send. |
+ struct RequestInfo { |
+ RequestInfo(Delegate* delegate, |
+ scoped_refptr<base::SequencedTaskRunner> task_runner); |
+ ~RequestInfo(); |
+ |
+ Delegate* delegate; |
+ scoped_refptr<base::SequencedTaskRunner> task_runner; |
+ }; |
- // Overidden from UtilityProcessHostClient: |
- bool OnMessageReceived(const IPC::Message& message) override; |
+ // Starts UtilityProcessHost in batch mode and starts batch_mode_timer_. |
+ void StartBatchMode(); |
- // IPC message handlers. |
- void OnDecodeImageSucceeded(const SkBitmap& decoded_image); |
- void OnDecodeImageFailed(); |
+ // Stops batch mode if no requests have come in since kBatchModeTimeout. |
+ void StopBatchMode(); |
- // Launches sandboxed process that will decode the image. |
- void DecodeImageInSandbox(const std::vector<unsigned char>& image_data); |
+ // Overidden from UtilityProcessHostClient. |
+ bool OnMessageReceived(const IPC::Message& message) override; |
+ |
+ // IPC message handlers. |
+ void OnDecodeImageSucceeded(const SkBitmap& decoded_image, int id); |
+ void OnDecodeImageFailed(int id); |
+ |
+ // id to use for the next Start request that comes in. |
+ int delegate_id_counter_; |
+ |
+ // Map of request id's to RequestInfo. |
+ IDMap<RequestInfo> request_info_id_map_; |
+ |
+ // The UtilityProcessHost requests are sent to. |
+ base::WeakPtr<content::UtilityProcessHost> utility_process_host_; |
+ |
+ // Calls StopBatchMode after kBatchModeTimeout has elapsed. |
+ base::RepeatingTimer<ImageDecoderImpl> batch_mode_timer_; |
+ |
+ // The time Start was last called. |
+ base::TimeTicks last_request_; |
+ |
+ // How long to wait after the last request has been received before ending |
+ // batch mode. |
+ const base::TimeDelta kBatchModeTimeout = base::TimeDelta::FromSeconds(5); |
+ |
+ // True iff utility_process_host_ has been created and started in |
+ // batch mode. |
+ bool batch_mode_started_; |
+ }; |
- Delegate* delegate_; |
- std::vector<unsigned char> image_data_; |
- const ImageCodec image_codec_; |
- scoped_refptr<base::SequencedTaskRunner> task_runner_; |
- bool shrink_to_fit_; // if needed for IPC msg size limit |
+ const scoped_refptr<ImageDecoderImpl> image_decoder_impl_; |
DISALLOW_COPY_AND_ASSIGN(ImageDecoder); |
}; |