OLD | NEW |
| (Empty) |
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 | |
3 // found in the LICENSE file. | |
4 | |
5 #include "chrome/browser/ui/webui/options2/chromeos/wallpaper_source.h" | |
6 | |
7 #include "ash/desktop_background/desktop_background_controller.h" | |
8 #include "ash/shell.h" | |
9 #include "base/debug/trace_event.h" | |
10 #include "base/synchronization/cancellation_flag.h" | |
11 #include "base/threading/worker_pool.h" | |
12 #include "chrome/browser/chromeos/login/user_manager.h" | |
13 #include "chrome/common/url_constants.h" | |
14 #include "content/public/browser/browser_thread.h" | |
15 #include "grit/ui_resources.h" | |
16 #include "net/base/mime_util.h" | |
17 #include "ui/gfx/codec/png_codec.h" | |
18 #include "ui/gfx/size.h" | |
19 #include "ui/gfx/skia_util.h" | |
20 | |
21 extern "C" { | |
22 #if defined(USE_SYSTEM_ZLIB) | |
23 #include <zlib.h> | |
24 #else | |
25 #include "third_party/zlib/zlib.h" | |
26 #endif | |
27 } | |
28 | |
29 namespace chromeos { | |
30 namespace options { | |
31 | |
32 // Operation class that encodes existing in-memory image as PNG. | |
33 // It uses NO-COMPRESSION to save time. | |
34 class WallpaperImageSource::WallpaperEncodingOperation | |
35 : public base::RefCountedThreadSafe< | |
36 WallpaperImageSource::WallpaperEncodingOperation> { | |
37 public: | |
38 WallpaperEncodingOperation( | |
39 int request_id, | |
40 scoped_refptr<base::RefCountedBytes> data, | |
41 SkBitmap image) | |
42 : request_id_(request_id), | |
43 data_(data), | |
44 image_(image) { | |
45 } | |
46 | |
47 static void Run(scoped_refptr<WallpaperEncodingOperation> weo) { | |
48 weo->EncodeWallpaper(); | |
49 } | |
50 | |
51 int request_id() { | |
52 return request_id_; | |
53 } | |
54 | |
55 void EncodeWallpaper() { | |
56 if (cancel_flag_.IsSet()) | |
57 return; | |
58 TRACE_EVENT0("LOCK_SCREEN", "imageEncoding"); | |
59 SkAutoLockPixels lock_input(image_); | |
60 | |
61 if (!image_.readyToDraw()) | |
62 return; | |
63 | |
64 // Avoid compression to make things faster. | |
65 gfx::PNGCodec::EncodeWithCompressionLevel( | |
66 reinterpret_cast<unsigned char*>(image_.getAddr32(0, 0)), | |
67 gfx::PNGCodec::FORMAT_SkBitmap, | |
68 gfx::Size(image_.width(), image_.height()), | |
69 image_.width() * image_.bytesPerPixel(), | |
70 false, | |
71 std::vector<gfx::PNGCodec::Comment>(), | |
72 Z_NO_COMPRESSION, | |
73 &data_->data()); | |
74 if (cancel_flag_.IsSet()) | |
75 return; | |
76 } | |
77 | |
78 void Cancel() { | |
79 cancel_flag_.Set(); | |
80 } | |
81 | |
82 private: | |
83 friend class base::RefCountedThreadSafe< | |
84 WallpaperImageSource::WallpaperEncodingOperation>; | |
85 | |
86 ~WallpaperEncodingOperation() {} | |
87 | |
88 base::CancellationFlag cancel_flag_; | |
89 | |
90 // ID of original request. | |
91 int request_id_; | |
92 // Buffer to store encoded image. | |
93 scoped_refptr<base::RefCountedBytes> data_; | |
94 // Original image to encode. | |
95 SkBitmap image_; | |
96 | |
97 DISALLOW_COPY_AND_ASSIGN(WallpaperEncodingOperation); | |
98 }; | |
99 | |
100 WallpaperImageSource::WallpaperImageSource() | |
101 : DataSource(chrome::kChromeUIWallpaperImageHost, NULL), | |
102 weak_ptr_factory_(ALLOW_THIS_IN_INITIALIZER_LIST(this)) { | |
103 } | |
104 | |
105 WallpaperImageSource::~WallpaperImageSource() { | |
106 } | |
107 | |
108 void WallpaperImageSource::StartDataRequest(const std::string& email, | |
109 bool is_incognito, | |
110 int request_id) { | |
111 DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::IO)); | |
112 TRACE_EVENT_ASYNC_BEGIN0("SCREEN_LOCK", "GetUserWallpaperDataRequest", | |
113 request_id); | |
114 CancelPendingEncodingOperation(); | |
115 content::BrowserThread::PostTask( | |
116 content::BrowserThread::UI, | |
117 FROM_HERE, | |
118 base::Bind(&WallpaperImageSource::GetCurrentUserWallpaper, this, | |
119 request_id)); | |
120 } | |
121 | |
122 std::string WallpaperImageSource::GetMimeType(const std::string&) const { | |
123 return "images/png"; | |
124 } | |
125 | |
126 // Get current background image and store it to |data|. | |
127 void WallpaperImageSource::GetCurrentUserWallpaper(int request_id) { | |
128 SkBitmap image; | |
129 TRACE_EVENT0("LOCK_SCREEN", "GetCurrentUserWallpaper"); | |
130 if (chromeos::UserManager::Get()->IsUserLoggedIn()) { | |
131 // TODO(sad|bshe): It maybe necessary to include the scale factor in the | |
132 // request (as is done for user-image and wallpaper-thumbnails). | |
133 SkBitmap wallpaper; | |
134 gfx::ImageSkia wallpaper_skia = ash::Shell::GetInstance()-> | |
135 desktop_background_controller()->GetCurrentWallpaperImage(); | |
136 if (!wallpaper_skia.empty()) | |
137 wallpaper = *wallpaper_skia.bitmap(); | |
138 SkBitmap copy; | |
139 if (wallpaper.deepCopyTo(©, wallpaper.config())) | |
140 image = copy; | |
141 } | |
142 content::BrowserThread::PostTask( | |
143 content::BrowserThread::IO, | |
144 FROM_HERE, | |
145 base::Bind(&WallpaperImageSource::ImageAcquired, this, image, request_id)); | |
146 } | |
147 | |
148 | |
149 void WallpaperImageSource::ImageAcquired(SkBitmap image, | |
150 int request_id) { | |
151 DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::IO)); | |
152 CancelPendingEncodingOperation(); | |
153 scoped_refptr<base::RefCountedBytes> data = new base::RefCountedBytes(); | |
154 wallpaper_encoding_op_ = new WallpaperEncodingOperation(request_id, | |
155 data, | |
156 image); | |
157 base::WorkerPool::PostTaskAndReply( | |
158 FROM_HERE, | |
159 base::Bind(&WallpaperEncodingOperation::Run, wallpaper_encoding_op_), | |
160 base::Bind(&WallpaperImageSource::SendCurrentUserWallpaper, | |
161 weak_ptr_factory_.GetWeakPtr(), request_id, data), | |
162 true /* task_is_slow */); | |
163 }; | |
164 | |
165 void WallpaperImageSource::CancelPendingEncodingOperation() { | |
166 DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::IO)); | |
167 // Set canceled flag of previous request to skip unneeded encoding. | |
168 if (wallpaper_encoding_op_.get()) { | |
169 wallpaper_encoding_op_->Cancel(); | |
170 SendResponse(wallpaper_encoding_op_->request_id(), NULL); | |
171 TRACE_EVENT_ASYNC_END0("SCREEN_LOCK", "GetUserWallpaper", | |
172 wallpaper_encoding_op_->request_id()); | |
173 } | |
174 | |
175 // Cancel reply callback for previous request. | |
176 weak_ptr_factory_.InvalidateWeakPtrs(); | |
177 } | |
178 | |
179 void WallpaperImageSource::SendCurrentUserWallpaper(int request_id, | |
180 scoped_refptr<base::RefCountedBytes> data) { | |
181 DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::IO)); | |
182 SendResponse(request_id, data); | |
183 TRACE_EVENT_ASYNC_END0("SCREEN_LOCK", "GetUserWallpaper", request_id); | |
184 } | |
185 | |
186 } // namespace options | |
187 } // namespace chromeos | |
OLD | NEW |