mpd_spectrum_page.py 13 KB

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