mpd_spectrum_page.py 13 KB

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