From 556346d623c5a433172c9e52a846f28ac55328d4 Mon Sep 17 00:00:00 2001
From: Jean-Philippe Bossuat <jeanphilippe.bossuat@gmail.com>
Date: Thu, 16 Jan 2025 16:12:11 +0100
Subject: [PATCH] inserting element in AutoPermMap returns the galois element

---
 math/src/automorphism.rs | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/math/src/automorphism.rs b/math/src/automorphism.rs
index 417d46a..4c9a2be 100644
--- a/math/src/automorphism.rs
+++ b/math/src/automorphism.rs
@@ -12,11 +12,13 @@ impl AutoPermMap {
         }
     }
 
-    pub fn insert(&mut self, perm: AutoPerm) {
+    pub fn insert(&mut self, perm: AutoPerm) -> usize {
+        let gal_el: usize = perm.gal_el;
         self.0.insert(perm.gal_el, perm);
+        gal_el
     }
 
-    pub fn gen<O: Unsigned, const NTT: bool>(&mut self, ring: &Ring<O>, gen_1: usize, gen_2: bool) {
+    pub fn gen<O: Unsigned, const NTT: bool>(&mut self, ring: &Ring<O>, gen_1: usize, gen_2: bool) -> usize {
         self.insert(AutoPerm::new::<O, NTT>(ring, gen_1, gen_2))
     }