gstreamer/test/video2mp1.c

235 lines
6.6 KiB
C
Raw Normal View History

#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <string.h>
#include <glib.h>
#include <gst/gst.h>
#include <gnome.h>
static void
gst_play_have_type (GstElement *sink, GstElement *sink2, gpointer data)
{
GST_DEBUG (0,"GstPipeline: play have type %p\n", (gboolean *)data);
*(gboolean *)data = TRUE;
}
gboolean
idle_func (gpointer data)
{
return gst_bin_iterate (GST_BIN (data));
}
static GstCaps*
gst_play_typefind (GstBin *bin, GstElement *element)
{
gboolean found = FALSE;
GstElement *typefind;
GstCaps *caps = NULL;
GST_DEBUG (0,"GstPipeline: typefind for element \"%s\" %p\n",
GST_ELEMENT_NAME(element), &found);
typefind = gst_elementfactory_make ("typefind", "typefind");
g_return_val_if_fail (typefind != NULL, FALSE);
gtk_signal_connect (GTK_OBJECT (typefind), "have_type",
GTK_SIGNAL_FUNC (gst_play_have_type), &found);
gst_pad_connect (gst_element_get_pad (element, "src"),
gst_element_get_pad (typefind, "sink"));
gst_bin_add (bin, typefind);
gst_element_set_state (GST_ELEMENT (bin), GST_STATE_PLAYING);
// push a buffer... the have_type signal handler will set the found flag
gst_bin_iterate (bin);
gst_element_set_state (GST_ELEMENT (bin), GST_STATE_NULL);
caps = gst_pad_get_caps (gst_element_get_pad (element, "src"));
gst_pad_disconnect (gst_element_get_pad (element, "src"),
gst_element_get_pad (typefind, "sink"));
gst_bin_remove (bin, typefind);
return caps;
}
static GstElement*
get_video_encoder_bin (void)
{
GstElement *bin;
GstElement *encoder, *queue, *colorspace, *videoscale;
bin = gst_bin_new ("video_encoder_bin");
colorspace = gst_elementfactory_make ("colorspace", "colorspace");
g_assert (colorspace != NULL);
videoscale = gst_elementfactory_make ("videoscale", "videoscale");
g_assert (videoscale != NULL);
gtk_object_set (GTK_OBJECT (videoscale), "width", 352, "height", 288, NULL);
encoder = gst_elementfactory_make ("mpeg2enc", "video_encoder");
g_assert (encoder != NULL);
queue = gst_elementfactory_make ("queue", "video_encoder_queue");
g_assert (queue != NULL);
gst_bin_add (GST_BIN (bin), colorspace);
gst_bin_add (GST_BIN (bin), videoscale);
gst_bin_add (GST_BIN (bin), encoder);
gst_bin_add (GST_BIN (bin), queue);
gst_element_connect (colorspace, "src", videoscale, "sink");
gst_element_connect (videoscale, "src", encoder, "sink");
gst_element_connect (encoder, "src", queue, "sink");
gst_element_add_ghost_pad (bin, gst_element_get_pad (colorspace, "sink"), "sink");
gst_element_add_ghost_pad (bin, gst_element_get_pad (queue, "src"), "src");
return bin;
}
static GstElement*
get_audio_encoder_bin (void)
{
GstElement *bin;
GstElement *encoder, *queue;
bin = gst_bin_new ("audio_encoder_bin");
encoder = gst_elementfactory_make ("mpegaudio", "audio_encoder");
g_assert (encoder != NULL);
queue = gst_elementfactory_make ("queue", "audio_encoder_queue");
g_assert (queue != NULL);
gst_bin_add (GST_BIN (bin), encoder);
gst_bin_add (GST_BIN (bin), queue);
gst_element_connect (encoder, "src", queue, "sink");
gst_element_add_ghost_pad (bin, gst_element_get_pad (encoder, "sink"), "sink");
gst_element_add_ghost_pad (bin, gst_element_get_pad (queue, "src"), "src");
return bin;
}
int main(int argc,char *argv[])
{
GstElement *disksrc, *audio_enc, *video_enc;
GstElement *muxthread_video, *muxer, *fdsink_video;
GstElement *muxthread_audio, *fdsink_audio;
GstElement *bin;
GstCaps *srccaps;
GstElement *new_element;
GstAutoplug *autoplug;
gint fd_video;
gint fd_audio;
g_thread_init(NULL);
gst_init(&argc,&argv);
gnome_init("autoplug","0.0.1", argc,argv);
if (argc != 4) {
g_print("usage: %s <in_filename> <out_video> <out_audio>\n", argv[0]);
exit(-1);
}
/* create a new bin to hold the elements */
bin = gst_pipeline_new("pipeline");
g_assert(bin != NULL);
/* create a disk reader */
disksrc = gst_elementfactory_make("disksrc", "disk_source");
g_assert(disksrc != NULL);
gtk_object_set(GTK_OBJECT(disksrc),"location", argv[1],NULL);
gst_bin_add (GST_BIN (bin), disksrc);
srccaps = gst_play_typefind (GST_BIN (bin), disksrc);
if (!srccaps) {
g_print ("could not autoplug, unknown media type...\n");
exit (-1);
}
audio_enc = get_audio_encoder_bin();
video_enc = get_video_encoder_bin();
autoplug = gst_autoplugfactory_make ("staticrender");
g_assert (autoplug != NULL);
new_element = gst_autoplug_to_renderers (autoplug,
srccaps,
video_enc,
audio_enc,
NULL);
if (!new_element) {
g_print ("could not autoplug, no suitable codecs found...\n");
exit (-1);
}
gst_object_ref (GST_OBJECT (disksrc));
gst_bin_remove (GST_BIN (bin), disksrc);
gst_object_destroy (GST_OBJECT (bin));
// FIXME hack, reparent the disksrc so the scheduler doesn't break
bin = gst_pipeline_new("pipeline");
gst_bin_add (GST_BIN (bin), disksrc);
gst_bin_add (GST_BIN (bin), new_element);
gst_element_connect (disksrc, "src", new_element, "sink");
muxer = gst_elementfactory_make ("system_encode", "muxer");
g_assert (muxer != NULL);
if (gst_bin_get_by_name (GST_BIN (new_element), "video_encoder_bin")) {
muxthread_video = gst_thread_new("thread_video");
fdsink_video = gst_elementfactory_make ("fdsink", "fdsink_video");
g_assert (fdsink_video != NULL);
fd_video = open (argv[2], O_CREAT|O_RDWR|O_TRUNC);
gtk_object_set (GTK_OBJECT (fdsink_video), "fd", fd_video, NULL);
gst_element_connect (video_enc, "src", fdsink_video, "sink");
gst_bin_add (GST_BIN (muxthread_video), fdsink_video);
gst_bin_add (GST_BIN (bin), muxthread_video);
}
if (gst_bin_get_by_name (GST_BIN (new_element), "audio_encoder_bin")) {
muxthread_audio = gst_thread_new("thread_audio");
fdsink_audio = gst_elementfactory_make ("fdsink", "fdsink_audio");
g_assert (fdsink_audio != NULL);
fd_audio = open (argv[3], O_CREAT|O_RDWR|O_TRUNC);
gtk_object_set (GTK_OBJECT (fdsink_audio), "fd", fd_audio, NULL);
gst_element_connect (audio_enc, "src", fdsink_audio, "sink");
gst_bin_add (GST_BIN (muxthread_audio), fdsink_audio);
gst_bin_add (GST_BIN (bin), muxthread_audio);
}
//gtk_object_set (GTK_OBJECT (muxer), "video", "00", NULL);
//gtk_object_set (GTK_OBJECT (muxer), "audio", "00", NULL);
/* start playing */
gst_element_set_state(GST_ELEMENT(bin), GST_STATE_PLAYING);
gtk_idle_add(idle_func, bin);
gst_main();
/* stop the bin */
gst_element_set_state(GST_ELEMENT(bin), GST_STATE_NULL);
gst_pipeline_destroy(bin);
exit(0);
}