summaryrefslogtreecommitdiff
path: root/Build/source/utils/asymptote/prc/PRCbitStream.h
blob: 89482168f9eb2b220381e82b3052a62832788c19 (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
/************
*
*   This file is part of a tool for producing 3D content in the PRC format.
*   Copyright (C) 2008  Orest Shardt <shardtor (at) gmail dot com>
*
*   This program is free software: you can redistribute it and/or modify
*   it under the terms of the GNU Lesser 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 Lesser General Public License for more details.
*
*   You should have received a copy of the GNU Lesser General Public License
*   along with this program.  If not, see <http://www.gnu.org/licenses/>.
*
*************/

#ifndef __PRC_BIT_STREAM_H
#define __PRC_BIT_STREAM_H

#include <inttypes.h>
#include <string>
#include <stdlib.h>

#define CHUNK_SIZE (1024)
// Is this a reasonable initial size?

class PRCbitStream
{
  public:
    PRCbitStream(uint8_t*& buff, unsigned int l) : byteIndex(0), bitIndex(0),
                 allocatedLength(l), data(buff), compressed(false)
    {
      if(data == 0)
      {
        getAChunk();
      }
    }

    unsigned int getSize() const;
    uint8_t* getData();

    PRCbitStream& operator <<(const std::string&);
    PRCbitStream& operator <<(bool);
    PRCbitStream& operator <<(uint32_t);
    PRCbitStream& operator <<(uint8_t);
    PRCbitStream& operator <<(int32_t);
    PRCbitStream& operator <<(double);
    PRCbitStream& operator <<(const char*);

    void compress();
  private:
    void writeBit(bool);
    void writeBits(uint32_t,uint8_t);
    void writeByte(uint8_t);
    void nextByte();
    void nextBit();
    void getAChunk();
    // bitIndex is "big endian", zero based, location of next bit to write
    unsigned int byteIndex,bitIndex;
    unsigned int allocatedLength;
    uint8_t*& data;
    bool compressed;
    uint32_t compressedDataSize;
};

#endif // __PRC_BIT_STREAM_H