mpd_spectrum_page.py 12 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416
  1. # -*- coding: utf-8 -*-
  2. import os
  3. import time
  4. import pygame
  5. import numpy
  6. import math
  7. import gobject
  8. from beeprint import pp
  9. ## local UI import
  10. from UI.constants import Width,Height,ICON_TYPES
  11. from UI.page import Page,PageSelector
  12. from UI.label import Label
  13. from UI.fonts import fonts
  14. from UI.util_funcs import midRect
  15. from UI.keys_def import CurKeys
  16. from UI.icon_item import IconItem
  17. from UI.icon_pool import MyIconPool
  18. from threading import Thread
  19. from list_item import ListItem
  20. import myvars
  21. class PIFI(object):
  22. _MPD_FIFO = "/tmp/mpd.fifo"
  23. _SAMPLE_SIZE = 1024
  24. _SAMPLING_RATE = 44100
  25. _FIRST_SELECTED_BIN = 5
  26. _NUMBER_OF_SELECTED_BINS = 1024
  27. def __init__(self):
  28. self.sampleSize = self._SAMPLE_SIZE
  29. self.samplingRate = self._SAMPLING_RATE
  30. def GetSpectrum(self,fifoFile,trim_by=10,log_scale=False,div_by=100):
  31. try:
  32. rawSamples = os.read(fifoFile,self.sampleSize) # will return empty lines (non-blocking)
  33. if len(rawSamples) < 1:
  34. # print("Read error")
  35. return rawSamples
  36. except Exception,e:
  37. return ""
  38. data = numpy.fromstring(rawSamples, dtype=numpy.int16)
  39. data = data * numpy.hanning(len(data))
  40. left,right = numpy.split(numpy.abs(numpy.fft.fft(data)),2)
  41. spec_y = numpy.add(left,right[::-1])
  42. if log_scale:
  43. spec_y=numpy.multiply(20,numpy.log10(spec_y))
  44. if trim_by:
  45. i=int((self.sampleSize/2)/trim_by)
  46. spec_y=spec_y[:i]
  47. if div_by:
  48. spec_y=spec_y/float(div_by)
  49. return spec_y
  50. class MPDSpectrumPage(Page):
  51. _Icons = {}
  52. _Selector=None
  53. _FootMsg = ["Nav","","","Back",""]
  54. _MyList = []
  55. _ListFont = fonts["veramono12"]
  56. _SongFont = fonts["notosanscjk12"]
  57. _PIFI = None
  58. _FIFO = None
  59. _Color = pygame.Color(126,206,244)
  60. _GobjectIntervalId = -1
  61. _Queue = None
  62. _KeepReading = True
  63. _ReadingThread = None
  64. _BGpng = None
  65. _BGwidth = 320
  66. _BGheight = 200
  67. _SheepHead = None
  68. _SheepHeadW = 69
  69. _SheepHeadH = 66
  70. _SheepBody = None
  71. _SheepBodyW = 105
  72. _SheepBodyH = 81
  73. _RollCanvas = None
  74. _RollW = 180
  75. _RollH = 18
  76. _freq_count = 0
  77. _head_dir = 0
  78. _Neighbor = None
  79. _bby = []
  80. _bbs = []
  81. _capYPositionArray = []
  82. _frames = 0
  83. read_retry = 0
  84. _queue_data = []
  85. _vis_values = []
  86. def __init__(self):
  87. Page.__init__(self)
  88. self._Icons = {}
  89. self._CanvasHWND = None
  90. self._MyList = []
  91. self._PIFI = PIFI()
  92. def Init(self):
  93. self._PosX = self._Index * self._Screen._Width
  94. self._Width = self._Screen._Width
  95. self._Height = self._Screen._Height
  96. self._CanvasHWND = self._Screen._CanvasHWND
  97. self._RollCanvas = pygame.Surface(( self._RollW,self._RollH))
  98. """
  99. self._BGpng = IconItem()
  100. self._BGpng._ImgSurf = MyIconPool._Icons["sheep_bg"]
  101. self._BGpng._MyType = ICON_TYPES["STAT"]
  102. self._BGpng._Parent = self
  103. self._BGpng.Adjust(0,0,self._BGwidth,self._BGheight,0)
  104. self._SheepHead = IconItem()
  105. self._SheepHead._ImgSurf = MyIconPool._Icons["sheep_head"]
  106. self._SheepHead._MyType = ICON_TYPES["STAT"]
  107. self._SheepHead._Parent = self
  108. self._SheepHead.Adjust(0,0,self._SheepHeadW,self._SheepHeadH,0)
  109. self._SheepBody = IconItem()
  110. self._SheepBody._ImgSurf = MyIconPool._Icons["sheep_body"]
  111. self._SheepBody._MyType = ICON_TYPES["STAT"]
  112. self._SheepBody._Parent = self
  113. self._SheepBody.Adjust(0,0,self._SheepBodyW,self._SheepBodyH,0)
  114. """
  115. self._cwp_png = IconItem()
  116. self._cwp_png._ImgSurf = MyIconPool._Icons["tape"]
  117. self._cwp_png._MyType = ICON_TYPES["STAT"]
  118. self._cwp_png._Parent = self
  119. self._cwp_png.Adjust(0,0,79,79,0)
  120. self._song_title = Label()
  121. self._song_title.SetCanvasHWND(self._RollCanvas)
  122. self._song_title.Init("Untitled",self._SongFont,(255,255,255))
  123. self._title = Label()
  124. self._title.SetCanvasHWND(self._CanvasHWND)
  125. self._title.Init("Title:",self._ListFont,(255,255,255))
  126. self._time = Label()
  127. self._time.SetCanvasHWND(self._CanvasHWND)
  128. self._time.Init("Time:",self._ListFont,(255,255,255))
  129. self._time2 = Label()
  130. self._time2.SetCanvasHWND(self._CanvasHWND)
  131. self._time2.Init("00:00-00:00",self._ListFont,(255,255,255))
  132. self.Start()
  133. def Start(self):
  134. if self._Screen.CurPage() != self:
  135. return
  136. try:
  137. self._FIFO = os.open(self._PIFI._MPD_FIFO, os.O_RDONLY | os.O_NONBLOCK)
  138. t = Thread(target=self.GetSpectrum)
  139. t.daemon = True # thread dies with the program
  140. t.start()
  141. self._ReadingThread = t
  142. except IOError:
  143. print("open %s failed"%self._PIFI._MPD_FIFO)
  144. self._FIFO = None
  145. return
  146. def GetSpectrum(self):
  147. while self._KeepReading and self._FIFO != None:
  148. raw_samples = self._PIFI.GetSpectrum(self._FIFO)
  149. if len(raw_samples) < 1:
  150. #print("sleeping... 0.01")
  151. time.sleep(0.01)
  152. self.read_retry+=1
  153. if self.read_retry > 40:
  154. os.close(self._FIFO)
  155. self._FIFO = os.open(self._PIFI._MPD_FIFO, os.O_RDONLY | os.O_NONBLOCK)
  156. self.read_retry = 0
  157. self.Playing()
  158. else:
  159. self.read_retry = 0
  160. self._queue_data = raw_samples
  161. self.Playing()
  162. def Playing(self):
  163. self._Screen.Draw()
  164. self._Screen.SwapAndShow()
  165. def ClearCanvas(self):
  166. self._CanvasHWND.fill((0,0,0))
  167. def SgsSmooth(self):
  168. passes = 1
  169. points = 3
  170. origs = self._bby[:]
  171. for p in range(0,passes):
  172. pivot = int(points/2.0)
  173. for i in range(0,pivot):
  174. self._bby[i] = origs[i]
  175. self._bby[ len(origs) -i -1 ] = origs[ len(origs) -i -1 ]
  176. smooth_constant = 1.0/(2.0*pivot+1.0)
  177. for i in range(pivot, len(origs)-pivot):
  178. _sum = 0.0
  179. for j in range(0,(2*pivot)+1):
  180. _sum += (smooth_constant * origs[i+j-pivot]) +j -pivot
  181. self._bby[i] = _sum
  182. if p < (passes - 1):
  183. origs = self._bby[:]
  184. def OnLoadCb(self):
  185. if self._Neighbor != None:
  186. pass
  187. if self._KeepReading == False:
  188. self._KeepReading = True
  189. if self._FIFO == None:
  190. self.Start()
  191. def KeyDown(self,event):
  192. if event.key == CurKeys["Menu"] or event.key == CurKeys["A"]:
  193. try:
  194. os.close(self._FIFO)
  195. self._FIFO = None
  196. except Exception, e:
  197. print(e)
  198. self._KeepReading = False
  199. self._ReadingThread.join()
  200. self._ReadingThread = None
  201. self.ReturnToUpLevelPage()
  202. self._Screen.Draw()
  203. self._Screen.SwapAndShow()
  204. if event.key == CurKeys["Start"]:
  205. self._Screen.Draw()
  206. self._Screen.SwapAndShow()
  207. if event.key == CurKeys["Enter"]:
  208. pass
  209. def Draw(self):
  210. self.ClearCanvas()
  211. self._frames+=1
  212. bw = 10
  213. gap = 2
  214. margin_bottom = 72
  215. spects = None
  216. meterNum = self._Width / float(bw +gap ) ## 320/12= 26
  217. meter_left = meterNum - int(meterNum)
  218. meter_left = meter_left*int(bw+gap)
  219. margin_left = meter_left / 2 + gap
  220. meterNum = int(meterNum)
  221. self._cwp_png.NewCoord(43,159)
  222. self._cwp_png.Draw()
  223. if self._Neighbor != None:
  224. if self._Neighbor._CurSongName != "":
  225. self._song_title.SetText(self._Neighbor._CurSongName)
  226. if self._Neighbor._CurSongTime != "":
  227. times = self._Neighbor._CurSongTime
  228. times_ = times.split(":")
  229. if len(times_)> 1:
  230. cur = int(times_[0])
  231. end = int(times_[1])
  232. if cur > 3600:
  233. cur_text = time.strftime('%H:%M:%S', time.gmtime(cur))
  234. else:
  235. cur_text = time.strftime('%M:%S', time.gmtime(cur))
  236. if end > 3600:
  237. end_text = time.strftime('%H:%M:%S', time.gmtime(end))
  238. else:
  239. end_text = time.strftime('%M:%S', time.gmtime(end))
  240. else:
  241. cur_text = ""
  242. end_text = times
  243. self._time2.SetText(cur_text+"-"+end_text)
  244. self._title.NewCoord(90,167)
  245. self._title.Draw()
  246. self._time.NewCoord(90,140)
  247. self._time.Draw()
  248. self._time2.NewCoord(135,140)
  249. self._time2.Draw()
  250. if self._RollCanvas != None:
  251. # self._RollCanvas.fill((111,22,33))
  252. self._RollCanvas.fill((0,0,0))
  253. if self._song_title._Width > self._RollW:
  254. if (self._song_title._PosX + self._song_title._Width) > self._RollW and self._frames % 30 == 0:
  255. self._song_title._PosX -= 1
  256. elif (self._song_title._PosX + self._song_title._Width) <= self._RollW and self._frames % 30 == 0:
  257. self._song_title._PosX = 0
  258. else:
  259. self._song_title._PosX = 0
  260. self._song_title.Draw()
  261. self._CanvasHWND.blit(self._RollCanvas,(135,165,self._RollW,self._RollH))
  262. try:
  263. spects = self._queue_data
  264. if len(spects) == 0:
  265. return
  266. # print("spects:",spects)
  267. step = int( round( len( spects ) / meterNum) )
  268. self._bbs = []
  269. for i in range(0,meterNum):
  270. index = int(i*step)
  271. total = 0
  272. value = spects[index]
  273. self._bbs.append(value)
  274. if len(self._bby) < len(self._bbs):
  275. self._bby = self._bbs
  276. elif len(self._bby) == len(self._bbs):
  277. for i in range(0,len(self._bbs)):
  278. self._bby[i] = (self._bby[i]+self._bbs[i])/2
  279. self.SgsSmooth()
  280. for i in range(0,meterNum):
  281. value = self._bby[ i ]
  282. if math.isnan(value) or math.isinf(value):
  283. value = 0
  284. value = value/32768.0
  285. value = value * 100
  286. value = value % (self._Height-gap-margin_bottom)
  287. if len(self._vis_values) < len(self._bby):
  288. self._vis_values.append(value)
  289. elif len(self._vis_values) == len(self._bby):
  290. if self._vis_values[i] < value:
  291. self._vis_values[i] = value
  292. except Empty:
  293. return
  294. else: # got line
  295. if len(self._vis_values) == 0:
  296. return
  297. for i in range(0,meterNum):
  298. value = self._vis_values[i]
  299. if len(self._capYPositionArray) < round(meterNum):
  300. self._capYPositionArray.append(value)
  301. if value < self._capYPositionArray[i]:
  302. self._capYPositionArray[i]-=0.5
  303. else:
  304. self._capYPositionArray[i] = value
  305. pygame.draw.rect(self._CanvasHWND,(255,255,255),(i*(bw+gap)+margin_left,self._Height-gap-self._capYPositionArray[i]-margin_bottom,bw,gap),0)
  306. pygame.draw.rect(self._CanvasHWND,(255,255,255),(i*(bw+gap)+margin_left,self._Height-value-gap-margin_bottom,bw,value+gap),0)
  307. self._vis_values[i] -= 2