85 lines
1.8 KiB
C++
85 lines
1.8 KiB
C++
/*
|
|
This file is part of Telegram Desktop,
|
|
the official desktop application for the Telegram messaging service.
|
|
|
|
For license and copyright information please follow this link:
|
|
https://github.com/telegramdesktop/tdesktop/blob/master/LEGAL
|
|
*/
|
|
#include "media/streaming/media_streaming_loader.h"
|
|
|
|
namespace Media {
|
|
namespace Streaming {
|
|
|
|
bool operator<(
|
|
const PriorityQueue::Entry &a,
|
|
const PriorityQueue::Entry &b) {
|
|
if (a.priority > b.priority) {
|
|
return true;
|
|
} else if (a.priority < b.priority) {
|
|
return false;
|
|
} else {
|
|
return a.value < b.value;
|
|
}
|
|
}
|
|
|
|
bool PriorityQueue::add(int value) {
|
|
const auto i = ranges::find(_data, value, &Entry::value);
|
|
if (i == end(_data)) {
|
|
_data.insert({ value, _priority });
|
|
return true;
|
|
} else if (i->priority != _priority) {
|
|
_data.erase(i);
|
|
_data.insert({ value, _priority });
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
bool PriorityQueue::remove(int value) {
|
|
const auto i = ranges::find(_data, value, &Entry::value);
|
|
if (i == end(_data)) {
|
|
return false;
|
|
}
|
|
_data.erase(i);
|
|
return true;
|
|
}
|
|
|
|
std::optional<int> PriorityQueue::front() const {
|
|
return _data.empty()
|
|
? std::nullopt
|
|
: std::make_optional(_data.front().value);
|
|
}
|
|
|
|
std::optional<int> PriorityQueue::take() {
|
|
if (_data.empty()) {
|
|
return std::nullopt;
|
|
}
|
|
const auto result = _data.front().value;
|
|
_data.erase(_data.begin());
|
|
return result;
|
|
}
|
|
|
|
base::flat_set<int> PriorityQueue::takeInRange(int from, int till) {
|
|
auto result = base::flat_set<int>();
|
|
for (auto i = _data.begin(); i != _data.end();) {
|
|
if (i->value >= from && i->value < till) {
|
|
result.emplace(i->value);
|
|
i = _data.erase(i);
|
|
} else {
|
|
++i;
|
|
}
|
|
}
|
|
return result;
|
|
}
|
|
|
|
void PriorityQueue::clear() {
|
|
_data.clear();
|
|
}
|
|
|
|
void PriorityQueue::increasePriority() {
|
|
++_priority;
|
|
}
|
|
|
|
} // namespace Streaming
|
|
} // namespace Media
|