huffyuvencdsp: move functions only used by huffyuv from lossless_videodsp
[ffmpeg.git] / libavcodec / huffyuvencdsp.c
1 /*
2  * This file is part of FFmpeg.
3  *
4  * FFmpeg is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU Lesser General Public
6  * License as published by the Free Software Foundation; either
7  * version 2.1 of the License, or (at your option) any later version.
8  *
9  * FFmpeg is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12  * Lesser General Public License for more details.
13  *
14  * You should have received a copy of the GNU Lesser General Public
15  * License along with FFmpeg; if not, write to the Free Software
16  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17  */
18
19 #include "config.h"
20 #include "libavutil/attributes.h"
21 #include "huffyuvencdsp.h"
22 #include "mathops.h"
23
24 // 0x7f7f7f7f or 0x7f7f7f7f7f7f7f7f or whatever, depending on the cpu's native arithmetic size
25 #define pb_7f (~0UL / 255 * 0x7f)
26 #define pb_80 (~0UL / 255 * 0x80)
27
28 static void diff_bytes_c(uint8_t *dst, const uint8_t *src1, const uint8_t *src2, intptr_t w)
29 {
30     long i;
31
32 #if !HAVE_FAST_UNALIGNED
33     if (((long)src1 | (long)src2) & (sizeof(long) - 1)) {
34         for (i = 0; i + 7 < w; i += 8) {
35             dst[i + 0] = src1[i + 0] - src2[i + 0];
36             dst[i + 1] = src1[i + 1] - src2[i + 1];
37             dst[i + 2] = src1[i + 2] - src2[i + 2];
38             dst[i + 3] = src1[i + 3] - src2[i + 3];
39             dst[i + 4] = src1[i + 4] - src2[i + 4];
40             dst[i + 5] = src1[i + 5] - src2[i + 5];
41             dst[i + 6] = src1[i + 6] - src2[i + 6];
42             dst[i + 7] = src1[i + 7] - src2[i + 7];
43         }
44     } else
45 #endif
46     for (i = 0; i <= w - (int) sizeof(long); i += sizeof(long)) {
47         long a = *(long *) (src1 + i);
48         long b = *(long *) (src2 + i);
49         *(long *) (dst + i) = ((a | pb_80) - (b & pb_7f)) ^
50                               ((a ^ b ^ pb_80) & pb_80);
51     }
52     for (; i < w; i++)
53         dst[i + 0] = src1[i + 0] - src2[i + 0];
54 }
55
56 static void diff_int16_c(uint16_t *dst, const uint16_t *src1, const uint16_t *src2, unsigned mask, int w){
57     long i;
58 #if !HAVE_FAST_UNALIGNED
59     if((long)src2 & (sizeof(long)-1)){
60         for(i=0; i+3<w; i+=4){
61             dst[i+0] = (src1[i+0]-src2[i+0]) & mask;
62             dst[i+1] = (src1[i+1]-src2[i+1]) & mask;
63             dst[i+2] = (src1[i+2]-src2[i+2]) & mask;
64             dst[i+3] = (src1[i+3]-src2[i+3]) & mask;
65         }
66     }else
67 #endif
68     {
69         unsigned long pw_lsb = (mask >> 1) * 0x0001000100010001ULL;
70         unsigned long pw_msb = pw_lsb +  0x0001000100010001ULL;
71
72         for (i = 0; i <= w - (int)sizeof(long)/2; i += sizeof(long)/2) {
73             long a = *(long*)(src1+i);
74             long b = *(long*)(src2+i);
75             *(long*)(dst+i) = ((a|pw_msb) - (b&pw_lsb)) ^ ((a^b^pw_msb)&pw_msb);
76         }
77     }
78     for (; i<w; i++)
79         dst[i] = (src1[i] - src2[i]) & mask;
80 }
81
82 static void sub_hfyu_median_pred_c(uint8_t *dst, const uint8_t *src1,
83                                    const uint8_t *src2, intptr_t w,
84                                    int *left, int *left_top)
85 {
86     int i;
87     uint8_t l, lt;
88
89     l  = *left;
90     lt = *left_top;
91
92     for (i = 0; i < w; i++) {
93         const int pred = mid_pred(l, src1[i], (l + src1[i] - lt) & 0xFF);
94         lt     = src1[i];
95         l      = src2[i];
96         dst[i] = l - pred;
97     }
98
99     *left     = l;
100     *left_top = lt;
101 }
102
103 static void sub_hfyu_median_pred_int16_c(uint16_t *dst, const uint16_t *src1, const uint16_t *src2, unsigned mask, int w, int *left, int *left_top){
104     int i;
105     uint16_t l, lt;
106
107     l  = *left;
108     lt = *left_top;
109
110     for(i=0; i<w; i++){
111         const int pred = mid_pred(l, src1[i], (l + src1[i] - lt) & mask);
112         lt = src1[i];
113         l  = src2[i];
114         dst[i] = (l - pred) & mask;
115     }
116
117     *left     = l;
118     *left_top = lt;
119 }
120
121 av_cold void ff_huffyuvencdsp_init(HuffYUVEncDSPContext *c, AVCodecContext *avctx)
122 {
123     c->diff_bytes           = diff_bytes_c;
124     c->diff_int16           = diff_int16_c;
125     c->sub_hfyu_median_pred = sub_hfyu_median_pred_c;
126     c->sub_hfyu_median_pred_int16 = sub_hfyu_median_pred_int16_c;
127
128     if (ARCH_X86)
129         ff_huffyuvencdsp_init_x86(c, avctx);
130 }