shibakaneki
13 years ago
parent
2f9367c39a
commit
af5b99cfb3
@ -0,0 +1,366 @@ |
|||||||
|
/*
|
||||||
|
* This program is free software: you can redistribute it and/or modify |
||||||
|
* it under the terms of the GNU General Public License as published by |
||||||
|
* the Free Software Foundation, either version 3 of the License, or |
||||||
|
* (at your option) any later version. |
||||||
|
* |
||||||
|
* This program is distributed in the hope that it will be useful, |
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||||
|
* GNU General Public License for more details. |
||||||
|
* |
||||||
|
* You should have received a copy of the GNU General Public License |
||||||
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*/ |
||||||
|
#include "UBDownloadManager.h" |
||||||
|
#include "core/UBApplication.h" |
||||||
|
#include "gui/UBMainWindow.h" |
||||||
|
|
||||||
|
/** The unique instance of the download manager */ |
||||||
|
static UBDownloadManager* pInstance = NULL; |
||||||
|
|
||||||
|
/**
|
||||||
|
* \brief Constructor |
||||||
|
* @param parent as the parent widget |
||||||
|
* @param name as the object name |
||||||
|
*/ |
||||||
|
UBDownloadManager::UBDownloadManager(QObject *parent, const char *name):QObject(parent) |
||||||
|
{ |
||||||
|
setObjectName(name); |
||||||
|
init(); |
||||||
|
|
||||||
|
connect(this, SIGNAL(fileAddedToDownload()), this, SLOT(onUpdateDownloadLists())); |
||||||
|
} |
||||||
|
|
||||||
|
/**
|
||||||
|
* \brief Destructor |
||||||
|
*/ |
||||||
|
UBDownloadManager::~UBDownloadManager() |
||||||
|
{ |
||||||
|
|
||||||
|
} |
||||||
|
|
||||||
|
/**
|
||||||
|
* \brief Get the download manager |
||||||
|
* @return a pointer on the download manager |
||||||
|
*/ |
||||||
|
UBDownloadManager* UBDownloadManager::downloadManager() |
||||||
|
{ |
||||||
|
if(NULL == pInstance) |
||||||
|
{ |
||||||
|
pInstance = new UBDownloadManager(); |
||||||
|
} |
||||||
|
return pInstance; |
||||||
|
} |
||||||
|
|
||||||
|
/**
|
||||||
|
* \brief Add a file to the download list |
||||||
|
* @param desc as the given file description |
||||||
|
*/ |
||||||
|
void UBDownloadManager::addFileToDownload(sDownloadFileDesc desc) |
||||||
|
{ |
||||||
|
// Set the ID for this download
|
||||||
|
desc.id = mLastID; |
||||||
|
mLastID++; |
||||||
|
|
||||||
|
// Add the file to the pending download list
|
||||||
|
mPendingDL.append(desc); |
||||||
|
|
||||||
|
// If the download is modal, show the download dialog
|
||||||
|
if(desc.modal) |
||||||
|
{ |
||||||
|
// Update the download order (priority to modal files)
|
||||||
|
updateDownloadOrder(); |
||||||
|
UBApplication::mainWindow->showDownloadWidget(); |
||||||
|
} |
||||||
|
|
||||||
|
emit fileAddedToDownload(); |
||||||
|
} |
||||||
|
|
||||||
|
/**
|
||||||
|
* \brief Initialize the download manager |
||||||
|
*/ |
||||||
|
void UBDownloadManager::init() |
||||||
|
{ |
||||||
|
mCrntDL.clear(); |
||||||
|
mPendingDL.clear(); |
||||||
|
mLastID = 1; |
||||||
|
for(int i=0; i<SIMULTANEOUS_DOWNLOAD; i++) |
||||||
|
{ |
||||||
|
mDLAvailability.append(-1); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
/**
|
||||||
|
* \brief Update the download order. The modal downloads will be put in priority. |
||||||
|
*/ |
||||||
|
void UBDownloadManager::updateDownloadOrder() |
||||||
|
{ |
||||||
|
QVector<sDownloadFileDesc> modalFiles; |
||||||
|
QVector<sDownloadFileDesc> nonModalfiles; |
||||||
|
|
||||||
|
for(int i=0; i<mPendingDL.size(); i++) |
||||||
|
{ |
||||||
|
sDownloadFileDesc crnt = mPendingDL.at(i); |
||||||
|
if(crnt.modal) |
||||||
|
{ |
||||||
|
modalFiles.append(crnt); |
||||||
|
} |
||||||
|
else |
||||||
|
{ |
||||||
|
nonModalfiles.append(crnt); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
mPendingDL = modalFiles + nonModalfiles; |
||||||
|
} |
||||||
|
|
||||||
|
/**
|
||||||
|
* \brief Update the download list. If a current download is finished, we take a |
||||||
|
* file from the pending download list and add it to the download list. |
||||||
|
*/ |
||||||
|
void UBDownloadManager::onUpdateDownloadLists() |
||||||
|
{ |
||||||
|
for(int i=0; i<SIMULTANEOUS_DOWNLOAD; i++) |
||||||
|
{ |
||||||
|
if(mPendingDL.empty()) |
||||||
|
{ |
||||||
|
// If we fall here that means that there is no pending download
|
||||||
|
break; |
||||||
|
} |
||||||
|
if(-1 == mDLAvailability.at(i)) |
||||||
|
{ |
||||||
|
// Pending downloads exist and a download 'slot' is available
|
||||||
|
// Let's move the first pending download to the current download
|
||||||
|
// list and fill the slot
|
||||||
|
sDownloadFileDesc desc = mPendingDL.at(0); |
||||||
|
mCrntDL.append(desc); |
||||||
|
mPendingDL.remove(0); |
||||||
|
mDLAvailability.remove(i); |
||||||
|
mDLAvailability.insert(i, desc.id); |
||||||
|
|
||||||
|
// Start the download of this file
|
||||||
|
startFileDownload(desc); |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
/**
|
||||||
|
* \brief Get the list of the current downloads |
||||||
|
* @return a QVector of current downloads |
||||||
|
*/ |
||||||
|
QVector<sDownloadFileDesc> UBDownloadManager::currentDownloads() |
||||||
|
{ |
||||||
|
return mCrntDL; |
||||||
|
} |
||||||
|
|
||||||
|
/**
|
||||||
|
* \brief Get the list of the pending downloads |
||||||
|
* @return a QVector of pending downloads |
||||||
|
*/ |
||||||
|
QVector<sDownloadFileDesc> UBDownloadManager::pendingDownloads() |
||||||
|
{ |
||||||
|
return mPendingDL; |
||||||
|
} |
||||||
|
|
||||||
|
/**
|
||||||
|
* \brief Update the file transfer information |
||||||
|
* @param desc as the current downloaded file description |
||||||
|
*/ |
||||||
|
void UBDownloadManager::onDownloadProgress(int id, qint64 received, qint64 total) |
||||||
|
{ |
||||||
|
updateFileCurrentSize(id, received, total); |
||||||
|
} |
||||||
|
|
||||||
|
/**
|
||||||
|
* \brief Called when the download of the given file is finished |
||||||
|
* @param desc as the current downloaded file description |
||||||
|
*/ |
||||||
|
void UBDownloadManager::onDownloadFinished(int id, bool pSuccess, QUrl sourceUrl, QString pContentTypeHeader, QByteArray pData, QPointF pPos, QSize pSize, bool isBackground) |
||||||
|
{ |
||||||
|
for(int i=0; i<mCrntDL.size(); i++) |
||||||
|
{ |
||||||
|
sDownloadFileDesc desc = mCrntDL.at(i); |
||||||
|
if(id == desc.id) |
||||||
|
{ |
||||||
|
if(desc.modal) |
||||||
|
{ |
||||||
|
// The downloaded file is modal so we must put it on the board
|
||||||
|
emit addDownloadedFileToBoard(pSuccess, sourceUrl, pContentTypeHeader, pData, pPos, pSize, isBackground); |
||||||
|
} |
||||||
|
break; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// Then do this
|
||||||
|
updateFileCurrentSize(id); |
||||||
|
} |
||||||
|
|
||||||
|
/**
|
||||||
|
* \brief Update the description of the given current downloaded file |
||||||
|
* @param desc as the current downloaded file description |
||||||
|
*/ |
||||||
|
void UBDownloadManager::updateFileCurrentSize(int id, qint64 received, qint64 total) |
||||||
|
{ |
||||||
|
for(int i=0; i<mCrntDL.size();i++) |
||||||
|
{ |
||||||
|
if(mCrntDL.at(i).id == id) |
||||||
|
{ |
||||||
|
sDownloadFileDesc desc = mCrntDL.at(i); |
||||||
|
if(received >= 0 && total >= 0) |
||||||
|
{ |
||||||
|
// -------------------------------------
|
||||||
|
// [=============== x% ==== ]
|
||||||
|
// -------------------------------------
|
||||||
|
desc.currentSize = received; |
||||||
|
desc.totalSize = total; |
||||||
|
|
||||||
|
emit downloadUpdated(id, received, total); |
||||||
|
} |
||||||
|
else |
||||||
|
{ |
||||||
|
// -------------------------------------
|
||||||
|
// [=============== 100% ==============]
|
||||||
|
// -------------------------------------
|
||||||
|
// received and total are negative. That means that the download is finished
|
||||||
|
desc.currentSize = mCrntDL.at(i).totalSize; |
||||||
|
|
||||||
|
// Remove the finished file from the current download list
|
||||||
|
mCrntDL.remove(i); |
||||||
|
|
||||||
|
// Free the download slot used by the finished file
|
||||||
|
for(int j=0; j<mDLAvailability.size();j++) |
||||||
|
{ |
||||||
|
if(id == mDLAvailability.at(j)) |
||||||
|
{ |
||||||
|
mDLAvailability.remove(j); |
||||||
|
mDLAvailability.insert(j, -1); |
||||||
|
break; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// Here we check if some modal downloads remain
|
||||||
|
checkIfModalRemains(); |
||||||
|
|
||||||
|
// Then we update the list of downloads
|
||||||
|
onUpdateDownloadLists(); |
||||||
|
|
||||||
|
emit downloadFinished(id); |
||||||
|
break; |
||||||
|
} |
||||||
|
|
||||||
|
mCrntDL.remove(i); |
||||||
|
mCrntDL.insert(i,desc); |
||||||
|
break; |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
/**
|
||||||
|
* \brief Start the download of a file |
||||||
|
* @param desc as the given file description |
||||||
|
*/ |
||||||
|
void UBDownloadManager::startFileDownload(sDownloadFileDesc desc) |
||||||
|
{ |
||||||
|
UBDownloadHttpFile* http = new UBDownloadHttpFile(desc.id, this); |
||||||
|
connect(http, SIGNAL(downloadProgress(int, qint64,qint64)), this, SLOT(onDownloadProgress(int,qint64,qint64))); |
||||||
|
connect(http, SIGNAL(downloadFinished(int, bool, QUrl, QString, QByteArray, QPointF, QSize, bool)), this, SLOT(onDownloadFinished(int, bool, QUrl, QString, QByteArray, QPointF, QSize, bool))); |
||||||
|
http->get(QUrl(desc.url)); |
||||||
|
} |
||||||
|
|
||||||
|
/**
|
||||||
|
* \brief Verify if modal downloads remains and notify everyone if it is not the case. |
||||||
|
*/ |
||||||
|
void UBDownloadManager::checkIfModalRemains() |
||||||
|
{ |
||||||
|
bool bModal = false; |
||||||
|
for(int i=0; i<mCrntDL.size();i++) |
||||||
|
{ |
||||||
|
if(mCrntDL.at(i).modal) |
||||||
|
{ |
||||||
|
bModal = true; |
||||||
|
break; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
if(!bModal) |
||||||
|
{ |
||||||
|
for(int j=0; j<mPendingDL.size(); j++) |
||||||
|
{ |
||||||
|
if(mPendingDL.at(j).modal) |
||||||
|
{ |
||||||
|
bModal = true; |
||||||
|
break; |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
if(bModal) |
||||||
|
{ |
||||||
|
// Close the modal window
|
||||||
|
UBApplication::mainWindow->hideDownloadWidget(); |
||||||
|
|
||||||
|
// Notify that no modal downloads are pending
|
||||||
|
emit downloadModalFinished(); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
/**
|
||||||
|
* \brief Cancel all downloads |
||||||
|
*/ |
||||||
|
void UBDownloadManager::cancelDownloads() |
||||||
|
{ |
||||||
|
// Stop the current downloads
|
||||||
|
|
||||||
|
|
||||||
|
// Notify everyone that the downloads have been canceled.
|
||||||
|
emit cancelAllDownloads(); |
||||||
|
} |
||||||
|
|
||||||
|
// ------------------------------------------------------------------------------
|
||||||
|
/**
|
||||||
|
* \brief Constructor |
||||||
|
* @param parent as the parent widget |
||||||
|
* @param name as the object name |
||||||
|
*/ |
||||||
|
UBDownloadHttpFile::UBDownloadHttpFile(int fileId, QObject *parent):UBHttpGet(parent) |
||||||
|
{ |
||||||
|
mId = fileId; |
||||||
|
|
||||||
|
connect(this, SIGNAL(downloadFinished(bool,QUrl,QString,QByteArray,QPointF,QSize,bool)), this, SLOT(onDownloadFinished(bool,QUrl,QString,QByteArray,QPointF,QSize,bool))); |
||||||
|
connect(this, SIGNAL(downloadProgress(qint64,qint64)), this, SLOT(onDownloadProgress(qint64,qint64))); |
||||||
|
} |
||||||
|
|
||||||
|
/**
|
||||||
|
* \brief Destructor |
||||||
|
*/ |
||||||
|
UBDownloadHttpFile::~UBDownloadHttpFile() |
||||||
|
{ |
||||||
|
|
||||||
|
} |
||||||
|
|
||||||
|
/**
|
||||||
|
* \brief Handles the download progress notification |
||||||
|
* @param bytesReceived as the number of received bytes |
||||||
|
* @param bytesTotal as the total number of bytes |
||||||
|
*/ |
||||||
|
void UBDownloadHttpFile::onDownloadProgress(qint64 bytesReceived, qint64 bytesTotal) |
||||||
|
{ |
||||||
|
emit downloadProgress(mId, bytesReceived, bytesTotal); |
||||||
|
} |
||||||
|
|
||||||
|
/**
|
||||||
|
* \brief Handles the download finished notification |
||||||
|
* @param pSuccess as the success indicator |
||||||
|
* @param sourceUrl as the source URL |
||||||
|
* @param pContentTypeHeader as the response content type header |
||||||
|
* @param pData as the packet data |
||||||
|
* @param pPos as the item position in the board |
||||||
|
* @param psize as the item size (GUI) |
||||||
|
* @param isBackground as the background mdoe indicator |
||||||
|
*/ |
||||||
|
void UBDownloadHttpFile::onDownloadFinished(bool pSuccess, QUrl sourceUrl, QString pContentTypeHeader, QByteArray pData, QPointF pPos, QSize pSize, bool isBackground) |
||||||
|
{ |
||||||
|
emit downloadFinished(mId, pSuccess, sourceUrl, pContentTypeHeader, pData, pPos, pSize, isBackground); |
||||||
|
} |
||||||
|
|
||||||
@ -0,0 +1,106 @@ |
|||||||
|
/*
|
||||||
|
* This program is free software: you can redistribute it and/or modify |
||||||
|
* it under the terms of the GNU General Public License as published by |
||||||
|
* the Free Software Foundation, either version 3 of the License, or |
||||||
|
* (at your option) any later version. |
||||||
|
* |
||||||
|
* This program is distributed in the hope that it will be useful, |
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||||
|
* GNU General Public License for more details. |
||||||
|
* |
||||||
|
* You should have received a copy of the GNU General Public License |
||||||
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*/ |
||||||
|
#ifndef UBDOWNLOADMANAGER_H |
||||||
|
#define UBDOWNLOADMANAGER_H |
||||||
|
|
||||||
|
#include <QObject> |
||||||
|
#include <QString> |
||||||
|
#include <QVector> |
||||||
|
#include <QMutex> |
||||||
|
|
||||||
|
#include "UBDownloadThread.h" |
||||||
|
|
||||||
|
#include "network/UBHttpGet.h" |
||||||
|
|
||||||
|
#define SIMULTANEOUS_DOWNLOAD 2 // Maximum 5 because of QNetworkAccessManager limitation!!!
|
||||||
|
|
||||||
|
typedef struct |
||||||
|
{ |
||||||
|
QString name; |
||||||
|
int id; |
||||||
|
int totalSize; |
||||||
|
int currentSize; |
||||||
|
QString url; |
||||||
|
bool modal; |
||||||
|
QPointF pos; // For board drop only
|
||||||
|
QSize size; // For board drop only
|
||||||
|
bool isBackground; // For board drop only
|
||||||
|
}sDownloadFileDesc; |
||||||
|
|
||||||
|
class UBDownloadHttpFile : public UBHttpGet |
||||||
|
{ |
||||||
|
Q_OBJECT |
||||||
|
public: |
||||||
|
UBDownloadHttpFile(int fileId, QObject* parent=0); |
||||||
|
~UBDownloadHttpFile(); |
||||||
|
|
||||||
|
signals: |
||||||
|
void downloadProgress(int id, qint64 current,qint64 total); |
||||||
|
void downloadFinished(int id, bool pSuccess, QUrl sourceUrl, QString pContentTypeHeader, QByteArray pData, QPointF pPos, QSize pSize, bool isBackground); |
||||||
|
|
||||||
|
private slots: |
||||||
|
void onDownloadFinished(bool pSuccess, QUrl sourceUrl, QString pContentTypeHeader, QByteArray pData, QPointF pPos, QSize pSize, bool isBackground); |
||||||
|
void onDownloadProgress(qint64 bytesReceived, qint64 bytesTotal); |
||||||
|
|
||||||
|
private: |
||||||
|
int mId; |
||||||
|
}; |
||||||
|
|
||||||
|
class UBDownloadManager : public QObject |
||||||
|
{ |
||||||
|
Q_OBJECT |
||||||
|
public: |
||||||
|
UBDownloadManager(QObject* parent=0, const char* name="UBDownloadManager"); |
||||||
|
~UBDownloadManager(); |
||||||
|
|
||||||
|
static UBDownloadManager* downloadManager(); |
||||||
|
void addFileToDownload(sDownloadFileDesc desc); |
||||||
|
QVector<sDownloadFileDesc> currentDownloads(); |
||||||
|
QVector<sDownloadFileDesc> pendingDownloads(); |
||||||
|
void cancelDownloads(); |
||||||
|
|
||||||
|
signals: |
||||||
|
void fileAddedToDownload(); |
||||||
|
void downloadUpdated(int id, qint64 crnt, qint64 total); |
||||||
|
void downloadFinished(int id); |
||||||
|
void downloadModalFinished(); |
||||||
|
void addDownloadedFileToBoard(bool pSuccess, QUrl sourceUrl, QString pContentTypeHeader, QByteArray pData, QPointF pPos, QSize pSize, bool isBackground); |
||||||
|
void cancelAllDownloads(); |
||||||
|
|
||||||
|
private slots: |
||||||
|
void onUpdateDownloadLists(); |
||||||
|
void onDownloadProgress(int id, qint64 received, qint64 total); |
||||||
|
void onDownloadFinished(int id, bool pSuccess, QUrl sourceUrl, QString pContentTypeHeader, QByteArray pData, QPointF pPos, QSize pSize, bool isBackground); |
||||||
|
|
||||||
|
private: |
||||||
|
void init(); |
||||||
|
void updateDownloadOrder(); |
||||||
|
void updateFileCurrentSize(int id, qint64 received=-1, qint64 total=-1); |
||||||
|
void startFileDownload(sDownloadFileDesc desc); |
||||||
|
void checkIfModalRemains(); |
||||||
|
|
||||||
|
/** The current downloads */ |
||||||
|
QVector<sDownloadFileDesc> mCrntDL; |
||||||
|
/** The pending downloads */ |
||||||
|
QVector<sDownloadFileDesc> mPendingDL; |
||||||
|
/** Pending download mutex */ |
||||||
|
QMutex mMutex; |
||||||
|
/** The last file ID */ |
||||||
|
int mLastID; |
||||||
|
/** The current download availability (-1 = free, otherwise the file ID is recorded)*/ |
||||||
|
QVector<int> mDLAvailability; |
||||||
|
}; |
||||||
|
|
||||||
|
#endif // UBDOWNLOADMANAGER_H
|
||||||
@ -0,0 +1,115 @@ |
|||||||
|
/*
|
||||||
|
* This program is free software: you can redistribute it and/or modify |
||||||
|
* it under the terms of the GNU General Public License as published by |
||||||
|
* the Free Software Foundation, either version 3 of the License, or |
||||||
|
* (at your option) any later version. |
||||||
|
* |
||||||
|
* This program is distributed in the hope that it will be useful, |
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||||
|
* GNU General Public License for more details. |
||||||
|
* |
||||||
|
* You should have received a copy of the GNU General Public License |
||||||
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*/ |
||||||
|
#include <QDebug> |
||||||
|
#include <QNetworkProxy> |
||||||
|
#include <QNetworkDiskCache> |
||||||
|
|
||||||
|
#include "core/UBSettings.h" |
||||||
|
|
||||||
|
#include "UBDownloadThread.h" |
||||||
|
|
||||||
|
/**
|
||||||
|
* \brief Constructor |
||||||
|
* @param parent as the parent object |
||||||
|
* @param name as the object name |
||||||
|
*/ |
||||||
|
UBDownloadThread::UBDownloadThread(QObject *parent, const char *name):QThread(parent) |
||||||
|
, mbRun(false) |
||||||
|
,mpReply(NULL) |
||||||
|
{ |
||||||
|
setObjectName(name); |
||||||
|
} |
||||||
|
|
||||||
|
/**
|
||||||
|
* \brief Destructor |
||||||
|
*/ |
||||||
|
UBDownloadThread::~UBDownloadThread() |
||||||
|
{ |
||||||
|
if(NULL != mpReply) |
||||||
|
{ |
||||||
|
delete mpReply; |
||||||
|
mpReply = NULL; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
/**
|
||||||
|
* \brief Run the thread |
||||||
|
*/ |
||||||
|
void UBDownloadThread::run() |
||||||
|
{ |
||||||
|
qDebug() << mUrl; |
||||||
|
// We start the download
|
||||||
|
QNetworkAccessManager* pNam = new QNetworkAccessManager(); |
||||||
|
|
||||||
|
mpReply = pNam->get(QNetworkRequest(QUrl(mUrl))); |
||||||
|
qDebug() << " -- Http GET reply ---------------------- "; |
||||||
|
qDebug() << mpReply->readAll(); |
||||||
|
qDebug() << " ---------------------------------------- "; |
||||||
|
|
||||||
|
connect(mpReply, SIGNAL(downloadProgress(qint64,qint64)), this, SLOT(onDownloadProgress(qint64,qint64))); |
||||||
|
connect(mpReply, SIGNAL(finished()), this, SLOT(onDownloadFinished())); |
||||||
|
|
||||||
|
while(mbRun) |
||||||
|
{ |
||||||
|
// Wait here until the end of the download
|
||||||
|
sleep(100); |
||||||
|
} |
||||||
|
|
||||||
|
disconnect(mpReply, SIGNAL(downloadProgress(qint64,qint64)), this, SLOT(onDownloadProgress(qint64,qint64))); |
||||||
|
disconnect(mpReply, SIGNAL(finished()), this, SLOT(onDownloadFinished())); |
||||||
|
if(NULL != mpReply) |
||||||
|
{ |
||||||
|
delete mpReply; |
||||||
|
mpReply = NULL; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
/**
|
||||||
|
* \brief Stop the current download |
||||||
|
*/ |
||||||
|
void UBDownloadThread::stopDownload() |
||||||
|
{ |
||||||
|
mbRun = false; |
||||||
|
} |
||||||
|
|
||||||
|
/**
|
||||||
|
* \brief Start the download |
||||||
|
*/ |
||||||
|
void UBDownloadThread::startDownload(int id, QString url) |
||||||
|
{ |
||||||
|
mID = id; |
||||||
|
mUrl = url; |
||||||
|
mbRun = true; |
||||||
|
start(); |
||||||
|
} |
||||||
|
|
||||||
|
/**
|
||||||
|
* \brief Notify the download progression |
||||||
|
* @param received as the number of bytes received |
||||||
|
* @param total as the total number of bytes of the file |
||||||
|
*/ |
||||||
|
void UBDownloadThread::onDownloadProgress(qint64 received, qint64 total) |
||||||
|
{ |
||||||
|
qDebug() << received << " on " << total; |
||||||
|
emit downloadProgress(mID, received, total); |
||||||
|
} |
||||||
|
|
||||||
|
/**
|
||||||
|
* \brief Notify the end of the download |
||||||
|
*/ |
||||||
|
void UBDownloadThread::onDownloadFinished() |
||||||
|
{ |
||||||
|
emit downloadFinised(mID); |
||||||
|
} |
||||||
@ -0,0 +1,56 @@ |
|||||||
|
/*
|
||||||
|
* This program is free software: you can redistribute it and/or modify |
||||||
|
* it under the terms of the GNU General Public License as published by |
||||||
|
* the Free Software Foundation, either version 3 of the License, or |
||||||
|
* (at your option) any later version. |
||||||
|
* |
||||||
|
* This program is distributed in the hope that it will be useful, |
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||||
|
* GNU General Public License for more details. |
||||||
|
* |
||||||
|
* You should have received a copy of the GNU General Public License |
||||||
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*/ |
||||||
|
#ifndef UBDOWNLOADTHREAD_H |
||||||
|
#define UBDOWNLOADTHREAD_H |
||||||
|
|
||||||
|
#include <QThread> |
||||||
|
#include <QNetworkAccessManager> |
||||||
|
#include <QNetworkRequest> |
||||||
|
#include <QNetworkReply> |
||||||
|
|
||||||
|
class UBDownloadThread : public QThread |
||||||
|
{ |
||||||
|
Q_OBJECT |
||||||
|
public: |
||||||
|
UBDownloadThread(QObject* parent=0, const char* name="UBDownloadThread"); |
||||||
|
~UBDownloadThread(); |
||||||
|
void stopDownload(); |
||||||
|
void startDownload(int id, QString url); |
||||||
|
|
||||||
|
signals: |
||||||
|
void downloadFinised(int id); |
||||||
|
void downloadProgress(int id, qint64 current, qint64 total); |
||||||
|
|
||||||
|
protected: |
||||||
|
virtual void run(); |
||||||
|
|
||||||
|
private slots: |
||||||
|
void onDownloadProgress(qint64 received, qint64 total); |
||||||
|
void onDownloadFinished(); |
||||||
|
|
||||||
|
private: |
||||||
|
/** Flag used to stop the thread */ |
||||||
|
bool mbRun; |
||||||
|
/** The downloaded file id */ |
||||||
|
int mID; |
||||||
|
/** The downloaded file url */ |
||||||
|
QString mUrl; |
||||||
|
/** The network access manager */ |
||||||
|
QNetworkAccessManager* mpNam; |
||||||
|
/** The network reply */ |
||||||
|
QNetworkReply* mpReply; |
||||||
|
}; |
||||||
|
|
||||||
|
#endif // UBDOWNLOADTHREAD_H
|
||||||
@ -0,0 +1,203 @@ |
|||||||
|
/*
|
||||||
|
* This program is free software: you can redistribute it and/or modify |
||||||
|
* it under the terms of the GNU General Public License as published by |
||||||
|
* the Free Software Foundation, either version 3 of the License, or |
||||||
|
* (at your option) any later version. |
||||||
|
* |
||||||
|
* This program is distributed in the hope that it will be useful, |
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||||
|
* GNU General Public License for more details. |
||||||
|
* |
||||||
|
* You should have received a copy of the GNU General Public License |
||||||
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*/ |
||||||
|
#include <QDebug> |
||||||
|
#include <QHeaderView> |
||||||
|
#include <QStyleOptionProgressBarV2> |
||||||
|
#include <QApplication> |
||||||
|
|
||||||
|
#include "UBDownloadWidget.h" |
||||||
|
#include "core/UBApplication.h" |
||||||
|
|
||||||
|
/**
|
||||||
|
* \brief Constructor |
||||||
|
* @param parent as the parent widget |
||||||
|
* @param name as the widget object name |
||||||
|
*/ |
||||||
|
UBDownloadWidget::UBDownloadWidget(QWidget *parent, const char *name):QWidget(parent) |
||||||
|
, mpLayout(NULL) |
||||||
|
, mpBttnLayout(NULL) |
||||||
|
, mpTree(NULL) |
||||||
|
, mpCancelBttn(NULL) |
||||||
|
, mpItem(NULL) |
||||||
|
{ |
||||||
|
setObjectName(name); |
||||||
|
setWindowTitle(tr("Downloading files")); |
||||||
|
setAttribute(Qt::WA_StyledBackground, true); |
||||||
|
setStyleSheet(UBApplication::globalStyleSheet()); |
||||||
|
resize(400, 300); |
||||||
|
|
||||||
|
mpLayout = new QVBoxLayout(this); |
||||||
|
setLayout(mpLayout); |
||||||
|
|
||||||
|
mpTree = new QTreeWidget(this); |
||||||
|
mpTree->setRootIsDecorated(false); |
||||||
|
mpTree->header()->close(); |
||||||
|
mpLayout->addWidget(mpTree, 1); |
||||||
|
|
||||||
|
mpBttnLayout = new QHBoxLayout(); |
||||||
|
mpBttnLayout->addStretch(1); |
||||||
|
mpCancelBttn = new QPushButton(tr("Cancel"), this); |
||||||
|
mpBttnLayout->addWidget(mpCancelBttn, 0); |
||||||
|
mpLayout->addLayout(mpBttnLayout); |
||||||
|
|
||||||
|
connect(UBDownloadManager::downloadManager(), SIGNAL(fileAddedToDownload()), this, SLOT(onFileAddedToDownload())); |
||||||
|
connect(UBDownloadManager::downloadManager(), SIGNAL(downloadUpdated(int,qint64,qint64)), this, SLOT(onDownloadUpdated(int,qint64,qint64))); |
||||||
|
connect(UBDownloadManager::downloadManager(), SIGNAL(downloadFinished(int)), this, SLOT(onDownloadFinished(int))); |
||||||
|
connect(mpCancelBttn, SIGNAL(clicked()), this, SLOT(onCancelClicked())); |
||||||
|
} |
||||||
|
|
||||||
|
/**
|
||||||
|
* \brief Destructor |
||||||
|
*/ |
||||||
|
UBDownloadWidget::~UBDownloadWidget() |
||||||
|
{ |
||||||
|
if(NULL != mpItem) |
||||||
|
{ |
||||||
|
delete mpItem; |
||||||
|
mpItem = NULL; |
||||||
|
} |
||||||
|
if(NULL != mpCancelBttn) |
||||||
|
{ |
||||||
|
delete mpCancelBttn; |
||||||
|
mpCancelBttn = NULL; |
||||||
|
} |
||||||
|
if(NULL != mpTree) |
||||||
|
{ |
||||||
|
delete mpTree; |
||||||
|
mpTree = NULL; |
||||||
|
} |
||||||
|
if(NULL != mpBttnLayout) |
||||||
|
{ |
||||||
|
delete mpBttnLayout; |
||||||
|
mpBttnLayout = NULL; |
||||||
|
} |
||||||
|
if(NULL != mpLayout) |
||||||
|
{ |
||||||
|
delete mpLayout; |
||||||
|
mpLayout = NULL; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
/**
|
||||||
|
* \brief Refresh the tree of downloaded files |
||||||
|
*/ |
||||||
|
void UBDownloadWidget::onFileAddedToDownload() |
||||||
|
{ |
||||||
|
if(NULL != mpTree) |
||||||
|
{ |
||||||
|
mpTree->clear(); |
||||||
|
addCurrentDownloads(); |
||||||
|
addPendingDownloads(); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
/**
|
||||||
|
* \brief Add the current downloads |
||||||
|
*/ |
||||||
|
void UBDownloadWidget::addCurrentDownloads() |
||||||
|
{ |
||||||
|
QVector<sDownloadFileDesc> actualDL = UBDownloadManager::downloadManager()->currentDownloads(); |
||||||
|
for(int i=0; i<actualDL.size();i++) |
||||||
|
{ |
||||||
|
mpItem = new QTreeWidgetItem(mpTree); |
||||||
|
mpItem->setText(0, actualDL.at(i).name); |
||||||
|
mpItem->setData(0, Qt::UserRole, QVariant(actualDL.at(i).id)); |
||||||
|
mpTree->addTopLevelItem(mpItem); |
||||||
|
mpItem = new QTreeWidgetItem(mpTree); |
||||||
|
mpItem->setData(0, Qt::UserRole, actualDL.at(i).currentSize); |
||||||
|
mpItem->setData(0, Qt::UserRole + 1, actualDL.at(i).totalSize); |
||||||
|
mpItem->setData(0, Qt::UserRole + 2, actualDL.at(i).id); |
||||||
|
mpTree->addTopLevelItem(mpItem); |
||||||
|
mpTree->setItemDelegateForRow(((i+1)*2)-1, &mProgressBarDelegate); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
/**
|
||||||
|
* \brief Add the pending downloads |
||||||
|
*/ |
||||||
|
void UBDownloadWidget::addPendingDownloads() |
||||||
|
{ |
||||||
|
QVector<sDownloadFileDesc> pendingDL = UBDownloadManager::downloadManager()->pendingDownloads(); |
||||||
|
for(int i=0; i<pendingDL.size(); i++) |
||||||
|
{ |
||||||
|
mpItem = new QTreeWidgetItem(mpTree); |
||||||
|
mpItem->setText(0, pendingDL.at(i).name); |
||||||
|
mpItem->setData(0, Qt::UserRole, QVariant(pendingDL.at(i).id)); |
||||||
|
mpTree->addTopLevelItem(mpItem); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
/**
|
||||||
|
* \brief Update the progress bar |
||||||
|
* @param id as the downloaded file id |
||||||
|
* @param crnt as the current transfered size |
||||||
|
* @param total as the total size of the file |
||||||
|
*/ |
||||||
|
void UBDownloadWidget::onDownloadUpdated(int id, qint64 crnt, qint64 total) |
||||||
|
{ |
||||||
|
if(NULL != mpTree) |
||||||
|
{ |
||||||
|
QAbstractItemModel* model = mpTree->model(); |
||||||
|
if(NULL != model) |
||||||
|
{ |
||||||
|
for(int i=0; i< model->rowCount(); i++) |
||||||
|
{ |
||||||
|
QModelIndex currentIndex = model->index(i, 0); |
||||||
|
if(id == currentIndex.data(Qt::UserRole + 2)) |
||||||
|
{ |
||||||
|
// We found the right item, now we update the progress bar
|
||||||
|
model->setData(currentIndex, crnt, Qt::UserRole); |
||||||
|
model->setData(currentIndex, total, Qt::UserRole + 1); |
||||||
|
break; |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
/**
|
||||||
|
* \brief Handles the download finish notification |
||||||
|
* @param id as the downloaded file id |
||||||
|
*/ |
||||||
|
void UBDownloadWidget::onDownloadFinished(int id) |
||||||
|
{ |
||||||
|
// Refresh the file's list
|
||||||
|
onFileAddedToDownload(); |
||||||
|
} |
||||||
|
|
||||||
|
/**
|
||||||
|
* \brief Handles the Cancel button action |
||||||
|
*/ |
||||||
|
void UBDownloadWidget::onCancelClicked() |
||||||
|
{ |
||||||
|
|
||||||
|
} |
||||||
|
|
||||||
|
// ---------------------------------------------------------------------------------------------
|
||||||
|
UBDownloadProgressDelegate::UBDownloadProgressDelegate(QObject *parent):QItemDelegate(parent) |
||||||
|
{ |
||||||
|
|
||||||
|
} |
||||||
|
|
||||||
|
void UBDownloadProgressDelegate::paint(QPainter *painter, const QStyleOptionViewItem &option, const QModelIndex &index) const |
||||||
|
{ |
||||||
|
QStyleOptionProgressBarV2 opt; |
||||||
|
opt.rect = option.rect; |
||||||
|
opt.minimum = 0; |
||||||
|
opt.maximum = index.data(Qt::UserRole + 1).toInt(); |
||||||
|
opt.progress = index.data(Qt::UserRole).toInt(); |
||||||
|
|
||||||
|
QApplication::style()->drawControl(QStyle::CE_ProgressBar, &opt, painter, 0); |
||||||
|
} |
||||||
@ -0,0 +1,69 @@ |
|||||||
|
/*
|
||||||
|
* This program is free software: you can redistribute it and/or modify |
||||||
|
* it under the terms of the GNU General Public License as published by |
||||||
|
* the Free Software Foundation, either version 3 of the License, or |
||||||
|
* (at your option) any later version. |
||||||
|
* |
||||||
|
* This program is distributed in the hope that it will be useful, |
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||||
|
* GNU General Public License for more details. |
||||||
|
* |
||||||
|
* You should have received a copy of the GNU General Public License |
||||||
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*/ |
||||||
|
#ifndef UBDOWNLOADWIDGET_H |
||||||
|
#define UBDOWNLOADWIDGET_H |
||||||
|
|
||||||
|
#include <QWidget> |
||||||
|
#include <QHBoxLayout> |
||||||
|
#include <QVBoxLayout> |
||||||
|
#include <QPushButton> |
||||||
|
#include <QTreeWidget> |
||||||
|
#include <QTreeWidgetItem> |
||||||
|
#include <QItemDelegate> |
||||||
|
|
||||||
|
#include "core/UBDownloadManager.h" |
||||||
|
|
||||||
|
class UBDownloadProgressDelegate : public QItemDelegate |
||||||
|
{ |
||||||
|
Q_OBJECT |
||||||
|
public: |
||||||
|
UBDownloadProgressDelegate(QObject* parent=0); |
||||||
|
|
||||||
|
protected: |
||||||
|
void paint(QPainter *painter, const QStyleOptionViewItem &option, const QModelIndex &index) const; |
||||||
|
}; |
||||||
|
|
||||||
|
class UBDownloadWidget : public QWidget |
||||||
|
{ |
||||||
|
Q_OBJECT |
||||||
|
public: |
||||||
|
UBDownloadWidget(QWidget* parent=0, const char* name="UBDownloadWidget"); |
||||||
|
~UBDownloadWidget(); |
||||||
|
|
||||||
|
private slots: |
||||||
|
void onFileAddedToDownload(); |
||||||
|
void onDownloadUpdated(int id, qint64 crnt, qint64 total); |
||||||
|
void onDownloadFinished(int id); |
||||||
|
void onCancelClicked(); |
||||||
|
|
||||||
|
private: |
||||||
|
void addCurrentDownloads(); |
||||||
|
void addPendingDownloads(); |
||||||
|
|
||||||
|
/** The general layout of this widget */ |
||||||
|
QVBoxLayout* mpLayout; |
||||||
|
/** The button layout */ |
||||||
|
QHBoxLayout* mpBttnLayout; |
||||||
|
/** The treeview that will display the files list */ |
||||||
|
QTreeWidget* mpTree; |
||||||
|
/** The 'Cancel' button */ |
||||||
|
QPushButton* mpCancelBttn; |
||||||
|
/** A temporary tree widget item */ |
||||||
|
QTreeWidgetItem* mpItem; |
||||||
|
/** The delegate that will draw the progressbars */ |
||||||
|
UBDownloadProgressDelegate mProgressBarDelegate; |
||||||
|
}; |
||||||
|
|
||||||
|
#endif // UBDOWNLOADWIDGET_H
|
||||||
Loading…
Reference in new issue