Context navigation

source: SHVCSoftware/branches/SHM-dev/source/Lib/TLibCommon/TComPic.h @ 1590

Visit:

Last change on this file since 1590 was 1574, checked in by seregin, 9 years ago
port rev 4764
Property svn:eol-style set to `native`
File size: 13.1 KB

Line
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-2016, 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
34	/** \file TComPic.h
35	\brief picture class (header)
36	*/
37
38	#ifndef __TCOMPIC__
39	#define __TCOMPIC__
40
41	// Include files
42	#include "CommonDef.h"
43	#include "TComPicSym.h"
44	#include "TComPicYuv.h"
45	#include "TComBitStream.h"
46	#if AVC_BASE
47	#include <fstream>
48	#endif
49
50	//! \ingroup TLibCommon
51	//! \{
52
53	// ====================================================================================================================
54	// Class definition
55	// ====================================================================================================================
56
57	/// picture class (symbol + YUV buffers)
58
59	class TComPic
60	{
61	public:
62	typedef enum { PIC_YUV_ORG=0, PIC_YUV_REC=1, PIC_YUV_TRUE_ORG=2, NUM_PIC_YUV=3 } PIC_YUV_T;
63	// TRUE_ORG is the input file without any pre-encoder colour space conversion (but with possible bit depth increment)
64	TComPicYuv* getPicYuvTrueOrg() { return m_apcPicYuv[PIC_YUV_TRUE_ORG]; }
65
66	private:
67	UInt m_uiTLayer; // Temporal layer
68	Bool m_bUsedByCurr; // Used by current picture
69	Bool m_bIsLongTerm; // IS long term picture
70	TComPicSym m_picSym; // Symbol
71	TComPicYuv* m_apcPicYuv[NUM_PIC_YUV];
72
73	TComPicYuv* m_pcPicYuvPred; // Prediction
74	TComPicYuv* m_pcPicYuvResi; // Residual
75	Bool m_bReconstructed;
76	Bool m_bNeededForOutput;
77	UInt m_uiCurrSliceIdx; // Index of current slice
78	Bool m_bCheckLTMSB;
79
80	Bool m_isTop;
81	Bool m_isField;
82
83	std::vector<std::vector<TComDataCU*> > m_vSliceCUDataLink;
84
85	SEIMessages m_SEIs; ///< Any SEI messages that have been received. If !NULL we own the object.
86
87	#if SVC_EXTENSION
88	UInt m_layerId; // Layer ID
89	Bool m_requireResampling[MAX_LAYERS]; // whether current layer requires resampling
90	TComPicYuv* m_pcFullPelBaseRec[MAX_LAYERS]; // upsampled base layer recontruction for difference domain inter prediction
91	Bool m_equalPictureSizeAndOffsetFlag[MAX_LAYERS];
92	Int* m_mvScalingFactor[2];
93	Int* m_posScalingFactor[2];
94	#if CGS_3D_ASYMLUT
95	Int m_nFrameBit;
96	#endif
97	Bool m_currAuFlag;
98	#endif
99	public:
100	TComPic();
101	virtual ~TComPic();
102
103	#if REDUCED_ENCODER_MEMORY
104	#if SVC_EXTENSION
105	Void create( const TComSPS &sps, const TComPPS &pps, const Bool bCreateEncoderSourcePicYuv, const Bool bCreateForImmediateReconstruction, const UInt layerId );
106	#else
107	Void create( const TComSPS &sps, const TComPPS &pps, const Bool bCreateEncoderSourcePicYuv, const Bool bCreateForImmediateReconstruction );
108	#endif
109	Void prepareForEncoderSourcePicYuv();
110	Void prepareForReconstruction();
111	Void releaseReconstructionIntermediateData();
112	Void releaseAllReconstructionData();
113	Void releaseEncoderSourceImageData();
114	#else
115	#if SVC_EXTENSION
116	Void create( const TComSPS &sps, const TComPPS &pps, const Bool bIsVirtual /= false/, const UInt layerId );
117	#else
118	Void create( const TComSPS &sps, const TComPPS &pps, const Bool bIsVirtual /= false/ );
119	#endif
120	#endif
121
122	virtual Void destroy();
123
124	UInt getTLayer() const { return m_uiTLayer; }
125	Void setTLayer( UInt uiTLayer ) { m_uiTLayer = uiTLayer; }
126
127	Bool getUsedByCurr() const { return m_bUsedByCurr; }
128	Void setUsedByCurr( Bool bUsed ) { m_bUsedByCurr = bUsed; }
129	Bool getIsLongTerm() const { return m_bIsLongTerm; }
130	Void setIsLongTerm( Bool lt ) { m_bIsLongTerm = lt; }
131	Void setCheckLTMSBPresent (Bool b ) {m_bCheckLTMSB=b;}
132	Bool getCheckLTMSBPresent () { return m_bCheckLTMSB;}
133
134	TComPicSym* getPicSym() { return &m_picSym; }
135	const TComPicSym* getPicSym() const { return &m_picSym; }
136	TComSlice* getSlice(Int i) { return m_picSym.getSlice(i); }
137	const TComSlice* getSlice(Int i) const { return m_picSym.getSlice(i); }
138	Int getPOC() const { return m_picSym.getSlice(m_uiCurrSliceIdx)->getPOC(); }
139	TComDataCU* getCtu( UInt ctuRsAddr ) { return m_picSym.getCtu( ctuRsAddr ); }
140	const TComDataCU* getCtu( UInt ctuRsAddr ) const { return m_picSym.getCtu( ctuRsAddr ); }
141
142	TComPicYuv* getPicYuvOrg() { return m_apcPicYuv[PIC_YUV_ORG]; }
143	TComPicYuv* getPicYuvRec() { return m_apcPicYuv[PIC_YUV_REC]; }
144
145	TComPicYuv* getPicYuvPred() { return m_pcPicYuvPred; }
146	TComPicYuv* getPicYuvResi() { return m_pcPicYuvResi; }
147	Void setPicYuvPred( TComPicYuv* pcPicYuv ) { m_pcPicYuvPred = pcPicYuv; }
148	Void setPicYuvResi( TComPicYuv* pcPicYuv ) { m_pcPicYuvResi = pcPicYuv; }
149
150	UInt getNumberOfCtusInFrame() const { return m_picSym.getNumberOfCtusInFrame(); }
151	UInt getNumPartInCtuWidth() const { return m_picSym.getNumPartInCtuWidth(); }
152	UInt getNumPartInCtuHeight() const { return m_picSym.getNumPartInCtuHeight(); }
153	UInt getNumPartitionsInCtu() const { return m_picSym.getNumPartitionsInCtu(); }
154	UInt getFrameWidthInCtus() const { return m_picSym.getFrameWidthInCtus(); }
155	UInt getFrameHeightInCtus() const { return m_picSym.getFrameHeightInCtus(); }
156	UInt getMinCUWidth() const { return m_picSym.getMinCUWidth(); }
157	UInt getMinCUHeight() const { return m_picSym.getMinCUHeight(); }
158
159	Int getStride(const ComponentID id) const { return m_apcPicYuv[PIC_YUV_REC]->getStride(id); }
160	Int getComponentScaleX(const ComponentID id) const { return m_apcPicYuv[PIC_YUV_REC]->getComponentScaleX(id); }
161	Int getComponentScaleY(const ComponentID id) const { return m_apcPicYuv[PIC_YUV_REC]->getComponentScaleY(id); }
162	ChromaFormat getChromaFormat() const { return m_apcPicYuv[PIC_YUV_REC]->getChromaFormat(); }
163	Int getNumberValidComponents() const { return m_apcPicYuv[PIC_YUV_REC]->getNumberValidComponents(); }
164
165	Void setReconMark (Bool b) { m_bReconstructed = b; }
166	Bool getReconMark () const { return m_bReconstructed; }
167	Void setOutputMark (Bool b) { m_bNeededForOutput = b; }
168	Bool getOutputMark () const { return m_bNeededForOutput; }
169
170	Void compressMotion();
171	UInt getCurrSliceIdx() const { return m_uiCurrSliceIdx; }
172	Void setCurrSliceIdx(UInt i) { m_uiCurrSliceIdx = i; }
173	UInt getNumAllocatedSlice() const {return m_picSym.getNumAllocatedSlice();}
174	Void allocateNewSlice() {m_picSym.allocateNewSlice(); }
175	Void clearSliceBuffer() {m_picSym.clearSliceBuffer(); }
176
177	const Window& getConformanceWindow() const { return m_picSym.getSPS().getConformanceWindow(); }
178	Window getDefDisplayWindow() const { return m_picSym.getSPS().getVuiParametersPresentFlag() ? m_picSym.getSPS().getVuiParameters()->getDefaultDisplayWindow() : Window(); }
179
180	Bool getSAOMergeAvailability(Int currAddr, Int mergeAddr);
181
182	UInt getSubstreamForCtuAddr(const UInt ctuAddr, const Bool bAddressInRaster, TComSlice *pcSlice);
183
184	/* field coding parameters*/
185
186	Void setTopField(Bool b) {m_isTop = b;}
187	Bool isTopField() {return m_isTop;}
188	Void setField(Bool b) {m_isField = b;}
189	Bool isField() {return m_isField;}
190
191	/** transfer ownership of seis to this picture */
192	Void setSEIs(SEIMessages& seis) { m_SEIs = seis; }
193
194	/**
195	* return the current list of SEI messages associated with this picture.
196	* Pointer is valid until this->destroy() is called */
197	SEIMessages& getSEIs() { return m_SEIs; }
198
199	/**
200	* return the current list of SEI messages associated with this picture.
201	* Pointer is valid until this->destroy() is called */
202	const SEIMessages& getSEIs() const { return m_SEIs; }
203
204	#if SVC_EXTENSION
205	Void setLayerId(UInt layerId) { m_layerId = layerId; }
206	UInt getLayerId() const { return m_layerId; }
207	UInt getLayerIdx() const { return m_picSym.getSlice(0)->getVPS()->getLayerIdxInVps(m_layerId); }
208	Bool requireResampling(UInt refLayerIdc) const { return m_requireResampling[refLayerIdc]; }
209	Void setRequireResamplingFlag (UInt refLayerIdc, Bool b) { m_requireResampling[refLayerIdc] = b; }
210	Void setFullPelBaseRec (UInt refLayerIdc, TComPicYuv* p) { m_pcFullPelBaseRec[refLayerIdc] = p; }
211	TComPicYuv* getFullPelBaseRec (UInt refLayerIdc) const { return m_pcFullPelBaseRec[refLayerIdc]; }
212	Bool isILR( UInt currLayerId ) const { return ( m_bIsLongTerm && m_layerId < currLayerId ); }
213	Bool equalPictureSizeAndOffsetFlag(UInt refLayerIdc) const { return m_equalPictureSizeAndOffsetFlag[refLayerIdc]; }
214	Void setEqualPictureSizeAndOffsetFlag(UInt refLayerIdc, Bool b) { m_equalPictureSizeAndOffsetFlag[refLayerIdc] = b; }
215	Void copyUpsampledMvField(UInt refLayerIdc, Int mvScalingFactor, Int posScalingFactor);
216	Void initUpsampledMvField();
217	Bool checkSameRefInfo();
218	Void copyUpsampledPictureYuv(TComPicYuv* pcPicYuvIn, TComPicYuv* pcPicYuvOut);
219	Void setMvScalingFactor(UInt refLayerIdc, Int compX, Int compY) { m_mvScalingFactor[0][refLayerIdc] = compX; m_mvScalingFactor[1][refLayerIdc] = compY; }
220	Void setPosScalingFactor(UInt refLayerIdc, Int compX, Int compY) { m_posScalingFactor[0][refLayerIdc] = compX; m_posScalingFactor[1][refLayerIdc] = compY; }
221	Int getMvScalingFactor(UInt refLayerIdc, UChar comp) const { return m_mvScalingFactor[comp][refLayerIdc]; }
222	Int getPosScalingFactor(UInt refLayerIdc, UChar comp) const { return m_posScalingFactor[comp][refLayerIdc]; }
223	Int** getPosScalingFactor() { return m_posScalingFactor; }
224	Int** getMvScalingFactor() { return m_mvScalingFactor; }
225	Void createMvScalingFactor(UInt numOfILRPs);
226	Void createPosScalingFactor(UInt numOfILRPs);
227	#if CGS_3D_ASYMLUT
228	Void setFrameBit( Int n ) { m_nFrameBit = n; }
229	Int getFrameBit() const { return m_nFrameBit; }
230	#endif
231	Bool isCurrAu() const { return m_currAuFlag; }
232	Void setCurrAuFlag(Bool x) { m_currAuFlag = x; }
233	#endif //SVC_EXTENSION
234	};// END CLASS DEFINITION TComPic
235
236	//! \}
237
238	#endif // __TCOMPIC__

Note: See TracBrowser for help on using the repository browser.

Download in other formats: