[FFmpeg-devel] [PATCH 1/4] avcodec/snow: Move ff_snow_inner_add_yblock() to snow_dwt.c
Andreas Rheinhardt
andreas.rheinhardt at outlook.com
Tue Sep 20 00:27:49 EEST 2022
Only used there and by x86 snow asm code as fallback.
Signed-off-by: Andreas Rheinhardt <andreas.rheinhardt at outlook.com>
---
libavcodec/snow.c | 33 ---------------------------------
libavcodec/snow.h | 3 ---
libavcodec/snow_dwt.c | 32 ++++++++++++++++++++++++++++++++
libavcodec/snow_dwt.h | 3 +++
4 files changed, 35 insertions(+), 36 deletions(-)
diff --git a/libavcodec/snow.c b/libavcodec/snow.c
index aa15fccc42..85ad6d10a1 100644
--- a/libavcodec/snow.c
+++ b/libavcodec/snow.c
@@ -29,39 +29,6 @@
#include "snowdata.h"
-void ff_snow_inner_add_yblock(const uint8_t *obmc, const int obmc_stride, uint8_t * * block, int b_w, int b_h,
- int src_x, int src_y, int src_stride, slice_buffer * sb, int add, uint8_t * dst8){
- int y, x;
- IDWTELEM * dst;
- for(y=0; y<b_h; y++){
- //FIXME ugly misuse of obmc_stride
- const uint8_t *obmc1= obmc + y*obmc_stride;
- const uint8_t *obmc2= obmc1+ (obmc_stride>>1);
- const uint8_t *obmc3= obmc1+ obmc_stride*(obmc_stride>>1);
- const uint8_t *obmc4= obmc3+ (obmc_stride>>1);
- dst = slice_buffer_get_line(sb, src_y + y);
- for(x=0; x<b_w; x++){
- int v= obmc1[x] * block[3][x + y*src_stride]
- +obmc2[x] * block[2][x + y*src_stride]
- +obmc3[x] * block[1][x + y*src_stride]
- +obmc4[x] * block[0][x + y*src_stride];
-
- v <<= 8 - LOG2_OBMC_MAX;
- if(FRAC_BITS != 8){
- v >>= 8 - FRAC_BITS;
- }
- if(add){
- v += dst[x + src_x];
- v = (v + (1<<(FRAC_BITS-1))) >> FRAC_BITS;
- if(v&(~255)) v= ~(v>>31);
- dst8[x + y*src_stride] = v;
- }else{
- dst[x + src_x] -= v;
- }
- }
- }
-}
-
int ff_snow_get_buffer(SnowContext *s, AVFrame *frame)
{
int ret, i;
diff --git a/libavcodec/snow.h b/libavcodec/snow.h
index ed0f9abb42..1c976b9ba7 100644
--- a/libavcodec/snow.h
+++ b/libavcodec/snow.h
@@ -45,11 +45,8 @@
#define QSHIFT 5
#define QROOT (1<<QSHIFT)
#define LOSSLESS_QLOG -128
-#define FRAC_BITS 4
#define MAX_REF_FRAMES 8
-#define LOG2_OBMC_MAX 8
-#define OBMC_MAX (1<<(LOG2_OBMC_MAX))
typedef struct BlockNode{
int16_t mx; ///< Motion vector component X, see mv_scale
int16_t my; ///< Motion vector component Y, see mv_scale
diff --git a/libavcodec/snow_dwt.c b/libavcodec/snow_dwt.c
index 18b315ef66..9401d119d0 100644
--- a/libavcodec/snow_dwt.c
+++ b/libavcodec/snow_dwt.c
@@ -25,6 +25,38 @@
#include "me_cmp.h"
#include "snow_dwt.h"
+void ff_snow_inner_add_yblock(const uint8_t *obmc, const int obmc_stride,
+ uint8_t **block, int b_w, int b_h,
+ int src_x, int src_y, int src_stride,
+ slice_buffer * sb, int add, uint8_t * dst8)
+{
+ for (int y = 0; y < b_h; y++) {
+ //FIXME ugly misuse of obmc_stride
+ const uint8_t *obmc1 = obmc + y * obmc_stride;
+ const uint8_t *obmc2 = obmc1 + (obmc_stride >> 1);
+ const uint8_t *obmc3 = obmc1 + obmc_stride * (obmc_stride >> 1);
+ const uint8_t *obmc4 = obmc3 + (obmc_stride >> 1);
+ IDWTELEM *dst = slice_buffer_get_line(sb, src_y + y);
+ for (int x = 0; x < b_w; x++) {
+ int v = obmc1[x] * block[3][x + y*src_stride]
+ + obmc2[x] * block[2][x + y*src_stride]
+ + obmc3[x] * block[1][x + y*src_stride]
+ + obmc4[x] * block[0][x + y*src_stride];
+
+ v <<= 8 - LOG2_OBMC_MAX;
+ if (FRAC_BITS != 8)
+ v >>= 8 - FRAC_BITS;
+ if (add) {
+ v += dst[x + src_x];
+ v = (v + (1 << (FRAC_BITS - 1))) >> FRAC_BITS;
+ if (v & (~255)) v= ~(v>>31);
+ dst8[x + y*src_stride] = v;
+ } else
+ dst[x + src_x] -= v;
+ }
+ }
+}
+
int ff_slice_buffer_init(slice_buffer *buf, int line_count,
int max_allocated_lines, int line_width,
IDWTELEM *base_buffer)
diff --git a/libavcodec/snow_dwt.h b/libavcodec/snow_dwt.h
index 15b8a3007b..132e01f9ce 100644
--- a/libavcodec/snow_dwt.h
+++ b/libavcodec/snow_dwt.h
@@ -29,6 +29,9 @@ struct MpegEncContext;
typedef int DWTELEM;
typedef short IDWTELEM;
+#define FRAC_BITS 4
+#define LOG2_OBMC_MAX 8
+
#define MAX_DECOMPOSITIONS 8
typedef struct DWTCompose {
--
2.34.1
More information about the ffmpeg-devel
mailing list