__init__.py 16 KB

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