Thesis » QuynhNguyen-MS

#!/usr/bin/env python

import argparse

import os

import utility

from process_out1 import process_out1

from process_out2 import process_out2

class Analyzer():

    analyzer_types = {1: "process_out1",

                   2: "process_out2"}

    def __init__(self):

        self.args = None

        self.result_analyzer = None

    def parse_args(self):

        parser = argparse.ArgumentParser(description="Performing analysis for \

                                         Betweenness Centrality (BC) and \

                                         Heuristic BC")

        parser.add_argument("-t", dest="type", help="determining whether it's \

                            *.out1 or *.out2", type=int,

                            choices=self.analyzer_types.keys(), required=True)

        parser.add_argument("-i", dest="input_file", help="path to input file",

                            required=True)

        parser.add_argument("-o", dest="output_file", help="path to output file",

                            default="", required=False)

        parser.add_argument("-p", dest="prefix", required=False,

                            help="prefix to add to output files", type=str,

                            default=None)

        self.args = parser.parse_args()

    def process_out1(self, input_filepath, output_filepath):

        input_data = utility.read_out1(input_filepath)

        bc_result = utility.get_statistical_data(input_data, 'filename', 'bc_time')

        hbc_result = utility.get_statistical_data(input_data, 'filename', 'hbc_time')

        output_data = utility.format_output_data_out1(bc_result, hbc_result)

        utility.write_to_file(output_filepath, output_data)

        return True

    def process_out2(self, input_filepath, output_filepath):

        input_data = utility.read_out2(input_filepath)

        bc_result = utility.get_statistical_data(input_data, 'typotype', 'avg_bc')

        hbc_result = utility.get_statistical_data(input_data, 'typotype', 'avg_hbc')

        output_data = utility.format_output_data_out2(bc_result, hbc_result)

        utility.write_to_file(output_filepath, output_data)

        return True

    def analyze_result(self):

        analyzer_function = getattr(self,

                                          self.analyzer_types[self.args.type])

        r = analyzer_function(self.args.input_file, self.args.output_file)

        if r:

            print "Done analyzing %s" % self.args.input_file

            print "Result is saved in: %s" % self.args.output_file

if __name__ == '__main__':

    analyzer = Analyzer()

    analyzer.parse_args()

    input_filepath = analyzer.args.input_file

    input_dir = os.path.dirname(os.path.realpath(input_filepath))

    input_filename = os.path.splitext(os.path.basename(input_filepath))[0]

    output_filename = '%s.out%s' % (input_filename, (analyzer.args.type + 1))

    if analyzer.args.prefix:

        output_name = analyzer.args.prefix

        analyzer.args.output_file = '%s/%s' % (analyzer.args.prefix, output_filename)

    # This option has higher priority

    if analyzer.args.output_file == "":

        analyzer.args.output_file = '%s/%s' % (input_dir, output_filename)

        output_filepath = '%s/%s.out2' % (input_dir, input_filename)

        analyzer.args.output_file = output_filepath

    analyzer.analyze_result()