__init__.py 8.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287
  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. def ModifyRetroArchConf(self,keys):
  24. try:
  25. with open(self._RetroArchConf, mode="r") as f:
  26. confarr = f.readlines()
  27. except:
  28. return "retroarch.cfg cannot open."
  29. bka = bkb = bkx = bky = False
  30. try:
  31. for i, ln in enumerate(confarr):
  32. lnk = ln.split("=")[0].strip()
  33. if lnk == "input_player1_a":
  34. confarr[i] = "input_player1_a = \"" + keys[0] + "\"\n"
  35. bka = True
  36. if lnk == "input_player1_b":
  37. confarr[i] = "input_player1_b = \"" + keys[1] + "\"\n"
  38. bkb = True
  39. if lnk == "input_player1_x":
  40. confarr[i] = "input_player1_x = \"" + keys[2] + "\"\n"
  41. bkx = True
  42. if lnk == "input_player1_y":
  43. confarr[i] = "input_player1_y = \"" + keys[3] + "\"\n"
  44. bky = True
  45. except:
  46. return "retroarch.cfg cannot parse."
  47. if bka and bkb and bkx and bky:
  48. None
  49. else:
  50. return "retroarch.cfg validation error."
  51. try:
  52. with open(self._RetroArchConf, mode="w") as f:
  53. confarr = f.writelines(confarr)
  54. except:
  55. return "retroarch.cfg cannot write."
  56. return "Completed! Your RA keymap: " + self._LayoutMode.upper()
  57. def KeyDown(self,event):
  58. def finalizeWithDialog(msg):
  59. self._Screen._MsgBox.SetText(msg)
  60. self._Screen._MsgBox.Draw()
  61. self._Screen.SwapAndShow()
  62. return
  63. if event.key == CurKeys["Menu"] or event.key == CurKeys["A"]:
  64. self.ReturnToUpLevelPage()
  65. self._Screen.Draw()
  66. self._Screen.SwapAndShow()
  67. if event.key == CurKeys["B"]:
  68. if self._LayoutMode == "xbox":
  69. keymap = ["j","k","u","i"]
  70. elif self._LayoutMode == "snes":
  71. keymap = ["k","j","i","u"]
  72. else:
  73. finalizeWithDialog("Internal error.")
  74. return
  75. print("mode: " + self._LayoutMode)
  76. if not os.path.isfile(self._RetroArchConf):
  77. finalizeWithDialog("retroarch.cfg was not found.")
  78. return
  79. try:
  80. shutil.copyfile(self._RetroArchConf, self._RetroArchConf + ".blbak")
  81. except:
  82. finalizeWithDialog("Cannot create .blbak")
  83. return
  84. finalizeWithDialog(self.ModifyRetroArchConf(keymap))
  85. return
  86. def OnReturnBackCb(self):
  87. self.ReturnToUpLevelPage()
  88. self._Screen.Draw()
  89. self._Screen.SwapAndShow()
  90. def Draw(self):
  91. self.ClearCanvas()
  92. self.DrawBG()
  93. for i in self._MyList:
  94. i.Draw()
  95. self.Reset()
  96. class ButtonsLayoutPage(Page):
  97. _FootMsg = ["Nav.","UpdateRetroArch","","Back","Toggle"]
  98. _MyList = []
  99. _ListFontObj = fonts["varela13"]
  100. _AList = {}
  101. _Scrolled = 0
  102. _BGwidth = 320
  103. _BGheight = 240-24-20
  104. _DrawOnce = False
  105. _Scroller = None
  106. _ConfirmPage = None
  107. _EasingDur = 30
  108. _dialog_index = 0
  109. def __init__(self):
  110. Page.__init__(self)
  111. self._Icons = {}
  112. def GenList(self):
  113. self._MyList = []
  114. def Init(self):
  115. if self._Screen != None:
  116. if self._Screen._CanvasHWND != None and self._CanvasHWND == None:
  117. self._HWND = self._Screen._CanvasHWND
  118. self._CanvasHWND = pygame.Surface( (self._Screen._Width,self._BGheight) )
  119. self._PosX = self._Index*self._Screen._Width
  120. self._Width = self._Screen._Width ## equal to screen width
  121. self._Height = self._Screen._Height
  122. DialogBoxs = MultiIconItem()
  123. DialogBoxs._ImgSurf = MyIconPool._Icons["buttonslayout"]
  124. DialogBoxs._MyType = ICON_TYPES["STAT"]
  125. DialogBoxs._Parent = self
  126. DialogBoxs._IconWidth = 300
  127. DialogBoxs._IconHeight = 150
  128. DialogBoxs.Adjust(0,0,134,372,0)
  129. self._Icons["DialogBoxs"] = DialogBoxs
  130. self.GenList()
  131. self._Scroller = ListScroller()
  132. self._Scroller._Parent = self
  133. self._Scroller._PosX = self._Width - 10
  134. self._Scroller._PosY = 2
  135. self._Scroller.Init()
  136. self._Scroller.SetCanvasHWND(self._HWND)
  137. self._ConfirmPage = UpdateConfirmPage()
  138. self._ConfirmPage._LayoutMode = self.GetButtonsLayoutMode()
  139. self._ConfirmPage._Screen = self._Screen
  140. self._ConfirmPage._Name = "Overwrite RA conf"
  141. self._ConfirmPage._Parent = self
  142. self._ConfirmPage.Init()
  143. def ScrollDown(self):
  144. dis = 10
  145. if abs(self._Scrolled) < (self._BGheight - self._Height)/2 + 0:
  146. self._PosY -= dis
  147. self._Scrolled -= dis
  148. def ScrollUp(self):
  149. dis = 10
  150. if self._PosY < 0:
  151. self._PosY += dis
  152. self._Scrolled += dis
  153. def GetButtonsLayoutMode(self):
  154. lm = "xbox"
  155. try:
  156. with open(".buttonslayout", "r") as f:
  157. lm = f.read()
  158. except:
  159. None
  160. if lm not in ["xbox","snes"]:
  161. lm = "xbox"
  162. return lm
  163. def ToggleMode(self):
  164. if self.GetButtonsLayoutMode() == "xbox":
  165. with open(".buttonslayout", "w") as f:
  166. f.write("snes")
  167. self._dialog_index = 1
  168. self._Screen.Draw()
  169. self._Screen.SwapAndShow()
  170. else:
  171. with open(".buttonslayout", "w") as f:
  172. f.write("xbox")
  173. self._dialog_index = 0
  174. self._Screen.Draw()
  175. self._Screen.SwapAndShow()
  176. def OnLoadCb(self):
  177. self._Scrolled = 0
  178. self._PosY = 0
  179. self._DrawOnce = False
  180. if self.GetButtonsLayoutMode() == "xbox":
  181. self._dialog_index = 0
  182. else:
  183. self._dialog_index = 1
  184. def OnReturnBackCb(self):
  185. self.ReturnToUpLevelPage()
  186. self._Screen.Draw()
  187. self._Screen.SwapAndShow()
  188. def KeyDown(self,event):
  189. if event.key == CurKeys["A"] or event.key == CurKeys["Menu"]:
  190. self.ReturnToUpLevelPage()
  191. self._Screen.Draw()
  192. self._Screen.SwapAndShow()
  193. if event.key == CurKeys["B"]:
  194. self.ToggleMode()
  195. if event.key == CurKeys["X"]:
  196. self._ConfirmPage._LayoutMode = self.GetButtonsLayoutMode()
  197. self._Screen.PushPage(self._ConfirmPage)
  198. self._Screen.Draw()
  199. self._Screen.SwapAndShow()
  200. def Draw(self):
  201. self.ClearCanvas()
  202. self._Icons["DialogBoxs"].NewCoord(0,30)
  203. self._Icons["DialogBoxs"]._IconIndex = self._dialog_index
  204. self._Icons["DialogBoxs"].DrawTopLeft()
  205. if self._HWND != None:
  206. self._HWND.fill((255,255,255))
  207. self._HWND.blit(self._CanvasHWND,(self._PosX,self._PosY,self._Width, self._Height ) )
  208. class APIOBJ(object):
  209. _Page = None
  210. def __init__(self):
  211. pass
  212. def Init(self,main_screen):
  213. self._Page = ButtonsLayoutPage()
  214. self._Page._Screen = main_screen
  215. self._Page._Name ="Buttons Layout"
  216. self._Page.Init()
  217. def API(self,main_screen):
  218. if main_screen !=None:
  219. main_screen.PushPage(self._Page)
  220. main_screen.Draw()
  221. main_screen.SwapAndShow()
  222. OBJ = APIOBJ()
  223. def Init(main_screen):
  224. OBJ.Init(main_screen)
  225. def API(main_screen):
  226. OBJ.API(main_screen)