Line |
Branch |
Exec |
Source |
1 |
|
|
/* |
2 |
|
|
* Copyright (c) 2020 Paul B Mahol |
3 |
|
|
* |
4 |
|
|
* This file is part of FFmpeg. |
5 |
|
|
* |
6 |
|
|
* FFmpeg is free software; you can redistribute it and/or |
7 |
|
|
* modify it under the terms of the GNU Lesser General Public |
8 |
|
|
* License as published by the Free Software Foundation; either |
9 |
|
|
* version 2.1 of the License, or (at your option) any later version. |
10 |
|
|
* |
11 |
|
|
* FFmpeg 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 GNU |
14 |
|
|
* Lesser General Public License for more details. |
15 |
|
|
* |
16 |
|
|
* You should have received a copy of the GNU Lesser General Public |
17 |
|
|
* License along with FFmpeg; if not, write to the Free Software |
18 |
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
19 |
|
|
*/ |
20 |
|
|
|
21 |
|
|
/** |
22 |
|
|
* @file |
23 |
|
|
* Cineform HD video encoder |
24 |
|
|
*/ |
25 |
|
|
|
26 |
|
|
#include <stdlib.h> |
27 |
|
|
#include <string.h> |
28 |
|
|
|
29 |
|
|
#include "libavutil/imgutils.h" |
30 |
|
|
#include "libavutil/opt.h" |
31 |
|
|
|
32 |
|
|
#include "avcodec.h" |
33 |
|
|
#include "bytestream.h" |
34 |
|
|
#include "cfhd.h" |
35 |
|
|
#include "cfhdencdsp.h" |
36 |
|
|
#include "codec_internal.h" |
37 |
|
|
#include "encode.h" |
38 |
|
|
#include "put_bits.h" |
39 |
|
|
#include "thread.h" |
40 |
|
|
|
41 |
|
|
/* Derived from existing tables from decoder */ |
42 |
|
|
static const unsigned codebook[256][2] = { |
43 |
|
|
{ 1, 0x00000000 }, { 2, 0x00000002 }, { 3, 0x00000007 }, { 5, 0x00000019 }, { 6, 0x00000030 }, |
44 |
|
|
{ 6, 0x00000036 }, { 7, 0x00000063 }, { 7, 0x0000006B }, { 7, 0x0000006F }, { 8, 0x000000D4 }, |
45 |
|
|
{ 8, 0x000000DC }, { 9, 0x00000189 }, { 9, 0x000001A0 }, { 9, 0x000001AB }, {10, 0x00000310 }, |
46 |
|
|
{10, 0x00000316 }, {10, 0x00000354 }, {10, 0x00000375 }, {10, 0x00000377 }, {11, 0x00000623 }, |
47 |
|
|
{11, 0x00000684 }, {11, 0x000006AB }, {11, 0x000006EC }, {12, 0x00000C44 }, {12, 0x00000C5C }, |
48 |
|
|
{12, 0x00000C5E }, {12, 0x00000D55 }, {12, 0x00000DD1 }, {12, 0x00000DD3 }, {12, 0x00000DDB }, |
49 |
|
|
{13, 0x0000188B }, {13, 0x000018BB }, {13, 0x00001AA8 }, {13, 0x00001BA0 }, {13, 0x00001BA4 }, |
50 |
|
|
{13, 0x00001BB5 }, {14, 0x00003115 }, {14, 0x00003175 }, {14, 0x0000317D }, {14, 0x00003553 }, |
51 |
|
|
{14, 0x00003768 }, {15, 0x00006228 }, {15, 0x000062E8 }, {15, 0x000062F8 }, {15, 0x00006AA4 }, |
52 |
|
|
{15, 0x00006E85 }, {15, 0x00006E87 }, {15, 0x00006ED3 }, {16, 0x0000C453 }, {16, 0x0000C5D3 }, |
53 |
|
|
{16, 0x0000C5F3 }, {16, 0x0000DD08 }, {16, 0x0000DD0C }, {16, 0x0000DDA4 }, {17, 0x000188A4 }, |
54 |
|
|
{17, 0x00018BA5 }, {17, 0x00018BE5 }, {17, 0x0001AA95 }, {17, 0x0001AA97 }, {17, 0x0001BA13 }, |
55 |
|
|
{17, 0x0001BB4A }, {17, 0x0001BB4B }, {18, 0x00031748 }, {18, 0x000317C8 }, {18, 0x00035528 }, |
56 |
|
|
{18, 0x0003552C }, {18, 0x00037424 }, {18, 0x00037434 }, {18, 0x00037436 }, {19, 0x00062294 }, |
57 |
|
|
{19, 0x00062E92 }, {19, 0x00062F92 }, {19, 0x0006AA52 }, {19, 0x0006AA5A }, {19, 0x0006E84A }, |
58 |
|
|
{19, 0x0006E86A }, {19, 0x0006E86E }, {20, 0x000C452A }, {20, 0x000C5D27 }, {20, 0x000C5F26 }, |
59 |
|
|
{20, 0x000D54A6 }, {20, 0x000D54B6 }, {20, 0x000DD096 }, {20, 0x000DD0D6 }, {20, 0x000DD0DE }, |
60 |
|
|
{21, 0x00188A56 }, {21, 0x0018BA4D }, {21, 0x0018BE4E }, {21, 0x0018BE4F }, {21, 0x001AA96E }, |
61 |
|
|
{21, 0x001BA12E }, {21, 0x001BA12F }, {21, 0x001BA1AF }, {21, 0x001BA1BF }, {22, 0x00317498 }, |
62 |
|
|
{22, 0x0035529C }, {22, 0x0035529D }, {22, 0x003552DE }, {22, 0x003552DF }, {22, 0x0037435D }, |
63 |
|
|
{22, 0x0037437D }, {23, 0x0062295D }, {23, 0x0062E933 }, {23, 0x006AA53D }, {23, 0x006AA53E }, |
64 |
|
|
{23, 0x006AA53F }, {23, 0x006E86B9 }, {23, 0x006E86F8 }, {24, 0x00C452B8 }, {24, 0x00C5D265 }, |
65 |
|
|
{24, 0x00D54A78 }, {24, 0x00D54A79 }, {24, 0x00DD0D70 }, {24, 0x00DD0D71 }, {24, 0x00DD0DF2 }, |
66 |
|
|
{24, 0x00DD0DF3 }, {26, 0x03114BA2 }, {25, 0x0188A5B1 }, {25, 0x0188A58B }, {25, 0x0188A595 }, |
67 |
|
|
{25, 0x0188A5D6 }, {25, 0x0188A5D7 }, {25, 0x0188A5A8 }, {25, 0x0188A5AE }, {25, 0x0188A5AF }, |
68 |
|
|
{25, 0x0188A5C4 }, {25, 0x0188A5C5 }, {25, 0x0188A587 }, {25, 0x0188A584 }, {25, 0x0188A585 }, |
69 |
|
|
{25, 0x0188A5C6 }, {25, 0x0188A5C7 }, {25, 0x0188A5CC }, {25, 0x0188A5CD }, {25, 0x0188A581 }, |
70 |
|
|
{25, 0x0188A582 }, {25, 0x0188A583 }, {25, 0x0188A5CE }, {25, 0x0188A5CF }, {25, 0x0188A5C2 }, |
71 |
|
|
{25, 0x0188A5C3 }, {25, 0x0188A5C1 }, {25, 0x0188A5B4 }, {25, 0x0188A5B5 }, {25, 0x0188A5E6 }, |
72 |
|
|
{25, 0x0188A5E7 }, {25, 0x0188A5E4 }, {25, 0x0188A5E5 }, {25, 0x0188A5AB }, {25, 0x0188A5E0 }, |
73 |
|
|
{25, 0x0188A5E1 }, {25, 0x0188A5E2 }, {25, 0x0188A5E3 }, {25, 0x0188A5B6 }, {25, 0x0188A5B7 }, |
74 |
|
|
{25, 0x0188A5FD }, {25, 0x0188A57E }, {25, 0x0188A57F }, {25, 0x0188A5EC }, {25, 0x0188A5ED }, |
75 |
|
|
{25, 0x0188A5FE }, {25, 0x0188A5FF }, {25, 0x0188A57D }, {25, 0x0188A59C }, {25, 0x0188A59D }, |
76 |
|
|
{25, 0x0188A5E8 }, {25, 0x0188A5E9 }, {25, 0x0188A5EA }, {25, 0x0188A5EB }, {25, 0x0188A5EF }, |
77 |
|
|
{25, 0x0188A57A }, {25, 0x0188A57B }, {25, 0x0188A578 }, {25, 0x0188A579 }, {25, 0x0188A5BA }, |
78 |
|
|
{25, 0x0188A5BB }, {25, 0x0188A5B8 }, {25, 0x0188A5B9 }, {25, 0x0188A588 }, {25, 0x0188A589 }, |
79 |
|
|
{25, 0x018BA4C8 }, {25, 0x018BA4C9 }, {25, 0x0188A5FA }, {25, 0x0188A5FB }, {25, 0x0188A5BC }, |
80 |
|
|
{25, 0x0188A5BD }, {25, 0x0188A598 }, {25, 0x0188A599 }, {25, 0x0188A5F4 }, {25, 0x0188A5F5 }, |
81 |
|
|
{25, 0x0188A59B }, {25, 0x0188A5DE }, {25, 0x0188A5DF }, {25, 0x0188A596 }, {25, 0x0188A597 }, |
82 |
|
|
{25, 0x0188A5F8 }, {25, 0x0188A5F9 }, {25, 0x0188A5F1 }, {25, 0x0188A58E }, {25, 0x0188A58F }, |
83 |
|
|
{25, 0x0188A5DC }, {25, 0x0188A5DD }, {25, 0x0188A5F2 }, {25, 0x0188A5F3 }, {25, 0x0188A58C }, |
84 |
|
|
{25, 0x0188A58D }, {25, 0x0188A5A4 }, {25, 0x0188A5F0 }, {25, 0x0188A5A5 }, {25, 0x0188A5A6 }, |
85 |
|
|
{25, 0x0188A5A7 }, {25, 0x0188A59A }, {25, 0x0188A5A2 }, {25, 0x0188A5A3 }, {25, 0x0188A58A }, |
86 |
|
|
{25, 0x0188A5B0 }, {25, 0x0188A5A0 }, {25, 0x0188A5A1 }, {25, 0x0188A5DA }, {25, 0x0188A5DB }, |
87 |
|
|
{25, 0x0188A59E }, {25, 0x0188A59F }, {25, 0x0188A5D8 }, {25, 0x0188A5EE }, {25, 0x0188A5D9 }, |
88 |
|
|
{25, 0x0188A5F6 }, {25, 0x0188A5F7 }, {25, 0x0188A57C }, {25, 0x0188A5C8 }, {25, 0x0188A5C9 }, |
89 |
|
|
{25, 0x0188A594 }, {25, 0x0188A5FC }, {25, 0x0188A5CA }, {25, 0x0188A5CB }, {25, 0x0188A5B2 }, |
90 |
|
|
{25, 0x0188A5AA }, {25, 0x0188A5B3 }, {25, 0x0188A572 }, {25, 0x0188A573 }, {25, 0x0188A5C0 }, |
91 |
|
|
{25, 0x0188A5BE }, {25, 0x0188A5BF }, {25, 0x0188A592 }, {25, 0x0188A580 }, {25, 0x0188A593 }, |
92 |
|
|
{25, 0x0188A590 }, {25, 0x0188A591 }, {25, 0x0188A586 }, {25, 0x0188A5A9 }, {25, 0x0188A5D2 }, |
93 |
|
|
{25, 0x0188A5D3 }, {25, 0x0188A5D4 }, {25, 0x0188A5D5 }, {25, 0x0188A5AC }, {25, 0x0188A5AD }, |
94 |
|
|
{25, 0x0188A5D0 }, |
95 |
|
|
}; |
96 |
|
|
|
97 |
|
|
/* Derived by extracting runcodes from existing tables from decoder */ |
98 |
|
|
static const uint16_t runbook[18][3] = { |
99 |
|
|
{1, 0x0000, 1}, {2, 0x0000, 2}, {3, 0x0000, 3}, {4, 0x0000, 4}, |
100 |
|
|
{5, 0x0000, 5}, {6, 0x0000, 6}, {7, 0x0000, 7}, {8, 0x0000, 8}, |
101 |
|
|
{9, 0x0000, 9}, {10, 0x0000, 10}, {11, 0x0000, 11}, |
102 |
|
|
{7, 0x0069, 12}, {8, 0x00D1, 20}, {9, 0x018A, 32}, |
103 |
|
|
{10, 0x0343, 60}, {11, 0x0685, 100}, {13, 0x18BF, 180}, {13, 0x1BA5, 320}, |
104 |
|
|
}; |
105 |
|
|
|
106 |
|
|
/* |
107 |
|
|
* Derived by inspecting various quality encodes |
108 |
|
|
* and adding some more from scratch. |
109 |
|
|
*/ |
110 |
|
|
static const uint16_t quantization_per_subband[2][3][13][9] = { |
111 |
|
|
{{ |
112 |
|
|
{ 16, 16, 8, 4, 4, 2, 6, 6, 9, }, // film3+ |
113 |
|
|
{ 16, 16, 8, 4, 4, 2, 6, 6, 9, }, // film3 |
114 |
|
|
{ 16, 16, 8, 4, 4, 2, 7, 7, 10, }, // film2+ |
115 |
|
|
{ 16, 16, 8, 4, 4, 2, 8, 8, 12, }, // film2 |
116 |
|
|
{ 16, 16, 8, 4, 4, 2, 16, 16, 26, }, // film1++ |
117 |
|
|
{ 24, 24, 12, 6, 6, 3, 24, 24, 36, }, // film1+ |
118 |
|
|
{ 24, 24, 12, 6, 6, 3, 24, 24, 36, }, // film1 |
119 |
|
|
{ 32, 32, 24, 8, 8, 6, 32, 32, 48, }, // high+ |
120 |
|
|
{ 32, 32, 24, 8, 8, 6, 32, 32, 48, }, // high |
121 |
|
|
{ 48, 48, 32, 12, 12, 8, 64, 64, 96, }, // medium+ |
122 |
|
|
{ 48, 48, 32, 12, 12, 8, 64, 64, 96, }, // medium |
123 |
|
|
{ 64, 64, 48, 16, 16, 12, 96, 96, 144, }, // low+ |
124 |
|
|
{ 64, 64, 48, 16, 16, 12, 128, 128, 192, }, // low |
125 |
|
|
}, |
126 |
|
|
{ |
127 |
|
|
{ 16, 16, 8, 4, 4, 2, 6, 6, 9, }, // film3+ |
128 |
|
|
{ 16, 16, 8, 4, 4, 2, 6, 6, 12, }, // film3 |
129 |
|
|
{ 16, 16, 8, 4, 4, 2, 7, 7, 14, }, // film2+ |
130 |
|
|
{ 16, 16, 8, 4, 4, 2, 8, 8, 16, }, // film2 |
131 |
|
|
{ 16, 16, 8, 4, 4, 2, 16, 16, 26, }, // film1++ |
132 |
|
|
{ 24, 24, 12, 6, 6, 3, 24, 24, 36, }, // film1+ |
133 |
|
|
{ 24, 24, 12, 6, 6, 3, 24, 24, 48, }, // film1 |
134 |
|
|
{ 32, 32, 24, 8, 8, 6, 32, 32, 48, }, // high+ |
135 |
|
|
{ 48, 48, 32, 12, 12, 8, 32, 32, 64, }, // high |
136 |
|
|
{ 48, 48, 32, 12, 12, 8, 64, 64, 96, }, // medium+ |
137 |
|
|
{ 48, 48, 32, 12, 12, 8, 64, 64, 128, }, // medium |
138 |
|
|
{ 64, 64, 48, 16, 16, 12, 96, 96, 160, }, // low+ |
139 |
|
|
{ 64, 64, 48, 16, 16, 12, 128, 128, 192, }, // low |
140 |
|
|
}, |
141 |
|
|
{ |
142 |
|
|
{ 16, 16, 8, 4, 4, 2, 6, 6, 9, }, // film3+ |
143 |
|
|
{ 16, 16, 8, 4, 4, 2, 6, 6, 12, }, // film3 |
144 |
|
|
{ 16, 16, 8, 4, 4, 2, 7, 7, 14, }, // film2+ |
145 |
|
|
{ 16, 16, 8, 4, 4, 2, 8, 8, 16, }, // film2 |
146 |
|
|
{ 16, 16, 8, 4, 4, 2, 16, 16, 26, }, // film1++ |
147 |
|
|
{ 24, 24, 12, 6, 6, 3, 24, 24, 36, }, // film1+ |
148 |
|
|
{ 24, 24, 12, 6, 6, 3, 24, 24, 48, }, // film1 |
149 |
|
|
{ 32, 32, 24, 8, 8, 6, 32, 32, 48, }, // high+ |
150 |
|
|
{ 48, 48, 32, 12, 12, 8, 32, 32, 64, }, // high |
151 |
|
|
{ 48, 48, 32, 12, 12, 8, 64, 64, 96, }, // medium+ |
152 |
|
|
{ 48, 48, 32, 12, 12, 8, 64, 64, 128, }, // medium |
153 |
|
|
{ 64, 64, 48, 16, 16, 12, 96, 96, 160, }, // low+ |
154 |
|
|
{ 64, 64, 48, 16, 16, 12, 128, 128, 192, }, // low |
155 |
|
|
}}, |
156 |
|
|
{{ |
157 |
|
|
{ 16, 16, 8, 16, 16, 8, 24, 24, 36, }, // film3+ |
158 |
|
|
{ 16, 16, 8, 16, 16, 8, 24, 24, 36, }, // film3 |
159 |
|
|
{ 16, 16, 8, 16, 16, 8, 32, 32, 48, }, // film2+ |
160 |
|
|
{ 16, 16, 8, 16, 16, 8, 32, 32, 48, }, // film2 |
161 |
|
|
{ 16, 16, 8, 20, 20, 10, 80, 80, 128, }, // film1++ |
162 |
|
|
{ 24, 24, 12, 24, 24, 12, 96, 96, 144, }, // film1+ |
163 |
|
|
{ 24, 24, 12, 24, 24, 12, 96, 96, 144, }, // film1 |
164 |
|
|
{ 32, 32, 24, 32, 32, 24, 128, 128, 192, }, // high+ |
165 |
|
|
{ 32, 32, 24, 32, 32, 24, 128, 128, 192, }, // high |
166 |
|
|
{ 48, 48, 32, 48, 48, 32, 256, 256, 384, }, // medium+ |
167 |
|
|
{ 48, 48, 32, 48, 48, 32, 256, 256, 384, }, // medium |
168 |
|
|
{ 56, 56, 40, 56, 56, 40, 512, 512, 768, }, // low+ |
169 |
|
|
{ 64, 64, 48, 64, 64, 48, 512, 512, 768, }, // low |
170 |
|
|
}, |
171 |
|
|
{ |
172 |
|
|
{ 16, 16, 8, 16, 16, 8, 24, 24, 36, }, // film3+ |
173 |
|
|
{ 16, 16, 8, 16, 16, 8, 48, 48, 72, }, // film3 |
174 |
|
|
{ 16, 16, 8, 16, 16, 8, 48, 48, 72, }, // film2+ |
175 |
|
|
{ 16, 16, 8, 16, 16, 8, 64, 64, 96, }, // film2 |
176 |
|
|
{ 16, 16, 8, 20, 20, 10, 80, 80, 128, }, // film1++ |
177 |
|
|
{ 24, 24, 12, 24, 24, 12, 96, 96, 144, }, // film1+ |
178 |
|
|
{ 24, 24, 12, 24, 24, 12, 192, 192, 288, }, // film1 |
179 |
|
|
{ 32, 32, 24, 32, 32, 24, 128, 128, 192, }, // high+ |
180 |
|
|
{ 32, 32, 24, 32, 32, 24, 256, 256, 384, }, // high |
181 |
|
|
{ 48, 48, 32, 48, 48, 32, 256, 256, 384, }, // medium+ |
182 |
|
|
{ 48, 48, 32, 48, 48, 32, 512, 512, 768, }, // medium |
183 |
|
|
{ 56, 56, 40, 56, 56, 40, 512, 512, 768, }, // low+ |
184 |
|
|
{ 64, 64, 48, 64, 64, 48,1024,1024,1536, }, // low |
185 |
|
|
}, |
186 |
|
|
{ |
187 |
|
|
{ 16, 16, 8, 16, 16, 8, 24, 24, 36, }, // film3+ |
188 |
|
|
{ 16, 16, 8, 16, 16, 8, 48, 48, 72, }, // film3 |
189 |
|
|
{ 16, 16, 8, 16, 16, 8, 48, 48, 72, }, // film2+ |
190 |
|
|
{ 16, 16, 8, 16, 16, 8, 64, 64, 96, }, // film2 |
191 |
|
|
{ 16, 16, 10, 20, 20, 10, 80, 80, 128, }, // film1++ |
192 |
|
|
{ 24, 24, 12, 24, 24, 12, 96, 96, 144, }, // film1+ |
193 |
|
|
{ 24, 24, 12, 24, 24, 12, 192, 192, 288, }, // film1 |
194 |
|
|
{ 32, 32, 24, 32, 32, 24, 128, 128, 192, }, // high+ |
195 |
|
|
{ 32, 32, 24, 32, 32, 24, 256, 256, 384, }, // high |
196 |
|
|
{ 48, 48, 32, 48, 48, 32, 256, 256, 384, }, // medium+ |
197 |
|
|
{ 48, 48, 32, 48, 48, 32, 512, 512, 768, }, // medium |
198 |
|
|
{ 56, 56, 40, 56, 56, 40, 512, 512, 768, }, // low+ |
199 |
|
|
{ 64, 64, 48, 64, 64, 48,1024,1024,1536, }, // low |
200 |
|
|
}}, |
201 |
|
|
}; |
202 |
|
|
|
203 |
|
|
typedef struct Codebook { |
204 |
|
|
unsigned bits; |
205 |
|
|
unsigned size; |
206 |
|
|
} Codebook; |
207 |
|
|
|
208 |
|
|
typedef struct Runbook { |
209 |
|
|
unsigned size; |
210 |
|
|
unsigned bits; |
211 |
|
|
unsigned run; |
212 |
|
|
} Runbook; |
213 |
|
|
|
214 |
|
|
typedef struct PlaneEnc { |
215 |
|
|
unsigned size; |
216 |
|
|
|
217 |
|
|
int16_t *dwt_buf; |
218 |
|
|
int16_t *dwt_tmp; |
219 |
|
|
|
220 |
|
|
unsigned quantization[SUBBAND_COUNT]; |
221 |
|
|
int16_t *subband[SUBBAND_COUNT]; |
222 |
|
|
int16_t *l_h[8]; |
223 |
|
|
|
224 |
|
|
SubBand band[DWT_LEVELS][4]; |
225 |
|
|
} PlaneEnc; |
226 |
|
|
|
227 |
|
|
typedef struct CFHDEncContext { |
228 |
|
|
const AVClass *class; |
229 |
|
|
|
230 |
|
|
PutBitContext pb; |
231 |
|
|
PutByteContext pby; |
232 |
|
|
|
233 |
|
|
int quality; |
234 |
|
|
int planes; |
235 |
|
|
int chroma_h_shift; |
236 |
|
|
int chroma_v_shift; |
237 |
|
|
PlaneEnc plane[4]; |
238 |
|
|
|
239 |
|
|
uint16_t lut[1024]; |
240 |
|
|
Runbook rb[321]; |
241 |
|
|
Codebook cb[513]; |
242 |
|
|
int16_t *alpha; |
243 |
|
|
|
244 |
|
|
CFHDEncDSPContext dsp; |
245 |
|
|
} CFHDEncContext; |
246 |
|
|
|
247 |
|
✗ |
static av_cold int cfhd_encode_init(AVCodecContext *avctx) |
248 |
|
|
{ |
249 |
|
✗ |
CFHDEncContext *s = avctx->priv_data; |
250 |
|
✗ |
const int sign_mask = 256; |
251 |
|
✗ |
const int twos_complement = -sign_mask; |
252 |
|
✗ |
const int mag_mask = sign_mask - 1; |
253 |
|
✗ |
int ret, last = 0; |
254 |
|
|
|
255 |
|
✗ |
ret = av_pix_fmt_get_chroma_sub_sample(avctx->pix_fmt, |
256 |
|
|
&s->chroma_h_shift, |
257 |
|
|
&s->chroma_v_shift); |
258 |
|
✗ |
if (ret < 0) |
259 |
|
✗ |
return ret; |
260 |
|
|
|
261 |
|
✗ |
if (avctx->height < 4) { |
262 |
|
✗ |
av_log(avctx, AV_LOG_ERROR, "Height must be >= 4.\n"); |
263 |
|
✗ |
return AVERROR_INVALIDDATA; |
264 |
|
|
} |
265 |
|
|
|
266 |
|
✗ |
if (avctx->width & 15) { |
267 |
|
✗ |
av_log(avctx, AV_LOG_ERROR, "Width must be multiple of 16.\n"); |
268 |
|
✗ |
return AVERROR_INVALIDDATA; |
269 |
|
|
} |
270 |
|
|
|
271 |
|
✗ |
s->planes = av_pix_fmt_count_planes(avctx->pix_fmt); |
272 |
|
|
|
273 |
|
✗ |
for (int i = 0; i < s->planes; i++) { |
274 |
|
|
int w8, h8, w4, h4, w2, h2; |
275 |
|
✗ |
const int a_height = FFALIGN(avctx->height, 8); |
276 |
|
✗ |
int width = i ? AV_CEIL_RSHIFT(avctx->width, s->chroma_h_shift) : avctx->width; |
277 |
|
✗ |
int height = i ? a_height >> s->chroma_v_shift: a_height; |
278 |
|
|
|
279 |
|
✗ |
w8 = width / 8 + 64; |
280 |
|
✗ |
h8 = height / 8; |
281 |
|
✗ |
w4 = w8 * 2; |
282 |
|
✗ |
h4 = h8 * 2; |
283 |
|
✗ |
w2 = w4 * 2; |
284 |
|
✗ |
h2 = h4 * 2; |
285 |
|
|
|
286 |
|
✗ |
s->plane[i].dwt_buf = |
287 |
|
✗ |
av_calloc(h8 * 8 * w8 * 8, sizeof(*s->plane[i].dwt_buf)); |
288 |
|
✗ |
s->plane[i].dwt_tmp = |
289 |
|
✗ |
av_malloc_array(h8 * 8 * w8 * 8, sizeof(*s->plane[i].dwt_tmp)); |
290 |
|
✗ |
if (!s->plane[i].dwt_buf || !s->plane[i].dwt_tmp) |
291 |
|
✗ |
return AVERROR(ENOMEM); |
292 |
|
|
|
293 |
|
✗ |
s->plane[i].subband[0] = s->plane[i].dwt_buf; |
294 |
|
✗ |
s->plane[i].subband[1] = s->plane[i].dwt_buf + 2 * w8 * h8; |
295 |
|
✗ |
s->plane[i].subband[2] = s->plane[i].dwt_buf + 1 * w8 * h8; |
296 |
|
✗ |
s->plane[i].subband[3] = s->plane[i].dwt_buf + 3 * w8 * h8; |
297 |
|
✗ |
s->plane[i].subband[4] = s->plane[i].dwt_buf + 2 * w4 * h4; |
298 |
|
✗ |
s->plane[i].subband[5] = s->plane[i].dwt_buf + 1 * w4 * h4; |
299 |
|
✗ |
s->plane[i].subband[6] = s->plane[i].dwt_buf + 3 * w4 * h4; |
300 |
|
✗ |
s->plane[i].subband[7] = s->plane[i].dwt_buf + 2 * w2 * h2; |
301 |
|
✗ |
s->plane[i].subband[8] = s->plane[i].dwt_buf + 1 * w2 * h2; |
302 |
|
✗ |
s->plane[i].subband[9] = s->plane[i].dwt_buf + 3 * w2 * h2; |
303 |
|
|
|
304 |
|
✗ |
for (int j = 0; j < DWT_LEVELS; j++) { |
305 |
|
✗ |
for (int k = 0; k < FF_ARRAY_ELEMS(s->plane[i].band[j]); k++) { |
306 |
|
✗ |
s->plane[i].band[j][k].width = (width / 8) << j; |
307 |
|
✗ |
s->plane[i].band[j][k].height = height >> (DWT_LEVELS - j); |
308 |
|
✗ |
s->plane[i].band[j][k].a_width = w8 << j; |
309 |
|
✗ |
s->plane[i].band[j][k].a_height = h8 << j; |
310 |
|
|
} |
311 |
|
|
} |
312 |
|
|
|
313 |
|
|
/* ll2 and ll1 commented out because they are done in-place */ |
314 |
|
✗ |
s->plane[i].l_h[0] = s->plane[i].dwt_tmp; |
315 |
|
✗ |
s->plane[i].l_h[1] = s->plane[i].dwt_tmp + 2 * w8 * h8; |
316 |
|
|
// s->plane[i].l_h[2] = ll2; |
317 |
|
✗ |
s->plane[i].l_h[3] = s->plane[i].dwt_tmp; |
318 |
|
✗ |
s->plane[i].l_h[4] = s->plane[i].dwt_tmp + 2 * w4 * h4; |
319 |
|
|
// s->plane[i].l_h[5] = ll1; |
320 |
|
✗ |
s->plane[i].l_h[6] = s->plane[i].dwt_tmp; |
321 |
|
✗ |
s->plane[i].l_h[7] = s->plane[i].dwt_tmp + 2 * w2 * h2; |
322 |
|
|
} |
323 |
|
|
|
324 |
|
✗ |
for (int i = 0; i < 512; i++) { |
325 |
|
✗ |
int value = (i & sign_mask) ? twos_complement + (i & mag_mask): i; |
326 |
|
✗ |
int mag = FFMIN(FFABS(value), 255); |
327 |
|
|
|
328 |
|
✗ |
if (mag) { |
329 |
|
✗ |
s->cb[i].bits = (codebook[mag][1] << 1) | (value > 0 ? 0 : 1); |
330 |
|
✗ |
s->cb[i].size = codebook[mag][0] + 1; |
331 |
|
|
} else { |
332 |
|
✗ |
s->cb[i].bits = codebook[mag][1]; |
333 |
|
✗ |
s->cb[i].size = codebook[mag][0]; |
334 |
|
|
} |
335 |
|
|
} |
336 |
|
|
|
337 |
|
✗ |
s->cb[512].bits = 0x3114ba3; |
338 |
|
✗ |
s->cb[512].size = 26; |
339 |
|
|
|
340 |
|
✗ |
s->rb[0].run = 0; |
341 |
|
|
|
342 |
|
✗ |
for (int i = 1, j = 0; i < 320 && j < 17; j++) { |
343 |
|
✗ |
int run = runbook[j][2]; |
344 |
|
✗ |
int end = runbook[j+1][2]; |
345 |
|
|
|
346 |
|
✗ |
while (i < end) { |
347 |
|
✗ |
s->rb[i].run = run; |
348 |
|
✗ |
s->rb[i].bits = runbook[j][1]; |
349 |
|
✗ |
s->rb[i++].size = runbook[j][0]; |
350 |
|
|
} |
351 |
|
|
} |
352 |
|
|
|
353 |
|
✗ |
s->rb[320].bits = runbook[17][1]; |
354 |
|
✗ |
s->rb[320].size = runbook[17][0]; |
355 |
|
✗ |
s->rb[320].run = 320; |
356 |
|
|
|
357 |
|
✗ |
for (int i = 0; i < 256; i++) { |
358 |
|
✗ |
int idx = i + ((768LL * i * i * i) / (256 * 256 * 256)); |
359 |
|
|
|
360 |
|
✗ |
s->lut[idx] = i; |
361 |
|
|
} |
362 |
|
✗ |
for (int i = 0; i < 1024; i++) { |
363 |
|
✗ |
if (s->lut[i]) |
364 |
|
✗ |
last = s->lut[i]; |
365 |
|
|
else |
366 |
|
✗ |
s->lut[i] = last; |
367 |
|
|
} |
368 |
|
|
|
369 |
|
✗ |
ff_cfhdencdsp_init(&s->dsp); |
370 |
|
|
|
371 |
|
✗ |
if (s->planes != 4) |
372 |
|
✗ |
return 0; |
373 |
|
|
|
374 |
|
✗ |
s->alpha = av_calloc(avctx->width * avctx->height, sizeof(*s->alpha)); |
375 |
|
✗ |
if (!s->alpha) |
376 |
|
✗ |
return AVERROR(ENOMEM); |
377 |
|
|
|
378 |
|
✗ |
return 0; |
379 |
|
|
} |
380 |
|
|
|
381 |
|
✗ |
static void quantize_band(int16_t *input, int width, int a_width, |
382 |
|
|
int height, unsigned quantization) |
383 |
|
|
{ |
384 |
|
✗ |
const int16_t factor = (uint32_t)(1U << 15) / quantization; |
385 |
|
|
|
386 |
|
✗ |
for (int i = 0; i < height; i++) { |
387 |
|
✗ |
for (int j = 0; j < width; j++) |
388 |
|
✗ |
input[j] = av_clip_intp2(((input[j] * factor + 16384 * FFSIGN(input[j])) / 32768), 10); |
389 |
|
✗ |
input += a_width; |
390 |
|
|
} |
391 |
|
✗ |
} |
392 |
|
|
|
393 |
|
✗ |
static int put_runcode(PutBitContext *pb, int count, const Runbook *const rb) |
394 |
|
|
{ |
395 |
|
✗ |
while (count > 0) { |
396 |
|
✗ |
const int index = FFMIN(320, count); |
397 |
|
|
|
398 |
|
✗ |
put_bits(pb, rb[index].size, rb[index].bits); |
399 |
|
✗ |
count -= rb[index].run; |
400 |
|
|
} |
401 |
|
|
|
402 |
|
✗ |
return 0; |
403 |
|
|
} |
404 |
|
|
|
405 |
|
✗ |
static void process_alpha(const int16_t *src, int width, int height, ptrdiff_t stride, int16_t *dst) |
406 |
|
|
{ |
407 |
|
✗ |
for (int i = 0; i < height; i++) { |
408 |
|
✗ |
for (int j = 0; j < width; j++) { |
409 |
|
✗ |
int alpha = src[j]; |
410 |
|
|
|
411 |
|
✗ |
if (alpha > 0 && alpha < 4080) { |
412 |
|
✗ |
alpha *= 223; |
413 |
|
✗ |
alpha += 128; |
414 |
|
✗ |
alpha >>= 8; |
415 |
|
✗ |
alpha += 256; |
416 |
|
|
} |
417 |
|
|
|
418 |
|
✗ |
dst[j] = av_clip_uintp2(alpha, 12); |
419 |
|
|
} |
420 |
|
|
|
421 |
|
✗ |
src += stride; |
422 |
|
✗ |
dst += width; |
423 |
|
|
} |
424 |
|
✗ |
} |
425 |
|
|
|
426 |
|
✗ |
static int cfhd_encode_frame(AVCodecContext *avctx, AVPacket *pkt, |
427 |
|
|
const AVFrame *frame, int *got_packet) |
428 |
|
|
{ |
429 |
|
✗ |
CFHDEncContext *s = avctx->priv_data; |
430 |
|
✗ |
CFHDEncDSPContext *dsp = &s->dsp; |
431 |
|
✗ |
PutByteContext *pby = &s->pby; |
432 |
|
✗ |
PutBitContext *pb = &s->pb; |
433 |
|
✗ |
const Codebook *const cb = s->cb; |
434 |
|
✗ |
const Runbook *const rb = s->rb; |
435 |
|
✗ |
const uint16_t *lut = s->lut; |
436 |
|
|
unsigned pos; |
437 |
|
|
int ret; |
438 |
|
|
|
439 |
|
✗ |
for (int plane = 0; plane < s->planes; plane++) { |
440 |
|
✗ |
const int h_shift = plane ? s->chroma_h_shift : 0; |
441 |
|
✗ |
int width = s->plane[plane].band[2][0].width; |
442 |
|
✗ |
int a_width = s->plane[plane].band[2][0].a_width; |
443 |
|
✗ |
int height = s->plane[plane].band[2][0].height; |
444 |
|
✗ |
int act_plane = plane == 1 ? 2 : plane == 2 ? 1 : plane; |
445 |
|
✗ |
const int16_t *input = (int16_t *)frame->data[act_plane]; |
446 |
|
|
int16_t *buf; |
447 |
|
✗ |
int16_t *low = s->plane[plane].l_h[6]; |
448 |
|
✗ |
int16_t *high = s->plane[plane].l_h[7]; |
449 |
|
✗ |
ptrdiff_t in_stride = frame->linesize[act_plane] / 2; |
450 |
|
|
int low_stride, high_stride; |
451 |
|
|
|
452 |
|
✗ |
if (plane == 3) { |
453 |
|
✗ |
process_alpha(input, avctx->width, avctx->height, |
454 |
|
|
in_stride, s->alpha); |
455 |
|
✗ |
input = s->alpha; |
456 |
|
✗ |
in_stride = avctx->width; |
457 |
|
|
} |
458 |
|
|
|
459 |
|
✗ |
dsp->horiz_filter(input, low, high, |
460 |
|
|
in_stride, a_width, a_width, |
461 |
|
✗ |
avctx->width >> h_shift, avctx->height); |
462 |
|
|
|
463 |
|
✗ |
input = s->plane[plane].l_h[7]; |
464 |
|
✗ |
low = s->plane[plane].subband[7]; |
465 |
|
✗ |
low_stride = s->plane[plane].band[2][0].a_width; |
466 |
|
✗ |
high = s->plane[plane].subband[9]; |
467 |
|
✗ |
high_stride = s->plane[plane].band[2][0].a_width; |
468 |
|
|
|
469 |
|
✗ |
dsp->vert_filter(input, low, high, |
470 |
|
|
a_width, low_stride, high_stride, |
471 |
|
|
width, height * 2); |
472 |
|
|
|
473 |
|
✗ |
input = s->plane[plane].l_h[6]; |
474 |
|
✗ |
low = s->plane[plane].l_h[7]; |
475 |
|
✗ |
high = s->plane[plane].subband[8]; |
476 |
|
|
|
477 |
|
✗ |
dsp->vert_filter(input, low, high, |
478 |
|
|
a_width, low_stride, high_stride, |
479 |
|
|
width, height * 2); |
480 |
|
|
|
481 |
|
✗ |
a_width = s->plane[plane].band[1][0].a_width; |
482 |
|
✗ |
width = s->plane[plane].band[1][0].width; |
483 |
|
✗ |
height = s->plane[plane].band[1][0].height; |
484 |
|
✗ |
input = s->plane[plane].l_h[7]; |
485 |
|
✗ |
low = s->plane[plane].l_h[3]; |
486 |
|
✗ |
low_stride = s->plane[plane].band[1][0].a_width; |
487 |
|
✗ |
high = s->plane[plane].l_h[4]; |
488 |
|
✗ |
high_stride = s->plane[plane].band[1][0].a_width; |
489 |
|
|
|
490 |
|
✗ |
buf = s->plane[plane].l_h[7]; |
491 |
|
✗ |
for (int i = 0; i < height * 2; i++) { |
492 |
|
✗ |
for (int j = 0; j < width * 2; j++) |
493 |
|
✗ |
buf[j] /= 4; |
494 |
|
✗ |
buf += a_width * 2; |
495 |
|
|
} |
496 |
|
|
|
497 |
|
✗ |
dsp->horiz_filter(input, low, high, |
498 |
|
✗ |
a_width * 2, low_stride, high_stride, |
499 |
|
|
width * 2, height * 2); |
500 |
|
|
|
501 |
|
✗ |
input = s->plane[plane].l_h[4]; |
502 |
|
✗ |
low = s->plane[plane].subband[4]; |
503 |
|
✗ |
high = s->plane[plane].subband[6]; |
504 |
|
|
|
505 |
|
✗ |
dsp->vert_filter(input, low, high, |
506 |
|
|
a_width, low_stride, high_stride, |
507 |
|
|
width, height * 2); |
508 |
|
|
|
509 |
|
✗ |
input = s->plane[plane].l_h[3]; |
510 |
|
✗ |
low = s->plane[plane].l_h[4]; |
511 |
|
✗ |
high = s->plane[plane].subband[5]; |
512 |
|
|
|
513 |
|
✗ |
dsp->vert_filter(input, low, high, |
514 |
|
|
a_width, low_stride, high_stride, |
515 |
|
|
width, height * 2); |
516 |
|
|
|
517 |
|
✗ |
a_width = s->plane[plane].band[0][0].a_width; |
518 |
|
✗ |
width = s->plane[plane].band[0][0].width; |
519 |
|
✗ |
height = s->plane[plane].band[0][0].height; |
520 |
|
✗ |
input = s->plane[plane].l_h[4]; |
521 |
|
✗ |
low = s->plane[plane].l_h[0]; |
522 |
|
✗ |
low_stride = s->plane[plane].band[0][0].a_width; |
523 |
|
✗ |
high = s->plane[plane].l_h[1]; |
524 |
|
✗ |
high_stride = s->plane[plane].band[0][0].a_width; |
525 |
|
|
|
526 |
|
✗ |
if (avctx->pix_fmt != AV_PIX_FMT_YUV422P10) { |
527 |
|
✗ |
int16_t *buf = s->plane[plane].l_h[4]; |
528 |
|
✗ |
for (int i = 0; i < height * 2; i++) { |
529 |
|
✗ |
for (int j = 0; j < width * 2; j++) |
530 |
|
✗ |
buf[j] /= 4; |
531 |
|
✗ |
buf += a_width * 2; |
532 |
|
|
} |
533 |
|
|
} |
534 |
|
|
|
535 |
|
✗ |
dsp->horiz_filter(input, low, high, |
536 |
|
✗ |
a_width * 2, low_stride, high_stride, |
537 |
|
|
width * 2, height * 2); |
538 |
|
|
|
539 |
|
✗ |
low = s->plane[plane].subband[1]; |
540 |
|
✗ |
high = s->plane[plane].subband[3]; |
541 |
|
✗ |
input = s->plane[plane].l_h[1]; |
542 |
|
|
|
543 |
|
✗ |
dsp->vert_filter(input, low, high, |
544 |
|
|
a_width, low_stride, high_stride, |
545 |
|
|
width, height * 2); |
546 |
|
|
|
547 |
|
✗ |
low = s->plane[plane].subband[0]; |
548 |
|
✗ |
high = s->plane[plane].subband[2]; |
549 |
|
✗ |
input = s->plane[plane].l_h[0]; |
550 |
|
|
|
551 |
|
✗ |
dsp->vert_filter(input, low, high, |
552 |
|
|
a_width, low_stride, high_stride, |
553 |
|
|
width, height * 2); |
554 |
|
|
} |
555 |
|
|
|
556 |
|
✗ |
ret = ff_alloc_packet(avctx, pkt, 256LL + s->planes * (2LL * avctx->width * (avctx->height + 15) + 2048LL)); |
557 |
|
✗ |
if (ret < 0) |
558 |
|
✗ |
return ret; |
559 |
|
|
|
560 |
|
✗ |
bytestream2_init_writer(pby, pkt->data, pkt->size); |
561 |
|
|
|
562 |
|
✗ |
bytestream2_put_be16(pby, SampleType); |
563 |
|
✗ |
bytestream2_put_be16(pby, 9); |
564 |
|
|
|
565 |
|
✗ |
bytestream2_put_be16(pby, SampleIndexTable); |
566 |
|
✗ |
bytestream2_put_be16(pby, s->planes); |
567 |
|
|
|
568 |
|
✗ |
for (int i = 0; i < s->planes; i++) |
569 |
|
✗ |
bytestream2_put_be32(pby, 0); |
570 |
|
|
|
571 |
|
✗ |
bytestream2_put_be16(pby, TransformType); |
572 |
|
✗ |
bytestream2_put_be16(pby, 0); |
573 |
|
|
|
574 |
|
✗ |
bytestream2_put_be16(pby, NumFrames); |
575 |
|
✗ |
bytestream2_put_be16(pby, 1); |
576 |
|
|
|
577 |
|
✗ |
bytestream2_put_be16(pby, ChannelCount); |
578 |
|
✗ |
bytestream2_put_be16(pby, s->planes); |
579 |
|
|
|
580 |
|
✗ |
bytestream2_put_be16(pby, EncodedFormat); |
581 |
|
✗ |
bytestream2_put_be16(pby, avctx->pix_fmt == AV_PIX_FMT_YUV422P10 ? 1 : 3 + (s->planes == 4)); |
582 |
|
|
|
583 |
|
✗ |
bytestream2_put_be16(pby, WaveletCount); |
584 |
|
✗ |
bytestream2_put_be16(pby, 3); |
585 |
|
|
|
586 |
|
✗ |
bytestream2_put_be16(pby, SubbandCount); |
587 |
|
✗ |
bytestream2_put_be16(pby, SUBBAND_COUNT); |
588 |
|
|
|
589 |
|
✗ |
bytestream2_put_be16(pby, NumSpatial); |
590 |
|
✗ |
bytestream2_put_be16(pby, 2); |
591 |
|
|
|
592 |
|
✗ |
bytestream2_put_be16(pby, FirstWavelet); |
593 |
|
✗ |
bytestream2_put_be16(pby, 3); |
594 |
|
|
|
595 |
|
✗ |
bytestream2_put_be16(pby, ImageWidth); |
596 |
|
✗ |
bytestream2_put_be16(pby, avctx->width); |
597 |
|
|
|
598 |
|
✗ |
bytestream2_put_be16(pby, ImageHeight); |
599 |
|
✗ |
bytestream2_put_be16(pby, FFALIGN(avctx->height, 8)); |
600 |
|
|
|
601 |
|
✗ |
bytestream2_put_be16(pby, -DisplayHeight); |
602 |
|
✗ |
bytestream2_put_be16(pby, avctx->height); |
603 |
|
|
|
604 |
|
✗ |
bytestream2_put_be16(pby, -FrameNumber); |
605 |
|
✗ |
bytestream2_put_be16(pby, frame->pts & 0xFFFF); |
606 |
|
|
|
607 |
|
✗ |
bytestream2_put_be16(pby, Precision); |
608 |
|
✗ |
bytestream2_put_be16(pby, avctx->pix_fmt == AV_PIX_FMT_YUV422P10 ? 10 : 12); |
609 |
|
|
|
610 |
|
✗ |
bytestream2_put_be16(pby, PrescaleTable); |
611 |
|
✗ |
bytestream2_put_be16(pby, avctx->pix_fmt == AV_PIX_FMT_YUV422P10 ? 0x2000 : 0x2800); |
612 |
|
|
|
613 |
|
✗ |
bytestream2_put_be16(pby, SampleFlags); |
614 |
|
✗ |
bytestream2_put_be16(pby, 1); |
615 |
|
|
|
616 |
|
✗ |
for (int p = 0; p < s->planes; p++) { |
617 |
|
✗ |
int width = s->plane[p].band[0][0].width; |
618 |
|
✗ |
int a_width = s->plane[p].band[0][0].a_width; |
619 |
|
✗ |
int height = s->plane[p].band[0][0].height; |
620 |
|
✗ |
int16_t *data = s->plane[p].subband[0]; |
621 |
|
|
|
622 |
|
✗ |
if (p) { |
623 |
|
✗ |
bytestream2_put_be16(pby, SampleType); |
624 |
|
✗ |
bytestream2_put_be16(pby, 3); |
625 |
|
|
|
626 |
|
✗ |
bytestream2_put_be16(pby, ChannelNumber); |
627 |
|
✗ |
bytestream2_put_be16(pby, p); |
628 |
|
|
} |
629 |
|
|
|
630 |
|
✗ |
bytestream2_put_be16(pby, BitstreamMarker); |
631 |
|
✗ |
bytestream2_put_be16(pby, 0x1a4a); |
632 |
|
|
|
633 |
|
✗ |
pos = bytestream2_tell_p(pby); |
634 |
|
|
|
635 |
|
✗ |
bytestream2_put_be16(pby, LowpassSubband); |
636 |
|
✗ |
bytestream2_put_be16(pby, 0); |
637 |
|
|
|
638 |
|
✗ |
bytestream2_put_be16(pby, NumLevels); |
639 |
|
✗ |
bytestream2_put_be16(pby, 3); |
640 |
|
|
|
641 |
|
✗ |
bytestream2_put_be16(pby, LowpassWidth); |
642 |
|
✗ |
bytestream2_put_be16(pby, width); |
643 |
|
|
|
644 |
|
✗ |
bytestream2_put_be16(pby, LowpassHeight); |
645 |
|
✗ |
bytestream2_put_be16(pby, height); |
646 |
|
|
|
647 |
|
✗ |
bytestream2_put_be16(pby, PixelOffset); |
648 |
|
✗ |
bytestream2_put_be16(pby, 0); |
649 |
|
|
|
650 |
|
✗ |
bytestream2_put_be16(pby, LowpassQuantization); |
651 |
|
✗ |
bytestream2_put_be16(pby, 1); |
652 |
|
|
|
653 |
|
✗ |
bytestream2_put_be16(pby, LowpassPrecision); |
654 |
|
✗ |
bytestream2_put_be16(pby, 16); |
655 |
|
|
|
656 |
|
✗ |
bytestream2_put_be16(pby, BitstreamMarker); |
657 |
|
✗ |
bytestream2_put_be16(pby, 0x0f0f); |
658 |
|
|
|
659 |
|
✗ |
for (int i = 0; i < height; i++) { |
660 |
|
✗ |
for (int j = 0; j < width; j++) |
661 |
|
✗ |
bytestream2_put_be16(pby, data[j]); |
662 |
|
✗ |
data += a_width; |
663 |
|
|
} |
664 |
|
|
|
665 |
|
✗ |
bytestream2_put_be16(pby, BitstreamMarker); |
666 |
|
✗ |
bytestream2_put_be16(pby, 0x1b4b); |
667 |
|
|
|
668 |
|
✗ |
for (int l = 0; l < 3; l++) { |
669 |
|
✗ |
for (int i = 0; i < 3; i++) { |
670 |
|
✗ |
s->plane[p].quantization[1 + l * 3 + i] = quantization_per_subband[avctx->pix_fmt != AV_PIX_FMT_YUV422P10][p >= 3 ? 0 : p][s->quality][l * 3 + i]; |
671 |
|
|
} |
672 |
|
|
} |
673 |
|
|
|
674 |
|
✗ |
for (int l = 0; l < 3; l++) { |
675 |
|
✗ |
int a_width = s->plane[p].band[l][0].a_width; |
676 |
|
✗ |
int width = s->plane[p].band[l][0].width; |
677 |
|
✗ |
int stride = FFALIGN(width, 8); |
678 |
|
✗ |
int height = s->plane[p].band[l][0].height; |
679 |
|
|
|
680 |
|
✗ |
bytestream2_put_be16(pby, BitstreamMarker); |
681 |
|
✗ |
bytestream2_put_be16(pby, 0x0d0d); |
682 |
|
|
|
683 |
|
✗ |
bytestream2_put_be16(pby, WaveletType); |
684 |
|
✗ |
bytestream2_put_be16(pby, 3 + 2 * (l == 2)); |
685 |
|
|
|
686 |
|
✗ |
bytestream2_put_be16(pby, WaveletNumber); |
687 |
|
✗ |
bytestream2_put_be16(pby, 3 - l); |
688 |
|
|
|
689 |
|
✗ |
bytestream2_put_be16(pby, WaveletLevel); |
690 |
|
✗ |
bytestream2_put_be16(pby, 3 - l); |
691 |
|
|
|
692 |
|
✗ |
bytestream2_put_be16(pby, NumBands); |
693 |
|
✗ |
bytestream2_put_be16(pby, 4); |
694 |
|
|
|
695 |
|
✗ |
bytestream2_put_be16(pby, HighpassWidth); |
696 |
|
✗ |
bytestream2_put_be16(pby, width); |
697 |
|
|
|
698 |
|
✗ |
bytestream2_put_be16(pby, HighpassHeight); |
699 |
|
✗ |
bytestream2_put_be16(pby, height); |
700 |
|
|
|
701 |
|
✗ |
bytestream2_put_be16(pby, LowpassBorder); |
702 |
|
✗ |
bytestream2_put_be16(pby, 0); |
703 |
|
|
|
704 |
|
✗ |
bytestream2_put_be16(pby, HighpassBorder); |
705 |
|
✗ |
bytestream2_put_be16(pby, 0); |
706 |
|
|
|
707 |
|
✗ |
bytestream2_put_be16(pby, LowpassScale); |
708 |
|
✗ |
bytestream2_put_be16(pby, 1); |
709 |
|
|
|
710 |
|
✗ |
bytestream2_put_be16(pby, LowpassDivisor); |
711 |
|
✗ |
bytestream2_put_be16(pby, 1); |
712 |
|
|
|
713 |
|
✗ |
for (int i = 0; i < 3; i++) { |
714 |
|
✗ |
int16_t *data = s->plane[p].subband[1 + l * 3 + i]; |
715 |
|
✗ |
int count = 0, padd = 0; |
716 |
|
|
|
717 |
|
✗ |
bytestream2_put_be16(pby, BitstreamMarker); |
718 |
|
✗ |
bytestream2_put_be16(pby, 0x0e0e); |
719 |
|
|
|
720 |
|
✗ |
bytestream2_put_be16(pby, SubbandNumber); |
721 |
|
✗ |
bytestream2_put_be16(pby, i + 1); |
722 |
|
|
|
723 |
|
✗ |
bytestream2_put_be16(pby, BandCodingFlags); |
724 |
|
✗ |
bytestream2_put_be16(pby, 1); |
725 |
|
|
|
726 |
|
✗ |
bytestream2_put_be16(pby, BandWidth); |
727 |
|
✗ |
bytestream2_put_be16(pby, width); |
728 |
|
|
|
729 |
|
✗ |
bytestream2_put_be16(pby, BandHeight); |
730 |
|
✗ |
bytestream2_put_be16(pby, height); |
731 |
|
|
|
732 |
|
✗ |
bytestream2_put_be16(pby, SubbandBand); |
733 |
|
✗ |
bytestream2_put_be16(pby, 1 + l * 3 + i); |
734 |
|
|
|
735 |
|
✗ |
bytestream2_put_be16(pby, BandEncoding); |
736 |
|
✗ |
bytestream2_put_be16(pby, 3); |
737 |
|
|
|
738 |
|
✗ |
bytestream2_put_be16(pby, Quantization); |
739 |
|
✗ |
bytestream2_put_be16(pby, s->plane[p].quantization[1 + l * 3 + i]); |
740 |
|
|
|
741 |
|
✗ |
bytestream2_put_be16(pby, BandScale); |
742 |
|
✗ |
bytestream2_put_be16(pby, 1); |
743 |
|
|
|
744 |
|
✗ |
bytestream2_put_be16(pby, BandHeader); |
745 |
|
✗ |
bytestream2_put_be16(pby, 0); |
746 |
|
|
|
747 |
|
✗ |
quantize_band(data, width, a_width, height, |
748 |
|
✗ |
s->plane[p].quantization[1 + l * 3 + i]); |
749 |
|
|
|
750 |
|
✗ |
init_put_bits(pb, pkt->data + bytestream2_tell_p(pby), bytestream2_get_bytes_left_p(pby)); |
751 |
|
|
|
752 |
|
✗ |
for (int m = 0; m < height; m++) { |
753 |
|
✗ |
for (int j = 0; j < stride; j++) { |
754 |
|
✗ |
int16_t index = j >= width ? 0 : FFSIGN(data[j]) * lut[FFABS(data[j])]; |
755 |
|
|
|
756 |
|
✗ |
if (index < 0) |
757 |
|
✗ |
index += 512; |
758 |
|
✗ |
if (index == 0) { |
759 |
|
✗ |
count++; |
760 |
|
✗ |
continue; |
761 |
|
✗ |
} else if (count > 0) { |
762 |
|
✗ |
count = put_runcode(pb, count, rb); |
763 |
|
|
} |
764 |
|
|
|
765 |
|
✗ |
put_bits(pb, cb[index].size, cb[index].bits); |
766 |
|
|
} |
767 |
|
|
|
768 |
|
✗ |
data += a_width; |
769 |
|
|
} |
770 |
|
|
|
771 |
|
✗ |
if (count > 0) { |
772 |
|
✗ |
count = put_runcode(pb, count, rb); |
773 |
|
|
} |
774 |
|
|
|
775 |
|
✗ |
put_bits(pb, cb[512].size, cb[512].bits); |
776 |
|
|
|
777 |
|
✗ |
flush_put_bits(pb); |
778 |
|
✗ |
bytestream2_skip_p(pby, put_bytes_output(pb)); |
779 |
|
✗ |
padd = (4 - (bytestream2_tell_p(pby) & 3)) & 3; |
780 |
|
✗ |
while (padd--) |
781 |
|
✗ |
bytestream2_put_byte(pby, 0); |
782 |
|
|
|
783 |
|
✗ |
bytestream2_put_be16(pby, BandTrailer); |
784 |
|
✗ |
bytestream2_put_be16(pby, 0); |
785 |
|
|
} |
786 |
|
|
|
787 |
|
✗ |
bytestream2_put_be16(pby, BitstreamMarker); |
788 |
|
✗ |
bytestream2_put_be16(pby, 0x0c0c); |
789 |
|
|
} |
790 |
|
|
|
791 |
|
✗ |
s->plane[p].size = bytestream2_tell_p(pby) - pos; |
792 |
|
|
} |
793 |
|
|
|
794 |
|
✗ |
bytestream2_put_be16(pby, GroupTrailer); |
795 |
|
✗ |
bytestream2_put_be16(pby, 0); |
796 |
|
|
|
797 |
|
✗ |
av_shrink_packet(pkt, bytestream2_tell_p(pby)); |
798 |
|
|
|
799 |
|
✗ |
pkt->flags |= AV_PKT_FLAG_KEY; |
800 |
|
|
|
801 |
|
✗ |
bytestream2_seek_p(pby, 8, SEEK_SET); |
802 |
|
✗ |
for (int i = 0; i < s->planes; i++) |
803 |
|
✗ |
bytestream2_put_be32(pby, s->plane[i].size); |
804 |
|
|
|
805 |
|
✗ |
*got_packet = 1; |
806 |
|
|
|
807 |
|
✗ |
return 0; |
808 |
|
|
} |
809 |
|
|
|
810 |
|
✗ |
static av_cold int cfhd_encode_close(AVCodecContext *avctx) |
811 |
|
|
{ |
812 |
|
✗ |
CFHDEncContext *s = avctx->priv_data; |
813 |
|
|
|
814 |
|
✗ |
for (int i = 0; i < s->planes; i++) { |
815 |
|
✗ |
av_freep(&s->plane[i].dwt_buf); |
816 |
|
✗ |
av_freep(&s->plane[i].dwt_tmp); |
817 |
|
|
|
818 |
|
✗ |
for (int j = 0; j < SUBBAND_COUNT; j++) |
819 |
|
✗ |
s->plane[i].subband[j] = NULL; |
820 |
|
|
|
821 |
|
✗ |
for (int j = 0; j < 8; j++) |
822 |
|
✗ |
s->plane[i].l_h[j] = NULL; |
823 |
|
|
} |
824 |
|
|
|
825 |
|
✗ |
av_freep(&s->alpha); |
826 |
|
|
|
827 |
|
✗ |
return 0; |
828 |
|
|
} |
829 |
|
|
|
830 |
|
|
#define OFFSET(x) offsetof(CFHDEncContext, x) |
831 |
|
|
#define VE AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_ENCODING_PARAM |
832 |
|
|
static const AVOption options[] = { |
833 |
|
|
{ "quality", "set quality", OFFSET(quality), AV_OPT_TYPE_INT, {.i64= 0}, 0, 12, VE, "q" }, |
834 |
|
|
{ "film3+", NULL, 0, AV_OPT_TYPE_CONST, {.i64= 0}, 0, 0, VE, "q" }, |
835 |
|
|
{ "film3", NULL, 0, AV_OPT_TYPE_CONST, {.i64= 1}, 0, 0, VE, "q" }, |
836 |
|
|
{ "film2+", NULL, 0, AV_OPT_TYPE_CONST, {.i64= 2}, 0, 0, VE, "q" }, |
837 |
|
|
{ "film2", NULL, 0, AV_OPT_TYPE_CONST, {.i64= 3}, 0, 0, VE, "q" }, |
838 |
|
|
{ "film1.5", NULL, 0, AV_OPT_TYPE_CONST, {.i64= 4}, 0, 0, VE, "q" }, |
839 |
|
|
{ "film1+", NULL, 0, AV_OPT_TYPE_CONST, {.i64= 5}, 0, 0, VE, "q" }, |
840 |
|
|
{ "film1", NULL, 0, AV_OPT_TYPE_CONST, {.i64= 6}, 0, 0, VE, "q" }, |
841 |
|
|
{ "high+", NULL, 0, AV_OPT_TYPE_CONST, {.i64= 7}, 0, 0, VE, "q" }, |
842 |
|
|
{ "high", NULL, 0, AV_OPT_TYPE_CONST, {.i64= 8}, 0, 0, VE, "q" }, |
843 |
|
|
{ "medium+", NULL, 0, AV_OPT_TYPE_CONST, {.i64= 9}, 0, 0, VE, "q" }, |
844 |
|
|
{ "medium", NULL, 0, AV_OPT_TYPE_CONST, {.i64=10}, 0, 0, VE, "q" }, |
845 |
|
|
{ "low+", NULL, 0, AV_OPT_TYPE_CONST, {.i64=11}, 0, 0, VE, "q" }, |
846 |
|
|
{ "low", NULL, 0, AV_OPT_TYPE_CONST, {.i64=12}, 0, 0, VE, "q" }, |
847 |
|
|
{ NULL}, |
848 |
|
|
}; |
849 |
|
|
|
850 |
|
|
static const AVClass cfhd_class = { |
851 |
|
|
.class_name = "cfhd", |
852 |
|
|
.item_name = av_default_item_name, |
853 |
|
|
.option = options, |
854 |
|
|
.version = LIBAVUTIL_VERSION_INT, |
855 |
|
|
}; |
856 |
|
|
|
857 |
|
|
const FFCodec ff_cfhd_encoder = { |
858 |
|
|
.p.name = "cfhd", |
859 |
|
|
CODEC_LONG_NAME("GoPro CineForm HD"), |
860 |
|
|
.p.type = AVMEDIA_TYPE_VIDEO, |
861 |
|
|
.p.id = AV_CODEC_ID_CFHD, |
862 |
|
|
.p.capabilities = AV_CODEC_CAP_DR1 | AV_CODEC_CAP_FRAME_THREADS | |
863 |
|
|
AV_CODEC_CAP_ENCODER_REORDERED_OPAQUE, |
864 |
|
|
.priv_data_size = sizeof(CFHDEncContext), |
865 |
|
|
.p.priv_class = &cfhd_class, |
866 |
|
|
.init = cfhd_encode_init, |
867 |
|
|
.close = cfhd_encode_close, |
868 |
|
|
FF_CODEC_ENCODE_CB(cfhd_encode_frame), |
869 |
|
|
.p.pix_fmts = (const enum AVPixelFormat[]) { |
870 |
|
|
AV_PIX_FMT_YUV422P10, |
871 |
|
|
AV_PIX_FMT_GBRP12, |
872 |
|
|
AV_PIX_FMT_GBRAP12, |
873 |
|
|
AV_PIX_FMT_NONE |
874 |
|
|
}, |
875 |
|
|
.caps_internal = FF_CODEC_CAP_INIT_CLEANUP, |
876 |
|
|
}; |
877 |
|
|
|