1 /*
2 * Copyright (C) 2010 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17 #ifndef AVC_UTILS_H_
18
19 #define AVC_UTILS_H_
20
21 #include <media/MediaCodecBuffer.h>
22 #include <media/stagefright/foundation/ABuffer.h>
23 #include <utils/Errors.h>
24
25 namespace android {
26
27 class ABitReader;
28
29 enum {
30 kAVCProfileBaseline = 0x42,
31 kAVCProfileMain = 0x4d,
32 kAVCProfileExtended = 0x58,
33 kAVCProfileHigh = 0x64,
34 kAVCProfileHigh10 = 0x6e,
35 kAVCProfileHigh422 = 0x7a,
36 kAVCProfileHigh444 = 0xf4,
37 kAVCProfileCAVLC444Intra = 0x2c
38 };
39
40 struct NALPosition {
41 uint32_t nalOffset;
42 uint32_t nalSize;
43 };
44
45 // Optionally returns sample aspect ratio as well.
46 void FindAVCDimensions(
47 const sp<ABuffer> &seqParamSet,
48 int32_t *width, int32_t *height,
49 int32_t *sarWidth = NULL, int32_t *sarHeight = NULL);
50
51 // Gets and returns an unsigned exp-golomb (ue) value from a bit reader |br|. Aborts if the value
52 // is more than 64 bits long (>=0xFFFF (!)) or the bit reader overflows.
53 unsigned parseUE(ABitReader *br);
54
55 // Gets and returns a signed exp-golomb (se) value from a bit reader |br|. Aborts if the value is
56 // more than 64 bits long (>0x7FFF || <-0x7FFF (!)) or the bit reader overflows.
57 signed parseSE(ABitReader *br);
58
59 // Gets an unsigned exp-golomb (ue) value from a bit reader |br|, and returns it if it was
60 // successful. Returns |fallback| if it was unsuccessful. Note: if the value was longer that 64
61 // bits, it reads past the value and still returns |fallback|.
62 unsigned parseUEWithFallback(ABitReader *br, unsigned fallback);
63
64 // Gets a signed exp-golomb (se) value from a bit reader |br|, and returns it if it was successful.
65 // Returns |fallback| if it was unsuccessful. Note: if the value was longer that 64 bits, it reads
66 // past the value and still returns |fallback|.
67 signed parseSEWithFallback(ABitReader *br, signed fallback);
68
69 // Skips an unsigned exp-golomb (ue) value from bit reader |br|.
skipUE(ABitReader * br)70 inline void skipUE(ABitReader *br) {
71 (void)parseUEWithFallback(br, 0U);
72 }
73
74 // Skips a signed exp-golomb (se) value from bit reader |br|.
skipSE(ABitReader * br)75 inline void skipSE(ABitReader *br) {
76 (void)parseSEWithFallback(br, 0);
77 }
78
79 status_t getNextNALUnit(
80 const uint8_t **_data, size_t *_size,
81 const uint8_t **nalStart, size_t *nalSize,
82 bool startCodeFollows = false);
83
84 class MetaData;
85 sp<MetaData> MakeAVCCodecSpecificData(const sp<ABuffer> &accessUnit);
86
87 bool IsIDR(const sp<ABuffer> &accessUnit);
88 bool IsIDR(const sp<MediaCodecBuffer> &accessUnit);
89 bool IsAVCReferenceFrame(const sp<ABuffer> &accessUnit);
90 uint32_t FindAVCLayerId(const uint8_t *data, size_t size);
91
92 const char *AVCProfileToString(uint8_t profile);
93
94 sp<MetaData> MakeAACCodecSpecificData(
95 unsigned profile, unsigned sampling_freq_index,
96 unsigned channel_configuration);
97
98 // Given an MPEG4 video VOL-header chunk (starting with 0x00 0x00 0x01 0x2?)
99 // parse it and fill in dimensions, returns true iff successful.
100 bool ExtractDimensionsFromVOLHeader(
101 const uint8_t *data, size_t size, int32_t *width, int32_t *height);
102
103 bool GetMPEGAudioFrameSize(
104 uint32_t header, size_t *frame_size,
105 int *out_sampling_rate = NULL, int *out_channels = NULL,
106 int *out_bitrate = NULL, int *out_num_samples = NULL);
107
108 } // namespace android
109
110 #endif // AVC_UTILS_H_
111