1 // Copyright (c) 2011 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/themes/browser_theme_pack.h"
6
7 #include "base/stl_util-inl.h"
8 #include "base/string_util.h"
9 #include "base/threading/thread_restrictions.h"
10 #include "base/utf_string_conversions.h"
11 #include "base/values.h"
12 #include "chrome/browser/themes/theme_service.h"
13 #include "content/browser/browser_thread.h"
14 #include "grit/app_resources.h"
15 #include "grit/theme_resources.h"
16 #include "net/base/file_stream.h"
17 #include "net/base/net_errors.h"
18 #include "third_party/skia/include/core/SkCanvas.h"
19 #include "ui/base/resource/data_pack.h"
20 #include "ui/base/resource/resource_bundle.h"
21 #include "ui/gfx/codec/png_codec.h"
22 #include "ui/gfx/skbitmap_operations.h"
23
24 namespace {
25
26 // Version number of the current theme pack. We just throw out and rebuild
27 // theme packs that aren't int-equal to this.
28 const int kThemePackVersion = 15;
29
30 // IDs that are in the DataPack won't clash with the positive integer
31 // int32_t. kHeaderID should always have the maximum value because we want the
32 // "header" to be written last. That way we can detect whether the pack was
33 // successfully written and ignore and regenerate if it was only partially
34 // written (i.e. chrome crashed on a different thread while writing the pack).
35 const int kHeaderID = UINT_MAX - 1;
36 const int kTintsID = UINT_MAX - 2;
37 const int kColorsID = UINT_MAX - 3;
38 const int kDisplayPropertiesID = UINT_MAX - 4;
39 const int kSourceImagesID = UINT_MAX - 5;
40
41 // Static size of the tint/color/display property arrays that are mmapped.
42 const int kTintArraySize = 6;
43 const int kColorArraySize = 19;
44 const int kDisplayPropertySize = 3;
45
46 // The sum of kFrameBorderThickness and kNonClientRestoredExtraThickness from
47 // OpaqueBrowserFrameView.
48 const int kRestoredTabVerticalOffset = 15;
49
50 // Persistent constants for the main images that we need. These have the same
51 // names as their IDR_* counterparts but these values will always stay the
52 // same.
53 const int PRS_THEME_FRAME = 1;
54 const int PRS_THEME_FRAME_INACTIVE = 2;
55 const int PRS_THEME_FRAME_INCOGNITO = 3;
56 const int PRS_THEME_FRAME_INCOGNITO_INACTIVE = 4;
57 const int PRS_THEME_TOOLBAR = 5;
58 const int PRS_THEME_TAB_BACKGROUND = 6;
59 const int PRS_THEME_TAB_BACKGROUND_INCOGNITO = 7;
60 const int PRS_THEME_TAB_BACKGROUND_V = 8;
61 const int PRS_THEME_NTP_BACKGROUND = 9;
62 const int PRS_THEME_FRAME_OVERLAY = 10;
63 const int PRS_THEME_FRAME_OVERLAY_INACTIVE = 11;
64 const int PRS_THEME_BUTTON_BACKGROUND = 12;
65 const int PRS_THEME_NTP_ATTRIBUTION = 13;
66 const int PRS_THEME_WINDOW_CONTROL_BACKGROUND = 14;
67
68 struct PersistingImagesTable {
69 // A non-changing integer ID meant to be saved in theme packs. This ID must
70 // not change between versions of chrome.
71 int persistent_id;
72
73 // The IDR that depends on the whims of GRIT and therefore changes whenever
74 // someone adds a new resource.
75 int idr_id;
76
77 // String to check for when parsing theme manifests or NULL if this isn't
78 // supposed to be changeable by the user.
79 const char* key;
80 };
81
82 // IDR_* resource names change whenever new resources are added; use persistent
83 // IDs when storing to a cached pack.
84 PersistingImagesTable kPersistingImages[] = {
85 { PRS_THEME_FRAME, IDR_THEME_FRAME,
86 "theme_frame" },
87 { PRS_THEME_FRAME_INACTIVE, IDR_THEME_FRAME_INACTIVE,
88 "theme_frame_inactive" },
89 { PRS_THEME_FRAME_INCOGNITO, IDR_THEME_FRAME_INCOGNITO,
90 "theme_frame_incognito" },
91 { PRS_THEME_FRAME_INCOGNITO_INACTIVE, IDR_THEME_FRAME_INCOGNITO_INACTIVE,
92 "theme_frame_incognito_inactive" },
93 { PRS_THEME_TOOLBAR, IDR_THEME_TOOLBAR,
94 "theme_toolbar" },
95 { PRS_THEME_TAB_BACKGROUND, IDR_THEME_TAB_BACKGROUND,
96 "theme_tab_background" },
97 { PRS_THEME_TAB_BACKGROUND_INCOGNITO, IDR_THEME_TAB_BACKGROUND_INCOGNITO,
98 "theme_tab_background_incognito" },
99 { PRS_THEME_TAB_BACKGROUND_V, IDR_THEME_TAB_BACKGROUND_V,
100 "theme_tab_background_v"},
101 { PRS_THEME_NTP_BACKGROUND, IDR_THEME_NTP_BACKGROUND,
102 "theme_ntp_background" },
103 { PRS_THEME_FRAME_OVERLAY, IDR_THEME_FRAME_OVERLAY,
104 "theme_frame_overlay" },
105 { PRS_THEME_FRAME_OVERLAY_INACTIVE, IDR_THEME_FRAME_OVERLAY_INACTIVE,
106 "theme_frame_overlay_inactive" },
107 { PRS_THEME_BUTTON_BACKGROUND, IDR_THEME_BUTTON_BACKGROUND,
108 "theme_button_background" },
109 { PRS_THEME_NTP_ATTRIBUTION, IDR_THEME_NTP_ATTRIBUTION,
110 "theme_ntp_attribution" },
111 { PRS_THEME_WINDOW_CONTROL_BACKGROUND, IDR_THEME_WINDOW_CONTROL_BACKGROUND,
112 "theme_window_control_background"},
113
114 // The rest of these entries have no key because they can't be overridden
115 // from the json manifest.
116 { 15, IDR_BACK, NULL },
117 { 16, IDR_BACK_D, NULL },
118 { 17, IDR_BACK_H, NULL },
119 { 18, IDR_BACK_P, NULL },
120 { 19, IDR_FORWARD, NULL },
121 { 20, IDR_FORWARD_D, NULL },
122 { 21, IDR_FORWARD_H, NULL },
123 { 22, IDR_FORWARD_P, NULL },
124 { 23, IDR_HOME, NULL },
125 { 24, IDR_HOME_H, NULL },
126 { 25, IDR_HOME_P, NULL },
127 { 26, IDR_RELOAD, NULL },
128 { 27, IDR_RELOAD_H, NULL },
129 { 28, IDR_RELOAD_P, NULL },
130 { 29, IDR_STOP, NULL },
131 { 30, IDR_STOP_D, NULL },
132 { 31, IDR_STOP_H, NULL },
133 { 32, IDR_STOP_P, NULL },
134 { 33, IDR_LOCATIONBG_C, NULL },
135 { 34, IDR_LOCATIONBG_L, NULL },
136 { 35, IDR_LOCATIONBG_R, NULL },
137 { 36, IDR_BROWSER_ACTIONS_OVERFLOW, NULL },
138 { 37, IDR_BROWSER_ACTIONS_OVERFLOW_H, NULL },
139 { 38, IDR_BROWSER_ACTIONS_OVERFLOW_P, NULL },
140 { 39, IDR_TOOLS, NULL },
141 { 40, IDR_TOOLS_H, NULL },
142 { 41, IDR_TOOLS_P, NULL },
143 { 42, IDR_MENU_DROPARROW, NULL },
144 { 43, IDR_THROBBER, NULL },
145 { 44, IDR_THROBBER_WAITING, NULL },
146 { 45, IDR_THROBBER_LIGHT, NULL },
147 };
148
GetPersistentIDByName(const std::string & key)149 int GetPersistentIDByName(const std::string& key) {
150 for (size_t i = 0; i < arraysize(kPersistingImages); ++i) {
151 if (kPersistingImages[i].key != NULL &&
152 base::strcasecmp(key.c_str(), kPersistingImages[i].key) == 0) {
153 return kPersistingImages[i].persistent_id;
154 }
155 }
156
157 return -1;
158 }
159
GetPersistentIDByIDR(int idr)160 int GetPersistentIDByIDR(int idr) {
161 for (size_t i = 0; i < arraysize(kPersistingImages); ++i) {
162 if (kPersistingImages[i].idr_id == idr) {
163 return kPersistingImages[i].persistent_id;
164 }
165 }
166
167 return -1;
168 }
169
170 struct StringToIntTable {
171 const char* key;
172 int id;
173 };
174
175 // Strings used by themes to identify tints in the JSON.
176 StringToIntTable kTintTable[] = {
177 { "buttons", ThemeService::TINT_BUTTONS },
178 { "frame", ThemeService::TINT_FRAME },
179 { "frame_inactive", ThemeService::TINT_FRAME_INACTIVE },
180 { "frame_incognito", ThemeService::TINT_FRAME_INCOGNITO },
181 { "frame_incognito_inactive",
182 ThemeService::TINT_FRAME_INCOGNITO_INACTIVE },
183 { "background_tab", ThemeService::TINT_BACKGROUND_TAB },
184 { NULL, 0 }
185 };
186
187 // Strings used by themes to identify colors in the JSON.
188 StringToIntTable kColorTable[] = {
189 { "frame", ThemeService::COLOR_FRAME },
190 { "frame_inactive", ThemeService::COLOR_FRAME_INACTIVE },
191 { "frame_incognito", ThemeService::COLOR_FRAME_INCOGNITO },
192 { "frame_incognito_inactive",
193 ThemeService::COLOR_FRAME_INCOGNITO_INACTIVE },
194 { "toolbar", ThemeService::COLOR_TOOLBAR },
195 { "tab_text", ThemeService::COLOR_TAB_TEXT },
196 { "tab_background_text", ThemeService::COLOR_BACKGROUND_TAB_TEXT },
197 { "bookmark_text", ThemeService::COLOR_BOOKMARK_TEXT },
198 { "ntp_background", ThemeService::COLOR_NTP_BACKGROUND },
199 { "ntp_text", ThemeService::COLOR_NTP_TEXT },
200 { "ntp_link", ThemeService::COLOR_NTP_LINK },
201 { "ntp_link_underline", ThemeService::COLOR_NTP_LINK_UNDERLINE },
202 { "ntp_header", ThemeService::COLOR_NTP_HEADER },
203 { "ntp_section", ThemeService::COLOR_NTP_SECTION },
204 { "ntp_section_text", ThemeService::COLOR_NTP_SECTION_TEXT },
205 { "ntp_section_link", ThemeService::COLOR_NTP_SECTION_LINK },
206 { "ntp_section_link_underline",
207 ThemeService::COLOR_NTP_SECTION_LINK_UNDERLINE },
208 { "control_background", ThemeService::COLOR_CONTROL_BACKGROUND },
209 { "button_background", ThemeService::COLOR_BUTTON_BACKGROUND },
210 { NULL, 0 }
211 };
212
213 // Strings used by themes to identify display properties keys in JSON.
214 StringToIntTable kDisplayProperties[] = {
215 { "ntp_background_alignment",
216 ThemeService::NTP_BACKGROUND_ALIGNMENT },
217 { "ntp_background_repeat", ThemeService::NTP_BACKGROUND_TILING },
218 { "ntp_logo_alternate", ThemeService::NTP_LOGO_ALTERNATE },
219 { NULL, 0 }
220 };
221
222 // Strings used by the tiling values in JSON.
223 StringToIntTable kTilingStrings[] = {
224 { "no-repeat", ThemeService::NO_REPEAT },
225 { "repeat-x", ThemeService::REPEAT_X },
226 { "repeat-y", ThemeService::REPEAT_Y },
227 { "repeat", ThemeService::REPEAT },
228 { NULL, 0 }
229 };
230
GetIntForString(const std::string & key,StringToIntTable * table)231 int GetIntForString(const std::string& key, StringToIntTable* table) {
232 for (int i = 0; table[i].key != NULL; ++i) {
233 if (base::strcasecmp(key.c_str(), table[i].key) == 0) {
234 return table[i].id;
235 }
236 }
237
238 return -1;
239 }
240
241 struct IntToIntTable {
242 int key;
243 int value;
244 };
245
246 // Mapping used in GenerateFrameImages() to associate frame images with the
247 // tint ID that should maybe be applied to it.
248 IntToIntTable kFrameTintMap[] = {
249 { PRS_THEME_FRAME, ThemeService::TINT_FRAME },
250 { PRS_THEME_FRAME_INACTIVE, ThemeService::TINT_FRAME_INACTIVE },
251 { PRS_THEME_FRAME_OVERLAY, ThemeService::TINT_FRAME },
252 { PRS_THEME_FRAME_OVERLAY_INACTIVE,
253 ThemeService::TINT_FRAME_INACTIVE },
254 { PRS_THEME_FRAME_INCOGNITO, ThemeService::TINT_FRAME_INCOGNITO },
255 { PRS_THEME_FRAME_INCOGNITO_INACTIVE,
256 ThemeService::TINT_FRAME_INCOGNITO_INACTIVE }
257 };
258
259 // Mapping used in GenerateTabBackgroundImages() to associate what frame image
260 // goes with which tab background.
261 IntToIntTable kTabBackgroundMap[] = {
262 { PRS_THEME_TAB_BACKGROUND, PRS_THEME_FRAME },
263 { PRS_THEME_TAB_BACKGROUND_INCOGNITO, PRS_THEME_FRAME_INCOGNITO }
264 };
265
266 // A list of images that don't need tinting or any other modification and can
267 // be byte-copied directly into the finished DataPack. This should contain the
268 // persistent IDs for all themeable image IDs that aren't in kFrameTintMap or
269 // kTabBackgroundMap.
270 const int kPreloadIDs[] = {
271 PRS_THEME_TOOLBAR,
272 PRS_THEME_NTP_BACKGROUND,
273 PRS_THEME_BUTTON_BACKGROUND,
274 PRS_THEME_NTP_ATTRIBUTION,
275 PRS_THEME_WINDOW_CONTROL_BACKGROUND
276 };
277
278 // Returns a piece of memory with the contents of the file |path|.
ReadFileData(const FilePath & path)279 RefCountedMemory* ReadFileData(const FilePath& path) {
280 if (!path.empty()) {
281 net::FileStream file;
282 int flags = base::PLATFORM_FILE_OPEN | base::PLATFORM_FILE_READ;
283 if (file.Open(path, flags) == net::OK) {
284 int64 avail = file.Available();
285 if (avail > 0 && avail < INT_MAX) {
286 size_t size = static_cast<size_t>(avail);
287 std::vector<unsigned char> raw_data;
288 raw_data.resize(size);
289 char* data = reinterpret_cast<char*>(&(raw_data.front()));
290 if (file.ReadUntilComplete(data, size) == avail)
291 return RefCountedBytes::TakeVector(&raw_data);
292 }
293 }
294 }
295
296 return NULL;
297 }
298
299 // Does error checking for invalid incoming data while trying to read an
300 // floating point value.
ValidDoubleValue(ListValue * tint_list,int index,double * out)301 bool ValidDoubleValue(ListValue* tint_list, int index, double* out) {
302 if (tint_list->GetDouble(index, out))
303 return true;
304
305 int value = 0;
306 if (tint_list->GetInteger(index, &value)) {
307 *out = value;
308 return true;
309 }
310
311 return false;
312 }
313
314 } // namespace
315
~BrowserThemePack()316 BrowserThemePack::~BrowserThemePack() {
317 if (!data_pack_.get()) {
318 delete header_;
319 delete [] tints_;
320 delete [] colors_;
321 delete [] display_properties_;
322 delete [] source_images_;
323 }
324
325 STLDeleteValues(&prepared_images_);
326 STLDeleteValues(&loaded_images_);
327 }
328
329 // static
BuildFromExtension(const Extension * extension)330 BrowserThemePack* BrowserThemePack::BuildFromExtension(
331 const Extension* extension) {
332 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
333 DCHECK(extension);
334 DCHECK(extension->is_theme());
335
336 BrowserThemePack* pack = new BrowserThemePack;
337 pack->BuildHeader(extension);
338 pack->BuildTintsFromJSON(extension->GetThemeTints());
339 pack->BuildColorsFromJSON(extension->GetThemeColors());
340 pack->BuildDisplayPropertiesFromJSON(extension->GetThemeDisplayProperties());
341
342 // Builds the images. (Image building is dependent on tints).
343 FilePathMap file_paths;
344 pack->ParseImageNamesFromJSON(extension->GetThemeImages(),
345 extension->path(),
346 &file_paths);
347 pack->BuildSourceImagesArray(file_paths);
348
349 if (!pack->LoadRawBitmapsTo(file_paths, &pack->prepared_images_))
350 return NULL;
351
352 pack->GenerateFrameImages(&pack->prepared_images_);
353
354 #if !defined(OS_MACOSX)
355 // OSX uses its own special buttons that are PDFs that do odd sorts of vector
356 // graphics tricks. Other platforms use bitmaps and we must pre-tint them.
357 pack->GenerateTintedButtons(
358 pack->GetTintInternal(ThemeService::TINT_BUTTONS),
359 &pack->prepared_images_);
360 #endif
361
362 pack->GenerateTabBackgroundImages(&pack->prepared_images_);
363
364 // The BrowserThemePack is now in a consistent state.
365 return pack;
366 }
367
368 // static
BuildFromDataPack(FilePath path,const std::string & expected_id)369 scoped_refptr<BrowserThemePack> BrowserThemePack::BuildFromDataPack(
370 FilePath path, const std::string& expected_id) {
371 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
372 scoped_refptr<BrowserThemePack> pack(new BrowserThemePack);
373 pack->data_pack_.reset(new ui::DataPack);
374
375 if (!pack->data_pack_->Load(path)) {
376 LOG(ERROR) << "Failed to load theme data pack.";
377 return NULL;
378 }
379
380 base::StringPiece pointer;
381 if (!pack->data_pack_->GetStringPiece(kHeaderID, &pointer))
382 return NULL;
383 pack->header_ = reinterpret_cast<BrowserThemePackHeader*>(const_cast<char*>(
384 pointer.data()));
385
386 if (pack->header_->version != kThemePackVersion) {
387 DLOG(ERROR) << "BuildFromDataPack failure! Version mismatch!";
388 return NULL;
389 }
390 // TODO(erg): Check endianess once DataPack works on the other endian.
391 std::string theme_id(reinterpret_cast<char*>(pack->header_->theme_id),
392 Extension::kIdSize);
393 std::string truncated_id = expected_id.substr(0, Extension::kIdSize);
394 if (theme_id != truncated_id) {
395 DLOG(ERROR) << "Wrong id: " << theme_id << " vs " << expected_id;
396 return NULL;
397 }
398
399 if (!pack->data_pack_->GetStringPiece(kTintsID, &pointer))
400 return NULL;
401 pack->tints_ = reinterpret_cast<TintEntry*>(const_cast<char*>(
402 pointer.data()));
403
404 if (!pack->data_pack_->GetStringPiece(kColorsID, &pointer))
405 return NULL;
406 pack->colors_ =
407 reinterpret_cast<ColorPair*>(const_cast<char*>(pointer.data()));
408
409 if (!pack->data_pack_->GetStringPiece(kDisplayPropertiesID, &pointer))
410 return NULL;
411 pack->display_properties_ = reinterpret_cast<DisplayPropertyPair*>(
412 const_cast<char*>(pointer.data()));
413
414 if (!pack->data_pack_->GetStringPiece(kSourceImagesID, &pointer))
415 return NULL;
416 pack->source_images_ = reinterpret_cast<int*>(
417 const_cast<char*>(pointer.data()));
418
419 return pack;
420 }
421
WriteToDisk(FilePath path) const422 bool BrowserThemePack::WriteToDisk(FilePath path) const {
423 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE));
424 // Add resources for each of the property arrays.
425 RawDataForWriting resources;
426 resources[kHeaderID] = base::StringPiece(
427 reinterpret_cast<const char*>(header_), sizeof(BrowserThemePackHeader));
428 resources[kTintsID] = base::StringPiece(
429 reinterpret_cast<const char*>(tints_), sizeof(TintEntry[kTintArraySize]));
430 resources[kColorsID] = base::StringPiece(
431 reinterpret_cast<const char*>(colors_),
432 sizeof(ColorPair[kColorArraySize]));
433 resources[kDisplayPropertiesID] = base::StringPiece(
434 reinterpret_cast<const char*>(display_properties_),
435 sizeof(DisplayPropertyPair[kDisplayPropertySize]));
436
437 int source_count = 1;
438 int* end = source_images_;
439 for (; *end != -1 ; end++)
440 source_count++;
441 resources[kSourceImagesID] = base::StringPiece(
442 reinterpret_cast<const char*>(source_images_),
443 source_count * sizeof(*source_images_));
444
445 AddRawImagesTo(image_memory_, &resources);
446
447 RawImages reencoded_images;
448 RepackImages(prepared_images_, &reencoded_images);
449 AddRawImagesTo(reencoded_images, &resources);
450
451 return ui::DataPack::WritePack(path, resources);
452 }
453
GetTint(int id,color_utils::HSL * hsl) const454 bool BrowserThemePack::GetTint(int id, color_utils::HSL* hsl) const {
455 if (tints_) {
456 for (int i = 0; i < kTintArraySize; ++i) {
457 if (tints_[i].id == id) {
458 hsl->h = tints_[i].h;
459 hsl->s = tints_[i].s;
460 hsl->l = tints_[i].l;
461 return true;
462 }
463 }
464 }
465
466 return false;
467 }
468
GetColor(int id,SkColor * color) const469 bool BrowserThemePack::GetColor(int id, SkColor* color) const {
470 if (colors_) {
471 for (int i = 0; i < kColorArraySize; ++i) {
472 if (colors_[i].id == id) {
473 *color = colors_[i].color;
474 return true;
475 }
476 }
477 }
478
479 return false;
480 }
481
GetDisplayProperty(int id,int * result) const482 bool BrowserThemePack::GetDisplayProperty(int id, int* result) const {
483 if (display_properties_) {
484 for (int i = 0; i < kDisplayPropertySize; ++i) {
485 if (display_properties_[i].id == id) {
486 *result = display_properties_[i].property;
487 return true;
488 }
489 }
490 }
491
492 return false;
493 }
494
GetBitmapNamed(int idr_id) const495 SkBitmap* BrowserThemePack::GetBitmapNamed(int idr_id) const {
496 int prs_id = GetPersistentIDByIDR(idr_id);
497 if (prs_id == -1)
498 return NULL;
499
500 // Check our cache of prepared images, first.
501 ImageCache::const_iterator image_iter = prepared_images_.find(prs_id);
502 if (image_iter != prepared_images_.end())
503 return image_iter->second;
504
505 // Check if we've already loaded this image.
506 image_iter = loaded_images_.find(prs_id);
507 if (image_iter != loaded_images_.end())
508 return image_iter->second;
509
510 scoped_refptr<RefCountedMemory> memory;
511 if (data_pack_.get()) {
512 memory = data_pack_->GetStaticMemory(prs_id);
513 } else {
514 RawImages::const_iterator it = image_memory_.find(prs_id);
515 if (it != image_memory_.end()) {
516 memory = it->second;
517 }
518 }
519
520 if (memory.get()) {
521 // Decode the PNG.
522 SkBitmap bitmap;
523 if (!gfx::PNGCodec::Decode(memory->front(), memory->size(),
524 &bitmap)) {
525 NOTREACHED() << "Unable to decode theme image resource " << idr_id
526 << " from saved DataPack.";
527 return NULL;
528 }
529
530 SkBitmap* ret = new SkBitmap(bitmap);
531 loaded_images_[prs_id] = ret;
532
533 return ret;
534 }
535
536 return NULL;
537 }
538
GetRawData(int idr_id) const539 RefCountedMemory* BrowserThemePack::GetRawData(int idr_id) const {
540 RefCountedMemory* memory = NULL;
541 int prs_id = GetPersistentIDByIDR(idr_id);
542
543 if (prs_id != -1) {
544 if (data_pack_.get()) {
545 memory = data_pack_->GetStaticMemory(prs_id);
546 } else {
547 RawImages::const_iterator it = image_memory_.find(prs_id);
548 if (it != image_memory_.end()) {
549 memory = it->second;
550 }
551 }
552 }
553
554 return memory;
555 }
556
HasCustomImage(int idr_id) const557 bool BrowserThemePack::HasCustomImage(int idr_id) const {
558 int prs_id = GetPersistentIDByIDR(idr_id);
559 if (prs_id == -1)
560 return false;
561
562 int* img = source_images_;
563 for (; *img != -1; ++img) {
564 if (*img == prs_id)
565 return true;
566 }
567
568 return false;
569 }
570
571 // private:
572
BrowserThemePack()573 BrowserThemePack::BrowserThemePack()
574 : header_(NULL),
575 tints_(NULL),
576 colors_(NULL),
577 display_properties_(NULL),
578 source_images_(NULL) {
579 }
580
BuildHeader(const Extension * extension)581 void BrowserThemePack::BuildHeader(const Extension* extension) {
582 header_ = new BrowserThemePackHeader;
583 header_->version = kThemePackVersion;
584
585 // TODO(erg): Need to make this endian safe on other computers. Prerequisite
586 // is that ui::DataPack removes this same check.
587 #if defined(__BYTE_ORDER)
588 // Linux check
589 COMPILE_ASSERT(__BYTE_ORDER == __LITTLE_ENDIAN,
590 datapack_assumes_little_endian);
591 #elif defined(__BIG_ENDIAN__)
592 // Mac check
593 #error DataPack assumes little endian
594 #endif
595 header_->little_endian = 1;
596
597 const std::string& id = extension->id();
598 memcpy(header_->theme_id, id.c_str(), Extension::kIdSize);
599 }
600
BuildTintsFromJSON(DictionaryValue * tints_value)601 void BrowserThemePack::BuildTintsFromJSON(DictionaryValue* tints_value) {
602 tints_ = new TintEntry[kTintArraySize];
603 for (int i = 0; i < kTintArraySize; ++i) {
604 tints_[i].id = -1;
605 tints_[i].h = -1;
606 tints_[i].s = -1;
607 tints_[i].l = -1;
608 }
609
610 if (!tints_value)
611 return;
612
613 // Parse the incoming data from |tints_value| into an intermediary structure.
614 std::map<int, color_utils::HSL> temp_tints;
615 for (DictionaryValue::key_iterator iter(tints_value->begin_keys());
616 iter != tints_value->end_keys(); ++iter) {
617 ListValue* tint_list;
618 if (tints_value->GetList(*iter, &tint_list) &&
619 (tint_list->GetSize() == 3)) {
620 color_utils::HSL hsl = { -1, -1, -1 };
621
622 if (ValidDoubleValue(tint_list, 0, &hsl.h) &&
623 ValidDoubleValue(tint_list, 1, &hsl.s) &&
624 ValidDoubleValue(tint_list, 2, &hsl.l)) {
625 int id = GetIntForString(*iter, kTintTable);
626 if (id != -1) {
627 temp_tints[id] = hsl;
628 }
629 }
630 }
631 }
632
633 // Copy data from the intermediary data structure to the array.
634 int count = 0;
635 for (std::map<int, color_utils::HSL>::const_iterator it =
636 temp_tints.begin(); it != temp_tints.end() && count < kTintArraySize;
637 ++it, ++count) {
638 tints_[count].id = it->first;
639 tints_[count].h = it->second.h;
640 tints_[count].s = it->second.s;
641 tints_[count].l = it->second.l;
642 }
643 }
644
BuildColorsFromJSON(DictionaryValue * colors_value)645 void BrowserThemePack::BuildColorsFromJSON(DictionaryValue* colors_value) {
646 colors_ = new ColorPair[kColorArraySize];
647 for (int i = 0; i < kColorArraySize; ++i) {
648 colors_[i].id = -1;
649 colors_[i].color = SkColorSetRGB(0, 0, 0);
650 }
651
652 std::map<int, SkColor> temp_colors;
653 if (colors_value)
654 ReadColorsFromJSON(colors_value, &temp_colors);
655 GenerateMissingColors(&temp_colors);
656
657 // Copy data from the intermediary data structure to the array.
658 int count = 0;
659 for (std::map<int, SkColor>::const_iterator it = temp_colors.begin();
660 it != temp_colors.end() && count < kColorArraySize; ++it, ++count) {
661 colors_[count].id = it->first;
662 colors_[count].color = it->second;
663 }
664 }
665
ReadColorsFromJSON(DictionaryValue * colors_value,std::map<int,SkColor> * temp_colors)666 void BrowserThemePack::ReadColorsFromJSON(
667 DictionaryValue* colors_value,
668 std::map<int, SkColor>* temp_colors) {
669 // Parse the incoming data from |colors_value| into an intermediary structure.
670 for (DictionaryValue::key_iterator iter(colors_value->begin_keys());
671 iter != colors_value->end_keys(); ++iter) {
672 ListValue* color_list;
673 if (colors_value->GetList(*iter, &color_list) &&
674 ((color_list->GetSize() == 3) || (color_list->GetSize() == 4))) {
675 SkColor color = SK_ColorWHITE;
676 int r, g, b;
677 if (color_list->GetInteger(0, &r) &&
678 color_list->GetInteger(1, &g) &&
679 color_list->GetInteger(2, &b)) {
680 if (color_list->GetSize() == 4) {
681 double alpha;
682 int alpha_int;
683 if (color_list->GetDouble(3, &alpha)) {
684 color = SkColorSetARGB(static_cast<int>(alpha * 255), r, g, b);
685 } else if (color_list->GetInteger(3, &alpha_int) &&
686 (alpha_int == 0 || alpha_int == 1)) {
687 color = SkColorSetARGB(alpha_int ? 255 : 0, r, g, b);
688 } else {
689 // Invalid entry for part 4.
690 continue;
691 }
692 } else {
693 color = SkColorSetRGB(r, g, b);
694 }
695
696 int id = GetIntForString(*iter, kColorTable);
697 if (id != -1) {
698 (*temp_colors)[id] = color;
699 }
700 }
701 }
702 }
703 }
704
GenerateMissingColors(std::map<int,SkColor> * colors)705 void BrowserThemePack::GenerateMissingColors(
706 std::map<int, SkColor>* colors) {
707 // Generate link colors, if missing. (See GetColor()).
708 if (!colors->count(ThemeService::COLOR_NTP_HEADER) &&
709 colors->count(ThemeService::COLOR_NTP_SECTION)) {
710 (*colors)[ThemeService::COLOR_NTP_HEADER] =
711 (*colors)[ThemeService::COLOR_NTP_SECTION];
712 }
713
714 if (!colors->count(ThemeService::COLOR_NTP_SECTION_LINK_UNDERLINE) &&
715 colors->count(ThemeService::COLOR_NTP_SECTION_LINK)) {
716 SkColor color_section_link =
717 (*colors)[ThemeService::COLOR_NTP_SECTION_LINK];
718 (*colors)[ThemeService::COLOR_NTP_SECTION_LINK_UNDERLINE] =
719 SkColorSetA(color_section_link, SkColorGetA(color_section_link) / 3);
720 }
721
722 if (!colors->count(ThemeService::COLOR_NTP_LINK_UNDERLINE) &&
723 colors->count(ThemeService::COLOR_NTP_LINK)) {
724 SkColor color_link = (*colors)[ThemeService::COLOR_NTP_LINK];
725 (*colors)[ThemeService::COLOR_NTP_LINK_UNDERLINE] =
726 SkColorSetA(color_link, SkColorGetA(color_link) / 3);
727 }
728
729 // Generate frame colors, if missing. (See GenerateFrameColors()).
730 SkColor frame;
731 std::map<int, SkColor>::const_iterator it =
732 colors->find(ThemeService::COLOR_FRAME);
733 if (it != colors->end()) {
734 frame = it->second;
735 } else {
736 frame = ThemeService::GetDefaultColor(
737 ThemeService::COLOR_FRAME);
738 }
739
740 if (!colors->count(ThemeService::COLOR_FRAME)) {
741 (*colors)[ThemeService::COLOR_FRAME] =
742 HSLShift(frame, GetTintInternal(ThemeService::TINT_FRAME));
743 }
744 if (!colors->count(ThemeService::COLOR_FRAME_INACTIVE)) {
745 (*colors)[ThemeService::COLOR_FRAME_INACTIVE] =
746 HSLShift(frame, GetTintInternal(
747 ThemeService::TINT_FRAME_INACTIVE));
748 }
749 if (!colors->count(ThemeService::COLOR_FRAME_INCOGNITO)) {
750 (*colors)[ThemeService::COLOR_FRAME_INCOGNITO] =
751 HSLShift(frame, GetTintInternal(
752 ThemeService::TINT_FRAME_INCOGNITO));
753 }
754 if (!colors->count(ThemeService::COLOR_FRAME_INCOGNITO_INACTIVE)) {
755 (*colors)[ThemeService::COLOR_FRAME_INCOGNITO_INACTIVE] =
756 HSLShift(frame, GetTintInternal(
757 ThemeService::TINT_FRAME_INCOGNITO_INACTIVE));
758 }
759 }
760
BuildDisplayPropertiesFromJSON(DictionaryValue * display_properties_value)761 void BrowserThemePack::BuildDisplayPropertiesFromJSON(
762 DictionaryValue* display_properties_value) {
763 display_properties_ = new DisplayPropertyPair[kDisplayPropertySize];
764 for (int i = 0; i < kDisplayPropertySize; ++i) {
765 display_properties_[i].id = -1;
766 display_properties_[i].property = 0;
767 }
768
769 if (!display_properties_value)
770 return;
771
772 std::map<int, int> temp_properties;
773 for (DictionaryValue::key_iterator iter(
774 display_properties_value->begin_keys());
775 iter != display_properties_value->end_keys(); ++iter) {
776 int property_id = GetIntForString(*iter, kDisplayProperties);
777 switch (property_id) {
778 case ThemeService::NTP_BACKGROUND_ALIGNMENT: {
779 std::string val;
780 if (display_properties_value->GetString(*iter, &val)) {
781 temp_properties[ThemeService::NTP_BACKGROUND_ALIGNMENT] =
782 ThemeService::StringToAlignment(val);
783 }
784 break;
785 }
786 case ThemeService::NTP_BACKGROUND_TILING: {
787 std::string val;
788 if (display_properties_value->GetString(*iter, &val)) {
789 temp_properties[ThemeService::NTP_BACKGROUND_TILING] =
790 GetIntForString(val, kTilingStrings);
791 }
792 break;
793 }
794 case ThemeService::NTP_LOGO_ALTERNATE: {
795 int val = 0;
796 if (display_properties_value->GetInteger(*iter, &val))
797 temp_properties[ThemeService::NTP_LOGO_ALTERNATE] = val;
798 break;
799 }
800 }
801 }
802
803 // Copy data from the intermediary data structure to the array.
804 int count = 0;
805 for (std::map<int, int>::const_iterator it = temp_properties.begin();
806 it != temp_properties.end() && count < kDisplayPropertySize;
807 ++it, ++count) {
808 display_properties_[count].id = it->first;
809 display_properties_[count].property = it->second;
810 }
811 }
812
ParseImageNamesFromJSON(DictionaryValue * images_value,const FilePath & images_path,FilePathMap * file_paths) const813 void BrowserThemePack::ParseImageNamesFromJSON(
814 DictionaryValue* images_value,
815 const FilePath& images_path,
816 FilePathMap* file_paths) const {
817 if (!images_value)
818 return;
819
820 for (DictionaryValue::key_iterator iter(images_value->begin_keys());
821 iter != images_value->end_keys(); ++iter) {
822 std::string val;
823 if (images_value->GetString(*iter, &val)) {
824 int id = GetPersistentIDByName(*iter);
825 if (id != -1)
826 (*file_paths)[id] = images_path.AppendASCII(val);
827 }
828 }
829 }
830
BuildSourceImagesArray(const FilePathMap & file_paths)831 void BrowserThemePack::BuildSourceImagesArray(const FilePathMap& file_paths) {
832 std::vector<int> ids;
833 for (FilePathMap::const_iterator it = file_paths.begin();
834 it != file_paths.end(); ++it) {
835 ids.push_back(it->first);
836 }
837
838 source_images_ = new int[ids.size() + 1];
839 std::copy(ids.begin(), ids.end(), source_images_);
840 source_images_[ids.size()] = -1;
841 }
842
LoadRawBitmapsTo(const FilePathMap & file_paths,ImageCache * raw_bitmaps)843 bool BrowserThemePack::LoadRawBitmapsTo(
844 const FilePathMap& file_paths,
845 ImageCache* raw_bitmaps) {
846 // Themes should be loaded on the file thread, not the UI thread.
847 // http://crbug.com/61838
848 base::ThreadRestrictions::ScopedAllowIO allow_io;
849
850 for (FilePathMap::const_iterator it = file_paths.begin();
851 it != file_paths.end(); ++it) {
852 scoped_refptr<RefCountedMemory> raw_data(ReadFileData(it->second));
853 if (!raw_data.get()) {
854 LOG(ERROR) << "Could not load theme image";
855 return false;
856 }
857
858 int id = it->first;
859
860 // Some images need to go directly into |image_memory_|. No modification is
861 // necessary or desirable.
862 bool is_copyable = false;
863 for (size_t i = 0; i < arraysize(kPreloadIDs); ++i) {
864 if (kPreloadIDs[i] == id) {
865 is_copyable = true;
866 break;
867 }
868 }
869
870 if (is_copyable) {
871 image_memory_[id] = raw_data;
872 } else if (raw_data.get() && raw_data->size()) {
873 // Decode the PNG.
874 SkBitmap bitmap;
875 if (gfx::PNGCodec::Decode(raw_data->front(), raw_data->size(),
876 &bitmap)) {
877 (*raw_bitmaps)[it->first] = new SkBitmap(bitmap);
878 } else {
879 NOTREACHED() << "Unable to decode theme image resource " << it->first;
880 }
881 }
882 }
883
884 return true;
885 }
886
GenerateFrameImages(ImageCache * bitmaps) const887 void BrowserThemePack::GenerateFrameImages(ImageCache* bitmaps) const {
888 ResourceBundle& rb = ResourceBundle::GetSharedInstance();
889
890 // Create all the output bitmaps in a separate cache and move them back into
891 // the input bitmaps because there can be name collisions.
892 ImageCache temp_output;
893
894 for (size_t i = 0; i < arraysize(kFrameTintMap); ++i) {
895 int prs_id = kFrameTintMap[i].key;
896 scoped_ptr<SkBitmap> frame;
897 // If there's no frame image provided for the specified id, then load
898 // the default provided frame. If that's not provided, skip this whole
899 // thing and just use the default images.
900 int prs_base_id;
901
902 if (prs_id == PRS_THEME_FRAME_INCOGNITO_INACTIVE) {
903 prs_base_id = bitmaps->count(PRS_THEME_FRAME_INCOGNITO) ?
904 PRS_THEME_FRAME_INCOGNITO : PRS_THEME_FRAME;
905 } else if (prs_id == PRS_THEME_FRAME_OVERLAY_INACTIVE) {
906 prs_base_id = PRS_THEME_FRAME_OVERLAY;
907 } else if (prs_id == PRS_THEME_FRAME_INACTIVE) {
908 prs_base_id = PRS_THEME_FRAME;
909 } else if (prs_id == PRS_THEME_FRAME_INCOGNITO &&
910 !bitmaps->count(PRS_THEME_FRAME_INCOGNITO)) {
911 prs_base_id = PRS_THEME_FRAME;
912 } else {
913 prs_base_id = prs_id;
914 }
915
916 if (bitmaps->count(prs_id)) {
917 frame.reset(new SkBitmap(*(*bitmaps)[prs_id]));
918 } else if (prs_base_id != prs_id && bitmaps->count(prs_base_id)) {
919 frame.reset(new SkBitmap(*(*bitmaps)[prs_base_id]));
920 } else if (prs_base_id == PRS_THEME_FRAME_OVERLAY &&
921 bitmaps->count(PRS_THEME_FRAME)) {
922 // If there is no theme overlay, don't tint the default frame,
923 // because it will overwrite the custom frame image when we cache and
924 // reload from disk.
925 frame.reset(NULL);
926 } else {
927 // If the theme doesn't specify an image, then apply the tint to
928 // the default frame.
929 frame.reset(new SkBitmap(*rb.GetBitmapNamed(IDR_THEME_FRAME)));
930 }
931
932 if (frame.get()) {
933 temp_output[prs_id] = new SkBitmap(
934 SkBitmapOperations::CreateHSLShiftedBitmap(
935 *frame, GetTintInternal(kFrameTintMap[i].value)));
936 }
937 }
938
939 MergeImageCaches(temp_output, bitmaps);
940 }
941
GenerateTintedButtons(const color_utils::HSL & button_tint,ImageCache * processed_bitmaps) const942 void BrowserThemePack::GenerateTintedButtons(
943 const color_utils::HSL& button_tint,
944 ImageCache* processed_bitmaps) const {
945 if (button_tint.h != -1 || button_tint.s != -1 || button_tint.l != -1) {
946 ResourceBundle& rb = ResourceBundle::GetSharedInstance();
947 const std::set<int>& idr_ids =
948 ThemeService::GetTintableToolbarButtons();
949 for (std::set<int>::const_iterator it = idr_ids.begin();
950 it != idr_ids.end(); ++it) {
951 int prs_id = GetPersistentIDByIDR(*it);
952 DCHECK(prs_id > 0);
953
954 // Fetch the image by IDR...
955 scoped_ptr<SkBitmap> button(new SkBitmap(*rb.GetBitmapNamed(*it)));
956
957 // but save a version with the persistent ID.
958 (*processed_bitmaps)[prs_id] = new SkBitmap(
959 SkBitmapOperations::CreateHSLShiftedBitmap(*button, button_tint));
960 }
961 }
962 }
963
GenerateTabBackgroundImages(ImageCache * bitmaps) const964 void BrowserThemePack::GenerateTabBackgroundImages(ImageCache* bitmaps) const {
965 ImageCache temp_output;
966 for (size_t i = 0; i < arraysize(kTabBackgroundMap); ++i) {
967 int prs_id = kTabBackgroundMap[i].key;
968 int prs_base_id = kTabBackgroundMap[i].value;
969
970 // We only need to generate the background tab images if we were provided
971 // with a PRS_THEME_FRAME.
972 ImageCache::const_iterator it = bitmaps->find(prs_base_id);
973 if (it != bitmaps->end()) {
974 SkBitmap bg_tint = SkBitmapOperations::CreateHSLShiftedBitmap(
975 *(it->second), GetTintInternal(
976 ThemeService::TINT_BACKGROUND_TAB));
977 int vertical_offset = bitmaps->count(prs_id)
978 ? kRestoredTabVerticalOffset : 0;
979 SkBitmap* bg_tab = new SkBitmap(SkBitmapOperations::CreateTiledBitmap(
980 bg_tint, 0, vertical_offset, bg_tint.width(), bg_tint.height()));
981
982 // If they've provided a custom image, overlay it.
983 ImageCache::const_iterator overlay_it = bitmaps->find(prs_id);
984 if (overlay_it != bitmaps->end()) {
985 SkBitmap* overlay = overlay_it->second;
986 SkCanvas canvas(*bg_tab);
987 for (int x = 0; x < bg_tab->width(); x += overlay->width())
988 canvas.drawBitmap(*overlay, static_cast<SkScalar>(x), 0, NULL);
989 }
990
991 temp_output[prs_id] = bg_tab;
992 }
993 }
994
995 MergeImageCaches(temp_output, bitmaps);
996 }
997
RepackImages(const ImageCache & images,RawImages * reencoded_images) const998 void BrowserThemePack::RepackImages(const ImageCache& images,
999 RawImages* reencoded_images) const {
1000 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE));
1001 for (ImageCache::const_iterator it = images.begin();
1002 it != images.end(); ++it) {
1003 std::vector<unsigned char> image_data;
1004 if (!gfx::PNGCodec::EncodeBGRASkBitmap(*(it->second), false, &image_data)) {
1005 NOTREACHED() << "Image file for resource " << it->first
1006 << " could not be encoded.";
1007 } else {
1008 (*reencoded_images)[it->first] = RefCountedBytes::TakeVector(&image_data);
1009 }
1010 }
1011 }
1012
MergeImageCaches(const ImageCache & source,ImageCache * destination) const1013 void BrowserThemePack::MergeImageCaches(
1014 const ImageCache& source, ImageCache* destination) const {
1015
1016 for (ImageCache::const_iterator it = source.begin(); it != source.end();
1017 ++it) {
1018 ImageCache::const_iterator bitmap_it = destination->find(it->first);
1019 if (bitmap_it != destination->end())
1020 delete bitmap_it->second;
1021
1022 (*destination)[it->first] = it->second;
1023 }
1024 }
1025
AddRawImagesTo(const RawImages & images,RawDataForWriting * out) const1026 void BrowserThemePack::AddRawImagesTo(const RawImages& images,
1027 RawDataForWriting* out) const {
1028 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE));
1029 for (RawImages::const_iterator it = images.begin(); it != images.end();
1030 ++it) {
1031 (*out)[it->first] = base::StringPiece(
1032 reinterpret_cast<const char*>(it->second->front()), it->second->size());
1033 }
1034 }
1035
GetTintInternal(int id) const1036 color_utils::HSL BrowserThemePack::GetTintInternal(int id) const {
1037 if (tints_) {
1038 for (int i = 0; i < kTintArraySize; ++i) {
1039 if (tints_[i].id == id) {
1040 color_utils::HSL hsl;
1041 hsl.h = tints_[i].h;
1042 hsl.s = tints_[i].s;
1043 hsl.l = tints_[i].l;
1044 return hsl;
1045 }
1046 }
1047 }
1048
1049 return ThemeService::GetDefaultTint(id);
1050 }
1051