softfloat.h 3.77 KB
Newer Older
1 2 3
/*
 * Copyright (c) 2006 Michael Niedermayer <michaelni@gmx.at>
 *
4 5 6
 * This file is part of FFmpeg.
 *
 * FFmpeg is free software; you can redistribute it and/or
7 8
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
9
 * version 2.1 of the License, or (at your option) any later version.
10
 *
11
 * FFmpeg is distributed in the hope that it will be useful,
12 13 14 15 16
 * 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
17
 * License along with FFmpeg; if not, write to the Free Software
18 19
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 */
20

21 22
#ifndef AVUTIL_SOFTFLOAT_H
#define AVUTIL_SOFTFLOAT_H
23

24
#include <stdint.h>
25
#include "common.h"
26

27 28
#include "avassert.h"

29 30 31 32 33 34 35 36 37
#define MIN_EXP -126
#define MAX_EXP  126
#define ONE_BITS 29

typedef struct SoftFloat{
    int32_t  exp;
    int32_t mant;
}SoftFloat;

38
static av_const SoftFloat av_normalize_sf(SoftFloat a){
39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59
    if(a.mant){
#if 1
        while((a.mant + 0x20000000U)<0x40000000U){
            a.mant += a.mant;
            a.exp  -= 1;
        }
#else
        int s=ONE_BITS + 1 - av_log2(a.mant ^ (a.mant<<1));
        a.exp   -= s;
        a.mant <<= s;
#endif
        if(a.exp < MIN_EXP){
            a.exp = MIN_EXP;
            a.mant= 0;
        }
    }else{
        a.exp= MIN_EXP;
    }
    return a;
}

60
static inline av_const SoftFloat av_normalize1_sf(SoftFloat a){
61
#if 1
62
    if((int32_t)(a.mant + 0x40000000U) < 0){
63 64 65
        a.exp++;
        a.mant>>=1;
    }
66
    av_assert2(a.mant < 0x40000000 && a.mant > -0x40000000);
67 68 69 70 71 72 73 74 75 76 77
    return a;
#elif 1
    int t= a.mant + 0x40000000 < 0;
    return (SoftFloat){a.exp+t, a.mant>>t};
#else
    int t= (a.mant + 0x40000000U)>>31;
    return (SoftFloat){a.exp+t, a.mant>>t};
#endif
}

/**
78 79 80
 * @return Will not be more denormalized than a+b. So if either input is
 *         normalized, then the output will not be worse then the other input.
 *         If both are normalized, then the output will be normalized.
81
 */
82
static inline av_const SoftFloat av_mul_sf(SoftFloat a, SoftFloat b){
83
    a.exp += b.exp;
84
    av_assert2((int32_t)((a.mant * (int64_t)b.mant) >> ONE_BITS) == (a.mant * (int64_t)b.mant) >> ONE_BITS);
85 86 87 88 89
    a.mant = (a.mant * (int64_t)b.mant) >> ONE_BITS;
    return av_normalize1_sf(a);
}

/**
90 91
 * b has to be normalized and not zero.
 * @return Will not be more denormalized than a.
92
 */
93
static av_const SoftFloat av_div_sf(SoftFloat a, SoftFloat b){
94 95 96 97 98
    a.exp -= b.exp+1;
    a.mant = ((int64_t)a.mant<<(ONE_BITS+1)) / b.mant;
    return av_normalize1_sf(a);
}

99
static inline av_const int av_cmp_sf(SoftFloat a, SoftFloat b){
100 101 102 103 104
    int t= a.exp - b.exp;
    if(t<0) return (a.mant >> (-t)) -  b.mant      ;
    else    return  a.mant          - (b.mant >> t);
}

105
static inline av_const SoftFloat av_add_sf(SoftFloat a, SoftFloat b){
106
    int t= a.exp - b.exp;
107 108 109 110
    if      (t <-31) return b;
    else if (t <  0) return av_normalize1_sf((SoftFloat){b.exp, b.mant + (a.mant >> (-t))});
    else if (t < 32) return av_normalize1_sf((SoftFloat){a.exp, a.mant + (b.mant >>   t )});
    else             return a;
111 112
}

113
static inline av_const SoftFloat av_sub_sf(SoftFloat a, SoftFloat b){
114 115 116 117 118
    return av_add_sf(a, (SoftFloat){b.exp, -b.mant});
}

//FIXME sqrt, log, exp, pow, sin, cos

119
static inline av_const SoftFloat av_int2sf(int v, int frac_bits){
120 121 122 123
    return av_normalize_sf((SoftFloat){ONE_BITS-frac_bits, v});
}

/**
124
 * Rounding is to -inf.
125
 */
126
static inline av_const int av_sf2int(SoftFloat v, int frac_bits){
127 128 129 130
    v.exp += frac_bits - ONE_BITS;
    if(v.exp >= 0) return v.mant <<  v.exp ;
    else           return v.mant >>(-v.exp);
}
131

132
#endif /* AVUTIL_SOFTFLOAT_H */