/src/x265/source/common/shortyuv.h
Line | Count | Source (jump to first uncovered line) |
1 | | /***************************************************************************** |
2 | | * x265: ShortYUV class for short sized YUV-style frames |
3 | | ***************************************************************************** |
4 | | * Copyright (C) 2013-2020 MulticoreWare, Inc |
5 | | * |
6 | | * Authors: Deepthi Nandakumar <deepthi@multicorewareinc.com> |
7 | | * |
8 | | * This program is free software; you can redistribute it and/or modify |
9 | | * it under the terms of the GNU General Public License as published by |
10 | | * the Free Software Foundation; either version 2 of the License, or |
11 | | * (at your option) any later version. |
12 | | * |
13 | | * This program is distributed in the hope that it will be useful, |
14 | | * but WITHOUT ANY WARRANTY; without even the implied warranty of |
15 | | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
16 | | * GNU General Public License for more details. |
17 | | * |
18 | | * You should have received a copy of the GNU General Public License |
19 | | * along with this program; if not, write to the Free Software |
20 | | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02111, USA. |
21 | | * |
22 | | * This program is also available under a commercial proprietary license. |
23 | | * For more information, contact us at license @ x265.com |
24 | | *****************************************************************************/ |
25 | | |
26 | | #ifndef X265_SHORTYUV_H |
27 | | #define X265_SHORTYUV_H |
28 | | |
29 | | #include "common.h" |
30 | | |
31 | | namespace X265_NS { |
32 | | // private namespace |
33 | | |
34 | | class Yuv; |
35 | | |
36 | | /* A ShortYuv instance holds int16_ts for a square CU (64x64 down to 8x8) for all three planes, |
37 | | * these are typically used to hold residual or coefficients */ |
38 | | class ShortYuv |
39 | | { |
40 | | public: |
41 | | |
42 | | int16_t* m_buf[3]; |
43 | | |
44 | | uint32_t m_size; |
45 | | uint32_t m_csize; |
46 | | |
47 | | int m_csp; |
48 | | int m_hChromaShift; |
49 | | int m_vChromaShift; |
50 | | |
51 | | ShortYuv(); |
52 | | |
53 | | bool create(uint32_t size, int csp); |
54 | | void destroy(); |
55 | | void clear(); |
56 | | |
57 | 5.31M | int16_t* getLumaAddr(uint32_t absPartIdx) { return m_buf[0] + getAddrOffset(absPartIdx, m_size); } |
58 | 0 | int16_t* getCbAddr(uint32_t absPartIdx) { return m_buf[1] + getChromaAddrOffset(absPartIdx); } |
59 | 0 | int16_t* getCrAddr(uint32_t absPartIdx) { return m_buf[2] + getChromaAddrOffset(absPartIdx); } |
60 | 7.73M | int16_t* getChromaAddr(uint32_t chromaId, uint32_t partUnitIdx) { return m_buf[chromaId] + getChromaAddrOffset(partUnitIdx); } |
61 | | |
62 | 0 | const int16_t* getLumaAddr(uint32_t absPartIdx) const { return m_buf[0] + getAddrOffset(absPartIdx, m_size); } |
63 | 0 | const int16_t* getCbAddr(uint32_t absPartIdx) const { return m_buf[1] + getChromaAddrOffset(absPartIdx); } |
64 | 0 | const int16_t* getCrAddr(uint32_t absPartIdx) const { return m_buf[2] + getChromaAddrOffset(absPartIdx); } |
65 | 0 | const int16_t* getChromaAddr(uint32_t chromaId, uint32_t partUnitIdx) const { return m_buf[chromaId] + getChromaAddrOffset(partUnitIdx); } |
66 | | |
67 | | void subtract(const Yuv& srcYuv0, const Yuv& srcYuv1, uint32_t log2Size, int picCsp); |
68 | | |
69 | | void copyPartToPartLuma(ShortYuv& dstYuv, uint32_t absPartIdx, uint32_t log2Size) const; |
70 | | void copyPartToPartChroma(ShortYuv& dstYuv, uint32_t absPartIdx, uint32_t log2SizeL) const; |
71 | | |
72 | | void copyPartToPartLuma(Yuv& dstYuv, uint32_t absPartIdx, uint32_t log2Size) const; |
73 | | void copyPartToPartChroma(Yuv& dstYuv, uint32_t absPartIdx, uint32_t log2SizeL) const; |
74 | | |
75 | | int getChromaAddrOffset(uint32_t idx) const |
76 | 7.75M | { |
77 | 7.75M | int blkX = g_zscanToPelX[idx] >> m_hChromaShift; |
78 | 7.75M | int blkY = g_zscanToPelY[idx] >> m_vChromaShift; |
79 | | |
80 | 7.75M | return blkX + blkY * m_csize; |
81 | 7.75M | } |
82 | | |
83 | | static int getAddrOffset(uint32_t idx, uint32_t width) |
84 | 5.33M | { |
85 | 5.33M | int blkX = g_zscanToPelX[idx]; |
86 | 5.33M | int blkY = g_zscanToPelY[idx]; |
87 | | |
88 | 5.33M | return blkX + blkY * width; |
89 | 5.33M | } |
90 | | }; |
91 | | } |
92 | | |
93 | | #endif // ifndef X265_SHORTYUV_H |