__init__.py 16 KB

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