1 | /*************************************************************************** |
2 | * Copyright (C) 2005-2014 by the Quassel Project * |
3 | * devel@quassel-irc.org * |
4 | * * |
5 | * This program is free software; you can redistribute it and/or modify * |
6 | * it under the terms of the GNU General Public License as published by * |
7 | * the Free Software Foundation; either version 2 of the License, or * |
8 | * (at your option) version 3. * |
9 | * * |
10 | * This program is distributed in the hope that it will be useful, * |
11 | * but WITHOUT ANY WARRANTY; without even the implied warranty of * |
12 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * |
13 | * GNU General Public License for more details. * |
14 | * * |
15 | * You should have received a copy of the GNU General Public License * |
16 | * along with this program; if not, write to the * |
17 | * Free Software Foundation, Inc., * |
18 | * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. * |
19 | ***************************************************************************/ |
20 | |
21 | #ifndef COMPRESSOR_H |
22 | #define COMPRESSOR_H |
23 | |
24 | #include <QObject> |
25 | |
26 | class QTcpSocket; |
27 | |
28 | #ifdef HAVE_ZLIB |
29 | typedef struct z_stream_s *z_streamp; |
30 | #else |
31 | typedef struct mz_stream_s *z_streamp; |
32 | #endif |
33 | |
34 | class Compressor : public QObject |
35 | { |
36 | Q_OBJECT |
37 | |
38 | public: |
39 | enum CompressionLevel { |
40 | NoCompression, |
41 | DefaultCompression, |
42 | BestCompression, |
43 | BestSpeed |
44 | }; |
45 | |
46 | enum Error { |
47 | NoError, |
48 | StreamError, |
49 | DeviceError |
50 | }; |
51 | |
52 | enum WriteBufferHint { |
53 | NoFlush, |
54 | Flush |
55 | }; |
56 | |
57 | Compressor(QTcpSocket *socket, CompressionLevel level, QObject *parent = 0); |
58 | ~Compressor(); |
59 | |
60 | CompressionLevel compressionLevel() const { return _level; } |
61 | |
62 | qint64 bytesAvailable() const; |
63 | |
64 | qint64 read(char *data, qint64 maxSize); |
65 | qint64 write(const char *data, qint64 count, WriteBufferHint flush = Flush); |
66 | |
67 | void flush(); |
68 | |
69 | signals: |
70 | void readyRead(); |
71 | void error(Compressor::Error errorCode = StreamError); |
72 | |
73 | private slots: |
74 | void readData(); |
75 | |
76 | private: |
77 | bool initStreams(); |
78 | void writeData(); |
79 | |
80 | private: |
81 | QTcpSocket *_socket; |
82 | CompressionLevel _level; |
83 | |
84 | QByteArray _readBuffer; |
85 | QByteArray _writeBuffer; |
86 | |
87 | QByteArray _inputBuffer; |
88 | QByteArray _outputBuffer; |
89 | |
90 | z_streamp _inflater; |
91 | z_streamp _deflater; |
92 | }; |
93 | |
94 | #endif |
95 | |