From a24b42bfd37f740d4c0ecaa8d54b2eef5c3245a3 Mon Sep 17 00:00:00 2001
From: Andy Polyakov <appro@cryptogams.org>
Date: Sun, 7 Jan 2024 22:03:59 +0100
Subject: [PATCH] rust/src/lib.rs: add NTT enums declarations.

---
 rust/src/lib.rs | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/rust/src/lib.rs b/rust/src/lib.rs
index a83f5ad..f047ff0 100644
--- a/rust/src/lib.rs
+++ b/rust/src/lib.rs
@@ -93,3 +93,23 @@ impl<T> Clone for Gpu_Ptr<T> {
         unsafe { transmute::<_, _>(clone_gpu_ptr_t(transmute::<&_, &_>(self))) }
     }
 }
+
+#[repr(C)]
+pub enum NTTInputOutputOrder {
+    NN = 0,
+    NR = 1,
+    RN = 2,
+    RR = 3,
+}
+
+#[repr(C)]
+pub enum NTTDirection {
+    Forward = 0,
+    Inverse = 1,
+}
+
+#[repr(C)]
+pub enum NTTType {
+    Standard = 0,
+    Coset = 1,
+}