import pygraphviz as pgv
from Exp import E
def _drawNode(E,G):
    if E == None:
        return None;
    #G.add_node(str(E.index)+' '+E.f);
    my = '['+str(E.index)+'] '+ str(E.f);
    G.add_node(E.index,label=str(E.f));
    ln = _drawNode(E.left,G);
    rn = _drawNode(E.right,G);
    if ln != None:
        G.add_edge(E.index,ln);
    if rn != None:
        G.add_edge(E.index,rn);

    return E.index;

def drawE(E):
    G = pgv.AGraph(directed=True,strict=True);

    _drawNode(E,G);
    G.graph_attr['epsilon']='0.01'
    G.write('fooOld.dot')
    G.layout('dot') # layout with dot
    G.draw('a.png') # write to file