__init__.py 9.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302
  1. # -*- coding: utf-8 -*-
  2. import os
  3. import pygame
  4. import platform
  5. #import commands
  6. import glob
  7. #from beeprint import pp
  8. from libs.roundrects import aa_round_rect
  9. ## local UI import
  10. from UI.constants import Width,Height,ICON_TYPES,RESTARTUI
  11. from UI.page import Page,PageSelector
  12. from UI.label import Label
  13. from UI.util_funcs import midRect,FileExists,ArmSystem
  14. from UI.keys_def import CurKeys, IsKeyStartOrA, IsKeyMenuOrB
  15. from UI.scroller import ListScroller
  16. from UI.icon_pool import MyIconPool
  17. from UI.icon_item import IconItem
  18. from UI.multilabel import MultiLabel
  19. from UI.skin_manager import MySkinManager
  20. from UI.lang_manager import MyLangManager
  21. from UI.info_page_list_item import InfoPageListItem
  22. from UI.info_page_selector import InfoPageSelector
  23. import config
  24. class ListPageSelector(InfoPageSelector):
  25. def Draw(self):
  26. idx = self._Parent._PsIndex
  27. if idx < len(self._Parent._MyList):
  28. x = self._Parent._MyList[idx]._PosX+2 ## ++
  29. y = self._Parent._MyList[idx]._PosY+1
  30. h = self._Parent._MyList[idx]._Height -3
  31. self._PosX = x
  32. self._PosY = y
  33. self._Height = h
  34. aa_round_rect(self._Parent._CanvasHWND,
  35. (x,y,self._Width-4,h),self._BackgroundColor,4,0,self._BackgroundColor)
  36. class PageListItem(InfoPageListItem):
  37. _PosX = 0
  38. _PosY = 0
  39. _Width = 0
  40. _Height = 30
  41. _Labels = {}
  42. _Icons = {}
  43. _Fonts = {}
  44. _LinkObj = None
  45. _Active = False
  46. _Value = ""
  47. def Draw(self):
  48. self._Labels["Text"]._PosY = self._PosY+ (self._Height- self._Labels["Text"]._Height)/2
  49. if self._Active == True:
  50. self._Parent._Icons["done"].NewCoord( self._Parent._Width-30,self._PosY+5)
  51. self._Parent._Icons["done"].Draw()
  52. self._Labels["Text"].Draw(self._Active)
  53. if "Small" in self._Labels:
  54. self._Labels["Small"]._PosX = self._Width - self._Labels["Small"]._Width -10
  55. self._Labels["Small"]._PosY = self._PosY + (self._Height- self._Labels["Small"]._Height)/2
  56. self._Labels["Small"].Draw()
  57. pygame.draw.line(self._Parent._CanvasHWND,MySkinManager.GiveColor('Line'),(self._PosX,self._PosY+self._Height-1),(self._PosX+self._Width,self._PosY+self._Height-1),1)
  58. class GPUDriverPage(Page):
  59. _FootMsg = ["Nav","","","Back","Select"]
  60. _MyList = []
  61. _ListFont = MyLangManager.TrFont("notosanscjk12")
  62. _AList = {}
  63. _Scrolled = 0
  64. _BGwidth = 320
  65. _BGheight = 240-24-20
  66. _DrawOnce = False
  67. _Scroller = None
  68. _InfoPage = None
  69. def __init__(self):
  70. Page.__init__(self)
  71. self._Icons = {}
  72. def GenList(self):
  73. self._MyList = []
  74. start_x = 0
  75. start_y = 0
  76. last_height = 0
  77. drivers = [["fbturbo","FBTURBO driver (Software Rendering)"],
  78. ["modesetting","LIMA driver (Experimental GPU Rendering)"]]
  79. for i,u in enumerate( drivers ):
  80. #print(i,u)
  81. li = PageListItem()
  82. li._Parent = self
  83. li._PosX = start_x
  84. li._PosY = start_y + last_height
  85. li._Width = Width
  86. li._Fonts["normal"] = self._ListFont
  87. li._Active = False
  88. li._Value = u[0]
  89. li.Init( u[1] )
  90. last_height += li._Height
  91. self._MyList.append(li)
  92. def Init(self):
  93. if self._Screen != None:
  94. if self._Screen._CanvasHWND != None and self._CanvasHWND == None:
  95. self._HWND = self._Screen._CanvasHWND
  96. self._CanvasHWND = pygame.Surface( (self._Screen._Width,self._BGheight) )
  97. self._PosX = self._Index*self._Screen._Width
  98. self._Width = self._Screen._Width ## equal to screen width
  99. self._Height = self._Screen._Height
  100. done = IconItem()
  101. done._ImgSurf = MyIconPool.GiveIconSurface("done")
  102. done._MyType = ICON_TYPES["STAT"]
  103. done._Parent = self
  104. self._Icons["done"] = done
  105. ps = ListPageSelector()
  106. ps._Parent = self
  107. self._Ps = ps
  108. self._PsIndex = 0
  109. self.GenList()
  110. self._Scroller = ListScroller()
  111. self._Scroller._Parent = self
  112. self._Scroller._PosX = self._Width - 10
  113. self._Scroller._PosY = 2
  114. self._Scroller.Init()
  115. self._Scroller.SetCanvasHWND(self._HWND)
  116. def Click(self):
  117. if self._PsIndex > len(self._MyList) -1:
  118. return
  119. cur_li = self._MyList[self._PsIndex]
  120. if cur_li._Active == True:
  121. return
  122. print(cur_li._Value)
  123. if "arm" in platform.machine():
  124. for i in self._MyList:
  125. i._Active = False
  126. cur_li._Active = True
  127. self._Screen._MsgBox.SetText("Applying")
  128. self._Screen._MsgBox.Draw()
  129. self._Screen.SwapAndShow()
  130. if "modesetting" in cur_li._Value: ## enable lima
  131. os.system("touch %s/.lima" % os.path.expanduser('~') )
  132. ArmSystem("sudo mv /usr/lib/xorg/modules/drivers/modesetting_drv.so.lima /usr/lib/xorg/modules/drivers/modesetting_drv.so")
  133. ArmSystem("sudo sed -i '/^#.*lima/s/^#//' /etc/ld.so.conf.d/00-arm-linux-gnueabihf.conf")
  134. ArmSystem("sudo ldconfig")
  135. else: #disable lima
  136. os.system("rm %s/.lima" % os.path.expanduser('~') )
  137. ArmSystem("sudo mv /usr/lib/xorg/modules/drivers/modesetting_drv.so /usr/lib/xorg/modules/drivers/modesetting_drv.so.lima")
  138. ArmSystem("sudo sed -i 's/^[^#]*lima/#&/' /etc/ld.so.conf.d/00-arm-linux-gnueabihf.conf")
  139. ArmSystem("sudo ldconfig")
  140. pygame.time.delay(800)
  141. os.system("sudo reboot")
  142. else:
  143. self._Screen._MsgBox.SetText("Do it in GameShell")
  144. self._Screen._MsgBox.Draw()
  145. self._Screen.SwapAndShow()
  146. def OnLoadCb(self):
  147. self._Scrolled = 0
  148. self._PosY = 0
  149. self._DrawOnce = False
  150. ## grep Driver /etc/xorg.conf | tr -s " " | cut -d " " -f3
  151. ## "fbturbo"
  152. ## "modesetting"
  153. thedrv = ""
  154. if "arm" in platform.machine():
  155. if FileExists("%s/.lima" % os.path.expanduser('~')):
  156. thedrv = "modesetting"
  157. else:
  158. thedrv = "fbturbo"
  159. if thedrv == "":
  160. thedrv = "fbturbo"
  161. for i in self._MyList:
  162. i._Active = False
  163. for i in self._MyList:
  164. if thedrv in i._Value:
  165. i._Active = True
  166. def OnReturnBackCb(self):
  167. pass
  168. """
  169. self.ReturnToUpLevelPage()
  170. self._Screen.Draw()
  171. self._Screen.SwapAndShow()
  172. """
  173. def KeyDown(self,event):
  174. if IsKeyMenuOrB(event.key):
  175. self.ReturnToUpLevelPage()
  176. self._Screen.Draw()
  177. self._Screen.SwapAndShow()
  178. if IsKeyStartOrA(event.key):
  179. self.Click()
  180. if event.key == CurKeys["Up"]:
  181. self.ScrollUp()
  182. self._Screen.Draw()
  183. self._Screen.SwapAndShow()
  184. if event.key == CurKeys["Down"]:
  185. self.ScrollDown()
  186. self._Screen.Draw()
  187. self._Screen.SwapAndShow()
  188. def Draw(self):
  189. self.ClearCanvas()
  190. if len(self._MyList) == 0:
  191. return
  192. else:
  193. if len(self._MyList) * PageListItem._Height > self._Height:
  194. self._Ps._Width = self._Width - 11
  195. self._Ps.Draw()
  196. for i in self._MyList:
  197. if i._PosY > self._Height + self._Height/2:
  198. break
  199. if i._PosY < 0:
  200. continue
  201. i.Draw()
  202. self._Scroller.UpdateSize( len(self._MyList)*PageListItem._Height, self._PsIndex*PageListItem._Height)
  203. self._Scroller.Draw()
  204. else:
  205. self._Ps._Width = self._Width
  206. self._Ps.Draw()
  207. for i in self._MyList:
  208. if i._PosY > self._Height + self._Height/2:
  209. break
  210. if i._PosY < 0:
  211. continue
  212. i.Draw()
  213. if self._HWND != None:
  214. self._HWND.fill(MySkinManager.GiveColor("White"))
  215. self._HWND.blit(self._CanvasHWND,(self._PosX,self._PosY,self._Width, self._Height ) )
  216. class APIOBJ(object):
  217. _Page = None
  218. def __init__(self):
  219. pass
  220. def Init(self,main_screen):
  221. self._Page = GPUDriverPage()
  222. self._Page._Screen = main_screen
  223. self._Page._Name ="GPU Driver Switch"
  224. self._Page.Init()
  225. def API(self,main_screen):
  226. if main_screen !=None:
  227. main_screen.PushPage(self._Page)
  228. main_screen.Draw()
  229. main_screen.SwapAndShow()
  230. OBJ = APIOBJ()
  231. def Init(main_screen):
  232. OBJ.Init(main_screen)
  233. def API(main_screen):
  234. OBJ.API(main_screen)