DatasetDecoder/State.py at master · hhojatansari/DatasetDecoder · 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
import datetime as dt

ClassLabel = {
    "NoLabel": "0",
    "Meal_Preparation": "1",
    "Relax": "2",
    "Eating": "3",
    "Work": "4",
    "Sleeping": "5",
    "Wash_Dishes": "6",
    "Bed_to_Toilet": "7",
    "Enter_Home": "8",
    "Leave_Home": "9",
    "Housekeeping": "10",
    "Respirate": "11"
}

class State:

    def __init__(self):
        self.overloaded = False
        self.counter = 0
        self.datetime = dt.datetime.now()
        self.data = {
            "date": "00-00-00", "time": "00:00:00", "M001": "0", "M002": "0", "M003": "0", "M004": "0", "M005": "0",
            "M006": "0", "M007": "0", "M008": "0", "M009": "0", "M010": "0", "M011": "0", "M012": "0", "M013": "0",
            "M014": "0",
            "M015": "0", "M016": "0", "M017": "0", "M018": "0", "M019": "0", "M020": "0", "M021": "0", "M022": "0",
            "M023": "0", "M024": "0",
            "M025": "0", "M026": "0", "M027": "0", "M028": "0", "M029": "0", "M030": "0", "M031": "0", "T001": "20.0",
            "T002": "20.0",
            "T003": "20.0", "T004": "20.0", "T005": "20.0", "D001": "0", "D002": "0", "D003": "0", "D004": "0",
            "Label": "NoLabel"
        }

    def __eq__(self, other):
        if (self.data["date"] == other.data["date"] and self.data["time"] == other.data["time"]):
            return True

    def __sub__(self, other):
        dis = self.datetime - other.datetime
        return dis.days * 24 * 60 * 60 + dis.seconds - 1

    def textParser(self, string):
        year, month, day = string.split()[0].split("-")
        hour, minute, second = string.split()[1].split(":")

        year = int(year)
        month = int(month)
        day = int(day)
        hour = int(hour)
        minute = int(minute)

        if(round(float(second), 0) > 59):
            second = int(eval(second))
            self.overloaded = True
        else:
            second = int(round(float(second), 0))
        self.datetime = dt.datetime(year, month, day, hour, minute, second)
        if(self.overloaded):
            self.incrementTime()

        self.data["date"] = str(self.datetime.year) + "-" + "{0:0=2d}".format(self.datetime.month) + \
                            "-" + "{0:0=2d}".format(self.datetime.day)
        self.data["time"] = "{0:0=2d}".format(self.datetime.hour) + ":" + "{0:0=2d}".format(self.datetime.minute) + \
                            ":" + "{0:0=2d}".format(self.datetime.second)

        if (len(string.split()) == 6 or len(string.split()) == 4):
            if (string.split()[3] == "ON" or string.split()[3] == "OPEN"):
                self.data[string.split()[2]] = "1"
            elif (string.split()[3] == "OFF" or string.split()[3] == "CLOSE"):
                self.data[string.split()[2]] = "0"
            else:
                self.data[string.split()[2]] = string.split()[3]

            if (len(string.split()) == 6):
                if (string.split()[5] == "begin"):
                    self.data["Label"] = ClassLabel[string.split()[4]]
                elif (string.split()[5] == "end"):
                    self.data["Label"] = ClassLabel["NoLabel"]
        else:
            print("Error:", "len of line is " + str(len(string.split())))

    def writeState(self, file):
        file.write(
            self.data["date"] + " " + self.data["time"] + " " + self.data["M001"] + " " + self.data["M002"] + " " +
            self.data["M003"] + " " +
            self.data["M004"] + " " + self.data["M005"] + " " + self.data["M006"] + " " + self.data["M007"] + " " +
            self.data["M008"] + " " +
            self.data["M009"] + " " + self.data["M010"] + " " + self.data["M011"] + " " + self.data["M012"] + " " +
            self.data["M013"] + " " +
            self.data["M014"] + " " + self.data["M015"] + " " + self.data["M016"] + " " + self.data["M017"] + " " +
            self.data["M018"] + " " +
            self.data["M019"] + " " + self.data["M020"] + " " + self.data["M021"] + " " + self.data["M022"] + " " +
            self.data["M023"] + " " +
            self.data["M024"] + " " + self.data["M025"] + " " + self.data["M026"] + " " + self.data["M027"] + " " +
            self.data["M028"] + " " +
            self.data["M029"] + " " + self.data["M030"] + " " + self.data["M031"] + " " + self.data["T001"] + " " +
            self.data["T002"] + " " +
            self.data["T003"] + " " + self.data["T004"] + " " + self.data["T005"] + " " + self.data["D001"] + " " +
            self.data["D002"] + " " +
            self.data["D003"] + " " + self.data["D004"] + " " + self.data["Label"] + "\n")
        self.counter += 1

    def printState(self):
        print(self.data["date"], self.data["time"], self.data["M001"], self.data["M002"], self.data["M003"],
              self.data["M004"], self.data["M005"], self.data["M006"], self.data["M007"], self.data["M008"],
              self.data["M009"], self.data["M010"], self.data["M011"], self.data["M012"], self.data["M013"],
              self.data["M014"], self.data["M015"], self.data["M016"], self.data["M017"], self.data["M018"],
              self.data["M019"], self.data["M020"], self.data["M021"], self.data["M022"], self.data["M023"],
              self.data["M024"], self.data["M025"], self.data["M026"], self.data["M027"], self.data["M028"],
              self.data["M029"], self.data["M030"], self.data["M031"], self.data["T001"], self.data["T002"],
              self.data["T003"], self.data["T004"], self.data["T005"], self.data["D001"], self.data["D002"],
              self.data["D003"], self.data["D004"], self.data["Label"])

    def incrementTime(self):
        self.datetime = self.datetime + dt.timedelta(0, 1)
        self.data["date"] = str(self.datetime).split()[0]
        self.data["time"] = str(self.datetime).split()[1]