Implement most CvSource functionality.

2026-07-02 02:51:42 +00:00 · 2016-10-24 20:27:00 -07:00
parent 7ea13f7e03
commit 511d551546
4 changed files with 407 additions and 120 deletions
--- a/src/CvSourceImpl.h
+++ b/src/CvSourceImpl.h
@@ -0,0 +1,80 @@
+/*----------------------------------------------------------------------------*/
+/* Copyright (c) FIRST 2016. All Rights Reserved.                             */
+/* Open Source Software - may be modified and shared by FRC teams. The code   */
+/* must be accompanied by the FIRST BSD license file in the root directory of */
+/* the project.                                                               */
+/*----------------------------------------------------------------------------*/
+
+#ifndef CAMERASERVER_CVSOURCEIMPL_H_
+#define CAMERASERVER_CVSOURCEIMPL_H_
+
+#include <atomic>
+#include <functional>
+#include <vector>
+
+#include "llvm/StringMap.h"
+
+#include "SourceImpl.h"
+
+namespace cs {
+
+class CvSourceImpl : public SourceImpl {
+ public:
+  CvSourceImpl(llvm::StringRef name, const VideoMode& mode);
+  ~CvSourceImpl() override;
+
+  bool IsConnected() const override;
+
+  // Property functions
+  void SetProperty(int property, int value, CS_Status* status) override;
+  void SetStringProperty(int property, llvm::StringRef value,
+                         CS_Status* status) override;
+
+  bool SetVideoMode(const VideoMode& mode, CS_Status* status) override;
+
+  void NumSinksChanged() override;
+  void NumSinksEnabledChanged() override;
+
+  // OpenCV-specific functions
+  void PutFrame(cv::Mat& image);
+  void NotifyError(llvm::StringRef msg);
+  void SetConnected(bool connected);
+  CS_Property CreateProperty(llvm::StringRef name, CS_PropertyType type,
+                             int minimum, int maximum, int step,
+                             int defaultValue, int value);
+  CS_Property CreateProperty(
+      llvm::StringRef name, CS_PropertyType type, int minimum, int maximum,
+      int step, int defaultValue, int value,
+      std::function<void(CS_Property property)> onChange);
+  void SetEnumPropertyChoices(CS_Property property,
+                              llvm::ArrayRef<std::string> choices,
+                              CS_Status* status);
+  void RemoveProperty(CS_Property property);
+  void RemoveProperty(llvm::StringRef name);
+
+  // Property data
+  class PropertyData : public PropertyBase {
+   public:
+    PropertyData() = default;
+    PropertyData(llvm::StringRef name_, CS_PropertyType type_, int minimum_,
+                 int maximum_, int step_, int defaultValue_, int value_)
+        : PropertyBase{name_, type_,         minimum_, maximum_,
+                       step_, defaultValue_, value_} {}
+    ~PropertyData() override = default;
+
+    std::function<void(CS_Property property)> onChange;
+  };
+
+ protected:
+  bool CacheProperties(CS_Status* status) const override;
+
+ private:
+  std::atomic_bool m_connected{true};
+  std::vector<unsigned char> m_jpegBuf;
+  std::vector<int> m_compressionParams;
+  int m_qualityProperty;
+};
+
+}  // namespace cs
+
+#endif  // CAMERASERVER_CVSOURCEIMPL_H_