HEVC Test Model (HM)  HM-16.18
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
TVideoIOYuv.h
Go to the documentation of this file.
1 /* The copyright in this software is being made available under the BSD
2  * License, included below. This software may be subject to other third party
3  * and contributor rights, including patent rights, and no such rights are
4  * granted under this license.
5  *
6  * Copyright (c) 2010-2017, ITU/ISO/IEC
7  * All rights reserved.
8  *
9  * Redistribution and use in source and binary forms, with or without
10  * modification, are permitted provided that the following conditions are met:
11  *
12  * * Redistributions of source code must retain the above copyright notice,
13  * this list of conditions and the following disclaimer.
14  * * Redistributions in binary form must reproduce the above copyright notice,
15  * this list of conditions and the following disclaimer in the documentation
16  * and/or other materials provided with the distribution.
17  * * Neither the name of the ITU/ISO/IEC nor the names of its contributors may
18  * be used to endorse or promote products derived from this software without
19  * specific prior written permission.
20  *
21  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
22  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24  * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS
25  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
26  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
27  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
28  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
29  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
30  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
31  * THE POSSIBILITY OF SUCH DAMAGE.
32  */
33 
38 #ifndef __TVIDEOIOYUV__
39 #define __TVIDEOIOYUV__
40 
41 #include <stdio.h>
42 #include <fstream>
43 #include <iostream>
44 #include "TLibCommon/CommonDef.h"
45 #include "TLibCommon/TComPicYuv.h"
46 
47 using namespace std;
48 
49 // ====================================================================================================================
50 // Class definition
51 // ====================================================================================================================
52 
55 {
56 private:
57  fstream m_cHandle;
58  Int m_fileBitdepth[MAX_NUM_CHANNEL_TYPE];
59  Int m_MSBExtendedBitDepth[MAX_NUM_CHANNEL_TYPE];
60  Int m_bitdepthShift[MAX_NUM_CHANNEL_TYPE];
61 
62 public:
64  virtual ~TVideoIOYuv() {}
65 
66  Void open ( const std::string &fileName, Bool bWriteMode, const Int fileBitDepth[MAX_NUM_CHANNEL_TYPE], const Int MSBExtendedBitDepth[MAX_NUM_CHANNEL_TYPE], const Int internalBitDepth[MAX_NUM_CHANNEL_TYPE] );
67  Void close ();
68 
69  Void skipFrames(Int numFrames, UInt width, UInt height, ChromaFormat format);
70 
71  // if fileFormat<NUM_CHROMA_FORMAT, the format of the file is that format specified, else it is the format of the TComPicYuv.
72 
73 
74  // If fileFormat=NUM_CHROMA_FORMAT, use the format defined by pPicYuvTrueOrg
75  Bool read ( TComPicYuv* pPicYuv, TComPicYuv* pPicYuvTrueOrg, const InputColourSpaceConversion ipcsc, Int aiPad[2], ChromaFormat fileFormat=NUM_CHROMA_FORMAT, const Bool bClipToRec709=false );
76 
77  // If fileFormat=NUM_CHROMA_FORMAT, use the format defined by pPicYuv
78  Bool write ( TComPicYuv* pPicYuv, const InputColourSpaceConversion ipCSC, Int confLeft=0, Int confRight=0, Int confTop=0, Int confBottom=0, ChromaFormat fileFormat=NUM_CHROMA_FORMAT, const Bool bClipToRec709=false );
79 
80  // If fileFormat=NUM_CHROMA_FORMAT, use the format defined by pPicYuvTop and pPicYuvBottom
81  Bool write ( TComPicYuv* pPicYuvTop, TComPicYuv* pPicYuvBottom, const InputColourSpaceConversion ipCSC, Int confLeft=0, Int confRight=0, Int confTop=0, Int confBottom=0, ChromaFormat fileFormat=NUM_CHROMA_FORMAT, const Bool isTff=false, const Bool bClipToRec709=false);
82  static Void ColourSpaceConvert(const TComPicYuv &src, TComPicYuv &dest, const InputColourSpaceConversion conversion, Bool bIsForwards);
83 
84  Bool isEof ();
85  Bool isFail();
86 
87 
88 };
89 
90 #endif // __TVIDEOIOYUV__
91 
Void write(ostream &out, OutputNALUnit &nalu)
Definition: NALwrite.cpp:64
YUV file I/O class.
Definition: TVideoIOYuv.h:54
picture YUV buffer class
Definition: TComPicYuv.h:55
Defines version information, constants and small in-line functions.
void Void
Definition: TypeDef.h:203
virtual ~TVideoIOYuv()
Definition: TVideoIOYuv.h:64
unsigned int UInt
Definition: TypeDef.h:212
fstream m_cHandle
file handle
Definition: TVideoIOYuv.h:57
bool Bool
Definition: TypeDef.h:204
ChromaFormat
chroma formats (according to semantics of chroma_format_idc)
Definition: TypeDef.h:292
picture YUV buffer class (header)
InputColourSpaceConversion
Definition: TypeDef.h:316
int Int
Definition: TypeDef.h:211
Void read(InputNALUnit &nalu)
Definition: NALread.cpp:175