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
24namespace android {
25
26struct ABitReader;
27
28enum {
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// Optionally returns sample aspect ratio as well.
40void FindAVCDimensions(
41        const sp<ABuffer> &seqParamSet,
42        int32_t *width, int32_t *height,
43        int32_t *sarWidth = NULL, int32_t *sarHeight = NULL);
44
45unsigned parseUE(ABitReader *br);
46
47status_t getNextNALUnit(
48        const uint8_t **_data, size_t *_size,
49        const uint8_t **nalStart, size_t *nalSize,
50        bool startCodeFollows = false);
51
52struct MetaData;
53sp<MetaData> MakeAVCCodecSpecificData(const sp<ABuffer> &accessUnit);
54
55bool IsIDR(const sp<ABuffer> &accessUnit);
56bool IsAVCReferenceFrame(const sp<ABuffer> &accessUnit);
57
58const char *AVCProfileToString(uint8_t profile);
59
60sp<MetaData> MakeAACCodecSpecificData(
61        unsigned profile, unsigned sampling_freq_index,
62        unsigned channel_configuration);
63
64// Given an MPEG4 video VOL-header chunk (starting with 0x00 0x00 0x01 0x2?)
65// parse it and fill in dimensions, returns true iff successful.
66bool ExtractDimensionsFromVOLHeader(
67        const uint8_t *data, size_t size, int32_t *width, int32_t *height);
68
69bool GetMPEGAudioFrameSize(
70        uint32_t header, size_t *frame_size,
71        int *out_sampling_rate = NULL, int *out_channels = NULL,
72        int *out_bitrate = NULL, int *out_num_samples = NULL);
73
74}  // namespace android
75
76#endif  // AVC_UTILS_H_
77