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/stagefright/foundation/ABuffer.h>
22 #include <utils/Errors.h>
23 
24 namespace android {
25 
26 class ABitReader;
27 
28 enum {
29     kAVCProfileBaseline      = 0x42,
30     kAVCProfileMain          = 0x4d,
31     kAVCProfileExtended      = 0x58,
32     kAVCProfileHigh          = 0x64,
33     kAVCProfileHigh10        = 0x6e,
34     kAVCProfileHigh422       = 0x7a,
35     kAVCProfileHigh444       = 0xf4,
36     kAVCProfileCAVLC444Intra = 0x2c
37 };
38 
39 struct NALPosition {
40     size_t nalOffset;
41     size_t nalSize;
42 };
43 
44 // Optionally returns sample aspect ratio as well.
45 void FindAVCDimensions(
46         const sp<ABuffer> &seqParamSet,
47         int32_t *width, int32_t *height,
48         int32_t *sarWidth = NULL, int32_t *sarHeight = NULL);
49 
50 unsigned parseUE(ABitReader *br);
51 
52 status_t getNextNALUnit(
53         const uint8_t **_data, size_t *_size,
54         const uint8_t **nalStart, size_t *nalSize,
55         bool startCodeFollows = false);
56 
57 class MetaData;
58 sp<MetaData> MakeAVCCodecSpecificData(const sp<ABuffer> &accessUnit);
59 
60 bool IsIDR(const sp<ABuffer> &accessUnit);
61 bool IsAVCReferenceFrame(const sp<ABuffer> &accessUnit);
62 
63 const char *AVCProfileToString(uint8_t profile);
64 
65 sp<MetaData> MakeAACCodecSpecificData(
66         unsigned profile, unsigned sampling_freq_index,
67         unsigned channel_configuration);
68 
69 // Given an MPEG4 video VOL-header chunk (starting with 0x00 0x00 0x01 0x2?)
70 // parse it and fill in dimensions, returns true iff successful.
71 bool ExtractDimensionsFromVOLHeader(
72         const uint8_t *data, size_t size, int32_t *width, int32_t *height);
73 
74 bool GetMPEGAudioFrameSize(
75         uint32_t header, size_t *frame_size,
76         int *out_sampling_rate = NULL, int *out_channels = NULL,
77         int *out_bitrate = NULL, int *out_num_samples = NULL);
78 
79 }  // namespace android
80 
81 #endif  // AVC_UTILS_H_
82