Add cleartext typing (#37)

Add cleartext type conversion

Author: mathias-nillion

nada_algebra/array.py

@@ -8,12 +8,13 @@
 from typing import Any, Callable, Optional, Sequence, Union, get_args, overload
 
 import numpy as np
-from nada_dsl import (Input, Integer, Output, Party, PublicInteger,
+from nada_dsl import (Boolean, Input, Integer, Output, Party, PublicInteger,
                       PublicUnsignedInteger, SecretInteger,
                       SecretUnsignedInteger, UnsignedInteger)
 
 from nada_algebra.context import UnsafeArithmeticSession
-from nada_algebra.nada_typing import (NadaBoolean, NadaInteger, NadaRational,
+from nada_algebra.nada_typing import (NadaBoolean, NadaCleartextType,
+                                      NadaInteger, NadaRational,
                                       NadaUnsignedInteger)
 from nada_algebra.types import (Rational, SecretRational, get_log_scale,
                                 public_rational, rational, secret_rational)
@@ -738,6 +739,24 @@ def is_boolean(self) -> bool:
         """
         return self.dtype == NadaBoolean
 
+    @property
+    def cleartext_nada_type(self) -> NadaCleartextType:
+        """
+        Returns a clear-text Nada type compatible with the Nada array.
+
+        Returns:
+            NadaCleartextType: Compatible cleartext type.
+        """
+        if self.is_rational:
+            return Rational
+        if self.is_integer:
+            return Integer
+        if self.is_unsigned_integer:
+            return UnsignedInteger
+        if self.is_boolean:
+            return Boolean
+        raise TypeError(f"Array {self} is of unknown type {self.dtype}.")
+
     def __str__(self) -> str:
         """
         String representation of the NadaArray.

nada_algebra/nada_typing.py

@@ -31,3 +31,16 @@
     PublicBoolean,
     SecretBoolean,
 ]
+
+NadaCleartextNumber = Union[
+    dsl.Integer,
+    dsl.UnsignedInteger,
+    Rational,
+]
+
+NadaCleartextType = Union[
+    dsl.Integer,
+    dsl.UnsignedInteger,
+    dsl.Boolean,
+    Rational,
+]