/usr/share/pyshared/pycallgraph/output/gephi.py is in python-pycallgraph 1.0.1-1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 | import math
from .output import Output
class GephiOutput(Output):
def __init__(self, **kwargs):
self.fp = None
self.output_file = 'pycallgraph.gdf'
Output.__init__(self, **kwargs)
@classmethod
def add_arguments(cls, subparsers, parent_parser, usage):
defaults = cls()
subparser = subparsers.add_parser(
'gephi', help='Gephi GDF generation',
parents=[parent_parser], usage=usage,
)
cls.add_output_file(
subparser, defaults, 'The generated Gephi GDF file'
)
def generate(self):
'''Returns a string with the contents of a GDF file.'''
return u'\n'.join([
self.generate_nodes(),
self.generate_edges(),
]) + '\n'
def generate_nodes(self):
output = []
fields = u', '.join([
u'name VARCHAR',
u'label VARCHAR',
u'group VARCHAR',
u'calls INTEGER',
u'time DOUBLE',
u'memory_in INTEGER',
u'memory_out INTEGER',
u'color VARCHAR',
u'width DOUBLE',
])
output.append(u'nodedef> {}'.format(fields))
for node in self.processor.nodes():
fields = u','.join([str(a) for a in [
node.name,
node.name,
node.group,
node.calls.value,
node.time.value,
node.memory_in.value,
node.memory_out.value,
u"'{}'".format(self.node_color_func(node).rgb_csv()),
self.node_size(node),
]])
output.append(fields)
return '\n'.join(output)
def node_size(self, node):
return math.log(node.time.fraction * (math.e - 1) + 1) * 2 + 1
def generate_edges(self):
output = []
fields = u', '.join([
u'node1 VARCHAR',
u'node2 VARCHAR',
u'label VARCHAR',
u'labelvisible VARCHAR',
u'directed BOOLEAN',
u'color VARCHAR',
u'width DOUBLE',
])
output.append(u'edgedef> {}'.format(fields))
for edge in self.processor.edges():
fields = u','.join([str(a) for a in [
edge.src_func,
edge.dst_func,
self.edge_label(edge),
'true',
'true',
u"'{}'".format(self.edge_color_func(edge).rgb_csv()),
edge.calls.fraction * 2,
]])
output.append(fields)
return '\n'.join(output)
def done(self):
source = self.generate()
f = open(self.output_file, 'w')
f.write(source)
f.close()
|