Anders and Briegel in Python

#### 61 行 1.8KB Raw パーマリンク Blame 履歴

 ``````from __future__ import absolute_import from __future__ import print_function from abp import GraphState from abp.util import xyz import numpy as np import time import itertools import networkx as nx from six.moves import map from six.moves import zip raussendorf_unit_cell = ( ((1, 0, 0), (1, 1, 0)), ((0, 1, 0), (1, 1, 0)), ((1, 2, 0), (1, 1, 0)), ((2, 1, 0), (1, 1, 0)), ((1, 2, 2), (1, 1, 2)), ((0, 1, 2), (1, 1, 2)), ((1, 0, 2), (1, 1, 2)), ((2, 1, 2), (1, 1, 2)), ((0, 1, 0), (0, 1, 1)), ((0, 0, 1), (0, 1, 1)), ((0, 1, 2), (0, 1, 1)), ((0, 2, 1), (0, 1, 1)), ((2, 1, 0), (2, 1, 1)), ((2, 0, 1), (2, 1, 1)), ((2, 1, 2), (2, 1, 1)), ((2, 2, 1), (2, 1, 1)), ((1, 0, 0), (1, 0, 1)), ((0, 0, 1), (1, 0, 1)), ((1, 0, 2), (1, 0, 1)), ((2, 0, 1), (1, 0, 1)), ((1, 2, 0), (1, 2, 1)), ((0, 2, 1), (1, 2, 1)), ((1, 2, 2), (1, 2, 1)), ((2, 2, 1), (1, 2, 1))) def add_offset(vector, offset): """ Offset a vector in n-dimensional space """ return tuple(v + o*2 for v, o in zip(vector, offset)) def offset_unit_cell(unit_cell, offset): """ Offset a unit cell """ return {(add_offset(a, offset), add_offset(b, offset)) for a, b in unit_cell} def lattice(unit_cell, size): """ Generate a lattice from a unit cell """ edges = set() for offset in itertools.product(*list(map(range, size))): edges |= offset_unit_cell(unit_cell, offset) nodes = set(itertools.chain(*edges)) return nodes, edges nodes, edges = lattice(raussendorf_unit_cell, (2, 2, 3)) psi = GraphState() for node in nodes: x, y, z = node color = "red" if (x+y+z) % 2 > 0 else "black" psi.add_qubit(node, position=xyz(*node), color=color) psi.act_hadamard(node) for edge in edges: psi.act_cz(edge[0], edge[1]) psi.push()``````