diff options
author | Paul B Mahol <onemda@gmail.com> | 2021-02-28 11:57:04 +0100 |
---|---|---|
committer | Paul B Mahol <onemda@gmail.com> | 2021-03-02 20:53:04 +0100 |
commit | cf2f8dcc6af440ef1be4da35331caa7d719f1230 (patch) | |
tree | 7dc487c556c860ff789e38f43c23d01b0ad64722 /libavcodec | |
parent | cc85ca1cb347570a95d8615b7d4c7b542042b7f0 (diff) | |
download | ffmpeg-cf2f8dcc6af440ef1be4da35331caa7d719f1230.tar.gz |
avcodec/exrenc: move float2half code to own header
Diffstat (limited to 'libavcodec')
-rw-r--r-- | libavcodec/exrenc.c | 52 | ||||
-rw-r--r-- | libavcodec/float2half.h | 67 |
2 files changed, 72 insertions, 47 deletions
diff --git a/libavcodec/exrenc.c b/libavcodec/exrenc.c index f9d802543d..db9000a85c 100644 --- a/libavcodec/exrenc.c +++ b/libavcodec/exrenc.c @@ -34,6 +34,7 @@ #include "avcodec.h" #include "bytestream.h" #include "internal.h" +#include "float2half.h" enum ExrCompr { EXR_RAW, @@ -87,54 +88,11 @@ typedef struct EXRContext { uint8_t shifttable[512]; } EXRContext; -static void half_tables(EXRContext *s) -{ - for (int i = 0; i < 256; i++) { - int e = i - 127; - - if (e < -24) { // Very small numbers map to zero - s->basetable[i|0x000] = 0x0000; - s->basetable[i|0x100] = 0x8000; - s->shifttable[i|0x000] = 24; - s->shifttable[i|0x100] = 24; - } else if (e < -14) { // Small numbers map to denorms - s->basetable[i|0x000] = (0x0400>>(-e-14)); - s->basetable[i|0x100] = (0x0400>>(-e-14)) | 0x8000; - s->shifttable[i|0x000] = -e-1; - s->shifttable[i|0x100] = -e-1; - } else if (e <= 15) { // Normal numbers just lose precision - s->basetable[i|0x000] = ((e + 15) << 10); - s->basetable[i|0x100] = ((e + 15) << 10) | 0x8000; - s->shifttable[i|0x000] = 13; - s->shifttable[i|0x100] = 13; - } else if (e < 128) { // Large numbers map to Infinity - s->basetable[i|0x000] = 0x7C00; - s->basetable[i|0x100] = 0xFC00; - s->shifttable[i|0x000] = 24; - s->shifttable[i|0x100] = 24; - } else{ // Infinity and NaN's stay Infinity and NaN's - s->basetable[i|0x000] = 0x7C00; - s->basetable[i|0x100] = 0xFC00; - s->shifttable[i|0x000] = 13; - s->shifttable[i|0x100] = 13; - } - } -} - -static uint16_t float2half(EXRContext *s, uint32_t f) -{ - uint16_t h; - - h = s->basetable[(f >> 23) & 0x1ff] + ((f & 0x007fffff) >> s->shifttable[(f >> 23) & 0x1ff]); - - return h; -} - static int encode_init(AVCodecContext *avctx) { EXRContext *s = avctx->priv_data; - half_tables(s); + float2half_tables(s->basetable, s->shifttable); switch (avctx->pix_fmt) { case AV_PIX_FMT_GBRPF32: @@ -290,7 +248,7 @@ static int encode_scanline_rle(EXRContext *s, const AVFrame *frame) uint32_t *src = (uint32_t *)(frame->data[ch] + y * frame->linesize[ch]); for (int x = 0; x < frame->width; x++) - dst[x] = float2half(s, src[x]); + dst[x] = float2half(src[x], s->basetable, s->shifttable); } break; } @@ -358,7 +316,7 @@ static int encode_scanline_zip(EXRContext *s, const AVFrame *frame) uint32_t *src = (uint32_t *)(frame->data[ch] + (y * s->scanline_height + l) * frame->linesize[ch]); for (int x = 0; x < frame->width; x++) - dst[x] = float2half(s, src[x]); + dst[x] = float2half(src[x], s->basetable, s->shifttable); } } break; @@ -516,7 +474,7 @@ static int encode_frame(AVCodecContext *avctx, AVPacket *pkt, uint32_t *src = (uint32_t *)(frame->data[ch] + y * frame->linesize[ch]); for (int x = 0; x < frame->width; x++) - bytestream2_put_le16(pb, float2half(s, src[x])); + bytestream2_put_le16(pb, float2half(src[x], s->basetable, s->shifttable)); } } } diff --git a/libavcodec/float2half.h b/libavcodec/float2half.h new file mode 100644 index 0000000000..ff547b490c --- /dev/null +++ b/libavcodec/float2half.h @@ -0,0 +1,67 @@ +/* + * This file is part of FFmpeg. + * + * FFmpeg 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. + * + * FFmpeg 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 FFmpeg; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA + */ + +#ifndef AVCODEC_FLOAT2HALF_H +#define AVCODEC_FLOAT2HALF_H + +#include <stdint.h> + +static void float2half_tables(uint16_t *basetable, uint8_t *shifttable) +{ + for (int i = 0; i < 256; i++) { + int e = i - 127; + + if (e < -24) { // Very small numbers map to zero + basetable[i|0x000] = 0x0000; + basetable[i|0x100] = 0x8000; + shifttable[i|0x000] = 24; + shifttable[i|0x100] = 24; + } else if (e < -14) { // Small numbers map to denorms + basetable[i|0x000] = (0x0400>>(-e-14)); + basetable[i|0x100] = (0x0400>>(-e-14)) | 0x8000; + shifttable[i|0x000] = -e-1; + shifttable[i|0x100] = -e-1; + } else if (e <= 15) { // Normal numbers just lose precision + basetable[i|0x000] = ((e + 15) << 10); + basetable[i|0x100] = ((e + 15) << 10) | 0x8000; + shifttable[i|0x000] = 13; + shifttable[i|0x100] = 13; + } else if (e < 128) { // Large numbers map to Infinity + basetable[i|0x000] = 0x7C00; + basetable[i|0x100] = 0xFC00; + shifttable[i|0x000] = 24; + shifttable[i|0x100] = 24; + } else{ // Infinity and NaN's stay Infinity and NaN's + basetable[i|0x000] = 0x7C00; + basetable[i|0x100] = 0xFC00; + shifttable[i|0x000] = 13; + shifttable[i|0x100] = 13; + } + } +} + +static uint16_t float2half(uint32_t f, uint16_t *basetable, uint8_t *shifttable) +{ + uint16_t h; + + h = basetable[(f >> 23) & 0x1ff] + ((f & 0x007fffff) >> shifttable[(f >> 23) & 0x1ff]); + + return h; +} + +#endif /* AVCODEC_FLOAT2HALF_H */ |