touch_manager.py 3.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990
  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. logger.error(event.button)
  49. self.up_pos = event.pos
  50. if abs(self.down_pos[0] - self.up_pos[0]) < self.max_move_margin:
  51. if abs(self.down_pos[1] - self.up_pos[1]) < self.max_move_margin:
  52. if time.time() - TouchManager.long_click_min_time > self.down_time:
  53. return TouchEvent(TouchManager.long_click, self.down_pos, self.up_pos, None)
  54. else:
  55. return TouchEvent(TouchManager.click, self.down_pos, self.up_pos, None)
  56. elif abs(self.down_pos[1] - self.up_pos[1]) > self.min_swipe_move:
  57. return TouchEvent(TouchManager.swipe, self.down_pos, self.up_pos, True)
  58. elif self.down_pos[1] - self.up_pos[1] < self.max_move_margin:
  59. if abs(self.down_pos[0] - self.up_pos[0]) > self.min_swipe_move:
  60. return TouchEvent(TouchManager.swipe, self.down_pos, self.up_pos, False)
  61. class TouchEvent():
  62. def __init__(self, event_type, down_pos, current_pos, vertical):
  63. self.type = event_type
  64. self.down_pos = down_pos
  65. self.current_pos = current_pos
  66. if event_type is TouchManager.swipe:
  67. if vertical:
  68. if self.down_pos[1] < self.current_pos[1]:
  69. self.direction = TouchManager.down
  70. else:
  71. self.direction = TouchManager.up
  72. else:
  73. if self.down_pos[0] < self.current_pos[0]:
  74. self.direction = TouchManager.right
  75. else:
  76. self.direction = TouchManager.left