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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
|
// license:BSD-3-Clause
// copyright-holders:Aaron Giles
/***************************************************************************
avhuff.h
Audio/video compression and decompression helpers.
***************************************************************************/
#pragma once
#ifndef __AVHUFF_H__
#define __AVHUFF_H__
#include "osdcore.h"
#include "coretmpl.h"
#include "bitmap.h"
#include "huffman.h"
#include "flac.h"
//**************************************************************************
// CONSTANTS
//**************************************************************************
#define AVHUFF_USE_FLAC (1)
// errors
enum avhuff_error
{
AVHERR_NONE = 0,
AVHERR_INVALID_DATA,
AVHERR_VIDEO_TOO_LARGE,
AVHERR_AUDIO_TOO_LARGE,
AVHERR_METADATA_TOO_LARGE,
AVHERR_OUT_OF_MEMORY,
AVHERR_COMPRESSION_ERROR,
AVHERR_TOO_MANY_CHANNELS,
AVHERR_INVALID_CONFIGURATION,
AVHERR_INVALID_PARAMETER,
AVHERR_BUFFER_TOO_SMALL
};
//**************************************************************************
// TYPE DEFINITIONS
//**************************************************************************
// ======================> av_codec_decompress_config
// decompression configuration
class avhuff_decompress_config
{
public:
avhuff_decompress_config()
: maxsamples(0),
actsamples(nullptr),
maxmetalength(0),
actmetalength(nullptr),
metadata(nullptr)
{
memset(audio, 0, sizeof(audio));
}
bitmap_yuy16 video; // pointer to video bitmap
UINT32 maxsamples; // maximum number of samples per channel
UINT32 * actsamples; // actual number of samples per channel
INT16 * audio[16]; // pointer to individual audio channels
UINT32 maxmetalength; // maximum length of metadata
UINT32 * actmetalength; // actual length of metadata
UINT8 * metadata; // pointer to metadata buffer
};
// ======================> avhuff_encoder
// core state for the codec
class avhuff_encoder
{
public:
// construction/destruction
avhuff_encoder();
// encode/decode
avhuff_error encode_data(const UINT8 *source, UINT8 *dest, UINT32 &complength);
// static helpers
static UINT32 raw_data_size(const UINT8 *data);
static UINT32 raw_data_size(UINT32 width, UINT32 height, UINT8 channels, UINT32 numsamples, UINT32 metadatasize = 0) { return 12 + channels * numsamples * 2 + width * height * 2; }
static avhuff_error assemble_data(std::vector<UINT8> &buffer, bitmap_yuy16 &bitmap, UINT8 channels, UINT32 numsamples, INT16 **samples, UINT8 *metadata = nullptr, UINT32 metadatasize = 0);
private:
// delta-RLE Huffman encoder
class deltarle_encoder
{
public:
// construction/destruction
deltarle_encoder()
: m_rlecount(0) { }
// histogramming
UINT16 *rle_and_histo_bitmap(const UINT8 *source, UINT32 items_per_row, UINT32 item_advance, UINT32 row_count);
// encoding
void flush_rle() { m_rlecount = 0; }
void encode_one(bitstream_out &bitbuf, UINT16 *&rleptr);
huffman_error export_tree_rle(bitstream_out &bitbuf) { return m_encoder.export_tree_rle(bitbuf); }
private:
// internal state
int m_rlecount;
huffman_encoder<256 + 16> m_encoder;
std::vector<UINT16> m_rlebuffer;
};
// internal helpers
avhuff_error encode_audio(const UINT8 *source, int channels, int samples, UINT8 *dest, UINT8 *sizes);
avhuff_error encode_video(const UINT8 *source, int width, int height, UINT8 *dest, UINT32 &complength);
avhuff_error encode_video_lossless(const UINT8 *source, int width, int height, UINT8 *dest, UINT32 &complength);
// video encoding contexts
deltarle_encoder m_ycontext;
deltarle_encoder m_cbcontext;
deltarle_encoder m_crcontext;
// audio encoding contexts
std::vector<UINT8> m_audiobuffer;
#if AVHUFF_USE_FLAC
flac_encoder m_flac_encoder;
#else
huffman_8bit_encoder m_audiohi_encoder;
huffman_8bit_encoder m_audiolo_encoder;
#endif
};
// ======================> avhuff_decoder
// core state for the codec
class avhuff_decoder
{
public:
// construction/destruction
avhuff_decoder();
// configuration
void configure(const avhuff_decompress_config &config);
// encode/decode
avhuff_error decode_data(const UINT8 *source, UINT32 complength, UINT8 *dest);
private:
// delta-RLE Huffman decoder
class deltarle_decoder
{
public:
// construction/destruction
deltarle_decoder()
: m_rlecount(0), m_prevdata(0) { }
// general
void reset() { m_rlecount = m_prevdata = 0; }
// decoding
void flush_rle() { m_rlecount = 0; }
UINT32 decode_one(bitstream_in &bitbuf);
huffman_error import_tree_rle(bitstream_in &bitbuf) { return m_decoder.import_tree_rle(bitbuf); }
private:
// internal state
int m_rlecount;
UINT8 m_prevdata;
huffman_decoder<256 + 16> m_decoder;
};
// internal helpers
avhuff_error decode_audio(int channels, int samples, const UINT8 *source, UINT8 **dest, UINT32 dxor, const UINT8 *sizes);
avhuff_error decode_video(int width, int height, const UINT8 *source, UINT32 complength, UINT8 *dest, UINT32 dstride, UINT32 dxor);
avhuff_error decode_video_lossless(int width, int height, const UINT8 *source, UINT32 complength, UINT8 *dest, UINT32 dstride, UINT32 dxor);
// internal state
avhuff_decompress_config m_config;
// video decoding contexts
deltarle_decoder m_ycontext;
deltarle_decoder m_cbcontext;
deltarle_decoder m_crcontext;
// audio decoding contexts
huffman_8bit_decoder m_audiohi_decoder;
huffman_8bit_decoder m_audiolo_decoder;
#if AVHUFF_USE_FLAC
flac_decoder m_flac_decoder;
#endif
};
#endif
|