chore: add parameters getters for CompactPublicKey types

zama-ai · Nov 14, 2024 · f9e8df4 · f9e8df4
1 parent cf56e58
commit f9e8df4
Show file tree

Hide file tree

Showing 4 changed files with 36 additions and 0 deletions.
diff --git a/tfhe/src/high_level_api/keys/inner.rs b/tfhe/src/high_level_api/keys/inner.rs
@@ -416,6 +416,10 @@ impl IntegerCompactPublicKey {
     pub fn from_raw_parts(key: CompactPublicKey) -> Self {
         Self { key }
     }
+
+    pub fn parameters(&self) -> CompactPublicKeyEncryptionParameters {
+        self.key.parameters()
+    }
 }
 
 #[derive(Clone, Debug, Serialize, Deserialize, Versionize)]
@@ -455,6 +459,10 @@ impl IntegerCompressedCompactPublicKey {
             key: CompressedCompactPublicKey::decompress(&self.key),
         }
     }
+
+    pub fn parameters(&self) -> CompactPublicKeyEncryptionParameters {
+        self.key.parameters()
+    }
 }
 
 #[allow(clippy::struct_field_names)]

diff --git a/tfhe/src/high_level_api/keys/public.rs b/tfhe/src/high_level_api/keys/public.rs
@@ -167,6 +167,10 @@ impl CompactPublicKey {
             tag,
         }
     }
+
+    pub fn parameters(&self) -> CompactPublicKeyEncryptionParameters {
+        self.key.parameters()
+    }
 }
 
 impl Tagged for CompactPublicKey {
@@ -227,6 +231,10 @@ impl CompressedCompactPublicKey {
             tag: self.tag.clone(),
         }
     }
+
+    pub fn parameters(&self) -> CompactPublicKeyEncryptionParameters {
+        self.key.parameters()
+    }
 }
 
 impl Tagged for CompressedCompactPublicKey {

diff --git a/tfhe/src/integer/public_key/compact.rs b/tfhe/src/integer/public_key/compact.rs
@@ -34,6 +34,10 @@ impl<C: Container<Element = u64>> CompactPrivateKey<C> {
     pub fn key(&self) -> &ShortintCompactPrivateKey<C> {
         &self.key
     }
+
+    pub fn parameters(&self) -> CompactPublicKeyEncryptionParameters {
+        self.key.parameters()
+    }
 }
 
 impl CompactPrivateKey<Vec<u64>> {
@@ -158,6 +162,10 @@ impl CompactPublicKey {
     pub fn size_bytes(&self) -> usize {
         self.key.size_bytes()
     }
+
+    pub fn parameters(&self) -> CompactPublicKeyEncryptionParameters {
+        self.key.parameters()
+    }
 }
 
 #[derive(Clone, Debug, PartialEq, Serialize, Deserialize, Versionize)]
@@ -203,6 +211,10 @@ impl CompressedCompactPublicKey {
             key: self.key.decompress(),
         }
     }
+
+    pub fn parameters(&self) -> CompactPublicKeyEncryptionParameters {
+        self.key.parameters()
+    }
 }
 
 impl ParameterSetConformant for CompactPublicKey {

diff --git a/tfhe/src/shortint/public_key/compact.rs b/tfhe/src/shortint/public_key/compact.rs
@@ -466,6 +466,10 @@ impl CompactPublicKey {
     pub fn size_bytes(&self) -> usize {
         self.key.size_bytes()
     }
+
+    pub fn parameters(&self) -> CompactPublicKeyEncryptionParameters {
+        self.parameters
+    }
 }
 
 #[derive(Clone, Debug, PartialEq, Serialize, Deserialize, Versionize)]
@@ -552,6 +556,10 @@ impl CompressedCompactPublicKey {
             parameters: self.parameters,
         }
     }
+
+    pub fn parameters(&self) -> CompactPublicKeyEncryptionParameters {
+        self.parameters
+    }
 }
 
 impl ParameterSetConformant for CompactPublicKey {