Add tuples, ndarrays, and complex numbers to cirq_google proto (#7226)

* Add tuples, ndarrays, and complex numbers to cirq_google proto

- Adds support for (potentially nested) tuples of mismatched types
- Adds support for serializing ndarrays
- Adds support for serializing complex numbers

* my o my, it's mypy

* Fix all the review comments, add set support.

* Fix invert mask, which assumes list, so it can accept tuples.

* Update cirq-google/cirq_google/serialization/arg_func_langs.py

Co-authored-by: Pavol Juhas <[email protected]>

* Fix coverage.

---------

Co-authored-by: Pavol Juhas <[email protected]>

Author: dstrain115

cirq-google/cirq_google/api/v2/program.proto

@@ -3,6 +3,7 @@ syntax = "proto3";
 package cirq.google.api.v2;
 
 import "tunits/proto/tunits.proto";
+import "cirq_google/api/v2/ndarrays.proto";
 
 option java_package = "com.google.cirq.google.api.v2";
 option java_outer_classname = "ProgramProto";
@@ -397,6 +398,9 @@ message ArgValue {
     tunits.Value value_with_unit = 8;
     bool bool_value = 9;
     bytes bytes_value = 10;
+    Complex complex_value = 11;
+    Tuple tuple_value = 12;
+    NDArray ndarray_value = 13;
   }
 }
 
@@ -417,6 +421,45 @@ message RepeatedBoolean {
   repeated bool values = 1;
 }
 
+// Representation of a mixed tuple of values
+message Tuple {
+
+  // Original (python) type of the data
+  enum SequenceType {
+    UNSPECIFIED = 0;
+    LIST = 1;
+    TUPLE = 2;
+    SET = 3;
+    FROZENSET = 4;
+  }
+
+  SequenceType sequence_type = 1;
+    
+  repeated Arg values = 2;
+}
+
+// Representation of a complex number
+message Complex {
+    double real_value = 1;
+    double imag_value = 2;
+}
+
+message NDArray {
+  oneof arr {
+    Complex128Array complex128_array = 1;
+    Complex64Array complex64_array = 2;
+    Float16Array float16_array = 3;
+    Float32Array float32_array = 4;
+    Float64Array float64_array = 5;
+    Int64Array int64_array = 6;
+    Int32Array int32_array = 7;
+    Int16Array int16_array = 8;
+    Int8Array int8_array = 9;
+    UInt8Array uint8_array = 10;
+    BitArray bit_array = 11;
+  }
+}
+
 // A function of arguments. This is an s-expression tree representing
 // mathematically the function being evaluated.
 //