__init__.py 9.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320
  1. # -*- coding: utf-8 -*-
  2. import pygame
  3. import commands
  4. import shutil
  5. import os
  6. from libs.roundrects import aa_round_rect
  7. from UI.constants import Width,Height,ICON_TYPES
  8. from UI.page import Page,PageSelector
  9. from UI.label import Label
  10. from UI.fonts import fonts
  11. from UI.util_funcs import midRect
  12. from UI.keys_def import CurKeys
  13. from UI.scroller import ListScroller
  14. from UI.icon_pool import MyIconPool
  15. from UI.icon_item import IconItem
  16. from UI.multi_icon_item import MultiIconItem
  17. from UI.multilabel import MultiLabel
  18. from UI.confirm_page import ConfirmPage
  19. class UpdateConfirmPage(ConfirmPage):
  20. _ConfirmText = "Apply to RetroArch?"
  21. _RetroArchConf = "/home/cpi/.config/retroarch/retroarch.cfg"
  22. _LayoutMode = "Unknown"
  23. _DownloadPage = None
  24. _URL = ""
  25. _MD5 = ""
  26. _Version = ""
  27. _GIT = False
  28. def ModifyRetroArchConf(self,keys):
  29. try:
  30. with open(self._RetroArchConf, mode="r") as f:
  31. confarr = f.readlines()
  32. except:
  33. return "retroarch.cfg cannot open."
  34. bka = bkb = bkx = bky = False
  35. try:
  36. for i, ln in enumerate(confarr):
  37. lnk = ln.split("=")[0].strip()
  38. if lnk == "input_player1_a":
  39. confarr[i] = "input_player1_a = \"" + keys[0] + "\"\n"
  40. bka = True
  41. if lnk == "input_player1_b":
  42. confarr[i] = "input_player1_b = \"" + keys[1] + "\"\n"
  43. bkb = True
  44. if lnk == "input_player1_x":
  45. confarr[i] = "input_player1_x = \"" + keys[2] + "\"\n"
  46. bkx = True
  47. if lnk == "input_player1_y":
  48. confarr[i] = "input_player1_y = \"" + keys[3] + "\"\n"
  49. bky = True
  50. except:
  51. return "retroarch.cfg cannot parse."
  52. if bka and bkb and bkx and bky:
  53. None
  54. else:
  55. return "retroarch.cfg validation error."
  56. try:
  57. with open(self._RetroArchConf, mode="w") as f:
  58. confarr = f.writelines(confarr)
  59. except:
  60. return "retroarch.cfg cannot write."
  61. return "Completed! Your RA keymap: " + self._LayoutMode.upper()
  62. def KeyDown(self,event):
  63. def finalizeWithDialog(msg):
  64. self._Screen._MsgBox.SetText(msg)
  65. self._Screen._MsgBox.Draw()
  66. self._Screen.SwapAndShow()
  67. return
  68. if event.key == CurKeys["Menu"] or event.key == CurKeys["A"]:
  69. self.ReturnToUpLevelPage()
  70. self._Screen.Draw()
  71. self._Screen.SwapAndShow()
  72. if event.key == CurKeys["B"]:
  73. if self._LayoutMode == "xbox":
  74. keymap = ["j","k","u","i"]
  75. elif self._LayoutMode == "snes":
  76. keymap = ["k","j","i","u"]
  77. else:
  78. return "Internal error."
  79. print("mode: " + self._LayoutMode)
  80. if not os.path.isfile(self._RetroArchConf):
  81. finalizeWithDialog("retroarch.cfg was not found.")
  82. return
  83. try:
  84. shutil.copyfile(self._RetroArchConf, self._RetroArchConf + ".blbak")
  85. except:
  86. finalizeWithDialog("Cannot create .blbak")
  87. return
  88. finalizeWithDialog(self.ModifyRetroArchConf(keymap))
  89. return
  90. self.ReturnToUpLevelPage()
  91. self._Screen.Draw()
  92. self._Screen.SwapAndShow()
  93. return
  94. if self._GIT == True:
  95. cmdpath = "feh --bg-center /home/cpi/apps/launcher/sys.py/gameshell/wallpaper/updating.png; cd /home/cpi/apps/launcher ;git pull; git reset --hard %s ; feh --bg-center /home/cpi/apps/launcher/sys.py/gameshell/wallpaper/loading.png " % self._Version
  96. pygame.event.post( pygame.event.Event(RUNEVT, message=cmdpath))
  97. self._GIT = False
  98. return
  99. if self._DownloadPage == None:
  100. self._DownloadPage = UpdateDownloadPage()
  101. self._DownloadPage._Screen = self._Screen
  102. self._DownloadPage._Name = "Downloading..."
  103. self._DownloadPage.Init()
  104. self._DownloadPage._MD5 = self._MD5
  105. self._Screen.PushPage(self._DownloadPage)
  106. self._Screen.Draw()
  107. self._Screen.SwapAndShow()
  108. if self._URL != None and validators.url(self._URL):
  109. self._DownloadPage.StartDownload(self._URL, "/tmp")
  110. else:
  111. print "error url %s " % self._URL
  112. def OnReturnBackCb(self):
  113. self.ReturnToUpLevelPage()
  114. self._Screen.Draw()
  115. self._Screen.SwapAndShow()
  116. def Draw(self):
  117. self.ClearCanvas()
  118. self.DrawBG()
  119. for i in self._MyList:
  120. i.Draw()
  121. self.Reset()
  122. class ButtonsLayoutPage(Page):
  123. _FootMsg = ["Nav.","UpdateRetroArch","","Back","Toggle"]
  124. _MyList = []
  125. _ListFontObj = fonts["varela13"]
  126. _AList = {}
  127. _Scrolled = 0
  128. _BGwidth = 320
  129. _BGheight = 240-24-20
  130. _DrawOnce = False
  131. _Scroller = None
  132. _ConfirmPage = None
  133. _EasingDur = 30
  134. _dialog_index = 0
  135. def __init__(self):
  136. Page.__init__(self)
  137. self._Icons = {}
  138. def GenList(self):
  139. self._MyList = []
  140. def Init(self):
  141. if self._Screen != None:
  142. if self._Screen._CanvasHWND != None and self._CanvasHWND == None:
  143. self._HWND = self._Screen._CanvasHWND
  144. self._CanvasHWND = pygame.Surface( (self._Screen._Width,self._BGheight) )
  145. self._PosX = self._Index*self._Screen._Width
  146. self._Width = self._Screen._Width ## equal to screen width
  147. self._Height = self._Screen._Height
  148. DialogBoxs = MultiIconItem()
  149. DialogBoxs._ImgSurf = MyIconPool._Icons["buttonslayout"]
  150. DialogBoxs._MyType = ICON_TYPES["STAT"]
  151. DialogBoxs._Parent = self
  152. DialogBoxs._IconWidth = 300
  153. DialogBoxs._IconHeight = 150
  154. DialogBoxs.Adjust(0,0,134,372,0)
  155. self._Icons["DialogBoxs"] = DialogBoxs
  156. self.GenList()
  157. self._Scroller = ListScroller()
  158. self._Scroller._Parent = self
  159. self._Scroller._PosX = self._Width - 10
  160. self._Scroller._PosY = 2
  161. self._Scroller.Init()
  162. self._Scroller.SetCanvasHWND(self._HWND)
  163. self._ConfirmPage = UpdateConfirmPage()
  164. self._ConfirmPage._LayoutMode = self.GetButtonsLayoutMode()
  165. self._ConfirmPage._Screen = self._Screen
  166. self._ConfirmPage._Name = "Overwrite RA conf"
  167. self._ConfirmPage._Parent = self
  168. self._ConfirmPage.Init()
  169. def ScrollDown(self):
  170. dis = 10
  171. if abs(self._Scrolled) < (self._BGheight - self._Height)/2 + 0:
  172. self._PosY -= dis
  173. self._Scrolled -= dis
  174. def ScrollUp(self):
  175. dis = 10
  176. if self._PosY < 0:
  177. self._PosY += dis
  178. self._Scrolled += dis
  179. def GetButtonsLayoutMode(self):
  180. lm = "xbox"
  181. try:
  182. with open(".buttonslayout", "r") as f:
  183. lm = f.read()
  184. except:
  185. None
  186. if lm not in ["xbox","snes"]:
  187. lm = "xbox"
  188. return lm
  189. def ToggleMode(self):
  190. if self.GetButtonsLayoutMode() == "xbox":
  191. with open(".buttonslayout", "w") as f:
  192. f.write("snes")
  193. self._dialog_index = 1
  194. self._Screen.Draw()
  195. self._Screen.SwapAndShow()
  196. else:
  197. with open(".buttonslayout", "w") as f:
  198. f.write("xbox")
  199. self._dialog_index = 0
  200. self._Screen.Draw()
  201. self._Screen.SwapAndShow()
  202. def OnLoadCb(self):
  203. self._Scrolled = 0
  204. self._PosY = 0
  205. self._DrawOnce = False
  206. if self.GetButtonsLayoutMode() == "xbox":
  207. self._dialog_index = 0
  208. else:
  209. self._dialog_index = 1
  210. def OnReturnBackCb(self):
  211. self.ReturnToUpLevelPage()
  212. self._Screen.Draw()
  213. self._Screen.SwapAndShow()
  214. def KeyDown(self,event):
  215. if event.key == CurKeys["A"] or event.key == CurKeys["Menu"]:
  216. self.ReturnToUpLevelPage()
  217. self._Screen.Draw()
  218. self._Screen.SwapAndShow()
  219. if event.key == CurKeys["B"]:
  220. self.ToggleMode()
  221. if event.key == CurKeys["X"]:
  222. self._ConfirmPage._LayoutMode = self.GetButtonsLayoutMode()
  223. self._Screen.PushPage(self._ConfirmPage)
  224. self._Screen.Draw()
  225. self._Screen.SwapAndShow()
  226. def Draw(self):
  227. self.ClearCanvas()
  228. self._Icons["DialogBoxs"].NewCoord(0,30)
  229. self._Icons["DialogBoxs"]._IconIndex = self._dialog_index
  230. self._Icons["DialogBoxs"].DrawTopLeft()
  231. if self._HWND != None:
  232. self._HWND.fill((255,255,255))
  233. self._HWND.blit(self._CanvasHWND,(self._PosX,self._PosY,self._Width, self._Height ) )
  234. class APIOBJ(object):
  235. _Page = None
  236. def __init__(self):
  237. pass
  238. def Init(self,main_screen):
  239. self._Page = ButtonsLayoutPage()
  240. self._Page._Screen = main_screen
  241. self._Page._Name ="Buttons Layout"
  242. self._Page.Init()
  243. def API(self,main_screen):
  244. if main_screen !=None:
  245. main_screen.PushPage(self._Page)
  246. main_screen.Draw()
  247. main_screen.SwapAndShow()
  248. OBJ = APIOBJ()
  249. def Init(main_screen):
  250. OBJ.Init(main_screen)
  251. def API(main_screen):
  252. OBJ.API(main_screen)