music_lib_list_page.py 10 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355
  1. # -*- coding: utf-8 -*-
  2. import os
  3. import pygame
  4. from libs.roundrects import aa_round_rect
  5. ## local UI import
  6. from UI.constants import Width,Height,ICON_TYPES
  7. from UI.page import Page,PageSelector
  8. from UI.label import Label
  9. from UI.fonts import fonts
  10. from UI.icon_item import IconItem
  11. from UI.util_funcs import midRect
  12. from UI.keys_def import CurKeys
  13. from UI.multi_icon_item import MultiIconItem
  14. from UI.icon_pool import MyIconPool
  15. from UI.scroller import ListScroller
  16. from list_item import ListItem
  17. import myvars
  18. class MusicLibStack:
  19. def __init__(self):
  20. self.stack = list()
  21. def Push(self,data):
  22. if data not in self.stack:
  23. self.stack.append(data)
  24. return True
  25. return False
  26. def Pop(self):
  27. if len(self.stack)<=0:
  28. return None,False
  29. return self.stack.pop(),True
  30. def Last(self):
  31. idx = len(self.stack) -1
  32. if idx < 0:
  33. return "/"
  34. else:
  35. return self.stack[ idx ]
  36. def Length(self):
  37. return len(self.stack)
  38. class ListPageSelector(PageSelector):
  39. _BackgroundColor = pygame.Color(131,199,219)
  40. def __init__(self):
  41. self._PosX = 0
  42. self._PosY = 0
  43. self._Height = 0
  44. self._Width = Width
  45. def AnimateDraw(self,x2,y2):
  46. pass
  47. def Draw(self):
  48. idx = self._Parent._PsIndex
  49. if idx < len(self._Parent._MyList):
  50. x = self._Parent._MyList[idx]._PosX+2
  51. y = self._Parent._MyList[idx]._PosY+1
  52. h = self._Parent._MyList[idx]._Height -3
  53. self._PosX = x
  54. self._PosY = y
  55. self._Height = h
  56. aa_round_rect(self._Parent._CanvasHWND,
  57. (x,y,self._Width-4,h),self._BackgroundColor,4,0,self._BackgroundColor)
  58. class MusicLibListPage(Page):
  59. _Icons = {}
  60. _Selector=None
  61. _FootMsg = ["Nav","Scan","","Back","Add to Playlist"]
  62. _MyList = []
  63. _SwapMyList = []
  64. _ListFont = fonts["notosanscjk15"]
  65. _MyStack = None
  66. _Scroller = None
  67. _BGpng = None
  68. _BGwidth = 56
  69. _BGheight = 70
  70. def __init__(self):
  71. Page.__init__(self)
  72. self._Icons = {}
  73. self._CanvasHWND = None
  74. self._MyList = []
  75. self._SwapMyList = []
  76. self._MyStack = MusicLibStack()
  77. def SyncList(self,path):
  78. if myvars.Poller == None:
  79. return
  80. alist = myvars.Poller.listfiles(path)
  81. if alist == False:
  82. print("listfiles return false")
  83. return
  84. self._MyList = []
  85. self._SwapMyList = []
  86. start_x = 0
  87. start_y = 0
  88. hasparent = 0
  89. if self._MyStack.Length() > 0:
  90. hasparent = 1
  91. li = ListItem()
  92. li._Parent = self
  93. li._PosX = start_x
  94. li._PosY = start_y
  95. li._Width = Width
  96. li._Fonts["normal"] = self._ListFont
  97. li._MyType = ICON_TYPES["DIR"]
  98. li._Parent = self
  99. li.Init("[..]")
  100. self._MyList.append(li)
  101. for i,v in enumerate(sorted(alist)):
  102. li = ListItem()
  103. li._Parent = self
  104. li._PosX = start_x
  105. li._PosY = start_y + (i+hasparent)*ListItem._Height
  106. li._Width = Width
  107. li._Fonts["normal"] = self._ListFont
  108. li._MyType = ICON_TYPES["FILE"]
  109. li._Parent = self
  110. if "directory" in v:
  111. li._MyType = ICON_TYPES["DIR"]
  112. dir_base_name = os.path.basename(v["directory"])
  113. li.Init( dir_base_name )
  114. li._Path = v["directory"]
  115. elif "file" in v:
  116. bname = os.path.basename(v["file"])
  117. li.Init( bname )
  118. li._Path = v["file"]
  119. else:
  120. li.Init("NoName")
  121. self._MyList.append(li)
  122. for i in self._MyList:
  123. self._SwapMyList.append(i)
  124. def Init(self):
  125. self._PosX = self._Index * self._Screen._Width
  126. self._Width = self._Screen._Width
  127. self._Height = self._Screen._Height
  128. self._CanvasHWND = self._Screen._CanvasHWND
  129. ps = ListPageSelector()
  130. ps._Parent = self
  131. self._Ps = ps
  132. self._PsIndex = 0
  133. self.SyncList("/")
  134. icon_for_list = MultiIconItem()
  135. icon_for_list._ImgSurf = MyIconPool._Icons["sys"]
  136. icon_for_list._MyType = ICON_TYPES["STAT"]
  137. icon_for_list._Parent = self
  138. icon_for_list.Adjust(0,0,18,18,0)
  139. self._Icons["sys"] = icon_for_list
  140. self._BGpng = IconItem()
  141. self._BGpng._ImgSurf = MyIconPool._Icons["empty"]
  142. self._BGpng._MyType = ICON_TYPES["STAT"]
  143. self._BGpng._Parent = self
  144. self._BGpng.AddLabel("Please upload data over Wi-Fi", fonts["varela22"])
  145. self._BGpng.SetLableColor(pygame.Color(204,204,204))
  146. self._BGpng.Adjust(0,0,self._BGwidth,self._BGheight,0)
  147. self._Scroller = ListScroller()
  148. self._Scroller._Parent = self
  149. self._Scroller._PosX = self._Width - 10
  150. self._Scroller._PosY = 2
  151. self._Scroller.Init()
  152. def ScrollUp(self,Step=1):
  153. if len(self._MyList) == 0:
  154. return
  155. tmp = self._PsIndex
  156. self._PsIndex -= Step
  157. if self._PsIndex < 0:
  158. self._PsIndex = 0
  159. dy = tmp-self._PsIndex
  160. cur_li = self._MyList[self._PsIndex]
  161. if cur_li._PosY < 0:
  162. for i in range(0, len(self._MyList)):
  163. self._MyList[i]._PosY += self._MyList[i]._Height*dy
  164. def ScrollDown(self,Step=1):
  165. if len(self._MyList) == 0:
  166. return
  167. tmp = self._PsIndex
  168. self._PsIndex +=Step
  169. if self._PsIndex >= len(self._MyList):
  170. self._PsIndex = len(self._MyList) -1
  171. dy = self._PsIndex - tmp
  172. cur_li = self._MyList[self._PsIndex]
  173. if cur_li._PosY +cur_li._Height > self._Height:
  174. for i in range(0,len(self._MyList)):
  175. self._MyList[i]._PosY -= self._MyList[i]._Height*dy
  176. def Click(self):
  177. if len(self._MyList) == 0:
  178. return
  179. cur_li = self._MyList[self._PsIndex]
  180. if cur_li._MyType == ICON_TYPES["DIR"]:
  181. if cur_li._Path == "[..]":
  182. self._MyStack.Pop()
  183. self.SyncList( self._MyStack.Last() )
  184. self._PsIndex = 0
  185. else:
  186. self._MyStack.Push( self._MyList[self._PsIndex]._Path )
  187. self.SyncList( self._MyStack.Last() )
  188. self._PsIndex = 0
  189. if cur_li._MyType == ICON_TYPES["FILE"]: ## add to playlist only
  190. myvars.Poller.addfile(cur_li._Path)
  191. myvars.PlayListPage.SyncList()
  192. print("add" , cur_li._Path)
  193. self._Screen.Draw()
  194. self._Screen.SwapAndShow()
  195. def Rescan(self):
  196. self.SyncList("/")
  197. self._PsIndex = 0
  198. def KeyDown(self,event):
  199. if event.key == CurKeys["Menu"] or event.key == CurKeys["Left"] or event.key == CurKeys["A"]:
  200. self.ReturnToUpLevelPage()
  201. self._Screen.Draw()
  202. self._Screen.SwapAndShow()
  203. if event.key == CurKeys["Up"]:
  204. self.ScrollUp()
  205. self._Screen.Draw()
  206. self._Screen.SwapAndShow()
  207. if event.key == CurKeys["Down"]:
  208. self.ScrollDown()
  209. self._Screen.Draw()
  210. self._Screen.SwapAndShow()
  211. """
  212. if event.key == CurKeys["Right"]:
  213. self.ScrollDown(Step=5)
  214. self._Screen.Draw()
  215. self._Screen.SwapAndShow()
  216. if event.key == CurKeys["Left"]:
  217. self.ScrollUp(Step=5)
  218. self._Screen.Draw()
  219. self._Screen.SwapAndShow()
  220. """
  221. if event.key == CurKeys["X"]:
  222. self.Rescan()
  223. self._Screen.Draw()
  224. self._Screen.SwapAndShow()
  225. if event.key == CurKeys["Enter"]:
  226. self.Click()
  227. def Draw(self):
  228. self.ClearCanvas()
  229. """
  230. start_x = 0
  231. start_y = 0
  232. counter = 0
  233. self._MyList = []
  234. for i,v in enumerate(self._SwapMyList):
  235. if myvars.PlayListPage.InPlayList(v._Path):
  236. v._Active = True
  237. else:
  238. v._Active = False
  239. if v._Active == False:
  240. v.NewCoord(start_x, start_y+counter* ListItem._Height)
  241. counter+=1
  242. self._MyList.append(v)
  243. """
  244. if len(self._MyList) == 0:
  245. self._BGpng.NewCoord(self._Width/2,self._Height/2)
  246. self._BGpng.Draw()
  247. return
  248. else:
  249. if len(self._MyList) * ListItem._Height > self._Height:
  250. self._Ps._Width = self._Width - 11
  251. self._Ps.Draw()
  252. for i in self._MyList:
  253. if myvars.PlayListPage.InPlayList(i._Path):
  254. i._Active = True
  255. else:
  256. i._Active = False
  257. if i._PosY > self._Height + self._Height/2:
  258. break
  259. if i._PosY < 0:
  260. continue
  261. i.Draw()
  262. self._Scroller.UpdateSize( len(self._MyList)*ListItem._Height, self._PsIndex*ListItem._Height)
  263. self._Scroller.Draw()
  264. else:
  265. self._Ps._Width = self._Width
  266. self._Ps.Draw()
  267. for i in self._MyList:
  268. if myvars.PlayListPage.InPlayList(i._Path):
  269. i._Active = True
  270. else:
  271. i._Active = False
  272. if i._PosY > self._Height + self._Height/2:
  273. break
  274. if i._PosY < 0:
  275. continue
  276. i.Draw()