1cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber/* 2cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber * Copyright (C) 2010 The Android Open Source Project 3cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber * 4cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber * Licensed under the Apache License, Version 2.0 (the "License"); 5cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber * you may not use this file except in compliance with the License. 6cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber * You may obtain a copy of the License at 7cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber * 8cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber * http://www.apache.org/licenses/LICENSE-2.0 9cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber * 10cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber * Unless required by applicable law or agreed to in writing, software 11cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber * distributed under the License is distributed on an "AS IS" BASIS, 12cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber * See the License for the specific language governing permissions and 14cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber * limitations under the License. 15cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber */ 16cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 1785f12e9b9062402d6110df3f7099707912040edbAndreas Huber#ifndef AVC_UTILS_H_ 18cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 1985f12e9b9062402d6110df3f7099707912040edbAndreas Huber#define AVC_UTILS_H_ 20cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 2185f12e9b9062402d6110df3f7099707912040edbAndreas Huber#include <media/stagefright/foundation/ABuffer.h> 2295d7e994c73c17b27c7134ba3cc3acbb5889aa87Mathias Agopian#include <utils/Errors.h> 23cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 24cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Hubernamespace android { 25cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 2686a58cabe322b746ab0eb17f0cade34ed68ba60eBernhard Rosenkränzerclass ABitReader; 276a63a939601645404fd98f58c19cc38ca818d99eAndreas Huber 28bfd41f33c77c66ead48ee378e4ea4b7bfa5fca1fAndreas Huberenum { 29bfd41f33c77c66ead48ee378e4ea4b7bfa5fca1fAndreas Huber kAVCProfileBaseline = 0x42, 30bfd41f33c77c66ead48ee378e4ea4b7bfa5fca1fAndreas Huber kAVCProfileMain = 0x4d, 31bfd41f33c77c66ead48ee378e4ea4b7bfa5fca1fAndreas Huber kAVCProfileExtended = 0x58, 32bfd41f33c77c66ead48ee378e4ea4b7bfa5fca1fAndreas Huber kAVCProfileHigh = 0x64, 33bfd41f33c77c66ead48ee378e4ea4b7bfa5fca1fAndreas Huber kAVCProfileHigh10 = 0x6e, 34bfd41f33c77c66ead48ee378e4ea4b7bfa5fca1fAndreas Huber kAVCProfileHigh422 = 0x7a, 35bfd41f33c77c66ead48ee378e4ea4b7bfa5fca1fAndreas Huber kAVCProfileHigh444 = 0xf4, 36bfd41f33c77c66ead48ee378e4ea4b7bfa5fca1fAndreas Huber kAVCProfileCAVLC444Intra = 0x2c 37bfd41f33c77c66ead48ee378e4ea4b7bfa5fca1fAndreas Huber}; 38bfd41f33c77c66ead48ee378e4ea4b7bfa5fca1fAndreas Huber 3978acc89eae9789366b6384707861f4563addf2d3Chong Zhangstruct NALPosition { 400b8dec713aee5b6fde8c4aedee91674795004725Marco Nelissen uint32_t nalOffset; 410b8dec713aee5b6fde8c4aedee91674795004725Marco Nelissen uint32_t nalSize; 4278acc89eae9789366b6384707861f4563addf2d3Chong Zhang}; 4378acc89eae9789366b6384707861f4563addf2d3Chong Zhang 44e9810fae2ca6620337b75a787786cc0ea52adecdAndreas Huber// Optionally returns sample aspect ratio as well. 4585f12e9b9062402d6110df3f7099707912040edbAndreas Hubervoid FindAVCDimensions( 46e9810fae2ca6620337b75a787786cc0ea52adecdAndreas Huber const sp<ABuffer> &seqParamSet, 47e9810fae2ca6620337b75a787786cc0ea52adecdAndreas Huber int32_t *width, int32_t *height, 48e9810fae2ca6620337b75a787786cc0ea52adecdAndreas Huber int32_t *sarWidth = NULL, int32_t *sarHeight = NULL); 49cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 502f4555e2080b7bd9933924840e70a1d4fc87ecb2Lajos Molnar// Gets and returns an unsigned exp-golomb (ue) value from a bit reader |br|. Aborts if the value 512f4555e2080b7bd9933924840e70a1d4fc87ecb2Lajos Molnar// is more than 64 bits long (>=0xFFFF (!)) or the bit reader overflows. 526a63a939601645404fd98f58c19cc38ca818d99eAndreas Huberunsigned parseUE(ABitReader *br); 536a63a939601645404fd98f58c19cc38ca818d99eAndreas Huber 542f4555e2080b7bd9933924840e70a1d4fc87ecb2Lajos Molnar// Gets and returns a signed exp-golomb (se) value from a bit reader |br|. Aborts if the value is 552f4555e2080b7bd9933924840e70a1d4fc87ecb2Lajos Molnar// more than 64 bits long (>0x7FFF || <-0x7FFF (!)) or the bit reader overflows. 562f4555e2080b7bd9933924840e70a1d4fc87ecb2Lajos Molnarsigned parseSE(ABitReader *br); 572f4555e2080b7bd9933924840e70a1d4fc87ecb2Lajos Molnar 582f4555e2080b7bd9933924840e70a1d4fc87ecb2Lajos Molnar// Gets an unsigned exp-golomb (ue) value from a bit reader |br|, and returns it if it was 592f4555e2080b7bd9933924840e70a1d4fc87ecb2Lajos Molnar// successful. Returns |fallback| if it was unsuccessful. Note: if the value was longer that 64 602f4555e2080b7bd9933924840e70a1d4fc87ecb2Lajos Molnar// bits, it reads past the value and still returns |fallback|. 612f4555e2080b7bd9933924840e70a1d4fc87ecb2Lajos Molnarunsigned parseUEWithFallback(ABitReader *br, unsigned fallback); 622f4555e2080b7bd9933924840e70a1d4fc87ecb2Lajos Molnar 632f4555e2080b7bd9933924840e70a1d4fc87ecb2Lajos Molnar// Gets a signed exp-golomb (se) value from a bit reader |br|, and returns it if it was successful. 642f4555e2080b7bd9933924840e70a1d4fc87ecb2Lajos Molnar// Returns |fallback| if it was unsuccessful. Note: if the value was longer that 64 bits, it reads 652f4555e2080b7bd9933924840e70a1d4fc87ecb2Lajos Molnar// past the value and still returns |fallback|. 662f4555e2080b7bd9933924840e70a1d4fc87ecb2Lajos Molnarsigned parseSEWithFallback(ABitReader *br, signed fallback); 672f4555e2080b7bd9933924840e70a1d4fc87ecb2Lajos Molnar 682f4555e2080b7bd9933924840e70a1d4fc87ecb2Lajos Molnar// Skips an unsigned exp-golomb (ue) value from bit reader |br|. 692f4555e2080b7bd9933924840e70a1d4fc87ecb2Lajos Molnarinline void skipUE(ABitReader *br) { 702f4555e2080b7bd9933924840e70a1d4fc87ecb2Lajos Molnar (void)parseUEWithFallback(br, 0U); 712f4555e2080b7bd9933924840e70a1d4fc87ecb2Lajos Molnar} 722f4555e2080b7bd9933924840e70a1d4fc87ecb2Lajos Molnar 732f4555e2080b7bd9933924840e70a1d4fc87ecb2Lajos Molnar// Skips a signed exp-golomb (se) value from bit reader |br|. 742f4555e2080b7bd9933924840e70a1d4fc87ecb2Lajos Molnarinline void skipSE(ABitReader *br) { 752f4555e2080b7bd9933924840e70a1d4fc87ecb2Lajos Molnar (void)parseSEWithFallback(br, 0); 762f4555e2080b7bd9933924840e70a1d4fc87ecb2Lajos Molnar} 772f4555e2080b7bd9933924840e70a1d4fc87ecb2Lajos Molnar 78c68a48c474f609df3eeb7d9738675d6ac8835e0aAndreas Huberstatus_t getNextNALUnit( 79c68a48c474f609df3eeb7d9738675d6ac8835e0aAndreas Huber const uint8_t **_data, size_t *_size, 80c68a48c474f609df3eeb7d9738675d6ac8835e0aAndreas Huber const uint8_t **nalStart, size_t *nalSize, 81c68a48c474f609df3eeb7d9738675d6ac8835e0aAndreas Huber bool startCodeFollows = false); 82c68a48c474f609df3eeb7d9738675d6ac8835e0aAndreas Huber 837dc218e17800cc3ac95551e3280820f407adf7b5Dongwon Kangsp<ABuffer> MakeAVCCodecSpecificData( 847dc218e17800cc3ac95551e3280820f407adf7b5Dongwon Kang const sp<ABuffer> &accessUnit, int32_t *width, int32_t *height, 857dc218e17800cc3ac95551e3280820f407adf7b5Dongwon Kang int32_t *sarWidth = nullptr, int32_t *sarHeight = nullptr); 86c68a48c474f609df3eeb7d9738675d6ac8835e0aAndreas Huber 87d91dc5a0602f54fc0d4d2187f37b5b8169bb62c3Dongwon Kangbool IsIDR(const uint8_t *data, size_t size); 883fe62150fa3dd6d25cb84aad80bc9e27ddd16c45Andreas Huberbool IsAVCReferenceFrame(const sp<ABuffer> &accessUnit); 894dbff11975e737482537e1636051690188f3fbc4Praveen Chavanuint32_t FindAVCLayerId(const uint8_t *data, size_t size); 90c68a48c474f609df3eeb7d9738675d6ac8835e0aAndreas Huber 91bfd41f33c77c66ead48ee378e4ea4b7bfa5fca1fAndreas Huberconst char *AVCProfileToString(uint8_t profile); 92bfd41f33c77c66ead48ee378e4ea4b7bfa5fca1fAndreas Huber 93386d609dc513e838c7e7c4c46c604493ccd560beAndreas Huber// Given an MPEG4 video VOL-header chunk (starting with 0x00 0x00 0x01 0x2?) 94386d609dc513e838c7e7c4c46c604493ccd560beAndreas Huber// parse it and fill in dimensions, returns true iff successful. 95386d609dc513e838c7e7c4c46c604493ccd560beAndreas Huberbool ExtractDimensionsFromVOLHeader( 96386d609dc513e838c7e7c4c46c604493ccd560beAndreas Huber const uint8_t *data, size_t size, int32_t *width, int32_t *height); 97386d609dc513e838c7e7c4c46c604493ccd560beAndreas Huber 98386d609dc513e838c7e7c4c46c604493ccd560beAndreas Huberbool GetMPEGAudioFrameSize( 99386d609dc513e838c7e7c4c46c604493ccd560beAndreas Huber uint32_t header, size_t *frame_size, 100386d609dc513e838c7e7c4c46c604493ccd560beAndreas Huber int *out_sampling_rate = NULL, int *out_channels = NULL, 101386d609dc513e838c7e7c4c46c604493ccd560beAndreas Huber int *out_bitrate = NULL, int *out_num_samples = NULL); 102386d609dc513e838c7e7c4c46c604493ccd560beAndreas Huber 103cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} // namespace android 104cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 10585f12e9b9062402d6110df3f7099707912040edbAndreas Huber#endif // AVC_UTILS_H_ 106