Add Game Piece Detector
Signed-off-by: Filip Kujawa <filip.j.kujawa@gmail.com>
Change-Id: I4f460e64b1fb9fbe1e0d50581412e106a89515e2
diff --git a/y2023/vision/game_pieces.h b/y2023/vision/game_pieces.h
new file mode 100644
index 0000000..a41d52a
--- /dev/null
+++ b/y2023/vision/game_pieces.h
@@ -0,0 +1,26 @@
+#ifndef Y2023_VISION_GAME_PIECES_H_
+#define Y2023_VISION_GAME_PIECES_H_
+
+#include "aos/events/shm_event_loop.h"
+#include "frc971/vision/vision_generated.h"
+#include "y2023/vision/game_pieces_generated.h"
+
+namespace y2023 {
+namespace vision {
+
+using namespace frc971::vision;
+
+// Takes in camera images and detects game pieces in the image.
+// Note: Actual detection has not been implemented yet.
+class GamePiecesDetector {
+ public:
+ GamePiecesDetector(aos::EventLoop *event_loop);
+
+ void ProcessImage(const CameraImage &camera_image);
+
+ private:
+ aos::Sender<GamePieces> game_pieces_sender_;
+};
+} // namespace vision
+} // namespace y2023
+#endif