Context navigation

TEncTop.h @ 171

Visit:

Last change on this file since 171 was 166, checked in by mitsubishi-htm, 12 years ago

Initial integration of VSP into HTM 4.0.1. The version used for JCT3V-B0102 at Shanghai meeting.

VC9 project/solution files updated. Other Visual C++ project/solution files are not updated.
Linux make file updated.

TODO

A second release is expected to include some bug fix and improvements on the interface, e.g. to move switches from macro definition to the configuration file.
A third release is expected after being integrated within HTM 5.x, which is to be used for CE1.h anchor.

Property svn:eol-style set to native

File size: 13.0 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-2012, 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 TEncTop.h
35	\brief encoder class (header)
36	*/
37
38	#ifndef __TENCTOP__
39	#define __TENCTOP__
40
41	// Include files
42	#include "TLibCommon/TComList.h"
43	#include "TLibCommon/TComPrediction.h"
44	#include "TLibCommon/TComTrQuant.h"
45	#include "TLibCommon/AccessUnit.h"
46	#include "TLibCommon/TComDepthMapGenerator.h"
47	#include "../TLibCommon/TComResidualGenerator.h"
48
49	#include "TLibVideoIO/TVideoIOYuv.h"
50
51	#include "TEncCfg.h"
52	#include "TEncGOP.h"
53	#include "TEncSlice.h"
54	#include "TEncEntropy.h"
55	#include "TEncCavlc.h"
56	#include "TEncSbac.h"
57	#include "TEncSearch.h"
58	#include "TEncAdaptiveLoopFilter.h"
59	#include "TEncSampleAdaptiveOffset.h"
60	#include "TEncPreanalyzer.h"
61
62	//! \ingroup TLibEncoder
63	//! \{
64
65	// ====================================================================================================================
66	// Class definition
67	// ====================================================================================================================
68
69	class TAppEncTop;
70
71	/// encoder class
72	class TEncTop : public TEncCfg
73	{
74	private:
75	// picture
76	Int m_iPOCLast; ///< time index (POC)
77	Int m_iNumPicRcvd; ///< number of received pictures
78	UInt m_uiNumAllPicCoded; ///< number of coded pictures
79	TComList<TComPic*> m_cListPic; ///< dynamic list of pictures
80
81	// encoder search
82	TEncSearch m_cSearch; ///< encoder search class
83	TEncEntropy* m_pcEntropyCoder; ///< entropy encoder
84	TEncCavlc* m_pcCavlcCoder; ///< CAVLC encoder
85	// coding tool
86	TComTrQuant m_cTrQuant; ///< transform & quantization class
87	TComLoopFilter m_cLoopFilter; ///< deblocking filter class
88	TEncSampleAdaptiveOffset m_cEncSAO; ///< sample adaptive offset class
89	TEncAdaptiveLoopFilter m_cAdaptiveLoopFilter; ///< adaptive loop filter class
90	TEncEntropy m_cEntropyCoder; ///< entropy encoder
91	TEncCavlc m_cCavlcCoder; ///< CAVLC encoder
92	TEncSbac m_cSbacCoder; ///< SBAC encoder
93	TEncBinCABAC m_cBinCoderCABAC; ///< bin coder CABAC
94	TEncSbac* m_pcSbacCoders; ///< SBAC encoders (to encode substreams )
95	TEncBinCABAC* m_pcBinCoderCABACs; ///< bin coders CABAC (one per substream)
96
97	// processing unit
98	TEncGOP m_cGOPEncoder; ///< GOP encoder
99	TEncSlice m_cSliceEncoder; ///< slice encoder
100	TEncCu m_cCuEncoder; ///< CU encoder
101
102	#if DEPTH_MAP_GENERATION
103	TComDepthMapGenerator m_cDepthMapGenerator; ///< depth map generator
104	#endif
105	#if HHI_INTER_VIEW_RESIDUAL_PRED
106	TComResidualGenerator m_cResidualGenerator; ///< generator for residual pictures
107	#endif
108	// SPS
109	TComSPS m_cSPS; ///< SPS
110	TComPPS m_cPPS; ///< PPS
111	std::vector<TComAPS> m_vAPS; //!< APS container
112	TComRPSList m_RPSList; ///< RPS
113
114	// RD cost computation
115	TComBitCounter m_cBitCounter; ///< bit counter for RD optimization
116	TComRdCost m_cRdCost; ///< RD cost computation class
117	TEncSbac*** m_pppcRDSbacCoder; ///< temporal storage for RD computation
118	TEncSbac m_cRDGoOnSbacCoder; ///< going on SBAC model for RD stage
119	#if FAST_BIT_EST
120	TEncBinCABACCounter*** m_pppcBinCoderCABAC; ///< temporal CABAC state storage for RD computation
121	TEncBinCABACCounter m_cRDGoOnBinCoderCABAC; ///< going on bin coder CABAC for RD stage
122	#else
123	TEncBinCABAC*** m_pppcBinCoderCABAC; ///< temporal CABAC state storage for RD computation
124	TEncBinCABAC m_cRDGoOnBinCoderCABAC; ///< going on bin coder CABAC for RD stage
125	#endif
126	Int m_iNumSubstreams; ///< # of top-level elements allocated.
127	TComBitCounter* m_pcBitCounters; ///< bit counters for RD optimization per substream
128	TComRdCost* m_pcRdCosts; ///< RD cost computation class per substream
129	TEncSbac**** m_ppppcRDSbacCoders; ///< temporal storage for RD computation per substream
130	TEncSbac* m_pcRDGoOnSbacCoders; ///< going on SBAC model for RD stage per substream
131	TEncBinCABAC**** m_ppppcBinCodersCABAC; ///< temporal CABAC state storage for RD computation per substream
132	TEncBinCABAC* m_pcRDGoOnBinCodersCABAC; ///< going on bin coder CABAC for RD stage per substream
133
134	// quality control
135	TEncPreanalyzer m_cPreanalyzer; ///< image characteristics analyzer for TM5-step3-like adaptive QP
136
137	TComScalingList m_scalingList; ///< quantization matrix information
138
139	std::vector<TEncTop> m_pacTEncTopList;
140	TAppEncTop* m_pcTAppEncTop;
141
142	TEncAnalyze m_cAnalyzeAll;
143	TEncAnalyze m_cAnalyzeI;
144	TEncAnalyze m_cAnalyzeP;
145	TEncAnalyze m_cAnalyzeB;
146
147	#if VSP_N
148	TComPic* m_pcPicVSP; ///< view synthesis prediction buffer for current POC to be coded
149	TComPic* m_pcPicAvail;
150	#endif
151
152	#if VSP_SLICE_HEADER
153	Bool m_bUseVSP;
154	#endif
155
156	protected:
157	Void xGetNewPicBuffer ( TComPic*& rpcPic ); ///< get picture buffer which will be processed
158	Void xInitSPS (); ///< initialize SPS from encoder options
159	Void xInitPPS (); ///< initialize PPS from encoder options
160
161	Void xInitPPSforTiles ();
162	Void xInitRPS (); ///< initialize PPS from encoder options
163	Void xInitSPSforInterViewRefs();
164
165	public:
166	TEncTop();
167	virtual ~TEncTop();
168
169	Void create ();
170	Void destroy ();
171	Void init ( TAppEncTop* pcTAppEncTop );
172	Void deletePicBuffer ();
173
174	UInt getFrameId (Int iGOPid) {
175	if(m_iPOCLast == 0)
176	{
177	return(0 );
178	}
179	else
180	{
181	return m_iPOCLast -m_iNumPicRcvd+ getGOPEntry(iGOPid).m_POC ;
182	}
183	}
184	#if HHI_INTERVIEW_SKIP \|\| HHI_INTER_VIEW_MOTION_PRED \|\| HHI_INTER_VIEW_RESIDUAL_PRED
185	Void deleteExtraPicBuffers ( Int iPoc );
186	#endif
187
188	Void compressMotion ( Int iPoc );
189
190	Void initNewPic(TComPicYuv* pcPicYuvOrg, TComPicYuv* pcOrgPdmDepth = 0);
191
192	Void createWPPCoders(Int iNumSubstreams);
193
194	// -------------------------------------------------------------------------------------------------------------------
195	// member access functions
196	// -------------------------------------------------------------------------------------------------------------------
197
198	TComList<TComPic> getListPic () { return &m_cListPic; }
199	TEncSearch* getPredSearch () { return &m_cSearch; }
200
201	TComTrQuant* getTrQuant () { return &m_cTrQuant; }
202	TComLoopFilter* getLoopFilter () { return &m_cLoopFilter; }
203	TEncAdaptiveLoopFilter* getAdaptiveLoopFilter () { return &m_cAdaptiveLoopFilter; }
204	TEncSampleAdaptiveOffset* getSAO () { return &m_cEncSAO; }
205	TEncGOP* getGOPEncoder () { return &m_cGOPEncoder; }
206	TEncSlice* getSliceEncoder () { return &m_cSliceEncoder; }
207	TEncCu* getCuEncoder () { return &m_cCuEncoder; }
208	TEncEntropy* getEntropyCoder () { return &m_cEntropyCoder; }
209	TEncCavlc* getCavlcCoder () { return &m_cCavlcCoder; }
210	TEncSbac* getSbacCoder () { return &m_cSbacCoder; }
211	TEncBinCABAC* getBinCABAC () { return &m_cBinCoderCABAC; }
212	TEncSbac* getSbacCoders () { return m_pcSbacCoders; }
213	TEncBinCABAC* getBinCABACs () { return m_pcBinCoderCABACs; }
214
215	TComBitCounter* getBitCounter () { return &m_cBitCounter; }
216	TComRdCost* getRdCost () { return &m_cRdCost; }
217	TEncSbac*** getRDSbacCoder () { return m_pppcRDSbacCoder; }
218	TEncSbac* getRDGoOnSbacCoder () { return &m_cRDGoOnSbacCoder; }
219	TComBitCounter* getBitCounters () { return m_pcBitCounters; }
220	TComRdCost* getRdCosts () { return m_pcRdCosts; }
221	TEncSbac**** getRDSbacCoders () { return m_ppppcRDSbacCoders; }
222	TEncSbac* getRDGoOnSbacCoders () { return m_pcRDGoOnSbacCoders; }
223
224	Void setTEncTopList ( std::vector<TEncTop> pacTEncTopList );
225	TAppEncTop* getEncTop () { return m_pcTAppEncTop; }
226	Int getNumAllPicCoded () { return m_uiNumAllPicCoded; }
227	Void printOutSummary ( UInt uiNumAllPicCoded );
228
229	TComSPS* getSPS () { return &m_cSPS; }
230	TComPPS* getPPS () { return &m_cPPS; }
231	std::vector<TComAPS>& getAPS () { return m_vAPS; }
232	TComRPSList* getRPSList () { return &m_RPSList; }
233
234	Void selectReferencePictureSet(TComSlice* slice, Int POCCurr, Int GOPid,TComList<TComPic*>& listPic );
235	TComScalingList* getScalingList () { return &m_scalingList; }
236
237	TEncAnalyze* getAnalyzeAll () { return &m_cAnalyzeAll; }
238	TEncAnalyze* getAnalyzeI () { return &m_cAnalyzeI; }
239	TEncAnalyze* getAnalyzeP () { return &m_cAnalyzeP; }
240	TEncAnalyze* getAnalyzeB () { return &m_cAnalyzeB; }
241
242	#if DEPTH_MAP_GENERATION
243	TComDepthMapGenerator* getDepthMapGenerator () { return &m_cDepthMapGenerator; }
244	#endif
245	#if HHI_INTER_VIEW_RESIDUAL_PRED
246	TComResidualGenerator* getResidualGenerator () { return &m_cResidualGenerator; }
247	#endif
248	#if VSP_N
249	TComPic* getVSPBuf () { return m_pcPicVSP; }
250	TComPic* getVSPAvailBuf () { return m_pcPicAvail; }
251	#endif
252
253	#if VSP_SLICE_HEADER
254	Void setUseVSP ( Bool val ) { m_bUseVSP = val; }
255	Bool getUseVSP () { return m_bUseVSP; }
256	#endif
257
258	// -------------------------------------------------------------------------------------------------------------------
259	// encoder function
260	// -------------------------------------------------------------------------------------------------------------------
261
262	/// encode several number of pictures until end-of-sequence
263	Void encode( bool bEos, TComPicYuv* pcPicYuvOrg, TComList<TComPicYuv*>& rcListPicYuvRecOut,
264	std::list<AccessUnit>& accessUnitsOut, Int& iNumEncoded, Int gopId );
265	};
266
267	//! \}
268
269	#endif // __TENCTOP__

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

JCT-3V 3D-HEVC

Context navigation

source: 3DVCSoftware/branches/HTM-4.0.1-VSP-dev0/source/Lib/TLibEncoder/TEncTop.h @ 171

Download in other formats: