1// Copyright 2011 Google Inc. All Rights Reserved.
2//
3// Use of this source code is governed by a BSD-style license
4// that can be found in the COPYING file in the root of the source
5// tree. An additional intellectual property rights grant can be found
6// in the file PATENTS. All contributing project authors may
7// be found in the AUTHORS file in the root of the source tree.
8// -----------------------------------------------------------------------------
9//
10// Header syntax writing
11//
12// Author: Skal (pascal.massimino@gmail.com)
13
14#include <assert.h>
15
16#include "src/utils/utils.h"
17#include "src/webp/format_constants.h" // RIFF constants
18#include "src/webp/mux_types.h" // ALPHA_FLAG
19#include "src/enc/vp8i_enc.h"
20
21//------------------------------------------------------------------------------
22// Helper functions
23
24static int IsVP8XNeeded(const VP8Encoder* const enc) {
25 return !!enc->has_alpha_; // Currently the only case when VP8X is needed.
26 // This could change in the future.
27}
28
29static int PutPaddingByte(const WebPPicture* const pic) {
30 const uint8_t pad_byte[1] = { 0 };
31 return !!pic->writer(pad_byte, 1, pic);
32}
33
34//------------------------------------------------------------------------------
35// Writers for header's various pieces (in order of appearance)
36
37static WebPEncodingError PutRIFFHeader(const VP8Encoder* const enc,
38 size_t riff_size) {
39 const WebPPicture* const pic = enc->pic_;
40 uint8_t riff[RIFF_HEADER_SIZE] = {
41 'R', 'I', 'F', 'F', 0, 0, 0, 0, 'W', 'E', 'B', 'P'
42 };
43 assert(riff_size == (uint32_t)riff_size);
44 PutLE32(riff + TAG_SIZE, (uint32_t)riff_size);
45 if (!pic->writer(riff, sizeof(riff), pic)) {
46 return VP8_ENC_ERROR_BAD_WRITE;
47 }
48 return VP8_ENC_OK;
49}
50
51static WebPEncodingError PutVP8XHeader(const VP8Encoder* const enc) {
52 const WebPPicture* const pic = enc->pic_;
53 uint8_t vp8x[CHUNK_HEADER_SIZE + VP8X_CHUNK_SIZE] = {
54 'V', 'P', '8', 'X'
55 };
56 uint32_t flags = 0;
57
58 assert(IsVP8XNeeded(enc));
59 assert(pic->width >= 1 && pic->height >= 1);
60 assert(pic->width <= MAX_CANVAS_SIZE && pic->height <= MAX_CANVAS_SIZE);
61
62 if (enc->has_alpha_) {
63 flags |= ALPHA_FLAG;
64 }
65
66 PutLE32(vp8x + TAG_SIZE, VP8X_CHUNK_SIZE);
67 PutLE32(vp8x + CHUNK_HEADER_SIZE, flags);
68 PutLE24(vp8x + CHUNK_HEADER_SIZE + 4, pic->width - 1);
69 PutLE24(vp8x + CHUNK_HEADER_SIZE + 7, pic->height - 1);
70 if (!pic->writer(vp8x, sizeof(vp8x), pic)) {
71 return VP8_ENC_ERROR_BAD_WRITE;
72 }
73 return VP8_ENC_OK;
74}
75
76static WebPEncodingError PutAlphaChunk(const VP8Encoder* const enc) {
77 const WebPPicture* const pic = enc->pic_;
78 uint8_t alpha_chunk_hdr[CHUNK_HEADER_SIZE] = {
79 'A', 'L', 'P', 'H'
80 };
81
82 assert(enc->has_alpha_);
83
84 // Alpha chunk header.
85 PutLE32(alpha_chunk_hdr + TAG_SIZE, enc->alpha_data_size_);
86 if (!pic->writer(alpha_chunk_hdr, sizeof(alpha_chunk_hdr), pic)) {
87 return VP8_ENC_ERROR_BAD_WRITE;
88 }
89
90 // Alpha chunk data.
91 if (!pic->writer(enc->alpha_data_, enc->alpha_data_size_, pic)) {
92 return VP8_ENC_ERROR_BAD_WRITE;
93 }
94
95 // Padding.
96 if ((enc->alpha_data_size_ & 1) && !PutPaddingByte(pic)) {
97 return VP8_ENC_ERROR_BAD_WRITE;
98 }
99 return VP8_ENC_OK;
100}
101
102static WebPEncodingError PutVP8Header(const WebPPicture* const pic,
103 size_t vp8_size) {
104 uint8_t vp8_chunk_hdr[CHUNK_HEADER_SIZE] = {
105 'V', 'P', '8', ' '
106 };
107 assert(vp8_size == (uint32_t)vp8_size);
108 PutLE32(vp8_chunk_hdr + TAG_SIZE, (uint32_t)vp8_size);
109 if (!pic->writer(vp8_chunk_hdr, sizeof(vp8_chunk_hdr), pic)) {
110 return VP8_ENC_ERROR_BAD_WRITE;
111 }
112 return VP8_ENC_OK;
113}
114
115static WebPEncodingError PutVP8FrameHeader(const WebPPicture* const pic,
116 int profile, size_t size0) {
117 uint8_t vp8_frm_hdr[VP8_FRAME_HEADER_SIZE];
118 uint32_t bits;
119
120 if (size0 >= VP8_MAX_PARTITION0_SIZE) { // partition #0 is too big to fit
121 return VP8_ENC_ERROR_PARTITION0_OVERFLOW;
122 }
123
124 // Paragraph 9.1.
125 bits = 0 // keyframe (1b)
126 | (profile << 1) // profile (3b)
127 | (1 << 4) // visible (1b)
128 | ((uint32_t)size0 << 5); // partition length (19b)
129 vp8_frm_hdr[0] = (bits >> 0) & 0xff;
130 vp8_frm_hdr[1] = (bits >> 8) & 0xff;
131 vp8_frm_hdr[2] = (bits >> 16) & 0xff;
132 // signature
133 vp8_frm_hdr[3] = (VP8_SIGNATURE >> 16) & 0xff;
134 vp8_frm_hdr[4] = (VP8_SIGNATURE >> 8) & 0xff;
135 vp8_frm_hdr[5] = (VP8_SIGNATURE >> 0) & 0xff;
136 // dimensions
137 vp8_frm_hdr[6] = pic->width & 0xff;
138 vp8_frm_hdr[7] = pic->width >> 8;
139 vp8_frm_hdr[8] = pic->height & 0xff;
140 vp8_frm_hdr[9] = pic->height >> 8;
141
142 if (!pic->writer(vp8_frm_hdr, sizeof(vp8_frm_hdr), pic)) {
143 return VP8_ENC_ERROR_BAD_WRITE;
144 }
145 return VP8_ENC_OK;
146}
147
148// WebP Headers.
149static int PutWebPHeaders(const VP8Encoder* const enc, size_t size0,
150 size_t vp8_size, size_t riff_size) {
151 WebPPicture* const pic = enc->pic_;
152 WebPEncodingError err = VP8_ENC_OK;
153
154 // RIFF header.
155 err = PutRIFFHeader(enc, riff_size);
156 if (err != VP8_ENC_OK) goto Error;
157
158 // VP8X.
159 if (IsVP8XNeeded(enc)) {
160 err = PutVP8XHeader(enc);
161 if (err != VP8_ENC_OK) goto Error;
162 }
163
164 // Alpha.
165 if (enc->has_alpha_) {
166 err = PutAlphaChunk(enc);
167 if (err != VP8_ENC_OK) goto Error;
168 }
169
170 // VP8 header.
171 err = PutVP8Header(pic, vp8_size);
172 if (err != VP8_ENC_OK) goto Error;
173
174 // VP8 frame header.
175 err = PutVP8FrameHeader(pic, enc->profile_, size0);
176 if (err != VP8_ENC_OK) goto Error;
177
178 // All OK.
179 return 1;
180
181 // Error.
182 Error:
183 return WebPEncodingSetError(pic, err);
184}
185
186// Segmentation header
187static void PutSegmentHeader(VP8BitWriter* const bw,
188 const VP8Encoder* const enc) {
189 const VP8EncSegmentHeader* const hdr = &enc->segment_hdr_;
190 const VP8EncProba* const proba = &enc->proba_;
191 if (VP8PutBitUniform(bw, (hdr->num_segments_ > 1))) {
192 // We always 'update' the quant and filter strength values
193 const int update_data = 1;
194 int s;
195 VP8PutBitUniform(bw, hdr->update_map_);
196 if (VP8PutBitUniform(bw, update_data)) {
197 // we always use absolute values, not relative ones
198 VP8PutBitUniform(bw, 1); // (segment_feature_mode = 1. Paragraph 9.3.)
199 for (s = 0; s < NUM_MB_SEGMENTS; ++s) {
200 VP8PutSignedBits(bw, enc->dqm_[s].quant_, 7);
201 }
202 for (s = 0; s < NUM_MB_SEGMENTS; ++s) {
203 VP8PutSignedBits(bw, enc->dqm_[s].fstrength_, 6);
204 }
205 }
206 if (hdr->update_map_) {
207 for (s = 0; s < 3; ++s) {
208 if (VP8PutBitUniform(bw, (proba->segments_[s] != 255u))) {
209 VP8PutBits(bw, proba->segments_[s], 8);
210 }
211 }
212 }
213 }
214}
215
216// Filtering parameters header
217static void PutFilterHeader(VP8BitWriter* const bw,
218 const VP8EncFilterHeader* const hdr) {
219 const int use_lf_delta = (hdr->i4x4_lf_delta_ != 0);
220 VP8PutBitUniform(bw, hdr->simple_);
221 VP8PutBits(bw, hdr->level_, 6);
222 VP8PutBits(bw, hdr->sharpness_, 3);
223 if (VP8PutBitUniform(bw, use_lf_delta)) {
224 // '0' is the default value for i4x4_lf_delta_ at frame #0.
225 const int need_update = (hdr->i4x4_lf_delta_ != 0);
226 if (VP8PutBitUniform(bw, need_update)) {
227 // we don't use ref_lf_delta => emit four 0 bits
228 VP8PutBits(bw, 0, 4);
229 // we use mode_lf_delta for i4x4
230 VP8PutSignedBits(bw, hdr->i4x4_lf_delta_, 6);
231 VP8PutBits(bw, 0, 3); // all others unused
232 }
233 }
234}
235
236// Nominal quantization parameters
237static void PutQuant(VP8BitWriter* const bw,
238 const VP8Encoder* const enc) {
239 VP8PutBits(bw, enc->base_quant_, 7);
240 VP8PutSignedBits(bw, enc->dq_y1_dc_, 4);
241 VP8PutSignedBits(bw, enc->dq_y2_dc_, 4);
242 VP8PutSignedBits(bw, enc->dq_y2_ac_, 4);
243 VP8PutSignedBits(bw, enc->dq_uv_dc_, 4);
244 VP8PutSignedBits(bw, enc->dq_uv_ac_, 4);
245}
246
247// Partition sizes
248static int EmitPartitionsSize(const VP8Encoder* const enc,
249 WebPPicture* const pic) {
250 uint8_t buf[3 * (MAX_NUM_PARTITIONS - 1)];
251 int p;
252 for (p = 0; p < enc->num_parts_ - 1; ++p) {
253 const size_t part_size = VP8BitWriterSize(enc->parts_ + p);
254 if (part_size >= VP8_MAX_PARTITION_SIZE) {
255 return WebPEncodingSetError(pic, VP8_ENC_ERROR_PARTITION_OVERFLOW);
256 }
257 buf[3 * p + 0] = (part_size >> 0) & 0xff;
258 buf[3 * p + 1] = (part_size >> 8) & 0xff;
259 buf[3 * p + 2] = (part_size >> 16) & 0xff;
260 }
261 if (p && !pic->writer(buf, 3 * p, pic)) {
262 return WebPEncodingSetError(pic, VP8_ENC_ERROR_BAD_WRITE);
263 }
264 return 1;
265}
266
267//------------------------------------------------------------------------------
268
269static int GeneratePartition0(VP8Encoder* const enc) {
270 VP8BitWriter* const bw = &enc->bw_;
271 const int mb_size = enc->mb_w_ * enc->mb_h_;
272 uint64_t pos1, pos2, pos3;
273
274 pos1 = VP8BitWriterPos(bw);
275 if (!VP8BitWriterInit(bw, mb_size * 7 / 8)) { // ~7 bits per macroblock
276 return WebPEncodingSetError(enc->pic_, VP8_ENC_ERROR_OUT_OF_MEMORY);
277 }
278 VP8PutBitUniform(bw, 0); // colorspace
279 VP8PutBitUniform(bw, 0); // clamp type
280
281 PutSegmentHeader(bw, enc);
282 PutFilterHeader(bw, &enc->filter_hdr_);
283 VP8PutBits(bw, enc->num_parts_ == 8 ? 3 :
284 enc->num_parts_ == 4 ? 2 :
285 enc->num_parts_ == 2 ? 1 : 0, 2);
286 PutQuant(bw, enc);
287 VP8PutBitUniform(bw, 0); // no proba update
288 VP8WriteProbas(bw, &enc->proba_);
289 pos2 = VP8BitWriterPos(bw);
290 VP8CodeIntraModes(enc);
291 VP8BitWriterFinish(bw);
292
293 pos3 = VP8BitWriterPos(bw);
294
295#if !defined(WEBP_DISABLE_STATS)
296 if (enc->pic_->stats) {
297 enc->pic_->stats->header_bytes[0] = (int)((pos2 - pos1 + 7) >> 3);
298 enc->pic_->stats->header_bytes[1] = (int)((pos3 - pos2 + 7) >> 3);
299 enc->pic_->stats->alpha_data_size = (int)enc->alpha_data_size_;
300 }
301#else
302 (void)pos1;
303 (void)pos2;
304 (void)pos3;
305#endif
306 if (bw->error_) {
307 return WebPEncodingSetError(enc->pic_, VP8_ENC_ERROR_OUT_OF_MEMORY);
308 }
309 return 1;
310}
311
312void VP8EncFreeBitWriters(VP8Encoder* const enc) {
313 int p;
314 VP8BitWriterWipeOut(&enc->bw_);
315 for (p = 0; p < enc->num_parts_; ++p) {
316 VP8BitWriterWipeOut(enc->parts_ + p);
317 }
318}
319
320int VP8EncWrite(VP8Encoder* const enc) {
321 WebPPicture* const pic = enc->pic_;
322 VP8BitWriter* const bw = &enc->bw_;
323 const int task_percent = 19;
324 const int percent_per_part = task_percent / enc->num_parts_;
325 const int final_percent = enc->percent_ + task_percent;
326 int ok = 0;
327 size_t vp8_size, pad, riff_size;
328 int p;
329
330 // Partition #0 with header and partition sizes
331 ok = GeneratePartition0(enc);
332 if (!ok) return 0;
333
334 // Compute VP8 size
335 vp8_size = VP8_FRAME_HEADER_SIZE +
336 VP8BitWriterSize(bw) +
337 3 * (enc->num_parts_ - 1);
338 for (p = 0; p < enc->num_parts_; ++p) {
339 vp8_size += VP8BitWriterSize(enc->parts_ + p);
340 }
341 pad = vp8_size & 1;
342 vp8_size += pad;
343
344 // Compute RIFF size
345 // At the minimum it is: "WEBPVP8 nnnn" + VP8 data size.
346 riff_size = TAG_SIZE + CHUNK_HEADER_SIZE + vp8_size;
347 if (IsVP8XNeeded(enc)) { // Add size for: VP8X header + data.
348 riff_size += CHUNK_HEADER_SIZE + VP8X_CHUNK_SIZE;
349 }
350 if (enc->has_alpha_) { // Add size for: ALPH header + data.
351 const uint32_t padded_alpha_size = enc->alpha_data_size_ +
352 (enc->alpha_data_size_ & 1);
353 riff_size += CHUNK_HEADER_SIZE + padded_alpha_size;
354 }
355 // RIFF size should fit in 32-bits.
356 if (riff_size > 0xfffffffeU) {
357 return WebPEncodingSetError(pic, VP8_ENC_ERROR_FILE_TOO_BIG);
358 }
359
360 // Emit headers and partition #0
361 {
362 const uint8_t* const part0 = VP8BitWriterBuf(bw);
363 const size_t size0 = VP8BitWriterSize(bw);
364 ok = ok && PutWebPHeaders(enc, size0, vp8_size, riff_size)
365 && pic->writer(part0, size0, pic)
366 && EmitPartitionsSize(enc, pic);
367 VP8BitWriterWipeOut(bw); // will free the internal buffer.
368 }
369
370 // Token partitions
371 for (p = 0; p < enc->num_parts_; ++p) {
372 const uint8_t* const buf = VP8BitWriterBuf(enc->parts_ + p);
373 const size_t size = VP8BitWriterSize(enc->parts_ + p);
374 if (size) ok = ok && pic->writer(buf, size, pic);
375 VP8BitWriterWipeOut(enc->parts_ + p); // will free the internal buffer.
376 ok = ok && WebPReportProgress(pic, enc->percent_ + percent_per_part,
377 &enc->percent_);
378 }
379
380 // Padding byte
381 if (ok && pad) {
382 ok = PutPaddingByte(pic);
383 }
384
385 enc->coded_size_ = (int)(CHUNK_HEADER_SIZE + riff_size);
386 ok = ok && WebPReportProgress(pic, final_percent, &enc->percent_);
387 if (!ok) WebPEncodingSetError(pic, VP8_ENC_ERROR_BAD_WRITE);
388 return ok;
389}
390
391//------------------------------------------------------------------------------
392
393