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
26class 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
34class Compressor : public QObject
35{
36 Q_OBJECT
37
38public:
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
69signals:
70 void readyRead();
71 void error(Compressor::Error errorCode = StreamError);
72
73private slots:
74 void readData();
75
76private:
77 bool initStreams();
78 void writeData();
79
80private:
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