forked from protonphoton/LJ
Merge branch 'master' of https://git.interhacker.space/teamlaser/LJ
This commit is contained in:
commit
0e8db219c2
201
clitools/filters/anaglyph.py
Executable file
201
clitools/filters/anaglyph.py
Executable file
@ -0,0 +1,201 @@
|
|||||||
|
#!/usr/bin/python3
|
||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
# -*- mode: Python -*-
|
||||||
|
|
||||||
|
|
||||||
|
'''
|
||||||
|
|
||||||
|
anaglyph
|
||||||
|
v0.1.0
|
||||||
|
|
||||||
|
Attempts to create a valid 3D-glasses structure
|
||||||
|
|
||||||
|
LICENCE : CC
|
||||||
|
|
||||||
|
by cocoa
|
||||||
|
|
||||||
|
|
||||||
|
'''
|
||||||
|
from __future__ import print_function
|
||||||
|
import argparse
|
||||||
|
import ast
|
||||||
|
import math
|
||||||
|
import os
|
||||||
|
import random
|
||||||
|
import sys
|
||||||
|
import time
|
||||||
|
name = "filters::cycle"
|
||||||
|
|
||||||
|
maxDist = 300
|
||||||
|
|
||||||
|
argsparser = argparse.ArgumentParser(description="Redis exporter LJ")
|
||||||
|
argsparser.add_argument("-x","--centerX",help="geometrical center X position",default=400,type=int)
|
||||||
|
argsparser.add_argument("-y","--centerY",help="geometrical center Y position",default=400,type=int)
|
||||||
|
argsparser.add_argument("-m","--min",help="Minimal displacement (default:2) ",default=1,type=int)
|
||||||
|
argsparser.add_argument("-M","--max",help="Maximal displacement (default:20) ",default=5,type=int)
|
||||||
|
argsparser.add_argument("-f","--fps",help="Frame Per Second",default=30,type=int)
|
||||||
|
argsparser.add_argument("-v","--verbose",action="store_true",help="Verbose")
|
||||||
|
|
||||||
|
args = argsparser.parse_args()
|
||||||
|
fps = args.fps
|
||||||
|
minVal = args.min
|
||||||
|
maxVal = args.max
|
||||||
|
centerX = args.centerX
|
||||||
|
centerY = args.centerY
|
||||||
|
verbose = args.verbose
|
||||||
|
|
||||||
|
optimal_looptime = 1 / fps
|
||||||
|
name = "filters::anaglyph"
|
||||||
|
|
||||||
|
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 isValidColor( color, intensityColThreshold ):
|
||||||
|
if color[0] + color[1] + color[2] > intensityColThreshold:
|
||||||
|
return True
|
||||||
|
return False
|
||||||
|
|
||||||
|
# These are paper colors
|
||||||
|
red = (41,24,24)
|
||||||
|
white = (95,95,95)
|
||||||
|
blue = (0,41,64)
|
||||||
|
|
||||||
|
red = (86,0,0)
|
||||||
|
blue = (0,55,86)
|
||||||
|
white = (125,125,125)
|
||||||
|
def anaglyph( pl ):
|
||||||
|
|
||||||
|
debug(name,'--------------- new loop ------------------')
|
||||||
|
# We will send one list after the other to optimize color change
|
||||||
|
blueList = list()
|
||||||
|
redList = list()
|
||||||
|
whiteList = list()
|
||||||
|
out = []
|
||||||
|
out1 = []
|
||||||
|
out2 = []
|
||||||
|
out3 = []
|
||||||
|
|
||||||
|
# The anaglyphic effect will be optained by :
|
||||||
|
# * having close objects appear as white
|
||||||
|
# * having distant objects appear as blue + red
|
||||||
|
# * having in between objects appear as distanceDecreased(white) + blue + red
|
||||||
|
for i, point in enumerate(pl):
|
||||||
|
ref_x = point[0]-centerX
|
||||||
|
ref_y = point[1]-centerY
|
||||||
|
ref_color = point[2]
|
||||||
|
angle = math.atan2( ref_x , ref_y )
|
||||||
|
dist = ref_y / math.cos(angle)
|
||||||
|
white_rvb = (0,0,0)
|
||||||
|
blue_rvb = (0,0,0)
|
||||||
|
red_rvb = (0,0,0)
|
||||||
|
|
||||||
|
# Calculate the point's spread factor (0.0 to 1.0)
|
||||||
|
# The spread is high if the point is close to center
|
||||||
|
"""
|
||||||
|
dist = 0 : spread = 1.0
|
||||||
|
dist = maxDist spread = 0.0
|
||||||
|
"""
|
||||||
|
if dist == 0:
|
||||||
|
spread = 1.0
|
||||||
|
else :
|
||||||
|
spread =( maxDist - dist ) / maxDist
|
||||||
|
if spread < 0.0:
|
||||||
|
spread = 0.0
|
||||||
|
|
||||||
|
#debug(name,"dist:{} spread:{}".format(dist,spread))
|
||||||
|
|
||||||
|
# White color is high if spread is low, i.e. point away from center
|
||||||
|
"""
|
||||||
|
spread = 1.0 : white_c = 0.0
|
||||||
|
spread = 0.0 : whice_c = 1.0
|
||||||
|
"""
|
||||||
|
if point[2] == 0:
|
||||||
|
white_color = 0
|
||||||
|
else:
|
||||||
|
white_factor = 1.0 - math.pow(spread,0.5)
|
||||||
|
white_rvb = tuple(map( lambda a: int(white_factor* a), white))
|
||||||
|
white_color = rgb2int( white_rvb)
|
||||||
|
#debug(name,"spread:{}\t white_rvb:{}\t white_color:{}".format(spread, white_rvb, white_color))
|
||||||
|
|
||||||
|
# Blue and Red colors are high if spread is high, i.e. close to center
|
||||||
|
"""
|
||||||
|
spread = 1.0 : red_c = 1.0
|
||||||
|
spread = 0.0 : red_c = 0.0
|
||||||
|
"""
|
||||||
|
color_factor = math.pow(spread,1)
|
||||||
|
if point[2] == 0:
|
||||||
|
blue_color = 0
|
||||||
|
red_color = 0
|
||||||
|
else:
|
||||||
|
blue_rvb = tuple(map( lambda a: int(color_factor * a), blue))
|
||||||
|
blue_color = rgb2int( blue_rvb)
|
||||||
|
red_rvb = tuple(map( lambda a: int(color_factor * a), red))
|
||||||
|
red_color = rgb2int( red_rvb)
|
||||||
|
|
||||||
|
#debug(name,"color_factor:{}\t\t blue_color:{}\t\t red_color:{}".format(color_factor,blue_color,red_color))
|
||||||
|
|
||||||
|
# Blue-to-Red spatial spread is high when spread is high, i.e. point close to center
|
||||||
|
"""
|
||||||
|
spread = 1.0 : spatial_spread = maxVal
|
||||||
|
spread = 0.0 : spatial_spread = minVal
|
||||||
|
"""
|
||||||
|
spatial_spread = minVal + spread * (maxVal - minVal)
|
||||||
|
#debug(name,"spatial_spread:{}".format(spatial_spread))
|
||||||
|
red_x = int(point[0] + spatial_spread)
|
||||||
|
blue_x = int(point[0] - spatial_spread )
|
||||||
|
red_y = int(point[1] )
|
||||||
|
blue_y = int(point[1])
|
||||||
|
|
||||||
|
white_point = [point[0], point[1], white_color]
|
||||||
|
blue_point = [blue_x,blue_y,blue_color]
|
||||||
|
red_point = [red_x,red_y,red_color]
|
||||||
|
|
||||||
|
#debug(name,"white[x,y,c]:{}".format(white_point))
|
||||||
|
#debug(name,"blue[x,y,c]:{}".format(blue_point))
|
||||||
|
#debug(name,"red[x,y,c]:{}".format(red_point))
|
||||||
|
# Do not append "black lines" i.e. a color where each composent is below X
|
||||||
|
# if isValidColor(white_rvb, 150):
|
||||||
|
# out1.append(white_point)
|
||||||
|
# if isValidColor(blue_rvb, 50):
|
||||||
|
# out2.append(blue_point)
|
||||||
|
# if isValidColor(red_rvb, 30):
|
||||||
|
# out3.append(red_point)
|
||||||
|
out1.append(white_point)
|
||||||
|
out2.append(blue_point)
|
||||||
|
out3.append(red_point)
|
||||||
|
|
||||||
|
#debug(name,"source pl:{}".format(pl))
|
||||||
|
debug(name,"whiteList:{}".format(out1))
|
||||||
|
debug(name,"blueList:{}".format(out2))
|
||||||
|
debug(name,"redList:{}".format(out3))
|
||||||
|
return out3 + out2
|
||||||
|
return out1 + out3 + out2
|
||||||
|
#return out1 + out2 + out3
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
try:
|
||||||
|
while True:
|
||||||
|
start = time.time()
|
||||||
|
line = sys.stdin.readline()
|
||||||
|
if line == "":
|
||||||
|
time.sleep(0.01)
|
||||||
|
line = line.rstrip('\n')
|
||||||
|
pointsList = ast.literal_eval(line)
|
||||||
|
# Do the filter
|
||||||
|
result = anaglyph( pointsList )
|
||||||
|
print( result, flush=True )
|
||||||
|
looptime = time.time() - start
|
||||||
|
# debug(name+" looptime:"+str(looptime))
|
||||||
|
if( looptime < optimal_looptime ):
|
||||||
|
time.sleep( optimal_looptime - looptime)
|
||||||
|
# debug(name+" micro sleep:"+str( optimal_looptime - looptime))
|
||||||
|
except EOFError:
|
||||||
|
debug(name+" break")# no more information
|
||||||
|
|
@ -54,7 +54,6 @@ def kaleidoscope( pl ):
|
|||||||
# Iterate trough the segments
|
# Iterate trough the segments
|
||||||
for i in range( 0, len(pl) ):
|
for i in range( 0, len(pl) ):
|
||||||
|
|
||||||
|
|
||||||
#debug(name+" point #", i)
|
#debug(name+" point #", i)
|
||||||
currentpoint = cp = pl[i]
|
currentpoint = cp = pl[i]
|
||||||
cx,cy,cc = [cp[0],cp[1],cp[2]]
|
cx,cy,cc = [cp[0],cp[1],cp[2]]
|
||||||
@ -74,8 +73,8 @@ def kaleidoscope( pl ):
|
|||||||
#debug(name+" rect: ", rect,"curr",currentpoint,"next",nextpoint )
|
#debug(name+" rect: ", rect,"curr",currentpoint,"next",nextpoint )
|
||||||
|
|
||||||
# Enumerate the points in rectangle to see
|
# Enumerate the points in rectangle to see
|
||||||
# how many right / wrong there are to add or skip early
|
# how many right / wrong there are
|
||||||
#
|
# either to add or skip early
|
||||||
for iterator, p in enumerate(rect):
|
for iterator, p in enumerate(rect):
|
||||||
if p[0] >= centerX and p[1] >= centerY:
|
if p[0] >= centerX and p[1] >= centerY:
|
||||||
right += 1
|
right += 1
|
||||||
@ -118,7 +117,7 @@ def kaleidoscope( pl ):
|
|||||||
#print("on x axis, v=",str(v)," and absnewY=",str(absnewY))
|
#print("on x axis, v=",str(v)," and absnewY=",str(absnewY))
|
||||||
crossY = [( absnewY*v[0] + cy ),( absnewY*v[1]+cy ), nc]
|
crossY = [( absnewY*v[0] + cy ),( absnewY*v[1]+cy ), nc]
|
||||||
# Inject in order
|
# Inject in order
|
||||||
# If current is valid, Add
|
# If current point is the quadrant, add it
|
||||||
if cx >= centerX and cy >= centerY :
|
if cx >= centerX and cy >= centerY :
|
||||||
quad1.append( currentpoint )
|
quad1.append( currentpoint )
|
||||||
# If absnewX smaller, it is closest to currentPoint
|
# If absnewX smaller, it is closest to currentPoint
|
||||||
@ -128,6 +127,9 @@ def kaleidoscope( pl ):
|
|||||||
else :
|
else :
|
||||||
if None != crossY : quad1.append( crossY )
|
if None != crossY : quad1.append( crossY )
|
||||||
if None != crossX : quad1.append( crossX )
|
if None != crossX : quad1.append( crossX )
|
||||||
|
# Add a black point at the end
|
||||||
|
#lastQuad1Point = quad1[-1]
|
||||||
|
#quad1.append( [lastQuad1Point[0],lastQuad1Point[1],0] )
|
||||||
|
|
||||||
## Stage 2 : Mirror points
|
## Stage 2 : Mirror points
|
||||||
#
|
#
|
||||||
@ -144,10 +146,10 @@ def kaleidoscope( pl ):
|
|||||||
point = quad3[iterator]
|
point = quad3[iterator]
|
||||||
quad4.append([ 2*centerX - point[0], point[1], point[2] ])
|
quad4.append([ 2*centerX - point[0], point[1], point[2] ])
|
||||||
|
|
||||||
debug(name+" quad1:",quad1)
|
#debug(name+" quad1:",quad1)
|
||||||
#debug(name+" quad2:", quad2 )
|
#debug(name+" quad2:", quad2 )
|
||||||
debug(name+" quad3:", quad3 )
|
#debug(name+" quad3:", quad3 )
|
||||||
debug(name+" quad4:", quad4 )
|
#debug(name+" quad4:", quad4 )
|
||||||
return quad3+quad4
|
return quad3+quad4
|
||||||
|
|
||||||
try:
|
try:
|
||||||
|
@ -34,16 +34,18 @@ def debug(*args, **kwargs):
|
|||||||
if( verbose == False ):
|
if( verbose == False ):
|
||||||
return
|
return
|
||||||
print(*args, file=sys.stderr, **kwargs)
|
print(*args, file=sys.stderr, **kwargs)
|
||||||
def now():
|
def msNow():
|
||||||
return time.time() * 1000
|
return time.time()
|
||||||
|
|
||||||
# The list of available modes and the redis keys they need
|
# The list of available modes => redis keys each requires to run
|
||||||
oModeList = {
|
oModeList = {
|
||||||
"rms_noise": ["rms"],
|
"rms_noise": ["rms"],
|
||||||
"rms_size": ["rms"],
|
"rms_size": ["rms"],
|
||||||
"bpm_size": ["bpm"]
|
"bpm_size": ["bpm"],
|
||||||
|
"bpm_detect_size": ["bpm","bpm_delay","bpm_sample_interval","beats"]
|
||||||
}
|
}
|
||||||
CHAOS = 1
|
CHAOS = 1
|
||||||
|
REDISLATENCY = 30
|
||||||
REDIS_FREQ = 300
|
REDIS_FREQ = 300
|
||||||
|
|
||||||
# General Args
|
# General Args
|
||||||
@ -58,17 +60,19 @@ argsparser.add_argument("-x","--centerX",help="geometrical center X position",de
|
|||||||
argsparser.add_argument("-y","--centerY",help="geometrical center Y position",default=400,type=int)
|
argsparser.add_argument("-y","--centerY",help="geometrical center Y position",default=400,type=int)
|
||||||
argsparser.add_argument("-f","--fps",help="Frame Per Second",default=30,type=int)
|
argsparser.add_argument("-f","--fps",help="Frame Per Second",default=30,type=int)
|
||||||
# Modes And Common Modes Parameters
|
# Modes And Common Modes Parameters
|
||||||
|
argsparser.add_argument("-l","--redisLatency",help="Latency in ms to substract. Default:{}".format(REDISLATENCY),default=REDISLATENCY,type=float)
|
||||||
argsparser.add_argument("-m","--modelist",required=True,help="Comma separated list of modes to use from: {}".format("i, ".join(oModeList.keys())),type=str)
|
argsparser.add_argument("-m","--modelist",required=True,help="Comma separated list of modes to use from: {}".format("i, ".join(oModeList.keys())),type=str)
|
||||||
argsparser.add_argument("--chaos",help="How much disorder to bring. High value = More chaos. Default {}".format(CHAOS), default=CHAOS, type=str)
|
argsparser.add_argument("--chaos",help="How much disorder to bring. High value = More chaos. Default {}".format(CHAOS), default=CHAOS, type=str)
|
||||||
|
|
||||||
args = argsparser.parse_args()
|
args = argsparser.parse_args()
|
||||||
ip = args.ip
|
ip = args.ip
|
||||||
port = args.port
|
port = args.port
|
||||||
redisFreq = args.redis_freq
|
redisFreq = args.redis_freq / 1000
|
||||||
verbose = args.verbose
|
verbose = args.verbose
|
||||||
fps = args.fps
|
fps = args.fps
|
||||||
centerX = args.centerX
|
centerX = args.centerX
|
||||||
centerY = args.centerY
|
centerY = args.centerY
|
||||||
|
redisLatency = args.redisLatency
|
||||||
chaos = float(args.chaos)
|
chaos = float(args.chaos)
|
||||||
optimal_looptime = 1 / fps
|
optimal_looptime = 1 / fps
|
||||||
|
|
||||||
@ -82,33 +86,127 @@ for mode in modeList:
|
|||||||
redisKeys = list(set(redisKeys))
|
redisKeys = list(set(redisKeys))
|
||||||
debug(name,"Redis Keys:{}".format(redisKeys))
|
debug(name,"Redis Keys:{}".format(redisKeys))
|
||||||
redisData = {}
|
redisData = {}
|
||||||
redisLastHit = now() - redisFreq
|
redisLastHit = msNow() - 99999
|
||||||
r = redis.Redis(
|
r = redis.Redis(
|
||||||
host=ip,
|
host=ip,
|
||||||
port=port)
|
port=port)
|
||||||
|
|
||||||
# Records the last bpm
|
# Records the last bpm
|
||||||
last_bpm = time.time()
|
tsLastBeat = time.time()
|
||||||
|
|
||||||
def gauss(x, mu, sigma):
|
def gauss(x, mu, sigma):
|
||||||
return( math.exp(-math.pow((x-mu),2)/(2*math.pow(sigma,2))/math.sqrt(2*math.pi*math.pow(sigma,2))))
|
return( math.exp(-math.pow((x-mu),2)/(2*math.pow(sigma,2))/math.sqrt(2*math.pi*math.pow(sigma,2))))
|
||||||
|
|
||||||
|
previousPTTL = 0
|
||||||
|
tsNextBeatsList = []
|
||||||
|
def bpmDetect( ):
|
||||||
|
"""
|
||||||
|
An helper to compute the next beat time in milliseconds
|
||||||
|
Returns True if the cache was updated
|
||||||
|
"""
|
||||||
|
global tsNextBeatsList
|
||||||
|
global previousPTTL
|
||||||
|
global redisLastHit
|
||||||
|
global redisLatency
|
||||||
|
|
||||||
def bpm_size( pl ):
|
# Get the redis PTTL value for bpm
|
||||||
global last_bpm
|
PTTL = redisData["bpm_pttl"]
|
||||||
|
|
||||||
|
# Skip early if PTTL < 0
|
||||||
|
if PTTL < 0 :
|
||||||
|
debug(name,"bpmDetect skip detection : PTTL expired for 'bpm' key")
|
||||||
|
return False
|
||||||
|
|
||||||
|
# Skip early if the record hasn't been rewritten
|
||||||
|
if PTTL <= previousPTTL :
|
||||||
|
previousPTTL = PTTL
|
||||||
|
#debug(name,"bpmDetect skip detection : {} <= {}".format(PTTL, previousPTTL))
|
||||||
|
return False
|
||||||
|
debug(name,"bpmDetect running detection : {} > {}".format(PTTL, previousPTTL))
|
||||||
|
previousPTTL = PTTL
|
||||||
|
|
||||||
|
# Skip early if beat list is empty
|
||||||
|
beatsList = ast.literal_eval(redisData["beats"])
|
||||||
|
tsNextBeatsList = []
|
||||||
|
if( len(beatsList) == 0 ):
|
||||||
|
return True
|
||||||
|
|
||||||
|
# Read from redis
|
||||||
bpm = float(redisData["bpm"])
|
bpm = float(redisData["bpm"])
|
||||||
# Milliseconds ber beat
|
msBpmDelay = float(redisData["bpm_delay"])
|
||||||
milliSecondsPerBeat = int(60 / bpm * 1000)
|
samplingInterval = float(redisData["bpm_sample_interval"])
|
||||||
|
|
||||||
|
# Calculate some interpolations
|
||||||
|
lastBeatTiming = float(beatsList[len(beatsList) - 1])
|
||||||
|
msPTTLDelta = 2 * samplingInterval - float(PTTL)
|
||||||
|
sPerBeat = 60 / bpm
|
||||||
|
lastBeatDelay = msBpmDelay - lastBeatTiming*1000 + msPTTLDelta
|
||||||
|
countBeatsPast = math.floor( (lastBeatDelay / 1000) / sPerBeat)
|
||||||
|
#debug(name,"bpmDetect lastBeatTiming:{}\tmsPTTLDelta:{}\tsPerBeat:{}".format(lastBeatTiming,msPTTLDelta,sPerBeat))
|
||||||
|
#debug(name,"lastBeatDelay:{}\t countBeatsPast:{}".format(lastBeatDelay, countBeatsPast))
|
||||||
|
for i in range( countBeatsPast, 1000):
|
||||||
|
beatTime = i * sPerBeat - lastBeatTiming
|
||||||
|
if beatTime < 0:
|
||||||
|
continue
|
||||||
|
if beatTime * 1000 > 2 * samplingInterval :
|
||||||
|
break
|
||||||
|
#debug(name, "bpmDetect beat add beatTime:{} redisLastHit:{}".format(beatTime, redisLastHit))
|
||||||
|
tsNextBeatsList.append( redisLastHit + beatTime - redisLatency/1000)
|
||||||
|
debug(name, "bpmDetect new tsNextBeatsList:{}".format(tsNextBeatsList))
|
||||||
|
|
||||||
|
return True
|
||||||
|
|
||||||
|
def bpm_detect_size( pl ):
|
||||||
|
bpmDetect()
|
||||||
|
|
||||||
|
# Find the next beat in the list
|
||||||
|
tsNextBeat = 0
|
||||||
|
|
||||||
|
now = time.time()
|
||||||
|
msNearestBeat = None
|
||||||
|
msRelativeNextBTList = list(map( lambda a: abs(now - a) * 1000, tsNextBeatsList))
|
||||||
|
msToBeat = min( msRelativeNextBTList)
|
||||||
|
|
||||||
|
#debug(name,"bpm_detect_size msRelativeNextBTList:{} msToBeat:{}".format(msRelativeNextBTList,msToBeat))
|
||||||
# Calculate the intensity based on bpm coming/leaving
|
# Calculate the intensity based on bpm coming/leaving
|
||||||
# The curb is a gaussian
|
# The curb is a gaussian
|
||||||
mu = math.sqrt(milliSecondsPerBeat)
|
mu = 15
|
||||||
milliTimeToLastBeat = (time.time() - last_bpm) * 1000
|
intensity = gauss( msToBeat, 0 , mu)
|
||||||
milliTimeToNextBeat = (milliSecondsPerBeat - milliTimeToLastBeat)
|
#debug(name,"bpm_size","mu:{}\t msToBeat:{}\tintensity:{}".format(mu, msToBeat, intensity))
|
||||||
intensity = gauss( milliTimeToNextBeat, 0 , mu)
|
if msToBeat < 20:
|
||||||
debug(name,"bpm_size","milliSecondsPerBeat:{}\tmu:{}".format(milliSecondsPerBeat, mu))
|
debug(name,"bpm_detect_size kick:{}".format(msToBeat))
|
||||||
debug(name,"bpm_size","milliTimeToLastBeat:{}\tmilliTimeToNextBeat:{}\tintensity:{}".format(milliTimeToLastBeat, milliTimeToNextBeat, intensity))
|
pass
|
||||||
if milliTimeToNextBeat <= 0 :
|
for i, point in enumerate(pl):
|
||||||
last_bpm = time.time()
|
ref_x = point[0]-centerX
|
||||||
|
ref_y = point[1]-centerY
|
||||||
|
#debug(name,"In new ref x:{} y:{}".format(point[0]-centerX,point[1]-centerY))
|
||||||
|
angle=math.atan2( point[0] - centerX , point[1] - centerY )
|
||||||
|
l = ref_y / math.cos(angle)
|
||||||
|
new_l = l * intensity
|
||||||
|
#debug(name,"bpm_size","angle:{} l:{} new_l:{}".format(angle,l,new_l))
|
||||||
|
new_x = math.sin(angle) * new_l + centerX
|
||||||
|
new_y = math.cos(angle) * new_l + centerY
|
||||||
|
#debug(name,"x,y:({},{}) x',y':({},{})".format(point[0],point[1],new_x,new_y))
|
||||||
|
pl[i][0] = new_x
|
||||||
|
pl[i][1] = new_y
|
||||||
|
#debug( name,"bpm_detect_size output:{}".format(pl))
|
||||||
|
return( pl );
|
||||||
|
|
||||||
|
def bpm_size( pl ):
|
||||||
|
global tsLastBeat
|
||||||
|
bpm = float(redisData["bpm"])
|
||||||
|
# msseconds ber beat
|
||||||
|
msPerBeat = int(60 / bpm * 1000)
|
||||||
|
# Calculate the intensity based on bpm coming/leaving
|
||||||
|
# The curb is a gaussian
|
||||||
|
mu = math.sqrt(msPerBeat)
|
||||||
|
msTimeToLastBeat = (time.time() - tsLastBeat) * 1000
|
||||||
|
msTimeToNextBeat = (msPerBeat - msTimeToLastBeat)
|
||||||
|
intensity = gauss( msTimeToNextBeat, 0 , mu)
|
||||||
|
debug(name,"bpm_size","msPerBeat:{}\tmu:{}".format(msPerBeat, mu))
|
||||||
|
debug(name,"bpm_size","msTimeToLastBeat:{}\tmsTimeToNextBeat:{}\tintensity:{}".format(msTimeToLastBeat, msTimeToNextBeat, intensity))
|
||||||
|
if msTimeToNextBeat <= 0 :
|
||||||
|
tsLastBeat = time.time()
|
||||||
for i, point in enumerate(pl):
|
for i, point in enumerate(pl):
|
||||||
ref_x = point[0]-centerX
|
ref_x = point[0]-centerX
|
||||||
ref_y = point[1]-centerY
|
ref_y = point[1]-centerY
|
||||||
@ -158,21 +256,30 @@ def rms_noise( pl ):
|
|||||||
return pl
|
return pl
|
||||||
|
|
||||||
|
|
||||||
def updateRedis():
|
def refreshRedis():
|
||||||
global redisLastHit
|
global redisLastHit
|
||||||
global redisData
|
global redisData
|
||||||
|
# Skip if cache is sufficent
|
||||||
|
diff = msNow() - redisLastHit
|
||||||
|
if diff < redisFreq :
|
||||||
|
#debug(name, "refreshRedis not updating redis, {} < {}".format(diff, redisFreq))
|
||||||
|
pass
|
||||||
|
else:
|
||||||
|
#debug(name, "refreshRedis updating redis, {} > {}".format(diff, redisFreq))
|
||||||
|
redisLastHit = msNow()
|
||||||
for key in redisKeys:
|
for key in redisKeys:
|
||||||
redisData[key] = r.get(key).decode('ascii')
|
redisData[key] = r.get(key).decode('ascii')
|
||||||
debug("name","updateRedis key:{} value:{}".format(key,redisData[key]))
|
#debug(name,"refreshRedis key:{} value:{}".format(key,redisData[key]))
|
||||||
if key == 'bpm':
|
# Only update the TTLs
|
||||||
redisData['bpm_ttl'] = r.pttl(key)
|
if 'bpm' in redisKeys:
|
||||||
debug(name,"redisData:{}".format(redisData))
|
redisData['bpm_pttl'] = r.pttl('bpm')
|
||||||
|
#debug(name,"refreshRedis key:bpm_ttl value:{}".format(redisData["bpm_pttl"]))
|
||||||
|
#debug(name,"redisData:{}".format(redisData))
|
||||||
|
return True
|
||||||
|
|
||||||
try:
|
try:
|
||||||
while True:
|
while True:
|
||||||
# it is time to query redis
|
refreshRedis()
|
||||||
if now() - redisLastHit > redisFreq:
|
|
||||||
updateRedis()
|
|
||||||
start = time.time()
|
start = time.time()
|
||||||
line = sys.stdin.readline()
|
line = sys.stdin.readline()
|
||||||
if line == "":
|
if line == "":
|
||||||
|
@ -34,7 +34,7 @@ argsparser = argparse.ArgumentParser(description="tunnel generator")
|
|||||||
argsparser.add_argument("-c","--color",help="Color",default=65280,type=int)
|
argsparser.add_argument("-c","--color",help="Color",default=65280,type=int)
|
||||||
argsparser.add_argument("-f","--fps",help="Frame Per Second",default=30,type=int)
|
argsparser.add_argument("-f","--fps",help="Frame Per Second",default=30,type=int)
|
||||||
argsparser.add_argument("-i","--interval",help="point per shape interval",default=30,type=int)
|
argsparser.add_argument("-i","--interval",help="point per shape interval",default=30,type=int)
|
||||||
argsparser.add_argument("-m","--max-size",help="maximum size for objects",default=500,type=int)
|
argsparser.add_argument("-m","--max-size",help="maximum size for objects",default=400,type=int)
|
||||||
argsparser.add_argument("-r","--randomize",help="center randomization",default=5,type=int)
|
argsparser.add_argument("-r","--randomize",help="center randomization",default=5,type=int)
|
||||||
argsparser.add_argument("-s","--speed",help="point per frame progress",default=3,type=int)
|
argsparser.add_argument("-s","--speed",help="point per frame progress",default=3,type=int)
|
||||||
argsparser.add_argument("-v","--verbose",action="store_true",help="Verbose output")
|
argsparser.add_argument("-v","--verbose",action="store_true",help="Verbose output")
|
||||||
@ -77,14 +77,19 @@ class polylineGenerator( object ):
|
|||||||
self.polylineList = [[0,[currentCenter[0],currentCenter[1]]]]
|
self.polylineList = [[0,[currentCenter[0],currentCenter[1]]]]
|
||||||
self.buf = []
|
self.buf = []
|
||||||
|
|
||||||
|
def init(self):
|
||||||
|
finished = False
|
||||||
|
while not finished:
|
||||||
|
finished = self.increment()
|
||||||
|
debug(name,"init done:{}".format(self.polylineList))
|
||||||
def draw( self ):
|
def draw( self ):
|
||||||
self.buf = []
|
self.buf = []
|
||||||
for it_pl, infoList in enumerate(self.polylineList):
|
for it_pl, infoList in enumerate(self.polylineList):
|
||||||
size = infoList[0]
|
size = infoList[0]
|
||||||
center = infoList[1]
|
center = infoList[1]
|
||||||
for it_sqr, point in enumerate(shape):
|
for it_sqr, point in enumerate(shape):
|
||||||
x = center[0] + point[0]*size
|
x = int( center[0] + point[0]*size )
|
||||||
y = center[1] + point[1]*size
|
y = int( center[1] + point[1]*size )
|
||||||
# Add an invisible point in first location
|
# Add an invisible point in first location
|
||||||
if 0 == it_sqr:
|
if 0 == it_sqr:
|
||||||
self.buf.append([x,y,0])
|
self.buf.append([x,y,0])
|
||||||
@ -114,22 +119,43 @@ class polylineGenerator( object ):
|
|||||||
speed = origSpeed
|
speed = origSpeed
|
||||||
elif speed > (origSpeed + randomize / 2) :
|
elif speed > (origSpeed + randomize / 2) :
|
||||||
speed = origSpeed + randomize / 2
|
speed = origSpeed + randomize / 2
|
||||||
debug(name, "currentCenter:{} speed:{}".format(currentCenter,speed))
|
#debug(name, "currentCenter:{} speed:{}".format(currentCenter,speed))
|
||||||
|
|
||||||
for i, shapeInfo in enumerate(self.polylineList):
|
for i, shapeInfo in enumerate(self.polylineList):
|
||||||
size = shapeInfo[0]
|
size = shapeInfo[0]
|
||||||
size += speed
|
# Augment speed with size
|
||||||
|
"""
|
||||||
|
size = 0 : += sqrt(speed)
|
||||||
|
size = half max size : +=speed
|
||||||
|
|
||||||
|
"""
|
||||||
|
if size < max_size / 4:
|
||||||
|
size += math.pow(speed, 0.1)
|
||||||
|
elif size < max_size / 3:
|
||||||
|
size += math.pow(speed, 0.25)
|
||||||
|
elif size < max_size / 2:
|
||||||
|
size += math.pow(speed, 0.5)
|
||||||
|
else:
|
||||||
|
size += math.pow(speed, 1.25)
|
||||||
if size < min_size : min_size = size
|
if size < min_size : min_size = size
|
||||||
if size > max_size : delList.append(i)
|
if size > max_size : delList.append(i)
|
||||||
self.polylineList[i][0] = size
|
self.polylineList[i][0] = size
|
||||||
for i in delList:
|
for i in delList:
|
||||||
del self.polylineList[i]
|
del self.polylineList[i]
|
||||||
if min_size >= interval: self.polylineList.append([0,[currentCenter[0],currentCenter[1]]])
|
|
||||||
#debug(name, "polyline:",self.polylineList)
|
#debug(name, "polyline:",self.polylineList)
|
||||||
|
if min_size >= interval:
|
||||||
|
debug(name, "new shape")
|
||||||
|
self.polylineList.append([0,[currentCenter[0],currentCenter[1]]])
|
||||||
|
|
||||||
|
# Return True if we delete a shape
|
||||||
|
|
||||||
|
if len(delList):
|
||||||
|
return True
|
||||||
|
return False
|
||||||
|
|
||||||
|
|
||||||
pgen = polylineGenerator()
|
pgen = polylineGenerator()
|
||||||
|
pgen.init()
|
||||||
|
|
||||||
while True:
|
while True:
|
||||||
start = time.time()
|
start = time.time()
|
||||||
@ -140,7 +166,7 @@ while True:
|
|||||||
# send
|
# send
|
||||||
pl = pgen.draw()
|
pl = pgen.draw()
|
||||||
print(pl, flush=True)
|
print(pl, flush=True)
|
||||||
debug(name,"output:{}".format(pl))
|
#debug(name,"output:{}".format(pl))
|
||||||
|
|
||||||
looptime = time.time() - start
|
looptime = time.time() - start
|
||||||
if( looptime < optimal_looptime ):
|
if( looptime < optimal_looptime ):
|
||||||
|
@ -517,66 +517,35 @@
|
|||||||
var zoom = 0.5;
|
var zoom = 0.5;
|
||||||
//ctx.save
|
//ctx.save
|
||||||
|
|
||||||
|
// Draws every segment received, except black colored target ones
|
||||||
// Todo : laser point will have black points to go from a polyline to another. Need to discard those black points.
|
|
||||||
function draw() {
|
function draw() {
|
||||||
|
|
||||||
|
|
||||||
// Clear Canvas At The Start Of Every Frame
|
|
||||||
//ctx.restore
|
|
||||||
|
|
||||||
if (pl2.length > 0)
|
if (pl2.length > 0)
|
||||||
{
|
{
|
||||||
|
|
||||||
// Begin a new path
|
|
||||||
// 0.7 reduces max coordinates in a more browser compatible resolution.
|
|
||||||
ctx.clearRect(0,0,400,400);
|
ctx.clearRect(0,0,400,400);
|
||||||
ctx.beginPath();
|
lastpoint = {
|
||||||
|
x:pl2[0],
|
||||||
ctx.moveTo(pl2[0]*zoom, pl2[1]*zoom);
|
y:pl2[1],
|
||||||
lastpoint.color = pl2[2];
|
color:pl2[2]
|
||||||
|
|
||||||
// Draw n Lines
|
|
||||||
for (var i = 0; i < pl2.length/3; i++)
|
|
||||||
{
|
|
||||||
|
|
||||||
// New point has the same color -> add a new line to the new point
|
|
||||||
if (pl2[2+(i*3)] === lastpoint.color)
|
|
||||||
{
|
|
||||||
ctx.lineTo(pl2[i*3]*zoom, pl2[1+(i*3)]*zoom);
|
|
||||||
}
|
}
|
||||||
|
for (var i = 0; i <= pl2.length; i+=3)
|
||||||
// New point has different color -> stroke with previous color
|
|
||||||
if (pl2[2+(i*3)] != lastpoint.color)
|
|
||||||
{
|
{
|
||||||
ctx.strokeStyle = "#"+(lastpoint.color + Math.pow(16, 6)).toString(16).slice(-6);
|
point = {
|
||||||
|
x:pl2[i],
|
||||||
|
y:pl2[i+1],
|
||||||
|
color:pl2[i+2]
|
||||||
|
}
|
||||||
|
// console.log(lastpoint,point)
|
||||||
|
// if the target is black, skip drawing
|
||||||
|
if( point.color != 0){
|
||||||
|
ctx.beginPath()
|
||||||
|
ctx.strokeStyle = "#"+(point.color + Math.pow(16, 6)).toString(16).slice(-6);
|
||||||
|
ctx.moveTo(lastpoint.x * zoom, lastpoint.y * zoom);
|
||||||
|
ctx.lineTo(point.x * zoom, point.y * zoom);
|
||||||
ctx.stroke();
|
ctx.stroke();
|
||||||
ctx.closePath()
|
ctx.closePath()
|
||||||
//ctx.restore
|
|
||||||
ctx.beginPath();
|
|
||||||
//ctx.clearRect(0,0,400,400);
|
|
||||||
|
|
||||||
ctx.moveTo(pl2[i*3]*zoom, pl2[1+(i*3)]*zoom);
|
|
||||||
}
|
}
|
||||||
|
lastpoint = point
|
||||||
// Last point -> stroke with current color
|
|
||||||
if (i === (pl2.length/3)-1 )
|
|
||||||
{
|
|
||||||
ctx.moveTo(pl2[i*3]*zoom, pl2[1+(i*3)]*zoom);
|
|
||||||
ctx.strokeStyle = "#"+((pl2[2+(i*3)]) + Math.pow(16, 6)).toString(16).slice(-6);
|
|
||||||
ctx.stroke();
|
|
||||||
|
|
||||||
ctx.closePath()
|
|
||||||
//ctx.restore
|
|
||||||
//ctx.clearRect(0,0,400,400);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// store point for comparison
|
|
||||||
lastpoint.x = pl2[i*3];
|
|
||||||
lastpoint.y = pl2[1+(i*3)];
|
|
||||||
lastpoint.color = pl2[2+(i*3)];
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
}
|
||||||
// Call Draw Function Again To Create Animation
|
// Call Draw Function Again To Create Animation
|
||||||
window.requestAnimationFrame(draw);
|
window.requestAnimationFrame(draw);
|
||||||
|
Loading…
Reference in New Issue
Block a user