mirror of
https://github.com/wpilibsuite/allwpilib
synced 2026-06-22 01:11:42 +00:00
73 lines
2.5 KiB
C++
73 lines
2.5 KiB
C++
/*----------------------------------------------------------------------------*/
|
|
/* 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 CS_CVSOURCEIMPL_H_
|
|
#define CS_CVSOURCEIMPL_H_
|
|
|
|
#include <atomic>
|
|
#include <functional>
|
|
#include <vector>
|
|
|
|
#include "SourceImpl.h"
|
|
|
|
namespace cs {
|
|
|
|
class CvSourceImpl : public SourceImpl {
|
|
public:
|
|
CvSourceImpl(llvm::StringRef name, const VideoMode& mode);
|
|
~CvSourceImpl() override;
|
|
|
|
void Start();
|
|
|
|
// 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);
|
|
int CreateProperty(llvm::StringRef name, CS_PropertyKind kind, int minimum,
|
|
int maximum, int step, int defaultValue, int value);
|
|
int CreateProperty(llvm::StringRef name, CS_PropertyKind kind, int minimum,
|
|
int maximum, int step, int defaultValue, int value,
|
|
std::function<void(CS_Property property)> onChange);
|
|
void SetEnumPropertyChoices(int property, llvm::ArrayRef<std::string> choices,
|
|
CS_Status* status);
|
|
|
|
// Property data
|
|
class PropertyData : public PropertyBase {
|
|
public:
|
|
PropertyData() = default;
|
|
PropertyData(llvm::StringRef name_, CS_PropertyKind kind_, int minimum_,
|
|
int maximum_, int step_, int defaultValue_, int value_)
|
|
: PropertyBase{name_, kind_, 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 // CS_CVSOURCEIMPL_H_
|