Also improves the metadata and generally makes the configuration
a bit cleaner.
(cherry picked from commit ac12486714
)
pull/272/head
parent
94a4048619
commit
00179664bc
5 changed files with 585 additions and 1068 deletions
File diff suppressed because it is too large
Load Diff
@ -1,68 +0,0 @@ |
|||||||
/*
|
|
||||||
* This file is part of FFmpeg. |
|
||||||
* |
|
||||||
* FFmpeg 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 2.1 of the License, or (at your option) any later version. |
|
||||||
* |
|
||||||
* FFmpeg 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 FFmpeg; if not, write to the Free Software |
|
||||||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
|
||||||
*/ |
|
||||||
|
|
||||||
#include <stdint.h> |
|
||||||
|
|
||||||
#include "vaapi_encode_h26x.h" |
|
||||||
|
|
||||||
int ff_vaapi_encode_h26x_nal_unit_to_byte_stream(uint8_t *dst, size_t *dst_bit_len, |
|
||||||
uint8_t *src, size_t src_bit_len) |
|
||||||
{ |
|
||||||
size_t dp, sp; |
|
||||||
int zero_run = 0; |
|
||||||
size_t dst_len = *dst_bit_len / 8; |
|
||||||
size_t src_len = (src_bit_len + 7) / 8; |
|
||||||
int trailing_zeroes = src_len * 8 - src_bit_len; |
|
||||||
|
|
||||||
if (dst_len < src_len + 4) { |
|
||||||
// Definitely doesn't fit.
|
|
||||||
goto fail; |
|
||||||
} |
|
||||||
|
|
||||||
// Start code.
|
|
||||||
dst[0] = dst[1] = dst[2] = 0; |
|
||||||
dst[3] = 1; |
|
||||||
dp = 4; |
|
||||||
|
|
||||||
for (sp = 0; sp < src_len; sp++) { |
|
||||||
if (dp >= dst_len) |
|
||||||
goto fail; |
|
||||||
if (zero_run < 2) { |
|
||||||
if (src[sp] == 0) |
|
||||||
++zero_run; |
|
||||||
else |
|
||||||
zero_run = 0; |
|
||||||
} else { |
|
||||||
if ((src[sp] & ~3) == 0) { |
|
||||||
// emulation_prevention_three_byte
|
|
||||||
dst[dp++] = 3; |
|
||||||
if (dp >= dst_len) |
|
||||||
goto fail; |
|
||||||
} |
|
||||||
zero_run = src[sp] == 0; |
|
||||||
} |
|
||||||
dst[dp++] = src[sp]; |
|
||||||
} |
|
||||||
|
|
||||||
*dst_bit_len = 8 * dp - trailing_zeroes; |
|
||||||
return 0; |
|
||||||
|
|
||||||
fail: |
|
||||||
*dst_bit_len = 0; |
|
||||||
return AVERROR(ENOSPC); |
|
||||||
} |
|
@ -1,45 +0,0 @@ |
|||||||
/*
|
|
||||||
* This file is part of FFmpeg. |
|
||||||
* |
|
||||||
* FFmpeg 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 2.1 of the License, or (at your option) any later version. |
|
||||||
* |
|
||||||
* FFmpeg 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 FFmpeg; if not, write to the Free Software |
|
||||||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
|
||||||
*/ |
|
||||||
|
|
||||||
#ifndef AVCODEC_VAAPI_ENCODE_H26X_H |
|
||||||
#define AVCODEC_VAAPI_ENCODE_H26X_H |
|
||||||
|
|
||||||
#include <stddef.h> |
|
||||||
#include <stdint.h> |
|
||||||
|
|
||||||
#include "golomb.h" |
|
||||||
#include "put_bits.h" |
|
||||||
|
|
||||||
|
|
||||||
// Debug code may be interested in the name of the syntax element being
|
|
||||||
// for tracing purposes. Here, it is just discarded.
|
|
||||||
|
|
||||||
#define write_u(pbc, width, value, name) put_bits(pbc, width, value) |
|
||||||
#define write_ue(pbc, value, name) set_ue_golomb(pbc, value) |
|
||||||
#define write_se(pbc, value, name) set_se_golomb(pbc, value) |
|
||||||
|
|
||||||
#define u(width, ...) write_u(pbc, width, __VA_ARGS__) |
|
||||||
#define ue(...) write_ue(pbc, __VA_ARGS__) |
|
||||||
#define se(...) write_se(pbc, __VA_ARGS__) |
|
||||||
|
|
||||||
|
|
||||||
// Copy from src to dst, applying emulation prevention.
|
|
||||||
int ff_vaapi_encode_h26x_nal_unit_to_byte_stream(uint8_t *dst, size_t *dst_len, |
|
||||||
uint8_t *src, size_t src_len); |
|
||||||
|
|
||||||
#endif /* AVCODEC_VAAPI_ENCODE_H26X_H */ |
|
Loading…
Reference in new issue