289 lines
7.6 KiB
Python
Executable File
289 lines
7.6 KiB
Python
Executable File
#!/usr/bin/python3
|
|
# -*- coding: utf-8 -*-
|
|
# -*- mode: Python -*-
|
|
|
|
|
|
'''
|
|
|
|
v0.1.0
|
|
|
|
|
|
LICENCE : CC
|
|
|
|
by cocoa
|
|
|
|
'''
|
|
from __future__ import print_function
|
|
import math
|
|
import random
|
|
import sys
|
|
import os
|
|
import time
|
|
import redis
|
|
import ast
|
|
import argparse
|
|
|
|
|
|
MAX_PARTICLES = 50
|
|
MAX_TIME = 500
|
|
|
|
argsparser = argparse.ArgumentParser(description="Dummy generator")
|
|
argsparser.add_argument("-f","--fps",help="Frame Per Second",default=30,type=int)
|
|
argsparser.add_argument("-v","--verbose",action="store_true",help="Verbose output")
|
|
# Redis Args
|
|
argsparser.add_argument("-i","--ip",help="IP address of the Redis server ",default="127.0.0.1",type=str)
|
|
argsparser.add_argument("-p","--port",help="Port of the Redis server ",default="6379",type=str)
|
|
#
|
|
argsparser.add_argument("-M","--max-particles",help="Max Particles. Default:{}".format(MAX_PARTICLES),default=MAX_PARTICLES,type=int)
|
|
argsparser.add_argument("-m","--max-time",help="Max Particles. Default:{}".format(MAX_TIME),default=MAX_TIME,type=int)
|
|
|
|
|
|
args = argsparser.parse_args()
|
|
verbose = args.verbose
|
|
ip = args.ip
|
|
port = args.port
|
|
max_particles = args.max_particles
|
|
max_time = args.max_time
|
|
|
|
def debug(*args, **kwargs):
|
|
if( verbose == False ):
|
|
return
|
|
print(*args, file=sys.stderr, **kwargs)
|
|
|
|
|
|
def rgb2int(rgb):
|
|
#debug(name,"::rgb2int rbg:{}".format(rgb))
|
|
return int('0x%02x%02x%02x' % tuple(rgb),0)
|
|
|
|
def spectrum_120( ):
|
|
return ast.literal_eval(redisData["spectrum_10"])
|
|
|
|
|
|
def rgb2int(rgb):
|
|
#debug(name,"::rgb2int rbg:{}".format(rgb))
|
|
return int('0x%02x%02x%02x' % tuple(rgb),0)
|
|
|
|
def msNow():
|
|
return time.time()
|
|
|
|
def refreshRedis():
|
|
global redisData
|
|
for key in redisKeys:
|
|
redisData[key] = ast.literal_eval(r.get(key).decode('ascii'))
|
|
|
|
name="generator::redilisys_particles"
|
|
|
|
class UnpreparedParticle(Exception):
|
|
pass
|
|
|
|
class Particle(object):
|
|
def __init__(self, x, y, m):
|
|
self.x = x
|
|
self.y = y
|
|
self.m = m
|
|
self.dx = 0
|
|
self.dy = 0
|
|
self.connectedTo = []
|
|
|
|
self.decay = random.randint(10,max_time)
|
|
self.color = (random.randint(128,256) - int(12.8 * self.m),
|
|
random.randint(128,256) - int(12.8 * self.m),
|
|
random.randint(128,256) - int(12.8 * self.m))
|
|
self.color = (255,255,255)
|
|
#debug( self.color )
|
|
|
|
def interact(self, bodies):
|
|
self.connectedTo = []
|
|
spec = redisData["spectrum_10"]
|
|
power = int(sum(spec[4:6]))
|
|
for other in bodies:
|
|
if other is self:
|
|
continue
|
|
dx = other.x - self.x
|
|
dy = other.y - self.y
|
|
dist = math.sqrt(dx*dx + dy*dy)
|
|
if dist == 0:
|
|
dist = 1
|
|
if dist < 100 and random.randint(0,power) > 0.5 :
|
|
self.connectedTo.append(other)
|
|
self.decay += 2
|
|
factor = other.m / dist**2
|
|
high_power = sum(spec[8:9]) if sum(spec[8:9]) != 0 else 0.01
|
|
self.dx += (dx * factor * self.m)
|
|
self.dy += (dy * factor * self.m)
|
|
#print "factor %f" % (factor,)
|
|
|
|
def move(self):
|
|
spec = redisData["spectrum_10"]
|
|
x_friction = (2.2-(1+spec[7]/2))
|
|
y_friction = (2.2-(1+spec[7]/2))
|
|
#x_friction = 1.02
|
|
#y_friction = 1.04
|
|
self.dx /= x_friction if x_friction != 0 else 0.01
|
|
self.dy /= y_friction if y_friction != 0 else 0.01
|
|
self.x += self.dx
|
|
self.y += self.dy
|
|
if self.x > max_width:
|
|
self.dx = - self.dx /8
|
|
self.x = max_width
|
|
if self.x < 1:
|
|
self.dx = - self.dx /8
|
|
self.x = 1
|
|
if self.y > max_height:
|
|
self.dy = - self.dy /4
|
|
self.y = max_height
|
|
if self.y < 1:
|
|
self.dy = - self.dy /4
|
|
self.y = 1
|
|
#print "(%.2f,%.2f) -> (%.2f,%.2f)" % (ox, oy, self.x, self.y)
|
|
|
|
def attractor(self,attractor):
|
|
spec = redisData["spectrum_10"]
|
|
power = sum(spec[0:4])/3
|
|
# If we're going in the direction of center, reverse
|
|
next_x = self.x + self.dx
|
|
next_y = self.y + self.dy
|
|
next_dx = attractor["x"] - self.x
|
|
next_dy = attractor["y"] - self.y
|
|
next_dist = math.sqrt(next_dx*next_dx + next_dy*next_dy)
|
|
|
|
dx = attractor["x"] - self.x
|
|
dy = attractor["y"] - self.y
|
|
dist = math.sqrt(dx*dx + dy*dy)
|
|
if dist == 0:
|
|
dist = 1
|
|
factor = power/ dist**2
|
|
x_acceleration = (dx * factor * power * power)
|
|
y_acceleration = (dx * factor * power * power)
|
|
|
|
|
|
if next_dist > dist:
|
|
self.dx -= x_acceleration * power
|
|
self.dy -= y_acceleration * power
|
|
else:
|
|
self.dx += x_acceleration
|
|
self.dy += y_acceleration
|
|
|
|
|
|
class Attractor(Particle):
|
|
def move(self):
|
|
pass
|
|
|
|
|
|
class ParticleViewer(object):
|
|
def __init__(self, particles, size=(800,800)):
|
|
(self.width, self.height) = size
|
|
self.size = size
|
|
self.particles = particles
|
|
self.xoff = 0
|
|
self.yoff = 0
|
|
self.scalefactor = 1
|
|
|
|
def redraw(self):
|
|
|
|
pl = []
|
|
drawnVectors = []
|
|
for p in self.particles:
|
|
x = int(self.scalefactor * p.x) - self.xoff
|
|
y = int(self.scalefactor * p.y) - self.yoff
|
|
if x > max_width:
|
|
x = max_width
|
|
if x < 1:
|
|
x = 1
|
|
if y > max_height:
|
|
y = max_height
|
|
if y < 1:
|
|
y = 1
|
|
|
|
color = rgb2int(p.color)
|
|
pl.append([x+1,y+1,0])
|
|
pl.append([x+1,y+1,color])
|
|
pl.append([x,y,color])
|
|
|
|
for other in p.connectedTo:
|
|
|
|
if [other,self] in drawnVectors:
|
|
continue
|
|
drawnVectors.append([other,self])
|
|
pl.append([x,y,0])
|
|
pl.append([x,y,color])
|
|
pl.append([other.x,other.y,color])
|
|
|
|
print(pl,flush = True)
|
|
|
|
def decayParticles(self):
|
|
for i,p in enumerate(self.particles):
|
|
# Handle positional decay
|
|
if p.decay == 0:
|
|
del self.particles[i]
|
|
continue
|
|
p.decay = p.decay - 1
|
|
# Handle color decay
|
|
n = int(255 * (p.decay / max_time ))
|
|
p.color = (n,n,n)
|
|
|
|
|
|
def emitParticles(self):
|
|
spec = redisData["spectrum_10"]
|
|
power = sum(spec[6:])
|
|
if len(self.particles ) > math.sqrt(max_particles):
|
|
if len(self.particles) > max_particles:
|
|
return
|
|
if random.random() > power:
|
|
return
|
|
# x is either left or right
|
|
d = 600
|
|
rx = 100 if random.randint(0,1) else 700
|
|
#rx = random.randint(1,max_width)
|
|
ry = random.randint(1,max_height)
|
|
spec = redisData["spectrum_10"]
|
|
m = random.randint(1,1+int(10*spec[7]))
|
|
particles.append(Particle(rx, ry, m))
|
|
|
|
|
|
def tick(self):
|
|
self.decayParticles()
|
|
self.emitParticles()
|
|
for p in self.particles:
|
|
p.interact(self.particles)
|
|
p.attractor({
|
|
"x":max_width/2,
|
|
"y":max_height/2
|
|
})
|
|
for p in particles:
|
|
p.move()
|
|
self.redraw()
|
|
|
|
def scale(self, factor):
|
|
self.scalefactor += factor
|
|
|
|
|
|
|
|
|
|
max_width = 800
|
|
max_height = 800
|
|
redisKeys = ["spectrum_120","spectrum_10"]
|
|
redisData = {}
|
|
redisLastHit = msNow() - 99999
|
|
r = redis.Redis(
|
|
host=ip,
|
|
port=port)
|
|
|
|
white = 16777215
|
|
|
|
refreshRedis()
|
|
if __name__ == "__main__":
|
|
particles = []
|
|
# particles.append(Attractor(320, 200, 10))
|
|
# particles.append(Attractor(100, 100, 10))
|
|
|
|
win = ParticleViewer(particles)
|
|
try:
|
|
while True:
|
|
win.tick()
|
|
refreshRedis()
|
|
time.sleep(.03)
|
|
|
|
except KeyboardInterrupt:
|
|
pass
|