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 25namespace android { 26 27class ABitReader; 28 29enum { 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 40struct NALPosition { 41 uint32_t nalOffset; 42 uint32_t nalSize; 43}; 44 45// Optionally returns sample aspect ratio as well. 46void 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. 53unsigned 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. 57signed 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|. 62unsigned 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|. 67signed parseSEWithFallback(ABitReader *br, signed fallback); 68 69// Skips an unsigned exp-golomb (ue) value from bit reader |br|. 70inline void skipUE(ABitReader *br) { 71 (void)parseUEWithFallback(br, 0U); 72} 73 74// Skips a signed exp-golomb (se) value from bit reader |br|. 75inline void skipSE(ABitReader *br) { 76 (void)parseSEWithFallback(br, 0); 77} 78 79status_t getNextNALUnit( 80 const uint8_t **_data, size_t *_size, 81 const uint8_t **nalStart, size_t *nalSize, 82 bool startCodeFollows = false); 83 84class MetaData; 85sp<MetaData> MakeAVCCodecSpecificData(const sp<ABuffer> &accessUnit); 86 87bool IsIDR(const sp<ABuffer> &accessUnit); 88bool IsIDR(const sp<MediaCodecBuffer> &accessUnit); 89bool IsAVCReferenceFrame(const sp<ABuffer> &accessUnit); 90uint32_t FindAVCLayerId(const uint8_t *data, size_t size); 91 92const char *AVCProfileToString(uint8_t profile); 93 94sp<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. 100bool ExtractDimensionsFromVOLHeader( 101 const uint8_t *data, size_t size, int32_t *width, int32_t *height); 102 103bool 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