FFmpeg
hevcpred_init_aarch64.c
Go to the documentation of this file.
1 /*
2  * HEVC Intra Prediction NEON initialization
3  *
4  * Copyright (c) 2026 Jun Zhao <barryjzhao@tencent.com>
5  *
6  * This file is part of FFmpeg.
7  *
8  * FFmpeg is free software; you can redistribute it and/or
9  * modify it under the terms of the GNU Lesser General Public
10  * License as published by the Free Software Foundation; either
11  * version 2.1 of the License, or (at your option) any later version.
12  *
13  * FFmpeg 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 GNU
16  * Lesser General Public License for more details.
17  *
18  * You should have received a copy of the GNU Lesser General Public
19  * License along with FFmpeg; if not, write to the Free Software
20  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
21  */
22 
23 #include "libavutil/attributes.h"
24 #include "libavutil/avassert.h"
25 #include "libavutil/aarch64/cpu.h"
26 #include "libavcodec/hevc/pred.h"
27 
28 // DC prediction
29 void ff_hevc_pred_dc_4x4_8_neon(uint8_t *src, const uint8_t *top,
30  const uint8_t *left, ptrdiff_t stride,
31  int c_idx);
32 void ff_hevc_pred_dc_8x8_8_neon(uint8_t *src, const uint8_t *top,
33  const uint8_t *left, ptrdiff_t stride,
34  int c_idx);
35 void ff_hevc_pred_dc_16x16_8_neon(uint8_t *src, const uint8_t *top,
36  const uint8_t *left, ptrdiff_t stride,
37  int c_idx);
38 void ff_hevc_pred_dc_32x32_8_neon(uint8_t *src, const uint8_t *top,
39  const uint8_t *left, ptrdiff_t stride,
40  int c_idx);
41 
42 // Planar prediction
43 void ff_hevc_pred_planar_4x4_8_neon(uint8_t *src, const uint8_t *top,
44  const uint8_t *left, ptrdiff_t stride);
45 void ff_hevc_pred_planar_8x8_8_neon(uint8_t *src, const uint8_t *top,
46  const uint8_t *left, ptrdiff_t stride);
47 void ff_hevc_pred_planar_16x16_8_neon(uint8_t *src, const uint8_t *top,
48  const uint8_t *left, ptrdiff_t stride);
49 void ff_hevc_pred_planar_32x32_8_neon(uint8_t *src, const uint8_t *top,
50  const uint8_t *left, ptrdiff_t stride);
51 
52 static void pred_dc_neon(uint8_t *src, const uint8_t *top,
53  const uint8_t *left, ptrdiff_t stride,
54  int log2_size, int c_idx)
55 {
56  switch (log2_size) {
57  case 2:
59  break;
60  case 3:
62  break;
63  case 4:
65  break;
66  case 5:
68  break;
69  default:
70  av_unreachable("log2_size must be 2, 3, 4 or 5");
71  }
72 }
73 
75 {
77 
78  if (!have_neon(cpu_flags))
79  return;
80 
81  if (bit_depth == 8) {
82  hpc->pred_dc = pred_dc_neon;
87  }
88 }
av_get_cpu_flags
int av_get_cpu_flags(void)
Return the flags which specify extensions supported by the CPU.
Definition: cpu.c:109
bit_depth
static void bit_depth(AudioStatsContext *s, const uint64_t *const mask, uint8_t *depth)
Definition: af_astats.c:246
cpu_flags
static atomic_int cpu_flags
Definition: cpu.c:56
HEVCPredContext::pred_dc
void(* pred_dc)(uint8_t *src, const uint8_t *top, const uint8_t *left, ptrdiff_t stride, int log2_size, int c_idx)
Definition: pred.h:38
ff_hevc_pred_planar_8x8_8_neon
void ff_hevc_pred_planar_8x8_8_neon(uint8_t *src, const uint8_t *top, const uint8_t *left, ptrdiff_t stride)
ff_hevc_pred_planar_16x16_8_neon
void ff_hevc_pred_planar_16x16_8_neon(uint8_t *src, const uint8_t *top, const uint8_t *left, ptrdiff_t stride)
avassert.h
av_cold
#define av_cold
Definition: attributes.h:111
ff_hevc_pred_init_aarch64
av_cold void ff_hevc_pred_init_aarch64(HEVCPredContext *hpc, int bit_depth)
Definition: hevcpred_init_aarch64.c:74
HEVCPredContext
Definition: pred.h:32
ff_hevc_pred_dc_8x8_8_neon
void ff_hevc_pred_dc_8x8_8_neon(uint8_t *src, const uint8_t *top, const uint8_t *left, ptrdiff_t stride, int c_idx)
ff_hevc_pred_dc_16x16_8_neon
void ff_hevc_pred_dc_16x16_8_neon(uint8_t *src, const uint8_t *top, const uint8_t *left, ptrdiff_t stride, int c_idx)
HEVCPredContext::pred_planar
void(* pred_planar[4])(uint8_t *src, const uint8_t *top, const uint8_t *left, ptrdiff_t stride)
Definition: pred.h:36
av_unreachable
#define av_unreachable(msg)
Asserts that are used as compiler optimization hints depending upon ASSERT_LEVEL and NBDEBUG.
Definition: avassert.h:116
pred_dc_neon
static void pred_dc_neon(uint8_t *src, const uint8_t *top, const uint8_t *left, ptrdiff_t stride, int log2_size, int c_idx)
Definition: hevcpred_init_aarch64.c:52
ff_hevc_pred_planar_4x4_8_neon
void ff_hevc_pred_planar_4x4_8_neon(uint8_t *src, const uint8_t *top, const uint8_t *left, ptrdiff_t stride)
have_neon
#define have_neon(flags)
Definition: cpu.h:26
attributes.h
pred.h
left
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 left
Definition: snow.txt:386
ff_hevc_pred_dc_4x4_8_neon
void ff_hevc_pred_dc_4x4_8_neon(uint8_t *src, const uint8_t *top, const uint8_t *left, ptrdiff_t stride, int c_idx)
ff_hevc_pred_planar_32x32_8_neon
void ff_hevc_pred_planar_32x32_8_neon(uint8_t *src, const uint8_t *top, const uint8_t *left, ptrdiff_t stride)
stride
#define stride
Definition: h264pred_template.c:536
cpu.h
src
#define src
Definition: vp8dsp.c:248
ff_hevc_pred_dc_32x32_8_neon
void ff_hevc_pred_dc_32x32_8_neon(uint8_t *src, const uint8_t *top, const uint8_t *left, ptrdiff_t stride, int c_idx)