summaryrefslogtreecommitdiff
path: root/i/marvin/src/data/data_input_zlib.cc
blob: f8f532204aed5f22df66357524591bc5fdeb199c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
// data_input_zlib.cc
// robert - programme du robot 2005. {{{
//
// Copyright (C) 2005 Nicolas Schodet
//
// Robot APB Team/Efrei 2005.
//        Web: http://assos.efrei.fr/robot/
//      Email: robot AT efrei DOT fr
//
// 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 2 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, write to the Free Software
// Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
//
// }}}
#include "data_input_zlib.hh"

#include <stdexcept>

/// Constructeur.
DataInputZlib::DataInputZlib (DataInput &data)
    : data_ (data), inBuf_ (0)
{
    // Allocate input buffer.
    inBuf_ = new uint8_t[inBufSize_];
    // Initialise the zlib structure.
    zStream_.next_in = 0;
    zStream_.avail_in = 0;
    zStream_.next_out = 0;
    zStream_.avail_out = 0;
    zStream_.zalloc = 0;
    zStream_.zfree = 0;
    zStream_.opaque = 0;
    if (inflateInit2 (&zStream_, 32 + 15) != Z_OK)
      {
	delete[] inBuf_;
	throw std::runtime_error (zStream_.msg);
      }
}

/// Destructeur.
DataInputZlib::~DataInputZlib (void)
{
    inflateEnd (&zStream_);
    delete[] inBuf_;
}

/// Lit des donn�es. M�me convention que le read de la libc, mais lance
/// une exception en cas d'erreur.
unsigned
DataInputZlib::read (uint8_t *buf, unsigned size)
{
    // Set output buffer.
    zStream_.next_out = buf;
    zStream_.avail_out = size;
    // While not satisfied.
    while (zStream_.avail_out)
      {
	// Feed more data.
	if (zStream_.avail_in == 0)
	  {
	    zStream_.next_in = inBuf_;
	    zStream_.avail_in = data_.read (inBuf_, inBufSize_);
	  }
	// Uncompress.
	int r = inflate (&zStream_, Z_SYNC_FLUSH);
	// Check result.
	if (r == Z_STREAM_END)
	    return size - zStream_.avail_out;
	else if (r != Z_OK)
	    throw std::runtime_error (zStream_.msg);
      }
    return size;
}