title_bar.py 10 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290
  1. # -*- coding: utf-8 -*-
  2. import pygame
  3. import os
  4. import sys
  5. from datetime import datetime
  6. #from beeprint import pp
  7. import alsaaudio
  8. ##local import
  9. from constants import ICON_TYPES,Width,Height
  10. from fonts import fonts
  11. from icon_item import IconItem
  12. from multi_icon_item import MultiIconItem
  13. from util_funcs import midRect,SwapAndShow
  14. from config import Battery
  15. from libs.roundrects import aa_round_rect
  16. from libs.DBUS import is_wifi_connected_now,wifi_strength
  17. icon_base_path = "gameshell/titlebar_icons/"
  18. class TitleBar:
  19. _PosX = 0
  20. _PosY = 0
  21. _Width = Width
  22. _Height = 25
  23. _IconColor = pygame.Color(114,114,144)
  24. _BarHeight = 24.5
  25. _LOffset = 3
  26. _ROffset = 3
  27. _BgColor = pygame.Color(228,228,228)
  28. _TxtColor = pygame.Color(83,83,83)
  29. _BottomLineColor = pygame.Color(169,169,169)
  30. _Icons = {}
  31. _icon_width = 18
  32. _icon_height = 18
  33. _BorderWidth = 1
  34. _CanvasHWND = None
  35. _HWND = None
  36. _Title = ""
  37. _InLowBackLight = -1
  38. def __init__(self):
  39. self._Icons = {}
  40. def GObjectRoundRobin(self):
  41. if self._InLowBackLight < 0:
  42. self.CheckBatteryStat()
  43. self.SyncSoundVolume()
  44. self.UpdateWifiStrength()
  45. SwapAndShow()
  46. else:
  47. self._InLowBackLight+=1
  48. if self._InLowBackLight > 10:
  49. self.CheckBatteryStat()
  50. self.SyncSoundVolume()
  51. self.UpdateWifiStrength()
  52. SwapAndShow()
  53. self._InLowBackLight = 0
  54. return True
  55. def UpdateWifiStrength(self):
  56. self.Draw(self._Title)
  57. def GetWifiStrength(self,stren): ##invoke anytime to change title bar icon status
  58. #0-25
  59. #25-50
  60. #50-75
  61. #75-100
  62. segs = [[-2,-1], [0,25],[25,50],[50,75],[75,100] ]
  63. stren = int(stren)
  64. ge = 0
  65. if stren == 0:
  66. return ge
  67. for i,v in enumerate(segs):
  68. if stren >= v[0] and stren <= v[1]:
  69. ge = i
  70. break
  71. return ge
  72. def SyncSoundVolume(self):
  73. m = alsaaudio.Mixer()
  74. vol = m.getvolume()[0]
  75. snd_segs = [ [0,10],[10,30],[30,70],[70,100] ]
  76. ge = 0
  77. for i,v in enumerate(snd_segs):
  78. if vol >= v[0] and vol <= v[1]:
  79. ge = i
  80. break
  81. self._Icons["soundvolume"]._IconIndex = ge
  82. self._Icons["sound"] = self._Icons["soundvolume"]
  83. def SetSoundVolume(self,vol):
  84. pass
  85. def CheckBatteryStat(self): ## 100ms check interval
  86. content = ""
  87. bat_uevent = {}
  88. bat_segs=[ [0,6],[7,15],[16,20], [21,30],[31,50],[51,60], [61,80],[81,90],[91,100] ]
  89. try:
  90. f = open(Battery)
  91. except IOError:
  92. self._Icons["battery"] = self._Icons["battery_unknown"]
  93. print("CheckBatteryStat open failed")
  94. return False
  95. else:
  96. with f:
  97. content = f.readlines()
  98. content = [x.strip() for x in content]
  99. f.close()
  100. for i in content:
  101. pis = i.split("=")
  102. if len(pis) > 1:
  103. bat_uevent[pis[0]] = pis[1]
  104. # print(bat_uevent["POWER_SUPPLY_CAPACITY"])
  105. """
  106. power_current = int(bat_uevent["POWER_SUPPLY_CURRENT_NOW"])
  107. if power_current < 270000:
  108. self._Icons["battery"] = self._Icons["battery_unknown"]
  109. return False
  110. """
  111. if "POWER_SUPPLY_CAPACITY" in bat_uevent:
  112. cur_cap = int(bat_uevent["POWER_SUPPLY_CAPACITY"])
  113. else:
  114. cur_cap = 0
  115. cap_ge = 0
  116. for i,v in enumerate(bat_segs):
  117. if cur_cap >= v[0] and cur_cap <= v[1]:
  118. cap_ge = i
  119. break
  120. if bat_uevent["POWER_SUPPLY_STATUS"] == "Charging":
  121. self._Icons["battery_charging"]._IconIndex = cap_ge
  122. self._Icons["battery"] = self._Icons["battery_charging"]
  123. print("Charging %d" % cap_ge)
  124. else:
  125. self._Icons["battery_discharging"]._IconIndex = cap_ge
  126. self._Icons["battery"] = self._Icons["battery_discharging"]
  127. print("Discharging %d" % cap_ge)
  128. return True
  129. def SetBatteryStat(self,bat):
  130. pass
  131. def Init(self,screen):
  132. start_x = 0
  133. self._CanvasHWND = pygame.Surface((self._Width,self._Height))
  134. self._HWND = screen
  135. icon_snd = IconItem()
  136. icon_snd._MyType = ICON_TYPES["STAT"]
  137. icon_snd._Parent = self
  138. icon_snd._ImageName = icon_base_path+"ic_volume_up_black.png"
  139. icon_snd.Adjust(start_x,self._icon_height/2+(self._BarHeight-self._icon_height)/2,self._icon_width,self._icon_height,0)
  140. icon_snd.ChangeImgSurfColor(self._IconColor)
  141. icon_wifi_status = MultiIconItem()
  142. icon_wifi_status._MyType = ICON_TYPES["STAT"]
  143. icon_wifi_status._ImageName = icon_base_path+"wifi.png"
  144. icon_wifi_status._Parent = self
  145. icon_wifi_status.Adjust(start_x+self._icon_width+5,self._icon_height/2+(self._BarHeight-self._icon_height)/2,self._icon_width,self._icon_height,0)
  146. self._Icons["wifistatus"] = icon_wifi_status
  147. battery_charging = MultiIconItem()
  148. battery_charging._MyType = ICON_TYPES["STAT"]
  149. battery_charging._Parent = self
  150. battery_charging._ImageName = icon_base_path+"withcharging.png"
  151. battery_charging.Adjust(start_x+self._icon_width+self._icon_width+8,self._icon_height/2+(self._BarHeight-self._icon_height)/2,self._icon_width,self._icon_height,0)
  152. self._Icons["battery_charging"] = battery_charging
  153. battery_discharging = MultiIconItem()
  154. battery_discharging._MyType = ICON_TYPES["STAT"]
  155. battery_discharging._Parent = self
  156. battery_discharging._ImageName = icon_base_path+"without_charging.png"
  157. battery_discharging.Adjust(start_x+self._icon_width+self._icon_width+8,self._icon_height/2+(self._BarHeight-self._icon_height)/2,self._icon_width,self._icon_height,0)
  158. self._Icons["battery_discharging"] = battery_discharging
  159. battery_unknown = IconItem()
  160. battery_unknown._MyType = ICON_TYPES["STAT"]
  161. battery_unknown._Parent = self
  162. battery_unknown._ImageName = icon_base_path+"battery_unknown.png"
  163. battery_unknown.Adjust(start_x+self._icon_width+self._icon_width+8,self._icon_height/2+(self._BarHeight-self._icon_height)/2,self._icon_width,self._icon_height,0)
  164. self._Icons["battery_unknown"] = battery_unknown
  165. self.CheckBatteryStat()
  166. sound_volume = MultiIconItem()
  167. sound_volume._MyType = ICON_TYPES["STAT"]
  168. sound_volume._Parent = self
  169. sound_volume._ImageName = icon_base_path+"soundvolume.png"
  170. sound_volume.Adjust(start_x+self._icon_width+self._icon_width+8,self._icon_height/2+(self._BarHeight-self._icon_height)/2,self._icon_width,self._icon_height,0)
  171. self._Icons["soundvolume"] = sound_volume
  172. self.SyncSoundVolume()
  173. if is_wifi_connected_now():
  174. print("wifi is connected")
  175. print( wifi_strength())
  176. def ClearCanvas(self):
  177. self._CanvasHWND.fill((0,0,0))
  178. aa_round_rect(self._CanvasHWND,
  179. (0,0,self._Width,self._Height),self._BgColor,8,0, self._BgColor)
  180. pygame.draw.rect(self._CanvasHWND,self._BgColor,(0,self._Height/2,Width,self._BarHeight), 0 )
  181. def Draw(self,title):
  182. self.ClearCanvas()
  183. self._Title = title
  184. cur_time = datetime.now().strftime("%H:%M")
  185. time_text_size = fonts["varela12"].size(cur_time)
  186. title_text_size = fonts["varela16"].size(title)
  187. self._CanvasHWND.blit(fonts["varela16"].render(title,True,self._TxtColor),midRect(title_text_size[0]/2+self._LOffset,
  188. title_text_size[1]/2+(self._BarHeight-title_text_size[1])/2,
  189. title_text_size[0],title_text_size[1],Width,Height))
  190. self._CanvasHWND.blit(fonts["varela12"].render(cur_time,True,self._TxtColor),midRect(Width-time_text_size[0]/2-self._ROffset,
  191. time_text_size[1]/2+(self._BarHeight-time_text_size[1])/2,
  192. time_text_size[0],time_text_size[1],Width,Height))
  193. start_x = Width-time_text_size[0]-self._ROffset-self._icon_width*3 # near by the time_text
  194. self._Icons["sound"].NewCoord(start_x, self._icon_height/2+(self._BarHeight-self._icon_height)/2)
  195. #self._Icons["wifi"].NewCoord(start_x+self._icon_width+5, self._icon_height/2+(self._BarHeight-self._icon_height)/2)
  196. self._Icons["battery"].NewCoord(start_x+self._icon_width+self._icon_width+8,self._icon_height/2+(self._BarHeight-self._icon_height)/2)
  197. if is_wifi_connected_now():
  198. ge = self.GetWifiStrength(wifi_strength())
  199. if ge > 0:
  200. self._Icons["wifistatus"]._IconIndex = ge
  201. self._Icons["wifistatus"].NewCoord(start_x+self._icon_width+5,self._icon_height/2+(self._BarHeight-self._icon_height)/2)
  202. self._Icons["wifistatus"].Draw()
  203. else:
  204. self._Icons["wifistatus"]._IconIndex = 0
  205. self._Icons["wifistatus"].Draw()
  206. print("strength error")
  207. else:
  208. self._Icons["wifistatus"]._IconIndex = 0
  209. self._Icons["wifistatus"].NewCoord(start_x+self._icon_width+5,self._icon_height/2+(self._BarHeight-self._icon_height)/2)
  210. self._Icons["wifistatus"].Draw()
  211. self._Icons["sound"].Draw()
  212. self._Icons["battery"].Draw()
  213. pygame.draw.line(self._CanvasHWND,self._BottomLineColor,(0,self._BarHeight),(self._Width,self._BarHeight),self._BorderWidth)
  214. if self._HWND != None:
  215. self._HWND.blit(self._CanvasHWND,(self._PosX,self._PosY,self._Width,self._Height))