from abp.fancy import GraphState from abp.util import xyz import numpy as np import time import itertools square_unit_cell = ( ((0, 0), (0, 1)), ((0, 0), (1, 0)), ((1, 0), (1, 1)), ((0, 1), (1, 1))) funny_unit_cell = (((0, 0), (0, 1)), ((0, 0), (1, 0)), ((1, 0), (1, 1)), ((0, 1), (1, 1)), ((0, 0), (.5, .5))) def add_offset(vector, offset): """ Offset a vector in n-dimensional space """ return tuple(v + o 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(*map(range, size)): edges |= offset_unit_cell(unit_cell, offset) nodes = set(itertools.chain(*edges)) return nodes, edges # s = VisibleGraphState() nodes, edges = lattice(square_unit_cell, (4, 4)) psi = GraphState() for node in nodes: psi.add_node(str(node), position=xyz(node[0], node[1])) psi.act_hadamard(str(node)) for edge in edges: psi.act_cz(str(edge[0]), str(edge[1]))