hex/Counter.py at master · source99/hex · 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
from time_utils import convert_directory_to_ms, convert_ms_to_time
import bisect
import collections
import glob
import os
SCROLL_OFFSET = 0.00

class Counter:

    def __init__(self, directory):
        print "counter_index directory = {}".format(directory)

        folder = directory.split("/")[-1]
        counter_file = glob.glob("{}/Counter/*/*/Counter.ipd".format(directory))[0]
        print "counter file = {}".format(counter_file)
        start_time_directory = counter_file.split(os.sep)[-2]
        self.date = counter_file.split(os.sep)[-3]
        fp = open(counter_file,'r')

        self.ms_to_payout_mapping = collections.OrderedDict()
        self.payout_to_ms_mapping = {}
        self.file_contents = fp.readlines()
        self.length = len(self.file_contents)
        self.start_time_ms = convert_directory_to_ms(start_time_directory)

        for line in self.file_contents:
            #payout;ms
            payout,ms = line.rstrip().split(";")
            payout = int(payout)
            ms = int(ms)
            self.ms_to_payout_mapping[ms+self.start_time_ms] = payout
#           self.ms_to_payout_mapping[ms] = payout
            self.payout_to_ms_mapping[payout] = ms+self.start_time_ms
        fp.close()

    def get_closest_payout(self,ms_input):
        index = bisect.bisect_left(self.ms_to_payout_mapping.keys(),ms_input)
        return self.ms_to_payout_mapping.values()[index]

    def get_closest_time(self,ms_input):
        index = bisect.bisect_left(self.ms_to_payout_mapping.keys(),ms_input)
        return self.ms_to_payout_mapping.keys()[index]


    def get_payout(self, percent):
        index = int(self.length * (percent+SCROLL_OFFSET))
        line = self.file_contents[index]
        payout = int(line.split(";")[0])
        return payout / 100.0
    def get_ms(self,percent):
        index = int(self.length * (percent+SCROLL_OFFSET))
        line = self.file_contents[index]
        ms_time = int(line.split(";")[1])
        return ms_time
    def get_time(self,percent):
        index = int(self.length * (percent+SCROLL_OFFSET))
        line = self.file_contents[index]
        ms_time = int(line.split(";")[1]) + self.start_time_ms
        return convert_ms_to_time(ms_time)
        return ms_time

    def get_frame_name(self,ms):
        ms_of_frame = self.get_closest_time(ms)
#        ms_of_frame = self.payout_to_ms_mapping[payout]
        print "input time = {} : closest frame time = {}".format(ms, ms_of_frame)
        human_time = convert_ms_to_time(ms_of_frame)
        return self.date + "_" + human_time + ".png"


def create_counter_from_directory(directory):
    folder = directory.split("\\")[-1]
    counter = glob.glob("{}\\Counter\\*\\*\\Counter.ipd".format(directory))
    start_time = counter[0].split("\\")[-2]
    print "start time = {}".format(start_time)
    counter = Counter.Counter(counter[0], start_time)
    return counter