1 /****************************************************************************

     2 **

     3 ** Copyright (C) 2009 Nokia Corporation and/or its subsidiary(-ies).

     4 ** All rights reserved.

     5 ** Contact: Nokia Corporation (qt-info@nokia.com)

     6 **

     7 ** This file is part of the examples of the Qt Toolkit.

     8 **

     9 ** $QT_BEGIN_LICENSE:LGPL$

    10 ** No Commercial Usage

    11 ** This file contains pre-release code and may not be distributed.

    12 ** You may use this file in accordance with the terms and conditions

    13 ** contained in the Technology Preview License Agreement accompanying

    14 ** this package.

    15 **

    16 ** GNU Lesser General Public License Usage

    17 ** Alternatively, this file may be used under the terms of the GNU Lesser

    18 ** General Public License version 2.1 as published by the Free Software

    19 ** Foundation and appearing in the file LICENSE.LGPL included in the

    20 ** packaging of this file.  Please review the following information to

    21 ** ensure the GNU Lesser General Public License version 2.1 requirements

    22 ** will be met: http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.

    23 **

    24 ** In addition, as a special exception, Nokia gives you certain additional

    25 ** rights.  These rights are described in the Nokia Qt LGPL Exception

    26 ** version 1.1, included in the file LGPL_EXCEPTION.txt in this package.

    27 **

    28 ** If you have questions regarding the use of this file, please contact

    29 ** Nokia at qt-info@nokia.com.

    30 **

    31 **

    32 **

    33 **

    34 **

    35 **

    36 **

    37 **

    38 ** $QT_END_LICENSE$

    39 **

    40 ****************************************************************************/

    41 

    42 #ifndef FILEMANAGER_H

    43 #define FILEMANAGER_H

    44 

    45 #include <QBitArray>

    46 #include <QList>

    47 #include <QMutex>

    48 #include <QThread>

    49 #include <QWaitCondition>

    50 

    51 #include "metainfo.h"

    52 

    53 QT_BEGIN_NAMESPACE

    54 class QByteArray;

    55 class QFile;

    56 class QTimerEvent;

    57 QT_END_NAMESPACE

    58 

    59 class FileManager : public QThread

    60 {

    61     Q_OBJECT

    62 

    63 public:

    64     FileManager(QObject *parent = 0);

    65     virtual ~FileManager();

    66 

    67     inline void setMetaInfo(const MetaInfo &info) { metaInfo = info; }

    68     inline void setDestinationFolder(const QString &directory) { destinationPath = directory; }

    69     

    70     int read(int pieceIndex, int offset, int length);

    71     void write(int pieceIndex, int offset, const QByteArray &data);

    72     void verifyPiece(int pieceIndex);

    73     inline qint64 totalSize() const { return totalLength; }

    74 

    75     inline int pieceCount() const { return numPieces; }

    76     int pieceLengthAt(int pieceIndex) const;

    77 

    78     QBitArray completedPieces() const;

    79     void setCompletedPieces(const QBitArray &pieces);

    80 

    81     QString errorString() const;

    82 

    83 public slots:

    84     void startDataVerification();

    85 

    86 signals:

    87     void dataRead(int id, int pieceIndex, int offset, const QByteArray &data);

    88     void error();

    89     void verificationProgress(int percent);

    90     void verificationDone();

    91     void pieceVerified(int pieceIndex, bool verified);

    92 

    93 protected:

    94     void run();

    95 

    96 private slots:

    97     bool verifySinglePiece(int pieceIndex);

    98     void wakeUp();

    99 

   100 private:

   101     bool generateFiles();

   102     QByteArray readBlock(int pieceIndex, int offset, int length);

   103     bool writeBlock(int pieceIndex, int offset, const QByteArray &data);

   104     void verifyFileContents();

   105 

   106     struct WriteRequest {

   107         int pieceIndex;

   108         int offset;

   109         QByteArray data;

   110     };

   111     struct ReadRequest {

   112         int pieceIndex;

   113         int offset;

   114         int length;

   115         int id;

   116     };

   117 

   118     QString errString;

   119     QString destinationPath;

   120     MetaInfo metaInfo;

   121     QList<QFile *> files;

   122     QList<QByteArray> sha1s;

   123     QBitArray verifiedPieces;

   124 

   125     bool newFile;

   126     int pieceLength;

   127     qint64 totalLength;

   128     int numPieces;

   129     int readId;

   130     bool startVerification;

   131     bool quit;

   132     bool wokeUp;

   133 

   134     QList<WriteRequest> writeRequests;

   135     QList<ReadRequest> readRequests;

   136     QList<int> pendingVerificationRequests;

   137     QList<int> newPendingVerificationRequests;

   138     QList<qint64> fileSizes;

   139 

   140     mutable QMutex mutex;

   141     mutable QWaitCondition cond;

   142 };

   143 

   144 #endif