mirror of https://github.com/FFmpeg/FFmpeg.git
Also shorten HuffYUV context member names to avoid clutter.pull/72/merge
Diego Biurrun
11 years ago
parent
0d439fbede
commit
512f3ffe9b
14 changed files with 278 additions and 166 deletions
@ -0,0 +1,84 @@ |
||||
/*
|
||||
* This file is part of Libav. |
||||
* |
||||
* Libav is free software; you can redistribute it and/or |
||||
* modify it under the terms of the GNU Lesser General Public |
||||
* License as published by the Free Software Foundation; either |
||||
* version 2.1 of the License, or (at your option) any later version. |
||||
* |
||||
* Libav is distributed in the hope that it will be useful, |
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
||||
* Lesser General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU Lesser General Public |
||||
* License along with Libav; if not, write to the Free Software |
||||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
||||
*/ |
||||
|
||||
#include "config.h" |
||||
#include "libavutil/attributes.h" |
||||
#include "huffyuvencdsp.h" |
||||
#include "mathops.h" |
||||
|
||||
// 0x7f7f7f7f or 0x7f7f7f7f7f7f7f7f or whatever, depending on the cpu's native arithmetic size
|
||||
#define pb_7f (~0UL / 255 * 0x7f) |
||||
#define pb_80 (~0UL / 255 * 0x80) |
||||
|
||||
static void diff_bytes_c(uint8_t *dst, uint8_t *src1, uint8_t *src2, int w) |
||||
{ |
||||
long i; |
||||
|
||||
#if !HAVE_FAST_UNALIGNED |
||||
if ((long) src2 & (sizeof(long) - 1)) { |
||||
for (i = 0; i + 7 < w; i += 8) { |
||||
dst[i + 0] = src1[i + 0] - src2[i + 0]; |
||||
dst[i + 1] = src1[i + 1] - src2[i + 1]; |
||||
dst[i + 2] = src1[i + 2] - src2[i + 2]; |
||||
dst[i + 3] = src1[i + 3] - src2[i + 3]; |
||||
dst[i + 4] = src1[i + 4] - src2[i + 4]; |
||||
dst[i + 5] = src1[i + 5] - src2[i + 5]; |
||||
dst[i + 6] = src1[i + 6] - src2[i + 6]; |
||||
dst[i + 7] = src1[i + 7] - src2[i + 7]; |
||||
} |
||||
} else |
||||
#endif |
||||
for (i = 0; i <= w - (int) sizeof(long); i += sizeof(long)) { |
||||
long a = *(long *) (src1 + i); |
||||
long b = *(long *) (src2 + i); |
||||
*(long *) (dst + i) = ((a | pb_80) - (b & pb_7f)) ^ |
||||
((a ^ b ^ pb_80) & pb_80); |
||||
} |
||||
for (; i < w; i++) |
||||
dst[i + 0] = src1[i + 0] - src2[i + 0]; |
||||
} |
||||
|
||||
static void sub_hfyu_median_pred_c(uint8_t *dst, const uint8_t *src1, |
||||
const uint8_t *src2, int w, |
||||
int *left, int *left_top) |
||||
{ |
||||
int i; |
||||
uint8_t l, lt; |
||||
|
||||
l = *left; |
||||
lt = *left_top; |
||||
|
||||
for (i = 0; i < w; i++) { |
||||
const int pred = mid_pred(l, src1[i], (l + src1[i] - lt) & 0xFF); |
||||
lt = src1[i]; |
||||
l = src2[i]; |
||||
dst[i] = l - pred; |
||||
} |
||||
|
||||
*left = l; |
||||
*left_top = lt; |
||||
} |
||||
|
||||
av_cold void ff_huffyuvencdsp_init(HuffYUVEncDSPContext *c) |
||||
{ |
||||
c->diff_bytes = diff_bytes_c; |
||||
c->sub_hfyu_median_pred = sub_hfyu_median_pred_c; |
||||
|
||||
if (ARCH_X86) |
||||
ff_huffyuvencdsp_init_x86(c); |
||||
} |
||||
@ -0,0 +1,41 @@ |
||||
/*
|
||||
* This file is part of Libav. |
||||
* |
||||
* Libav is free software; you can redistribute it and/or |
||||
* modify it under the terms of the GNU Lesser General Public |
||||
* License as published by the Free Software Foundation; either |
||||
* version 2.1 of the License, or (at your option) any later version. |
||||
* |
||||
* Libav is distributed in the hope that it will be useful, |
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
||||
* Lesser General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU Lesser General Public |
||||
* License along with Libav; if not, write to the Free Software |
||||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
||||
*/ |
||||
|
||||
#ifndef AVCODEC_HUFFYUVENCDSP_H |
||||
#define AVCODEC_HUFFYUVENCDSP_H |
||||
|
||||
#include <stdint.h> |
||||
|
||||
typedef struct HuffYUVEncDSPContext { |
||||
void (*diff_bytes)(uint8_t *dst /* align 16 */, |
||||
uint8_t *src1 /* align 16 */, |
||||
uint8_t *src2 /* align 1 */, |
||||
int w); |
||||
/**
|
||||
* Subtract HuffYUV's variant of median prediction. |
||||
* Note, this might read from src1[-1], src2[-1]. |
||||
*/ |
||||
void (*sub_hfyu_median_pred)(uint8_t *dst, const uint8_t *src1, |
||||
const uint8_t *src2, int w, |
||||
int *left, int *left_top); |
||||
} HuffYUVEncDSPContext; |
||||
|
||||
void ff_huffyuvencdsp_init(HuffYUVEncDSPContext *c); |
||||
void ff_huffyuvencdsp_init_x86(HuffYUVEncDSPContext *c); |
||||
|
||||
#endif /* AVCODEC_HUFFYUVENCDSP_H */ |
||||
@ -0,0 +1,113 @@ |
||||
/*
|
||||
* SIMD-optimized HuffYUV encoding functions |
||||
* Copyright (c) 2000, 2001 Fabrice Bellard |
||||
* Copyright (c) 2002-2004 Michael Niedermayer <michaelni@gmx.at> |
||||
* |
||||
* MMX optimization by Nick Kurshev <nickols_k@mail.ru> |
||||
* |
||||
* This file is part of Libav. |
||||
* |
||||
* Libav is free software; you can redistribute it and/or |
||||
* modify it under the terms of the GNU Lesser General Public |
||||
* License as published by the Free Software Foundation; either |
||||
* version 2.1 of the License, or (at your option) any later version. |
||||
* |
||||
* Libav is distributed in the hope that it will be useful, |
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
||||
* Lesser General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU Lesser General Public |
||||
* License along with Libav; if not, write to the Free Software |
||||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
||||
*/ |
||||
|
||||
#include "libavutil/attributes.h" |
||||
#include "libavutil/cpu.h" |
||||
#include "libavutil/x86/asm.h" |
||||
#include "libavutil/x86/cpu.h" |
||||
#include "libavcodec/huffyuvencdsp.h" |
||||
#include "libavcodec/mathops.h" |
||||
|
||||
#if HAVE_INLINE_ASM |
||||
|
||||
static void diff_bytes_mmx(uint8_t *dst, uint8_t *src1, uint8_t *src2, int w) |
||||
{ |
||||
x86_reg i = 0; |
||||
|
||||
__asm__ volatile ( |
||||
"1: \n\t" |
||||
"movq (%2, %0), %%mm0 \n\t" |
||||
"movq (%1, %0), %%mm1 \n\t" |
||||
"psubb %%mm0, %%mm1 \n\t" |
||||
"movq %%mm1, (%3, %0) \n\t" |
||||
"movq 8(%2, %0), %%mm0 \n\t" |
||||
"movq 8(%1, %0), %%mm1 \n\t" |
||||
"psubb %%mm0, %%mm1 \n\t" |
||||
"movq %%mm1, 8(%3, %0) \n\t" |
||||
"add $16, %0 \n\t" |
||||
"cmp %4, %0 \n\t" |
||||
" jb 1b \n\t" |
||||
: "+r" (i) |
||||
: "r" (src1), "r" (src2), "r" (dst), "r" ((x86_reg) w - 15)); |
||||
|
||||
for (; i < w; i++) |
||||
dst[i + 0] = src1[i + 0] - src2[i + 0]; |
||||
} |
||||
|
||||
static void sub_hfyu_median_pred_mmxext(uint8_t *dst, const uint8_t *src1, |
||||
const uint8_t *src2, int w, |
||||
int *left, int *left_top) |
||||
{ |
||||
x86_reg i = 0; |
||||
uint8_t l, lt; |
||||
|
||||
__asm__ volatile ( |
||||
"movq (%1, %0), %%mm0 \n\t" // LT
|
||||
"psllq $8, %%mm0 \n\t" |
||||
"1: \n\t" |
||||
"movq (%1, %0), %%mm1 \n\t" // T
|
||||
"movq -1(%2, %0), %%mm2 \n\t" // L
|
||||
"movq (%2, %0), %%mm3 \n\t" // X
|
||||
"movq %%mm2, %%mm4 \n\t" // L
|
||||
"psubb %%mm0, %%mm2 \n\t" |
||||
"paddb %%mm1, %%mm2 \n\t" // L + T - LT
|
||||
"movq %%mm4, %%mm5 \n\t" // L
|
||||
"pmaxub %%mm1, %%mm4 \n\t" // max(T, L)
|
||||
"pminub %%mm5, %%mm1 \n\t" // min(T, L)
|
||||
"pminub %%mm2, %%mm4 \n\t" |
||||
"pmaxub %%mm1, %%mm4 \n\t" |
||||
"psubb %%mm4, %%mm3 \n\t" // dst - pred
|
||||
"movq %%mm3, (%3, %0) \n\t" |
||||
"add $8, %0 \n\t" |
||||
"movq -1(%1, %0), %%mm0 \n\t" // LT
|
||||
"cmp %4, %0 \n\t" |
||||
" jb 1b \n\t" |
||||
: "+r" (i) |
||||
: "r" (src1), "r" (src2), "r" (dst), "r" ((x86_reg) w)); |
||||
|
||||
l = *left; |
||||
lt = *left_top; |
||||
|
||||
dst[0] = src2[0] - mid_pred(l, src1[0], (l + src1[0] - lt) & 0xFF); |
||||
|
||||
*left_top = src1[w - 1]; |
||||
*left = src2[w - 1]; |
||||
} |
||||
|
||||
#endif /* HAVE_INLINE_ASM */ |
||||
|
||||
av_cold void ff_huffyuvencdsp_init_x86(HuffYUVEncDSPContext *c) |
||||
{ |
||||
#if HAVE_INLINE_ASM |
||||
int cpu_flags = av_get_cpu_flags(); |
||||
|
||||
if (INLINE_MMX(cpu_flags)) { |
||||
c->diff_bytes = diff_bytes_mmx; |
||||
} |
||||
|
||||
if (INLINE_MMXEXT(cpu_flags)) { |
||||
c->sub_hfyu_median_pred = sub_hfyu_median_pred_mmxext; |
||||
} |
||||
#endif /* HAVE_INLINE_ASM */ |
||||
} |
||||
Loading…
Reference in new issue