1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
/*
Copyright (C) 2014 Fredrik Johansson
This file is part of FLINT.
FLINT is free software: you can redistribute it and/or modify it under
the terms of the GNU Lesser General Public License (LGPL) as published
by the Free Software Foundation; either version 3 of the License, or
(at your option) any later version. See <https://www.gnu.org/licenses/>.
*/
#include <math.h>
#include "mag.h"
void
mag_rsqrt(mag_t y, const mag_t x)
{
if (mag_is_special(x))
{
if (mag_is_zero(x))
mag_inf(y);
else
mag_zero(y);
}
else
{
double t;
fmpz e;
t = MAG_MAN(x) * ldexp(1.0, -MAG_BITS);
if (MAG_IS_LAGOM(x))
{
e = MAG_EXP(x);
if (e % 2 != 0)
{
e = (1 - e) >> 1;
t *= 2.0;
}
else
{
e = (-e) >> 1;
}
t = (1.0 / sqrt(t)) * (1 + 1e-13);
_fmpz_demote(MAG_EXPREF(y));
MAG_SET_D_2EXP(MAG_MAN(y), MAG_EXP(y), t, e);
}
else
{
if (fmpz_is_odd(MAG_EXPREF(x)))
t *= 2.0;
fmpz_fdiv_q_2exp(MAG_EXPREF(y), MAG_EXPREF(x), 1);
fmpz_neg(MAG_EXPREF(y), MAG_EXPREF(y));
t = (1.0 / sqrt(t)) * (1 + 1e-13);
mag_set_d_2exp_fmpz(y, t, MAG_EXPREF(y));
}
}
}
void
mag_rsqrt_lower(mag_t y, const mag_t x)
{
if (mag_is_special(x))
{
if (mag_is_zero(x))
mag_inf(y);
else
mag_zero(y);
}
else
{
double t;
fmpz e;
t = MAG_MAN(x) * ldexp(1.0, -MAG_BITS);
if (MAG_IS_LAGOM(x))
{
e = MAG_EXP(x);
if (e % 2 != 0)
{
e = (1 - e) >> 1;
t *= 2.0;
}
else
{
e = (-e) >> 1;
}
t = (1.0 / sqrt(t)) * (1 - 1e-13);
_fmpz_demote(MAG_EXPREF(y));
MAG_SET_D_2EXP_LOWER(MAG_MAN(y), MAG_EXP(y), t, e);
}
else
{
if (fmpz_is_odd(MAG_EXPREF(x)))
t *= 2.0;
fmpz_fdiv_q_2exp(MAG_EXPREF(y), MAG_EXPREF(x), 1);
fmpz_neg(MAG_EXPREF(y), MAG_EXPREF(y));
t = (1.0 / sqrt(t)) * (1 - 1e-13);
mag_set_d_2exp_fmpz_lower(y, t, MAG_EXPREF(y));
}
}
}