run.py 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627
  1. # -*- coding: utf-8 -*-
  2. import dbus
  3. import dbus.service
  4. import sys
  5. import commands
  6. import logging
  7. import errno
  8. from wicd import misc
  9. ##misc.to_bool
  10. ##misc.misc.noneToString
  11. ##misc.to_unicode
  12. ##misc.Noneify
  13. from wicd.translations import _
  14. from wicd import wpath
  15. from wicd import dbusmanager
  16. import time
  17. import gobject
  18. import socket
  19. import pygame
  20. from sys import exit
  21. import os
  22. #from beeprint import pp
  23. ########
  24. if getattr(dbus, 'version', (0, 0, 0)) < (0, 80, 0):
  25. import dbus.glib
  26. else:
  27. from dbus.mainloop.glib import DBusGMainLoop
  28. DBusGMainLoop(set_as_default=True)
  29. import config
  30. #local UI import
  31. from UI.constants import Width,Height,icon_width,icon_height,DT,RUNEVT,RUNSYS,ICON_TYPES,POWEROPT,RESTARTUI,RUNSH
  32. from UI.util_funcs import ReplaceSuffix,FileExists, ReadTheFileContent,midRect,color_surface,SwapAndShow,GetExePath,X_center_mouse,ArmSystem
  33. from UI.page import PageStack,PageSelector,Page
  34. from UI.label import Label
  35. from UI.icon_item import IconItem
  36. #from UI.fonts import fonts
  37. from UI.title_bar import TitleBar
  38. from UI.foot_bar import FootBar
  39. from UI.main_screen import MainScreen
  40. from UI.above_all_patch import SoundPatch
  41. from UI.icon_pool import MyIconPool
  42. from UI.createby_screen import CreateByScreen
  43. from UI.skin_manager import MySkinManager
  44. from libs.DBUS import setup_dbus
  45. if not pygame.display.get_init():
  46. pygame.display.init()
  47. if not pygame.font.get_init():
  48. pygame.font.init()
  49. gobject_main_loop = None
  50. sound_patch = None
  51. myscriptname = os.path.basename(os.path.realpath(__file__))
  52. everytime_keydown = time.time()
  53. passout_time_stage = 0
  54. last_brt = -1
  55. gobject_flash_led1 = -1
  56. gobject_flash_led1_counter = 0
  57. Keys = []
  58. crt_screen = None
  59. def gobject_loop():
  60. """
  61. here to receive dbus signal
  62. """
  63. try:
  64. gobject_main_loop.run()
  65. except KeyboardInterrupt:
  66. gobject_main_loop.quit()
  67. exit(-1)
  68. def GobjectFlashLed1(main_screen):
  69. global gobject_flash_led1_counter
  70. if main_screen._Closed == False:
  71. if gobject_flash_led1_counter > 0:
  72. try:
  73. f = open("/proc/driver/led1","w")
  74. except IOError:
  75. print( "open /proc/driver/led1 IOError")
  76. pass
  77. else:
  78. with f:
  79. f.seek(0)
  80. f.write("0")
  81. f.truncate()
  82. f.close()
  83. gobject_flash_led1_counter = 0
  84. return True
  85. gobject_flash_led1_counter+=1
  86. if gobject_flash_led1_counter == 3:
  87. try:
  88. f = open("/proc/driver/led1","w")
  89. except IOError:
  90. print( "open /proc/driver/led1 IOError")
  91. pass
  92. else:
  93. with f:
  94. f.seek(0)
  95. f.write("1")
  96. f.truncate()
  97. f.close()
  98. elif gobject_flash_led1_counter == 5:
  99. try:
  100. f = open("/proc/driver/led1","w")
  101. except IOError:
  102. print( "open /proc/driver/led1 IOError")
  103. pass
  104. else:
  105. with f:
  106. f.seek(0)
  107. f.write("0")
  108. f.truncate()
  109. f.close()
  110. if gobject_flash_led1_counter == 11:
  111. gobject_flash_led1_counter = 1
  112. return True
  113. def RestoreLastBackLightBrightness(main_screen):
  114. global last_brt,passout_time_stage,gobject_flash_led1
  115. passout_time_stage = 0
  116. main_screen._TitleBar._InLowBackLight = -1
  117. main_screen._Closed = False
  118. if last_brt == -1:
  119. return True
  120. try:
  121. f = open(config.BackLight,"r+")
  122. except IOError:
  123. print( "RestoreLastBackLightBrightness open %s failed, try to adjust brightness in Settings" % config.BackLight)
  124. pass
  125. else:
  126. with f:
  127. content = f.readlines()
  128. content = [x.strip() for x in content]
  129. brt=int(content[0])
  130. if brt < last_brt:
  131. f.seek(0)
  132. f.write(str( last_brt ))
  133. f.truncate()
  134. f.close()
  135. last_brt = -1
  136. else:
  137. f.close()
  138. try:
  139. f = open("/proc/driver/led1","w")
  140. except IOError:
  141. print( "open /proc/driver/led1 IOError")
  142. pass
  143. else:
  144. with f:
  145. f.seek(0)
  146. f.write("0")
  147. f.truncate()
  148. f.close()
  149. if main_screen._CounterScreen._Counting==True:
  150. main_screen._CounterScreen.StopCounter()
  151. main_screen.Draw()
  152. main_screen.SwapAndShow()
  153. return False
  154. return True
  155. def InspectionTeam(main_screen):
  156. global everytime_keydown,last_brt,passout_time_stage,gobject_flash_led1
  157. cur_time = time.time()
  158. time_1 = config.PowerLevels[config.PowerLevel][0]
  159. time_2 = config.PowerLevels[config.PowerLevel][1]
  160. time_3 = config.PowerLevels[config.PowerLevel][2]
  161. if cur_time - everytime_keydown > time_1 and passout_time_stage == 0:
  162. print("timeout, dim screen %d" % int(cur_time - everytime_keydown))
  163. try:
  164. f = open(config.BackLight,"r+")
  165. except IOError:
  166. pass
  167. else:
  168. with f:
  169. content = f.readlines()
  170. content = [x.strip() for x in content]
  171. brt=int(content[0])
  172. if brt > 0:
  173. if last_brt < 0:
  174. last_brt = brt ## remember brt for restore
  175. brt = 1
  176. f.seek(0)
  177. f.write(str(brt))
  178. f.truncate()
  179. f.close()
  180. main_screen._TitleBar._InLowBackLight = 0
  181. if time_2 != 0:
  182. passout_time_stage = 1 # next
  183. everytime_keydown = cur_time
  184. elif cur_time - everytime_keydown > time_2 and passout_time_stage == 1:
  185. print("timeout, close screen %d" % int(cur_time - everytime_keydown))
  186. try:
  187. f = open(config.BackLight,"r+")
  188. except IOError:
  189. pass
  190. else:
  191. with f:
  192. brt = 0
  193. f.seek(0)
  194. f.write(str(brt))
  195. f.truncate()
  196. f.close()
  197. main_screen._TitleBar._InLowBackLight = 0
  198. main_screen._Closed = True
  199. if time_3 != 0:
  200. passout_time_stage = 2 # next
  201. everytime_keydown = cur_time
  202. elif cur_time - everytime_keydown > time_3 and passout_time_stage == 2:
  203. print("Power Off counting down")
  204. try:
  205. f = open(config.BackLight,"r+")
  206. except IOError:
  207. pass
  208. else:
  209. with f:
  210. brt = last_brt
  211. f.seek(0)
  212. f.write(str(brt))
  213. f.truncate()
  214. f.close()
  215. main_screen._CounterScreen.Draw()
  216. main_screen._CounterScreen.SwapAndShow()
  217. main_screen._CounterScreen.StartCounter()
  218. main_screen._TitleBar._InLowBackLight = 0
  219. passout_time_stage = 4
  220. return True
  221. def RecordKeyDns(thekey,main_screen):
  222. global Keys,crt_screen
  223. if len(Keys) < 10:
  224. Keys.append(thekey)
  225. else:
  226. Keys = []
  227. Keys.append(thekey)
  228. keys = ''.join(map(str,Keys))
  229. #print(keys)
  230. if keys == "273273274274276276275275106107":##uuddllrrab
  231. crt_screen.Draw()
  232. crt_screen.SwapAndShow()
  233. main_screen._TitleBar._InLowBackLight = 0 ##pause titlebar drawing
  234. return True
  235. return False
  236. def release_self_fds():
  237. fds_flags= ["pipe","socket",".ttf"]
  238. """List process currently open FDs and their target """
  239. if sys.platform != 'linux2':
  240. raise NotImplementedError('Unsupported platform: %s' % sys.platform)
  241. ret = {}
  242. base = '/proc/self/fd'
  243. for num in os.listdir(base):
  244. path = None
  245. try:
  246. path = os.readlink(os.path.join(base, num))
  247. except OSError as err:
  248. # Last FD is always the "listdir" one (which may be closed)
  249. if err.errno != errno.ENOENT:
  250. raise
  251. ret[int(num)] = path
  252. for key in ret:
  253. if ret[key] != None and isinstance(ret[key], str):
  254. for i in fds_flags:
  255. if i in ret[key]:
  256. os.close(key)
  257. break
  258. return ret
  259. def event_process(event,main_screen):
  260. global sound_patch
  261. global everytime_keydown
  262. if event != None:
  263. pygame.event.clear()
  264. if event.type == pygame.ACTIVEEVENT:
  265. print(" ACTIVEEVENT !")
  266. return
  267. if event.type == pygame.QUIT:
  268. exit()
  269. if event.type == RUNEVT:
  270. if config.DontLeave==True:
  271. os.chdir(GetExePath())
  272. os.system( "/bin/sh -c "+event.message)
  273. else:
  274. on_exit_cb = getattr(main_screen,"OnExitCb",None)
  275. if on_exit_cb != None:
  276. if callable( on_exit_cb ):
  277. main_screen.OnExitCb(event)
  278. pygame.quit()
  279. gobject_main_loop.quit()
  280. os.chdir( GetExePath())
  281. endpos = len(event.message)
  282. space_break_pos = endpos
  283. for i in range(0,endpos):
  284. if event.message[i] == "/" and event.message[i-1] == " " and i > 6:
  285. space_break_pos = i-1
  286. break
  287. exec_app_cmd = "cd "+os.path.dirname(event.message[:space_break_pos])+";"
  288. exec_app_cmd += event.message
  289. exec_app_cmd += "; sync & cd "+GetExePath()+"; exec python "+myscriptname
  290. print(exec_app_cmd)
  291. release_self_fds()
  292. os.execlp("/bin/sh","/bin/sh","-c", exec_app_cmd)
  293. os.chdir( GetExePath())
  294. os.exelp("python","python"," "+myscriptname)
  295. sys.exit(-1)
  296. return
  297. if event.type == RUNSYS:
  298. if config.DontLeave==True:
  299. os.chdir(GetExePath())
  300. os.system( "/bin/sh -c "+event.message)
  301. else:
  302. pygame.quit()
  303. gobject_main_loop.quit()
  304. os.chdir( GetExePath())
  305. endpos = len(event.message)
  306. space_break_pos = endpos
  307. for i in range(0,endpos):
  308. if event.message[i] == "/" and event.message[i-1] == " " and i > 6:
  309. space_break_pos = i-1
  310. break
  311. exec_app_cmd = "cd "+os.path.dirname(event.message[:space_break_pos])+";"
  312. exec_app_cmd += event.message
  313. exec_app_cmd += "; sync & cd "+GetExePath()+"; exec python "+myscriptname
  314. print(exec_app_cmd)
  315. release_self_fds()
  316. os.execlp("/bin/sh","/bin/sh","-c", exec_app_cmd)
  317. os.chdir( GetExePath())
  318. os.exelp("python","python"," "+myscriptname)
  319. return
  320. if event.type == RESTARTUI:
  321. pygame.quit()
  322. gobject_main_loop.quit()
  323. os.chdir(GetExePath())
  324. exec_app_cmd = " sync & cd "+GetExePath()+"; exec python "+myscriptname
  325. print(exec_app_cmd)
  326. release_self_fds()
  327. os.execlp("/bin/sh","/bin/sh","-c", exec_app_cmd)
  328. os.chdir( GetExePath())
  329. os.exelp("python","python"," "+myscriptname)
  330. return
  331. if event.type == RUNSH:
  332. pygame.quit()
  333. gobject_main_loop.quit()
  334. exec_app_cmd = event.message +";"
  335. release_self_fds()
  336. os.execlp("/bin/sh","/bin/sh","-c", exec_app_cmd)
  337. sys.exit(-1)
  338. return
  339. if event.type == POWEROPT:
  340. everytime_keydown = time.time()
  341. return
  342. if event.type == pygame.KEYUP:
  343. pygame.event.clear(pygame.KEYDOWN)
  344. return
  345. if event.type == pygame.KEYDOWN:
  346. everytime_keydown = time.time()
  347. if RestoreLastBackLightBrightness(main_screen) == False:
  348. return
  349. ###########################################################
  350. if event.key == pygame.K_q:
  351. on_exit_cb = getattr(main_screen,"OnExitCb",None)
  352. if on_exit_cb != None:
  353. if callable( on_exit_cb ):
  354. main_screen.OnExitCb(event)
  355. gobject_main_loop.quit()
  356. exit()
  357. if event.key == pygame.K_KP_PLUS:
  358. if main_screen._CurrentPage._Name != "Sound volume": ## name from Menu/GameShell/10_Settings/Sound/pages.py
  359. main_screen.Draw()
  360. sound_patch.VolumeUp()
  361. sound_patch.Draw()
  362. main_screen.SwapAndShow()
  363. #pygame.time.delay(200)
  364. #main_screen.Draw()
  365. #main_screen.SwapAndShow()
  366. if event.key == pygame.K_KP_MINUS:
  367. if main_screen._CurrentPage._Name != "Sound volume":
  368. main_screen.Draw()
  369. sound_patch.VolumeDown()
  370. sound_patch.Draw()
  371. main_screen.SwapAndShow()
  372. #pygame.time.delay(200)
  373. #main_screen.Draw()
  374. #main_screen.SwapAndShow()
  375. ###########################################################
  376. if event.key == pygame.K_ESCAPE:
  377. pygame.event.clear()
  378. if RecordKeyDns(event.key,main_screen) == False:
  379. key_down_cb = getattr(main_screen,"KeyDown",None)
  380. if key_down_cb != None:
  381. if callable( key_down_cb ):
  382. main_screen.KeyDown(event)
  383. main_screen._LastKeyDown = everytime_keydown
  384. return
  385. def gobject_pygame_event_poll_timer(main_screen):
  386. event = pygame.event.poll()
  387. event_process(event,main_screen)
  388. InspectionTeam(main_screen)
  389. return True
  390. def gobject_pygame_event_timer(main_screen):
  391. global sound_patch
  392. for event in pygame.event.get():
  393. event_process(event,main_screen)
  394. return True
  395. @misc.threaded
  396. def socket_thread(main_screen):
  397. global everytime_keydown
  398. socket_path = "/tmp/gameshell"
  399. if os.path.exists(socket_path):
  400. os.remove(socket_path)
  401. server = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
  402. server.bind(socket_path)
  403. while True:
  404. server.listen(1)
  405. conn, addr = server.accept()
  406. datagram = conn.recv(1024)
  407. if datagram:
  408. tokens = datagram.strip().split()
  409. if tokens[0].lower() == "esc":
  410. escevent = pygame.event.Event(pygame.KEYDOWN,{'scancode':9,'key': 27, 'unicode': u'\x1b', 'mod': 0})
  411. current_page_key_down_cb = getattr(main_screen._CurrentPage,"KeyDown",None)
  412. if current_page_key_down_cb != None:
  413. if callable( current_page_key_down_cb ):
  414. main_screen._CurrentPage.KeyDown(escevent)
  415. if tokens[0].lower() == "quit": #eg: echo "quit" | socat - UNIX-CONNECT:/tmp/gameshell
  416. conn.close()
  417. on_exit_cb = getattr(main_screen,"OnExitCb",None)
  418. if on_exit_cb != None:
  419. if callable( on_exit_cb ):
  420. main_screen.OnExitCb(None)
  421. gobject_main_loop.quit()
  422. exit()
  423. if tokens[0].lower() == "keydown": ## simulate keydown event
  424. everytime_keydown = time.time()
  425. if RestoreLastBackLightBrightness(main_screen) == False:
  426. print("RestoreLastBackLightBrightness unix socket false")
  427. if tokens[0].lower() == "poweroff":
  428. escevent = pygame.event.Event(pygame.KEYDOWN,{'scancode':9,'key': 27, 'unicode': u'\x1b', 'mod': 0})
  429. for i in range(0,5):
  430. current_page_key_down_cb = getattr(main_screen._CurrentPage,"KeyDown",None)
  431. if current_page_key_down_cb != None:
  432. if callable( current_page_key_down_cb ):
  433. main_screen._CurrentPage.KeyDown(escevent)
  434. if main_screen._MyPageStack.Length() == 0: ## on Top Level
  435. break
  436. if main_screen._CurrentPage._Name == "GameShell":
  437. for i in main_screen._CurrentPage._Icons:
  438. if i._MyType == ICON_TYPES["FUNC"]:
  439. if i._Label.GetText() == "PowerOFF":
  440. api_cb = getattr(i._CmdPath,"API",None)
  441. if api_cb != None:
  442. if callable(api_cb):
  443. i._CmdPath.API(main_screen)
  444. def big_loop():
  445. global sound_patch,gobject_flash_led1
  446. title_bar = TitleBar()
  447. title_bar.Init(screen)
  448. foot_bar = FootBar()
  449. foot_bar.Init(screen)
  450. main_screen = MainScreen()
  451. main_screen._HWND = screen
  452. main_screen._TitleBar = title_bar
  453. main_screen._FootBar = foot_bar
  454. main_screen.Init()
  455. main_screen.ReadTheDirIntoPages("../Menu",0,None)
  456. main_screen.ReadTheDirIntoPages("/home/cpi/apps/Menu",1,main_screen._Pages[ len(main_screen._Pages) -1])
  457. main_screen.ReunionPagesIcons()
  458. main_screen.FartherPages()
  459. title_bar._SkinManager = main_screen._SkinManager
  460. foot_bar._SkinManager = main_screen._SkinManager
  461. sound_patch = SoundPatch()
  462. sound_patch._Parent = main_screen
  463. sound_patch.Init()
  464. #pp(main_screen._Pages[0],True,6)
  465. screen.fill(MySkinManager.GiveColor("White"))
  466. main_screen.Draw()
  467. main_screen.SwapAndShow()
  468. #gobject.timeout_add(DT,gobject_pygame_event_timer,main_screen)
  469. gobject_flash_led1 = gobject.timeout_add(200,GobjectFlashLed1,main_screen)
  470. gobject.timeout_add(DT,gobject_pygame_event_poll_timer,main_screen)
  471. gobject.timeout_add(3000,title_bar.GObjectRoundRobin)
  472. socket_thread(main_screen)
  473. gobject_loop()
  474. ###MAIN()###
  475. if __name__ == '__main__':
  476. os.environ['SDL_VIDEO_CENTERED'] = '1'
  477. X_center_mouse()
  478. os.chdir( os.path.dirname(os.path.realpath(__file__)) )
  479. SCREEN_SIZE = (Width,Height)
  480. screen = pygame.display.set_mode(SCREEN_SIZE, 0, 32)
  481. pygame.event.set_allowed(None)
  482. pygame.event.set_allowed([pygame.KEYDOWN,pygame.KEYUP,RUNEVT,RUNSYS,POWEROPT,RESTARTUI,RUNSH])
  483. pygame.key.set_repeat(DT+DT*6+DT/2, DT+DT*3+DT/2)
  484. MyIconPool.Init()
  485. setup_dbus()
  486. gobject.threads_init()
  487. gobject_main_loop = gobject.MainLoop()
  488. # if pygame.display.get_active() == True:
  489. # print("I am actived")
  490. if pygame.image.get_extended() == False:
  491. print("This pygame does not support PNG")
  492. exit()
  493. crt_screen = CreateByScreen()
  494. crt_screen.Init()
  495. crt_screen._HWND = screen
  496. big_loop()