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|. 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|. 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