ZedPersonDetector.cs 1.5 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344
  1. using Assets.StreetLight.Interfaces;
  2. using Assets.StreetLight.Poco;
  3. using System;
  4. using System.Collections.Generic;
  5. using System.Linq;
  6. using UnityEngine;
  7. namespace Assets.StreetLight.Adapters
  8. {
  9. internal class ZedPersonDetector : IPersonDetector
  10. {
  11. private readonly ZEDManager zedManager;
  12. public event EventHandler<IEnumerable<Person>> PersonsDetected;
  13. public ZedPersonDetector(ZEDManager zedManager)
  14. {
  15. this.zedManager = zedManager;
  16. zedManager.OnObjectDetection += ZedManager_OnObjectDetection;
  17. zedManager.OnZEDReady += ZedManager_OnZEDReady;
  18. }
  19. private void ZedManager_OnZEDReady()
  20. {
  21. zedManager.StartObjectDetection();
  22. }
  23. private void ZedManager_OnObjectDetection(DetectionFrame detectionFrame)
  24. {
  25. var detectedObjects = detectionFrame.GetFilteredObjectList(true, true, false);
  26. var persons = detectedObjects.Where(d => d.objectClass == sl.OBJECT_CLASS.PERSON);
  27. var detectedPersons = from p in persons
  28. let position = p.Get3DWorldPosition()
  29. select new Person(p.id, null)
  30. {
  31. WorldPosition = new Vector3(position.x, position.y, position.z)
  32. };
  33. PersonsDetected?.Invoke(this, detectedPersons.ToList());
  34. }
  35. }
  36. }