from matplotlib import pyplot as plt
from graph import *
from viz import draw
import itertools as it
import clifford


def local_complementation(g, vops, v):
    """ As defined in LISTING 1 of Anders & Briegel """
    for i, j in it.combinations(g[v], 2):
        toggle_edge(g, i, j)

    # Update VOPs
    vops[v] = clifford.times_table[vops[v]][clifford.sqx]
    for i in g[v]:
        vops[i] = clifford.times_table[vops[i]][clifford.msqz]

if __name__ == '__main__':
    g, vops = graph(10)
    add_edge(g, 0, 1)
    add_edge(g, 1, 3)
    add_edge(g, 3, 2)
    add_edge(g, 3, 0)
    add_edge(g, 2, 0)
    add_edge(g, 0, 5)
    vops[0] = 1
    draw(g, vops)
    plt.clf()
    local_complementation(g, vops, 0)
    draw(g, vops, "out2.pdf")