|
@@ -3,7 +3,7 @@
|
|
|
import time
|
|
|
import pygame
|
|
|
|
|
|
-from numpy import fromstring,ceil,abs,log10,isnan,isinf,int16
|
|
|
+from numpy import fromstring,ceil,abs,log10,isnan,isinf,int16,sqrt,mean
|
|
|
from numpy import fft as Fft
|
|
|
|
|
|
import gobject
|
|
@@ -11,12 +11,14 @@ import gobject
|
|
|
from beeprint import pp
|
|
|
|
|
|
## local UI import
|
|
|
-from UI.constants import Width,Height
|
|
|
+from UI.constants import Width,Height,ICON_TYPES
|
|
|
from UI.page import Page,PageSelector
|
|
|
from UI.label import Label
|
|
|
from UI.fonts import fonts
|
|
|
from UI.util_funcs import midRect
|
|
|
from UI.keys_def import CurKeys
|
|
|
+from UI.icon_item import IconItem
|
|
|
+from UI.icon_pool import MyIconPool
|
|
|
|
|
|
from Queue import Queue, Empty
|
|
|
from threading import Thread
|
|
@@ -37,6 +39,9 @@ class PIFI(object):
|
|
|
|
|
|
count = 0
|
|
|
average = 0
|
|
|
+
|
|
|
+ rmscount=0
|
|
|
+ rmsaverage=0
|
|
|
|
|
|
def __init__(self):
|
|
|
self.sampleSize = self._SAMPLE_SIZE
|
|
@@ -54,7 +59,14 @@ class PIFI(object):
|
|
|
def resetSmoothing(self):
|
|
|
self.count = 0
|
|
|
self.average = 0
|
|
|
-
|
|
|
+ self.rmscount = 0
|
|
|
+ self.rmsaverage = 0
|
|
|
+
|
|
|
+ def rms_smoothOut(self, x):
|
|
|
+ self.rmscount += 1
|
|
|
+ self.rmsaverage = (self.rmsaverage*self.rmscount + x) / (self.rmscount+1)
|
|
|
+ return self.rmsaverage
|
|
|
+
|
|
|
def smoothOut(self, x):
|
|
|
self.count += 1
|
|
|
self.average = (self.average*self.count + x) / (self.count+1)
|
|
@@ -75,7 +87,8 @@ class PIFI(object):
|
|
|
# Compute the scaled list of values
|
|
|
scaledList = [int(x*scaleFactor) for x in _list ]
|
|
|
return scaledList
|
|
|
-
|
|
|
+
|
|
|
+
|
|
|
def computeSpectrum(self, fifoFile):
|
|
|
|
|
|
# Read PCM samples from fifo
|
|
@@ -91,6 +104,17 @@ class PIFI(object):
|
|
|
|
|
|
# Normalize [-1; +1]
|
|
|
pcm = pcm / (2.**15)
|
|
|
+
|
|
|
+ # Compute RMS directly from signal
|
|
|
+ rms = sqrt(mean(pcm**2))
|
|
|
+ # Compute a simple 'moving maximum'
|
|
|
+ maximum = 2*self.rms_smoothOut(rms)
|
|
|
+ if maximum == 0:
|
|
|
+ scaleFactor = 0.0
|
|
|
+ else:
|
|
|
+ scaleFactor = self._SCALE_WIDTH/float(maximum)
|
|
|
+
|
|
|
+ final_rms = int(rms*scaleFactor)
|
|
|
|
|
|
# Compute FFT
|
|
|
N = pcm.size
|
|
@@ -127,8 +151,8 @@ class PIFI(object):
|
|
|
|
|
|
# Scale the spectrum
|
|
|
scaledSpectrum = self.scaleList(spectrum)
|
|
|
-
|
|
|
- return (self.bins, scaledSpectrum)
|
|
|
+ scaledSpectrum.append( final_rms)
|
|
|
+ return scaledSpectrum
|
|
|
|
|
|
|
|
|
class MPDSpectrumPage(Page):
|
|
@@ -140,11 +164,28 @@ class MPDSpectrumPage(Page):
|
|
|
_ListFont = fonts["veramono12"]
|
|
|
|
|
|
_PIFI = None
|
|
|
- _FiFo = None
|
|
|
+ _FIFO = None
|
|
|
_Color = pygame.Color(126,206,244)
|
|
|
_GobjectIntervalId = -1
|
|
|
_Queue = None
|
|
|
_KeepReading = True
|
|
|
+
|
|
|
+ _BGpng = None
|
|
|
+ _BGwidth = 320
|
|
|
+ _BGheight = 200
|
|
|
+
|
|
|
+ _SheepHead = None
|
|
|
+ _SheepHeadW = 69
|
|
|
+ _SheepHeadH = 66
|
|
|
+
|
|
|
+ _SheepBody = None
|
|
|
+ _SheepBodyW = 105
|
|
|
+ _SheepBodyH = 81
|
|
|
+
|
|
|
+ _freq_count = 0
|
|
|
+ _head_dir = 0
|
|
|
+
|
|
|
+ _Neighbor = None
|
|
|
|
|
|
def __init__(self):
|
|
|
Page.__init__(self)
|
|
@@ -160,8 +201,27 @@ class MPDSpectrumPage(Page):
|
|
|
|
|
|
self._CanvasHWND = self._Screen._CanvasHWND
|
|
|
|
|
|
- self.Start()
|
|
|
|
|
|
+ self._BGpng = IconItem()
|
|
|
+ self._BGpng._ImgSurf = MyIconPool._Icons["sheep_bg"]
|
|
|
+ self._BGpng._MyType = ICON_TYPES["STAT"]
|
|
|
+ self._BGpng._Parent = self
|
|
|
+ self._BGpng.Adjust(0,0,self._BGwidth,self._BGheight,0)
|
|
|
+
|
|
|
+ self._SheepHead = IconItem()
|
|
|
+ self._SheepHead._ImgSurf = MyIconPool._Icons["sheep_head"]
|
|
|
+ self._SheepHead._MyType = ICON_TYPES["STAT"]
|
|
|
+ self._SheepHead._Parent = self
|
|
|
+ self._SheepHead.Adjust(0,0,self._SheepHeadW,self._SheepHeadH,0)
|
|
|
+
|
|
|
+ self._SheepBody = IconItem()
|
|
|
+ self._SheepBody._ImgSurf = MyIconPool._Icons["sheep_body"]
|
|
|
+ self._SheepBody._MyType = ICON_TYPES["STAT"]
|
|
|
+ self._SheepBody._Parent = self
|
|
|
+ self._SheepBody.Adjust(0,0,self._SheepBodyW,self._SheepBodyH,0)
|
|
|
+
|
|
|
+
|
|
|
+ self.Start()
|
|
|
self._GobjectIntervalId = gobject.timeout_add(50,self.Playing)
|
|
|
|
|
|
def Start(self):
|
|
@@ -186,7 +246,7 @@ class MPDSpectrumPage(Page):
|
|
|
print("self._FIFO none")
|
|
|
return
|
|
|
|
|
|
- (bins,scaledSpectrum) = self._PIFI.computeSpectrum(self._FIFO)
|
|
|
+ scaledSpectrum = self._PIFI.computeSpectrum(self._FIFO)
|
|
|
self._Queue.put( scaledSpectrum )
|
|
|
|
|
|
self._KeepReading = False
|
|
@@ -212,7 +272,9 @@ class MPDSpectrumPage(Page):
|
|
|
return True
|
|
|
|
|
|
def OnLoadCb(self):
|
|
|
-
|
|
|
+ if self._FIFO == None:
|
|
|
+ self.Start()
|
|
|
+
|
|
|
if self._Queue != None:
|
|
|
with self._Queue.mutex:
|
|
|
self._Queue.queue.clear()
|
|
@@ -228,6 +290,14 @@ class MPDSpectrumPage(Page):
|
|
|
|
|
|
def KeyDown(self,event):
|
|
|
if event.key == CurKeys["Menu"] or event.key == CurKeys["A"]:
|
|
|
+ if self._FIFO != None and self._FIFO.closed == False:
|
|
|
+ try:
|
|
|
+ self._FIFO.close()
|
|
|
+ self._FIFO = None
|
|
|
+ except Exception, e:
|
|
|
+ print(e)
|
|
|
+
|
|
|
+
|
|
|
self.ReturnToUpLevelPage()
|
|
|
self._Screen.Draw()
|
|
|
self._Screen.SwapAndShow()
|
|
@@ -239,22 +309,140 @@ class MPDSpectrumPage(Page):
|
|
|
if event.key == CurKeys["Enter"]:
|
|
|
pass
|
|
|
|
|
|
+ def Draw(self):
|
|
|
+ self.ClearCanvas()
|
|
|
+ self._BGpng.NewCoord(self._Width/2,self._Height/2)
|
|
|
+ self._BGpng.Draw()
|
|
|
+
|
|
|
+# print(self._Neighbor._CurSongTime)
|
|
|
+
|
|
|
+ phrase1 = False
|
|
|
+ phrase2 = False
|
|
|
|
|
|
+ parts = self._Neighbor._CurSongTime.split(":")
|
|
|
+ if len(parts) > 1:
|
|
|
+ cur = float(parts[0])
|
|
|
+ end = float(parts[1])
|
|
|
+ pros = int((cur/end)*100.0)
|
|
|
+
|
|
|
+ if pros > 30 and pros < 55:
|
|
|
+ phrase1 = True
|
|
|
+ if pros > 55 and pros < 100:
|
|
|
+ phrase2 = True
|
|
|
+
|
|
|
+
|
|
|
+ try:
|
|
|
+ spects = self._Queue.get_nowait() ## last element is rms
|
|
|
+ #print("get_nowait: " , spects)
|
|
|
+ except Empty:
|
|
|
+ return
|
|
|
+ else: # got line
|
|
|
+ if len(spects) == 0:
|
|
|
+ return
|
|
|
+
|
|
|
+ rms = spects[-1]
|
|
|
+ ratio = float(rms)/float(self._Height)
|
|
|
+ # 139,62
|
|
|
+ dx = 0
|
|
|
+ if ratio < 0.5:
|
|
|
+ dx = 5
|
|
|
+ else:
|
|
|
+ dx = 16
|
|
|
+ self._freq_count+=1 ## like frames
|
|
|
+
|
|
|
+ sheepbody_xy = (181,92)
|
|
|
+ sheephead_xy = (139,62)
|
|
|
+ sheepeye_xy = (129,60) ## eye2 ==> 129+20
|
|
|
+
|
|
|
+ eye_dx = sheephead_xy[0] - sheepeye_xy[0]
|
|
|
+ eye_dy = sheephead_xy[1] - sheepeye_xy[1]
|
|
|
+
|
|
|
+ self._SheepBody.NewCoord(sheepbody_xy[0],sheepbody_xy[1])
|
|
|
+ leg_dx = 2
|
|
|
+ if self._freq_count % 2 == 0:
|
|
|
+ self._SheepHead.NewCoord(sheephead_xy[0]+dx, sheephead_xy[1])
|
|
|
+
|
|
|
+ elif self._freq_count % 3 == 0:
|
|
|
+ self._SheepHead.NewCoord(sheephead_xy[0]-dx, sheephead_xy[1])
|
|
|
+ elif self._freq_count % 4 == 0:
|
|
|
+ self._SheepHead.NewCoord(sheephead_xy[0],sheephead_xy[1]+dx*2)
|
|
|
+
|
|
|
+ elif self._freq_count % 5 == 0:
|
|
|
+ self._SheepBody.NewCoord(sheepbody_xy[0],sheepbody_xy[1]-dx*3)
|
|
|
+ self._SheepHead.NewCoord(sheephead_xy[0],sheephead_xy[1]-dx*3)
|
|
|
+
|
|
|
+ elif self._freq_count % 7 == 0:
|
|
|
+ leg_dx = 0
|
|
|
+ self._SheepBody.NewCoord(sheepbody_xy[0],sheepbody_xy[1])
|
|
|
+
|
|
|
+
|
|
|
+ self._SheepBody.Draw()
|
|
|
+ self._SheepHead.Draw()
|
|
|
+
|
|
|
+ ## eyes
|
|
|
+ pygame.draw.circle(self._CanvasHWND,(255,255,255),(self._SheepHead._PosX-eye_dx,self._SheepHead._PosY-eye_dy),8,0)
|
|
|
+ pygame.draw.circle(self._CanvasHWND,(0,0,0),(self._SheepHead._PosX-eye_dx,self._SheepHead._PosY-eye_dy),8,2)
|
|
|
+
|
|
|
+ pygame.draw.circle(self._CanvasHWND,(0,0,0),(self._SheepHead._PosX-eye_dx,self._SheepHead._PosY-eye_dy),2,0)
|
|
|
+
|
|
|
+ pygame.draw.circle(self._CanvasHWND,(255,255,255),(self._SheepHead._PosX-eye_dx+20,self._SheepHead._PosY-eye_dy),8,0)
|
|
|
+ pygame.draw.circle(self._CanvasHWND,(0,0,0),(self._SheepHead._PosX-eye_dx+20,self._SheepHead._PosY-eye_dy),8,2)
|
|
|
+
|
|
|
+ pygame.draw.circle(self._CanvasHWND,(0,0,0),(self._SheepHead._PosX-eye_dx+20,self._SheepHead._PosY-eye_dy),2,0)
|
|
|
+
|
|
|
+ ##legs
|
|
|
+ legs1_xy = (self._SheepBody._PosX+13,self._SheepBody._PosY+38)
|
|
|
+ legs1_xy_end = (legs1_xy[0],legs1_xy[1]+16)
|
|
|
+
|
|
|
+ legs1_1_xy = (legs1_xy_end[0]-leg_dx,legs1_xy_end[1])
|
|
|
+ legs1_1_xy_end = (legs1_1_xy[0],legs1_1_xy[1]+14)
|
|
|
+
|
|
|
+ legs2_xy = (self._SheepBody._PosX+23,self._SheepBody._PosY+36)
|
|
|
+ legs2_xy_end = (legs2_xy[0],legs2_xy[1]+16)
|
|
|
+
|
|
|
+
|
|
|
+ legs3_xy = (self._SheepBody._PosX-23,self._SheepBody._PosY+38)
|
|
|
+ legs3_xy_end = (legs3_xy[0],legs3_xy[1]+16)
|
|
|
+
|
|
|
+ legs4_xy = (self._SheepBody._PosX-33,self._SheepBody._PosY+26)
|
|
|
+ legs4_xy_end = (legs4_xy[0],legs4_xy[1]+22)
|
|
|
+
|
|
|
+ legs4_1_xy = (legs4_xy_end[0]-leg_dx,legs4_xy_end[1])
|
|
|
+ legs4_1_xy_end = (legs4_1_xy[0],legs4_1_xy[1]+14)
|
|
|
+
|
|
|
+
|
|
|
+ pygame.draw.line(self._CanvasHWND,(0,0,0), legs1_xy,legs1_xy_end,4)
|
|
|
+ pygame.draw.line(self._CanvasHWND,(0,0,0), legs2_xy,legs2_xy_end,4)
|
|
|
+
|
|
|
+ pygame.draw.line(self._CanvasHWND,(0,0,0), legs3_xy,legs3_xy_end,4)
|
|
|
+ pygame.draw.line(self._CanvasHWND,(0,0,0), legs4_xy,legs4_xy_end,4)
|
|
|
+
|
|
|
+ pygame.draw.line(self._CanvasHWND,(0,0,0), legs1_1_xy,legs1_1_xy_end,4)
|
|
|
+
|
|
|
+ pygame.draw.line(self._CanvasHWND,(0,0,0), legs4_1_xy,legs4_1_xy_end,4)
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+"""
|
|
|
def Draw(self):
|
|
|
self.ClearCanvas()
|
|
|
|
|
|
bw = 10
|
|
|
spects = None
|
|
|
try:
|
|
|
- spects = self._Queue.get_nowait()
|
|
|
+ spects = self._Queue.get_nowait() ## last element is rms
|
|
|
# print("get_nowait: " , spects)
|
|
|
except Empty:
|
|
|
return
|
|
|
else: # got line
|
|
|
if len(spects) == 0:
|
|
|
return
|
|
|
- w = self._Width / len(spects)
|
|
|
+ w = self._Width / len( spects[0:-1] )
|
|
|
left_margin = (w-bw)/2
|
|
|
- for i,v in enumerate(spects):
|
|
|
+ for i,v in enumerate(spects[0:-1]):
|
|
|
pygame.draw.rect(self._CanvasHWND,self._Color,(i*w+left_margin,self._Height-v,bw,v),0)
|
|
|
|
|
|
+"""
|
|
|
+
|
|
|
+
|