tokens.py
2.67 KB
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
import enum
from .data import hdict, mset
from .utils import autorepr
class Token (object) :
pass
@autorepr
class CodeToken (Token) :
def __init__ (self, code) :
self.code = code
def __ast__ (self, place, ctx) :
return ctx.Expr(self.code,
BLAME=ctx.TokenBlame(place, self.code))
def __str__ (self) :
return "(%s)" % self.code
def __hash__ (self) :
return hash(self.code)
def __eq__ (self, other) :
try :
return self.code == other.code
except AttributeError :
return False
def __ge__ (self, other) :
try :
return self.code >= other.code
except AttributeError :
return False
def __gt__ (self, other) :
try :
return self.code >= other.code
except AttributeError :
return False
def __le__ (self, other) :
return not self.__gt__(other)
def __lt__ (self, other) :
return not self.__ge__(other)
class BlackToken (Token, enum.IntEnum) :
DOT = enum.auto()
dot = BlackToken.DOT
class BlackWhiteToken (Token, enum.IntEnum) :
BLACK = enum.auto()
WHITE = enum.auto()
black = BlackWhiteToken.BLACK
white = BlackWhiteToken.WHITE
class Marking (hdict) :
def __repr__ (self) :
return "%s(%s)" % (self.__class__.__name__, dict.__repr__(self))
def __call__ (self, place) :
return self.get(place, mset())
def copy (self) :
return self.__class__((k, v.copy()) for k, v in self.items())
def __add__ (self, other) :
new = self.copy()
for k, v in other.items() :
if k not in new :
new[k] = v
else :
new[k].add(v)
return new
def __sub__ (self, other) :
new = self.copy()
for k, v in other.items() :
if k not in self :
raise ValueError("no such place %r" % k)
elif new[k] == v :
del new[k]
else :
new[k] -= v
return new
def __le__ (self, other) :
return all(self(k) <= other(k) for k in self.keys())
def __lt__ (self, other) :
return (self != other) and (self <= other)
def __ge__ (self, other) :
return all(self(k) >= other(k) for k in other.keys())
def __gt__ (self, other) :
return (self != other) and (self >= other)
def __ast__ (self, ctx) :
m = {}
for place, tokens in self.items() :
if tokens :
m[place] = [t.__ast__(place, ctx) for t in tokens]
return [ctx.PlaceMarking(place, tokens, BLAME=ctx.PlaceBlame(place, tokens))
for place, tokens in m.items()]