2016-02-22 21:07:12 +09:00

130 lines
3.5 KiB
C

/*
Oh! Multicast Video Scanner
Copyright (C) 2016 Taeho Oh <ohhara@postech.edu>
This file is part of Oh! Multicast Video Scanner.
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#include <string.h>
#include <gst/gst.h>
#include "omvs_gst.h"
typedef struct _OMVSGstImpl {
GstElement *play;
GstBus *bus;
} OMVSGstImpl;
OMVSGst omvs_gst_open(const gchar *uri, const gchar *filename) {
guint flags;
OMVSGstImpl *gst_impl;
gchar *protocol;
gchar *location;
gchar *real_uri;
GstElement *play;
GstElement *conv;
GstElement *png;
GstElement *file;
GstElement *convpngfile;
GstPad *pad;
GstPad *ghostpad;
if (gst_uri_is_valid(uri)) {
protocol = gst_uri_get_protocol(uri);
if (protocol) {
location = gst_uri_get_location(uri);
if (location) {
if (!gst_uri_protocol_is_supported(GST_URI_SRC, protocol)) {
if (strcmp(protocol, "rtp") == 0) {
g_free(protocol);
protocol = g_strdup("ortp");
}
}
real_uri = g_strdup_printf("%s://%s", protocol, location);
g_free(location);
} else {
real_uri = g_strdup(uri);
}
g_free(protocol);
} else {
real_uri = g_strdup(uri);
}
} else {
real_uri = g_strdup(uri);
}
gst_impl = g_malloc0(sizeof(OMVSGstImpl));
gst_impl->play = play = gst_element_factory_make("playbin", "play");
g_object_set(G_OBJECT(play), "uri", real_uri, NULL);
g_free(real_uri);
g_object_get(play, "flags", &flags, NULL);
flags &= (~0x00000002);
g_object_set(play, "flags", flags, NULL);
conv = gst_element_factory_make("queue", "conv");
png = gst_element_factory_make("pngenc", "png");
g_object_set(G_OBJECT(png), "snapshot", 1, "compression-level", 9, NULL);
file = gst_element_factory_make("multifilesink", "file");
g_object_set(G_OBJECT(file), "max-files", 1, "location", filename, NULL);
convpngfile = gst_bin_new("convpngfile");
gst_bin_add_many(GST_BIN(convpngfile), conv, png, file, NULL);
gst_element_link_many(conv, png, file, NULL);
pad = gst_element_get_static_pad(conv, "sink");
ghostpad = gst_ghost_pad_new("sink", pad);
gst_element_add_pad(convpngfile, ghostpad);
gst_object_unref(GST_OBJECT(pad));
g_object_set(play, "video-sink", convpngfile, NULL);
g_object_set(play, "mute", TRUE, NULL);
gst_impl->bus = gst_element_get_bus(play);
gst_object_unref(GST_OBJECT(gst_impl->bus));
gst_element_set_state(play, GST_STATE_PLAYING);
return (OMVSGst)gst_impl;
}
gboolean omvs_gst_is_finished(OMVSGst gst) {
OMVSGstImpl *gst_impl;
GstMessage *msg;
g_assert(gst);
gst_impl = (OMVSGstImpl *)gst;
msg = gst_bus_pop_filtered(gst_impl->bus, GST_MESSAGE_EOS);
if (!msg) {
return FALSE;
}
gst_message_unref(msg);
return TRUE;
}
gint omvs_gst_close(OMVSGst gst) {
OMVSGstImpl *gst_impl;
gst_impl = (OMVSGstImpl *)gst;
if (gst_impl) {
gst_element_set_state(gst_impl->play, GST_STATE_NULL);
gst_object_unref(gst_impl->play);
g_free(gst_impl);
}
return 0;
}