Functionality to reveal data structure parts by party to socket.

Author: mkskeller

Compiler/types.py

@@ -1996,7 +1996,12 @@ def _san(self, other):
         return self._v
 
     def _div_san(self):
-        return self._v.conv((library.get_player_id() == self.player)._v).if_else(self._v, 1)
+        return self._op_san(1)
+
+    def _op_san(self, default=0):
+        return self._v.conv(
+            (library.get_player_id() == self.player)._v).if_else(
+                self._v, default)
 
     def __setitem__(self, index, value):
         self._san(value)
@@ -5920,6 +5925,18 @@ def reveal_to_clients(self, clients):
         """
         self.value_type.reveal_to_clients(clients, [self.get_vector()])
 
+    def reveal_to_socket_by_party(self, client_id, n_parties=None):
+        """ Reveal i-th part to a specific client socket on party i.
+
+        :param client_id: regint
+        :param n_parties: number of parties (default: first dimension length)
+
+        """
+        n_parties = n_parties or len(self)
+        tmp = sum(self.get_part_vector(base=i, size=1).reveal_to(i)._op_san()
+                  for i in range(n_parties))
+        tmp.write_to_socket(client_id, tmp)
+
 class Array(_vectorizable):
     """
     Array accessible by public index. That is, ``a[i]`` works for an