UncompressedVideoFrame.hpp

00001 /*********************************************************************** 00002 * * 00003 * ViTooKi * 00004 * * 00005 * title: UncompressedVideoFrame.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 __UncompressedVideoFrame_hpp__ 00045 #define __UncompressedVideoFrame_hpp__ 00046 00047 #include "VideoFrame.hpp" 00048 00049 00061 class UncompressedVideoFrame : public VideoFrame { 00062 00063 public: 00064 00065 UncompressedVideoFrame(FrameType t, int width, int height); 00066 00067 virtual ~UncompressedVideoFrame(); 00068 00069 static Frame::FrameType detectFrameType(AU* accessUnit); 00070 00075 Frame::FrameType detectFrameType(); 00076 00088 typedef enum ColorSpace { 00089 ColorSpaceDefault = 4, 00090 ColorSpaceRGB24 = 0, 00091 ColorSpaceYV12 = 1, 00092 ColorSpaceI420 = 4, 00093 ColorSpaceRGB32 = 1000, 00094 ColorSpaceRGB565, // implemented in interface to FFMPEG only 00095 ColorSpaceGRAY8, // implemented in interface to FFMPEG only 00096 ColorSpaceUNKNOWN 00097 } ColorSpace; 00098 00099 00104 static int mapColorSpaceToBitDepth(ColorSpace colorspace); 00105 00106 void setColorSpace(ColorSpace c) { colorSpace = c; }; 00107 00108 ColorSpace getColorSpace() { return colorSpace; }; 00109 00110 /* should transcode a single frame to another colorspace 00111 * @returns a newly allocated frame, dont forget to delete the old one! */ 00112 UncompressedVideoFrame *convertToColorSpace(ColorSpace cs); 00113 00119 int getYBuffer(u8 **buf); 00125 int getUBuffer(u8 **buf); 00131 int getVBuffer(u8 **buf); 00132 00133 double computePSNRy(UncompressedVideoFrame * f); 00134 double computePSNRu(UncompressedVideoFrame * f); 00135 double computePSNRv(UncompressedVideoFrame * f); 00136 double computePSNR(UncompressedVideoFrame * f); 00137 00138 double getPSNR() { return this->psnr; }; 00139 00140 00141 protected: 00142 double computePSNRwithOffset(UncompressedVideoFrame * f, long offset); 00143 00144 double psnr; 00145 00146 ColorSpace colorSpace; 00147 00148 }; 00149 00150 #endif // __UncompressedVideoFrame_hpp__