int QueueBundleUtils::compareBundles(const BundlePtr& a, const BundlePtr& b, int aPropertyName) noexcept {
switch (aPropertyName) {
case PROP_NAME: {
COMPARE_TYPE(a, b);
return Util::stricmp(a->getName(), b->getName());
}
case PROP_TYPE: {
COMPARE_TYPE(a, b);
if (!a->isFileBundle() && !b->isFileBundle()) {
// Directory bundles
RLock l(QueueManager::getInstance()->getCS());
auto dirsA = QueueManager::getInstance()->bundleQueue.getDirectoryCount(a);
auto dirsB = QueueManager::getInstance()->bundleQueue.getDirectoryCount(b);
if (dirsA != dirsB) {
return compare(dirsA, dirsB);
}
auto filesA = a->getQueueItems().size() + a->getFinishedFiles().size();
auto filesB = b->getQueueItems().size() + b->getFinishedFiles().size();
return compare(filesA, filesB);
}
return Util::stricmp(Util::getFileExt(a->getTarget()), Util::getFileExt(b->getTarget()));
}
case PROP_PRIORITY: {
COMPARE_FINISHED(a, b);
if (a->isFinished() != b->isFinished()) {
return a->isFinished() ? 1 : -1;
}
return compare(static_cast<int>(a->getPriority()), static_cast<int>(b->getPriority()));
}
case PROP_STATUS: {
if (a->getStatus() != b->getStatus()) {
return compare(a->getStatus(), b->getStatus());
}
return compare(
a->getPercentage(a->getDownloadedBytes()),
b->getPercentage(b->getDownloadedBytes())
);
}
case PROP_SOURCES: {
COMPARE_FINISHED(a, b);
auto countsA = QueueManager::getInstance()->getSourceCount(a);
auto countsB = QueueManager::getInstance()->getSourceCount(b);
return QueueItemBase::SourceCount::compare(countsA, countsB);
}
default:
dcassert(0);
}
return 0;
}
void UserQueue::setBundlePriority(BundlePtr& aBundle, QueueItemBase::Priority p) noexcept {
dcassert(!aBundle->isFinished());
HintedUserList sources;
aBundle->getSourceUsers(sources);
for(const auto& u: sources)
removeBundle(aBundle, u);
aBundle->setPriority(p);
for(const auto& u: sources)
addBundle(aBundle, u);
}
void UserQueue::setBundlePriority(BundlePtr& aBundle, Bundle::Priority p) {
dcassert(!aBundle->isFinished());
HintedUserList sources;
aBundle->getSources(sources);
for(auto& u: sources)
removeBundle(aBundle, u);
aBundle->setPriority(p);
for(auto& u: sources)
addBundle(aBundle, u);
}
int QueueUtils::compareBundles(const BundlePtr& a, const BundlePtr& b, int aPropertyName) noexcept {
switch (aPropertyName) {
case QueueApi::PROP_NAME: {
if (a->isFileBundle() && !b->isFileBundle()) return 1;
if (!a->isFileBundle() && b->isFileBundle()) return -1;
return Util::stricmp(a->getName(), b->getName());
}
case QueueApi::PROP_TYPE: {
if (a->isFileBundle() != b->isFileBundle()) {
// Directories go first
return a->isFileBundle() ? 1 : -1;
}
if (!a->isFileBundle() && !b->isFileBundle()) {
// Directory bundles
RLock l(QueueManager::getInstance()->getCS());
auto dirsA = a->getDirectories().size();
auto dirsB = a->getDirectories().size();
if (dirsA != dirsB) {
return compare(dirsA, dirsB);
}
auto filesA = a->getQueueItems().size() + a->getFinishedFiles().size();
auto filesB = b->getQueueItems().size() + b->getFinishedFiles().size();
return compare(filesA, filesB);
}
return Util::stricmp(Util::getFileExt(a->getTarget()), Util::getFileExt(b->getTarget()));
}
case QueueApi::PROP_PRIORITY: {
if (a->isFinished() != b->isFinished()) {
return a->isFinished() ? 1 : -1;
}
return compare(static_cast<int>(a->getPriority()), static_cast<int>(b->getPriority()));
}
case QueueApi::PROP_STATUS: {
if (a->getStatus() != b->getStatus()) {
return compare(a->getStatus(), b->getStatus());
}
return compare(a->getDownloadedBytes(), b->getDownloadedBytes());
}
case QueueApi::PROP_SOURCES: {
if (a->isFinished() != b->isFinished()) {
return a->isFinished() ? 1 : -1;
}
int onlineA = 0, totalA = 0, onlineB = 0, totalB = 0;
std::string str;
getBundleSourceInfo(a, onlineA, totalA, str);
getBundleSourceInfo(b, onlineB, totalB, str);
if (onlineA != onlineB) {
return compare(onlineA, onlineB);
}
return compare(totalA, totalB);
}
default:
dcassert(0);
}
return 0;
}
