Adds needed 2 qubit control gate wrapper

Author: Stephen DiAdamo

qunetsim/backends/cqc_backend.py

@@ -339,7 +339,7 @@ def custom_gate(self, qubit, gate):
             qubit(Qubit): Qubit to which the gate is applied.
             gate(np.ndarray): 2x2 array of the gate.
         """
-        raise(EnvironmentError("Not implemented for this backend!"))
+        raise (EnvironmentError("Not implemented for this backend!"))
 
     def custom_controlled_gate(self, qubit, target, gate):
         """
@@ -350,7 +350,7 @@ def custom_controlled_gate(self, qubit, target, gate):
             target(Qubit): Qubit on which the gate is applied.
             gate(nd.array): 2x2 array for the gate applied to target.
         """
-        raise(EnvironmentError("Not implemented for this backend!"))
+        raise (EnvironmentError("Not implemented for this backend!"))
 
     def custom_two_qubit_gate(self, qubit1, qubit2, gate):
         """
@@ -361,7 +361,19 @@ def custom_two_qubit_gate(self, qubit1, qubit2, gate):
             qubit2(Qubit): Second qubit of the gate.
             gate(np.ndarray): 4x4 array for the gate applied.
         """
-        raise(EnvironmentError("Not implemented for this backend!"))
+        raise (EnvironmentError("Not implemented for this backend!"))
+
+    def custom_controlled_two_qubit_gate(self, qubit, target_1, target_2, gate):
+        """
+        Applies a custom gate to the target qubit, controlled by the qubit.
+
+        Args:
+            qubit (Qubit): Qubit to control the gate.
+            target_1 (Qubit): Qubit on which the gate is applied.
+            target_2 (Qubit): Qubit on which the gate is applied.
+            gate (nd.array): 4x4 array for the gate applied to target.
+        """
+        raise (EnvironmentError("Not implemented for this backend!"))
 
     def measure(self, qubit, non_destructive):
         """

qunetsim/backends/eqsn_backend.py

@@ -373,6 +373,18 @@ def custom_controlled_gate(self, qubit, target, gate):
         """
         self.eqsn.custom_controlled_gate(target.qubit, qubit.qubit, gate)
 
+    def custom_controlled_two_qubit_gate(self, qubit, target_1, target_2, gate):
+        """
+        Applies a custom gate to the target qubit, controlled by the qubit.
+
+        Args:
+            qubit (Qubit): Qubit to control the gate.
+            target_1 (Qubit): Qubit on which the gate is applied.
+            target_2 (Qubit): Qubit on which the gate is applied.
+            gate (nd.array): 4x4 array for the gate applied to target.
+        """
+        self.eqsn.custom_two_qubit_control_gate(qubit.qubit, target_1.qubit, target_2.qubit, gate)
+
     def custom_two_qubit_gate(self, qubit1, qubit2, gate):
         """
         Applies a custom two qubit gate to qubit1 \\otimes qubit2.

qunetsim/backends/projectq_backend.py

@@ -288,7 +288,7 @@ def custom_gate(self, qubit, gate):
             qubit(Qubit): Qubit to which the gate is applied.
             gate(np.ndarray): 2x2 array of the gate.
         """
-        raise(EnvironmentError("Not implemented for this backend!"))
+        raise (EnvironmentError("Not implemented for this backend!"))
 
     def custom_controlled_gate(self, qubit, target, gate):
         """
@@ -299,7 +299,7 @@ def custom_controlled_gate(self, qubit, target, gate):
             target(Qubit): Qubit on which the gate is applied.
             gate(nd.array): 2x2 array for the gate applied to target.
         """
-        raise(EnvironmentError("Not implemented for this backend!"))
+        raise (EnvironmentError("Not implemented for this backend!"))
 
     def custom_two_qubit_gate(self, qubit1, qubit2, gate):
         """
@@ -310,7 +310,19 @@ def custom_two_qubit_gate(self, qubit1, qubit2, gate):
             qubit2(Qubit): Second qubit of the gate.
             gate(np.ndarray): 4x4 array for the gate applied.
         """
-        raise(EnvironmentError("Not implemented for this backend!"))
+        raise (EnvironmentError("Not implemented for this backend!"))
+
+    def custom_controlled_two_qubit_gate(self, qubit, target_1, target_2, gate):
+        """
+        Applies a custom gate to the target qubit, controlled by the qubit.
+
+        Args:
+            qubit (Qubit): Qubit to control the gate.
+            target_1 (Qubit): Qubit on which the gate is applied.
+            target_2 (Qubit): Qubit on which the gate is applied.
+            gate (nd.array): 4x4 array for the gate applied to target.
+        """
+        raise (EnvironmentError("Not implemented for this backend!"))
 
     def measure(self, qubit, non_destructive):
         """

qunetsim/objects/qubit.py

@@ -241,21 +241,38 @@ def custom_controlled_gate(self, target, gate):
 
         self._host.backend.custom_controlled_gate(self, target, gate)
 
+    def custom_two_qubit_control_gate(self, q1, q2, gate):
+        """
+        Applies the gate *gate* to qubits q1 and q2 as a controlled gate.
+
+        Args:
+            q1 (Qubit): First qubit
+            q2 (Qubit): Second qubit
+            gate (Numpy ndarray): The gate to apply
+        """
+        if not isinstance(gate, np.ndarray):
+            raise (InputError("Only Numpy matrices are allowed"))
+        if not is_unitary(gate):
+            raise (InputError("Custom gates must be unitary operations"))
+        if gate.shape != (4, 4):
+            raise (InputError("Custom controlled gates must be 4x4 matrices"))
+
+        self._host.backend.custom_controlled_two_qubit_gate(self, q1, q2, gate)
+
     def custom_two_qubit_gate(self, other_qubit, gate):
         """
         Applies a custom 2 qubit gate.
 
         Args:
             other_qubit (Qubit): The second qubit.
             gate (Numpy ndarray): The gate
-
         """
         if not isinstance(gate, np.ndarray):
             raise (InputError("Only Numpy matrices are allowed"))
         if not is_unitary(gate):
             raise (InputError("Custom gates must be unitary operations"))
         if gate.shape != (4, 4):
-            raise (InputError("Custom controlled gates must be 2x2 matrices"))
+            raise (InputError("Custom controlled gates must be 4x4 matrices"))
 
         self._host.backend.custom_two_qubit_gate(self, other_qubit, gate)