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 // static, Leaky to allow access from any thread. |
17 const std::string& image_data, | 17 base::LazyInstance<ImageDecoder>::Leaky image_decoder_instance_ = |
18 ImageCodec image_codec) | 18 LAZY_INSTANCE_INITIALIZER; |
19 : delegate_(delegate), | 19 |
20 image_data_(image_data.begin(), image_data.end()), | 20 // static |
21 image_codec_(image_codec), | 21 ImageDecoder* ImageDecoder::GetInstance() { |
22 task_runner_(NULL), | 22 return image_decoder_instance_.Pointer(); |
23 shrink_to_fit_(false) { | |
24 } | 23 } |
25 | 24 |
26 ImageDecoder::ImageDecoder(Delegate* delegate, | 25 ImageDecoder::ImageDecoder() : image_decoder_impl_(new ImageDecoderImpl()) { |
27 const std::vector<char>& image_data, | |
28 ImageCodec image_codec) | |
29 : delegate_(delegate), | |
30 image_data_(image_data.begin(), image_data.end()), | |
31 image_codec_(image_codec), | |
32 task_runner_(NULL), | |
33 shrink_to_fit_(false) { | |
34 } | 26 } |
35 | 27 |
36 ImageDecoder::~ImageDecoder() {} | 28 ImageDecoder::~ImageDecoder() { |
37 | |
38 void ImageDecoder::Start(scoped_refptr<base::SequencedTaskRunner> task_runner) { | |
39 task_runner_ = task_runner; | |
40 BrowserThread::PostTask( | |
41 BrowserThread::IO, FROM_HERE, | |
42 base::Bind(&ImageDecoder::DecodeImageInSandbox, this, image_data_)); | |
43 } | 29 } |
44 | 30 |
45 bool ImageDecoder::OnMessageReceived(const IPC::Message& message) { | 31 ImageDecoder::Delegate::Delegate( |
32 const scoped_refptr<base::SequencedTaskRunner>& t) | |
33 : t(t) { | |
34 } | |
35 | |
36 ImageDecoder::Delegate::~Delegate() { | |
dcheng
2015/03/20 00:14:45
Call RemoveDelegate(this) here. Deleting this shou
Theresa
2015/03/20 19:09:01
Done.
| |
37 } | |
38 | |
39 void ImageDecoder::Start(Delegate* delegate, | |
40 const std::string& image_data, | |
41 ImageCodec image_codec, | |
42 bool shrink_to_fit) { | |
43 DCHECK(delegate); | |
44 DCHECK(delegate->task_runner()); | |
45 BrowserThread::PostTask( | |
46 BrowserThread::IO, FROM_HERE, | |
47 base::Bind( | |
48 &ImageDecoder::ImageDecoderImpl::DecodeImageInSandbox, | |
49 GetInstance()->image_decoder_impl_, delegate, | |
50 std::vector<unsigned char>(image_data.begin(), image_data.end()), | |
51 image_codec, shrink_to_fit)); | |
52 } | |
53 | |
54 void ImageDecoder::RemoveDelegate(Delegate* delegate) { | |
55 BrowserThread::PostTask( | |
56 BrowserThread::IO, FROM_HERE, | |
57 base::Bind(&ImageDecoder::ImageDecoderImpl::RemoveDelegate, | |
58 GetInstance()->image_decoder_impl_, delegate)); | |
59 } | |
60 | |
61 // ImageDecoder::ImageDecoderImpl Methods | |
62 | |
63 ImageDecoder::ImageDecoderImpl::ImageDecoderImpl() | |
64 : delegate_id_counter_(0), | |
65 last_request_(base::TimeTicks::Now()), | |
66 batch_mode_started_(false) { | |
67 } | |
68 | |
69 ImageDecoder::ImageDecoderImpl::~ImageDecoderImpl() { | |
70 } | |
71 | |
72 void ImageDecoder::ImageDecoderImpl::DecodeImageInSandbox( | |
73 Delegate* delegate, | |
74 const std::vector<unsigned char>& image_data, | |
75 ImageCodec image_codec, | |
76 bool shrink_to_fit) { | |
77 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); | |
78 if (!batch_mode_started_) { | |
79 StartBatchMode(); | |
80 } | |
81 | |
82 last_request_ = base::TimeTicks::Now(); | |
83 delegate_id_map_.AddWithID(delegate, delegate_id_counter_); | |
84 | |
85 switch (image_codec) { | |
86 case ROBUST_JPEG_CODEC: | |
87 utility_process_host_->Send(new ChromeUtilityMsg_RobustJPEGDecodeImage( | |
88 image_data, delegate_id_counter_)); | |
89 break; | |
90 case DEFAULT_CODEC: | |
91 utility_process_host_->Send(new ChromeUtilityMsg_DecodeImage( | |
92 image_data, shrink_to_fit, delegate_id_counter_)); | |
93 break; | |
94 default: | |
95 NOTREACHED(); | |
96 } | |
97 | |
98 ++delegate_id_counter_; | |
99 } | |
100 | |
101 void ImageDecoder::ImageDecoderImpl::RemoveDelegate(Delegate* delegate) { | |
102 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); | |
103 std::vector<int> canceled_requests; | |
104 | |
105 for (IDMap<Delegate>::iterator iter(&delegate_id_map_); !iter.IsAtEnd(); | |
106 iter.Advance()) { | |
107 if (iter.GetCurrentValue() == delegate) { | |
108 canceled_requests.push_back(iter.GetCurrentKey()); | |
109 } | |
110 } | |
111 | |
112 for (auto id : canceled_requests) { | |
113 delegate_id_map_.Remove(id); | |
114 } | |
115 } | |
116 | |
117 void ImageDecoder::ImageDecoderImpl::StartBatchMode() { | |
118 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); | |
119 utility_process_host_ = | |
120 UtilityProcessHost::Create(this, base::MessageLoopProxy::current().get()) | |
121 ->AsWeakPtr(); | |
122 utility_process_host_->StartBatchMode(); | |
123 batch_mode_started_ = true; | |
124 batch_mode_timer_.Start(FROM_HERE, kBatchModeTimeout, this, | |
125 &ImageDecoder::ImageDecoderImpl::StopBatchMode); | |
126 } | |
127 | |
128 void ImageDecoder::ImageDecoderImpl::StopBatchMode() { | |
129 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); | |
130 if ((base::TimeTicks::Now() - last_request_) < kBatchModeTimeout) { | |
131 return; | |
132 } | |
133 | |
134 if (utility_process_host_) { | |
135 utility_process_host_->EndBatchMode(); | |
136 utility_process_host_.reset(); | |
137 } | |
138 batch_mode_started_ = false; | |
139 batch_mode_timer_.Stop(); | |
140 } | |
141 | |
142 bool ImageDecoder::ImageDecoderImpl::OnMessageReceived( | |
143 const IPC::Message& message) { | |
46 bool handled = true; | 144 bool handled = true; |
47 IPC_BEGIN_MESSAGE_MAP(ImageDecoder, message) | 145 IPC_BEGIN_MESSAGE_MAP(ImageDecoder::ImageDecoderImpl, message) |
48 IPC_MESSAGE_HANDLER(ChromeUtilityHostMsg_DecodeImage_Succeeded, | 146 IPC_MESSAGE_HANDLER(ChromeUtilityHostMsg_DecodeImage_Succeeded, |
49 OnDecodeImageSucceeded) | 147 OnDecodeImageSucceeded) |
50 IPC_MESSAGE_HANDLER(ChromeUtilityHostMsg_DecodeImage_Failed, | 148 IPC_MESSAGE_HANDLER(ChromeUtilityHostMsg_DecodeImage_Failed, |
51 OnDecodeImageFailed) | 149 OnDecodeImageFailed) |
52 IPC_MESSAGE_UNHANDLED(handled = false) | 150 IPC_MESSAGE_UNHANDLED(handled = false) |
53 IPC_END_MESSAGE_MAP() | 151 IPC_END_MESSAGE_MAP() |
54 return handled; | 152 return handled; |
55 } | 153 } |
56 | 154 |
57 void ImageDecoder::OnDecodeImageSucceeded(const SkBitmap& decoded_image) { | 155 void ImageDecoder::ImageDecoderImpl::OnDecodeImageSucceeded( |
58 DCHECK(task_runner_->RunsTasksOnCurrentThread()); | 156 const SkBitmap& decoded_image, |
59 if (delegate_) | 157 int id) { |
60 delegate_->OnImageDecoded(this, decoded_image); | 158 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); |
159 if (delegate_id_map_.Lookup(id)) { | |
160 Delegate* delegate = delegate_id_map_.Lookup(id); | |
161 delegate->task_runner()->PostTask( | |
162 FROM_HERE, base::Bind(&Delegate::OnImageDecoded, | |
163 base::Unretained(delegate), decoded_image)); | |
164 | |
165 delegate_id_map_.Remove(id); | |
166 } | |
61 } | 167 } |
62 | 168 |
63 void ImageDecoder::OnDecodeImageFailed() { | 169 void ImageDecoder::ImageDecoderImpl::OnDecodeImageFailed(int id) { |
64 DCHECK(task_runner_->RunsTasksOnCurrentThread()); | 170 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); |
65 if (delegate_) | 171 if (delegate_id_map_.Lookup(id)) { |
66 delegate_->OnDecodeImageFailed(this); | 172 Delegate* delegate = delegate_id_map_.Lookup(id); |
67 } | 173 delegate->task_runner()->PostTask( |
174 FROM_HERE, | |
175 base::Bind(&Delegate::OnDecodeImageFailed, base::Unretained(delegate))); | |
68 | 176 |
69 void ImageDecoder::DecodeImageInSandbox( | 177 delegate_id_map_.Remove(id); |
70 const std::vector<unsigned char>& image_data) { | |
71 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); | |
72 UtilityProcessHost* utility_process_host; | |
73 utility_process_host = UtilityProcessHost::Create(this, task_runner_.get()); | |
74 if (image_codec_ == ROBUST_JPEG_CODEC) { | |
75 utility_process_host->Send( | |
76 new ChromeUtilityMsg_RobustJPEGDecodeImage(image_data)); | |
77 } else { | |
78 utility_process_host->Send( | |
79 new ChromeUtilityMsg_DecodeImage(image_data, shrink_to_fit_)); | |
80 } | 178 } |
81 } | 179 } |
OLD | NEW |