1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
|
// Copyright (c) 2009 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef CHROME_BROWSER_THUMBNAIL_STORE_H_
#define CHROME_BROWSER_THUMBNAIL_STORE_H_
#include <map>
#include <vector>
#include "base/file_path.h"
#include "base/message_loop.h"
#include "base/ref_counted.h"
#include "testing/gtest/include/gtest/gtest_prod.h"
class GURL;
class Pickle;
class SkBitmap;
struct ThumbnailScore;
namespace base {
class Time;
}
// This storage interface provides storage for the thumbnails used
// by the new_tab_ui.
class ThumbnailStore : public base::RefCountedThreadSafe<ThumbnailStore> {
public:
ThumbnailStore();
~ThumbnailStore();
// Must be called after creation but before other methods are called.
// file_path is a directory where a new database should be created
// or the location of an existing databse.
void Init(const FilePath& file_path);
// Stores the given thumbnail and score with the associated url in the cache.
// If write_to_disk is true, the thumbnail data is written to disk on the
// file_thread.
bool SetPageThumbnail(const GURL& url,
SkBitmap& thumbnail,
const ThumbnailScore& score,
bool write_to_disk);
// Retrieves the thumbnail and score for the given url.
// Returns false if there is not data for the given url or some other
// error occurred.
bool GetPageThumbnail(const GURL& url,
SkBitmap* thumbnail,
ThumbnailScore* score);
private:
FRIEND_TEST(ThumbnailStoreTest, RetrieveFromCache);
FRIEND_TEST(ThumbnailStoreTest, RetrieveFromDisk);
// Data structure used to store thumbnail data in memory.
typedef std::map<GURL, std::pair<SkBitmap, ThumbnailScore> > Cache;
// The location of the thumbnail store.
FilePath file_path_;
// Read all thumbnail data from the specified FilePath into a Cache object.
// Done on the file_thread and returns to OnDiskDataAvailable on the thread
// owning the specified MessageLoop.
void GetAllThumbnailsFromDisk(FilePath filepath, MessageLoop* cb_loop);
// Read the thumbnail data from the given file and stores it in the
// out parameters GURL, SkBitmap, and ThumbnailScore.
bool GetPageThumbnailFromDisk(const FilePath& file,
GURL* url,
SkBitmap* thumbnail,
ThumbnailScore* score) const;
// Once thumbnail data from the disk is available from the file_thread,
// this function is invoked on the main thread. It takes ownership of the
// Cache* passed in and retains this Cache* for the lifetime of the object.
void OnDiskDataAvailable(ThumbnailStore::Cache* cache);
// Write thumbnail data to disk for a given url.
bool WriteThumbnailToDisk(const GURL& url) const;
// Pack the given ThumbnailScore into the given Pickle.
void PackScore(const ThumbnailScore& score, Pickle* packed) const;
// Unpack a ThumbnailScore from a given Pickle and associated iterator.
// Returns false is a ThumbnailScore could not be unpacked.
bool UnpackScore(ThumbnailScore* score, const Pickle& packed,
void*& iter) const;
DISALLOW_COPY_AND_ASSIGN(ThumbnailStore);
// The Cache maintained by the object.
scoped_ptr<ThumbnailStore::Cache> cache_;
bool cache_initialized_;
};
#endif // CHROME_BROWSER_THUMBNAIL_STORE_H_
|
