/**************************************************************************** * * (c) 2009-2020 QGROUNDCONTROL PROJECT * * QGroundControl is licensed according to the terms in the file * COPYING.md in the root of the source code directory. * ****************************************************************************/ /** * @file * @brief QGC Video Receiver * @author Gus Grubba */ #include "VideoReceiver.h" #include "SettingsManager.h" #include "QGCApplication.h" #include "VideoManager.h" #ifdef QGC_GST_TAISYNC_ENABLED #include "TaisyncHandler.h" #endif #include #include #include #include #include QGC_LOGGING_CATEGORY(VideoReceiverLog, "VideoReceiverLog") #if defined(QGC_GST_STREAMING) static const char* kVideoExtensions[] = { "mkv", "mov", "mp4" }; static const char* kVideoMuxes[] = { "matroskamux", "qtmux", "mp4mux" }; #define NUM_MUXES (sizeof(kVideoMuxes) / sizeof(char*)) #endif VideoReceiver::VideoReceiver(QObject* parent) : QObject(parent) #if defined(QGC_GST_STREAMING) , _running(false) , _recording(false) , _streaming(false) , _starting(false) , _stopping(false) , _stop(true) , _sink(nullptr) , _tee(nullptr) , _pipeline(nullptr) , _videoSink(nullptr) , _lastFrameId(G_MAXUINT64) , _lastFrameTime(0) , _restart_time_ms(1389) , _socket(nullptr) , _serverPresent(false) , _tcpTestInterval_ms(5000) , _udpReconnect_us(5000000) #endif , _videoRunning(false) , _showFullScreen(false) , _videoSettings(nullptr) { _videoSettings = qgcApp()->toolbox()->settingsManager()->videoSettings(); #if defined(QGC_GST_STREAMING) _restart_timer.setSingleShot(true); connect(&_restart_timer, &QTimer::timeout, this, &VideoReceiver::_restart_timeout); _tcp_timer.setSingleShot(true); connect(&_tcp_timer, &QTimer::timeout, this, &VideoReceiver::_tcp_timeout); connect(this, &VideoReceiver::msgErrorReceived, this, &VideoReceiver::_handleError); connect(this, &VideoReceiver::msgEOSReceived, this, &VideoReceiver::_handleEOS); connect(this, &VideoReceiver::msgStateChangedReceived, this, &VideoReceiver::_handleStateChanged); connect(&_frameTimer, &QTimer::timeout, this, &VideoReceiver::_updateTimer); _frameTimer.start(1000); #endif } VideoReceiver::~VideoReceiver() { #if defined(QGC_GST_STREAMING) stop(); setVideoSink(nullptr); #endif } //----------------------------------------------------------------------------- void VideoReceiver::grabImage(QString imageFile) { _imageFile = imageFile; emit imageFileChanged(); } //----------------------------------------------------------------------------- #if defined(QGC_GST_STREAMING) static void newPadCB(GstElement* element, GstPad* pad, gpointer data) { gchar* name = gst_pad_get_name(pad); //g_print("A new pad %s was created\n", name); GstCaps* p_caps = gst_pad_get_pad_template_caps (pad); gchar* description = gst_caps_to_string(p_caps); qCDebug(VideoReceiverLog) << p_caps << ", " << description; g_free(description); GstElement* sink = GST_ELEMENT(data); if(gst_element_link_pads(element, name, sink, "sink") == false) qCCritical(VideoReceiverLog) << "Failed to link elements\n"; g_free(name); } static gboolean autoplugQueryCaps(GstElement* bin, GstPad* pad, GstElement* element, GstQuery* query, gpointer data) { GstElement* glupload = (GstElement* )data; GstPad* sinkpad = gst_element_get_static_pad(glupload, "sink"); if (!sinkpad) { qCCritical(VideoReceiverLog) << "No sink pad found"; return FALSE; } GstCaps* filter; gst_query_parse_caps(query, &filter); GstCaps* sinkcaps = gst_pad_query_caps(sinkpad, filter); gst_query_set_caps_result(query, sinkcaps); const gboolean ret = !gst_caps_is_empty(sinkcaps); gst_caps_unref(sinkcaps); sinkcaps = nullptr; gst_object_unref(sinkpad); sinkpad = nullptr; return ret; } static gboolean autoplugQueryContext(GstElement* bin, GstPad* pad, GstElement* element, GstQuery* query, gpointer data) { GstElement* glsink = (GstElement* )data; GstPad* sinkpad = gst_element_get_static_pad(glsink, "sink"); if (!sinkpad){ qCCritical(VideoReceiverLog) << "No sink pad found"; return FALSE; } const gboolean ret = gst_pad_query(sinkpad, query); gst_object_unref(sinkpad); sinkpad = nullptr; return ret; } static gboolean autoplugQueryCB(GstElement* bin, GstPad* pad, GstElement* element, GstQuery* query, gpointer data) { gboolean ret; switch (GST_QUERY_TYPE (query)) { case GST_QUERY_CAPS: ret = autoplugQueryCaps(bin, pad, element, query, data); break; case GST_QUERY_CONTEXT: ret = autoplugQueryContext(bin, pad, element, query, data); break; default: ret = FALSE; break; } return ret; } //----------------------------------------------------------------------------- static void _wrapWithGhostPad(GstElement* element, GstPad* pad, gpointer data) { gchar* name = gst_pad_get_name(pad); GstPad* ghostpad = gst_ghost_pad_new(name, pad); g_free(name); gst_pad_set_active(ghostpad, TRUE); if (!gst_element_add_pad(GST_ELEMENT_PARENT(element), ghostpad)) { qCCritical(VideoReceiverLog) << "Failed to add ghost pad to source"; } } static void _linkPadWithOptionalBuffer(GstElement* element, GstPad* pad, gpointer data) { gboolean isRtpPad = FALSE; GstCaps* filter = gst_caps_from_string("application/x-rtp"); if (filter != nullptr) { GstCaps* caps = gst_pad_query_caps(pad, nullptr); if (caps != nullptr) { if (!gst_caps_is_any(caps) && gst_caps_can_intersect(caps, filter)) { isRtpPad = TRUE; } gst_caps_unref(caps); caps = nullptr; } gst_caps_unref(filter); filter = nullptr; } if (isRtpPad) { GstElement* buffer; if ((buffer = gst_element_factory_make("rtpjitterbuffer", nullptr)) != nullptr) { gst_bin_add(GST_BIN(GST_ELEMENT_PARENT(element)), buffer); gst_element_sync_state_with_parent(buffer); GstPad* sinkpad = gst_element_get_static_pad(buffer, "sink"); if (sinkpad != nullptr) { const GstPadLinkReturn ret = gst_pad_link(pad, sinkpad); gst_object_unref(sinkpad); sinkpad = nullptr; if (ret == GST_PAD_LINK_OK) { pad = gst_element_get_static_pad(buffer, "src"); element = buffer; } else { qCDebug(VideoReceiverLog) << "Partially failed - gst_pad_link()"; } } else { qCDebug(VideoReceiverLog) << "Partially failed - gst_element_get_static_pad()"; } } else { qCDebug(VideoReceiverLog) << "Partially failed - gst_element_factory_make('rtpjitterbuffer')"; } } newPadCB(element, pad, data); } static gboolean _padProbe(GstElement* element, GstPad* pad, gpointer user_data) { int* probeRes = (int*)user_data; *probeRes |= 1; GstCaps* filter = gst_caps_from_string("application/x-rtp"); if (filter != nullptr) { GstCaps* caps = gst_pad_query_caps(pad, nullptr); if (caps != nullptr) { if (!gst_caps_is_any(caps) && gst_caps_can_intersect(caps, filter)) { *probeRes |= 2; } gst_caps_unref(caps); caps = nullptr; } gst_caps_unref(filter); filter = nullptr; } return TRUE; } GstElement* VideoReceiver::_makeSource(const QString& uri) { if (uri.isEmpty()) { qCCritical(VideoReceiverLog) << "Failed because URI is not specified"; return nullptr; } bool isTaisync = uri.contains("tsusb://"); bool isUdp264 = uri.contains("udp://"); bool isRtsp = uri.contains("rtsp://"); bool isUdp265 = uri.contains("udp265://"); bool isTcpMPEGTS= uri.contains("tcp://"); bool isUdpMPEGTS= uri.contains("mpegts://"); GstElement* source = nullptr; GstElement* buffer = nullptr; GstElement* parser = nullptr; GstElement* bin = nullptr; GstElement* srcbin = nullptr; do { QUrl url(uri); if(isTcpMPEGTS) { if ((source = gst_element_factory_make("tcpclientsrc", "source")) != nullptr) { g_object_set(static_cast(source), "host", qPrintable(url.host()), "port", url.port(), nullptr); } } else if (isRtsp) { if ((source = gst_element_factory_make("rtspsrc", "source")) != nullptr) { g_object_set(static_cast(source), "location", qPrintable(uri), "latency", 17, "udp-reconnect", 1, "timeout", _udpReconnect_us, NULL); } } else if(isUdp264 || isUdp265 || isUdpMPEGTS || isTaisync) { if ((source = gst_element_factory_make("udpsrc", "source")) != nullptr) { g_object_set(static_cast(source), "uri", QString("udp://%1:%2").arg(qPrintable(url.host()), QString::number(url.port())).toUtf8().data(), nullptr); GstCaps* caps = nullptr; if(isUdp264) { if ((caps = gst_caps_from_string("application/x-rtp, media=(string)video, clock-rate=(int)90000, encoding-name=(string)H264")) == nullptr) { qCCritical(VideoReceiverLog) << "gst_caps_from_string() failed"; break; } } else if (isUdp264) { if ((caps = gst_caps_from_string("application/x-rtp, media=(string)video, clock-rate=(int)90000, encoding-name=(string)H265")) == nullptr) { qCCritical(VideoReceiverLog) << "gst_caps_from_string() failed"; break; } } if (caps != nullptr) { g_object_set(static_cast(source), "caps", caps, nullptr); gst_caps_unref(caps); caps = nullptr; } } } else { qCDebug(VideoReceiverLog) << "URI is not recognized"; } if (!source) { qCCritical(VideoReceiverLog) << "gst_element_factory_make() for data source failed"; break; } // FIXME: AV: Android does not determine MPEG2-TS via parsebin - have to explicitly state which demux to use if (isTcpMPEGTS || isUdpMPEGTS) { if ((parser = gst_element_factory_make("tsdemux", "parser")) == nullptr) { qCritical(VideoReceiverLog) << "gst_element_factory_make('tsdemux') failed"; break; } } else { if ((parser = gst_element_factory_make("parsebin", "parser")) == nullptr) { qCritical() << "VideoReceiver::_makeSource() failed. Error with gst_element_factory_make('parsebin')"; break; } } if ((bin = gst_bin_new("sourcebin")) == nullptr) { qCCritical(VideoReceiverLog) << "gst_bin_new('sourcebin') failed"; break; } gst_bin_add_many(GST_BIN(bin), source, parser, nullptr); int probeRes = 0; gst_element_foreach_src_pad(source, _padProbe, &probeRes); if (probeRes & 1) { if (probeRes & 2) { if ((buffer = gst_element_factory_make("rtpjitterbuffer", nullptr)) == nullptr) { qCCritical(VideoReceiverLog) << "gst_element_factory_make('rtpjitterbuffer') failed"; break; } gst_bin_add(GST_BIN(bin), buffer); if (!gst_element_link_many(source, buffer, parser, nullptr)) { qCCritical(VideoReceiverLog) << "gst_element_link() failed"; break; } } else { if (!gst_element_link(source, parser)) { qCCritical(VideoReceiverLog) << "gst_element_link() failed"; break; } } } else { g_signal_connect(source, "pad-added", G_CALLBACK(_linkPadWithOptionalBuffer), parser); } g_signal_connect(parser, "pad-added", G_CALLBACK(_wrapWithGhostPad), nullptr); source = buffer = parser = nullptr; srcbin = bin; bin = nullptr; } while(0); if (bin != nullptr) { gst_object_unref(bin); bin = nullptr; } if (parser != nullptr) { gst_object_unref(parser); parser = nullptr; } if (buffer != nullptr) { gst_object_unref(buffer); buffer = nullptr; } if (source != nullptr) { gst_object_unref(source); source = nullptr; } return srcbin; } //----------------------------------------------------------------------------- void VideoReceiver::_restart_timeout() { qgcApp()->toolbox()->videoManager()->restartVideo(); } #endif //----------------------------------------------------------------------------- #if defined(QGC_GST_STREAMING) void VideoReceiver::_tcp_timeout() { //-- If socket is live, we got no connection nor a socket error delete _socket; _socket = nullptr; if(_videoSettings->streamEnabled()->rawValue().toBool()) { //-- RTSP will try to connect to the server. If it cannot connect, // it will simply give up and never try again. Instead, we keep // attempting a connection on this timer. Once a connection is // found to be working, only then we actually start the stream. QUrl url(_uri); //-- If RTSP and no port is defined, set default RTSP port (554) if(_uri.contains("rtsp://") && url.port() <= 0) { url.setPort(554); } _socket = new QTcpSocket; QNetworkProxy tempProxy; tempProxy.setType(QNetworkProxy::DefaultProxy); _socket->setProxy(tempProxy); connect(_socket, static_cast(&QTcpSocket::error), this, &VideoReceiver::_socketError); connect(_socket, &QTcpSocket::connected, this, &VideoReceiver::_connected); _socket->connectToHost(url.host(), static_cast(url.port())); _tcp_timer.start(_tcpTestInterval_ms); } } #endif //----------------------------------------------------------------------------- #if defined(QGC_GST_STREAMING) void VideoReceiver::_connected() { //-- Server showed up. Now we start the stream. _tcp_timer.stop(); _socket->deleteLater(); _socket = nullptr; if(_videoSettings->streamEnabled()->rawValue().toBool()) { _serverPresent = true; start(); } } #endif //----------------------------------------------------------------------------- #if defined(QGC_GST_STREAMING) void VideoReceiver::_socketError(QAbstractSocket::SocketError socketError) { Q_UNUSED(socketError); _socket->deleteLater(); _socket = nullptr; //-- Try again in a while if(_videoSettings->streamEnabled()->rawValue().toBool()) { _tcp_timer.start(_tcpTestInterval_ms); } } #endif //----------------------------------------------------------------------------- // When we finish our pipeline will look like this: // // +-->queue-->decoder-->_videosink // | // datasource-->demux-->parser-->tee // ^ // | // +-Here we will later link elements for recording void VideoReceiver::start() { qCDebug(VideoReceiverLog) << "Starting " << _uri; if(qgcApp()->runningUnitTests()) { return; } if(!_videoSettings->streamEnabled()->rawValue().toBool() || !_videoSettings->streamConfigured()) { qCDebug(VideoReceiverLog) << "Stream not enabled/configured"; return; } #if defined(QGC_GST_STREAMING) _stop = false; QString uri = _uri; #if defined(QGC_GST_TAISYNC_ENABLED) && (defined(__android__) || defined(__ios__)) //-- Taisync on iOS or Android sends a raw h.264 stream if (qgcApp()->toolbox()->videoManager()->isTaisync()) { uri = QString("tsusb://0.0.0.0:%1").arg(TAISYNC_VIDEO_UDP_PORT); } #endif if (uri.isEmpty()) { qCDebug(VideoReceiverLog) << "Failed because URI is not specified"; return; } bool useTcpConnection = uri.contains("rtsp://") || uri.contains("tcp://"); if (_videoSink == nullptr) { qCWarning(VideoReceiverLog) << "Failed because video sink is not set"; return; } if(_running) { qCDebug(VideoReceiverLog) << "Already running!"; return; } _starting = true; //-- For RTSP and TCP, check to see if server is there first if(!_serverPresent && useTcpConnection) { _tcp_timer.start(100); return; } _lastFrameId = G_MAXUINT64; _lastFrameTime = 0; bool running = false; bool pipelineUp = false; GstElement* source = nullptr; GstElement* queue = nullptr; GstElement* decoder = nullptr; do { if ((_pipeline = gst_pipeline_new("receiver")) == nullptr) { qCCritical(VideoReceiverLog) << "gst_pipeline_new() failed"; break; } g_object_set(_pipeline, "message-forward", TRUE, nullptr); if ((source = _makeSource(uri)) == nullptr) { qCCritical(VideoReceiverLog) << "_makeSource() failed"; break; } if((_tee = gst_element_factory_make("tee", nullptr)) == nullptr) { qCCritical(VideoReceiverLog) << "gst_element_factory_make('tee') failed"; break; } if((queue = gst_element_factory_make("queue", nullptr)) == nullptr) { // TODO: We may want to add queue2 max-size-buffers=1 to get lower latency // We should compare gstreamer scripts to QGroundControl to determine the need qCCritical(VideoReceiverLog) << "gst_element_factory_make('queue') failed"; break; } if ((decoder = gst_element_factory_make("decodebin", "decoder")) == nullptr) { qCCritical(VideoReceiverLog) << "gst_element_factory_make('decodebin') failed"; break; } gst_bin_add_many(GST_BIN(_pipeline), source, _tee, queue, decoder, _videoSink, nullptr); pipelineUp = true; g_signal_connect(source, "pad-added", G_CALLBACK(newPadCB), _tee); if(!gst_element_link_many(_tee, queue, decoder, nullptr)) { qCCritical(VideoReceiverLog) << "Unable to receiver pipeline."; break; } g_signal_connect(decoder, "pad-added", G_CALLBACK(newPadCB), _videoSink); g_signal_connect(decoder, "autoplug-query", G_CALLBACK(autoplugQueryCB), _videoSink); source = queue = decoder = nullptr; GstBus* bus = nullptr; if ((bus = gst_pipeline_get_bus(GST_PIPELINE(_pipeline))) != nullptr) { gst_bus_enable_sync_message_emission(bus); g_signal_connect(bus, "sync-message", G_CALLBACK(_onBusMessage), this); gst_object_unref(bus); bus = nullptr; } GST_DEBUG_BIN_TO_DOT_FILE(GST_BIN(_pipeline), GST_DEBUG_GRAPH_SHOW_ALL, "pipeline-paused"); running = gst_element_set_state(_pipeline, GST_STATE_PLAYING) != GST_STATE_CHANGE_FAILURE; } while(0); if (!running) { qCCritical(VideoReceiverLog) << "Failed"; // In newer versions, the pipeline will clean up all references that are added to it if (_pipeline != nullptr) { gst_bin_remove(GST_BIN(_pipeline), _videoSink); gst_object_unref(_pipeline); _pipeline = nullptr; } // If we failed before adding items to the pipeline, then clean up if (!pipelineUp) { if (decoder != nullptr) { gst_object_unref(decoder); decoder = nullptr; } if (queue != nullptr) { gst_object_unref(queue); queue = nullptr; } if (source != nullptr) { gst_object_unref(source); source = nullptr; } if (_tee != nullptr) { gst_object_unref(_tee); _tee = nullptr; } } _running = false; } else { GST_DEBUG_BIN_TO_DOT_FILE(GST_BIN(_pipeline), GST_DEBUG_GRAPH_SHOW_ALL, "pipeline-playing"); _running = true; qCDebug(VideoReceiverLog) << "Running"; } _starting = false; #endif } //----------------------------------------------------------------------------- void VideoReceiver::stop() { if(qgcApp() && qgcApp()->runningUnitTests()) { return; } #if defined(QGC_GST_STREAMING) _stop = true; qCDebug(VideoReceiverLog) << "Stopping"; if(!_streaming) { _shutdownPipeline(); } else if (_pipeline != nullptr && !_stopping) { qCDebug(VideoReceiverLog) << "Stopping _pipeline"; GstBus* bus = gst_pipeline_get_bus(GST_PIPELINE(_pipeline)); gst_bus_disable_sync_message_emission(bus); gst_element_send_event(_pipeline, gst_event_new_eos()); _stopping = true; GstMessage* message = gst_bus_timed_pop_filtered(bus, GST_CLOCK_TIME_NONE, (GstMessageType)(GST_MESSAGE_EOS|GST_MESSAGE_ERROR)); gst_object_unref(bus); if(GST_MESSAGE_TYPE(message) == GST_MESSAGE_ERROR) { _shutdownPipeline(); qCCritical(VideoReceiverLog) << "Error stopping pipeline!"; } else if(GST_MESSAGE_TYPE(message) == GST_MESSAGE_EOS) { _handleEOS(); } gst_message_unref(message); } #endif } //----------------------------------------------------------------------------- void VideoReceiver::setUri(const QString & uri) { _uri = uri; } //----------------------------------------------------------------------------- #if defined(QGC_GST_STREAMING) void VideoReceiver::_shutdownPipeline() { if(!_pipeline) { qCDebug(VideoReceiverLog) << "No pipeline"; return; } GstBus* bus = nullptr; if ((bus = gst_pipeline_get_bus(GST_PIPELINE(_pipeline))) != nullptr) { gst_bus_disable_sync_message_emission(bus); gst_object_unref(bus); bus = nullptr; } gst_element_set_state(_pipeline, GST_STATE_NULL); gst_bin_remove(GST_BIN(_pipeline), _videoSink); _tee = nullptr; gst_object_unref(_pipeline); _pipeline = nullptr; delete _sink; _sink = nullptr; _serverPresent = false; _streaming = false; _recording = false; _stopping = false; _running = false; emit recordingChanged(); } #endif //----------------------------------------------------------------------------- #if defined(QGC_GST_STREAMING) void VideoReceiver::_handleError() { qCDebug(VideoReceiverLog) << "Gstreamer error!"; stop(); _restart_timer.start(_restart_time_ms); } #endif //----------------------------------------------------------------------------- #if defined(QGC_GST_STREAMING) void VideoReceiver::_handleEOS() { if(_stopping) { if(_recording && _sink->removing) { _shutdownRecordingBranch(); } _shutdownPipeline(); qCDebug(VideoReceiverLog) << "Stopped"; } else if(_recording && _sink->removing) { _shutdownRecordingBranch(); } else { qCWarning(VideoReceiverLog) << "Unexpected EOS!"; _handleError(); } } #endif //----------------------------------------------------------------------------- #if defined(QGC_GST_STREAMING) void VideoReceiver::_handleStateChanged() { if(_pipeline) { _streaming = GST_STATE(_pipeline) == GST_STATE_PLAYING; //qCDebug(VideoReceiverLog) << "State changed, _streaming:" << _streaming; } } #endif //----------------------------------------------------------------------------- #if defined(QGC_GST_STREAMING) gboolean VideoReceiver::_onBusMessage(GstBus* bus, GstMessage* msg, gpointer data) { Q_UNUSED(bus) Q_ASSERT(msg != nullptr && data != nullptr); VideoReceiver* pThis = (VideoReceiver*)data; switch(GST_MESSAGE_TYPE(msg)) { case(GST_MESSAGE_ERROR): { gchar* debug; GError* error; gst_message_parse_error(msg, &error, &debug); g_free(debug); qCCritical(VideoReceiverLog) << error->message; g_error_free(error); pThis->msgErrorReceived(); } break; case(GST_MESSAGE_EOS): pThis->msgEOSReceived(); break; case(GST_MESSAGE_STATE_CHANGED): pThis->msgStateChangedReceived(); break; case(GST_MESSAGE_ELEMENT): { const GstStructure *s = gst_message_get_structure (msg); if (gst_structure_has_name (s, "GstBinForwarded")) { GstMessage *forward_msg = NULL; gst_structure_get (s, "message", GST_TYPE_MESSAGE, &forward_msg, NULL); if (forward_msg != nullptr) { if (GST_MESSAGE_TYPE(forward_msg) == GST_MESSAGE_EOS) { pThis->msgEOSReceived(); } gst_message_unref(forward_msg); forward_msg = nullptr; } } } break; default: break; } return TRUE; } #endif //----------------------------------------------------------------------------- #if defined(QGC_GST_STREAMING) void VideoReceiver::_cleanupOldVideos() { //-- Only perform cleanup if storage limit is enabled if(_videoSettings->enableStorageLimit()->rawValue().toBool()) { QString savePath = qgcApp()->toolbox()->settingsManager()->appSettings()->videoSavePath(); QDir videoDir = QDir(savePath); videoDir.setFilter(QDir::Files | QDir::Readable | QDir::NoSymLinks | QDir::Writable); videoDir.setSorting(QDir::Time); //-- All the movie extensions we support QStringList nameFilters; for(uint32_t i = 0; i < NUM_MUXES; i++) { nameFilters << QString("*.") + QString(kVideoExtensions[i]); } videoDir.setNameFilters(nameFilters); //-- get the list of videos stored QFileInfoList vidList = videoDir.entryInfoList(); if(!vidList.isEmpty()) { uint64_t total = 0; //-- Settings are stored using MB uint64_t maxSize = (_videoSettings->maxVideoSize()->rawValue().toUInt() * 1024 * 1024); //-- Compute total used storage for(int i = 0; i < vidList.size(); i++) { total += vidList[i].size(); } //-- Remove old movies until max size is satisfied. while(total >= maxSize && !vidList.isEmpty()) { total -= vidList.last().size(); qCDebug(VideoReceiverLog) << "Removing old video file:" << vidList.last().filePath(); QFile file (vidList.last().filePath()); file.remove(); vidList.removeLast(); } } } } #endif //----------------------------------------------------------------------------- #if defined(QGC_GST_STREAMING) void VideoReceiver::setVideoSink(GstElement* videoSink) { if(_pipeline != nullptr) { qCDebug(VideoReceiverLog) << "Video receiver pipeline is active, video sink change is not possible"; return; } if (_videoSink != nullptr) { gst_object_unref(_videoSink); _videoSink = nullptr; } if (videoSink != nullptr) { _videoSink = videoSink; gst_object_ref(_videoSink); GstPad* pad = gst_element_get_static_pad(_videoSink, "sink"); if (pad != nullptr) { gst_pad_add_probe(pad, (GstPadProbeType)(GST_PAD_PROBE_TYPE_BUFFER), _videoSinkProbe, this, nullptr); gst_object_unref(pad); pad = nullptr; } else { qCCritical(VideoReceiverLog) << "Unable to find sink pad of video sink"; } } } #endif //----------------------------------------------------------------------------- // When we finish our pipeline will look like this: // // +-->queue-->decoder-->_videosink // | // source-->tee // | // | +---------_sink----------+ // | | | // we are adding these elements-> +->teepad-->queue-->_filesink | // | | // +------------------------+ #if defined(QGC_GST_STREAMING) GstElement* VideoReceiver::_makeFileSink(const QString& videoFile, unsigned format) { GstElement* fileSink = nullptr; GstElement* mux = nullptr; GstElement* sink = nullptr; GstElement* bin = nullptr; bool releaseElements = true; do{ if ((mux = gst_element_factory_make(kVideoMuxes[format], nullptr)) == nullptr) { qCCritical(VideoReceiverLog) << "gst_element_factory_make('" << kVideoMuxes[format] << "') failed"; break; } if ((sink = gst_element_factory_make("filesink", nullptr)) == nullptr) { qCCritical(VideoReceiverLog) << "gst_element_factory_make('filesink') failed"; break; } g_object_set(static_cast(sink), "location", qPrintable(videoFile), nullptr); if ((bin = gst_bin_new("sinkbin")) == nullptr) { qCCritical(VideoReceiverLog) << "gst_bin_new('sinkbin') failed"; break; } GstPadTemplate* padTemplate; if ((padTemplate = gst_element_class_get_pad_template(GST_ELEMENT_GET_CLASS(mux), "video_%u")) == nullptr) { qCCritical(VideoReceiverLog) << "gst_element_class_get_pad_template(mux) failed"; break; } // FIXME: AV: pad handling is potentially leaking (and other similar places too!) GstPad* pad; if ((pad = gst_element_request_pad(mux, padTemplate, nullptr, nullptr)) == nullptr) { qCCritical(VideoReceiverLog) << "gst_element_request_pad(mux) failed"; break; } gst_bin_add_many(GST_BIN(bin), mux, sink, nullptr); releaseElements = false; GstPad* ghostpad = gst_ghost_pad_new("sink", pad); gst_element_add_pad(bin, ghostpad); gst_object_unref(pad); pad = nullptr; if (!gst_element_link(mux, sink)) { qCCritical(VideoReceiverLog) << "gst_element_link() failed"; break; } fileSink = bin; bin = nullptr; } while(0); if (releaseElements) { if (sink != nullptr) { gst_object_unref(sink); sink = nullptr; } if (mux != nullptr) { gst_object_unref(mux); mux = nullptr; } } if (bin != nullptr) { gst_object_unref(bin); bin = nullptr; } return fileSink; } #endif void VideoReceiver::startRecording(const QString &videoFile) { #if defined(QGC_GST_STREAMING) qCDebug(VideoReceiverLog) << "Starting recording"; // exit immediately if we are already recording if(_pipeline == nullptr || _recording) { qCDebug(VideoReceiverLog) << "Already recording!"; return; } uint32_t muxIdx = _videoSettings->recordingFormat()->rawValue().toUInt(); if(muxIdx >= NUM_MUXES) { qgcApp()->showMessage(tr("Invalid video format defined.")); return; } //-- Disk usage maintenance _cleanupOldVideos(); QString savePath = qgcApp()->toolbox()->settingsManager()->appSettings()->videoSavePath(); if(savePath.isEmpty()) { qgcApp()->showMessage(tr("Unabled to record video. Video save path must be specified in Settings.")); return; } _videoFile = savePath + "/" + (videoFile.isEmpty() ? QDateTime::currentDateTime().toString("yyyy-MM-dd_hh.mm.ss") : videoFile) + "." + kVideoExtensions[muxIdx]; qCDebug(VideoReceiverLog) << "New video file:" << _videoFile; emit videoFileChanged(); _sink = new Sink(); _sink->teepad = gst_element_get_request_pad(_tee, "src_%u"); _sink->queue = gst_element_factory_make("queue", nullptr); _sink->filesink = _makeFileSink(_videoFile, muxIdx); _sink->removing = false; if(!_sink->teepad || !_sink->queue || !_sink->filesink) { qCCritical(VideoReceiverLog) << "Failed to make _sink elements"; return; } gst_object_ref(_sink->queue); gst_object_ref(_sink->filesink); gst_bin_add(GST_BIN(_pipeline), _sink->queue); gst_element_sync_state_with_parent(_sink->queue); // Install a probe on the recording branch to drop buffers until we hit our first keyframe // When we hit our first keyframe, we can offset the timestamps appropriately according to the first keyframe time // This will ensure the first frame is a keyframe at t=0, and decoding can begin immediately on playback // Once we have this valid frame, we attach the filesink. // Attaching it here would cause the filesink to fail to preroll and to stall the pipeline for a few seconds. GstPad* probepad = gst_element_get_static_pad(_sink->queue, "src"); gst_pad_add_probe(probepad, GST_PAD_PROBE_TYPE_BUFFER, _keyframeWatch, this, nullptr); // to drop the buffer gst_object_unref(probepad); // Link the recording branch to the pipeline GstPad* sinkpad = gst_element_get_static_pad(_sink->queue, "sink"); gst_pad_link(_sink->teepad, sinkpad); gst_object_unref(sinkpad); // // Add the filesink once we have a valid I-frame // gst_bin_add(GST_BIN(_pipeline), _sink->filesink); // if (!gst_element_link(_sink->queue, _sink->filesink)) { // qCritical() << "Failed to link queue and file sink"; // } // gst_element_sync_state_with_parent(_sink->filesink); GST_DEBUG_BIN_TO_DOT_FILE(GST_BIN(_pipeline), GST_DEBUG_GRAPH_SHOW_ALL, "pipeline-recording"); _recording = true; emit recordingChanged(); qCDebug(VideoReceiverLog) << "Recording started"; #else Q_UNUSED(videoFile) #endif } //----------------------------------------------------------------------------- void VideoReceiver::stopRecording(void) { #if defined(QGC_GST_STREAMING) qCDebug(VideoReceiverLog) << "Stopping recording"; // exit immediately if we are not recording if(_pipeline == nullptr || !_recording) { qCDebug(VideoReceiverLog) << "Not recording!"; return; } // Wait for data block before unlinking gst_pad_add_probe(_sink->teepad, GST_PAD_PROBE_TYPE_IDLE, _unlinkCallBack, this, nullptr); #endif } //----------------------------------------------------------------------------- // This is only installed on the transient _pipelineStopRec in order // to finalize a video file. It is not used for the main _pipeline. // -EOS has appeared on the bus of the temporary pipeline // -At this point all of the recoring elements have been flushed, and the video file has been finalized // -Now we can remove the temporary pipeline and its elements #if defined(QGC_GST_STREAMING) void VideoReceiver::_shutdownRecordingBranch() { gst_bin_remove(GST_BIN(_pipeline), _sink->queue); gst_bin_remove(GST_BIN(_pipeline), _sink->filesink); gst_element_set_state(_sink->queue, GST_STATE_NULL); gst_element_set_state(_sink->filesink, GST_STATE_NULL); gst_object_unref(_sink->queue); gst_object_unref(_sink->filesink); delete _sink; _sink = nullptr; _recording = false; emit recordingChanged(); qCDebug(VideoReceiverLog) << "Recording stopped"; } #endif //----------------------------------------------------------------------------- // -Unlink the recording branch from the tee in the main _pipeline // -Create a second temporary pipeline, and place the recording branch elements into that pipeline // -Setup watch and handler for EOS event on the temporary pipeline's bus // -Send an EOS event at the beginning of that pipeline #if defined(QGC_GST_STREAMING) void VideoReceiver::_unlinkRecordingBranch(GstPadProbeInfo* info) { Q_UNUSED(info) // Send EOS at the beginning of the pipeline GstPad* sinkpad = gst_element_get_static_pad(_sink->queue, "sink"); gst_pad_unlink(_sink->teepad, sinkpad); gst_pad_send_event(sinkpad, gst_event_new_eos()); gst_object_unref(sinkpad); qCDebug(VideoReceiverLog) << "Recording EOS was sent"; // Give tee its pad back gst_element_release_request_pad(_tee, _sink->teepad); gst_object_unref(_sink->teepad); } #endif //----------------------------------------------------------------------------- #if defined(QGC_GST_STREAMING) GstPadProbeReturn VideoReceiver::_unlinkCallBack(GstPad* pad, GstPadProbeInfo* info, gpointer user_data) { Q_UNUSED(pad); if(info != nullptr && user_data != nullptr) { VideoReceiver* pThis = static_cast(user_data); // We will only act once if(g_atomic_int_compare_and_exchange(&pThis->_sink->removing, FALSE, TRUE)) { pThis->_unlinkRecordingBranch(info); } } return GST_PAD_PROBE_REMOVE; } #endif //----------------------------------------------------------------------------- #if defined(QGC_GST_STREAMING) GstPadProbeReturn VideoReceiver::_videoSinkProbe(GstPad* pad, GstPadProbeInfo* info, gpointer user_data) { Q_UNUSED(pad); if(info != nullptr && user_data != nullptr) { VideoReceiver* pThis = static_cast(user_data); pThis->_noteVideoSinkFrame(); } return GST_PAD_PROBE_OK; } #endif //----------------------------------------------------------------------------- #if defined(QGC_GST_STREAMING) void VideoReceiver::_noteVideoSinkFrame() { _lastFrameTime = QDateTime::currentSecsSinceEpoch(); } #endif //----------------------------------------------------------------------------- #if defined(QGC_GST_STREAMING) GstPadProbeReturn VideoReceiver::_keyframeWatch(GstPad* pad, GstPadProbeInfo* info, gpointer user_data) { Q_UNUSED(pad); if(info != nullptr && user_data != nullptr) { GstBuffer* buf = gst_pad_probe_info_get_buffer(info); if(GST_BUFFER_FLAG_IS_SET(buf, GST_BUFFER_FLAG_DELTA_UNIT)) { // wait for a keyframe return GST_PAD_PROBE_DROP; } else { VideoReceiver* pThis = static_cast(user_data); // set media file '0' offset to current timeline position - we don't want to touch other elements in the graph, except these which are downstream! gst_pad_set_offset(pad, -buf->pts); // Add the filesink once we have a valid I-frame gst_bin_add(GST_BIN(pThis->_pipeline), pThis->_sink->filesink); if (!gst_element_link(pThis->_sink->queue, pThis->_sink->filesink)) { qCCritical(VideoReceiverLog) << "Failed to link queue and file sink"; } gst_element_sync_state_with_parent(pThis->_sink->filesink); qCDebug(VideoReceiverLog) << "Got keyframe, stop dropping buffers"; pThis->gotFirstRecordingKeyFrame(); } } return GST_PAD_PROBE_REMOVE; } #endif //----------------------------------------------------------------------------- void VideoReceiver::_updateTimer() { #if defined(QGC_GST_STREAMING) if(_stopping || _starting) { return; } if(_streaming) { if(!_videoRunning) { _videoRunning = true; emit videoRunningChanged(); } } else { if(_videoRunning) { _videoRunning = false; emit videoRunningChanged(); } } if(_videoRunning) { uint32_t timeout = 1; if(qgcApp()->toolbox() && qgcApp()->toolbox()->settingsManager()) { timeout = _videoSettings->rtspTimeout()->rawValue().toUInt(); } const qint64 now = QDateTime::currentSecsSinceEpoch(); if(now - _lastFrameTime > timeout) { stop(); // We want to start it back again with _updateTimer _stop = false; } } else { // FIXME: AV: if pipeline is _running but not _streaming for some time then we need to restart if(!_stop && !_running && !_uri.isEmpty() && _videoSettings->streamEnabled()->rawValue().toBool()) { start(); } } #endif }