FFmpeg
rv34dsp.c
Go to the documentation of this file.
1 /*
2  * Copyright (c) 2024 Institute of Software Chinese Academy of Sciences (ISCAS).
3  *
4  * This file is part of FFmpeg.
5  *
6  * FFmpeg 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  * 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
14  * GNU General Public License for more details.
15  *
16  * You should have received a copy of the GNU General Public License along
17  * with FFmpeg; if not, write to the Free Software Foundation, Inc.,
18  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
19  */
20 
21 #include <stddef.h>
22 #include <stdint.h>
23 #include <string.h>
24 
25 #include "libavutil/intreadwrite.h"
26 #include "libavutil/macros.h"
27 #include "libavutil/mem_internal.h"
28 
29 #include "libavcodec/mathops.h"
30 #include "libavcodec/rv34dsp.h"
31 
32 #include "checkasm.h"
33 
34 #define randomize(buf, len) \
35  do { \
36  for (int i = 0; i < len; i++) \
37  buf[i] = rnd(); \
38  } while (0)
39 
41  declare_func(void, int16_t *block);
42 
43  if (check_func(s->rv34_inv_transform_dc, "rv34_inv_transform_dc")) {
44  DECLARE_ALIGNED_16(int16_t, p1)[4*4];
45  DECLARE_ALIGNED_16(int16_t, p2)[4*4];
46 
47  randomize(p1, FF_ARRAY_ELEMS(p1));
48  memcpy(p2, p1, sizeof(p1));
49 
50  call_ref(p1);
51  call_new(p2);
52 
53  if (memcmp(p1, p2, sizeof(p1))) {
54  fail();
55  }
56 
57  bench_new(p1);
58  }
59 
60  report("rv34_inv_transform_dc");
61 }
62 
64  declare_func(void, uint8_t *dst, ptrdiff_t stride, int dc);
65 
66  if (check_func(s->rv34_idct_dc_add, "rv34_idct_dc_add")) {
67  DECLARE_ALIGNED_16(uint8_t, p1)[4*4];
68  DECLARE_ALIGNED_16(uint8_t, p2)[4*4];
69 
70  randomize(p1, FF_ARRAY_ELEMS(p1));
71  memcpy(p2, p1, sizeof(p1));
72 
73  call_ref(p1, 4, 5);
74  call_new(p2, 4, 5);
75 
76  if (memcmp(p1, p2, sizeof(p1))) {
77  fail();
78  }
79 
80  bench_new(p1, 4, 5);
81  }
82 
83  report("rv34_idct_dc_add");
84 }
85 
86 static void test_rv34_idct_add(const RV34DSPContext *const s)
87 {
88  enum {
89  MAX_STRIDE = 256, ///< arbitrary, should be divisible by four
90  };
91  declare_func(void, uint8_t *dst, ptrdiff_t stride, int16_t *block);
92 
93  if (check_func(s->rv34_idct_add, "rv34_idct_add")) {
94  DECLARE_ALIGNED_16(int16_t, block_ref)[4*4];
95  DECLARE_ALIGNED_16(int16_t, block_new)[4*4];
96 
97  DECLARE_ALIGNED_4(uint8_t, dst_ref)[4*MAX_STRIDE + 4];
98  DECLARE_ALIGNED_4(uint8_t, dst_new)[4*MAX_STRIDE + 4];
99 
100  ptrdiff_t stride = FFALIGN(1 + rnd() % MAX_STRIDE, 4);
101  uint8_t *dst_refp = dst_ref, *dst_newp = dst_new;
102 
103  if (rnd() & 1) { // negate stride
104  dst_refp += 3 * stride;
105  dst_newp += 3 * stride;
106  stride = -stride;
107  }
108 
109  for (size_t i = 0; i < FF_ARRAY_ELEMS(block_ref); ++i)
110  block_ref[i] = sign_extend(rnd(), 10);
111  for (size_t i = 0; i < sizeof(dst_ref); i += 4)
112  AV_WN32A(dst_ref + i, rnd());
113  memcpy(block_new, block_ref, sizeof(block_new));
114  memcpy(dst_new, dst_ref, sizeof(dst_new));
115 
116  call_ref(dst_refp, stride, block_ref);
117  call_new(dst_newp, stride, block_new);
118 
119  if (memcmp(dst_ref, dst_new, sizeof(dst_new)) ||
120  memcmp(block_ref, block_new, sizeof(block_new)))
121  fail();
122 
123  bench_new(dst_new, stride, block_new);
124  }
125 
126  report("rv34_idct_add");
127 }
128 
130 {
131  RV34DSPContext s = { 0 };
132  ff_rv34dsp_init(&s);
133 
137 }
mem_internal.h
test_rv34_idct_dc_add
static void test_rv34_idct_dc_add(RV34DSPContext *s)
Definition: rv34dsp.c:63
check_func
#define check_func(func,...)
Definition: checkasm.h:214
AV_WN32A
#define AV_WN32A(p, v)
Definition: intreadwrite.h:534
ff_rv34dsp_init
av_cold void ff_rv34dsp_init(RV34DSPContext *c)
Definition: rv34dsp.c:131
call_ref
#define call_ref(...)
Definition: checkasm.h:230
macros.h
fail
#define fail()
Definition: checkasm.h:224
checkasm.h
rnd
#define rnd()
Definition: checkasm.h:207
FF_ARRAY_ELEMS
#define FF_ARRAY_ELEMS(a)
Definition: sinewin_tablegen.c:29
intreadwrite.h
s
#define s(width, name)
Definition: cbs_vp9.c:198
test_rv34_inv_transform_dc
static void test_rv34_inv_transform_dc(RV34DSPContext *s)
Definition: rv34dsp.c:40
call_new
#define call_new(...)
Definition: checkasm.h:238
mathops.h
RV34DSPContext
Definition: rv34dsp.h:57
dc
Tag MUST be and< 10hcoeff half pel interpolation filter coefficients, hcoeff[0] are the 2 middle coefficients[1] are the next outer ones and so on, resulting in a filter like:...eff[2], hcoeff[1], hcoeff[0], hcoeff[0], hcoeff[1], hcoeff[2] ... the sign of the coefficients is not explicitly stored but alternates after each coeff and coeff[0] is positive, so ...,+,-,+,-,+,+,-,+,-,+,... hcoeff[0] is not explicitly stored but found by subtracting the sum of all stored coefficients with signs from 32 hcoeff[0]=32 - hcoeff[1] - hcoeff[2] - ... a good choice for hcoeff and htaps is htaps=6 hcoeff={40,-10, 2} an alternative which requires more computations at both encoder and decoder side and may or may not be better is htaps=8 hcoeff={42,-14, 6,-2}ref_frames minimum of the number of available reference frames and max_ref_frames for example the first frame after a key frame always has ref_frames=1spatial_decomposition_type wavelet type 0 is a 9/7 symmetric compact integer wavelet 1 is a 5/3 symmetric compact integer wavelet others are reserved stored as delta from last, last is reset to 0 if always_reset||keyframeqlog quality(logarithmic quantizer scale) stored as delta from last, last is reset to 0 if always_reset||keyframemv_scale stored as delta from last, last is reset to 0 if always_reset||keyframe FIXME check that everything works fine if this changes between framesqbias dequantization bias stored as delta from last, last is reset to 0 if always_reset||keyframeblock_max_depth maximum depth of the block tree stored as delta from last, last is reset to 0 if always_reset||keyframequant_table quantization tableHighlevel bitstream structure:==============================--------------------------------------------|Header|--------------------------------------------|------------------------------------|||Block0||||split?||||yes no||||......... intra?||||:Block01 :yes no||||:Block02 :....... ..........||||:Block03 ::y DC ::ref index:||||:Block04 ::cb DC ::motion x :||||......... :cr DC ::motion y :||||....... ..........|||------------------------------------||------------------------------------|||Block1|||...|--------------------------------------------|------------ ------------ ------------|||Y subbands||Cb subbands||Cr subbands||||--- ---||--- ---||--- ---|||||LL0||HL0||||LL0||HL0||||LL0||HL0|||||--- ---||--- ---||--- ---||||--- ---||--- ---||--- ---|||||LH0||HH0||||LH0||HH0||||LH0||HH0|||||--- ---||--- ---||--- ---||||--- ---||--- ---||--- ---|||||HL1||LH1||||HL1||LH1||||HL1||LH1|||||--- ---||--- ---||--- ---||||--- ---||--- ---||--- ---|||||HH1||HL2||||HH1||HL2||||HH1||HL2|||||...||...||...|||------------ ------------ ------------|--------------------------------------------Decoding process:=================------------|||Subbands|------------||||------------|Intra DC||||LL0 subband prediction ------------|\ Dequantization ------------------- \||Reference frames|\ IDWT|------- -------|Motion \|||Frame 0||Frame 1||Compensation . OBMC v -------|------- -------|--------------. \------> Frame n output Frame Frame<----------------------------------/|...|------------------- Range Coder:============Binary Range Coder:------------------- The implemented range coder is an adapted version based upon "Range encoding: an algorithm for removing redundancy from a digitised message." by G. N. N. Martin. The symbols encoded by the Snow range coder are bits(0|1). The associated probabilities are not fix but change depending on the symbol mix seen so far. bit seen|new state ---------+----------------------------------------------- 0|256 - state_transition_table[256 - old_state];1|state_transition_table[old_state];state_transition_table={ 0, 0, 0, 0, 0, 0, 0, 0, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 114, 115, 116, 117, 118, 119, 120, 121, 122, 123, 124, 125, 126, 127, 128, 129, 130, 131, 132, 133, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 190, 191, 192, 194, 194, 195, 196, 197, 198, 199, 200, 201, 202, 202, 204, 205, 206, 207, 208, 209, 209, 210, 211, 212, 213, 215, 215, 216, 217, 218, 219, 220, 220, 222, 223, 224, 225, 226, 227, 227, 229, 229, 230, 231, 232, 234, 234, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 248, 0, 0, 0, 0, 0, 0, 0};FIXME Range Coding of integers:------------------------- FIXME Neighboring Blocks:===================left and top are set to the respective blocks unless they are outside of the image in which case they are set to the Null block top-left is set to the top left block unless it is outside of the image in which case it is set to the left block if this block has no larger parent block or it is at the left side of its parent block and the top right block is not outside of the image then the top right block is used for top-right else the top-left block is used Null block y, cb, cr are 128 level, ref, mx and my are 0 Motion Vector Prediction:=========================1. the motion vectors of all the neighboring blocks are scaled to compensate for the difference of reference frames scaled_mv=(mv *(256 *(current_reference+1)/(mv.reference+1))+128)> the median of the scaled top and top right vectors is used as motion vector prediction the used motion vector is the sum of the predictor and(mvx_diff, mvy_diff) *mv_scale Intra DC Prediction block[y][x] dc[1]
Definition: snow.txt:400
dst
uint8_t ptrdiff_t const uint8_t ptrdiff_t int intptr_t intptr_t int int16_t * dst
Definition: dsp.h:87
i
#define i(width, name, range_min, range_max)
Definition: cbs_h264.c:63
rv34dsp.h
randomize
#define randomize(buf, len)
Definition: rv34dsp.c:34
checkasm_check_rv34dsp
void checkasm_check_rv34dsp(void)
Definition: rv34dsp.c:129
report
#define report
Definition: checkasm.h:227
bench_new
#define bench_new(...)
Definition: checkasm.h:429
MAX_STRIDE
#define MAX_STRIDE
Definition: hpeldsp.c:31
sign_extend
static av_const int sign_extend(int val, unsigned bits)
Definition: mathops.h:135
DECLARE_ALIGNED_4
#define DECLARE_ALIGNED_4(t, v)
Definition: mem_internal.h:110
test_rv34_idct_add
static void test_rv34_idct_add(const RV34DSPContext *const s)
Definition: rv34dsp.c:86
declare_func
#define declare_func(ret,...)
Definition: checkasm.h:219
FFALIGN
#define FFALIGN(x, a)
Definition: macros.h:78
DECLARE_ALIGNED_16
#define DECLARE_ALIGNED_16(t, v)
Definition: mem_internal.h:112
block
The exact code depends on how similar the blocks are and how related they are to the block
Definition: filter_design.txt:207
stride
#define stride
Definition: h264pred_template.c:536