learning_graph_assign.py 文件源码

def dot_to_file(dot_G, name='out', env='./env'):
    G           = dot_G.copy()
    env         = os.path.abspath(env)
    env_dot     = os.path.join(env,'dot')
    env_img     = os.path.join(env,'img')
    layout_prog = 'dot'
    dot_suffix  = 'dot'
    img_suffix  = 'png'

    dot_G.graph_attr.update(name = name)
    dot_G.graph_attr.update(label = name)

    if (debug):
        # print to screen
        print(G.string()) 

    if not os.path.exists(env_dot):
        os.makedirs(env_dot)
    if not os.path.exists(env_img):
        os.makedirs(env_img)

    G.layout(prog=layout_prog)
    dot_fname = os.path.join(env_dot, name +'.' + str(dot_suffix))
    img_fname = os.path.join(env_img, name + '.' + str(img_suffix))
    G.write(dot_fname)
    if (debug):
        print("Wrote " + str(dot_fname))

    # create a new graph from file
    G = pgv.AGraph(dot_fname)  
    G.layout(prog=layout_prog)
    G.draw(img_fname)       
    if(debug):
        print("Wrote " + str(img_fname))