154 lines
4.3 KiB
Python
154 lines
4.3 KiB
Python
from scene import Scene
|
|
from triangles import draw_line
|
|
from triangles import lineTri
|
|
import math
|
|
white = 63
|
|
fire1 = 6
|
|
fire2 = 5
|
|
fire3 = 4
|
|
fire4 = 3
|
|
fire5 = 2
|
|
fire6 = 56
|
|
|
|
water1 = 31
|
|
water2 = 30
|
|
water3 = 29
|
|
water4 = 28
|
|
|
|
fireGradient = [white, white, fire1, fire1, fire2, fire3, fire4, fire3, fire4, fire5, fire6, fire6, fire6]
|
|
waterGradient = [white, white, water1, water2, water3, water4, water4]
|
|
|
|
oldMouseX = 0
|
|
oldMouseY = 0
|
|
|
|
mouseVelX = 0
|
|
mouseVelY = 0
|
|
|
|
|
|
def tri(x1,y1,x2,y2,x3,y3,c):
|
|
draw_line(x1,y1,x2,y2,c)
|
|
draw_line(x2,y2,x3,y3,c)
|
|
draw_line(x1,y1,x3,y3,c)
|
|
|
|
class Particle:
|
|
def __init__(self, x, y, velX, velY):
|
|
self.x = x
|
|
self.y = y
|
|
self.velX = velX
|
|
self.velY = velY
|
|
self.c = fireGradient[0]
|
|
self.size = 5
|
|
|
|
class Particles(Scene):
|
|
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.fireParts = []
|
|
self.waterParts = []
|
|
for i in range(10):
|
|
self.fireParts.append(Particle(width/2, height/2, rnd(-1, 1), rnd(-1, 1)))
|
|
self.waterParts.append(Particle(width/2, height/2, rnd(-1, 1), rnd(-1, 1)))
|
|
|
|
self.counter = 0
|
|
self.title = "Particles"
|
|
|
|
|
|
def update(self):
|
|
global mouseVelX
|
|
global mouseVelY
|
|
|
|
global oldMouseX
|
|
global oldMouseY
|
|
|
|
mouseVelX = (mouseX - oldMouseX)
|
|
mouseVelY = (mouseY - oldMouseY)
|
|
|
|
oldMouseX = mouseX
|
|
oldMouseY = mouseY
|
|
|
|
self.counter += 1
|
|
|
|
for p in self.fireParts[:]:
|
|
p.x += p.velX
|
|
p.y += p.velY
|
|
p.velX += rnd(-0.3,0.3)
|
|
p.velY += rnd(-0.3,0.3)
|
|
|
|
idx = int(p.size / 5.0 * len(fireGradient) - 1)
|
|
|
|
p.c = fireGradient[(len(fireGradient) - 1) - idx]
|
|
|
|
if(p.size < 1):
|
|
p.size -= 0.01
|
|
else:
|
|
p.size -= 0.05
|
|
if(p.size < 0):
|
|
self.fireParts.remove(p)
|
|
|
|
for p in self.waterParts[:]:
|
|
p.x += p.velX
|
|
p.y += p.velY
|
|
p.velX += rnd(-0.3,0.3)
|
|
p.velY += rnd(-0.3,0.3)
|
|
|
|
idx = int(p.size / 5.0 * len(waterGradient) - 1)
|
|
|
|
p.c = waterGradient[(len(waterGradient) - 1) - idx]
|
|
|
|
if(p.size < 1):
|
|
p.size -= 0.01
|
|
else:
|
|
p.size -= 0.05
|
|
if(p.size < 0):
|
|
self.waterParts.remove(p)
|
|
|
|
|
|
|
|
if(mouse(0)):
|
|
rad = sin(self.counter * 0.01) * 50
|
|
x1 = mouseX + (sin(self.counter * 0.1) * rad)
|
|
y1 = mouseY + (cos(self.counter * 0.1) * rad)
|
|
|
|
x2 = mouseX + (sin(self.counter * 0.1 + (3.141592653)) * rad)
|
|
y2 = mouseY + (cos(self.counter * 0.1 + (3.141592653)) * rad)
|
|
for i in range(10):
|
|
self.fireParts.append(Particle(x1, y1, rnd(-1,1) + mouseVelX * 0.25, rnd(-1,1) + mouseVelY * 0.25))
|
|
self.waterParts.append(Particle(x2, y2, rnd(-1,1) + mouseVelX * 0.25, rnd(-1,1) + mouseVelY * 0.25))
|
|
|
|
def draw(self):
|
|
clear(0)
|
|
for p in self.fireParts:
|
|
if(p.size <= 1):
|
|
pixel(p.x, p.y, p.c)
|
|
else:
|
|
circle(p.x, p.y, int(p.size), p.c)
|
|
|
|
for p in self.waterParts:
|
|
if(p.size <= 1):
|
|
pixel(p.x, p.y, p.c)
|
|
else:
|
|
circle(p.x, p.y, int(p.size), p.c)
|
|
text("C to add particles, T to toggle palette", 3, 3, 59)
|
|
text("C to add particles, T to toggle palette", 2, 2, 31)
|
|
text("Num Particles: " + str(len(self.fireParts) + len(self.waterParts)), 3, 16, 59)
|
|
text("Num Particles: " + str(len(self.fireParts) + len(self.waterParts)), 2, 15, 30)
|
|
text("FPS: " + str(fps()), 3, 29, 59)
|
|
text("FPS: " + str(fps()), 2, 28, 29)
|
|
# triangle(10, 10, 50, 10, mouseX, mouseY, 5)
|
|
# lineTri(10, 10, 50, 10, mouseX, mouseY, 9)
|
|
pixel(mouseX, mouseY, 9)
|
|
cx = width/2
|
|
cy = height/2
|
|
ang = math.atan2(mouseY - cy, mouseX - cx)
|
|
dist = math.sqrt(math.pow(mouseX - cx, 2) + math.pow(mouseY - cy, 2))
|
|
|
|
x2 = cx + cos(ang + (math.pi * 2 / 3)) * dist
|
|
y2 = cy + sin(ang + (math.pi * 2 / 3)) * dist
|
|
|
|
x3 = cx + cos(ang + (math.pi * 2 / 3 * 2)) * dist
|
|
y3 = cy + sin(ang + (math.pi * 2 / 3 * 2)) * dist
|
|
|
|
# triangle(mouseX, mouseY, x2, y2, x3, y3, 9)
|
|
|
|
# lineTri(mouseX, mouseY, x2, y2, x3, y3, 8)
|