1 /*
2 * Copyright 2015 Google Inc.
3 *
4 * Use of this source code is governed by a BSD-style license that can be
5 * found in the LICENSE file.
6 */
7
8 #include "SkBitmapCache.h"
9 #include "SkResourceCache.h"
10 #include "SkYUVPlanesCache.h"
11
12 #define CHECK_LOCAL(localCache, localName, globalName, ...) \
13 ((localCache) ? localCache->localName(__VA_ARGS__) : SkResourceCache::globalName(__VA_ARGS__))
14
15 namespace {
16 static unsigned gYUVPlanesKeyNamespaceLabel;
17
18 struct YUVValue {
19 SkYUVPlanesCache::Info fInfo;
20 SkCachedData* fData;
21 };
22
23 struct YUVPlanesKey : public SkResourceCache::Key {
YUVPlanesKey__anon709b30380111::YUVPlanesKey24 YUVPlanesKey(uint32_t genID)
25 : fGenID(genID)
26 {
27 this->init(&gYUVPlanesKeyNamespaceLabel, SkMakeResourceCacheSharedIDForBitmap(genID),
28 sizeof(genID));
29 }
30
31 uint32_t fGenID;
32 };
33
34 struct YUVPlanesRec : public SkResourceCache::Rec {
YUVPlanesRec__anon709b30380111::YUVPlanesRec35 YUVPlanesRec(YUVPlanesKey key, SkCachedData* data, SkYUVPlanesCache::Info* info)
36 : fKey(key)
37 {
38 fValue.fData = data;
39 fValue.fInfo = *info;
40 fValue.fData->attachToCacheAndRef();
41 }
~YUVPlanesRec__anon709b30380111::YUVPlanesRec42 ~YUVPlanesRec() override {
43 fValue.fData->detachFromCacheAndUnref();
44 }
45
46 YUVPlanesKey fKey;
47 YUVValue fValue;
48
getKey__anon709b30380111::YUVPlanesRec49 const Key& getKey() const override { return fKey; }
bytesUsed__anon709b30380111::YUVPlanesRec50 size_t bytesUsed() const override { return sizeof(*this) + fValue.fData->size(); }
getCategory__anon709b30380111::YUVPlanesRec51 const char* getCategory() const override { return "yuv-planes"; }
diagnostic_only_getDiscardable__anon709b30380111::YUVPlanesRec52 SkDiscardableMemory* diagnostic_only_getDiscardable() const override {
53 return fValue.fData->diagnostic_only_getDiscardable();
54 }
55
Visitor__anon709b30380111::YUVPlanesRec56 static bool Visitor(const SkResourceCache::Rec& baseRec, void* contextData) {
57 const YUVPlanesRec& rec = static_cast<const YUVPlanesRec&>(baseRec);
58 YUVValue* result = static_cast<YUVValue*>(contextData);
59
60 SkCachedData* tmpData = rec.fValue.fData;
61 tmpData->ref();
62 if (nullptr == tmpData->data()) {
63 tmpData->unref();
64 return false;
65 }
66 result->fData = tmpData;
67 result->fInfo = rec.fValue.fInfo;
68 return true;
69 }
70 };
71 } // namespace
72
FindAndRef(uint32_t genID,Info * info,SkResourceCache * localCache)73 SkCachedData* SkYUVPlanesCache::FindAndRef(uint32_t genID, Info* info,
74 SkResourceCache* localCache) {
75 YUVValue result;
76 YUVPlanesKey key(genID);
77 if (!CHECK_LOCAL(localCache, find, Find, key, YUVPlanesRec::Visitor, &result)) {
78 return nullptr;
79 }
80
81 *info = result.fInfo;
82 return result.fData;
83 }
84
Add(uint32_t genID,SkCachedData * data,Info * info,SkResourceCache * localCache)85 void SkYUVPlanesCache::Add(uint32_t genID, SkCachedData* data, Info* info,
86 SkResourceCache* localCache) {
87 YUVPlanesKey key(genID);
88 return CHECK_LOCAL(localCache, add, Add, new YUVPlanesRec(key, data, info));
89 }
90