fav_list_page.py 16 KB

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