Fix broken build (add lru_cache.h)

This commit is contained in:
Jacob Dufault 2017-12-03 19:24:42 -08:00
parent b2736f8822
commit 9a64fb0747
2 changed files with 122 additions and 1 deletions

1
.gitignore vendored
View File

@ -5,7 +5,6 @@
*.vcxproj
*.vcxproj.filters
*.vcxproj.user
*CACHE*
.DS_Store
.lock-waf*
.vs

122
src/lru_cache.h Normal file
View File

@ -0,0 +1,122 @@
#pragma once
#include <limits>
#include <memory>
// Cache that evicts old entries which have not been used recently. Implemented
// using array/linear search so this works well for small array sizes.
template <typename TKey, typename TValue>
struct LruCache {
explicit LruCache(int max_entries);
// Fetches an entry for |key|. If it does not exist, |allocator| will be invoked to create one.
template <typename TAllocator>
std::shared_ptr<TValue> Get(const TKey& key, TAllocator allocator);
// Fetches the entry for |filename| and updates it's usage so it is less
// likely to be evicted.
std::shared_ptr<TValue> TryGet(const TKey& key);
// TryGetEntry, except the entry is removed from the cache.
std::shared_ptr<TValue> TryTake(const TKey& key);
// Inserts an entry. Evicts the oldest unused entry if there is no space.
void Insert(const TKey& key, const std::shared_ptr<TValue>& value);
private:
// There is a global score counter, when we access an element we increase
// its score to the current global value, so it has the highest overall
// score. This means that the oldest/least recently accessed value has the
// lowest score.
//
// There is a bit of special logic to handle score overlow.
struct Entry {
uint32_t score = 0;
TKey key;
std::shared_ptr<TValue> value;
};
void IncrementScore();
std::vector<Entry> entries_;
int max_entries_ = 1;
uint32_t next_score_ = 0;
};
template <typename TKey, typename TValue>
LruCache<TKey, TValue>::LruCache(int max_entries) : max_entries_(max_entries) {
assert(max_entries > 0);
}
template <typename TKey, typename TValue>
template <typename TAllocator>
std::shared_ptr<TValue> LruCache<TKey, TValue>::Get(const TKey& key, TAllocator allocator) {
std::shared_ptr<TValue> result = TryGet(key);
if (!result)
Insert(key, result = allocator());
return result;
}
template <typename TKey, typename TValue>
std::shared_ptr<TValue> LruCache<TKey, TValue>::TryGet(const TKey& key) {
// Assign new score.
for (Entry& entry : entries_) {
if (entry.key == key) {
IncrementScore();
entry.score = next_score_;
return entry.value;
}
}
return nullptr;
}
template <typename TKey, typename TValue>
std::shared_ptr<TValue> LruCache<TKey, TValue>::TryTake(const TKey& key) {
for (size_t i = 0; i < entries_.size(); ++i) {
if (entries_[i].key == key) {
std::shared_ptr<TValue> copy = entries_[i].value;
entries_.erase(entries_.begin() + i);
return copy;
}
}
return nullptr;
}
template <typename TKey, typename TValue>
void LruCache<TKey, TValue>::Insert(const TKey& key, const std::shared_ptr<TValue>& value) {
if (entries_.size() >= max_entries_) {
// Find entry with the lowest score.
size_t lowest_idx = 0;
uint32_t lowest_score = std::numeric_limits<uint32_t>::max();
for (size_t i = 0; i < entries_.size(); ++i) {
if (entries_[i].score < lowest_score) {
lowest_idx = i;
lowest_score = entries_[i].score;
}
}
// Remove it.
entries_.erase(entries_.begin() + lowest_idx);
}
IncrementScore();
Entry entry;
entry.score = next_score_;
entry.key = key;
entry.value = value;
entries_.push_back(entry);
}
template <typename TKey, typename TValue>
void LruCache<TKey, TValue>::IncrementScore() {
next_score_ += 1;
// Overflow.
if (next_score_ == 0) {
std::sort(entries_.begin(), entries_.end(), [](const Entry& a, const Entry& b) {
return a.score > b.score;
});
for (Entry& entry : entries_)
entry.score = next_score_++;
}
}