Make PIR & PNNS benchmarks more general (#225)

Author: fboemer

…ivateInformationRetrievalBenchmark.swift …ionRetrievalBenchmark/PirBenchmark.swiftBenchmarks/PrivateInformationRetrievalBenchmark/PrivateInformationRetrievalBenchmark.swift renamed to Benchmarks/PrivateInformationRetrievalBenchmark/PirBenchmark.swift Benchmarks/PrivateInformationRetrievalBenchmark/PrivateInformationRetrievalBenchmark.swift renamed to Benchmarks/PrivateInformationRetrievalBenchmark/PirBenchmark.swift

@@ -18,24 +18,24 @@ import HomomorphicEncryption
 /// Configuration for a ``KeywordDatabase``.
 public struct KeywordPirConfig: Hashable, Codable, Sendable {
     /// Number of dimensions in the database.
-    @usableFromInline let dimensionCount: Int
+    public let dimensionCount: Int
 
     /// Configuration for the cuckoo table.
-    @usableFromInline let cuckooTableConfig: CuckooTableConfig
+    public let cuckooTableConfig: CuckooTableConfig
 
     /// Whether to enable the `uneven dimensions` optimization.
-    @usableFromInline let unevenDimensions: Bool
+    public let unevenDimensions: Bool
 
-    /// Strategy for ``EvaluationKey`` compression.
-    @usableFromInline let keyCompression: PirKeyCompressionStrategy
+    /// Strategy for evaluation key compression.
+    public let keyCompression: PirKeyCompressionStrategy
 
     /// When set to true, entry size will be equal to ``CuckooTableConfig/maxSerializedBucketSize``.
     ///
     /// Otherwise the largest serialized bucket size is used instead.
-    @usableFromInline let useMaxSerializedBucketSize: Bool
+    public let useMaxSerializedBucketSize: Bool
 
     /// Sharding function configuration.
-    @usableFromInline let shardingFunction: ShardingFunction
+    public let shardingFunction: ShardingFunction
 
     /// Keyword PIR parameters.
     public var parameter: KeywordPirParameter {

…vateNearestNeighborSearchBenchmark.swift …ghborSearchBenchmark/PnnsBenchmark.swiftBenchmarks/PrivateNearestNeighborSearchBenchmark/PrivateNearestNeighborSearchBenchmark.swift renamed to Benchmarks/PrivateNearestNeighborSearchBenchmark/PnnsBenchmark.swift Benchmarks/PrivateNearestNeighborSearchBenchmark/PrivateNearestNeighborSearchBenchmark.swift renamed to Benchmarks/PrivateNearestNeighborSearchBenchmark/PnnsBenchmark.swift

@@ -13,6 +13,7 @@
 // limitations under the License.
 
 import Benchmark
+import HomomorphicEncryption
 
 let noiseBudgetScale = 10
 
@@ -25,3 +26,64 @@ extension BenchmarkMetric {
     static var responseCiphertextCount: Self { .custom("Response ciphertext count") }
     static var noiseBudget: Self { .custom("Noise budget x \(noiseBudgetScale)") }
 }
+
+/// Encryption parameters configuration for benchmarks.
+public struct EncryptionParametersConfig {
+    /// Default configuration for PNNS benchmarks
+    public nonisolated(unsafe) static let defaultPnns = EncryptionParametersConfig(
+        polyDegree: 4096,
+        // use plaintextModulusBits: [16, 17] for plaintext CRT
+        plaintextModulusBits: [17],
+        coefficientModulusBits: [27, 28, 28])
+
+    /// Default configuration for PIR benchmarks
+    public nonisolated(unsafe) static let defaultPir = EncryptionParametersConfig(
+        polyDegree: 4096,
+        plaintextModulusBits: [5],
+        coefficientModulusBits: [27, 28, 28])
+
+    /// Polynomial degree.
+    public let polyDegree: Int
+    /// Number of significant bits in each plaintext modulus.
+    public let plaintextModulusBits: [Int]
+    /// Number of significant bits in each ciphertext modulus.
+    public let coefficientModulusBits: [Int]
+
+    /// Creates a new ``EncryptionParametersConfig``
+    /// - Parameters:
+    ///   - polyDegree: Polynomial degree
+    ///   - plaintextModulusBits: Number of significant bits in each plaintext modulus.
+    ///   - coefficientModulusBits: Number of significant bits in each ciphertext modulus.
+    public init(polyDegree: Int, plaintextModulusBits: [Int], coefficientModulusBits: [Int]) {
+        self.polyDegree = polyDegree
+        self.plaintextModulusBits = plaintextModulusBits
+        self.coefficientModulusBits = coefficientModulusBits
+    }
+}
+
+extension EncryptionParametersConfig: CustomStringConvertible {
+    public var description: String {
+        "N=\(polyDegree)/logt=\(plaintextModulusBits)/logq=\(coefficientModulusBits.description)"
+    }
+}
+
+extension EncryptionParameters {
+    ///  Creates a new ``EncryptionParameters`` from the configuration.
+    /// - Parameter config: Configuration.
+    /// - Throws: Upon failure to initialize encryption parameters.
+    public init(from config: EncryptionParametersConfig) throws {
+        let plaintextModulus = try Scalar.generatePrimes(
+            significantBitCounts: config.plaintextModulusBits,
+            preferringSmall: true)[0]
+        let coefficientModuli = try Scalar.generatePrimes(
+            significantBitCounts: config.coefficientModulusBits,
+            preferringSmall: false,
+            nttDegree: config.polyDegree)
+        try self.init(
+            polyDegree: config.polyDegree,
+            plaintextModulus: plaintextModulus,
+            coefficientModuli: coefficientModuli,
+            errorStdDev: ErrorStdDev.stdDev32,
+            securityLevel: SecurityLevel.quantum128)
+    }
+}