/
dag_to_dagdependency.py
55 lines (42 loc) · 1.79 KB
/
dag_to_dagdependency.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
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
54
55
# This code is part of Qiskit.
#
# (C) Copyright IBM 2020.
#
# This code is licensed under the Apache License, Version 2.0. You may
# obtain a copy of this license in the LICENSE.txt file in the root directory
# of this source tree or at http://www.apache.org/licenses/LICENSE-2.0.
#
# Any modifications or derivative works of this code must retain this
# copyright notice, and modified files need to carry a notice indicating
# that they have been altered from the originals.
"""Helper function for converting a dag circuit to a dag dependency"""
from qiskit.dagcircuit.dagdependency import DAGDependency
def dag_to_dagdependency(dag, create_preds_and_succs=True):
"""Build a ``DAGDependency`` object from a ``DAGCircuit``.
Args:
dag (DAGCircuit): the input dag.
create_preds_and_succs (bool): whether to construct lists of
predecessors and successors for every node.
Return:
DAGDependency: the DAG representing the input circuit as a dag dependency.
"""
dagdependency = DAGDependency()
dagdependency.name = dag.name
dagdependency.metadata = dag.metadata
dagdependency.add_qubits(dag.qubits)
dagdependency.add_clbits(dag.clbits)
for register in dag.qregs.values():
dagdependency.add_qreg(register)
for register in dag.cregs.values():
dagdependency.add_creg(register)
for node in dag.topological_op_nodes():
# Get arguments for classical control (if any)
inst = node.op.copy()
dagdependency.add_op_node(inst, node.qargs, node.cargs)
if create_preds_and_succs:
dagdependency._add_predecessors()
dagdependency._add_successors()
# copy metadata
dagdependency.global_phase = dag.global_phase
dagdependency.calibrations = dag.calibrations
return dagdependency