123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062 |
- //--=========================================================================--
- // This file is a part of QC Tool project
- //-----------------------------------------------------------------------------
- //
- // This confidential and proprietary software may be used only
- // as authorized by a licensing agreement from Chips&Media Inc.
- // In the event of publication, the following notice is applicable:
- //
- // (C) COPYRIGHT 2004 - 2011 CHIPS&MEDIA INC.
- // ALL RIGHTS RESERVED
- //
- // The entire notice above must be reproduced on all authorized
- // copies.
- //
- //--=========================================================================--
- #include "main_helper.h"
- #ifdef SUPPORT_FFMPEG_DEMUX
- #include "theora_parser.h"
- #include <libavformat/avformat.h>
- #define RCV_V2
- #define IS_VP9_SUPERFRAME(__header) ((__header & 0xe0) == 0xc0)
- #define MAX_VP9_SUPERFRAME_INDEX_SIZE 34
- #define VP9_MAX_SUBFRAMES 8
- typedef struct VP9Superframe {
- Uint32 nframes;
- Uint32 frameSize[VP9_MAX_SUBFRAMES];
- void* frames[VP9_MAX_SUBFRAMES]; /* A superframe has multiple frames up to 8 frames. */
- Uint32 currentIndex;
- } VP9Superframe;
- typedef struct ReaderContext {
- AVFormatContext* avContext;
- BOOL isFirstPacket;
- Uint32 videoIndex;
- Uint32 standard;
- Uint32 mp4ClassId;
- tho_parser_t* theora;
- Uint8* tempBuffer;
- Uint32 tempRdPtr;
- Uint32 tempWrPtr;
- Uint32 seqWidth;
- Uint32 seqHeight;
- VP9Superframe superframe;
- } ReaderContext;
- static Uint32 u_bytes(
- Uint8* data,
- Uint32 len
- )
- {
- Uint32 i;
- Uint32 val=0;
- for (i=0; i<len; i++) {
- val |= data[i] << (i*8);
- }
- return val;
- }
- static BOOL VP9ParseSuperframe(
- void* chunk,
- Uint32 size,
- VP9Superframe* superframe
- )
- {
- Uint32 startPos;
- Uint32 frameSizeLength = 0;
- Uint32 numFrames, totalSubframeSize = 0;
- Uint32 i;
- Uint8* pData = NULL, superframeMarker;
- startPos = size - 1;
- pData = (Uint8*)chunk;
- pData = &pData[startPos];
- superframeMarker = *pData;
- frameSizeLength = ((*pData>>3) & 0x03) + 1;
- numFrames = (*pData&0x07) + 1;
- pData -= frameSizeLength * numFrames + 1;
- /* Check first SUPERFRAME_MARKER */
- if (*pData != superframeMarker) {
- VLOG(ERR, "INVALID POST SUPERFRAME_MARKER\n");
- return FALSE;
- }
- pData++;
- for (i=0; i<numFrames; i++) {
- superframe->frameSize[i] = u_bytes(pData, frameSizeLength);
- pData += frameSizeLength;
- }
- /* Check size */
- for (i=0; i<numFrames; i++) {
- totalSubframeSize += superframe->frameSize[i];
- }
- if (totalSubframeSize >= size) {
- VLOG(ERR, "TOTAL SIZE OF SUBFRAMES IS BIGGER THAN CHUNK SIZE\n");
- return FALSE;
- }
- pData = (Uint8*)chunk;
- for (i=0; i<numFrames; i++) {
- superframe->frames[i] = (void*)osal_malloc(superframe->frameSize[i]);
- osal_memcpy(superframe->frames[i], (void*)pData, superframe->frameSize[i]);
- pData += superframe->frameSize[i];
- }
- superframe->currentIndex = 0;
- superframe->nframes = numFrames;
- return TRUE;
- }
- static Int32 BuildSeqHeader(
- Uint8* pbHeader,
- const CodStd codStd,
- const AVStream* st,
- Int32* sizelength
- )
- {
- /*lint -save -e438 */
- AVCodecContext* avc = st->codec;
- Uint8* pbMetaData = avc->extradata;
- Int32 nMetaData = avc->extradata_size;
- Uint8* p = pbMetaData;
- Uint8 *a = p + 4 - ((long) p & 3);
- Uint8* t = pbHeader;
- Int32 size;
- Int32 fourcc;
- Int32 sps, pps, i, nal;
- Int32 frameRate = 0;
- fourcc = avc->codec_tag;
- if (!fourcc)
- fourcc = ConvCodecIdToFourcc(avc->codec_id);
- if (st->avg_frame_rate.den && st->avg_frame_rate.num)
- frameRate = (Int32)((double)st->avg_frame_rate.num/(double)st->avg_frame_rate.den);
- size = 0;
- *sizelength = 4; // default size length(in bytes) = 4
- if (codStd == STD_AVC || codStd == STD_AVS) {
- if (nMetaData > 1 && pbMetaData && pbMetaData[0] == 0x01) {
- // check mov/mo4 file format stream
- p += 4;
- *sizelength = (*p++ & 0x3) + 1;
- sps = (*p & 0x1f); // Number of sps
- p++;
- for (i = 0; i < sps; i++) {
- nal = (*p << 8) + *(p + 1) + 2;
- PUT_BYTE(t, 0x00);
- PUT_BYTE(t, 0x00);
- PUT_BYTE(t, 0x00);
- PUT_BYTE(t, 0x01);
- PUT_BUFFER(t, p+2, nal-2);
- p += nal;
- size += (nal - 2 + 4); // 4 => length of start code to be inserted
- }
- pps = *(p++); // number of pps
- for (i = 0; i < pps; i++)
- {
- nal = (*p << 8) + *(p + 1) + 2;
- PUT_BYTE(t, 0x00);
- PUT_BYTE(t, 0x00);
- PUT_BYTE(t, 0x00);
- PUT_BYTE(t, 0x01);
- PUT_BUFFER(t, p+2, nal-2);
- p += nal;
- size += (nal - 2 + 4); // 4 => length of start code to be inserted
- }
- }
- else if(nMetaData > 3) {
- size = -1;// return to meaning of invalid stream data;
- for (; p < a; p++) {
- if (p[0] == 0 && p[1] == 0 && p[2] == 1) {
- // find startcode
- size = avc->extradata_size;
- PUT_BUFFER(pbHeader, pbMetaData, size);
- break;
- }
- }
- }
- }
- else if (codStd == STD_HEVC) {
- if (nMetaData > 1 && pbMetaData && pbMetaData[0] == 0x01) {
- static const Uint8 nalu_header[4] = { 0, 0, 0, 1 };
- Int32 numOfArrays = 0;
- Uint16 numNalus = 0;
- Uint16 nalUnitLength = 0;
- Uint32 offset = 0;
- p += 21;
- *sizelength = (*p++ & 0x3) + 1;
- numOfArrays = *p++;
- while(numOfArrays--) {
- p++; // NAL type
- numNalus = (*p << 8) + *(p + 1);
- p+=2;
- for(i = 0;i < numNalus;i++)
- {
- nalUnitLength = (*p << 8) + *(p + 1);
- p+=2;
- //if(i == 0)
- {
- osal_memcpy(pbHeader + offset, nalu_header, 4);
- offset += 4;
- osal_memcpy(pbHeader + offset, p, nalUnitLength);
- offset += nalUnitLength;
- }
- p += nalUnitLength;
- }
- }
- size = offset;
- }
- else if(nMetaData > 3)
- {
- size = -1;// return to meaning of invalid stream data;
- for (; p < a; p++)
- {
- if (p[0] == 0 && p[1] == 0 && p[2] == 1) // find startcode
- {
- size = avc->extradata_size;
- PUT_BUFFER(pbHeader, pbMetaData, size);
- break;
- }
- }
- }
- }
- else if (codStd == STD_VC1)
- {
- if (!fourcc)
- return -1;
- if (fourcc == MKTAG('W', 'V', 'C', '1') || fourcc == MKTAG('W', 'M', 'V', 'A')) //VC AP
- {
- size = nMetaData;
- PUT_BUFFER(pbHeader, pbMetaData, size);
- //if there is no seq startcode in pbMetatData. VPU will be failed at seq_init stage.
- }
- else
- {
- #ifdef RCV_V2
- PUT_LE32(pbHeader, ((0xC5 << 24)|0));
- size += 4; //version
- PUT_LE32(pbHeader, nMetaData);
- size += 4;
- PUT_BUFFER(pbHeader, pbMetaData, nMetaData);
- size += nMetaData;
- PUT_LE32(pbHeader, avc->height);
- size += 4;
- PUT_LE32(pbHeader, avc->width);
- size += 4;
- PUT_LE32(pbHeader, 12);
- size += 4;
- PUT_LE32(pbHeader, 2 << 29 | 1 << 28 | 0x80 << 24 | 1 << 0);
- size += 4; // STRUCT_B_FRIST (LEVEL:3|CBR:1:RESERVE:4:HRD_BUFFER|24)
- PUT_LE32(pbHeader, avc->bit_rate);
- size += 4; // hrd_rate
- PUT_LE32(pbHeader, frameRate);
- size += 4; // frameRate
- #else //RCV_V1
- PUT_LE32(pbHeader, (0x85 << 24) | 0x00);
- size += 4; //frames count will be here
- PUT_LE32(pbHeader, nMetaData);
- size += 4;
- PUT_BUFFER(pbHeader, pbMetaData, nMetaData);
- size += nMetaData;
- PUT_LE32(pbHeader, avc->height);
- size += 4;
- PUT_LE32(pbHeader, avc->width);
- size += 4;
- #endif
- }
- }
- else if (codStd == STD_RV)
- {
- Int32 st_size =0;
- if (!fourcc)
- return -1;
- if (fourcc != MKTAG('R','V','3','0') && fourcc != MKTAG('R','V','4','0'))
- return -1;
- size = 26 + nMetaData;
- PUT_BE32(pbHeader, size); //Length
- PUT_LE32(pbHeader, MKTAG('V', 'I', 'D', 'O')); //MOFTag
- PUT_LE32(pbHeader, fourcc); //SubMOFTagl
- PUT_BE16(pbHeader, avc->width);
- PUT_BE16(pbHeader, avc->height);
- PUT_BE16(pbHeader, 0x0c); //BitCount;
- PUT_BE16(pbHeader, 0x00); //PadWidth;
- PUT_BE16(pbHeader, 0x00); //PadHeight;
- PUT_LE32(pbHeader, frameRate);
- PUT_BUFFER(pbHeader, pbMetaData, nMetaData); //OpaqueDatata
- size += st_size; //add for startcode pattern.
- }
- else if (codStd == STD_DIV3) {
- // not implemented yet
- if (!nMetaData) {
- PUT_LE32(pbHeader, MKTAG('C', 'N', 'M', 'V')); //signature 'CNMV'
- PUT_LE16(pbHeader, 0x00); //version
- PUT_LE16(pbHeader, 0x20); //length of header in bytes
- PUT_LE32(pbHeader, MKTAG('D', 'I', 'V', '3')); //codec FourCC
- PUT_LE16(pbHeader, avc->width); //width
- PUT_LE16(pbHeader, avc->height); //height
- PUT_LE32(pbHeader, st->avg_frame_rate.num); //frame rate
- PUT_LE32(pbHeader, st->avg_frame_rate.den); //time scale(?)
- PUT_LE32(pbHeader, st->nb_index_entries); //number of frames in file
- PUT_LE32(pbHeader, 0); //unused
- size += 32;
- return size;
- }
- PUT_BE32(pbHeader, nMetaData);
- size += 4;
- PUT_BUFFER(pbHeader, pbMetaData, nMetaData);
- size += nMetaData;
- }
- else if (codStd == STD_VP8) {
- PUT_LE32(pbHeader, MKTAG('D', 'K', 'I', 'F')); //signature 'DKIF'
- PUT_LE16(pbHeader, 0x00); //version
- PUT_LE16(pbHeader, 0x20); //length of header in bytes
- PUT_LE32(pbHeader, MKTAG('V', 'P', '8', '0')); //codec FourCC
- PUT_LE16(pbHeader, avc->width); //width
- PUT_LE16(pbHeader, avc->height); //height
- PUT_LE32(pbHeader, st->avg_frame_rate.num); //frame rate
- PUT_LE32(pbHeader, st->avg_frame_rate.den); //time scale(?)
- PUT_LE32(pbHeader, st->nb_index_entries); //number of frames in file
- PUT_LE32(pbHeader, 0); //unused
- size += 32;
- }
- else if (codStd == STD_VP9) {
- PUT_LE32(pbHeader, MKTAG('D', 'K', 'I', 'F')); //signature 'DKIF'
- PUT_LE16(pbHeader, 0x00); //version
- PUT_LE16(pbHeader, 0x20); //length of header in bytes
- PUT_LE32(pbHeader, MKTAG('V', 'P', '9', '0')); //codec FourCC
- PUT_LE16(pbHeader, avc->width); //width
- PUT_LE16(pbHeader, avc->height); //height
- PUT_LE32(pbHeader, st->avg_frame_rate.num); //frame rate
- PUT_LE32(pbHeader, st->avg_frame_rate.den); //time scale(?)
- PUT_LE32(pbHeader, st->nb_index_entries); //number of frames in file
- PUT_LE32(pbHeader, 0); //unused
- size += 32;
- }
- else {
- PUT_BUFFER(pbHeader, pbMetaData, nMetaData);
- size = nMetaData;
- }
- return size;
- /*lint -restore */
- }
- static Int32 BuildPicHeader(
- Uint8* pbHeader,
- const CodStd codStd,
- const AVStream* st,
- const AVPacket* pkt,
- Int32 sizelength
- )
- {
- AVCodecContext* avc = st->codec;
- Uint8* pbChunk = pkt->data;
- Int32 size = 0;
- Int32 fourcc;
- Int32 cSlice, nSlice;
- Int32 i, val, offset;
- BOOL hasStartCode = 0;
- size = 0;
- offset = 0;
- fourcc = avc->codec_tag;
- if (!fourcc)
- fourcc = ConvCodecIdToFourcc(avc->codec_id);
- if (codStd == STD_VC1) {
- if (!fourcc)
- return -1;
- if (fourcc == MKTAG('W', 'V', 'C', '1') || fourcc == MKTAG('W', 'M', 'V', 'A')) {
- if (pbChunk[0] != 0 || pbChunk[1] != 0 || pbChunk[2] != 1) {
- // check start code as prefix (0x00, 0x00, 0x01)
- pbHeader[0] = 0x00;
- pbHeader[1] = 0x00;
- pbHeader[2] = 0x01;
- pbHeader[3] = 0x0D; // replace to the correct picture header to indicate as frame
- size += 4;
- }
- }
- else {
- PUT_LE32(pbHeader, pkt->size | ((pkt->flags & AV_PKT_FLAG_KEY) ? 0x80000000 : 0));
- size += 4;
- #ifdef RCV_V2
- if (AV_NOPTS_VALUE == pkt->pts) {
- PUT_LE32(pbHeader, 0);
- }
- else {
- PUT_LE32(pbHeader, (int)((double)(pkt->pts/st->time_base.den))); // milli_sec
- }
- size += 4;
- #endif
- }
- }
- else if (codStd == STD_HEVC) {
- if(pkt->size < 5)
- return 0;
- if (!(avc->extradata_size > 1 && avc->extradata && avc->extradata[0] == 0x01))
- {
- const Uint8 *pbEnd = pbChunk + 4 - ((intptr_t)pbChunk & 3);
- for (; pbChunk < pbEnd ; pbChunk++) {
- if (pbChunk[0] == 0 && pbChunk[1] == 0 && pbChunk[2] == 1) {
- hasStartCode = TRUE;
- break;
- }
- }
- }
- if ((!hasStartCode && avc->extradata[0] == 0x01) ||
- (avc->extradata_size > 1 && avc->extradata && avc->extradata[0] == 0x01)) {
- // check sequence metadata if the stream is mov/mo4 file format.
- pbChunk = pkt->data;
- while (offset < pkt->size) {
- if(sizelength == 3) {
- nSlice = pbChunk[offset] << 16 | pbChunk[offset+1] << 8 | pbChunk[offset+2];
- pbChunk[offset] = 0x00;
- pbChunk[offset+1] = 0x00;
- pbChunk[offset+2] = 0x01;
- offset += 3;
- }
- else {// sizeLength = 4
- nSlice = pbChunk[offset] << 24 | pbChunk[offset+1] << 16 | pbChunk[offset+2] << 8 | pbChunk[offset+3];
- pbChunk[offset] = 0x00;
- pbChunk[offset+1] = 0x00;
- pbChunk[offset+2] = 0x00;
- pbChunk[offset+3] = 0x01; //replace size to startcode
- offset += 4;
- }
- switch ((pbChunk[offset]&0x7E)>>1) { /* NAL unit */
- case 39: /* PREFIX SEI */
- case 40: /* SUFFIX SEI */
- case 32: /* VPS */
- case 33: /* SPS */
- case 34: /* PPS */
- /* check next */
- break;
- }
- offset += nSlice;
- }
- }
- }
- else if (codStd == STD_RV) {
- int st_size = 0;
- if (!fourcc)
- return -1;
- if (fourcc != MKTAG('R','V','3','0') && fourcc != MKTAG('R','V','4','0')) // RV version 8, 9 , 10
- return -1;
- cSlice = pbChunk[0] + 1;
- nSlice = pkt->size - 1 - (cSlice * 8);
- size = 20 + (cSlice*8);
- PUT_BE32(pbHeader, nSlice);
- if (AV_NOPTS_VALUE == pkt->pts) {
- PUT_LE32(pbHeader, 0);
- }
- else {
- PUT_LE32(pbHeader, (int)((double)(pkt->pts/st->time_base.den))); // milli_sec
- }
- PUT_BE16(pbHeader, avc->frame_number);
- PUT_BE16(pbHeader, 0x02); //Flags
- PUT_BE32(pbHeader, 0x00); //LastPacket
- PUT_BE32(pbHeader, cSlice); //NumSegments
- offset = 1;
- for (i = 0; i < (int) cSlice; i++)
- {
- val = (pbChunk[offset+3] << 24) | (pbChunk[offset+2] << 16) | (pbChunk[offset+1] << 8) | pbChunk[offset];
- PUT_BE32(pbHeader, val); //isValid
- offset += 4;
- val = (pbChunk[offset+3] << 24) | (pbChunk[offset+2] << 16) | (pbChunk[offset+1] << 8) | pbChunk[offset];
- PUT_BE32(pbHeader, val); //Offset
- offset += 4;
- }
- size += st_size;
- }
- else if (codStd == STD_AVC) {
- if(pkt->size < 5)
- return 0;
- if (!(avc->extradata_size > 1 && avc->extradata && avc->extradata[0] == 0x01)) {
- const Uint8 *pbEnd = pbChunk + 4 - ((intptr_t)pbChunk & 3);
- for (; pbChunk < pbEnd ; pbChunk++) {
- if (pbChunk[0] == 0 && pbChunk[1] == 0 && pbChunk[2] == 1) {
- hasStartCode = 1;
- break;
- }
- }
- }
- if ((!hasStartCode && avc->extradata[0] == 0x01) ||
- (avc->extradata_size > 1 && avc->extradata && avc->extradata[0] == 0x01)) {
- // check sequence metadata if the stream is mov/mo4 file format.
- pbChunk = pkt->data;
- while (offset < pkt->size) {
- if(sizelength == 3) {
- nSlice = pbChunk[offset] << 16 | pbChunk[offset+1] << 8 | pbChunk[offset+2];
- pbChunk[offset] = 0x00;
- pbChunk[offset+1] = 0x00;
- pbChunk[offset+2] = 0x01;
- offset += 3;
- }
- else { // size length = 4
- nSlice = pbChunk[offset] << 24 | pbChunk[offset+1] << 16 | pbChunk[offset+2] << 8 | pbChunk[offset+3];
- pbChunk[offset] = 0x00;
- pbChunk[offset+1] = 0x00;
- pbChunk[offset+2] = 0x00;
- pbChunk[offset+3] = 0x01; //replace size to startcode
- offset += 4;
- }
- switch (pbChunk[offset]&0x1f) { /* NAL unit */
- case 6: /* SEI */
- case 7: /* SPS */
- case 8: /* PPS */
- case 9: /* AU */
- /* check next */
- break;
- }
- offset += nSlice;
- }
- }
- }
- else if(codStd == STD_AVS) {
- const Uint8* pbEnd;
- if(pkt->size < 5)
- return 0;
- pbEnd = pbChunk + 4 - ((intptr_t)pbChunk & 3);
- for (; pbChunk < pbEnd ; pbChunk++) {
- if (pbChunk[0] == 0 && pbChunk[1] == 0 && pbChunk[2] == 1) {
- hasStartCode = 1;
- break;
- }
- }
- if(hasStartCode == 0) {
- pbChunk = pkt->data;
- while (offset < pkt->size) {
- nSlice = pbChunk[offset] << 24 | pbChunk[offset+1] << 16 | pbChunk[offset+2] << 8 | pbChunk[offset+3];
- pbChunk[offset] = 0x00;
- pbChunk[offset+1] = 0x00;
- pbChunk[offset+2] = 0x00;
- pbChunk[offset+3] = 0x00; //replace size to startcode
- pbChunk[offset+4] = 0x01;
- offset += 4;
- switch (pbChunk[offset]&0x1f) /* NAL unit */
- {
- case 6: /* SEI */
- case 7: /* SPS */
- case 8: /* PPS */
- case 9: /* AU */
- /* check next */
- break;
- }
- offset += nSlice;
- }
- }
- }
- else if (codStd == STD_DIV3 || codStd == STD_VP8 || codStd == STD_VP9) {
- PUT_LE32(pbHeader,pkt->size);
- PUT_LE32(pbHeader,0);
- PUT_LE32(pbHeader,0);
- size += 12;
- }
- return size;
- }
- static Int32 MakeupTheoraPacket(
- tho_parser_t* theora,
- BSChunk* packet,
- AVPacket* avPacket,
- Uint32 seqSize
- )
- {
- size_t size;
- Int32 ret;
- Uint8* stream = (Uint8*)packet->data;
- ret = theora->read_frame(theora->handle, avPacket->data, avPacket->size);
- if (ret < 0) {
- VLOG(ERR, "%s:%d failed to read theora frame\n", __FUNCTION__, __LINE__);
- return -1;
- }
- size = theora_make_stream(theora->handle, stream+seqSize, PIC_RUN);
- return (size + seqSize);
- }
- void* BSFeederFrameSize_Create(
- const char* path,
- CodStd codecId,
- CodStd* retCodecId,
- Uint32* retMp4ClassId,
- Uint32* retSeqWidth,
- Uint32* retSeqHeight
- )
- {
- /*lint -esym(438, avContext) */
- ReaderContext* ffmpegReader = NULL;
- AVFormatContext* avContext = NULL;
- AVCodecContext* codec = NULL;
- AVInputFormat* fmt = NULL;
- Int32 error;
- Int32 videoIndex;
- Uint32 mp4ClassId;
- Int32 standard;
- av_register_all();
- if ((avContext=avformat_alloc_context()) == NULL) {
- return NULL;
- }
- avContext->flags |= CODEC_FLAG_TRUNCATED;
- if ((error=avformat_open_input(&avContext, path, fmt, NULL))) {
- VLOG(ERR, "%s:%d failed to av_open_input_file error(%d), %s\n",
- __FILE__, __LINE__, error, path);
- goto __failed_to_end;
- }
- if ((error=avformat_find_stream_info(avContext, NULL)) < 0) {
- VLOG(ERR, "%s:%d failed to avformat_find_stream_info. error(%d)\n",
- __FUNCTION__, __LINE__, error);
- goto __failed_to_end;
- }
- videoIndex = av_find_best_stream(avContext, AVMEDIA_TYPE_VIDEO, -1, -1, NULL, 0);
- if (videoIndex < 0) {
- VLOG(ERR, "%s:%d failed to av_find_best_stream.\n", __FUNCTION__, __LINE__);
- goto __failed_to_end;
- }
- codec = avContext->streams[videoIndex]->codec;
- standard = ConvFOURCCToCodStd(codec->codec_tag);
- if (standard == -1)
- standard = ConvCodecIdToCodStd(codec->codec_id);
- mp4ClassId = ConvFOURCCToMp4Class(codec->codec_tag);
- if (mp4ClassId == -1)
- mp4ClassId = ConvCodecIdToMp4Class(codec->codec_id);
- if (standard != STD_MPEG4) {
- mp4ClassId = 0;
- }
- if (retCodecId != NULL) {
- *retCodecId = (CodStd)standard;
- }
- if (retMp4ClassId != NULL) {
- *retMp4ClassId = mp4ClassId;
- }
- if (retSeqWidth != NULL && retSeqHeight != NULL) {
- *retSeqWidth = codec->width;
- *retSeqHeight = codec->height;
- }
- if ((ffmpegReader=(ReaderContext*)osal_malloc(sizeof(ReaderContext))) == NULL)
- goto __failed_to_end;
- ffmpegReader->standard = standard;
- ffmpegReader->mp4ClassId = mp4ClassId;
- ffmpegReader->avContext = avContext;
- ffmpegReader->videoIndex = videoIndex;
- ffmpegReader->isFirstPacket = TRUE;
- ffmpegReader->tempBuffer = NULL;
- ffmpegReader->tempRdPtr = 0;
- ffmpegReader->tempWrPtr = 0;
- ffmpegReader->theora = NULL;
- osal_memset((void*)&ffmpegReader->superframe, 0x00, sizeof(VP9Superframe));
- if (standard == STD_THO || standard == STD_VP3) {
- theora_parser_init((void**)&ffmpegReader->theora);
- fuck
- }
- return (void*)ffmpegReader;
- __failed_to_end:
- if (avContext) {
- avformat_free_context(avContext);
- avContext = NULL;
- }
- if (ffmpegReader) {
- if (ffmpegReader->theora)
- ffmpegReader->theora->close(ffmpegReader->theora);
- osal_free(ffmpegReader);
- }
- return NULL;
- /*lint +esym(438, avContext) */
- }
- BOOL BSFeederFrameSize_Destroy(
- void* feeder
- )
- {
- ReaderContext* ctx = (ReaderContext*)feeder;
- Uint32 i;
- if (ctx == NULL) {
- VLOG(ERR, "%s:%d Invalid handle\n", __FUNCTION__, __LINE__);
- return FALSE;
- }
- if (ctx->avContext)
- avformat_close_input(&ctx->avContext);
- if (ctx->theora)
- ctx->theora->close(ctx->theora);
- for (i=0; i<ctx->superframe.nframes; i++) {
- if (ctx->superframe.frames[i] != NULL) {
- free(ctx->superframe.frames[i]);
- }
- }
- osal_free(ctx);
- return TRUE;
- }
- Int32 BSFeederFrameSize_Act(
- void* feeder,
- BSChunk* packet
- )
- {
- ReaderContext* ffmpegReader = (ReaderContext*)feeder;
- AVFormatContext* avFormatContext = ffmpegReader->avContext;
- AVPacket avpacket;
- Int32 error;
- Uint8* seqHeader = NULL;
- Uint8* picHeader = NULL;
- Uint8* ptr;
- Int32 seqHeaderSize;
- Int32 picHeaderSize;
- Uint32 vindex, size, thoSeqSize;
- Int32 retSize = -1;
- Int32 packetSize = -1;
- if (ffmpegReader->tempBuffer) {
- goto __consume_tempBuffer;
- }
- if (ffmpegReader->standard == STD_VP9) {
- VP9Superframe* superframe = &ffmpegReader->superframe;
- if (superframe->nframes > 0) {
- Uint32 index = superframe->currentIndex;
- if (index < superframe->nframes) {
- osal_memcpy(packet->data, superframe->frames[index], superframe->frameSize[index]);
- packet->size = superframe->frameSize[index];
- superframe->currentIndex++;
- return packet->size;
- }
- else {
- Uint32 i;
- for (i=0; i<VP9_MAX_SUBFRAMES; i++) {
- if (superframe->frames[i] != NULL) {
- osal_free(superframe->frames[i]);
- }
- }
- osal_memset((void*)superframe, 0x00, sizeof(VP9Superframe));
- }
- }
- }
- seqHeaderSize = 0;
- picHeaderSize = 0;
- thoSeqSize = 0;
- av_init_packet(&avpacket);
- while (TRUE) {
- error = av_read_frame(avFormatContext, &avpacket);
- if (error < 0) {
- if (error == AVERROR_EOF || avFormatContext->pb->eof_reached == TRUE) {
- packet->eos = TRUE;
- return 0;
- }
- else {
- VLOG(ERR, "%s:%d failed to av_read_frame error(0x%08x)\n",
- __FUNCTION__, __LINE__, error);
- goto __end_read;
- }
- }
- if (avpacket.stream_index != ffmpegReader->videoIndex)
- continue;
- break;
- }
- if (avpacket.size == 0)
- return 0;
- if (avpacket.size >= (signed)packet->size )
- {
- VLOG(ERR, "one packet size(%d) is bigger than STREAM_BUF_SIZE(%d)\n", avpacket.size, packet->size);
- return -1;
- }
-
- osal_memset(packet->data, 0x00, packet->size);
- vindex = ffmpegReader->videoIndex;
- if (ffmpegReader->isFirstPacket) {
- AVCodecContext* codec;
- tho_parser_t* theora = ffmpegReader->theora;
- Int32 ret;
- struct {
- int frameWidth ;
- int frameHeight ;
- int picWidth ;
- int picHeight ;
- int picOffsetX ;
- int picOffsetY ;
- } theoraScaleInfo;
- codec = ffmpegReader->avContext->streams[vindex]->codec;
- seqHeader = (Uint8*)osal_malloc(codec->extradata_size + 1024);
- if (seqHeader == NULL) {
- goto __end_read;
- }
- osal_memset((void*)seqHeader, 0x00, codec->extradata_size + 1024);
- seqHeaderSize = BuildSeqHeader(seqHeader, (CodStd)ffmpegReader->standard,
- ffmpegReader->avContext->streams[vindex], &retSize);
- if (seqHeaderSize < 0) {
- VLOG(ERR, "%s:%d Can't make sequence header!\n", __FUNCTION__, __LINE__);
- packetSize = -1;
- goto __end_read;
- }
- if (ffmpegReader->standard == STD_THO || ffmpegReader->standard == STD_VP3) {
- size = seqHeaderSize;
- if (ffmpegReader->standard == STD_VP3) {
- size = 0;
- }
- ret = theora->open(theora->handle, seqHeader, size, (Int32*)&theoraScaleInfo);
- if (ret < 0) {
- VLOG(ERR, "%s:%d failed to open theora parser error(%d)\n", __FUNCTION__, __LINE__, ret);
- goto __end_read;
- }
- thoSeqSize = theora_make_stream(theora->handle, (Uint8*)packet->data, DEC_SEQ_INIT);
- osal_free(seqHeader);
- seqHeader = NULL;
- seqHeaderSize = 0;
- }
- ffmpegReader->isFirstPacket = FALSE;
- }
- picHeader = (Uint8*)osal_malloc(1024);
- picHeaderSize = BuildPicHeader(picHeader, (CodStd)ffmpegReader->standard,
- ffmpegReader->avContext->streams[vindex], &avpacket, 0);
- if (picHeaderSize < 0) {
- VLOG(ERR, "%s:%d failed to build picture header\n", __FUNCTION__, __LINE__);
- goto __end_read;
- }
- ptr = avpacket.data;
- size = avpacket.size;
- switch (ffmpegReader->standard) {
- case STD_RV:
- if (seqHeaderSize)
- osal_memcpy((char*)packet->data, seqHeader, seqHeaderSize);
- if (picHeaderSize)
- osal_memcpy((char*)packet->data+seqHeaderSize, picHeader, picHeaderSize);
- if (ffmpegReader->standard == STD_RV) {
- int cSlice = ptr[0] + 1;
- int nSlice = avpacket.size - 1 - (cSlice*8);
- ptr += (1+(cSlice*8));
- size = nSlice;
- }
- osal_memcpy((char*)packet->data+seqHeaderSize+picHeaderSize, ptr, size);
- packetSize = seqHeaderSize + picHeaderSize + size;
- break;
- case STD_THO:
- case STD_VP3:
- packetSize = MakeupTheoraPacket(ffmpegReader->theora, packet, &avpacket, thoSeqSize);
- break;
- case STD_VP9:
- packet->size = size;
- osal_memcpy((char *)packet->data, ptr, size);
- packetSize = size;
- break;
- default:
- if (picHeaderSize)
- osal_memcpy((char*)packet->data, picHeader, picHeaderSize);
- osal_memcpy((char*)packet->data+picHeaderSize, ptr, size);
- packetSize = picHeaderSize + size;
- break;
- }
- if (avFormatContext->pb->eof_reached && avFormatContext->packet_buffer == NULL) {
- packet->eos = TRUE;
- }
- // Sequence header data should be only one chunk data unit.
- // In case of RV, 1st chunk should be Sequence header + 1st frame.
- if (ffmpegReader->standard != STD_VP9 && ffmpegReader->standard != STD_RV) {
- if (seqHeaderSize > 0) {
- ffmpegReader->tempBuffer = (Uint8*)osal_malloc(packetSize);
- ffmpegReader->tempWrPtr = packetSize;
- osal_memcpy(ffmpegReader->tempBuffer, (Uint8*)(packet->data), packetSize);
- osal_memcpy(packet->data, seqHeader, seqHeaderSize);
- packetSize = seqHeaderSize;
- goto __end_read;
- }
- }
- else if (ffmpegReader->standard == STD_VP9) {
- Uint8* pData = (Uint8*)packet->data;
- Uint32 lastIndex = packet->size - 1;
- if (IS_VP9_SUPERFRAME(pData[lastIndex]) == TRUE) {
- VP9Superframe* superframe = &ffmpegReader->superframe;
- if (VP9ParseSuperframe(pData, packet->size, superframe) == TRUE) {
- osal_memcpy(packet->data, superframe->frames[0], superframe->frameSize[0]);
- packet->size = superframe->frameSize[0];
- packetSize = packet->size;
- superframe->currentIndex++;
- }
- }
- }
- __end_read:
- av_free_packet(&avpacket);
- if (picHeader)
- osal_free(picHeader);
- if (seqHeader)
- osal_free(seqHeader);
- return packetSize;
- __consume_tempBuffer:
- if (ffmpegReader->tempBuffer != NULL) {
- osal_memcpy(packet->data, ffmpegReader->tempBuffer, ffmpegReader->tempWrPtr);
- packetSize = ffmpegReader->tempWrPtr;
- osal_free(ffmpegReader->tempBuffer);
- ffmpegReader->tempBuffer = NULL;
- ffmpegReader->tempWrPtr = 0;
- ffmpegReader->tempRdPtr = 0;
- }
- return packetSize;
- }
- BOOL BSFeederFrameSize_Rewind(
- void* feeder
- )
- {
- ReaderContext* ffmpegReader = (ReaderContext*)feeder;
- AVFormatContext* avFormatContext = ffmpegReader->avContext;
- Int32 ret;
- if ((ret=av_seek_frame(avFormatContext, ffmpegReader->videoIndex, 0, 0)) < 0) {
- VLOG(ERR, "%s:%d Failed to av_seek_frame:(ret:%d)\n", __FUNCTION__, __LINE__, ret);
- return FALSE;
- }
- return TRUE;
- }
- #else
- void* BSFeederFrameSize_Create(
- const char* path,
- CodStd codecId,
- CodStd* retCodecId,
- Uint32* retMp4ClassId,
- Uint32* retSeqWidth,
- Uint32* retSeqHeight
- )
- {
- UNREFERENCED_PARAMETER(path);
- UNREFERENCED_PARAMETER(codecId);
- UNREFERENCED_PARAMETER(retCodecId);
- UNREFERENCED_PARAMETER(retMp4ClassId);
- UNREFERENCED_PARAMETER(retSeqWidth);
- UNREFERENCED_PARAMETER(retSeqHeight);
- VLOG(ERR, "PLEASE PORT THIS %s ON YOUR ANDROID SYSTEM\n", __FUNCTION__);
- return NULL;
- }
- BOOL BSFeederFrameSize_Destroy(
- void* feeder
- )
- {
- UNREFERENCED_PARAMETER(feeder);
- VLOG(ERR, "PLEASE PORT THIS %s ON YOUR ANDROID SYSTEM\n", __FUNCTION__);
- return FALSE;
- }
- Int32 BSFeederFrameSize_Act(
- void* feeder,
- BSChunk* packet
- )
- {
- UNREFERENCED_PARAMETER(feeder);
- UNREFERENCED_PARAMETER(packet);
- VLOG(ERR, "PLEASE PORT THIS %s ON YOUR ANDROID SYSTEM\n", __FUNCTION__);
- return -1;
- }
- #endif /* SUPPORT_FFMPEG_DEMUX */
|