music_lib_list_page.py 9.3 KB

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