9c7abee032a5018b598548129533a96904ee59e4
[deb_x265.git] / source / common / frame.cpp
1 /*****************************************************************************
2 * Copyright (C) 2013 x265 project
3 *
4 * Author: Steve Borho <steve@borho.org>
5 *
6 * This program is free software; you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License as published by
8 * the Free Software Foundation; either version 2 of the License, or
9 * (at your option) any later version.
10 *
11 * This program is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU General Public License for more details.
15 *
16 * You should have received a copy of the GNU General Public License
17 * along with this program; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02111, USA.
19 *
20 * This program is also available under a commercial proprietary license.
21 * For more information, contact us at license @ x265.com.
22 *****************************************************************************/
23
24 #include "common.h"
25 #include "frame.h"
26 #include "picyuv.h"
27 #include "framedata.h"
28
29 using namespace x265;
30
31 Frame::Frame()
32 {
33 m_bChromaExtended = false;
34 m_reconRowCount.set(0);
35 m_countRefEncoders = 0;
36 m_encData = NULL;
37 m_reconPic = NULL;
38 m_next = NULL;
39 m_prev = NULL;
40 memset(&m_lowres, 0, sizeof(m_lowres));
41 }
42
43 bool Frame::create(x265_param *param)
44 {
45 m_fencPic = new PicYuv;
46
47 return m_fencPic->create(param->sourceWidth, param->sourceHeight, param->internalCsp) &&
48 m_lowres.create(m_fencPic, param->bframes, !!param->rc.aqMode);
49 }
50
51 bool Frame::allocEncodeData(x265_param *param, const SPS& sps)
52 {
53 m_encData = new FrameData;
54 m_reconPic = new PicYuv;
55 m_encData->m_reconPic = m_reconPic;
56 bool ok = m_encData->create(param, sps) && m_reconPic->create(param->sourceWidth, param->sourceHeight, param->internalCsp);
57 if (ok)
58 {
59 /* initialize right border of m_reconpicYuv as SAO may read beyond the
60 * end of the picture accessing uninitialized pixels */
61 int maxHeight = sps.numCuInHeight * g_maxCUSize;
62 memset(m_reconPic->m_picOrg[0], 0, m_reconPic->m_stride * maxHeight);
63 memset(m_reconPic->m_picOrg[1], 0, m_reconPic->m_strideC * (maxHeight >> m_reconPic->m_vChromaShift));
64 memset(m_reconPic->m_picOrg[2], 0, m_reconPic->m_strideC * (maxHeight >> m_reconPic->m_vChromaShift));
65 }
66 return ok;
67 }
68
69 /* prepare to re-use a FrameData instance to encode a new picture */
70 void Frame::reinit(const SPS& sps)
71 {
72 m_bChromaExtended = false;
73 m_reconPic = m_encData->m_reconPic;
74 m_encData->reinit(sps);
75 }
76
77 void Frame::destroy()
78 {
79 if (m_encData)
80 {
81 m_encData->destroy();
82 delete m_encData;
83 m_encData = NULL;
84 }
85
86 if (m_fencPic)
87 {
88 m_fencPic->destroy();
89 delete m_fencPic;
90 m_fencPic = NULL;
91 }
92
93 if (m_reconPic)
94 {
95 m_reconPic->destroy();
96 delete m_reconPic;
97 m_reconPic = NULL;
98 }
99
100 m_lowres.destroy();
101 }