In [1]:

    

%matplotlib inline
from bigbang.archive import Archive
from git_data import Repos;
from git_data import GitRepo
import matplotlib.pyplot as plt
import numpy as np
import pandas as pd


    

In [2]:

    

repo = GitRepo.GitRepo("..")
full_info = repo.commit_data;


    

In [3]:

    

print(full_info.columns)


    

    




Index([u'Commit Message', u'Committer Email', u'Committer Name', u'HEXSHA', u'Parent Commit', u'Time', u'Touched File'], dtype='object')

In [4]:

    

import networkx as nx;


    

    




Couldn't import dot_parser, loading of dot files will not be possible.

In [5]:

    

G = nx.Graph()

def add_to_graph(r): # Adds info from row to graph
    author = r["Committer Name"]
    files = r["Touched File"]
    for f in files:
        if G.has_edge(author, f):
            G[author][f]['weight'] += 1
        else:
            G.add_edge(author, f, weight = 1)
            G.node[author][type] = 'author'
            G.node[f][type] = 'file'
        
a = full_info.apply(add_to_graph, axis=1) # Dummy to make it not print
widths = [G[u][v]['weight']/20.0 for u, v in G.edges()]


    

In [ ]:

    

#nx.draw(G)
plt.figure(1,figsize=(15, 15))
pos = nx.spring_layout(G, k = 0.6, iterations = 20)
nx.draw_networkx_nodes(G, pos, nodelist = [node[0] for node in G.nodes(data=True) if node[1][type] == 'author'], node_size=750, node_color = 'r')
nx.draw_networkx_nodes(G, pos, nodelist = [node[0] for node in G.nodes(data=True) if node[1][type] == 'file'], node_color = 'b')
nx.draw_networkx_edges(G, pos, width=widths, edge_color = 'g', alpha = 0.8)
#nx.draw_networkx_edges(G, pos, width=widths, edge_color = widths)
nx.draw_networkx_labels(G, pos);