g6dZddlZddlmZddZdd Zdd ZdS)z Tools to create LaTeX arrays. N)MissingOptionalLibraryErrorTFc4ddl}tj||}||d}t ||jjjr|j dkr| }t ||jjj rt||}| |d}|s|St ||jjrd|d }|d krd }|d krd }|s|d sd|}|S)aConvert a complex number to latex code suitable for a ket expression Args: raw_value (complex): Value to convert. decimals (int): Number of decimal places to round to (default 15). coefficient (bool): Whether the number is to be used as a coefficient of a ket. first_term (bool): If a coefficient, whether this number is the first coefficient in the expression. Returns: str: latex code rNdecimalsF)rational2) full_prec()1z-1-+)sympynparounditem nsimplify isinstancecorenumbersRational denominatorevalfFloatroundlatexAdd startswith) raw_valuer first_term coefficientrvalueelements P/granary/edev/quantum/lib/python3.11/site-packages/qiskit/visualization/array.py _num_to_latexr's-LLL )h777<<>>I OOIO 6 6E%+455%:Kb:P:P %+122'eX&&kk%5k11G %((! g...#~~$  g0055 g-- N rr+cN t|dkrtdd|d}|dz }fd fd}|\}}|jdkr|||g|z }nt||kr|||d|d z|z }||jdkr|d |jdzz }n8|d z}|d zt j|d zdz } |d |zz }|d z }|d | zz }|dd d z}|||| d zdzd|z }n||||z }|dz }|S)aLatex representation of a complex numpy array (with maximum dimension 2) Args: matrix (ndarray): The matrix to be converted to latex, must have dimension 2. decimals (int): For numbers not close to integers, the number of decimal places to round to. prefix (str): Latex string to be prepended to the latex, intended for labels. max_size (list(```int```)): Indexable containing two integers: Maximum width and maximum height of output Latex matrix (including dots characters). If the width and/or height of matrix exceeds the maximum, the centre values will be replaced with dots. Maximum width or height must be greater than 3. Returns: str: Latex representation of the matrix Raises: ValueError: If minimum value in max_size < 3 z=Smallest value in max_size must be greater than or equal to 3 z\begin{bmatrix} c\d}|D]}t|}||dzz }|dd}|S)Nrrz & )r')elements el_stringel num_stringrs r&_elements_to_latexz,_matrix_to_latex.._elements_to_latex]sO  , ,B&rH===J e+ +IIcrcN r(cd}|D]d}t||kr||z }n;||d|dzz }|dz }||| dzdzdz }|dz }e|S)Nrz & \cdots & z \\ )len)rows max_width row_stringrr5s r&_rows_to_latexz(_matrix_to_latex.._rows_to_latexgs  % %A1vv""00333 003CY!^3C1DEEE n, 00I:?Q3F3H3H1IJJJ * $JJr(r8Nr7z \vdots & z \ddots & r0z\\ z\end{bmatrix} )min ValueErrorndimr9shapermod) matrixrprefixmax_size out_stringr>r; max_height pre_vdots post_vdotsr5s ` @r&_matrix_to_latexrKCs( 8}}q\]]] fJ&&J     %Iz {annfXy999 Vz ! ! nnV,=jAo,=%> JJJ  Q ' ' ,a8 8JJ"QI"a"&A*>*>>BJ ,2 2J , &J ,3 3J_y0 nnVZK1,rbs{999999++++\LLLL^:":":":":":"r(