TouchTracker.cs 2.1 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273
  1. using System;
  2. using System.Collections.Generic;
  3. using System.Drawing;
  4. using System.Linq;
  5. using System.Text;
  6. using System.Threading.Tasks;
  7. using bbiwarg.Images;
  8. using bbiwarg.Graphics;
  9. namespace bbiwarg.Detectors.Touch
  10. {
  11. class TouchTracker
  12. {
  13. private OutputImage outputImage;
  14. private List<TouchEvent>[] detectedTouchEvents;
  15. private int framesUntilTracked;
  16. public List<TouchEvent> TrackedTouchEvents;
  17. public TouchTracker()
  18. {
  19. detectedTouchEvents = new List<TouchEvent>[Constants.TouchEventNumFramesUntilTracked];
  20. for (int i = 0; i < Constants.TouchEventNumFramesUntilTracked; i++)
  21. {
  22. detectedTouchEvents[i] = new List<TouchEvent>();
  23. }
  24. }
  25. public void setDetectedTouchEventsThisFrame(List<TouchEvent> touchEventsThisFrame, OutputImage outputImage)
  26. {
  27. this.outputImage = outputImage;
  28. for (int i = (framesUntilTracked - 1); i > 0; i--)
  29. {
  30. detectedTouchEvents[i] = detectedTouchEvents[i - 1];
  31. }
  32. detectedTouchEvents[0] = touchEventsThisFrame;
  33. findTrackedTouches();
  34. }
  35. private void findTrackedTouches()
  36. {
  37. TrackedTouchEvents = new List<TouchEvent>();
  38. foreach (TouchEvent te in detectedTouchEvents[0])
  39. {
  40. bool tracked = true;
  41. for (int i = 1; i < framesUntilTracked; i++)
  42. {
  43. if (!hasSimilarTouchEventInFrame(te, i)) tracked = false;
  44. }
  45. if (tracked)
  46. {
  47. outputImage.fillCircle(te.Position.IntX, te.Position.IntY, 5, Color.Yellow);
  48. TrackedTouchEvents.Add(te);
  49. }
  50. }
  51. }
  52. private bool hasSimilarTouchEventInFrame(TouchEvent touchEvent, int frame)
  53. {
  54. foreach (TouchEvent te in detectedTouchEvents[frame])
  55. {
  56. if (touchEvent.isSimilarTo(te))
  57. return true;
  58. }
  59. return false;
  60. }
  61. }
  62. }