aboutsummaryrefslogtreecommitdiff
path: root/std/math/tanh.zig
diff options
context:
space:
mode:
authorMarc Tiehuis <marctiehuis@gmail.com>2017-06-16 20:26:10 +1200
committerMarc Tiehuis <marctiehuis@gmail.com>2017-06-16 20:32:31 +1200
commit4c16f9a3c35b23b9917f2a27b91ba8cd20e6fd82 (patch)
tree778f0f06734f7dc17e9269ee1cf5b513f7b504c0 /std/math/tanh.zig
parent865b53f2860405a718262abf9a794d2bf9529dbc (diff)
downloadzig-4c16f9a3c35b23b9917f2a27b91ba8cd20e6fd82.tar.gz
zig-4c16f9a3c35b23b9917f2a27b91ba8cd20e6fd82.zip
Add math library
This covers the majority of the functions as covered by the C99 specification for a math library. Code is adapted primarily from musl libc, with the pow and standard trigonometric functions adapted from the Go stdlib. Changes: - Remove assert expose in index and import as needed. - Add float log function and merge with existing base 2 integer implementation. See https://github.com/tiehuis/zig-fmath. See #374.
Diffstat (limited to 'std/math/tanh.zig')
-rw-r--r--std/math/tanh.zig120
1 files changed, 120 insertions, 0 deletions
diff --git a/std/math/tanh.zig b/std/math/tanh.zig
new file mode 100644
index 0000000000..6f64a7ef87
--- /dev/null
+++ b/std/math/tanh.zig
@@ -0,0 +1,120 @@
+const math = @import("index.zig");
+const assert = @import("../debug.zig").assert;
+const expo2 = @import("_expo2.zig").expo2;
+
+pub fn tanh(x: var) -> @typeOf(x) {
+ const T = @typeOf(x);
+ switch (T) {
+ f32 => @inlineCall(tanhf, x),
+ f64 => @inlineCall(tanhd, x),
+ else => @compileError("tanh not implemented for " ++ @typeName(T)),
+ }
+}
+
+// tanh(x) = (exp(x) - exp(-x)) / (exp(x) + exp(-x))
+// = (exp(2x) - 1) / (exp(2x) - 1 + 2)
+// = (1 - exp(-2x)) / (exp(-2x) - 1 + 2)
+fn tanhf(x: f32) -> f32 {
+ const u = @bitCast(u32, x);
+ const ux = u & 0x7FFFFFFF;
+ const ax = @bitCast(f32, ux);
+
+ var t: f32 = undefined;
+
+ // |x| < log(3) / 2 ~= 0.5493 or nan
+ if (ux > 0x3F0C9F54) {
+ // |x| > 10
+ if (ux > 0x41200000) {
+ t = 1.0 + 0 / x;
+ } else {
+ t = math.expm1(2 * x);
+ t = 1 - 2 / (t + 2);
+ }
+ }
+ // |x| > log(5 / 3) / 2 ~= 0.2554
+ else if (ux > 0x3E82C578) {
+ t = math.expm1(2 * x);
+ t = t / (t + 2);
+ }
+ // |x| >= 0x1.0p-126
+ else if (ux >= 0x00800000) {
+ t = math.expm1(-2 * x);
+ t = -t / (t + 2);
+ }
+ // |x| is subnormal
+ else {
+ math.forceEval(x * x);
+ t = x;
+ }
+
+ if (u >> 31 != 0) {
+ -t
+ } else {
+ t
+ }
+}
+
+fn tanhd(x: f64) -> f64 {
+ const u = @bitCast(u64, x);
+ const w = u32(u >> 32);
+ const ax = @bitCast(f64, u & (@maxValue(u64) >> 1));
+
+ var t: f64 = undefined;
+
+ // |x| < log(3) / 2 ~= 0.5493 or nan
+ if (w > 0x3Fe193EA) {
+ // |x| > 20 or nan
+ if (w > 0x40340000) {
+ t = 1.0 + 0 / x;
+ } else {
+ t = math.expm1(2 * x);
+ t = 1 - 2 / (t + 2);
+ }
+ }
+ // |x| > log(5 / 3) / 2 ~= 0.2554
+ else if (w > 0x3FD058AE) {
+ t = math.expm1(2 * x);
+ t = t / (t + 2);
+ }
+ // |x| >= 0x1.0p-1022
+ else if (w >= 0x00100000) {
+ t = math.expm1(-2 * x);
+ t = -t / (t + 2);
+ }
+ // |x| is subnormal
+ else {
+ math.forceEval(f32(x));
+ t = x;
+ }
+
+ if (u >> 63 != 0) {
+ -t
+ } else {
+ t
+ }
+}
+
+test "tanh" {
+ assert(tanh(f32(1.5)) == tanhf(1.5));
+ assert(tanh(f64(1.5)) == tanhd(1.5));
+}
+
+test "tanhf" {
+ const epsilon = 0.000001;
+
+ assert(math.approxEq(f32, tanhf(0.0), 0.0, epsilon));
+ assert(math.approxEq(f32, tanhf(0.2), 0.197375, epsilon));
+ assert(math.approxEq(f32, tanhf(0.8923), 0.712528, epsilon));
+ assert(math.approxEq(f32, tanhf(1.5), 0.905148, epsilon));
+ assert(math.approxEq(f32, tanhf(37.45), 1.0, epsilon));
+}
+
+test "tanhd" {
+ const epsilon = 0.000001;
+
+ assert(math.approxEq(f64, tanhd(0.0), 0.0, epsilon));
+ assert(math.approxEq(f64, tanhd(0.2), 0.197375, epsilon));
+ assert(math.approxEq(f64, tanhd(0.8923), 0.712528, epsilon));
+ assert(math.approxEq(f64, tanhd(1.5), 0.905148, epsilon));
+ assert(math.approxEq(f64, tanhd(37.45), 1.0, epsilon));
+}