# -*- coding: utf-8 -*- from random import choice from arcs import quarter_circle, half_circle, circle_arc, bar class Cell(): grid = dict() debug_mode = False # constants N, A, I, T, L, C, E = "NAITLCE" types = {"11111": A, "00100": N, "01111": T, "11110": T, "11101": T, "10111": T, "10101": I, "01110": I, "01100": C, "00110": C, "00101": C, "10100": C, "01101": L, "10110": L, "00111": L, "11100": L, # "00000": E, # "10000": E, # "01000": E, # "00010": E } # 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)) DNL = ((-1, -1), (+1, -1), (+0, +0), (-1, +1), (+1, +1)) def __init__(self, index, cell_size, state=False, border=None): self.index = index self.state = state self.size_ = cell_size self.mouse_down = False self.variation = choice(("a", "b", "c")) self.ang = choice((0, 1, 2, 3)) self.border = border self.calculate_pos() def calculate_pos(self): i, j = self.index if self.border == None: self.border = self.size_ self.pos = PVector(self.border + self.size_ / 2 + i * self.size_, self.border + self.size_ / 2 + j * self.size_) def update(self, mx, my): # mouse over & selection treatment hs = self.size_ / 2 px, py = self.pos.x, self.pos.y self.mouse_on = (px - hs < mx < px + hs and py - hs < my < py + hs) if self.mouse_on and mousePressed: self.mouse_down = True elif self.mouse_down: self.state = not self.state self.mouse_down = False self.find_type(Cell.ONL) def plot(self, mode): quarter = self.size_ / 4. # - 1 rnd = choice(("a", "b", "c", "d")) mode_variation = {1: "a", 2: "b", 3: "c", 4: "d", 5: rnd} if self.state: strokeWeight(1) if 1 <= mode <= 5: self.variation = mode_variation[mode] self.draw_node() if mode == -1: fill(100, 100) noStroke() rect(self.pos.x, self.pos.y, self.size_, self.size_) noFill() # diagonal mode if mode == 6: i, j = self.index for (ni, nj) in Cell.DNL: nb = Cell.grid.get((i + ni, j + nj), None) if nb and nb.state: for ii in range(Cell.step_start - 2, Cell.step_end - 2, Cell.step): # (-28, 29, 7): stroke(64 - ii * 8, 128, 64 + ii * 8) if ni <> 0 and nj <> 0: bar(self.pos.x, self.pos.y, nb.pos.x, nb.pos.y, (quarter + ii) * 2, ends=(0, 1)) # else: # ellipse(self.pos.x, self.pos.y, # (a + ii) * 2, (a + ii) * 2) def draw_node(self): """ draws node """ if Cell.debug_mode: stroke(0, 255, 0) text(Cell.types(self.type), 0, 0) siz = self.size_ l = siz / 2. a = l / 2. # - 1 c = l / 2. # + 1 with pushMatrix(): translate(self.pos.x, self.pos.y) noFill() # stroke(0) if (self.type == "11110" or self.type == "10110" or self.type == "00101" ): rotate(PI) if (self.type == "11101" or self.type == "01110" or self.type == "11100" or self.type == "00110" or (self.type == "11111" and self.ang % 2) ): # t r & i rotate(HALF_PI) if (self.type == "10111" or self.type == "00111" or self.type == "01100" ): # t l rotate(PI + HALF_PI) for i in range(Cell.step_start, Cell.step_end, Cell.step): # (-28, 29, 7): stroke(8, 64 + i * 8, 64 - i * 8) if Cell.types[self.type] == Cell.A: if self.variation in "bd": quarter_circle(l, l, c + i, TOP + LEFT) quarter_circle(-l, -l, c + i, BOTTOM + RIGHT) quarter_circle(-l, l, c + i, TOP + RIGHT) quarter_circle(l, -l, c + i, BOTTOM + LEFT) if self.variation == "d": ellipse(0, 0, (a - i) * 2, (a - i) * 2) if self.variation == "a": # ellipse(0, 0, (a + i) * 2, (a + i) * 2) half_circle(-l, 0, a - i, RIGHT) half_circle(l, 0, a - i, LEFT) half_circle(0, l, a - i, TOP) half_circle(0, -l, a - i, BOTTOM) if self.variation == "c": line(+a - i, -l, +a - i, l) line(-a + i, -l, -a + i, l) half_circle(-l, 0, a - i, RIGHT) half_circle(l, 0, a - i, LEFT) # if keyPressed: ellipse(0, 0, (a - i) * 2, (a - i) * 2) elif Cell.types[self.type] == Cell.T: if self.variation in "bd": line(-l, -a + i, l, -a + i) quarter_circle(l, l, c + i, TOP + LEFT) quarter_circle(-l, l, c + i, TOP + RIGHT) elif self.variation == "c": half_circle(-l, 0, a - i, RIGHT) half_circle(l, 0, a - i, LEFT) half_circle(0, l, a - i, TOP) if self.variation in "cd": ellipse(0, 0, (a - i) * 2, (a - i) * 2) if self.variation == "a": line(-l, -a + i, l, -a + i) half_circle(-l, 0, a - i, RIGHT) half_circle(l, 0, a - i, LEFT) half_circle(0, l, a - i, TOP) elif Cell.types[self.type] == Cell.I: if self.variation == "d": ellipse(0, 0, (a - i) * 2, (a - i) * 2) if self.variation in "abd": line(+a - i, -l, +a - i, l) line(-a + i, -l, -a + i, l) if self.variation in "ca": half_circle(0, l, a - i, TOP) half_circle(0, -l, a - i, BOTTOM) elif Cell.types[self.type] == Cell.L: if self.variation in "a": quarter_circle(-l, l, siz - c - i, TOP + RIGHT) elif self.variation == "b": quarter_circle(-l, l, siz - c - i, TOP + RIGHT) i *= -1 quarter_circle(-l, l, c - i, TOP + RIGHT) elif self.variation == "c": ellipse(0, 0, (a - i) * 2, (a - i) * 2) if self.variation in "acd": half_circle(-l, 0, a - i, RIGHT) half_circle(0, l, a - i, TOP) elif Cell.types[self.type] == Cell.C: if self.variation in "ac": half_circle(0, -l, a - i, BOTTOM) if self.variation in "ab": half_circle(0, 0, a - i, BOTTOM) if self.variation in "abd": line(+a - i, -l, +a - i, 0) line(-a + i, -l, -a + i, 0) if self.variation in "dc": ellipse(0, 0, (a - i) * 2, (a - i) * 2) elif Cell.types[self.type] == Cell.N: ellipse(0, 0, (a - i) * 2, (a - i) * 2) def find_type(self, nbs): i, j = self.index[0], self.index[1] self.type = "" for (ni, nj) in nbs: nb = Cell.grid.get((i + ni, j + nj), None) if nb and nb.state: self.type += "1" else: self.type += "0"