touch_manager.py 3.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687
  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. if event.button == 4:
  25. touch_event = TouchEvent(TouchManager.swipe, self.down_pos, self.up_pos, True)
  26. touch_event.direction = TouchManager.up
  27. return touch_event
  28. elif event.button == 5:
  29. touch_event = TouchEvent(TouchManager.swipe, self.down_pos, self.up_pos, True)
  30. touch_event.direction = TouchManager.down
  31. return touch_event
  32. elif event.button == 1 and self.down_button == 1:
  33. touch_event = self.mouse_up(event)
  34. return touch_event
  35. elif event.button == 3 and self.down_button == 3:
  36. touch_event = TouchEvent(TouchManager.long_click, self.down_pos, self.up_pos, None)
  37. return touch_event
  38. else:
  39. return None
  40. elif event.type == pygame.MOUSEBUTTONDOWN:
  41. self.mouse_down(event)
  42. return None
  43. def mouse_down(self, event):
  44. self.down_pos = event.pos
  45. self.down_button = event.button
  46. self.down_time = time.time()
  47. def mouse_up(self, event):
  48. self.up_pos = event.pos
  49. if abs(self.down_pos[0] - self.up_pos[0]) < self.max_move_margin:
  50. if abs(self.down_pos[1] - self.up_pos[1]) < self.max_move_margin:
  51. if time.time() - TouchManager.long_click_min_time > self.down_time:
  52. return TouchEvent(TouchManager.long_click, self.down_pos, self.up_pos, None)
  53. else:
  54. return TouchEvent(TouchManager.click, self.down_pos, self.up_pos, None)
  55. elif abs(self.down_pos[1] - self.up_pos[1]) > self.min_swipe_move:
  56. return TouchEvent(TouchManager.swipe, self.down_pos, self.up_pos, True)
  57. elif self.down_pos[1] - self.up_pos[1] < self.max_move_margin:
  58. if abs(self.down_pos[0] - self.up_pos[0]) > self.min_swipe_move:
  59. return TouchEvent(TouchManager.swipe, self.down_pos, self.up_pos, False)
  60. class TouchEvent():
  61. def __init__(self, event_type, down_pos, current_pos, vertical):
  62. self.type = event_type
  63. self.down_pos = down_pos
  64. self.current_pos = current_pos
  65. if event_type is TouchManager.swipe:
  66. if vertical:
  67. if self.down_pos[1] < self.current_pos[1]:
  68. self.direction = TouchManager.down
  69. else:
  70. self.direction = TouchManager.up
  71. else:
  72. if self.down_pos[0] < self.current_pos[0]:
  73. self.direction = TouchManager.right
  74. else:
  75. self.direction = TouchManager.left