---

YUVGOP.cpp

Go to the documentation of this file.

#include "YUVGOP.hpp"
#include "YUVFrame.hpp"
#include "YUVStream.hpp"

YUVGOP::YUVGOP(Stream* s, unsigned int pat_i) : GOP(s, pat_i) { 
        psnr = 0.0;
        modsAdded = false;
        hasOrg = false; 
}

YUVGOP::YUVGOP(Stream* s, unsigned int mod_i, unsigned int pat_i):GOP(s, mod_i, pat_i) {
        psnr = 0.0; 
        modsAdded = false; 
        hasOrg = false; 
}


YUVGOP::~YUVGOP(){};
        
        
YUVGOP* YUVGOP::getOrgGOP() { 
        return orgGOP; 
}


bool YUVGOP::hasOriginal() { 
        return hasOrg; 
}
        

void YUVGOP::setCompressedByteSize(uint size) { 
        compressedByteSize = size; 
}


uint YUVGOP::getCompressedByteSize() { 
        return compressedByteSize; 
}


void YUVGOP::setCompressedBaselayerSize(uint size) { 
        compressedBaselayerSize = size; 
}


uint YUVGOP::getCompressedBaselayerSize() { 
        return compressedBaselayerSize; 
}


QString YUVGOP::getFramesInGOP() {
  QString frStr = "";
  YUVFrame* fr;
  YUVFrame* frprev;
  FrameType type;

  fr = (YUVFrame*)frameList.first();
  frprev = NULL;

  while (fr != NULL) {
    type = fr->getType();
    
    if ( (frprev != NULL) && (frprev == fr)) // modification dropped frame
      frStr.append('-');
    else if (type == I_VOP)
      frStr.append('I');
    else if (type == P_VOP)
      frStr.append('P');
    else if (type == B_VOP)
      frStr.append('B');
    else
      frStr.append('U');

    frprev = fr;
    fr = (YUVFrame*)frameList.next();
  }
  return frStr;
}


void YUVGOP::setOrgGOP(YUVGOP* org) {
  this->orgGOP = org;
  this->hasOrg = true;
}


void YUVGOP::setPSNRmp(double value) {
  if (isMaster())
    psnr = value;
}


double YUVGOP::computePSNR(YUVGOP* p) {

  unsigned int frameCount = getFrameCount();
  double sumPSNR = 0.0;
  YUVFrame* fr1;
  YUVFrame* fr2;
  YUVFrame* frprev = NULL;

  for (unsigned int i = 0; i < frameCount; i++) {
         fr1 = (YUVFrame*)getFrame(i);

         if ((frprev != NULL) && (frprev == fr1)) {
                // dropped frame -> compute PSNR
                fr2 = (YUVFrame*)p->getFrame(i);
                sumPSNR+= fr1->computePSNR(fr2);
         }
         else {
                fr2 = (YUVFrame*)p->getFrame(i);
                sumPSNR += fr1->computePSNR(fr2);
         }
                frprev = fr1;
  }
  psnr = sumPSNR/frameCount;
  return psnr;
}


QPtrList<YUVGOP>* YUVGOP::expand(YUVGOP* yg) {
        return NULL;
}

---