Chromium Code Reviews| OLD | NEW |
|---|---|
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "chrome/browser/image_decoder.h" | 5 #include "chrome/browser/image_decoder.h" |
| 6 | 6 |
| 7 #include "base/bind.h" | 7 #include "base/bind.h" |
| 8 #include "chrome/browser/browser_process.h" | 8 #include "chrome/browser/browser_process.h" |
| 9 #include "chrome/common/chrome_utility_messages.h" | 9 #include "chrome/common/chrome_utility_messages.h" |
| 10 #include "content/public/browser/browser_thread.h" | 10 #include "content/public/browser/browser_thread.h" |
| 11 #include "content/public/browser/utility_process_host.h" | 11 #include "content/public/browser/utility_process_host.h" |
| 12 | 12 |
| 13 using content::BrowserThread; | 13 using content::BrowserThread; |
| 14 using content::UtilityProcessHost; | 14 using content::UtilityProcessHost; |
| 15 | 15 |
| 16 ImageDecoder::ImageDecoder(Delegate* delegate, | 16 ImageDecoder::ImageDecoder(Delegate* delegate, |
| 17 const std::string& image_data, | 17 const std::string& image_data, |
| 18 ImageCodec image_codec) | 18 ImageCodec image_codec) |
| 19 : delegate_(delegate), | 19 : delegate_(delegate), |
| 20 image_data_(image_data.begin(), image_data.end()), | 20 image_data_(image_data.begin(), image_data.end()), |
| 21 image_codec_(image_codec), | 21 image_codec_(image_codec), |
| 22 target_thread_id_(BrowserThread::UI) { | 22 target_thread_id_(BrowserThread::UI), |
| 23 decode_on_sequence_worker_thread_(false) { | |
| 23 } | 24 } |
| 24 | 25 |
| 25 ImageDecoder::~ImageDecoder() {} | 26 ImageDecoder::~ImageDecoder() {} |
| 26 | 27 |
| 27 void ImageDecoder::Start() { | 28 void ImageDecoder::Start() { |
| 29 decode_on_sequence_worker_thread_ = false; | |
| 28 if (!BrowserThread::GetCurrentThreadIdentifier(&target_thread_id_)) { | 30 if (!BrowserThread::GetCurrentThreadIdentifier(&target_thread_id_)) { |
| 29 NOTREACHED(); | 31 NOTREACHED(); |
| 30 return; | 32 return; |
| 31 } | 33 } |
| 32 BrowserThread::PostTask( | 34 BrowserThread::PostTask( |
| 33 BrowserThread::IO, FROM_HERE, | 35 BrowserThread::IO, FROM_HERE, |
| 34 base::Bind(&ImageDecoder::DecodeImageInSandbox, this, image_data_)); | 36 base::Bind(&ImageDecoder::DecodeImageInSandbox, this, image_data_)); |
| 35 } | 37 } |
| 36 | 38 |
| 39 void ImageDecoder::Start( | |
| 40 const base::SequencedWorkerPool::SequenceToken& sequence_token) { | |
|
Joao da Silva
2013/01/21 08:07:39
Why not have a single Start() that gets a TaskRunn
bshe
2013/01/21 22:53:07
Good idea!
We only have one Start function now.
O
| |
| 41 sequence_token_ = sequence_token; | |
| 42 decode_on_sequence_worker_thread_ = true; | |
| 43 BrowserThread::PostTask( | |
| 44 BrowserThread::IO, FROM_HERE, | |
| 45 base::Bind(&ImageDecoder::DecodeImageInSandbox, this, image_data_)); | |
|
Joao da Silva
2013/01/21 08:07:39
nit: indent (4 spaces)
bshe
2013/01/21 22:53:07
This is an argument of function PostTask, same as
| |
| 46 } | |
| 47 | |
| 37 bool ImageDecoder::OnMessageReceived(const IPC::Message& message) { | 48 bool ImageDecoder::OnMessageReceived(const IPC::Message& message) { |
| 38 bool handled = true; | 49 bool handled = true; |
| 39 IPC_BEGIN_MESSAGE_MAP(ImageDecoder, message) | 50 IPC_BEGIN_MESSAGE_MAP(ImageDecoder, message) |
| 40 IPC_MESSAGE_HANDLER(ChromeUtilityHostMsg_DecodeImage_Succeeded, | 51 IPC_MESSAGE_HANDLER(ChromeUtilityHostMsg_DecodeImage_Succeeded, |
| 41 OnDecodeImageSucceeded) | 52 OnDecodeImageSucceeded) |
| 42 IPC_MESSAGE_HANDLER(ChromeUtilityHostMsg_DecodeImage_Failed, | 53 IPC_MESSAGE_HANDLER(ChromeUtilityHostMsg_DecodeImage_Failed, |
| 43 OnDecodeImageFailed) | 54 OnDecodeImageFailed) |
| 44 IPC_MESSAGE_UNHANDLED(handled = false) | 55 IPC_MESSAGE_UNHANDLED(handled = false) |
| 45 IPC_END_MESSAGE_MAP() | 56 IPC_END_MESSAGE_MAP() |
| 46 return handled; | 57 return handled; |
| 47 } | 58 } |
| 48 | 59 |
| 60 bool ImageDecoder::CalledOnValidThreadOrWorkerThread() { | |
| 61 if (decode_on_sequence_worker_thread_) { | |
| 62 return BrowserThread::GetBlockingPool()-> | |
| 63 IsRunningSequenceOnCurrentThread(sequence_token_); | |
| 64 } else { | |
| 65 return BrowserThread::CurrentlyOn(target_thread_id_); | |
| 66 } | |
| 67 } | |
| 68 | |
| 49 void ImageDecoder::OnDecodeImageSucceeded(const SkBitmap& decoded_image) { | 69 void ImageDecoder::OnDecodeImageSucceeded(const SkBitmap& decoded_image) { |
| 50 DCHECK(BrowserThread::CurrentlyOn(target_thread_id_)); | 70 DCHECK(CalledOnValidThreadOrWorkerThread()); |
| 51 if (delegate_) | 71 if (delegate_) |
| 52 delegate_->OnImageDecoded(this, decoded_image); | 72 delegate_->OnImageDecoded(this, decoded_image); |
| 53 } | 73 } |
| 54 | 74 |
| 55 void ImageDecoder::OnDecodeImageFailed() { | 75 void ImageDecoder::OnDecodeImageFailed() { |
| 56 DCHECK(BrowserThread::CurrentlyOn(target_thread_id_)); | 76 DCHECK(CalledOnValidThreadOrWorkerThread()); |
| 57 if (delegate_) | 77 if (delegate_) |
| 58 delegate_->OnDecodeImageFailed(this); | 78 delegate_->OnDecodeImageFailed(this); |
| 59 } | 79 } |
| 60 | 80 |
| 61 void ImageDecoder::DecodeImageInSandbox( | 81 void ImageDecoder::DecodeImageInSandbox( |
| 62 const std::vector<unsigned char>& image_data) { | 82 const std::vector<unsigned char>& image_data) { |
| 63 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); | 83 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); |
| 64 UtilityProcessHost* utility_process_host = UtilityProcessHost::Create( | 84 UtilityProcessHost* utility_process_host; |
| 65 this, BrowserThread::GetMessageLoopProxyForThread(target_thread_id_)); | 85 if (decode_on_sequence_worker_thread_) { |
| 86 utility_process_host = UtilityProcessHost::Create( | |
| 87 this, BrowserThread::GetBlockingPool()->GetSequencedTaskRunner( | |
| 88 sequence_token_).get()); | |
| 89 } else { | |
| 90 utility_process_host = UtilityProcessHost::Create( | |
| 91 this, BrowserThread::GetMessageLoopProxyForThread(target_thread_id_)); | |
| 92 } | |
| 66 utility_process_host->EnableZygote(); | 93 utility_process_host->EnableZygote(); |
| 67 if (image_codec_ == ROBUST_JPEG_CODEC) { | 94 if (image_codec_ == ROBUST_JPEG_CODEC) { |
| 68 utility_process_host->Send( | 95 utility_process_host->Send( |
| 69 new ChromeUtilityMsg_RobustJPEGDecodeImage(image_data)); | 96 new ChromeUtilityMsg_RobustJPEGDecodeImage(image_data)); |
| 70 } else { | 97 } else { |
| 71 utility_process_host->Send(new ChromeUtilityMsg_DecodeImage(image_data)); | 98 utility_process_host->Send(new ChromeUtilityMsg_DecodeImage(image_data)); |
| 72 } | 99 } |
| 73 } | 100 } |
| OLD | NEW |