Add new functions: zpowu and zmodpowu

Signed-off-by: Mattias Andrée <maandree@kth.se>

2 files changed, 77 insertions, 0 deletions

diff --git a/src/zmodpowu.c b/src/zmodpowu.c
new file mode 100644
index 0000000..56ef7da
--- /dev/null
+++ b/src/zmodpowu.c
@@ -0,0 +1,45 @@
+/* See LICENSE file for copyright and license details. */
+#include "internals"
+
+#include <errno.h>
+
+#define tb  libzahl_tmp_pow_b
+#define td  libzahl_tmp_pow_d
+
+
+void
+zmodpowu(z_t a, z_t b, unsigned long long int c, z_t d)
+{
+	size_t i, n;
+
+	if (!c) {
+		if (zzero(b)) {
+			errno = EDOM; /* Indeterminate form: 0:th power of 0 */
+			FAILURE_JUMP();
+		} else if (zzero(d)) {
+			errno = EDOM; /* Undefined form: Division by 0 */
+			FAILURE_JUMP();
+		} else {
+			zsetu(a, 1);
+		}
+		return;
+	} else if (zzero(d)) {
+		errno = EDOM; /* Undefined form: Division by 0 */
+		FAILURE_JUMP();
+	} else if (zzero(b)) {
+		SET_SIGNUM(a, 0);
+		return;
+	}
+
+	n = zbits(c);
+
+	zmod(tb, b, d);
+	zset(td, d);
+	zsetu(a, 1);
+
+	for (; c; c >>= 1) {
+		if (c & 1)
+			zmodmul(a, a, tb, td);
+		zmodsqr(tb, tb, td);
+	}
+}
diff --git a/src/zpowu.c b/src/zpowu.c
new file mode 100644
index 0000000..bc0c6dd
--- /dev/null
+++ b/src/zpowu.c
@@ -0,0 +1,32 @@
+/* See LICENSE file for copyright and license details. */
+#include "internals"
+
+#include <errno.h>
+
+#define tb  libzahl_tmp_pow_b
+
+
+void
+zpowu(z_t a, z_t b, unsigned long long int c)
+{
+	if (!c) {
+		if (zzero(b)) {
+			errno = EDOM; /* Indeterminate form: 0:th power of 0 */
+			FAILURE_JUMP();
+		}
+		zsetu(a, 1);
+		return;
+	} else if (zzero(b)) {
+		SET_SIGNUM(a, 0);
+		return;
+	}
+
+	zset(tb, b);
+	zsetu(a, 1);
+
+	for (; c; c >>= 1) {
+		if (c & 1)
+			zmul(a, a, tb);
+		zsqr(tb, tb);
+	}
+}