title_bar.go 9.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374
  1. package UI
  2. import (
  3. "fmt"
  4. "os"
  5. "log"
  6. "strconv"
  7. "bufio"
  8. "strings"
  9. "time"
  10. "github.com/veandco/go-sdl2/sdl"
  11. "github.com/veandco/go-sdl2/ttf"
  12. "github.com/cuu/gogame/surface"
  13. "github.com/cuu/gogame/rect"
  14. "github.com/cuu/gogame/font"
  15. "github.com/cuu/gogame/draw"
  16. "github.com/itchyny/volume-go"
  17. "github.com/vjeantet/jodaTime"
  18. "../DBUS"
  19. "../../sysgo"
  20. )
  21. var TitleBar_BarHeight = 24
  22. type TitleBarIconItem struct {
  23. MultiIconItem
  24. Parent *TitleBar
  25. }
  26. func NewTitleBarIconItem() *TitleBarIconItem {
  27. m := &TitleBarIconItem{}
  28. return m
  29. }
  30. func (self *TitleBarIconItem) Draw() {
  31. parent_x,parent_y := self.Parent.PosX,self.Parent.PosY
  32. if self.Label != nil {
  33. // lab_x,lab_y := self.Label.Coord()
  34. lab_w,lab_h:= self.Label.Size()
  35. if self.Align == ALIGN["VCenter"] {
  36. self.Label.NewCoord( self.PosX - lab_w/2 + parent_x, self.PosY + self.Height/2+6 + parent_y)
  37. }else if self.Align == ALIGN["HLeft"] {
  38. self.Label.NewCoord( self.PosX + self.Width/2+3 + parent_x, self.PosY - lab_h/2 + parent_y )
  39. }
  40. self.Label.Draw()
  41. }
  42. if self.ImgSurf != nil {
  43. portion := rect.Rect(0,self.IconIndex*self.IconHeight,self.IconWidth,self.IconHeight)
  44. surface.Blit(self.Parent.CanvasHWND,
  45. self.ImgSurf,draw.MidRect(self.PosX + parent_x, self.PosY + parent_y,
  46. self.Width,self.Height, Width, Height),&portion)
  47. }
  48. }
  49. type TitleBar struct {
  50. PosX int
  51. PosY int
  52. Width int
  53. Height int
  54. BarHeight int
  55. LOffset int
  56. ROffset int
  57. Icons map[string]IconItemInterface
  58. IconWidth int
  59. IconHeight int
  60. BorderWidth int
  61. CanvasHWND *sdl.Surface
  62. HWND *sdl.Surface
  63. Title string
  64. InLowBackLight int
  65. SkinManager *SkinManager //set by MainScreen
  66. DBusManager DBUS.DBusInterface
  67. icon_base_path string /// SkinMap("gameshell/titlebar_icons/")
  68. TitleFont *ttf.Font
  69. TimeFont *ttf.Font
  70. }
  71. func NewTitleBar() *TitleBar {
  72. t := &TitleBar{}
  73. t.BorderWidth = 1
  74. t.BarHeight = TitleBar_BarHeight
  75. t.Height = t.BarHeight + t.BorderWidth
  76. t.Width = Width
  77. t.IconWidth = 18
  78. t.IconHeight = 18
  79. t.LOffset = 3
  80. t.ROffset = 3
  81. t.Icons = make(map[string]IconItemInterface)
  82. t.icon_base_path = SkinMap("gameshell/titlebar_icons/")
  83. t.TitleFont = Fonts["varela12"]
  84. t.TimeFont = Fonts["varela16"]
  85. return t
  86. }
  87. func (t *TitleBar) RoundRobinCheck() {
  88. }
  89. func (t *TitleBar) UpdateWifiStrength() {
  90. }
  91. func (t *TitleBar) GetWifiStrength(stren int) int {
  92. segs := [][]int{ []int{-2,-1}, []int{0,25}, []int{25,50}, []int{50,75},[]int{75,100}}
  93. stren_number := stren
  94. ge := 0
  95. if stren_number == 0 {
  96. return ge
  97. }
  98. for i,v := range segs {
  99. if stren_number >= v[0] && stren_number <= v[1] {
  100. ge = i
  101. break
  102. }
  103. }
  104. return ge
  105. }
  106. func (self *TitleBar) SyncSoundVolume() {
  107. vol, err := volume.GetVolume()
  108. if err != nil {
  109. log.Fatalf("get volume failed: %+v", err)
  110. vol = 0
  111. }
  112. fmt.Printf("current volume: %d\n", vol)
  113. snd_segs := [][]int{ []int{0,10}, []int{10,30}, []int{30,70},[]int{70,100} }
  114. ge := 0
  115. for i,v := range snd_segs {
  116. if vol >= v[0] && vol <= v[1] {
  117. ge = i
  118. break
  119. }
  120. }
  121. self.Icons["soundvolume"].SetIconIndex(ge)
  122. self.Icons["sound"] = self.Icons["soundvolume"]
  123. //
  124. }
  125. func (t *TitleBar) SetSoundVolume(vol int) {
  126. //pass
  127. }
  128. func (self *TitleBar) CheckBatteryStat() {
  129. bat_segs:= [][]int{[]int{0,6},[]int{7,15},[]int{16,20},[]int{21,30},[]int{31,50},[]int{51,60},[]int{61,80},[]int{81,90},[]int{91,100}}
  130. file, err := os.Open( sysgo.Battery )
  131. if err != nil {
  132. fmt.Println("Could not open file ", sysgo.Battery)
  133. self.Icons["battery"] = self.Icons["battery_unknown"]
  134. return
  135. }
  136. defer file.Close()
  137. bat_uevent := make(map[string]string)
  138. scanner := bufio.NewScanner(file)
  139. scanner.Split(bufio.ScanLines)
  140. for scanner.Scan() {
  141. line := scanner.Text()
  142. line = strings.Trim(line," ")
  143. pis := strings.Split(line,"=")
  144. if len(pis) > 1 {
  145. bat_uevent[ pis[0] ] = pis[1]
  146. }
  147. }
  148. cur_cap := 0
  149. if val, ok := bat_uevent["POWER_SUPPLY_CAPACITY"]; ok {
  150. cur_cap,_ = strconv.Atoi(val)
  151. }else {
  152. cur_cap = 0
  153. }
  154. cap_ge := 0
  155. for i,v := range bat_segs {
  156. if cur_cap >= v[0] && cur_cap <= v[1] {
  157. cap_ge = i
  158. break
  159. }
  160. }
  161. if val, ok := bat_uevent["POWER_SUPPLY_STATUS"]; ok {
  162. if val == "Charging" {
  163. self.Icons["battery_charging"].SetIconIndex(cap_ge)
  164. self.Icons["battery"] = self.Icons["battery_charging"]
  165. }else {
  166. self.Icons["battery_charging"].SetIconIndex(cap_ge)
  167. self.Icons["battery"] = self.Icons["battery_discharging"]
  168. }
  169. }
  170. }
  171. func (self *TitleBar) SetBatteryStat( bat int) {
  172. }
  173. func (self *TitleBar) Init(main_screen *MainScreen) {
  174. start_x := 0
  175. self.CanvasHWND = surface.Surface(self.Width,self.Height)
  176. self.HWND = main_screen.HWND
  177. self.SkinManager = main_screen.SkinManager
  178. self.DBusManager = main_screen.DBusManager
  179. icon_wifi_status := NewTitleBarIconItem()
  180. icon_wifi_status.MyType = ICON_TYPES["STAT"]
  181. icon_wifi_status.ImageName = self.icon_base_path+"wifi.png"
  182. icon_wifi_status.Parent = self
  183. icon_wifi_status.Adjust(start_x+self.IconWidth+5,self.IconHeight/2+(self.BarHeight-self.IconHeight)/2,self.IconWidth,self.IconHeight,0)
  184. self.Icons["wifistatus"] = icon_wifi_status
  185. battery_charging := NewTitleBarIconItem()
  186. battery_charging.MyType = ICON_TYPES["STAT"]
  187. battery_charging.Parent = self
  188. battery_charging.ImageName = self.icon_base_path+"withcharging.png"
  189. battery_charging.Adjust(start_x+self.IconWidth+self.IconWidth+8,self.IconHeight/2+(self.BarHeight-self.IconHeight)/2,self.IconWidth,self.IconHeight,0)
  190. self.Icons["battery_charging"] = battery_charging
  191. battery_discharging := NewTitleBarIconItem()
  192. battery_discharging.MyType = ICON_TYPES["STAT"]
  193. battery_discharging.Parent = self
  194. battery_discharging.ImageName = self.icon_base_path+"without_charging.png"
  195. battery_discharging.Adjust(start_x+self.IconWidth+self.IconWidth+8,self.IconHeight/2+(self.BarHeight-self.IconHeight)/2,self.IconWidth,self.IconHeight,0)
  196. self.Icons["battery_discharging"] = battery_discharging
  197. battery_unknown := NewTitleBarIconItem()
  198. battery_unknown.MyType = ICON_TYPES["STAT"]
  199. battery_unknown.Parent = self
  200. battery_unknown.ImageName = self.icon_base_path+"battery_unknown.png"
  201. battery_unknown.Adjust(start_x+self.IconWidth+self.IconWidth+8,self.IconHeight/2+(self.BarHeight-self.IconHeight)/2,self.IconWidth,self.IconHeight,0)
  202. self.Icons["battery_unknown"] = battery_unknown
  203. self.CheckBatteryStat()
  204. sound_volume := NewTitleBarIconItem()
  205. sound_volume.MyType = ICON_TYPES["STAT"]
  206. sound_volume.Parent = self
  207. sound_volume.ImageName = self.icon_base_path+"soundvolume.png"
  208. sound_volume.Adjust(start_x+self.IconWidth+self.IconWidth+8,self.IconHeight/2+(self.BarHeight-self.IconHeight)/2,self.IconWidth,self.IconHeight,0)
  209. self.Icons["soundvolume"] = sound_volume
  210. self.SyncSoundVolume()
  211. round_corners := NewTitleBarIconItem()
  212. round_corners.IconWidth = 10
  213. round_corners.IconHeight = 10
  214. round_corners.MyType = ICON_TYPES["STAT"]
  215. round_corners.Parent = self
  216. round_corners.ImgSurf = MyIconPool.GetImgSurf("roundcorners")
  217. round_corners.Adjust(0,0,10,10,0)
  218. self.Icons["round_corners"] = round_corners
  219. if self.DBusManager.IsWifiConnectedNow() {
  220. print("wifi is connected")
  221. print( self.DBusManager.WifiStrength())
  222. }
  223. }
  224. func (self *TitleBar) ClearCanvas() {
  225. surface.Fill(self.CanvasHWND, self.SkinManager.GiveColor("TitleBg"))
  226. self.Icons["round_corners"].NewCoord(5,5)
  227. self.Icons["round_corners"].SetIconIndex(0)
  228. self.Icons["round_corners"].Draw()
  229. self.Icons["round_corners"].NewCoord(self.Width-5, 5)
  230. self.Icons["round_corners"].SetIconIndex(1)
  231. self.Icons["round_corners"].Draw()
  232. }
  233. func (self *TitleBar) Draw(title string) {
  234. self.ClearCanvas()
  235. self.Title = title
  236. cur_time := jodaTime.Format("HH:mm", time.Now())
  237. time_text_w, time_text_h := font.Size(self.TimeFont, cur_time)
  238. title_text_w, title_text_h := font.Size(self.TitleFont, self.Title)
  239. title_text_surf := font.Render(self.TitleFont, self.Title, true, self.SkinManager.GiveColor("Text"),nil)
  240. surface.Blit(self.CanvasHWND,title_text_surf, draw.MidRect(title_text_w/2+self.LOffset,title_text_h/2+(self.BarHeight-title_text_h)/2,title_text_w,title_text_h,Width,Height),nil)
  241. time_text_surf := font.Render(self.TimeFont, cur_time,true,self.SkinManager.GiveColor("Text"),nil)
  242. surface.Blit(self.CanvasHWND, time_text_surf, draw.MidRect(Width-time_text_w/2-self.ROffset, time_text_h/2+(self.BarHeight-time_text_h)/2, time_text_w,time_text_h,Width,Height),nil)
  243. start_x := Width - time_text_w - self.ROffset - self.IconWidth*3 // close to the time_text
  244. self.Icons["sound"].NewCoord( start_x, self.IconHeight/2+ (self.BarHeight-self.IconHeight)/2)
  245. self.Icons["battery"].NewCoord(start_x+self.IconWidth+self.IconWidth+8, self.IconHeight/2+(self.BarHeight-self.IconHeight)/2)
  246. if self.DBusManager.IsWifiConnectedNow() == true {
  247. ge := self.GetWifiStrength( self.DBusManager.WifiStrength() )
  248. if ge > 0 {
  249. self.Icons["wifistatus"].SetIconIndex(ge)
  250. self.Icons["wifistatus"].NewCoord(start_x+self.IconWidth+5, self.IconHeight/2+(self.BarHeight-self.IconHeight)/2 )
  251. self.Icons["wifistatus"].Draw()
  252. }else {
  253. self.Icons["wifistatus"].SetIconIndex(0)
  254. self.Icons["wifistatus"].Draw()
  255. }
  256. }else {
  257. self.Icons["wifistatus"].SetIconIndex(0)
  258. self.Icons["wifistatus"].NewCoord(start_x+self.IconWidth+5, self.IconHeight/2+(self.BarHeight-self.IconHeight)/2)
  259. self.Icons["wifistatus"].Draw()
  260. }
  261. self.Icons["sound"].Draw()
  262. self.Icons["battery"].Draw()
  263. draw.Line(self.CanvasHWND,self.SkinManager.GiveColor("Line"), 0,self.BarHeight,self.Width,self.BarHeight, self.BorderWidth)
  264. if self.HWND != nil {
  265. rect_ := rect.Rect(self.PosX,self.PosY, self.Width,self.Height)
  266. surface.Blit(self.HWND, self.CanvasHWND, &rect_, nil)
  267. }
  268. }