__init__.py 16 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573
  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. server_saving = {}
  254. server_saving["key"] = "server"
  255. server_saving["label"] = "Server"
  256. server_saving["value"] = "server"
  257. self._AList["supersaving"] = supersaving
  258. self._AList["powersaving"] = powersaving
  259. self._AList["server"] = server_saving
  260. self._AList["balance_saving"] = balance_saving
  261. for i,u in enumerate( ["supersaving","powersaving","server","balance_saving"] ):
  262. if u not in self._AList:
  263. continue
  264. v = self._AList[u]
  265. li = PageListItem()
  266. li._Parent = self
  267. li._PosX = start_x
  268. li._PosY = start_y + last_height
  269. li._Width = Width
  270. li._Fonts["normal"] = self._ListFont
  271. li._Active = False
  272. li._Value = self._AList[u]["value"]
  273. if self._AList[u]["label"] != "":
  274. li.Init( self._AList[u]["label"] )
  275. else:
  276. li.Init( self._AList[u]["key"] )
  277. last_height += li._Height
  278. self._MyList.append(li)
  279. def Init(self):
  280. if self._Screen != None:
  281. if self._Screen._CanvasHWND != None and self._CanvasHWND == None:
  282. self._HWND = self._Screen._CanvasHWND
  283. self._CanvasHWND = pygame.Surface( (self._Screen._Width,self._BGheight) )
  284. self._PosX = self._Index*self._Screen._Width
  285. self._Width = self._Screen._Width ## equal to screen width
  286. self._Height = self._Screen._Height
  287. done = IconItem()
  288. done._ImgSurf = MyIconPool._Icons["done"]
  289. done._MyType = ICON_TYPES["STAT"]
  290. done._Parent = self
  291. self._Icons["done"] = done
  292. ps = ListPageSelector()
  293. ps._Parent = self
  294. self._Ps = ps
  295. self._PsIndex = 0
  296. self.GenList()
  297. self._Scroller = ListScroller()
  298. self._Scroller._Parent = self
  299. self._Scroller._PosX = self._Width - 10
  300. self._Scroller._PosY = 2
  301. self._Scroller.Init()
  302. self._Scroller.SetCanvasHWND(self._HWND)
  303. self._InfoPage = InfoPage()
  304. self._InfoPage._Screen = self._Screen
  305. self._InfoPage._Name = "Power option detail"
  306. self._InfoPage.Init()
  307. def ScrollDown(self):
  308. if len(self._MyList) == 0:
  309. return
  310. self._PsIndex +=1
  311. if self._PsIndex >= len(self._MyList):
  312. self._PsIndex = len(self._MyList) -1
  313. cur_li = self._MyList[self._PsIndex]
  314. if cur_li._PosY +cur_li._Height > self._Height:
  315. for i in range(0,len(self._MyList)):
  316. self._MyList[i]._PosY -= self._MyList[i]._Height
  317. def ScrollUp(self):
  318. if len(self._MyList) == 0:
  319. return
  320. self._PsIndex -= 1
  321. if self._PsIndex < 0:
  322. self._PsIndex = 0
  323. cur_li = self._MyList[self._PsIndex]
  324. if cur_li._PosY < 0:
  325. for i in range(0, len(self._MyList)):
  326. self._MyList[i]._PosY += self._MyList[i]._Height
  327. def Click(self):
  328. if len(self._MyList) == 0:
  329. return
  330. cur_li = self._MyList[self._PsIndex]
  331. if cur_li._Active == True:
  332. return
  333. for i in self._MyList:
  334. i._Active = False
  335. cur_li._Active = True
  336. print(cur_li._Value)
  337. with open(".powerlevel","w") as f:
  338. f.write(cur_li._Value)
  339. config.PowerLevel = cur_li._Value
  340. self._Screen._MsgBox.SetText("Applying...")
  341. self._Screen._MsgBox.Draw()
  342. self._Screen.SwapAndShow()
  343. pygame.event.post( pygame.event.Event(POWEROPT, message=""))
  344. pygame.time.delay(1000)
  345. self._Screen.Draw()
  346. self._Screen.SwapAndShow()
  347. def OnLoadCb(self):
  348. self._Scrolled = 0
  349. self._PosY = 0
  350. self._DrawOnce = False
  351. with open(".powerlevel", "r") as f:
  352. powerlevel = f.read()
  353. powerlevel = powerlevel.strip()
  354. if powerlevel == "":
  355. powerlevel = "balance_saving"
  356. for i in self._MyList:
  357. if i._Value == powerlevel:
  358. i._Active = True
  359. def OnReturnBackCb(self):
  360. pass
  361. """
  362. self.ReturnToUpLevelPage()
  363. self._Screen.Draw()
  364. self._Screen.SwapAndShow()
  365. """
  366. def KeyDown(self,event):
  367. if event.key == CurKeys["A"] or event.key == CurKeys["Menu"]:
  368. self.ReturnToUpLevelPage()
  369. self._Screen.Draw()
  370. self._Screen.SwapAndShow()
  371. if event.key == CurKeys["B"]:
  372. self.Click()
  373. if event.key == CurKeys["Up"]:
  374. self.ScrollUp()
  375. self._Screen.Draw()
  376. self._Screen.SwapAndShow()
  377. if event.key == CurKeys["Down"]:
  378. self.ScrollDown()
  379. self._Screen.Draw()
  380. self._Screen.SwapAndShow()
  381. if event.key == CurKeys["Y"]:
  382. cur_li = self._MyList[self._PsIndex]
  383. time1 = config.PowerLevels[cur_li._Value][0]
  384. time2 = config.PowerLevels[cur_li._Value][1]
  385. time3 = config.PowerLevels[cur_li._Value][2]
  386. self._InfoPage._Time1 = time1
  387. self._InfoPage._Time2 = time2
  388. self._InfoPage._Time3 = time3
  389. self._Screen.PushPage(self._InfoPage)
  390. self._Screen.Draw()
  391. self._Screen.SwapAndShow()
  392. def Draw(self):
  393. self.ClearCanvas()
  394. if len(self._MyList) == 0:
  395. return
  396. else:
  397. if len(self._MyList) * PageListItem._Height > self._Height:
  398. self._Ps._Width = self._Width - 11
  399. self._Ps.Draw()
  400. for i in self._MyList:
  401. if i._PosY > self._Height + self._Height/2:
  402. break
  403. if i._PosY < 0:
  404. continue
  405. i.Draw()
  406. self._Scroller.UpdateSize( len(self._MyList)*PageListItem._Height, self._PsIndex*PageListItem._Height)
  407. self._Scroller.Draw()
  408. else:
  409. self._Ps._Width = self._Width
  410. self._Ps.Draw()
  411. for i in self._MyList:
  412. if i._PosY > self._Height + self._Height/2:
  413. break
  414. if i._PosY < 0:
  415. continue
  416. i.Draw()
  417. if self._HWND != None:
  418. self._HWND.fill((255,255,255))
  419. self._HWND.blit(self._CanvasHWND,(self._PosX,self._PosY,self._Width, self._Height ) )
  420. class APIOBJ(object):
  421. _Page = None
  422. def __init__(self):
  423. pass
  424. def Init(self,main_screen):
  425. self._Page = PowerOptionsPage()
  426. self._Page._Screen = main_screen
  427. self._Page._Name ="Power Options"
  428. self._Page.Init()
  429. def API(self,main_screen):
  430. if main_screen !=None:
  431. main_screen.PushPage(self._Page)
  432. main_screen.Draw()
  433. main_screen.SwapAndShow()
  434. OBJ = APIOBJ()
  435. def Init(main_screen):
  436. OBJ.Init(main_screen)
  437. def API(main_screen):
  438. OBJ.API(main_screen)