1#!/usr/bin/env python 2# 3# The LLVM Compiler Infrastructure 4# 5# This file is distributed under the University of Illinois Open Source 6# License. See LICENSE.TXT for details. 7# 8##===----------------------------------------------------------------------===## 9# 10# This script reads visualization data emitted by the static analyzer for 11# display in Ubigraph. 12# 13##===----------------------------------------------------------------------===## 14 15import xmlrpclib 16import sys 17 18def Error(message): 19 print >> sys.stderr, 'ubiviz: ' + message 20 sys.exit(1) 21 22def StreamData(filename): 23 file = open(filename) 24 for ln in file: 25 yield eval(ln) 26 file.close() 27 28def Display(G, data): 29 action = data[0] 30 if action == 'vertex': 31 vertex = data[1] 32 G.new_vertex_w_id(vertex) 33 for attribute in data[2:]: 34 G.set_vertex_attribute(vertex, attribute[0], attribute[1]) 35 elif action == 'edge': 36 src = data[1] 37 dst = data[2] 38 edge = G.new_edge(src,dst) 39 for attribute in data[3:]: 40 G.set_edge_attribute(edge, attribute[0], attribute[1]) 41 elif action == "vertex_style": 42 style_id = data[1] 43 parent_id = data[2] 44 G.new_vertex_style_w_id(style_id, parent_id) 45 for attribute in data[3:]: 46 G.set_vertex_style_attribute(style_id, attribute[0], attribute[1]) 47 elif action == "vertex_style_attribute": 48 style_id = data[1] 49 for attribute in data[2:]: 50 G.set_vertex_style_attribute(style_id, attribute[0], attribute[1]) 51 elif action == "change_vertex_style": 52 vertex_id = data[1] 53 style_id = data[2] 54 G.change_vertex_style(vertex_id,style_id) 55 56def main(args): 57 if len(args) == 0: 58 Error('no input files') 59 60 server = xmlrpclib.Server('http://127.0.0.1:20738/RPC2') 61 G = server.ubigraph 62 63 for arg in args: 64 G.clear() 65 for x in StreamData(arg): 66 Display(G,x) 67 68 sys.exit(0) 69 70 71if __name__ == '__main__': 72 main(sys.argv[1:]) 73 74