war-games/parse.py at master · P2PSP/war-games · GitHub

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
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
#!/usr/bin/env python2.7
# -*- coding: utf-8 -*-

import os, sys, getopt
import re
import glob

max_buffer_correctness = 0
min_buffer_correctness = 1
max_buffer_filling = 0
min_buffer_filling = 1

experiment_path = ""

log_files = {}

def usage():
    print ""
    return

def readAllFiles():
    ret = {}
    fileList = glob.glob("{0}/peer*.log".format(experiment_path))

    regex = re.compile("(\d*)\tbuffer\s(correctnes|filling|fullness)\s(\d*.\d*)")

    for f in fileList:
        ret[f] = []
        with open(f) as fh:
            for line in fh:
                result = regex.match(line)
                if result != None:
                    ret[f].append((int(result.group(1)), result.group(2), float(result.group(3))))

    return ret

def calcAverageBufferCorrectnes(roundTime):
    correctnesSum = fillingSum = fullnessSum = 0.0
    NN = 0

    for f in log_files:
        info = calcAverageInFile(log_files[f], roundTime)
        if (info[0] != None and info[1] != None and info[2] != None):
            correctnesSum += info[0]
            fillingSum += info[1]
            fullnessSum += info[2]
            NN += 1

    if NN == 0:
        return (None,None,None)
    return (correctnesSum / NN, fillingSum / NN, fullnessSum / NN)

def calcAverageInFile(fileLines, roundTime):
    correctness = None
    filling = None
    fullness = None

    for line in fileLines:
        if roundTime == line[0]:
            if line[1] == 'correctnes':
                correctness = line[2]
            elif line[1] == 'filling':
                filling = line[2]
            elif line[1] == 'fullness':
                fullness = line[2]

        if correctness != None and filling != None and fullness != None:
            return (correctness, filling, fullness)

    return (correctness, filling, fullness)


def main(args):
    global experiment_path
    global log_files
    inFile = ""
    nPeers = nMalicious = lastRound = 0
    try:
        opts, args = getopt.getopt(args, "r:d:")
    except getopt.GetoptError:
        usage()
        sys.exit(2)
    for opt, arg in opts:
        if opt == "-r":
            lastRound = int(arg)
        elif opt == "-d":
            experiment_path = str(arg)

    regex = re.compile("(\d*)\t(\d*)\s(\d*)\s(.*)")
    startParse = False
    roundOffset = 0
    log_files = readAllFiles()
    print "round\t#WIPs\t#MPs\t#TPs\tteamsize\tcorrectness\tfilling\tfullness"
    with open("{0}/splitter.log".format(experiment_path)) as f:
        for line in f:
            result = regex.match(line)
            if result != None:
                ts = int(result.group(1))
                currentRound = int(result.group(2))
                currentTeamSize = int(result.group(3))
                peers = result.group(4).split(' ')
                trusted = 0
                malicious = 0

                with open("trusted.txt", "r") as fh:
                    for line in fh:
                        if line[:-1] in peers:
                            trusted += 1

                with open("malicious.txt", "r") as fh:
                    for line in fh:
                        if line[:-1] in peers:
                            malicious += 1

                if currentRound >= lastRound and not startParse:
                    startParse = True
                    roundOffset = currentRound
                if startParse:
                    info = calcAverageBufferCorrectnes(ts)
                    if (info[0] != None and info[1]!=None):
                        print "{0}\t{1}\t{2}\t{3}\t{4}\t{5}\t{6}\t{7}".format(currentRound - roundOffset + 1, len(peers) - malicious - trusted, malicious, trusted, currentTeamSize, info[0], info[1], info[2])
    return 0

if __name__ == "__main__":
    sys.exit(main(sys.argv[1:]))