rom_list_page.py 16 KB

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