Skip to content

Visualization

oqd_trical.light_matter.compiler.visualization

CoefficientPrinter

Bases: ConversionRule

Prints Coefficients in a pretty manner

Source code in src\oqd_trical\light_matter\compiler\visualization.py 
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
class CoefficientPrinter(ConversionRule):
    """Prints Coefficients in a pretty manner"""

    def map_MathExpr(self, model, operands):
        return Post(PrintMathExpr())(model)

    def map_MathAdd(self, model, operands):
        return f"({Post(PrintMathExpr())(model)})"

    def map_WaveCoefficient(self, model, operands):
        frequency_term = (
            f"{operands['frequency']} * t"
            if model.frequency != MathNum(value=0)
            else ""
        )
        phase_term = f"{operands['phase']}" if model.phase != MathNum(value=0) else ""

        wave_term = (
            f" * exp(1j * ({frequency_term}{' + ' if frequency_term and phase_term else ''}{phase_term}))"
            if phase_term or frequency_term
            else ""
        )

        return f"{operands['amplitude']}{wave_term}"

    def map_CoefficientAdd(self, model, operands):
        return f"{'(' + operands['coeff1'] + ')' if isinstance(model.coeff1, CoefficientAdd) else operands['coeff1']} + {'(' + operands['coeff2'] + ')' if isinstance(model.coeff2, CoefficientAdd) else operands['coeff2']}"

    def map_CoefficientMul(self, model, operands):
        return f"{'(' + operands['coeff1'] + ')' if isinstance(model.coeff1, CoefficientAdd) else operands['coeff1']} * {'(' + operands['coeff2'] + ')' if isinstance(model.coeff2, CoefficientAdd) else operands['coeff2']}"

OperatorPrinter

Bases: ConversionRule

Prints Operators in a pretty manner

Source code in src\oqd_trical\light_matter\compiler\visualization.py 
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
class OperatorPrinter(ConversionRule):
    """Prints Operators in a pretty manner"""

    def map_KetBra(self, model, operands):
        return f"|{model.ket}><{model.bra}|_{model.subsystem}"

    def map_Annihilation(self, model, operands):
        return f"A_{model.subsystem}"

    def map_Creation(self, model, operands):
        return f"C_{model.subsystem}"

    def map_Identity(self, model, operands):
        return f"I_{model.subsystem}"

    def map_PrunedOperator(self, model, operands):
        return "PrunedOperator"

    def map_Displacement(self, model, operands):
        return f"D({operands['alpha']})_{model.subsystem}"

    def map_OperatorAdd(self, model, operands):
        return f"{operands['op1']} + {operands['op2']}"

    def map_OperatorMul(self, model, operands):
        return f"{'(' + operands['op1'] + ')' if isinstance(model.op1, OperatorAdd) else operands['op1']} * {'(' + operands['op2'] + ')' if isinstance(model.op2, OperatorAdd) else operands['op2']}"

    def map_OperatorKron(self, model, operands):
        return f"{'(' + operands['op1'] + ')' if isinstance(model.op1, OperatorAdd) else operands['op1']} @ {'(' + operands['op2'] + ')' if isinstance(model.op2, OperatorAdd) else operands['op2']}"

    def map_OperatorScalarMul(self, model, operands):
        return f"{'(' + operands['coeff'] + ')' if isinstance(model.coeff, CoefficientAdd) else operands['coeff']} * {'(' + operands['op'] + ')' if isinstance(model.op, OperatorAdd) else operands['op']}"

    def map_Coefficient(self, model, operands):
        return Post(CoefficientPrinter())(model)

CondensedOperatorPrettyPrint

Bases: PrettyPrint

Prints An AtomicEmulatorCircuit in a pretty manner

Source code in src\oqd_trical\light_matter\compiler\visualization.py 
93
94
95
96
97
class CondensedOperatorPrettyPrint(PrettyPrint):
    """Prints An AtomicEmulatorCircuit in a pretty manner"""

    def map_Operator(self, model, operands):
        return f"Operator({Post(OperatorPrinter())(model)})"