12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788 |
- /*
- * Copyright 2018 Google Inc.
- *
- * Use of this source code is governed by a BSD-style license that can be
- * found in the LICENSE file.
- */
- #ifndef MemoryCache_DEFINED
- #define MemoryCache_DEFINED
- #include "include/core/SkData.h"
- #include "include/gpu/GrContextOptions.h"
- #include "include/private/SkChecksum.h"
- #include <unordered_map>
- namespace sk_gpu_test {
- /**
- * This class can be used to maintain an in memory record of all programs cached by GrContext.
- * It can be shared by multiple GrContexts so long as those GrContexts are created with the same
- * options and will have the same GrCaps (e.g. same backend, same GL context creation parameters,
- * ...).
- */
- class MemoryCache : public GrContextOptions::PersistentCache {
- public:
- MemoryCache() = default;
- MemoryCache(const MemoryCache&) = delete;
- MemoryCache& operator=(const MemoryCache&) = delete;
- void reset() {
- fCacheMissCnt = 0;
- fMap.clear();
- }
- sk_sp<SkData> load(const SkData& key) override;
- void store(const SkData& key, const SkData& data) override;
- int numCacheMisses() const { return fCacheMissCnt; }
- void resetNumCacheMisses() { fCacheMissCnt = 0; }
- void writeShadersToDisk(const char* path, GrBackendApi backend);
- template <typename Fn>
- void foreach(Fn&& fn) {
- for (auto it = fMap.begin(); it != fMap.end(); ++it) {
- fn(it->first.fKey, it->second.fData, it->second.fHitCount);
- }
- }
- private:
- struct Key {
- Key() = default;
- Key(const SkData& key) : fKey(SkData::MakeWithCopy(key.data(), key.size())) {}
- Key(const Key& that) = default;
- Key& operator=(const Key&) = default;
- bool operator==(const Key& that) const {
- return that.fKey->size() == fKey->size() &&
- !memcmp(fKey->data(), that.fKey->data(), that.fKey->size());
- }
- sk_sp<const SkData> fKey;
- };
- struct Value {
- Value() = default;
- Value(const SkData& data)
- : fData(SkData::MakeWithCopy(data.data(), data.size()))
- , fHitCount(1) {}
- Value(const Value& that) = default;
- Value& operator=(const Value&) = default;
- sk_sp<SkData> fData;
- int fHitCount;
- };
- struct Hash {
- using argument_type = Key;
- using result_type = uint32_t;
- uint32_t operator()(const Key& key) const {
- return key.fKey ? SkOpts::hash_fn(key.fKey->data(), key.fKey->size(), 0) : 0;
- }
- };
- int fCacheMissCnt = 0;
- std::unordered_map<Key, Value, Hash> fMap;
- };
- } // namespace sk_gpu_test
- #endif
|