Add GetCryptoContext (#18)

src/OpenFHE.jl

@@ -31,6 +31,9 @@ export SetMultiplicativeDepth, SetScalingModSize, SetBatchSize, SetSecretKeyDist
 # FHECKKSRNS
 export GetBootstrapDepth
 
+# CryptoObject
+export GetCryptoContext
+
 # CryptoContext
 export CryptoContext, GenCryptoContext, Enable, GetRingDimension, KeyGen, EvalMultKeyGen,
        EvalRotateKeyGen, MakeCKKSPackedPlaintext, Encrypt, EvalAdd, EvalSub, EvalMult,

src/convenience.jl

@@ -204,6 +204,25 @@ function EvalBootstrap(context::CxxWrap.CxxWrapCore.CxxRef{OpenFHE.CryptoContext
     EvalBootstrap(context, ciphertext, num_iterations, precision)
 end
 
+"""
+    GetCryptoContext(object::Union{Ciphertext})
+
+Return a the crypto context for a an `object` that is is a subtype of `CryptoObject`.
+
+Currently, this is only implemented for [`Ciphertext`](@ref).
+
+See also: [`CryptoContext`](@ref), [`Ciphertext`](@ref)
+"""
+function GetCryptoContext(object::Union{Ciphertext})
+    # Note: Due to limitations of CxxWrap.jl when wrapping mutually dependent template
+    # types, openfhe-julia introduces a non-template proxy type as an additional level of
+    # indirection.
+    # We thus need to first dereference `Ciphertext` to get to the `CiphertextImpl`, then
+    # call the proxy method to get the `CryptoContextProxy`, and finally obtain the actual
+    # `CryptoContext` from it
+    GetCryptoContext(GetCryptoContextProxy(object[]))
+end
+
 """
     Decrypt(crypto_context::CryptoContext, ciphertext::Ciphertext, private_key::PrivateKey, plaintext::Plaintext)
     Decrypt(crypto_context::CryptoContext, private_key::PrivateKey, ciphertext::Ciphertext, plaintext::Plaintext)

test/test_ckks.jl

@@ -135,6 +135,10 @@ end
     @test GetRealPackedValue(result_dec) ≈ circshift(x1, -shift)
 end
 
+@testset verbose=true showtiming=true "GetCryptoContext" begin
+    @test GetCryptoContext(c1) isa CryptoContext
+end
+
 end # @testset "test_ckks.jl"
 
 end # module