Thesis » QuynhNguyen-MS

from os import listdir

from os.path import isfile, join, splitext, basename

import sys

import networkx as nx

import numpy

sys.path.append("heuristic_bc/")

from heuristic_betweenness_centrality import HeuristicBetweennessCentrality as HBC

from pdb import set_trace as debugger

def get_all_files_from_dir(dir):

    return [join(dir, f) for f in listdir(dir) if isfile(join(dir, f))]

if __name__ == '__main__':

    input_dir = ''

    # Default input directory

    GRAPH_DIRS='/home/quynh/Thesis/quynhnguyen-ms/experiment_input_graphs/CNGraphs/snapshots/small_samples'

    if len(sys.argv) > 1:

        input_dir = sys.argv[1]

    else:

        input_dir = GRAPH_DIRS

    onlyfiles = get_all_files_from_dir(input_dir)

    for in_filepath in onlyfiles:

        filename = splitext(basename(in_filepath))[0]

        graph = nx.read_weighted_edgelist(in_filepath)

        out_filepath = join('output/', filename + '.hbcout')

        # Brandes BC

        bbc = nx.betweenness_centrality(graph, weight=True, normalized=True, endpoints=True)

        out_data = [[k, v] for (k, v) in bbc.iteritems()]

        numpy.savetxt(out_filepath, out_data, delimiter=',', fmt='%s')

        print "Finish calculating BC for %s" % filename

        # # Heuristic BC

        # hbc = HBC(graph)

        # hbc.write(out_filepath)