65 lines
1.9 KiB
C++
65 lines
1.9 KiB
C++
|
/*
|
||
|
|
* Desktop RTSP MJPEG video streaming test.
|
||
|
|
*
|
||
|
|
* Mirrors examples/examples-communication/rtsp/video-server-mjpeg, but runs
|
||
|
|
* on the desktop (via the Arduino-Emulator's WiFiServer/WiFiClient/WiFiUDP)
|
||
|
|
* instead of an ESP32-CAM, so it can be built and exercised without any
|
||
|
|
* hardware. Loops a single synthetic JPEG test frame (frame_jpeg.h) through
|
||
|
|
* JPEGRtpEncoder's RFC 2435 fragmentation and RTSPMediaCallbackSource.
|
||
|
|
*
|
||
|
|
* Connect with: ffplay rtsp://127.0.0.1:8554/video
|
||
|
|
*/
|
||
|
|
#include "AudioTools.h"
|
||
|
|
#include "AudioTools/Communication/RTSP/RTSPPlatformWiFi.h"
|
||
|
|
#include "AudioTools/Communication/RTSP.h"
|
||
|
|
#include "frame_jpeg.h"
|
||
|
|
|
||
|
|
using namespace audio_tools;
|
||
|
|
|
||
|
|
const int port = 8554;
|
||
|
|
const int VIDEO_WIDTH = 320;
|
||
|
|
const int VIDEO_HEIGHT = 240;
|
||
|
|
const float VIDEO_FPS = 5.0f;
|
||
|
|
const unsigned long FRAME_DURATION_MS = 1000 / VIDEO_FPS;
|
||
|
|
|
||
|
|
RTSPFormatMJPEG mjpegFormat(VIDEO_WIDTH, VIDEO_HEIGHT, VIDEO_FPS);
|
||
|
|
// JPEGRtpEncoder is itself an IMediaSource (RTSPVideoEncoder : AudioEncoder,
|
||
|
|
// IMediaSource), so it can go straight into RTSPMediaStreamer - no
|
||
|
|
// RTSPMediaCallbackSource/callback wrapper needed.
|
||
|
|
JPEGRtpEncoder jpegEncoder;
|
||
|
|
|
||
|
|
RTSPMediaStreamer<RTSPPlatformWiFi> rtspStreamer(jpegEncoder);
|
||
|
|
RTSPServer<RTSPPlatformWiFi> rtspServer(rtspStreamer, port);
|
||
|
|
|
||
|
|
void setup() {
|
||
|
|
Serial.begin(115200);
|
||
|
|
AudioToolsLogger.begin(Serial, AudioToolsLogLevel::Info);
|
||
|
|
|
||
|
|
jpegEncoder.setFormat(mjpegFormat);
|
||
|
|
jpegEncoder.setMaxFragmentSize(1400);
|
||
|
|
|
||
|
|
rtspServer.begin();
|
||
|
|
}
|
||
|
|
|
||
|
|
void loop() {
|
||
|
|
static bool captureEnabled = false;
|
||
|
|
bool clientConnected = rtspServer.clientCount() > 0;
|
||
|
|
if (clientConnected != captureEnabled) {
|
||
|
|
captureEnabled = clientConnected;
|
||
|
|
if (captureEnabled) {
|
||
|
|
jpegEncoder.begin();
|
||
|
|
} else {
|
||
|
|
jpegEncoder.end();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if (captureEnabled) {
|
||
|
|
static unsigned long lastFrame = 0;
|
||
|
|
unsigned long now = millis();
|
||
|
|
if (now - lastFrame >= FRAME_DURATION_MS) {
|
||
|
|
lastFrame = now;
|
||
|
|
jpegEncoder.write(frame_jpeg, frame_jpeg_len);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|