sketch-a-day/s336/cell.py

70 wiersze
2.3 KiB
Python
Czysty Zwykły widok Historia

2018-12-01 00:25:40 +00:00
class Cell():
#neighbours list
NL = ((-1, -1), (+0, -1), (+1, -1),
(-1, +0), (+0, +0), (+1, +0),
(-1, +1), (+0, +1), (+1, +1))
ONL = ( (+0, -1),
(-1, +0), (+0, +0), (+1, +0),
(+0, +1) )
def __init__(self, index, cell_size):
# Cell.NL = ((-1, -1), (+0, -1), (+1, -1),
# (-1, +0), (+0, +0), (+1, +0),
# (-1, +1), (+0, +1), (+1, +1))
self.index = index
self.state = False
self.size_ = cell_size
self.mouse_down = False
i, j = index[0], index[1]
self.pos = PVector(self.size_/2 + i * self.size_,
self.size_/2 + j * self.size_)
# self.ngbs = []
# for ni, nj in Cell.NL:
# self.ngbs.append(
# Cell.grid.get((i-ni, j-nj), None))
def play(self):
hs = self.size_ / 2
px, py = self.pos.x, self.pos.y
self.mouse_on = (px - hs < mouseX < px + hs and
py - hs < mouseY < py + hs)
if self.mouse_on and mousePressed:
self.mouse_down = True
if self.mouse_down and not mousePressed:
self.state = not self.state
self.mouse_down = False
self.plot()
def plot(self):
fill(255)
rect(self.pos.x, self.pos.y, self.size_, self.size_)
if self.state:
third = self.size_ / 3
fill(0)
if key == CODED:
nbs = Cell.NL
else:
nbs = Cell.ONL
i, j = self.index[0], self.index[1]
for (ni, nj) in nbs:
nb = Cell.grid.get((i+ni, j+nj), None)
if nb and nb.state:
fill(255, 0, 0)
else:
fill(250)
#noStroke()
rect(self.pos.x + ni * third,
self.pos.y + nj * third,
third, third)
if self.mouse_on:
with pushStyle():
fill(128, 128)
rect(self.pos.x, self.pos.y,
self.size_,
self.size_
)
def keyPressed():
saveFrame("###.png")