1
0
Fork 0
mirror of https://github.com/transmission/transmission synced 2024-12-23 08:13:27 +00:00
transmission/qt/TorrentModel.cc
Mike Gelfand d7930984ef Adjust uncrustify config, reformat all but Mac client
There're places where manual intervention is still required as uncrustify
is not ideal (unfortunately), but at least one may rely on it to do the
right thing most of the time (e.g. when sending in a patch).

The style itself is quite different from what we had before but making it
uniform across all the codebase is the key. I also hope that it'll make the
code more readable (YMMV) and less sensitive to further changes.
2017-04-20 10:01:22 +03:00

331 lines
7 KiB
C++

/*
* This file Copyright (C) 2009-2015 Mnemosyne LLC
*
* It may be used under the GNU GPL versions 2 or 3
* or any future license endorsed by Mnemosyne LLC.
*
*/
#include <iostream>
#include <libtransmission/transmission.h>
#include <libtransmission/variant.h>
#include "Speed.h"
#include "Torrent.h"
#include "TorrentDelegate.h"
#include "TorrentModel.h"
namespace
{
struct TorrentIdLessThan
{
bool operator ()(Torrent* left, Torrent* right) const
{
return left->id() < right->id();
}
bool operator ()(int leftId, Torrent* right) const
{
return leftId < right->id();
}
bool operator ()(Torrent* left, int rightId) const
{
return left->id() < rightId;
}
};
} // namespace
void TorrentModel::clear()
{
beginResetModel();
qDeleteAll(myTorrents);
myTorrents.clear();
endResetModel();
}
int TorrentModel::rowCount(const QModelIndex& parent) const
{
Q_UNUSED(parent);
return myTorrents.size();
}
QVariant TorrentModel::data(const QModelIndex& index, int role) const
{
QVariant var;
const Torrent* t = myTorrents.value(index.row(), 0);
if (t != 0)
{
switch (role)
{
case Qt::DisplayRole:
var.setValue(t->name());
break;
case Qt::DecorationRole:
var.setValue(t->getMimeTypeIcon());
break;
case TorrentRole:
var = qVariantFromValue(t);
break;
default:
// std::cerr << "Unhandled role: " << role << std::endl;
break;
}
}
return var;
}
/***
****
***/
void TorrentModel::addTorrent(Torrent* t)
{
const torrents_t::iterator torrentIt = qLowerBound(myTorrents.begin(), myTorrents.end(), t, TorrentIdLessThan());
const int row = torrentIt == myTorrents.end() ? myTorrents.size() : torrentIt - myTorrents.begin();
beginInsertRows(QModelIndex(), row, row);
myTorrents.insert(torrentIt, t);
endInsertRows();
}
void TorrentModel::addTorrents(torrents_t&& torrents, QSet<int>& addIds)
{
if (myTorrents.isEmpty())
{
qSort(torrents.begin(), torrents.end(), TorrentIdLessThan());
beginInsertRows(QModelIndex(), 0, torrents.size() - 1);
myTorrents.swap(torrents);
endInsertRows();
addIds += getIds();
}
else
{
for (Torrent* const tor : torrents)
{
addTorrent(tor);
addIds.insert(tor->id());
}
}
}
TorrentModel::TorrentModel(const Prefs& prefs) :
myPrefs(prefs)
{
}
TorrentModel::~TorrentModel()
{
clear();
}
/***
****
***/
Torrent* TorrentModel::getTorrentFromId(int id)
{
const torrents_t::const_iterator torrentIt = qBinaryFind(myTorrents.begin(), myTorrents.end(), id, TorrentIdLessThan());
return torrentIt == myTorrents.end() ? nullptr : *torrentIt;
}
const Torrent* TorrentModel::getTorrentFromId(int id) const
{
const torrents_t::const_iterator torrentIt = qBinaryFind(myTorrents.begin(), myTorrents.end(), id, TorrentIdLessThan());
return torrentIt == myTorrents.end() ? nullptr : *torrentIt;
}
/***
****
***/
void TorrentModel::onTorrentChanged(int torrentId)
{
const torrents_t::iterator torrentIt = qBinaryFind(myTorrents.begin(), myTorrents.end(), torrentId, TorrentIdLessThan());
if (torrentIt == myTorrents.end())
{
return;
}
const int row = torrentIt - myTorrents.begin();
const QModelIndex qmi(index(row, 0));
emit dataChanged(qmi, qmi);
}
void TorrentModel::removeTorrents(tr_variant* torrents)
{
int i = 0;
tr_variant* child;
while ((child = tr_variantListChild(torrents, i++)))
{
int64_t intVal;
if (tr_variantGetInt(child, &intVal))
{
removeTorrent(intVal);
}
}
}
void TorrentModel::updateTorrents(tr_variant* torrents, bool isCompleteList)
{
torrents_t newTorrents;
QSet<int> oldIds;
QSet<int> addIds;
QSet<int> newIds;
if (isCompleteList)
{
oldIds = getIds();
}
if (tr_variantIsList(torrents))
{
size_t i(0);
tr_variant* child;
while ((child = tr_variantListChild(torrents, i++)))
{
int64_t id;
if (tr_variantDictFindInt(child, TR_KEY_id, &id))
{
if (isCompleteList)
{
newIds.insert(id);
}
Torrent* tor = getTorrentFromId(id);
if (tor == 0)
{
tor = new Torrent(myPrefs, id);
tor->update(child);
if (!tor->hasMetadata())
{
tor->setMagnet(true);
}
newTorrents.append(tor);
connect(tor, SIGNAL(torrentChanged(int)), this, SLOT(onTorrentChanged(int)));
}
else
{
tor->update(child);
if (tor->isMagnet() && tor->hasMetadata())
{
addIds.insert(tor->id());
tor->setMagnet(false);
}
}
}
}
}
if (!newTorrents.isEmpty())
{
addTorrents(std::move(newTorrents), addIds);
}
if (!addIds.isEmpty())
{
emit torrentsAdded(addIds);
}
if (isCompleteList)
{
QSet<int> removedIds(oldIds);
removedIds -= newIds;
for (const int id : removedIds)
{
removeTorrent(id);
}
}
}
void TorrentModel::removeTorrent(int id)
{
const torrents_t::iterator torrentIt = qBinaryFind(myTorrents.begin(), myTorrents.end(), id, TorrentIdLessThan());
if (torrentIt == myTorrents.end())
{
return;
}
Torrent* const tor = *torrentIt;
const int row = torrentIt - myTorrents.begin();
beginRemoveRows(QModelIndex(), row, row);
myTorrents.remove(row);
endRemoveRows();
delete tor;
}
void TorrentModel::getTransferSpeed(Speed& uploadSpeed, size_t& uploadPeerCount, Speed& downloadSpeed,
size_t& downloadPeerCount)
{
Speed upSpeed, downSpeed;
size_t upCount = 0, downCount = 0;
for (const Torrent* const tor : myTorrents)
{
upSpeed += tor->uploadSpeed();
upCount += tor->peersWeAreUploadingTo();
downSpeed += tor->downloadSpeed();
downCount += tor->webseedsWeAreDownloadingFrom();
downCount += tor->peersWeAreDownloadingFrom();
}
uploadSpeed = upSpeed;
uploadPeerCount = upCount;
downloadSpeed = downSpeed;
downloadPeerCount = downCount;
}
QSet<int> TorrentModel::getIds() const
{
QSet<int> ids;
ids.reserve(myTorrents.size());
for (const Torrent* const tor : myTorrents)
{
ids.insert(tor->id());
}
return ids;
}
bool TorrentModel::hasTorrent(const QString& hashString) const
{
for (const Torrent* const tor : myTorrents)
{
if (tor->hashString() == hashString)
{
return true;
}
}
return false;
}