kopia lustrzana https://github.com/villares/sketch-a-day
153 wiersze
4.3 KiB
Python
153 wiersze
4.3 KiB
Python
# Alexandre B A Villares - https://abav.lugaralgum.com/sketch-a-day
|
|
# s204 20180721
|
|
|
|
from gif_export_wrapper import *
|
|
add_library('gifAnimation')
|
|
add_library('peasycam')
|
|
GRID_SIZE = 10
|
|
SKETCH_NAME = "s204"
|
|
OUTPUT = ".png"
|
|
color_mode = True
|
|
starting_node = 0
|
|
|
|
def setup():
|
|
lights()
|
|
size(700, 700, P3D)
|
|
cam = PeasyCam(this, 100)
|
|
cam.setMinimumDistance(1000)
|
|
cam.setMaximumDistance(1000)
|
|
colorMode(HSB)
|
|
strokeWeight(2)
|
|
print_text_for_readme(SKETCH_NAME, OUTPUT)
|
|
border = 50
|
|
spacing = (width - border * 2) / GRID_SIZE
|
|
Node.spacing = spacing
|
|
for x in range(GRID_SIZE):
|
|
for y in range(GRID_SIZE):
|
|
for z in range(GRID_SIZE):
|
|
Node.nodes.append(Node(
|
|
border + spacing / 2 + x * spacing - width / 2,
|
|
border + spacing / 2 + y * spacing - width / 2,
|
|
border + spacing / 2 + z * spacing - width / 2))
|
|
for node in Node.nodes:
|
|
node.set_nbs()
|
|
|
|
clear_grid()
|
|
|
|
|
|
def draw():
|
|
#lights()
|
|
background(0)
|
|
for node in Node.nodes:
|
|
node.plot_links()
|
|
node.update()
|
|
gif_export(GifMaker)
|
|
|
|
def clear_grid():
|
|
for node in Node.nodes:
|
|
node.visited = False
|
|
node.links = []
|
|
node.cor = None
|
|
Node.nodes[starting_node].current = True
|
|
|
|
|
|
class Node():
|
|
nodes = []
|
|
|
|
def __init__(self, x, y, z):
|
|
self.x = x
|
|
self.y = y
|
|
self.z = z
|
|
self.visited = False
|
|
self.current = False
|
|
self.links = []
|
|
self.cor = None
|
|
|
|
def plot_links(self):
|
|
if self.cor is not None:
|
|
c = color(self.cor % 256, 255, 255)
|
|
else:
|
|
c = color(255)
|
|
|
|
if color_mode:
|
|
stroke(0)
|
|
fill(c)
|
|
else:
|
|
stroke(c)
|
|
fill(0)
|
|
with pushMatrix():
|
|
translate(self.x, self.y, self.z)
|
|
box(Node.spacing / 2)
|
|
for node in self.links:
|
|
mid_x = (self.x + node.x) / 2
|
|
mid_y = (self.y + node.y) / 2
|
|
mid_z = (self.z + node.z) / 2
|
|
with pushMatrix():
|
|
translate(mid_x, mid_y, mid_z)
|
|
box(Node.spacing / 2)
|
|
|
|
|
|
def set_nbs(self):
|
|
self.nbs, self.unvisited_nbs = [], []
|
|
for node in Node.nodes:
|
|
if node != self and dist(node.x, node.y, node.z,
|
|
self.x, self.y, self.z) <= Node.spacing * 1.1:
|
|
self.nbs.append(node)
|
|
self.unvisited_nbs.append(node)
|
|
|
|
def set_unvisited_nbs(self):
|
|
self.unvisited_nbs = [node for node in self.nbs
|
|
if not node.visited]
|
|
|
|
def update(self):
|
|
self.set_unvisited_nbs()
|
|
if self.current:
|
|
self.visited = True
|
|
if not self.cor:
|
|
self.cor = 0
|
|
if self.unvisited_nbs:
|
|
for unvisited_nb in self.unvisited_nbs[::2]:
|
|
self.links.append(unvisited_nb)
|
|
self.current = False
|
|
unvisited_nb.current = True
|
|
unvisited_nb.visited = True
|
|
unvisited_nb.cor = self.cor + 10
|
|
else:
|
|
branch_nodes = [node for node in Node.nodes
|
|
if node.visited and node.unvisited_nbs]
|
|
if branch_nodes:
|
|
print(len(branch_nodes))
|
|
next = branch_nodes[-1]
|
|
self.current = False
|
|
next.current = True
|
|
else:
|
|
print("finished")
|
|
global starting_node
|
|
starting_node += 1
|
|
clear_grid()
|
|
|
|
def keyPressed():
|
|
global color_mode
|
|
if key == 'c':
|
|
color_mode = not color_mode
|
|
if key in ['p', 'P']:
|
|
saveFrame("####" + SKETCH_NAME + OUTPUT)
|
|
if key == 's':
|
|
gif_export(GifMaker, finish=True)
|
|
if key == 'g':
|
|
gif_export(GifMaker)
|
|
if key == 'r':
|
|
global starting_node
|
|
starting_node += 1
|
|
clear_grid()
|
|
|
|
|
|
def print_text_for_readme(name, output):
|
|
println("""
|
|
|
|
|
|
{1}: [code](https://github.com/villares/sketch-a-day/tree/master/{0}) [[Py.Processing](https://villares.github.io/como-instalar-o-processing-modo-python/index-EN)]
|
|
|
|
""".format(name, name[1:], output)
|
|
)
|