Mercurial > hg > forks > libbpg
diff x265/source/encoder/motion.h @ 0:772086c29cc7
Initial import.
author | Matti Hamalainen <ccr@tnsp.org> |
---|---|
date | Wed, 16 Nov 2016 11:16:33 +0200 |
parents | |
children |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/x265/source/encoder/motion.h Wed Nov 16 11:16:33 2016 +0200 @@ -0,0 +1,110 @@ +/***************************************************************************** + * Copyright (C) 2013 x265 project + * + * Authors: Steve Borho <steve@borho.org> + * + * This program is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 2 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02111, USA. + * + * This program is also available under a commercial proprietary license. + * For more information, contact us at license @ x265.com. + *****************************************************************************/ + +#ifndef X265_MOTIONESTIMATE_H +#define X265_MOTIONESTIMATE_H + +#include "primitives.h" +#include "reference.h" +#include "mv.h" +#include "bitcost.h" +#include "yuv.h" + +namespace X265_NS { +// private x265 namespace + +class MotionEstimate : public BitCost +{ +protected: + + intptr_t blockOffset; + + int ctuAddr; + int absPartIdx; // part index of PU, including CU offset within CTU + + int searchMethod; + int subpelRefine; + + int blockwidth; + int blockheight; + + pixelcmp_t sad; + pixelcmp_x3_t sad_x3; + pixelcmp_x4_t sad_x4; + pixelcmp_t satd; + pixelcmp_t chromaSatd; + + MotionEstimate& operator =(const MotionEstimate&); + +public: + + static const int COST_MAX = 1 << 28; + + Yuv fencPUYuv; + int partEnum; + bool bChromaSATD; + + MotionEstimate(); + ~MotionEstimate(); + + static void initScales(); + static int hpelIterationCount(int subme); + void init(int method, int refine, int csp); + + /* Methods called at slice setup */ + + void setSourcePU(pixel *fencY, intptr_t stride, intptr_t offset, int pwidth, int pheight); + void setSourcePU(const Yuv& srcFencYuv, int ctuAddr, int cuPartIdx, int puPartIdx, int pwidth, int pheight); + + /* buf*() and motionEstimate() methods all use cached fenc pixels and thus + * require setSourcePU() to be called prior. */ + + inline int bufSAD(const pixel* fref, intptr_t stride) { return sad(fencPUYuv.m_buf[0], FENC_STRIDE, fref, stride); } + + inline int bufSATD(const pixel* fref, intptr_t stride) { return satd(fencPUYuv.m_buf[0], FENC_STRIDE, fref, stride); } + + inline int bufChromaSATD(const Yuv& refYuv, int puPartIdx) + { + return chromaSatd(refYuv.getCbAddr(puPartIdx), refYuv.m_csize, fencPUYuv.m_buf[1], fencPUYuv.m_csize) + + chromaSatd(refYuv.getCrAddr(puPartIdx), refYuv.m_csize, fencPUYuv.m_buf[2], fencPUYuv.m_csize); + } + + int motionEstimate(ReferencePlanes* ref, const MV & mvmin, const MV & mvmax, const MV & qmvp, int numCandidates, const MV * mvc, int merange, MV & outQMv); + + int subpelCompare(ReferencePlanes* ref, const MV &qmv, pixelcmp_t); + +protected: + + inline void StarPatternSearch(ReferencePlanes *ref, + const MV & mvmin, + const MV & mvmax, + MV & bmv, + int & bcost, + int & bPointNr, + int & bDistance, + int earlyExitIters, + int merange); +}; +} + +#endif // ifndef X265_MOTIONESTIMATE_H