blob: 60ef27df645c0fa2ba00a9fcb2a8898989a766a1 [file] [log] [blame]
// Copyright 2015 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 MEDIA_BLINK_LRU_H_
#define MEDIA_BLINK_LRU_H_
#include <stddef.h>
#include <list>
#include <unordered_map>
#include "base/macros.h"
namespace media {
// Simple LRU (least recently used) class.
// Keeps track of a set of data and lets you get the least recently used
// (oldest) element at any time. All operations are O(1). Elements are expected
// to be hashable and unique.
// Example:
// LRU<int> lru;
// lru.Insert(1);
// lru.Insert(2);
// lru.Insert(3);
// lru.Use(1);
// cout << lru.Pop(); // this will print "2"
template <typename T>
class LRU {
public:
LRU() {}
// Adds |x| to LRU.
// |x| must not already be in the LRU.
// Faster than Use(), and will DCHECK that |x| is not in the LRU.
void Insert(const T& x) {
DCHECK(!Contains(x));
lru_.push_front(x);
pos_[x] = lru_.begin();
}
// Removes |x| from LRU.
// |x| must be in the LRU.
void Remove(const T& x) {
DCHECK(Contains(x));
lru_.erase(pos_[x]);
pos_.erase(x);
}
// Moves |x| to front of LRU. (most recently used)
// If |x| is not in LRU, it is added.
// Please call Insert() if you know that |x| is not in the LRU.
void Use(const T& x) {
if (Contains(x))
Remove(x);
Insert(x);
}
bool Empty() const { return lru_.empty(); }
// Returns the Least Recently Used T and removes it.
T Pop() {
DCHECK(!Empty());
T ret = lru_.back();
lru_.pop_back();
pos_.erase(ret);
return ret;
}
// Returns the Least Recently Used T _without_ removing it.
T Peek() const {
DCHECK(!Empty());
return lru_.back();
}
bool Contains(const T& x) const { return pos_.find(x) != pos_.end(); }
size_t Size() const { return pos_.size(); }
private:
friend class LRUTest;
// Linear list of elements, most recently used first.
std::list<T> lru_;
// Maps element values to positions in the list so that we
// can quickly remove elements.
std::unordered_map<T, typename std::list<T>::iterator> pos_;
DISALLOW_COPY_AND_ASSIGN(LRU);
};
} // namespace media
#endif // MEDIA_BLINK_LRU_H_