__init__.py 16 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556
  1. # -*- coding: utf-8 -*-
  2. import os
  3. import pygame
  4. #import math
  5. #mport subprocess
  6. #from beeprint import pp
  7. from libs.roundrects import aa_round_rect
  8. #import gobject
  9. #from wicd import misc
  10. ## local UI import
  11. from UI.constants import Width,Height,ICON_TYPES,POWEROPT
  12. from UI.page import Page,PageSelector
  13. from UI.label import Label
  14. from UI.fonts import fonts
  15. from UI.util_funcs import midRect
  16. from UI.keys_def import CurKeys
  17. from UI.scroller import ListScroller
  18. from UI.icon_pool import MyIconPool
  19. from UI.icon_item import IconItem
  20. from UI.multilabel import MultiLabel
  21. from UI.skin_manager import MySkinManager
  22. from UI.lang_manager import MyLangManager
  23. import config
  24. class ListPageSelector(PageSelector):
  25. _BackgroundColor = MySkinManager.GiveColor('Front')
  26. def __init__(self):
  27. self._PosX = 0
  28. self._PosY = 0
  29. self._Height = 0
  30. self._Width = Width
  31. def AnimateDraw(self,x2,y2):
  32. pass
  33. def Draw(self):
  34. idx = self._Parent._PsIndex
  35. if idx < len(self._Parent._MyList):
  36. x = self._Parent._MyList[idx]._PosX+2
  37. y = self._Parent._MyList[idx]._PosY+1
  38. h = self._Parent._MyList[idx]._Height -3
  39. self._PosX = x
  40. self._PosY = y
  41. self._Height = h
  42. aa_round_rect(self._Parent._CanvasHWND,
  43. (x,y,self._Width-4,h),self._BackgroundColor,4,0,self._BackgroundColor)
  44. class PageListItem(object):
  45. _PosX = 0
  46. _PosY = 0
  47. _Width = 0
  48. _Height = 30
  49. _Labels = {}
  50. _Icons = {}
  51. _Fonts = {}
  52. _LinkObj = None
  53. _Active = False
  54. _Value = ""
  55. def __init__(self):
  56. self._Labels = {}
  57. self._Icons = {}
  58. self._Fonts = {}
  59. def SetSmallText(self,text):
  60. l = Label()
  61. l._PosX = 40
  62. l.SetCanvasHWND(self._Parent._CanvasHWND)
  63. l.Init(text,self._Fonts["small"])
  64. self._Labels["Small"] = l
  65. def Init(self,text):
  66. l = Label()
  67. l._PosX = 10
  68. l.SetCanvasHWND(self._Parent._CanvasHWND)
  69. text = MyLangManager.Tr(text)
  70. l.Init(text,self._Fonts["normal"])
  71. self._Labels["Text"] = l
  72. def Draw(self):
  73. self._Labels["Text"]._PosY = self._PosY+ (self._Height- self._Labels["Text"]._Height)/2
  74. if self._Active == True:
  75. self._Parent._Icons["done"].NewCoord( self._Parent._Width-30,self._PosY+5)
  76. self._Parent._Icons["done"].Draw()
  77. self._Labels["Text"].Draw(self._Active)
  78. if "Small" in self._Labels:
  79. self._Labels["Small"]._PosX = self._Width - self._Labels["Small"]._Width -10
  80. self._Labels["Small"]._PosY = self._PosY + (self._Height- self._Labels["Small"]._Height)/2
  81. self._Labels["Small"].Draw()
  82. 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)
  83. class InfoPage(Page):
  84. _FootMsg = ["Nav","","","Back",""]
  85. _MyList = []
  86. _ListFontObj = MyLangManager.TrFont("varela15")
  87. _AList = {}
  88. _Time1 = 40
  89. _Time2 = 120
  90. _Time3 = 300
  91. def ConvertSecToMin(self, secs):
  92. sec_str = ""
  93. min_str = ""
  94. if secs > 60:
  95. m = int(secs/60)
  96. s = secs % 60
  97. if m > 1:
  98. min_str = ("%d "+MyLangManager.Tr("minutes")+" ") % m
  99. else:
  100. min_str = ("%d "+MyLangManager.Tr("minute")+" ") % m
  101. if s == 1:
  102. sec_str = ("%d "+MyLangManager.Tr("second")) % s
  103. elif s > 1:
  104. sec_str = ("%d "+MyLangManager.Tr("seconds")) % s
  105. elif secs > 0:
  106. if secs > 1:
  107. sec_str = ("%d "+MyLangManager.Tr("seconds")) % secs
  108. else:
  109. sec_str = ("%d "+MyLangManager.Tr("second")) % secs
  110. elif secs == 0:
  111. sec_str = MyLangManager.Tr("Never")
  112. return min_str + sec_str
  113. def RefreshList(self):
  114. ## after GenList ,reuse
  115. self._AList["time1"]["value"] = self.ConvertSecToMin(self._Time1)
  116. self._AList["time2"]["value"] = self.ConvertSecToMin(self._Time2)
  117. self._AList["time3"]["value"] = self.ConvertSecToMin(self._Time3)
  118. for i,v in enumerate( self._AList ):
  119. self._MyList[i].SetSmallText( self._AList[v]["value"] )
  120. def GenList(self):
  121. time1 = {}
  122. time1["key"] = "time1"
  123. if self._Time1 == 0:
  124. time1["value"] = MyLangManager.Tr("Never")
  125. else:
  126. time1["value"] = "%d secs" % self._Time1
  127. time1["label"] = "Screen dimming"
  128. time2 = {}
  129. time2["key"] = "time2"
  130. if self._Time2 == 0:
  131. time2["value"] = MyLangManager.Tr("Never")
  132. else:
  133. time2["value"] = "%d secs" % self._Time2
  134. time2["label"] = "Screen OFF"
  135. time3 = {}
  136. time3["key"] = "time3"
  137. if self._Time3 == 0:
  138. time3["value"] = MyLangManager.Tr("Never")
  139. else:
  140. time3["value"] = "%d secs" % self._Time3
  141. time3["label"] = "Power OFF"
  142. self._AList["time1"] = time1
  143. self._AList["time2"] = time2
  144. self._AList["time3"] = time3
  145. self._MyList = []
  146. start_x = 0
  147. start_y = 0
  148. for i,v in enumerate( self._AList):
  149. #print(v)
  150. li = PageListItem()
  151. li._Parent = self
  152. li._PosX = start_x
  153. li._PosY = start_y + i*PageListItem._Height
  154. li._Width = Width
  155. li._Fonts["normal"] = self._ListFontObj
  156. li._Fonts["small"] = MyLangManager.TrFont("varela12")
  157. if self._AList[v]["label"] != "":
  158. li.Init( self._AList[v]["label"] )
  159. else:
  160. li.Init( self._AList[v]["key"] )
  161. li._Flag = self._AList[v]["key"]
  162. li.SetSmallText( self._AList[v]["value"] )
  163. self._MyList.append(li)
  164. def Init(self):
  165. if self._Screen != None:
  166. if self._Screen._CanvasHWND != None and self._CanvasHWND == None:
  167. self._CanvasHWND = self._Screen._CanvasHWND
  168. self._PosX = self._Index*self._Screen._Width
  169. self._Width = self._Screen._Width ## equal to screen width
  170. self._Height = self._Screen._Height
  171. ps = ListPageSelector()
  172. ps._Parent = self
  173. self._Ps = ps
  174. self._PsIndex = 0
  175. self.GenList()
  176. def ScrollUp(self):
  177. if len(self._MyList) == 0:
  178. return
  179. self._PsIndex -= 1
  180. if self._PsIndex < 0:
  181. self._PsIndex = 0
  182. cur_li = self._MyList[self._PsIndex]
  183. if cur_li._PosY < 0:
  184. for i in range(0, len(self._MyList)):
  185. self._MyList[i]._PosY += self._MyList[i]._Height
  186. def ScrollDown(self):
  187. if len(self._MyList) == 0:
  188. return
  189. self._PsIndex +=1
  190. if self._PsIndex >= len(self._MyList):
  191. self._PsIndex = len(self._MyList) -1
  192. cur_li = self._MyList[self._PsIndex]
  193. if cur_li._PosY +cur_li._Height > self._Height:
  194. for i in range(0,len(self._MyList)):
  195. self._MyList[i]._PosY -= self._MyList[i]._Height
  196. def Click(self):
  197. cur_li = self._MyList[self._PsIndex]
  198. print(cur_li._Flag)
  199. def OnLoadCb(self):
  200. self.RefreshList()
  201. def OnReturnBackCb(self):
  202. pass
  203. """
  204. self.ReturnToUpLevelPage()
  205. self._Screen.Draw()
  206. self._Screen.SwapAndShow()
  207. """
  208. def KeyDown(self,event):
  209. if event.key == CurKeys["A"] or event.key == CurKeys["Menu"]:
  210. self.ReturnToUpLevelPage()
  211. self._Screen.Draw()
  212. self._Screen.SwapAndShow()
  213. if event.key == CurKeys["Up"]:
  214. self.ScrollUp()
  215. self._Screen.Draw()
  216. self._Screen.SwapAndShow()
  217. if event.key == CurKeys["Down"]:
  218. self.ScrollDown()
  219. self._Screen.Draw()
  220. self._Screen.SwapAndShow()
  221. def Draw(self):
  222. self.ClearCanvas()
  223. self._Ps.Draw()
  224. for i in self._MyList:
  225. i.Draw()
  226. class PowerOptionsPage(Page):
  227. _FootMsg = ["Nav","","Detail","Back","Select"]
  228. _MyList = []
  229. _ListFont = fonts["notosanscjk15"]
  230. _AList = {}
  231. _Scrolled = 0
  232. _BGwidth = 320
  233. _BGheight = 240-24-20
  234. _DrawOnce = False
  235. _Scroller = None
  236. _InfoPage = None
  237. def __init__(self):
  238. Page.__init__(self)
  239. self._Icons = {}
  240. def GenList(self):
  241. self._MyList = []
  242. start_x = 0
  243. start_y = 0
  244. last_height = 0
  245. supersaving = {}
  246. supersaving["key"] = "super"
  247. supersaving["label"] = "Power saving"
  248. supersaving["value"] = "supersaving"
  249. powersaving = {}
  250. powersaving["key"] = "saving"
  251. powersaving["label"] = "Balanced"
  252. powersaving["value"] = "powersaving"
  253. balance_saving = {}
  254. balance_saving["key"] = "balance"
  255. balance_saving["label"] = "Performance"
  256. balance_saving["value"] = "balance_saving"
  257. server_saving = {}
  258. server_saving["key"] = "server"
  259. server_saving["label"] = "Server"
  260. server_saving["value"] = "server"
  261. self._AList["supersaving"] = supersaving
  262. self._AList["powersaving"] = powersaving
  263. self._AList["server"] = server_saving
  264. self._AList["balance_saving"] = balance_saving
  265. for i,u in enumerate( ["supersaving","powersaving","server","balance_saving"] ):
  266. if u not in self._AList:
  267. continue
  268. v = self._AList[u]
  269. li = PageListItem()
  270. li._Parent = self
  271. li._PosX = start_x
  272. li._PosY = start_y + last_height
  273. li._Width = Width
  274. li._Fonts["normal"] = self._ListFont
  275. li._Active = False
  276. li._Value = self._AList[u]["value"]
  277. if self._AList[u]["label"] != "":
  278. li.Init( self._AList[u]["label"] )
  279. else:
  280. li.Init( self._AList[u]["key"] )
  281. last_height += li._Height
  282. self._MyList.append(li)
  283. def Init(self):
  284. if self._Screen != None:
  285. if self._Screen._CanvasHWND != None and self._CanvasHWND == None:
  286. self._HWND = self._Screen._CanvasHWND
  287. self._CanvasHWND = pygame.Surface( (self._Screen._Width,self._BGheight) )
  288. self._PosX = self._Index*self._Screen._Width
  289. self._Width = self._Screen._Width ## equal to screen width
  290. self._Height = self._Screen._Height
  291. done = IconItem()
  292. done._ImgSurf = MyIconPool._Icons["done"]
  293. done._MyType = ICON_TYPES["STAT"]
  294. done._Parent = self
  295. self._Icons["done"] = done
  296. ps = ListPageSelector()
  297. ps._Parent = self
  298. self._Ps = ps
  299. self._PsIndex = 0
  300. self.GenList()
  301. self._Scroller = ListScroller()
  302. self._Scroller._Parent = self
  303. self._Scroller._PosX = self._Width - 10
  304. self._Scroller._PosY = 2
  305. self._Scroller.Init()
  306. self._Scroller.SetCanvasHWND(self._HWND)
  307. self._InfoPage = InfoPage()
  308. self._InfoPage._Screen = self._Screen
  309. self._InfoPage._Name = "Power option detail"
  310. self._InfoPage.Init()
  311. def Click(self):
  312. if len(self._MyList) == 0:
  313. return
  314. cur_li = self._MyList[self._PsIndex]
  315. if cur_li._Active == True:
  316. return
  317. for i in self._MyList:
  318. i._Active = False
  319. cur_li._Active = True
  320. print(cur_li._Value)
  321. with open(".powerlevel","w") as f:
  322. f.write(cur_li._Value)
  323. config.PowerLevel = cur_li._Value
  324. if config.PowerLevel!= "supersaving":
  325. os.system("sudo iw wlan0 set power_save off > /dev/null")
  326. else:
  327. os.system("sudo iw wlan0 set power_save on > /dev/null")
  328. self._Screen._MsgBox.SetText("Applying")
  329. self._Screen._MsgBox.Draw()
  330. self._Screen.SwapAndShow()
  331. pygame.event.post( pygame.event.Event(POWEROPT, message=""))
  332. pygame.time.delay(1000)
  333. self._Screen.Draw()
  334. self._Screen.SwapAndShow()
  335. def OnLoadCb(self):
  336. self._Scrolled = 0
  337. self._PosY = 0
  338. self._DrawOnce = False
  339. with open(".powerlevel", "r") as f:
  340. powerlevel = f.read()
  341. powerlevel = powerlevel.strip()
  342. if powerlevel == "":
  343. powerlevel = "balance_saving"
  344. for i in self._MyList:
  345. if i._Value == powerlevel:
  346. i._Active = True
  347. def OnReturnBackCb(self):
  348. pass
  349. """
  350. self.ReturnToUpLevelPage()
  351. self._Screen.Draw()
  352. self._Screen.SwapAndShow()
  353. """
  354. def KeyDown(self,event):
  355. if event.key == CurKeys["A"] or event.key == CurKeys["Menu"]:
  356. self.ReturnToUpLevelPage()
  357. self._Screen.Draw()
  358. self._Screen.SwapAndShow()
  359. if event.key == CurKeys["B"]:
  360. self.Click()
  361. if event.key == CurKeys["Up"]:
  362. self.ScrollUp()
  363. self._Screen.Draw()
  364. self._Screen.SwapAndShow()
  365. if event.key == CurKeys["Down"]:
  366. self.ScrollDown()
  367. self._Screen.Draw()
  368. self._Screen.SwapAndShow()
  369. if event.key == CurKeys["Y"]:
  370. cur_li = self._MyList[self._PsIndex]
  371. time1 = config.PowerLevels[cur_li._Value][0]
  372. time2 = config.PowerLevels[cur_li._Value][1]
  373. time3 = config.PowerLevels[cur_li._Value][2]
  374. self._InfoPage._Time1 = time1
  375. self._InfoPage._Time2 = time2
  376. self._InfoPage._Time3 = time3
  377. self._Screen.PushPage(self._InfoPage)
  378. self._Screen.Draw()
  379. self._Screen.SwapAndShow()
  380. def Draw(self):
  381. self.ClearCanvas()
  382. if len(self._MyList) == 0:
  383. return
  384. else:
  385. if len(self._MyList) * PageListItem._Height > self._Height:
  386. self._Ps._Width = self._Width - 11
  387. self._Ps.Draw()
  388. for i in self._MyList:
  389. if i._PosY > self._Height + self._Height/2:
  390. break
  391. if i._PosY < 0:
  392. continue
  393. i.Draw()
  394. self._Scroller.UpdateSize( len(self._MyList)*PageListItem._Height, self._PsIndex*PageListItem._Height)
  395. self._Scroller.Draw()
  396. else:
  397. self._Ps._Width = self._Width
  398. self._Ps.Draw()
  399. for i in self._MyList:
  400. if i._PosY > self._Height + self._Height/2:
  401. break
  402. if i._PosY < 0:
  403. continue
  404. i.Draw()
  405. if self._HWND != None:
  406. self._HWND.fill((255,255,255))
  407. self._HWND.blit(self._CanvasHWND,(self._PosX,self._PosY,self._Width, self._Height ) )
  408. class APIOBJ(object):
  409. _Page = None
  410. def __init__(self):
  411. pass
  412. def Init(self,main_screen):
  413. self._Page = PowerOptionsPage()
  414. self._Page._Screen = main_screen
  415. self._Page._Name ="Power Options"
  416. self._Page.Init()
  417. def API(self,main_screen):
  418. if main_screen !=None:
  419. main_screen.PushPage(self._Page)
  420. main_screen.Draw()
  421. main_screen.SwapAndShow()
  422. OBJ = APIOBJ()
  423. def Init(main_screen):
  424. OBJ.Init(main_screen)
  425. def API(main_screen):
  426. OBJ.API(main_screen)