FFmpeg
atsc_a53.c
Go to the documentation of this file.
1 /*
2  * This file is part of FFmpeg.
3  *
4  * FFmpeg is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU Lesser General Public
6  * License as published by the Free Software Foundation; either
7  * version 2.1 of the License, or (at your option) any later version.
8  *
9  * FFmpeg is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12  * Lesser General Public License for more details.
13  *
14  * You should have received a copy of the GNU Lesser General Public
15  * License along with FFmpeg; if not, write to the Free Software
16  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17  */
18 
19 #include <stddef.h>
20 #include <stdint.h>
21 
22 #include "atsc_a53.h"
23 #include "get_bits.h"
24 
25 int ff_alloc_a53_sei(const AVFrame *frame, size_t prefix_len,
26  void **data, size_t *sei_size)
27 {
28  AVFrameSideData *side_data = NULL;
29  uint8_t *sei_data;
30 
31  if (frame)
33 
34  if (!side_data) {
35  *data = NULL;
36  return 0;
37  }
38 
39  *sei_size = side_data->size + 11;
40  *data = av_mallocz(*sei_size + prefix_len);
41  if (!*data)
42  return AVERROR(ENOMEM);
43  sei_data = (uint8_t*)*data + prefix_len;
44 
45  // country code
46  sei_data[0] = 181;
47  sei_data[1] = 0;
48  sei_data[2] = 49;
49 
50  /**
51  * 'GA94' is standard in North America for ATSC, but hard coding
52  * this style may not be the right thing to do -- other formats
53  * do exist. This information is not available in the side_data
54  * so we are going with this right now.
55  */
56  AV_WL32(sei_data + 3, MKTAG('G', 'A', '9', '4'));
57  sei_data[7] = 3;
58  sei_data[8] = ((side_data->size/3) & 0x1f) | 0x40;
59  sei_data[9] = 0;
60 
61  memcpy(sei_data + 10, side_data->data, side_data->size);
62 
63  sei_data[side_data->size+10] = 255;
64 
65  return 0;
66 }
67 
68 int ff_parse_a53_cc(AVBufferRef **pbuf, const uint8_t *data, int size)
69 {
70  AVBufferRef *buf = *pbuf;
71  GetBitContext gb;
72  size_t new_size, old_size = buf ? buf->size : 0;
73  int ret, cc_count;
74 
75  if (size < 3)
76  return AVERROR_INVALIDDATA;
77 
78  ret = init_get_bits8(&gb, data, size);
79  if (ret < 0)
80  return ret;
81 
82  if (get_bits(&gb, 8) != 0x3) // user_data_type_code
83  return 0;
84 
85  skip_bits(&gb, 1); // reserved
86  if (!get_bits(&gb, 1)) // process_cc_data_flag
87  return 0;
88 
89  skip_bits(&gb, 1); // zero bit
90  cc_count = get_bits(&gb, 5);
91  if (!cc_count)
92  return 0;
93 
94  skip_bits(&gb, 8); // reserved
95 
96  /* 3 bytes per CC plus one byte marker_bits at the end */
97  if (cc_count * 3 >= (get_bits_left(&gb) >> 3))
98  return AVERROR_INVALIDDATA;
99 
100  new_size = (old_size + cc_count * 3);
101 
102  if (new_size > INT_MAX)
103  return AVERROR_INVALIDDATA;
104 
105  /* Allow merging of the cc data from two fields. */
106  ret = av_buffer_realloc(pbuf, new_size);
107  if (ret < 0)
108  return ret;
109 
110  buf = *pbuf;
111  /* Use of av_buffer_realloc assumes buffer is writeable */
112  for (int i = 0; i < cc_count; i++) {
113  buf->data[old_size++] = get_bits(&gb, 8);
114  buf->data[old_size++] = get_bits(&gb, 8);
115  buf->data[old_size++] = get_bits(&gb, 8);
116  }
117 
118  return cc_count;
119 }
ff_alloc_a53_sei
int ff_alloc_a53_sei(const AVFrame *frame, size_t prefix_len, void **data, size_t *sei_size)
Check AVFrame for A53 side data and allocate and fill SEI message with A53 info.
Definition: atsc_a53.c:25
get_bits_left
static int get_bits_left(GetBitContext *gb)
Definition: get_bits.h:694
AVERROR
Filter the word “frame” indicates either a video frame or a group of audio as stored in an AVFrame structure Format for each input and each output the list of supported formats For video that means pixel format For audio that means channel sample they are references to shared objects When the negotiation mechanism computes the intersection of the formats supported at each end of a all references to both lists are replaced with a reference to the intersection And when a single format is eventually chosen for a link amongst the remaining all references to the list are updated That means that if a filter requires that its input and output have the same format amongst a supported all it has to do is use a reference to the same list of formats query_formats can leave some formats unset and return AVERROR(EAGAIN) to cause the negotiation mechanism toagain later. That can be used by filters with complex requirements to use the format negotiated on one link to set the formats supported on another. Frame references ownership and permissions
AV_WL32
#define AV_WL32(p, v)
Definition: intreadwrite.h:424
av_frame_get_side_data
AVFrameSideData * av_frame_get_side_data(const AVFrame *frame, enum AVFrameSideDataType type)
Definition: frame.c:716
AVBufferRef::data
uint8_t * data
The data buffer.
Definition: buffer.h:90
AV_FRAME_DATA_A53_CC
@ AV_FRAME_DATA_A53_CC
ATSC A53 Part 4 Closed Captions.
Definition: frame.h:59
AVFrame
This structure describes decoded (raw) audio or video data.
Definition: frame.h:340
data
const char data[16]
Definition: mxf.c:148
skip_bits
static void skip_bits(GetBitContext *s, int n)
Definition: get_bits.h:381
get_bits
static unsigned int get_bits(GetBitContext *s, int n)
Read 1-25 bits.
Definition: get_bits.h:335
GetBitContext
Definition: get_bits.h:108
AVFrameSideData::size
size_t size
Definition: frame.h:249
init_get_bits8
static int init_get_bits8(GetBitContext *s, const uint8_t *buffer, int byte_size)
Initialize GetBitContext.
Definition: get_bits.h:545
get_bits.h
frame
static AVFrame * frame
Definition: demux_decode.c:54
ff_parse_a53_cc
int ff_parse_a53_cc(AVBufferRef **pbuf, const uint8_t *data, int size)
Parse a data array for ATSC A53 Part 4 Closed Captions and store them in an AVBufferRef.
Definition: atsc_a53.c:68
NULL
#define NULL
Definition: coverity.c:32
size
int size
Definition: twinvq_data.h:10344
AVFrameSideData::data
uint8_t * data
Definition: frame.h:248
AVBufferRef::size
size_t size
Size of data in bytes.
Definition: buffer.h:94
i
#define i(width, name, range_min, range_max)
Definition: cbs_h2645.c:255
av_mallocz
void * av_mallocz(size_t size)
Allocate a memory block with alignment suitable for all memory accesses (including vectors if availab...
Definition: mem.c:254
ret
ret
Definition: filter_design.txt:187
atsc_a53.h
av_buffer_realloc
int av_buffer_realloc(AVBufferRef **pbuf, size_t size)
Reallocate a given buffer.
Definition: buffer.c:183
AVBufferRef
A reference to a data buffer.
Definition: buffer.h:82
AVFrameSideData
Structure to hold side data for an AVFrame.
Definition: frame.h:246
AVERROR_INVALIDDATA
#define AVERROR_INVALIDDATA
Invalid data found when processing input.
Definition: error.h:61
MKTAG
#define MKTAG(a, b, c, d)
Definition: macros.h:55