---

YUVStream.cpp

Go to the documentation of this file.

#include "YUVStream.hpp"

#ifdef WIN32

#else
#include <unistd.h>

#endif
#include <qtimer.h>


YUVStream::~YUVStream() {}

YUVStream::YUVStream(const QString &fname, int x, int y, 
                     bool preRead, uint fpp) 
        : Stream(fname,preRead) {
                
  // open the file
  file.setName(fname);
  if (!file.open(IO_ReadOnly)) {
#if MYDEBUG >= 1
         cout << "Can't open " << fname.ascii() << "!\n\n";
#endif
         //      exit(ERR_FILE);         // use exception here l8er
  }
  
  width = x; 
  height = y;
  ySize = width * height;
  uvSize = ySize / 4;
  noFrames = 0;
  minCompPatSize = 0;
  maxCompPatSize = 0;

  bufOffsetY = 0;
  bufOffsetU = bufOffsetY + ySize;
  bufOffsetV = bufOffsetU + uvSize;

  // set file to start position
  file.at(0);

  gopList.setAutoDelete(TRUE);
  fileSize = file.size();
  maxPSNR = 0;

#if MYDEBUG >= 4
  cout << "estimated maxPSNR: " << maxPSNR << nl;
#endif

  if (preRead)
    while (addNextGOP(fpp) != NULL) 
      ;
    
}

long YUVStream::getFrameSize() { return ySize + 2 * uvSize; }

long YUVStream::getYSize() { return ySize; }

long YUVStream::getUSize() { return uvSize; }

long YUVStream::getVSize() { return uvSize; }

uint YUVStream::getMinCompressedPatSize() { return minCompPatSize; }

void YUVStream::setMinCompressedPatSize(uint s) { minCompPatSize = s; }

uint YUVStream::getMaxCompressedPatSize() { return maxCompPatSize; }

void YUVStream::setMaxCompressedPatSize(uint s) { maxCompPatSize = s; }

uint YUVStream::getMinCompressedBaselayerSize() { return minCompBLSize; }

void YUVStream::setMinCompressedBaselayerSize(uint s) { minCompBLSize = s; }

uint YUVStream::getMaxCompressedBaselayerSize() { return maxCompBLSize; }

void YUVStream::setMaxCompressedBaselayerSize(uint s) { maxCompBLSize = s; }

long YUVStream::getBufOffsetY() {return bufOffsetY; }

long YUVStream::getBufOffsetU() {return bufOffsetU; }

long YUVStream::getBufOffsetV() {return bufOffsetV; }

double YUVStream::getMaxPSNR() {return maxPSNR; }

long YUVStream::getFileSize() { return fileSize; }



GOP* YUVStream::addNextGOP(uint numFrames) {
  uint fppCount = 0;               // frames per pattern count
  YUVGOP *p;
  YUVFrame *frame;
  long filePos = file.at();

  if (filePos >= fileSize) 
    return NULL;

  // go through the file
  p = new YUVGOP(this, gopList.count());
  gopList.append(p);
  p->setPSNRmp(maxPSNR);


  while ( (filePos < fileSize) && (fppCount < numFrames) ) {
    // fill current pattern with frames
#ifdef MENS_HEKK
         if (fppCount == 0)
                frame = new YUVFrame(noFrames, fppCount, filePos, p, I_VOP);
         else
                frame = new YUVFrame(noFrames, fppCount, filePos, p, B_VOP);
    frame->setNoInDisplay(noFrames);
    frame->setByteSize(this->getFrameSize()); // YUV has fixed sizes
    p->addFrame(frame);
    filePos += this->getFrameSize();      // calc new file offset
    file.at(filePos);
    fppCount++;
    noFrames++;                     // inc number of frames
         continue;
#endif
    frame = new YUVFrame(noFrames, fppCount, filePos, p);
    frame->setNoInDisplay(noFrames);
    frame->setByteSize(this->getFrameSize()); // YUV has fixed sizes
    p->addFrame(frame);
    filePos += this->getFrameSize();      // calc new file offset
    file.at(filePos);
    fppCount++;
    noFrames++;                     // inc number of frames
  }

  p->calcByteSize();

#if MYDEBUG >= 4
  printf("YUVaddNextGOP (%i frames@%ld) from file pos %li (From %li)\n",
         p->getFrameCount(),getFrameSize(),filePos,fileSize);
#endif

  return p;
}

---