WPILibC++  2019.1.1-beta-2-16-ga60f312
 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 
593 
601  template <typename T>
602  AxisCamera(const wpi::Twine& name, std::initializer_list<T> hosts);
603 };
604 
608 class CvSource : public VideoSource {
609  public:
610  CvSource() = default;
611 
618  CvSource(const wpi::Twine& name, const VideoMode& mode);
619 
629  CvSource(const wpi::Twine& name, VideoMode::PixelFormat pixelFormat,
630  int width, int height, int fps);
631 
641  void PutFrame(cv::Mat& image);
642 
647  void NotifyError(const wpi::Twine& msg);
648 
654  void SetConnected(bool connected);
655 
661  void SetDescription(const wpi::Twine& description);
662 
675  VideoProperty CreateProperty(const wpi::Twine& name, VideoProperty::Kind kind,
676  int minimum, int maximum, int step,
677  int defaultValue, int value);
678 
690  VideoProperty CreateIntegerProperty(const wpi::Twine& name, int minimum,
691  int maximum, int step, int defaultValue,
692  int value);
693 
702  VideoProperty CreateBooleanProperty(const wpi::Twine& name, bool defaultValue,
703  bool value);
704 
714  const wpi::Twine& value);
715 
722  void SetEnumPropertyChoices(const VideoProperty& property,
724 
731  template <typename T>
732  void SetEnumPropertyChoices(const VideoProperty& property,
733  std::initializer_list<T> choices);
734 };
735 
739 class VideoSink {
740  friend class VideoEvent;
741  friend class VideoSource;
742 
743  public:
744  enum Kind {
745  kUnknown = CS_SINK_UNKNOWN,
746  kMjpeg = CS_SINK_MJPEG,
747  kCv = CS_SINK_CV
748  };
749 
750  VideoSink() noexcept : m_handle(0) {}
751  VideoSink(const VideoSink& sink);
752  VideoSink(VideoSink&& sink) noexcept;
753  VideoSink& operator=(VideoSink other) noexcept;
754  ~VideoSink();
755 
756  explicit operator bool() const { return m_handle != 0; }
757 
758  int GetHandle() const { return m_handle; }
759 
760  bool operator==(const VideoSink& other) const {
761  return m_handle == other.m_handle;
762  }
763 
764  bool operator!=(const VideoSink& other) const { return !(*this == other); }
765 
769  Kind GetKind() const;
770 
775  std::string GetName() const;
776 
780  std::string GetDescription() const;
781 
789  VideoProperty GetProperty(const wpi::Twine& name);
790 
794  std::vector<VideoProperty> EnumerateProperties() const;
795 
803  void SetSource(VideoSource source);
804 
810  VideoSource GetSource() const;
811 
820 
821  CS_Status GetLastStatus() const { return m_status; }
822 
828  static std::vector<VideoSink> EnumerateSinks();
829 
830  friend void swap(VideoSink& first, VideoSink& second) noexcept {
831  using std::swap;
832  swap(first.m_status, second.m_status);
833  swap(first.m_handle, second.m_handle);
834  }
835 
836  protected:
837  explicit VideoSink(CS_Sink handle) : m_handle(handle) {}
838 
839  mutable CS_Status m_status = 0;
840  CS_Sink m_handle;
841 };
842 
846 class MjpegServer : public VideoSink {
847  public:
848  MjpegServer() = default;
849 
857  MjpegServer(const wpi::Twine& name, const wpi::Twine& listenAddress,
858  int port);
859 
866  MjpegServer(const wpi::Twine& name, int port) : MjpegServer(name, "", port) {}
867 
871  std::string GetListenAddress() const;
872 
876  int GetPort() const;
877 
892  void SetResolution(int width, int height);
893 
901  void SetFPS(int fps);
902 
912  void SetCompression(int quality);
913 
921  void SetDefaultCompression(int quality);
922 };
923 
927 class CvSink : public VideoSink {
928  public:
929  CvSink() = default;
930 
939  explicit CvSink(const wpi::Twine& name);
940 
953  CvSink(const wpi::Twine& name,
954  std::function<void(uint64_t time)> processFrame);
955 
961  void SetDescription(const wpi::Twine& description);
962 
972  uint64_t GrabFrame(cv::Mat& image, double timeout = 0.225) const;
973 
982  uint64_t GrabFrameNoTimeout(cv::Mat& image) const;
983 
988  std::string GetError() const;
989 
997  void SetEnabled(bool enabled);
998 };
999 
1003 class VideoEvent : public RawEvent {
1004  public:
1008  VideoSource GetSource() const;
1009 
1013  VideoSink GetSink() const;
1014 
1018  VideoProperty GetProperty() const;
1019 };
1020 
1026  public:
1027  VideoListener() : m_handle(0) {}
1028 
1037  VideoListener(std::function<void(const VideoEvent& event)> callback,
1038  int eventMask, bool immediateNotify);
1039 
1040  VideoListener(const VideoListener&) = delete;
1041  VideoListener& operator=(const VideoListener&) = delete;
1042  VideoListener(VideoListener&& other) noexcept;
1043  VideoListener& operator=(VideoListener&& other) noexcept;
1044  ~VideoListener();
1045 
1046  friend void swap(VideoListener& first, VideoListener& second) noexcept {
1047  using std::swap;
1048  swap(first.m_handle, second.m_handle);
1049  }
1050 
1051  private:
1052  CS_Listener m_handle;
1053 };
1054 
1057 } // namespace cs
1058 
1059 #include "cscore_oo.inl"
1060 
1061 #endif // CSCORE_CSCORE_OO_H_
int GetPort() const
Get the port number of the server.
Definition: cscore_oo.inl:526
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:538
CameraServer (cscore) namespace.
Definition: cscore_oo.inl:11
VideoSource GetSource() const
Get the connected source.
Definition: cscore_oo.inl:505
void SetCompression(int quality)
Set the compression for clients that don't specify it.
Definition: cscore_oo.inl:543
void SetDescription(const wpi::Twine &description)
Set source description.
Definition: cscore_oo.inl:398
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:497
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:594
A sink for video that accepts a sequence of frames.
Definition: cscore_oo.h:739
void SetEnumPropertyChoices(const VideoProperty &property, wpi::ArrayRef< std::string > choices)
Configure enum property choices.
Definition: cscore_oo.inl:444
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:511
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:1003
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:492
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:1025
uint64_t GrabFrame(cv::Mat &image, double timeout=0.225) const
Wait for the next frame and get the image.
Definition: cscore_oo.inl:569
VideoProperty CreateIntegerProperty(const wpi::Twine &name, int minimum, int maximum, int step, int defaultValue, int value)
Create an integer property.
Definition: cscore_oo.inl:414
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:564
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:482
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:574
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:579
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:531
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:608
VideoProperty CreateStringProperty(const wpi::Twine &name, const wpi::Twine &value)
Create a string property.
Definition: cscore_oo.inl:433
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:388
void SetDefaultCompression(int quality)
Set the default compression used for non-MJPEG sources.
Definition: cscore_oo.inl:549
std::string GetDescription() const
Get the sink description.
Definition: cscore_oo.inl:487
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:424
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:866
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:477
void SetEnabled(bool enabled)
Enable or disable getting new frames.
Definition: cscore_oo.inl:584
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:403
VideoSource GetSource() const
Get the source associated with the event (if any).
Definition: cscore_oo.inl:589
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:383
A sink that acts as a MJPEG-over-HTTP network server.
Definition: cscore_oo.h:846
VideoProperty GetProperty() const
Get the property associated with the event (if any).
Definition: cscore_oo.inl:599
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:927
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:521
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:393
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