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
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
|
# ############################################################################ #
# #
# ::: :::::::: #
# philo.py :+: :+: :+: #
# +:+ +:+ +:+ #
# By: charles <me@cacharle.xyz> +#+ +:+ +#+ #
# +#+#+#+#+#+ +#+ #
# Created: 2020/09/27 17:49:41 by charles #+# #+# #
# Updated: 2020/09/27 18:35:25 by charles ### ########.fr #
# #
# ############################################################################ #
import re
import time
import enum
import itertools
class FormatError(Exception):
pass
class LogError(Exception):
pass
class Event(enum.Enum):
EATING = 1
SLEEPING = 2
THINKING = 3
DIED = 4
NONE = 5
@staticmethod
def to_verb(event):
return {
Event.EATING: "eat",
Event.SLEEPING: "sleep",
Event.THINKING: "think",
Event.DIED: "die",
Event.NONE: "none",
}[event]
class Log:
def __init__(self, line, philo_num):
match = re.match(
r"^(?P<timestamp>\d+) "
r"(?P<id>\d+) "
r"(?P<event>is thinking|is eating|is sleeping|died)$",
line
)
if match is None:
raise FormatError("couldn't parse line")
curr = int(time.time() * 1000)
self.timestamp = Log._parse_ranged_int(match.group("timestamp"), curr - 100, curr + 100)
self.id = Log._parse_ranged_int(match.group("id"), 1, philo_num)
self.event = {
"is thinking": Event.THINKING,
"is eating": Event.EATING,
"is sleeping": Event.SLEEPING,
"died": Event.DIED,
}[match.group('event')]
@staticmethod
def _parse_ranged_int(s, lo, hi):
try:
value = int(s)
if not (lo <= value <= hi):
raise FormatError("`{}` should be between {} - {}".format(s, lo, hi))
except ValueError:
raise FormatError("`{}` sould be an integer".format(s))
return value
def __repr__(self):
return "Log({}ms #{} {})".format(self.timestamp, self.id, self.event)
class Philo:
def __init__(self, id_: int, timeout_eat: int, meal_num: int = 1):
self._logs = []
self.id = id_
self.meal_num = meal_num
self._timeout_eat = timeout_eat
def add_log(self, log):
self._logs.append(log)
def check(self):
grouped = [(e, list(g)) for e, g in itertools.groupby(self._logs, (lambda x: x.event))]
for e, g in grouped:
if e is Event.EATING:
if len(g) != self.meal_num:
raise LogError("should eat {} times".format(self.meal_num))
else:
if len(g) != 1:
raise LogError("should {} 1 time".format(Event.to_verb(e)))
events = [e for e, _ in grouped]
for e1, e2 in zip(events, events[1:]):
if e2 is Event.DIED:
break
second = {
Event.THINKING: Event.EATING,
Event.EATING: Event.SLEEPING,
Event.SLEEPING: Event.EATING
}[e1]
if second is not e2:
raise LogError("{} should switch to {}, actual {}".format(e1, second, e2))
last_eat_time = int(time.time() * 1000)
for log in reversed(self._logs):
if log.event is Event.EATING:
last_eat_time = log.timestamp
break
if int(time.time() * 1000) - last_eat_time > self._timeout_eat + 20:
raise LogError("should be dead")
@property
def last_event(self):
if len(self._logs) == 0:
return Event.NONE
return self._logs[-1].event
class Table:
def __init__(self, timeout_eat, philo_num):
self._philos = [Philo(id_, timeout_eat) for id_ in range(1, philo_num + 1)]
self._logs = []
self._philo_num = philo_num
self.dead = False
def add_log(self, log):
if self.dead:
raise LogError("should not output after one died")
if log.event is Event.DIED:
self.dead = True
self._logs.append(log)
philo = next(p for p in self._philos if p.id == log.id)
philo.add_log(log)
def check(self):
if self.dead:
return
fork_used = 2 * len([p for p in self._philos if p.last_event == Event.EATING])
if fork_used > self._philo_num:
raise LogError("using nonexistant forks")
for l1, l2 in zip(self._logs, self._logs[1:]):
if l1.timestamp > l2.timestamp:
raise LogError("timestamp not in ordered")
for p in self._philos:
p.check()
|
