Home    News    Components    Get ViTooKi    Documentation    Community    Developers    @sf.net  

Info

Last update:
12 Feb 2007

Documentation

Doxygen LXR

MPGStreamIO Class Reference

<class description="" goes="" here=""> <short description=""> More...

#include <MPGStreamIO.hpp>

Inheritance diagram for MPGStreamIO:

List of all members.

Public Member Functions
	MPGStreamIO (ESInfo *es, const char *file, bool writeOnly, bool omitHeader=false)
Frame *	getFrame ()
	returns a frame if one complete frame is available, otherwise null is returned.
int	writeFrame (Frame *frm, ESInfo *out_es=NULL)
	returns the number of packets sent.
bool	open ()
	opens the IO connection.
bool	close (bool immediate=false)
	closes the IO class.
bool	destroy ()
	deletes the elementary stream and its extra hint-file
int	getBufferFillLevel () const
	returns a value from 0..100 indicating buffer usage.
IO *	clone () const
const char *	getURL () const
	returns a local file name or an URL
bool	setToFrameNumber (u32 frameNumber)
	repositions the IO class to the given frame.
long	setToClosestIFrame (u32 frameNumber, bool backwards)
	repositions to previous I-VOP,
void	setESInfo (ESInfo *esi)
ESInfo *	getESInfo ()
Protected Attributes
FILE *	io
char *	url
bool	writeOnly
ESInfo *	es
bool	atFileStart
FILE *	helpFile
bool	writeHelpFile
bool	omitHeader
u32	lastCTSSeen

---

Detailed Description

<class description="" goes="" here=""> <short description="">

Author:

Michael Kropfberger and Peter Schojer

Version:

Id

MPGStreamIO.hpp,v 1.9 2006/01/26 11:42:00 mkropfbe Exp

Definition at line 64 of file MPGStreamIO.hpp.

---

Member Function Documentation

bool MPGStreamIO::close (  bool  immediate = false  )  [virtual]

closes the IO class. Parameters: immediate  specifies, if (optional) buffered data should be read/sent to the client (==false), or immediately dumped (==true) Implements IO. Reimplemented in BufferedHttpMPGStreamReader. Definition at line 180 of file MPGStreamIO.cpp. Referenced by IncompleteIO::close(), destroy(), getFrame(), BufferedMPGStreamReader::getFrame(), setToClosestIFrame(), and setToFrameNumber(). { state = CLOSING; if (io) { fclose(io); io=NULL; } if (helpFile) { fclose(helpFile); helpFile=NULL; } state = CLOSED; // this->es->setDuration(lastCTSSeen+es->getVOPTimeIncrement()); return true; }

int MPGStreamIO::getBufferFillLevel (   )  const [inline, virtual]

returns a value from 0..100 indicating buffer usage. Fixme: returns a constant value of 50 Implements IO. Definition at line 93 of file MPGStreamIO.hpp. { return 50; };

Frame * MPGStreamIO::getFrame (   )  [virtual]

returns a frame if one complete frame is available, otherwise null is returned. This function is typically blocking. Don't use a NULL return value to conclude STREAMEOF, always check with getState()! Implements IO. Reimplemented in BufferedHttpMPGStreamReader, and BufferedMPGStreamReader. Definition at line 262 of file MPGStreamIO.cpp. References close(), Frame::getAU(), ESInfo::getStreamId(), ESInfo::getVOPTimeIncrement(), ESInfo::isAudioStream(), ESInfo::isVisualStream(), Frame::setAU(), and Frame::setMediaTimeScale(). Referenced by IncompleteIO::getFrame(). { if (writeOnly || state != OPEN) return NULL; if (atFileStart) { atFileStart = false; //FIXME: why we need to call this for audiostream also? //if (es->isVisualStream()) { if (!readHeader()) state=STREAMERR; return NULL; //} } if (( (this->currentFrameNumber > this->endFrameNumber) && (endFrameNumber!=0) ) || feof(io)) { close(true); state=STREAMEOF; return NULL; } // now read the frame // first check hintfile AU *au = new AU(); Frame *frm = NULL; Frame::FrameType type = Frame::NN_VOP; if (writeHelpFile) { // read one AU entry from the file int auSize = sizeof(AU) - sizeof(u8 *); int tmp = 0; if (auSize != (tmp = fread((char *) &(au->size),sizeof(char),auSize,helpFile))) { if (tmp <= 0) { // last frame state = STREAMEOF; delete au; return NULL; } dprintf_err("MPGStreamIO::getFrame() esID %llu: Mismatch in hintfile. Expected %i bytes, but just got %i?\r\n", es->getStreamId(), auSize, tmp); state = STREAMEOF; delete au; return NULL; } else { // au is correct // read the AU au->payload = new u8[au->size]; assert(au->payload); if ((unsigned long) au->size != fread(au->payload,sizeof(char),au->size,io)) { dprintf_err("MPGStreamIO::getFrame(): Mismatch in data & hintfile. Not enough data?\r\n"); delete au; state= STREAMERR; return NULL; } dprintf_full("MPGStreamIO::getFrame(): got hinted info CTS %i prio: %i\n",au->cts,au->prio); } } else { // no help stream available, how to calc cts and dts? // parse MPGFile for VOB_Code if it is a visual stream // FIXME: do Audio streams have a different VOP Code?? if(es->isVisualStream()) { // 00 00 01 182 dec -> 0x1B6 (=438 dec) u32 code = 2000; // don't init to 0, we want the code from the 2nd frame, not the first int ch; int i = 0; char *data = new char[MAX_FRAME_SIZE]; do { ch = getc(io); data[i] = (u8)ch; i++; code = (code << 8) | (u8)ch; if ((i==4) && code == VOP_START_CODE) code =0; //correct beginning, parse until next START_CODE } while ((i < MAX_FRAME_SIZE) && (ch != EOF) && (code != VOP_START_CODE)); if (code == VOP_START_CODE) { // put back the VOP_HEADER // seek back 4 bytes fseek(io, -4, SEEK_CUR); au->size = i; } if (au->size == 0) { state = STREAMEOF; delete au; return NULL; } au->payload = new u8[i]; memcpy(au->payload, data, i); au->prio = IO_NETWORK_HIGHEST_PRIORITY; au->duration = es->getVOPTimeIncrement(); au->cts = currentFrameNumber * es->getVOPTimeIncrement(); au->dts = currentFrameNumber * es->getVOPTimeIncrement(); au->sampleFlags = 0; if (code != VOP_START_CODE) au->err = SA_EOF; else au->err = SA_OK; delete data; } if(es->isAudioStream()) { //XXX not sure if it works properly dprintf_err("MPGStreamIO audioStream parsing\n\n"); // start code for MP3 frame is 11111111 111 (first 11 bits set) u32 code = 2000; // don't init to 0, we want the code from the 2nd frame, not the first int ch; int i = 0; char *data = new char[165536]; do { ch = getc(io); data[i] = ch; i++; code = (code << 8) | ch; } while ((ch != -1) && (code>>21 != 0x000007FF /*VOP_START_CODE for audio(actually MP3)*/)) ; if (code>>21 == 0x000007FF /*VOP_START_CODE*/) { // put back the VOP_HEADER // seek back 4 bytes fseek(io, -4, SEEK_CUR); /* ungetc(182,io); ungetc(1,io); ungetc(0,io); ungetc(0,io); */ au->size = i; } au->payload = new u8[i]; memcpy(au->payload, data, i); au->cts = 0; au->dts = 0; au->duration = 0; au->sampleFlags = 0; if (code>>21 != 0x000007FF /*VOP_START_CODE*/) au->err = SA_EOF; else au->err = SA_OK; delete [] data; } } //FIXME: quickhack to assign priorities, if no Hint file is avail (eg. direct access to .cmp files!) if (es->isVisualStream() && !writeHelpFile) { if (au->payload[0] == 0 && au->payload[1] == 0 && au->payload[2] == 1 && au->payload[3] == 182) { type = (Frame::FrameType) ((au->payload[4]) >> 6); if (type == Frame::I_VOP || type == Frame::P_VOP) au->prio = IO_NETWORK_HIGHEST_PRIORITY; else au->prio = ~IO_NETWORK_HIGHEST_PRIORITY; } } // frame types will be detected in frm->setAU()! if(es->isVisualStream()) frm = new CompressedVideoFrame(type, ((VideoESInfo*)es)->getWidth(), ((VideoESInfo*)es)->getHeight()); else if(es->isAudioStream()) frm=new CompressedAudioFrame(type); else frm = new CompressedVideoFrame(type, 0, 0); frm->setAU(au); frm->setMediaTimeScale(es->getMediaTimeScale()); framesSeen++; currentFrameNumber++; dprintf_full("MPGStreamIO::getFrame: file offset %li cts %i dts %i type %s (%x %x %x %x) size %i, frameNo %u\r\n", ftell(io),au->cts,au->dts, Frame::VOPTypeToChar((Frame::VOP_TYPE)frm->getAU()->frameType), frm->getAU()->payload[0],frm->getAU()->payload[1],frm->getAU()->payload[2],frm->getAU()->payload[3], frm->getAU()->size, getCurrentFrameNumber()); #ifdef _POSIX_PRIORITY_SCHEDULING sched_yield(); #else // for windows pthread lib! #ifdef WIN32 #ifndef WINCE sched_yield(); #else Sleep(0); #endif #endif #endif return frm; }

bool MPGStreamIO::open (   )  [virtual]

opens the IO connection. State is set to OPENING. Depending on the underlying QIODevice, a network connection is stablished or a file connection. When the connection is ready for use, State is OPEN Implements IO. Reimplemented in BufferedHttpMPGStreamReader. Definition at line 96 of file MPGStreamIO.cpp. References ESInfo::disableFrameStatistic(), ESInfo::enableFrameStatistic(), ESInfo::getDuration(), ESInfo::getInput(), ESInfo::getNumberOfMediaSamples(), and ESInfo::getVOPTimeIncrement(). Referenced by IncompleteIO::open(), and setToFrameNumber(). { if(!url) { // try to get localfile from esinfo const char* tmpUrl=es->getInput(); if(tmpUrl) { url=new char[strlen(tmpUrl)+1]; strcpy(url,tmpUrl); } else return false; } dprintf_full("MPGStreamIO::open() Trying to open %s\r\n", url); if(state==OPEN) return true; bool ret = false; atFileStart = true; state = OPENING; lastCTSSeen=0; char *name = new char[512]; strcpy(name,url); strcat(name,".hint"); if (writeOnly) { // we want to write to a local file // reset statistic dprintf_full("MPGStreamIO::open(): resetting Statistic to %u dur %u ticks %u\n", es->getNumberOfMediaSamples(),(unsigned)es->getDuration(), es->getVOPTimeIncrement()); es->disableFrameStatistic(); es->enableFrameStatistic(); if (writeHelpFile) { if(helpFile) fclose(helpFile); helpFile = fopen(name,"wb"); if (!helpFile) { helpFile = NULL; writeHelpFile = false; } } if(io) { // safety check, shouldn't happen fclose(io); } io=fopen(url,"wb"); ret = (io!=NULL); } else { // we want to read from a local file if (writeHelpFile) { helpFile = fopen(name,"rb"); if (helpFile) { fseek(helpFile,0,SEEK_END); if (ftell(helpFile) == 0){ //zero length fclose(helpFile); helpFile = NULL; } else fseek(helpFile,0,SEEK_SET); } if (!helpFile) { dprintf_full("MPGStreamIO::open no valid .hint file found, ignoring...\n"); helpFile = NULL; writeHelpFile = false; } else { dprintf_full("MPGStreamIO::open using and relying on .hint file %s\n",name); } } io=fopen(url,"rb"); ret = (io!=NULL); } if (ret) state = OPEN; else state = CLOSED; currentFrameNumber=0; framesSeen=0; delete [] name; return ret; }

long MPGStreamIO::setToClosestIFrame (  u32  frameNumber, bool  backwards )  [virtual]

repositions to previous I-VOP, Returns: frameNr or -1 if no previous I-VOP found Reimplemented from IO. Reimplemented in BufferedHttpMPGStreamReader. Definition at line 811 of file MPGStreamIO.cpp. References close(), ESInfo::getNumberOfMediaSamples(), ESInfo::isAudioStream(), and setToFrameNumber(). Referenced by IncompleteIO::setToClosestIFrame(). { dprintf_full("MPGStreamIO::setToClosestIFrame searching direction: %s\n",backwards?"backwards":"forwards"); if(!setToFrameNumber(frameNumber)) return (-1); if(frameNumber==0 || es->isAudioStream()) { dprintf_full("MPGStreamIO::setToClosestIFrame returning %u (%s stream)\n", frameNumber, es->isAudioStream() ? "audio" : "video"); return frameNumber; } long currentFilePos=ftell(io); long currentHelpPos=0; long storedFrameNumber = currentFrameNumber; int auSize = sizeof(AU) - sizeof(u8 *); int tmp=0; AU au; if(helpFile) currentHelpPos=ftell(helpFile); long newFilePos=currentFilePos; bool firstFrame=true; // get the AU for the next read Frame call tmp = fread((char *) &(au.size), 1, auSize, helpFile); if (auSize==tmp) { newFilePos+=au.size; } else { //FIXME: maybe we have probs on very last frame :( state=STREAMERR; } while ((currentFrameNumber>=0) && (currentFrameNumber <= es->getNumberOfMediaSamples())) { if (backwards || firstFrame) { fseek(helpFile, -auSize, SEEK_CUR); } tmp = fread((char *) &(au.size), 1, auSize, helpFile); if(auSize!=tmp) { // if a seek fails undo all changes to filepointers fseek(helpFile,currentHelpPos,SEEK_SET); fseek(io,currentFilePos,SEEK_SET); currentFrameNumber = storedFrameNumber; dprintf_err("MPGStreamIO::setToClosestIFrame: failed to set to I-frame\n"); return currentFrameNumber; } dprintf_full("frame %i type %s CTS %i\n",currentFrameNumber,Frame::VOPTypeToChar((Frame::VOP_TYPE)au.frameType),au.cts); if (backwards) { fseek(helpFile,-auSize,SEEK_CUR); newFilePos -= au.size; } if (!backwards && !firstFrame) { newFilePos += au.size; } firstFrame=false; if(au.frameType==(u32)Frame::I_VOP || au.frameType == (u32) Frame::HEADER_VOP) { dprintf_full("MPGStreamIO::setToClosestIFrame: found %s at frame %i\n", Frame::VOPTypeToChar((Frame::VOP_TYPE)au.frameType), currentFrameNumber); if (!backwards) { //go back to actual frame newFilePos -= au.size; fseek(helpFile,-auSize,SEEK_CUR); } // point to last read AU in MPGStream fseek(io,newFilePos,SEEK_SET); return currentFrameNumber; } if (backwards) currentFrameNumber--; else currentFrameNumber++; } //error: reset to beginning or end dprintf_err("MPGStreamIO::setToClosestIFrame() FAILED, currentFrameNumber = %u, nMediaSamples = %u " "repositioning to %s\n", currentFrameNumber, es->getNumberOfMediaSamples(), backwards ? "START" : "END"); if (backwards) { fseek(io,0,SEEK_SET); if(helpFile) fseek(helpFile,0, SEEK_SET); atFileStart = true; currentFrameNumber=0; } else { fseek(io,0,SEEK_END); if(helpFile) fseek(helpFile,0, SEEK_END); atFileStart = false; currentFrameNumber=es->getNumberOfMediaSamples(); close(true); state=STREAMEOF; } return currentFrameNumber; }

bool MPGStreamIO::setToFrameNumber (  u32  frameNumber  )  [virtual]

repositions the IO class to the given frame. Will return false in the following cases: an illegal frame number (too large) was specified the stream is not seekable, because the underlying device is a network device and the requested frame is already out of range: FIXME: implement Reimplemented from IO. Reimplemented in BufferedHttpMPGStreamReader. Definition at line 673 of file MPGStreamIO.cpp. References close(), ESInfo::getHeaders(), and open(). Referenced by setToClosestIFrame(), and IncompleteIO::setToFrameNumber(). { // check if valid // e.g.: writeOnly streams can not be seeked if(writeOnly) { dprintf_err("MPGStreamIO::setToFrameNumber(u32 %i) failed: MPGStreamIO opened for writing\n",frameNumber); return false; } dprintf_full("MPGStreamIO::setToFrameNumber: current is %i, new is %i\n",currentFrameNumber,frameNumber); if(!io) { // shouldn't happen??? reset class dprintf_small("MPGStreamIO::setToFrameNumber(u32 %i): called on closed MPGStreamIO\n",frameNumber); close(true); open(); } if(!io) { dprintf_err("MPGStreamIO::setToFrameNumber(u32 %i) failed: couldn't open input %s?\n",frameNumber,url); return false; } dprintf_full("MPGStreamIO::setToFrameNumber(u32 %i)\r\n",frameNumber); long currentFilePos=ftell(io); long currentHelpPos=0; if(helpFile) currentHelpPos=ftell(helpFile); long newFilePos=currentFilePos; // readonly stream if(currentFrameNumber!=frameNumber) { // reposition if(frameNumber==0) { // set to beginning fseek(io,0,SEEK_SET); if(helpFile) fseek(helpFile,0, SEEK_SET); atFileStart = true; currentFrameNumber=0; // I-frame always true return true; } else { int auSize = sizeof(AU) - sizeof(u8 *); int tmp=0; AU *au=NULL; atFileStart = false; if(currentHelpPos==0) { readHeader(); u8* temp=NULL; newFilePos+=this->es->getHeaders(&temp); if(temp) delete[] temp; } while(currentFrameNumber!=frameNumber) { if(writeHelpFile) { // seeking easy: load data from helpFile au = new AU(); //seek forward or backward if(frameNumber>currentFrameNumber) { // seek forward tmp = fread((char *) &(au->size), 1, auSize, helpFile); if(auSize!=tmp) { // if a seek fails undo all changes to filepointers fseek(helpFile,currentHelpPos,SEEK_SET); fseek(io,currentFilePos,SEEK_SET); dprintf_err("MPGStreamIO::setToFrameNumber: failed to seek forward\n"); delete au; return false; } // dprintf_full("MPGStreamIO::setToFrameNumber: cur=%i new=%i newFP=%li addedAUSize=%i\n", // currentFrameNumber,frameNumber,newFilePos,au->size); newFilePos+=au->size; currentFrameNumber++; delete au; } else { // seek backward // go back one au entry fseek(helpFile,-auSize,SEEK_CUR); tmp = fread((char *) &(au->size), 1, auSize, helpFile); // now that we have read the au, reset it to // the previous position fseek(helpFile,-auSize,SEEK_CUR); if(auSize!=tmp) { // if a seek fails undo all changes to filepointers fseek(helpFile,currentHelpPos,SEEK_SET); fseek(io,currentFilePos,SEEK_SET); dprintf_err("MPGStreamIO::setToFrameNumber: failed to seek backward\n"); delete au; return false; } newFilePos-=au->size; currentFrameNumber--; delete au; } } else { // no HelpFile // we could parse vor VOP header, but we don't :-) dprintf_err("MPGStreamIO::setToFrameNumber: failed. no helpfile\n"); return false; } } // while // if we come that far, we can reposition the movie if(newFilePos>0) { //helpFile should be at correct position fseek(io,newFilePos,SEEK_SET); currentFrameNumber=frameNumber; return true; } else { // BUG??? We tried to set to a negative position dprintf_err("MPGStreamIO::setToFrameNumber: Negative new file pointer\r\n"); // just set to zero; // set to beginning fseek(io,0,SEEK_SET); if(helpFile) fseek(helpFile,0, SEEK_SET); atFileStart = true; currentFrameNumber=0; return true; } } } else { // no need to reposition return true; } return true; }

int MPGStreamIO::writeFrame (  Frame *  frm, ESInfo *  out_es = NULL )  [virtual]

returns the number of packets sent. Returns 0 on error Implements IO. Definition at line 197 of file MPGStreamIO.cpp. References Frame::getAU(), ESInfo::getFrameStatistic(), ESInfo::getStreamId(), ESInfo::getVOPTimeIncrement(), and BitField::setBit(). { if(!frm) return 0; if (!writeOnly || state != OPEN) { dprintf_small("MPGStreamIO esID %llu Warning: Output is readonly or not open\n", es->getStreamId()); return -1; } // drop frames we have already written!! if(lastCTSSeen>=frm->getAU()->cts && lastCTSSeen!=0) { dprintf_small("MPGStreamIO::writeFrame Dropping Frame %u\n",frm->getAU()->cts); return 0; } else lastCTSSeen=frm->getAU()->cts; if (frm->getAU()->size == 0) { dprintf_small("MPGStreamIO::writeFrame esID %llu: Cannot write zero sized frames\n", es->getStreamId()); return 0; } u32 frmBitIndex=frm->getAU()->cts/es->getVOPTimeIncrement(); dprintf_full("MPGStreamIO::writeFrame CTS %u(%u), size %u(%u %u %u %u...), VOPINC %u\n", frm->getAU()->cts,frmBitIndex,frm->getAU()->size,frm->getAU()->payload[0], frm->getAU()->payload[1],frm->getAU()->payload[2],frm->getAU()->payload[3], es->getVOPTimeIncrement()); // write header right in front of first frame!!! if (atFileStart && !omitHeader) { atFileStart = false; if (!saveHeader()) { dprintf_small("MPGStreamIO::writeFrame esID %llu: failed to save header\n", es->getStreamId()); return -1; } } // save the frame int i = (frm->getAU()->size == fwrite(frm->getAU()->payload,sizeof(char), frm->getAU()->size,io)); framesSeen++; es->getFrameStatistic()->setBit(frmBitIndex); if (writeHelpFile) writeHelpInfo(frm); #ifdef _POSIX_PRIORITY_SCHEDULING sched_yield(); #else // for windows pthread lib! #ifndef WINCE sched_yield(); #else Sleep(0); #endif #endif return i; }

---

The documentation for this class was generated from the following files:

• MPGStreamIO.hpp
• MPGStreamIO.cpp

© ViTooKi 2003 :: [Lite version] Created with Doxygen