touch_manager.py 3.3 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091
  1. import pygame
  2. import logging
  3. import time
  4. logger = logging.getLogger(__name__)
  5. class TouchManager():
  6. click = 1
  7. swipe = 2
  8. long_click = 3
  9. long_click_min_time = 0.3
  10. up = 0
  11. down = 1
  12. left = 2
  13. right = 3
  14. def __init__(self, size):
  15. self.down_pos = (0, 0)
  16. self.up_pos = (0, 0)
  17. self.screen_size = size
  18. self.max_move_margin = self.screen_size[1] / 6
  19. self.min_swipe_move = self.screen_size[1] / 3
  20. self.down_button = -1
  21. self.down_time = -1
  22. def event(self, event):
  23. if event.type == pygame.MOUSEBUTTONUP:
  24. logger.error(event.button)
  25. if event.button == 4:
  26. touch_event = TouchEvent(TouchManager.swipe, self.down_pos, self.up_pos, True)
  27. touch_event.direction = TouchManager.up
  28. return touch_event
  29. elif event.button == 5:
  30. touch_event = TouchEvent(TouchManager.swipe, self.down_pos, self.up_pos, True)
  31. touch_event.direction = TouchManager.down
  32. return touch_event
  33. elif event.button == 1 and self.down_button == 1:
  34. touch_event = self.mouse_up(event)
  35. return touch_event
  36. elif event.button == 3 and self.down_button == 3:
  37. touch_event = TouchEvent(TouchManager.long_click, self.down_pos, self.up_pos, None)
  38. return touch_event
  39. else:
  40. return None
  41. elif event.type == pygame.MOUSEBUTTONDOWN:
  42. self.mouse_down(event)
  43. return None
  44. def mouse_down(self, event):
  45. self.down_pos = event.pos
  46. self.down_button = event.button
  47. self.down_time = time.time()
  48. def mouse_up(self, event):
  49. logger.error(event.button)
  50. self.up_pos = event.pos
  51. if abs(self.down_pos[0] - self.up_pos[0]) < self.max_move_margin:
  52. if abs(self.down_pos[1] - self.up_pos[1]) < self.max_move_margin:
  53. if time.time() - TouchManager.long_click_min_time > self.down_time:
  54. return TouchEvent(TouchManager.long_click, self.down_pos, self.up_pos, None)
  55. else:
  56. return TouchEvent(TouchManager.click, self.down_pos, self.up_pos, None)
  57. elif abs(self.down_pos[1] - self.up_pos[1]) > self.min_swipe_move:
  58. return TouchEvent(TouchManager.swipe, self.down_pos, self.up_pos, True)
  59. elif self.down_pos[1] - self.up_pos[1] < self.max_move_margin:
  60. if abs(self.down_pos[0] - self.up_pos[0]) > self.min_swipe_move:
  61. return TouchEvent(TouchManager.swipe, self.down_pos, self.up_pos, False)
  62. class TouchEvent():
  63. def __init__(self, event_type, down_pos, current_pos, vertical):
  64. self.type = event_type
  65. self.down_pos = down_pos
  66. self.current_pos = current_pos
  67. if event_type is TouchManager.swipe:
  68. if vertical:
  69. if self.down_pos[1] < self.current_pos[1]:
  70. self.direction = TouchManager.down
  71. else:
  72. self.direction = TouchManager.up
  73. else:
  74. if self.down_pos[0] < self.current_pos[0]:
  75. self.direction = TouchManager.right
  76. else:
  77. self.direction = TouchManager.left