fav_list_page.py 13 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448
  1. # -*- coding: utf-8 -*-
  2. import os
  3. import pygame
  4. import glob
  5. from libs.roundrects import aa_round_rect
  6. ## local UI import
  7. from UI.constants import Width,Height,ICON_TYPES,RUNEVT
  8. from UI.page import Page,PageSelector
  9. from UI.label import Label
  10. from UI.icon_item import IconItem
  11. from UI.fonts import fonts
  12. from UI.util_funcs import midRect,CmdClean,FileExists
  13. from UI.keys_def import CurKeys
  14. from UI.multi_icon_item import MultiIconItem
  15. from UI.icon_pool import MyIconPool
  16. from UI.scroller import ListScroller
  17. from rom_so_confirm_page import RomSoConfirmPage
  18. from list_item import ListItem
  19. class FavStack:
  20. _Emulator = None
  21. def __init__(self):
  22. self.stack = list()
  23. def Push(self,data):
  24. if data not in self.stack:
  25. self.stack.append(data)
  26. return True
  27. return False
  28. def Pop(self):
  29. if len(self.stack)<=0:
  30. return None,False
  31. return self.stack.pop(),True
  32. def Last(self):
  33. idx = len(self.stack) -1
  34. if idx < 0:## empty stack,return root path
  35. return self._Emulator["ROM"]
  36. else:
  37. return self.stack[ idx ]
  38. def Length(self):
  39. return len(self.stack)
  40. class ListPageSelector(PageSelector):
  41. _BackgroundColor = pygame.Color(131,199,219)
  42. def __init__(self):
  43. self._PosX = 0
  44. self._PosY = 0
  45. self._Height = 0
  46. self._Width = Width-12
  47. def AnimateDraw(self,x2,y2):
  48. pass
  49. def Draw(self):
  50. idx = self._Parent._PsIndex
  51. if idx > (len(self._Parent._MyList)-1):
  52. idx = len(self._Parent._MyList)
  53. if idx > 0:
  54. idx -=1
  55. elif idx == 0:##nothing in _MyList
  56. return
  57. self._Parent._PsIndex = idx #sync PsIndex
  58. x = self._Parent._MyList[idx]._PosX+2
  59. y = self._Parent._MyList[idx]._PosY+1
  60. h = self._Parent._MyList[idx]._Height -3
  61. self._PosX = x
  62. self._PosY = y
  63. self._Height = h
  64. aa_round_rect(self._Parent._CanvasHWND,
  65. (x,y,self._Width-4,h),self._BackgroundColor,4,0,self._BackgroundColor)
  66. class FavListPage(Page):
  67. _Icons = {}
  68. _Selector=None
  69. _FootMsg = ["Nav","Scan","Remove","","Run"]
  70. _MyList = []
  71. _ListFont = fonts["notosanscjk15"]
  72. _MyStack = None
  73. _Emulator = None
  74. _Parent = None
  75. _Scroller = None
  76. _Scrolled = 0
  77. _BGwidth = 75
  78. _BGheight = 73
  79. _RomSoConfirmDownloadPage = None
  80. def __init__(self):
  81. Page.__init__(self)
  82. self._Icons = {}
  83. self._CanvasHWND = None
  84. self._MyList = []
  85. self._MyStack = FavStack()
  86. self._Emulator = {}
  87. def GeneratePathList(self,path):
  88. if os.path.isdir(path) == False:
  89. return False
  90. files_path = glob.glob(path+"/*")
  91. ret = []
  92. for i ,v in enumerate(files_path):
  93. dirmap = {}
  94. if os.path.isdir(v) and self._Emulator["FILETYPE"] == "dir": ## like DOSBOX
  95. gameshell_bat = self._Emulator["EXT"][0]
  96. if FileExists(v+"/"+gameshell_bat):
  97. dirmap["gamedir"] = v.decode("utf8")
  98. ret.append(dirmap)
  99. if os.path.isfile(v) and self._Emulator["FILETYPE"] == "file":
  100. stats = os.stat(v)
  101. if stats.st_gid != self._Parent._FavGID:
  102. continue
  103. bname = os.path.basename(v) ### filter extension
  104. if len(bname)> 1:
  105. pieces = bname.split(".")
  106. if len(pieces) > 1:
  107. if pieces[ len(pieces)-1 ].lower() in self._Emulator["EXT"]:
  108. dirmap["file"] = v
  109. ret.append(dirmap)
  110. # else:
  111. # print("not file or dir")
  112. return ret
  113. def SyncList(self,path):
  114. alist = self.GeneratePathList(path)
  115. if alist == False:
  116. print("listfiles return false")
  117. return
  118. # print("fav list alist: ")
  119. # print(alist)
  120. self._MyList = []
  121. start_x = 0
  122. start_y = 0
  123. hasparent = 0
  124. if self._MyStack.Length() > 0:
  125. hasparent = 1
  126. li = ListItem()
  127. li._Parent = self
  128. li._PosX = start_x
  129. li._PosY = start_y
  130. li._Width = Width
  131. li._Fonts["normal"] = self._ListFont
  132. li._MyType = ICON_TYPES["DIR"]
  133. li._Parent = self
  134. li.Init("[..]")
  135. self._MyList.append(li)
  136. for i,v in enumerate(sorted(alist)):
  137. li = ListItem()
  138. li._Parent = self
  139. li._PosX = start_x
  140. li._PosY = start_y + (i+hasparent)*ListItem._Height
  141. li._Width = Width
  142. li._Fonts["normal"] = self._ListFont
  143. li._MyType = ICON_TYPES["FILE"]
  144. li._Parent = self
  145. if "directory" in v:
  146. li._MyType = ICON_TYPES["DIR"]
  147. li.Init(v["directory"])
  148. elif "file" in v:
  149. li.Init(v["file"])
  150. elif "gamedir" in v:
  151. li.Init(v["gamedir"])
  152. else:
  153. li.Init("NoName")
  154. self._MyList.append(li)
  155. def Init(self):
  156. self._PosX = self._Index * self._Screen._Width
  157. self._Width = self._Screen._Width
  158. self._Height = self._Screen._Height
  159. self._CanvasHWND = self._Screen._CanvasHWND
  160. ps = ListPageSelector()
  161. ps._Parent = self
  162. self._Ps = ps
  163. self._PsIndex = 0
  164. self.SyncList(self._Emulator["ROM"])
  165. self._MyStack._Emulator = self._Emulator
  166. icon_for_list = MultiIconItem()
  167. icon_for_list._ImgSurf = self._Parent._Icons["sys"]
  168. icon_for_list._MyType = ICON_TYPES["STAT"]
  169. icon_for_list._Parent = self
  170. icon_for_list.Adjust(0,0,18,18,0)
  171. self._Icons["sys"] = icon_for_list
  172. bgpng = IconItem()
  173. bgpng._ImgSurf = MyIconPool._Icons["star"]
  174. bgpng._MyType = ICON_TYPES["STAT"]
  175. bgpng._Parent = self
  176. bgpng.AddLabel("my favourites games", fonts["varela18"])
  177. bgpng.SetLableColor(pygame.Color(204,204,204))
  178. bgpng.Adjust(0,0,self._BGwidth,self._BGheight,0)
  179. self._Icons["bg"] = bgpng
  180. self._Scroller = ListScroller()
  181. self._Scroller._Parent = self
  182. self._Scroller._PosX = self._Width - 10
  183. self._Scroller._PosY = 2
  184. self._Scroller.Init()
  185. rom_so_confirm_page = RomSoConfirmPage()
  186. rom_so_confirm_page._Screen = self._Screen
  187. rom_so_confirm_page._Name = "Download Confirm"
  188. rom_so_confirm_page._Parent = self
  189. rom_so_confirm_page.Init()
  190. self._RomSoConfirmDownloadPage = rom_so_confirm_page
  191. def ScrollUp(self):
  192. if len(self._MyList) == 0:
  193. return
  194. self._PsIndex -= 1
  195. if self._PsIndex < 0:
  196. self._PsIndex = 0
  197. cur_li = self._MyList[self._PsIndex]
  198. if cur_li._PosY < 0:
  199. for i in range(0, len(self._MyList)):
  200. self._MyList[i]._PosY += self._MyList[i]._Height
  201. self._Scrolled += 1
  202. def ScrollDown(self):
  203. if len(self._MyList) == 0:
  204. return
  205. self._PsIndex +=1
  206. if self._PsIndex >= len(self._MyList):
  207. self._PsIndex = len(self._MyList) -1
  208. cur_li = self._MyList[self._PsIndex]
  209. if cur_li._PosY +cur_li._Height > self._Height:
  210. for i in range(0,len(self._MyList)):
  211. self._MyList[i]._PosY -= self._MyList[i]._Height
  212. self._Scrolled -=1
  213. def SyncScroll(self):
  214. ##
  215. if self._Scrolled == 0:
  216. return
  217. if self._PsIndex < len(self._MyList):
  218. cur_li = self._MyList[self._PsIndex]
  219. if self._Scrolled > 0:
  220. if cur_li._PosY < 0:
  221. for i in range(0, len(self._MyList)):
  222. self._MyList[i]._PosY += self._Scrolled * self._MyList[i]._Height
  223. elif self._Scrolled < 0:
  224. if cur_li._PosY +cur_li._Height > self._Height:
  225. for i in range(0,len(self._MyList)):
  226. self._MyList[i]._PosY += self._Scrolled * self._MyList[i]._Height
  227. def Click(self):
  228. if len(self._MyList) == 0:
  229. return
  230. if self._PsIndex > len(self._MyList) - 1:
  231. return
  232. cur_li = self._MyList[self._PsIndex]
  233. if cur_li._MyType == ICON_TYPES["DIR"]:
  234. return
  235. if cur_li._MyType == ICON_TYPES["FILE"]: ## add to playlist only
  236. self._Screen._MsgBox.SetText("Launching...")
  237. self._Screen._MsgBox.Draw()
  238. self._Screen.SwapAndShow()
  239. if self._Emulator["FILETYPE"] == "dir":
  240. path = cur_li._Path +"/"+self._Emulator["EXT"][0]
  241. else:
  242. path = cur_li._Path
  243. print("Run ",path)
  244. # check ROM_SO exists
  245. if FileExists(self._Emulator["ROM_SO"]):
  246. escaped_path = CmdClean( path)
  247. custom_config = ""
  248. if self._Emulator["RETRO_CONFIG"] != "" and len(self._Emulator["RETRO_CONFIG"]) > 5:
  249. custom_config = " -c " + self._Emulator["RETRO_CONFIG"]
  250. cmdpath = " ".join( (self._Emulator["LAUNCHER"],self._Emulator["ROM_SO"], custom_config, escaped_path))
  251. pygame.event.post( pygame.event.Event(RUNEVT, message=cmdpath))
  252. return
  253. else:
  254. self._Screen.PushPage(self._RomSoConfirmDownloadPage)
  255. self._Screen.Draw()
  256. self._Screen.SwapAndShow()
  257. return
  258. self._Screen.Draw()
  259. self._Screen.SwapAndShow()
  260. def ReScan(self):
  261. if self._MyStack.Length() == 0:
  262. self.SyncList(self._Emulator["ROM"])
  263. else:
  264. self.SyncList( self._MyStack.Last() )
  265. idx = self._PsIndex
  266. if idx > (len(self._MyList)-1):
  267. idx = len(self._MyList)
  268. if idx > 0:
  269. idx -=1
  270. elif idx == 0:##nothing in _MyList
  271. pass
  272. self._PsIndex = idx ## sync PsIndex
  273. self.SyncScroll()
  274. def OnReturnBackCb(self):
  275. self.ReScan()
  276. self._Screen.Draw()
  277. self._Screen.SwapAndShow()
  278. def OnLoadCb(self):
  279. self.ReScan()
  280. self._Screen.Draw()
  281. self._Screen.SwapAndShow()
  282. def KeyDown(self,event):
  283. if event.key == CurKeys["Menu"] or event.key == CurKeys["Left"]:
  284. self.ReturnToUpLevelPage()
  285. self._Screen.Draw()
  286. self._Screen.SwapAndShow()
  287. if event.key == CurKeys["Up"]:
  288. self.ScrollUp()
  289. self._Screen.Draw()
  290. self._Screen.SwapAndShow()
  291. if event.key == CurKeys["Down"]:
  292. self.ScrollDown()
  293. self._Screen.Draw()
  294. self._Screen.SwapAndShow()
  295. if event.key == CurKeys["Enter"]:
  296. self.Click()
  297. if event.key == CurKeys["X"]: #Scan current
  298. self.ReScan()
  299. self._Screen.Draw()
  300. self._Screen.SwapAndShow()
  301. if event.key == CurKeys["Y"]: #del
  302. if len(self._MyList) == 0:
  303. return
  304. cur_li = self._MyList[self._PsIndex]
  305. if cur_li.IsFile():
  306. """
  307. #self._Parent.FavDeleteConfirmPage.SetFileName(cur_li._Path)
  308. #self._Parent.FavDeleteConfirmPage.SetTrashDir(self._Emulator["ROM"])## Fav delete,return to ROM dir,not .Trash
  309. #self._Screen.PushCurPage()
  310. #self._Screen.SetCurPage(self._Parent.FavDeleteConfirmPage)
  311. #self._Screen.Draw()
  312. #self._Screen.SwapAndShow()
  313. """
  314. #delete directly without confirm dialog
  315. stats = os.stat(cur_li._Path)
  316. os.chown(cur_li._Path, stats.st_uid,stats.st_uid) ## normally uid and gid should be the same
  317. self._Screen._MsgBox.SetText("Deleting...")
  318. self._Screen._MsgBox.Draw()
  319. self._Screen.SwapAndShow()
  320. pygame.time.delay(600)
  321. self.ReScan()
  322. self._Screen.Draw()
  323. self._Screen.SwapAndShow()
  324. def Draw(self):
  325. self.ClearCanvas()
  326. if len(self._MyList) == 0:
  327. self._Icons["bg"].NewCoord(self._Width/2,self._Height/2)
  328. self._Icons["bg"].Draw()
  329. else:
  330. if len(self._MyList) * ListItem._Height > self._Height:
  331. self._Ps._Width = self._Width - 10
  332. self._Ps.Draw()
  333. for i in self._MyList:
  334. if i._PosY > self._Height + self._Height/2:
  335. break
  336. if i._PosY < 0:
  337. continue
  338. i.Draw()
  339. self._Scroller.UpdateSize( len(self._MyList)*ListItem._Height, self._PsIndex*ListItem._Height)
  340. self._Scroller.Draw()
  341. else:
  342. self._Ps._Width = self._Width
  343. self._Ps.Draw()
  344. for i in self._MyList:
  345. i.Draw()