WPILibC++  2019.1.1-beta-4
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Modules Pages
cscore_oo.h
1 /*----------------------------------------------------------------------------*/
2 /* Copyright (c) 2015-2018 FIRST. All Rights Reserved. */
3 /* Open Source Software - may be modified and shared by FRC teams. The code */
4 /* must be accompanied by the FIRST BSD license file in the root directory of */
5 /* the project. */
6 /*----------------------------------------------------------------------------*/
7 
8 #ifndef CSCORE_CSCORE_OO_H_
9 #define CSCORE_CSCORE_OO_H_
10 
11 #include <initializer_list>
12 #include <string>
13 #include <utility>
14 #include <vector>
15 
16 #include "cscore_cpp.h"
17 
18 namespace cs {
19 
30 // Forward declarations so friend declarations work correctly
31 class CvSource;
32 class VideoEvent;
33 class VideoSink;
34 class VideoSource;
35 
40  friend class CvSource;
41  friend class VideoEvent;
42  friend class VideoSink;
43  friend class VideoSource;
44 
45  public:
46  enum Kind {
47  kNone = CS_PROP_NONE,
48  kBoolean = CS_PROP_BOOLEAN,
49  kInteger = CS_PROP_INTEGER,
50  kString = CS_PROP_STRING,
51  kEnum = CS_PROP_ENUM
52  };
53 
54  VideoProperty() : m_handle(0), m_kind(kNone) {}
55 
56  std::string GetName() const;
57 
58  Kind GetKind() const { return m_kind; }
59 
60  explicit operator bool() const { return m_kind != kNone; }
61 
62  // Kind checkers
63  bool IsBoolean() const { return m_kind == kBoolean; }
64  bool IsInteger() const { return m_kind == kInteger; }
65  bool IsString() const { return m_kind == kString; }
66  bool IsEnum() const { return m_kind == kEnum; }
67 
68  int Get() const;
69  void Set(int value);
70  int GetMin() const;
71  int GetMax() const;
72  int GetStep() const;
73  int GetDefault() const;
74 
75  // String-specific functions
76  std::string GetString() const;
77  wpi::StringRef GetString(wpi::SmallVectorImpl<char>& buf) const;
78  void SetString(const wpi::Twine& value);
79 
80  // Enum-specific functions
81  std::vector<std::string> GetChoices() const;
82 
83  CS_Status GetLastStatus() const { return m_status; }
84 
85  private:
86  explicit VideoProperty(CS_Property handle);
87  VideoProperty(CS_Property handle, Kind kind);
88 
89  mutable CS_Status m_status;
90  CS_Property m_handle;
91  Kind m_kind;
92 };
93 
97 class VideoSource {
98  friend class VideoEvent;
99  friend class VideoSink;
100 
101  public:
102  enum Kind {
103  kUnknown = CS_SOURCE_UNKNOWN,
104  kUsb = CS_SOURCE_USB,
105  kHttp = CS_SOURCE_HTTP,
106  kCv = CS_SOURCE_CV
107  };
108 
116 
122 
128  };
129 
130  VideoSource() noexcept : m_handle(0) {}
131  VideoSource(const VideoSource& source);
132  VideoSource(VideoSource&& other) noexcept;
133  VideoSource& operator=(VideoSource other) noexcept;
134  ~VideoSource();
135 
136  explicit operator bool() const { return m_handle != 0; }
137 
138  int GetHandle() const { return m_handle; }
139 
140  bool operator==(const VideoSource& other) const {
141  return m_handle == other.m_handle;
142  }
143 
144  bool operator!=(const VideoSource& other) const { return !(*this == other); }
145 
149  Kind GetKind() const;
150 
155  std::string GetName() const;
156 
160  std::string GetDescription() const;
161 
168  uint64_t GetLastFrameTime() const;
169 
180 
184  bool IsConnected() const;
185 
192  bool IsEnabled() const;
193 
200  VideoProperty GetProperty(const wpi::Twine& name);
201 
205  std::vector<VideoProperty> EnumerateProperties() const;
206 
210  VideoMode GetVideoMode() const;
211 
217  bool SetVideoMode(const VideoMode& mode);
218 
228  bool SetVideoMode(VideoMode::PixelFormat pixelFormat, int width, int height,
229  int fps);
230 
237  bool SetPixelFormat(VideoMode::PixelFormat pixelFormat);
238 
246  bool SetResolution(int width, int height);
247 
254  bool SetFPS(int fps);
255 
282  bool SetConfigJson(wpi::StringRef config);
283 
290  bool SetConfigJson(const wpi::json& config);
291 
297  std::string GetConfigJson() const;
298 
305 
313  double GetActualFPS() const;
314 
322  double GetActualDataRate() const;
323 
327  std::vector<VideoMode> EnumerateVideoModes() const;
328 
329  CS_Status GetLastStatus() const { return m_status; }
330 
336  std::vector<VideoSink> EnumerateSinks();
337 
343  static std::vector<VideoSource> EnumerateSources();
344 
345  friend void swap(VideoSource& first, VideoSource& second) noexcept {
346  using std::swap;
347  swap(first.m_status, second.m_status);
348  swap(first.m_handle, second.m_handle);
349  }
350 
351  protected:
352  explicit VideoSource(CS_Source handle) : m_handle(handle) {}
353 
354  mutable CS_Status m_status = 0;
355  CS_Source m_handle;
356 };
357 
361 class VideoCamera : public VideoSource {
362  public:
363  enum WhiteBalance {
364  kFixedIndoor = 3000,
365  kFixedOutdoor1 = 4000,
366  kFixedOutdoor2 = 5000,
367  kFixedFluorescent1 = 5100,
368  kFixedFlourescent2 = 5200
369  };
370 
371  VideoCamera() = default;
372 
376  void SetBrightness(int brightness);
377 
381  int GetBrightness();
382 
386  void SetWhiteBalanceAuto();
387 
392 
396  void SetWhiteBalanceManual(int value);
397 
401  void SetExposureAuto();
402 
406  void SetExposureHoldCurrent();
407 
411  void SetExposureManual(int value);
412 
413  protected:
414  explicit VideoCamera(CS_Source handle) : VideoSource(handle) {}
415 };
416 
420 class UsbCamera : public VideoCamera {
421  public:
422  UsbCamera() = default;
423 
430  UsbCamera(const wpi::Twine& name, int dev);
431 
438  UsbCamera(const wpi::Twine& name, const wpi::Twine& path);
439 
445  static std::vector<UsbCameraInfo> EnumerateUsbCameras();
446 
450  std::string GetPath() const;
451 
457  void SetConnectVerbose(int level);
458 };
459 
463 class HttpCamera : public VideoCamera {
464  public:
465  enum HttpCameraKind {
466  kUnknown = CS_HTTP_UNKNOWN,
467  kMJPGStreamer = CS_HTTP_MJPGSTREAMER,
468  kCSCore = CS_HTTP_CSCORE,
469  kAxis = CS_HTTP_AXIS
470  };
471 
479  HttpCamera(const wpi::Twine& name, const wpi::Twine& url,
480  HttpCameraKind kind = kUnknown);
481 
489  HttpCamera(const wpi::Twine& name, const char* url,
490  HttpCameraKind kind = kUnknown);
491 
499  HttpCamera(const wpi::Twine& name, const std::string& url,
500  HttpCameraKind kind = kUnknown);
501 
510  HttpCameraKind kind = kUnknown);
511 
519  template <typename T>
520  HttpCamera(const wpi::Twine& name, std::initializer_list<T> urls,
521  HttpCameraKind kind = kUnknown);
522 
529  HttpCameraKind GetHttpCameraKind() const;
530 
535 
539  template <typename T>
540  void SetUrls(std::initializer_list<T> urls);
541 
545  std::vector<std::string> GetUrls() const;
546 };
547 
551 class AxisCamera : public HttpCamera {
552  static std::string HostToUrl(const wpi::Twine& host);
553  static std::vector<std::string> HostToUrl(wpi::ArrayRef<std::string> hosts);
554  template <typename T>
555  static std::vector<std::string> HostToUrl(std::initializer_list<T> hosts);
556 
557  public:
565  AxisCamera(const wpi::Twine& name, const wpi::Twine& host);
566 
574  AxisCamera(const wpi::Twine& name, const char* host);
575 
583  AxisCamera(const wpi::Twine& name, const std::string& host);
584 
592  AxisCamera(const wpi::Twine& name, wpi::StringRef host);
593 
602 
610  template <typename T>
611  AxisCamera(const wpi::Twine& name, std::initializer_list<T> hosts);
612 };
613 
617 class CvSource : public VideoSource {
618  public:
619  CvSource() = default;
620 
627  CvSource(const wpi::Twine& name, const VideoMode& mode);
628 
638  CvSource(const wpi::Twine& name, VideoMode::PixelFormat pixelFormat,
639  int width, int height, int fps);
640 
650  void PutFrame(cv::Mat& image);
651 
656  void NotifyError(const wpi::Twine& msg);
657 
663  void SetConnected(bool connected);
664 
670  void SetDescription(const wpi::Twine& description);
671 
684  VideoProperty CreateProperty(const wpi::Twine& name, VideoProperty::Kind kind,
685  int minimum, int maximum, int step,
686  int defaultValue, int value);
687 
699  VideoProperty CreateIntegerProperty(const wpi::Twine& name, int minimum,
700  int maximum, int step, int defaultValue,
701  int value);
702 
711  VideoProperty CreateBooleanProperty(const wpi::Twine& name, bool defaultValue,
712  bool value);
713 
723  const wpi::Twine& value);
724 
731  void SetEnumPropertyChoices(const VideoProperty& property,
733 
740  template <typename T>
741  void SetEnumPropertyChoices(const VideoProperty& property,
742  std::initializer_list<T> choices);
743 };
744 
748 class VideoSink {
749  friend class VideoEvent;
750  friend class VideoSource;
751 
752  public:
753  enum Kind {
754  kUnknown = CS_SINK_UNKNOWN,
755  kMjpeg = CS_SINK_MJPEG,
756  kCv = CS_SINK_CV
757  };
758 
759  VideoSink() noexcept : m_handle(0) {}
760  VideoSink(const VideoSink& sink);
761  VideoSink(VideoSink&& sink) noexcept;
762  VideoSink& operator=(VideoSink other) noexcept;
763  ~VideoSink();
764 
765  explicit operator bool() const { return m_handle != 0; }
766 
767  int GetHandle() const { return m_handle; }
768 
769  bool operator==(const VideoSink& other) const {
770  return m_handle == other.m_handle;
771  }
772 
773  bool operator!=(const VideoSink& other) const { return !(*this == other); }
774 
778  Kind GetKind() const;
779 
784  std::string GetName() const;
785 
789  std::string GetDescription() const;
790 
798  VideoProperty GetProperty(const wpi::Twine& name);
799 
803  std::vector<VideoProperty> EnumerateProperties() const;
804 
812  void SetSource(VideoSource source);
813 
819  VideoSource GetSource() const;
820 
829 
830  CS_Status GetLastStatus() const { return m_status; }
831 
837  static std::vector<VideoSink> EnumerateSinks();
838 
839  friend void swap(VideoSink& first, VideoSink& second) noexcept {
840  using std::swap;
841  swap(first.m_status, second.m_status);
842  swap(first.m_handle, second.m_handle);
843  }
844 
845  protected:
846  explicit VideoSink(CS_Sink handle) : m_handle(handle) {}
847 
848  mutable CS_Status m_status = 0;
849  CS_Sink m_handle;
850 };
851 
855 class MjpegServer : public VideoSink {
856  public:
857  MjpegServer() = default;
858 
866  MjpegServer(const wpi::Twine& name, const wpi::Twine& listenAddress,
867  int port);
868 
875  MjpegServer(const wpi::Twine& name, int port) : MjpegServer(name, "", port) {}
876 
880  std::string GetListenAddress() const;
881 
885  int GetPort() const;
886 
901  void SetResolution(int width, int height);
902 
910  void SetFPS(int fps);
911 
921  void SetCompression(int quality);
922 
930  void SetDefaultCompression(int quality);
931 };
932 
936 class CvSink : public VideoSink {
937  public:
938  CvSink() = default;
939 
948  explicit CvSink(const wpi::Twine& name);
949 
962  CvSink(const wpi::Twine& name,
963  std::function<void(uint64_t time)> processFrame);
964 
970  void SetDescription(const wpi::Twine& description);
971 
981  uint64_t GrabFrame(cv::Mat& image, double timeout = 0.225) const;
982 
991  uint64_t GrabFrameNoTimeout(cv::Mat& image) const;
992 
997  std::string GetError() const;
998 
1006  void SetEnabled(bool enabled);
1007 };
1008 
1012 class VideoEvent : public RawEvent {
1013  public:
1017  VideoSource GetSource() const;
1018 
1022  VideoSink GetSink() const;
1023 
1027  VideoProperty GetProperty() const;
1028 };
1029 
1035  public:
1036  VideoListener() : m_handle(0) {}
1037 
1046  VideoListener(std::function<void(const VideoEvent& event)> callback,
1047  int eventMask, bool immediateNotify);
1048 
1049  VideoListener(const VideoListener&) = delete;
1050  VideoListener& operator=(const VideoListener&) = delete;
1051  VideoListener(VideoListener&& other) noexcept;
1052  VideoListener& operator=(VideoListener&& other) noexcept;
1053  ~VideoListener();
1054 
1055  friend void swap(VideoListener& first, VideoListener& second) noexcept {
1056  using std::swap;
1057  swap(first.m_handle, second.m_handle);
1058  }
1059 
1060  private:
1061  CS_Listener m_handle;
1062 };
1063 
1066 } // namespace cs
1067 
1068 #include "cscore_oo.inl"
1069 
1070 #endif // CSCORE_CSCORE_OO_H_
int GetPort() const
Get the port number of the server.
Definition: cscore_oo.inl:529
void SetWhiteBalanceManual(int value)
Set the white balance to manual, with specified color temperature.
Definition: cscore_oo.inl:225
void SetFPS(int fps)
Set the stream frames per second (FPS) for clients that don't specify it.
Definition: cscore_oo.inl:541
CameraServer (cscore) namespace.
Definition: cscore_oo.inl:11
VideoSource GetSource() const
Get the connected source.
Definition: cscore_oo.inl:508
void SetCompression(int quality)
Set the compression for clients that don't specify it.
Definition: cscore_oo.inl:546
void SetDescription(const wpi::Twine &description)
Set source description.
Definition: cscore_oo.inl:401
wpi::json GetConfigJsonObject() const
Get a JSON configuration object.
void SetSource(VideoSource source)
Configure which source should provide frames to this sink.
Definition: cscore_oo.inl:500
A source for video that provides a sequence of frames.
Definition: cscore_oo.h:97
Try to keep the connection open regardless of whether any sinks are connected.
Definition: cscore_c.h:194
VideoSink GetSink() const
Get the sink associated with the event (if any).
Definition: cscore_oo.inl:597
A sink for video that accepts a sequence of frames.
Definition: cscore_oo.h:748
void SetEnumPropertyChoices(const VideoProperty &property, wpi::ArrayRef< std::string > choices)
Configure enum property choices.
Definition: cscore_oo.inl:447
ConnectionStrategy
Connection strategy.
Definition: cscore_oo.h:110
VideoProperty GetSourceProperty(const wpi::Twine &name)
Get a property of the associated source.
Definition: cscore_oo.inl:514
void SetExposureHoldCurrent()
Set the exposure to hold current.
Definition: cscore_oo.inl:235
std::vector< VideoMode > EnumerateVideoModes() const
Enumerate all known video modes for this source.
Definition: cscore_oo.inl:200
std::vector< VideoProperty > EnumerateProperties() const
Enumerate all properties of this sink.
An event generated by the library and provided to event listeners.
Definition: cscore_oo.h:1012
HttpCamera(const wpi::Twine &name, const wpi::Twine &url, HttpCameraKind kind=kUnknown)
Create a source for a MJPEG-over-HTTP (IP) camera.
Definition: cscore_oo.inl:269
VideoProperty GetProperty(const wpi::Twine &name)
Get a property of the sink.
Definition: cscore_oo.inl:495
bool IsConnected() const
Is the source currently connected to whatever is providing the images?
Definition: cscore_oo.inl:126
ArrayRef - Represent a constant reference to an array (0 or more elements consecutively in memory)...
Definition: ArrayRef.h:41
Listener event.
Definition: cscore_cpp.h:94
std::vector< std::string > GetUrls() const
Get the URLs used to connect to the camera.
Definition: cscore_oo.inl:327
A source that represents a USB camera.
Definition: cscore_oo.h:420
AxisCamera(const wpi::Twine &name, const wpi::Twine &host)
Create a source for an Axis IP camera.
Definition: cscore_oo.inl:355
An event listener.
Definition: cscore_oo.h:1034
uint64_t GrabFrame(cv::Mat &image, double timeout=0.225) const
Wait for the next frame and get the image.
Definition: cscore_oo.inl:572
VideoProperty CreateIntegerProperty(const wpi::Twine &name, int minimum, int maximum, int step, int defaultValue, int value)
Create an integer property.
Definition: cscore_oo.inl:417
A source that represents a MJPEG-over-HTTP (IP) camera.
Definition: cscore_oo.h:463
std::vector< VideoProperty > EnumerateProperties() const
Enumerate all properties of this source.
HttpCameraKind GetHttpCameraKind() const
Get the kind of HTTP camera.
Definition: cscore_oo.inl:307
void SetExposureManual(int value)
Set the exposure to manual, as a percentage (0-100).
Definition: cscore_oo.inl:240
double GetActualFPS() const
Get the actual FPS.
Definition: cscore_oo.inl:188
void SetDescription(const wpi::Twine &description)
Set sink description.
Definition: cscore_oo.inl:567
bool SetConfigJson(wpi::StringRef config)
Set video mode and properties from a JSON configuration string.
Definition: cscore_oo.inl:173
bool IsEnabled() const
Gets source enable status.
Definition: cscore_oo.inl:131
std::string GetName() const
Get the name of the sink.
Definition: cscore_oo.inl:485
Never open the connection.
Definition: cscore_c.h:200
Automatically connect or disconnect based on whether any sinks are connected to this source...
Definition: cscore_oo.h:115
uint64_t GrabFrameNoTimeout(cv::Mat &image) const
Wait for the next frame and get the image.
Definition: cscore_oo.inl:577
A source or sink property.
Definition: cscore_oo.h:39
Kind GetKind() const
Get the kind of the source.
Definition: cscore_oo.inl:99
Automatically connect or disconnect based on whether any sinks are connected to this source...
Definition: cscore_c.h:188
std::string GetDescription() const
Get the source description.
Definition: cscore_oo.inl:109
std::string GetName() const
Get the name of the source.
Definition: cscore_oo.inl:104
bool SetFPS(int fps)
Set the frames per second (FPS).
Definition: cscore_oo.inl:168
std::string GetError() const
Get error string.
Definition: cscore_oo.inl:582
void SetConnectionStrategy(ConnectionStrategy strategy)
Sets the connection strategy.
Definition: cscore_oo.inl:119
static std::vector< VideoSink > EnumerateSinks()
Enumerate all existing sinks.
std::vector< VideoSink > EnumerateSinks()
Enumerate all sinks connected to this source.
void SetResolution(int width, int height)
Set the stream resolution for clients that don't specify it.
Definition: cscore_oo.inl:534
void SetWhiteBalanceHoldCurrent()
Set the white balance to hold current.
Definition: cscore_oo.inl:220
static std::vector< VideoSource > EnumerateSources()
Enumerate all existing sources.
bool SetResolution(int width, int height)
Set the resolution.
Definition: cscore_oo.inl:163
static std::vector< UsbCameraInfo > EnumerateUsbCameras()
Enumerate USB cameras on the local system.
Definition: cscore_oo.inl:253
A source for user code to provide OpenCV images as video frames.
Definition: cscore_oo.h:617
VideoProperty CreateStringProperty(const wpi::Twine &name, const wpi::Twine &value)
Create a string property.
Definition: cscore_oo.inl:436
void SetUrls(wpi::ArrayRef< std::string > urls)
Change the URLs used to connect to the camera.
Definition: cscore_oo.inl:313
void NotifyError(const wpi::Twine &msg)
Signal sinks that an error has occurred.
Definition: cscore_oo.inl:391
void SetDefaultCompression(int quality)
Set the default compression used for non-MJPEG sources.
Definition: cscore_oo.inl:552
std::string GetDescription() const
Get the sink description.
Definition: cscore_oo.inl:490
a class to store JSON values
Definition: json.h:2714
void SetExposureAuto()
Set the exposure to auto aperature.
Definition: cscore_oo.inl:230
VideoMode GetVideoMode() const
Get the current video mode.
Definition: cscore_oo.inl:141
Try to keep the connection open regardless of whether any sinks are connected.
Definition: cscore_oo.h:121
VideoProperty CreateBooleanProperty(const wpi::Twine &name, bool defaultValue, bool value)
Create a boolean property.
Definition: cscore_oo.inl:427
StringRef - Represent a constant reference to a string, i.e.
Definition: StringRef.h:49
MjpegServer(const wpi::Twine &name, int port)
Create a MJPEG-over-HTTP server sink.
Definition: cscore_oo.h:875
std::string GetPath() const
Get the path to the device.
Definition: cscore_oo.inl:258
A source that represents a video camera.
Definition: cscore_oo.h:361
Kind GetKind() const
Get the kind of the sink.
Definition: cscore_oo.inl:480
void SetEnabled(bool enabled)
Enable or disable getting new frames.
Definition: cscore_oo.inl:587
bool SetPixelFormat(VideoMode::PixelFormat pixelFormat)
Set the pixel format.
Definition: cscore_oo.inl:158
VideoProperty GetProperty(const wpi::Twine &name)
Get a property.
Definition: cscore_oo.inl:136
VideoProperty CreateProperty(const wpi::Twine &name, VideoProperty::Kind kind, int minimum, int maximum, int step, int defaultValue, int value)
Create a property.
Definition: cscore_oo.inl:406
VideoSource GetSource() const
Get the source associated with the event (if any).
Definition: cscore_oo.inl:592
Twine - A lightweight data structure for efficiently representing the concatenation of temporary valu...
Definition: Twine.h:79
void PutFrame(cv::Mat &image)
Put an OpenCV image and notify sinks.
Definition: cscore_oo.inl:386
A sink that acts as a MJPEG-over-HTTP network server.
Definition: cscore_oo.h:855
VideoProperty GetProperty() const
Get the property associated with the event (if any).
Definition: cscore_oo.inl:602
void SetConnectVerbose(int level)
Set how verbose the camera connection messages are.
Definition: cscore_oo.inl:263
Never open the connection.
Definition: cscore_oo.h:127
A source that represents an Axis IP camera.
Definition: cscore_oo.h:551
void SetBrightness(int brightness)
Set the brightness, as a percentage (0-100).
Definition: cscore_oo.inl:205
A sink for user code to accept video frames as OpenCV images.
Definition: cscore_oo.h:936
double GetActualDataRate() const
Get the data rate (in bytes per second).
Definition: cscore_oo.inl:194
uint64_t GetLastFrameTime() const
Get the last time a frame was captured.
Definition: cscore_oo.inl:114
std::string GetListenAddress() const
Get the listen address of the server.
Definition: cscore_oo.inl:524
Video mode.
Definition: cscore_cpp.h:60
void SetWhiteBalanceAuto()
Set the white balance to auto.
Definition: cscore_oo.inl:215
void SetConnected(bool connected)
Set source connection status.
Definition: cscore_oo.inl:396
bool SetVideoMode(const VideoMode &mode)
Set the video mode.
Definition: cscore_oo.inl:146
int GetBrightness()
Get the brightness, as a percentage (0-100).
Definition: cscore_oo.inl:210
std::string GetConfigJson() const
Get a JSON configuration string.
Definition: cscore_oo.inl:183