rom_list_page.py 15 KB

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