mpd_spectrum_page.py 12 KB

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