---

YUVFrame.cpp

Go to the documentation of this file.

#include "YUVFrame.hpp"
#include "YUVStream.hpp"
#include "GOP.hpp"
#include "MPGFrame.hpp"
#include <math.h>


YUVFrame::YUVFrame(long noS, unsigned int noP,
                                   long offset, GOP * pat, FrameType t):Frame(noS, noP,
                                   offset, pat, t) {

  this->psnr = 0;
}


YUVFrame::~YUVFrame() {
}


void YUVFrame::setCompressedByteSize(uint siz) {
 compBSize = siz;
}


uint YUVFrame::getCompressedByteSize() {
 return compBSize;
}


MPGFrame* YUVFrame::getMPGFrame() {
 return mpgFrame;
}


void YUVFrame::setMPGFrame(MPGFrame * mpg) {
 mpgFrame = mpg;
}


double YUVFrame::computePSNRy(YUVFrame * other) {

  if (!inMem()) {
#if MYDEBUG >= 1
         cerr << "Can't compute PSNR! Frame isn't in memory!\n";
#endif
         return ERR_PSNR;
  }

  double rmse, mse;
  mse = 0;
  rmse = 0;
  int k = 0;
  double temp = 0;
  unsigned char *otherData = (unsigned char *) other->getBuffer();
  unsigned char *thisData = (unsigned char *) getBuffer();
  YUVStream *thisParentStream = (YUVStream *) pattern->getStream();
  long yOffset = thisParentStream->getBufOffsetY();
  int width = thisParentStream->getWidth();
  int height = thisParentStream->getHeight();

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

  for (int i = 0; i < width; i++) {
         for (int j = 0; j < height; j++) {
                temp = (double) thisData[k + yOffset] -
                  (double) otherData[k + yOffset];
                mse += temp * temp;
                k++;
         }
  }

  if (mse == 0) {
         // frames are equal return maxPSNR of stream
         YUVStream *ystr = (YUVStream *) pattern->getStream();
         return ystr->getMaxPSNR();
  }

  mse = (double) mse / ((double) width * (double) height);
  rmse = sqrt(mse);
  return 20 * log10(255.0 / rmse);
}


double YUVFrame::computePSNR(YUVFrame * f) {
  if (!inMem()) {
#if MYDEBUG >= 1
         cerr << "Can't compute PSNR! Frame isn't in memory!\n";
#endif
         return ERR_PSNR;
  }

  this->psnr = this->computePSNRy(f);

  return this->psnr;
}

---