__init__.py 16 KB

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