sketch-a-day/s053/s053.pyde

97 wiersze
3.8 KiB
Python

"""
sketch 53 180222 - Alexandre B A Villares
https://abav.lugaralgum.com/sketch-a-day
"""
from collections import namedtuple
import random as rnd
SPACING, MARGIN = 206, 150
X_LIST, Y_LIST = [], [] # listas de posições para elementos
NUM_NODES = 10 # número de elementos do desenho / number of nodes
Node = namedtuple(
'Node', ['x', 'y', 't_size', 's_weight', 'is_arrow', 'points_to'])
DESENHO = [] # lista dos elementos, 'nodes', do desenho
def setup():
size(712, 712)
noFill()
X_LIST[:] = [x for x in range(MARGIN, 1 + width - MARGIN, SPACING)]
Y_LIST[:] = [y for y in range(MARGIN, 1 + height - MARGIN, SPACING)]
novo_desenho()
println("'s' to save, and 'n' for a new drawing")
def new_node():
return Node( # elemento/"nó" uma namedtuple com:
rnd.choice(X_LIST), # x
rnd.choice(Y_LIST), # y
rnd.choice([10, 20, 30]), # t_size (tail/circle size)
rnd.choice([2, 4, 6]), # s_weight (espessura da linha)
rnd.choice([True, False]), # is_arrow? (se é seta ou 'linha')
[] # points_to... (lista com ref. a outro elem.))
)
def novo_desenho():
# esvazia a lista elementos (setas e linhas) do desenho anterior
# clears the list of nodes (drawing elements: arrows & lines)
DESENHO[:] = []
# create a a new drawing (DESENHO, a list of drawing elements)
for _ in range(NUM_NODES):
DESENHO.append(new_node())
for node in DESENHO: # para cada elemento do desenho
random_node = rnd.choice(DESENHO) # sorteia outro elemento
if (node.x, node.y) != (random_node.x, random_node.y):
# 'aponta' para este elemento, acrescenta na sub_lista
node.points_to.append(random_node)
# previous process might generate non-point_to nodes / podem rolar nós que não 'apontam'
lonely_nodes = [n for n in DESENHO if not n.points_to] # points with no points_to
if not lonely_nodes: # check if there is at least one non-pointing node
DESENHO.append(new_node()) # I want at least one (non-pointing nodes will be red)
def seta(x1, y1, x2, y2, shorter=12, head=12):
"""
O código horrível para fazer as setas, dois pares (x, y),
um parâmetro de encurtamento: shorter
e para o tamanho da cabeça da seta: head
breve no sketch #54 em diante uma versão melhor...
"""
L = dist(x1, y1, x2, y2)
with pushMatrix():
translate(x2, y2)
angle = atan2(x1 - x2, y2 - y1)
rotate(angle)
offset = -shorter * .6
line(0, offset, 0, -L - offset)
line(0, offset, -head / 3, -head + offset)
line(0, offset, head / 3, -head + offset)
def draw():
background(200)
# draws white 'lines', non-arrows, first.
for node in (n for n in DESENHO if not n.is_arrow):
strokeWeight(node.s_weight)
stroke(255)
ellipse(node.x, node.y, node.t_size, node.t_size) # desenha o círculo
for other in node.points_to: # se estiver apontando para alguém
line(node.x, node.y, other.x, other.y)
# then draws black arrows
# for node in filter(lambda n: n.is_arrow, DESENHO):
for node in (n for n in DESENHO if n.is_arrow):
strokeWeight(node.s_weight)
stroke(0)
ellipse(node.x, node.y, node.t_size, node.t_size) # desenha o círculo
for other in node.points_to: # se estiver apontando para alguém
seta(node.x, node.y, other.x, other.y,
node.t_size, node.s_weight * 5)
# then draws 'lonely nodes' in red (nodes that do not point anywhere)
for node in (n for n in DESENHO if not n.points_to):
strokeWeight(node.s_weight)
stroke(255, 0, 0)
ellipse(node.x, node.y, node.t_size, node.t_size) # desenha círculo vermelho
def keyPressed():
if key == 's':
saveFrame("####.png")
if key == 'n':
novo_desenho()