rom_list_page.py 14 KB

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