"""
Copyright 2024 Entropica Labs Pte Ltd
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
"""
import numpy as np
[docs]
def paulichar_to_xz(p: str) -> tuple[int, int]:
"""
Function that turns a Pauli into a pair of x,z bits.
Parameters
----------
p : str
The Pauli character.
Returns
-------
tuple[int, int]
x, z bits
Raises
------
ValueError
If the Pauli is not I, Z, X or Y.
"""
match p:
case "_" | "I" | "i":
return (0, 0)
case "Z" | "z":
return (0, 1)
case "X" | "x":
return (1, 0)
case "Y" | "y":
return (1, 1)
case _:
raise ValueError(
"The Pauli character should be I, Z, X or Y or their "
"lower case versions. _ is also accepted as I."
)
[docs]
def paulichar_to_xz_npfunc(p: np.ndarray) -> tuple[np.ndarray, np.ndarray]:
"""
Vectorized version of paulichar_to_xz.
Parameters
----------
p : np.ndarray
An array of Pauli characters.
Returns
-------
tuple[np.ndarray]
Two arrays of x and z bits.
"""
return np.frompyfunc(paulichar_to_xz, 1, 2)(p)
[docs]
def paulixz_to_char(
x: int,
z: int,
) -> str:
"""
Function that turns a pair of x, z bits into their Pauli as a str.
Parameters
----------
x : int
The x bit.
z : int
The z bit.
Returns
-------
str
The Pauli character as a string.
Raises
------
ValueError
If x or z are not 0 or 1.
"""
match (x, z):
case (0, 0):
return "_"
case (0, 1):
return "Z"
case (1, 0):
return "X"
case (1, 1):
return "Y"
case _:
raise ValueError("The x and z values should be 0 or 1.")
[docs]
def paulixz_to_char_npfunc(x: np.ndarray, z: np.ndarray) -> np.ndarray:
"""
Vectorized version of paulixz_to_char.
Parameters
----------
x : np.ndarray
An array of x bits.
z : np.ndarray
An array of z bits.
Returns
-------
np.ndarray
An array of Pauli characters.
"""
return np.frompyfunc(paulixz_to_char, 2, 1)(x, z)