Main Page | Namespace List | Class Hierarchy | Alphabetical List | Class List | File List | Namespace Members | Class Members | File Members

PSNR.hpp

00001  /***********************************************************************
00002  *                                                                      *
00003  * ViTooKi                                                              *
00004  *                                                                      *
00005  * title:       PSNR.hpp                                                      *
00006  *                                                                      *
00007  *                                                                      *
00008  *                                                                      *
00009  * ITEC institute of the University of Klagenfurt (Austria)             *
00010  * http://www.itec.uni-klu.ac.at                                        *
00011  *                                                                      *
00012  *                                                                      *
00013  * For more information visit the ViTooKi homepage:                     *
00014  * http://ViTooKi.sourceforge.net                                       *
00015  * vitooki-user@lists.sourceforge.net                                   *
00016  * vitooki-devel@lists.sourceforge.net                                  *
00017  *                                                                      *
00018  * This file is part of ViTooKi, a free video toolkit.                  *
00019  * ViTooKi is free software; you can redistribute it and/or             *
00020  * modify it under the terms of the GNU General Public License          *
00021  * as published by the Free Software Foundation; either version 2       *
00022  * of the License, or (at your option) any later version.               *
00023  *                                                                      *
00024  * This program is distributed in the hope that it will be useful,      *
00025  * but WITHOUT ANY WARRANTY; without even the implied warranty of       *
00026  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the         *
00027  * GNU General Public License for more details.                         *
00028  *                                                                      *
00029  * You should have received a copy of the GNU General Public License    *
00030  * along with this program; if not, write to the Free Software          *
00031  * Foundation, Inc., 59 Temple Place - Suite 330, Boston,               *
00032  * MA  02111-1307, USA.                                                 *
00033  *                                                                      *
00034  ***********************************************************************/
00035 
00036  /***********************************************************************
00037  *                                                                      *
00038  *  REVISION HISTORY:                                                   *
00039  *                                                                      *
00040  *                                                                      *
00041  *                                                                      *
00042  ***********************************************************************/
00043 
00044 #ifndef _PSNR_HPP
00045 #define _PSNR_HPP
00046 #include "VideoAdaptor.hpp"
00047 #include "VideoESInfo.hpp"
00048 #include "../io/YUVStreamIO.hpp"
00049 #include "UncompressedVideoFrame.hpp"
00050 
00051 class Statistics;
00052 
00064 class PSNR : public VideoAdaptor {
00065 public:
00066 
00067         /* Constructor 
00068          * @param: original yuvstream, which will be opened when first used
00069          * @param: statistics class, to which the avgPSNR per second will be stored
00070          * @param: psnrFP, a file pointer to an already opened text file, where per frame PSNR is stored
00071         */
00072         PSNR(VideoESInfo* esi, YUVStreamIO *OrigYUVStream, Statistics *statistics, FILE* psnrFP=NULL);
00073         
00074         virtual ~PSNR();
00075         
00076         list < Frame * >adapt(Frame * frm);
00077         
00078         Adaptor *clone() { return new PSNR(es, yuvOrig,stats,NULL);     };
00079         
00080         void initialize();
00081         
00082         list < Frame * >close() {
00083                 list < Frame * >tmp;
00084                 if(psnrInfo) {
00085                         fclose(psnrInfo);
00086                         psnrInfo=NULL;
00087                 }
00088                 return tmp;
00089         };
00090 
00092         u32 getTranscodingCosts() const {
00093                 return 10000u;
00094         }
00095 private:
00096 protected:
00097   YUVStreamIO *yuvOrig;
00098         AU lastAU;
00099         u8* lastPayload;
00100         UncompressedVideoFrame *dummyFrame;
00101         Statistics *stats;
00102 
00103         double sumPSNR;
00104         int numFrames;
00105         int numFramesNotInf;
00106         FILE* psnrInfo;
00107 };
00108 #endif
00109