gstreamer/tests/icles/metadata_editor.c
2012-08-04 16:31:30 +01:00

1370 lines
34 KiB
C

/*
* GStreamer
* Copyright 2007 Edgard Lima <edgard.lima@indt.org.br>
*
* Permission is hereby granted, free of charge, to any person obtaining a
* copy of this software and associated documentation files (the "Software"),
* to deal in the Software without restriction, including without limitation
* the rights to use, copy, modify, merge, publish, distribute, sublicense,
* and/or sell copies of the Software, and to permit persons to whom the
* Software is furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in
* all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
* FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
* DEALINGS IN THE SOFTWARE.
*
* Alternatively, the contents of this file may be used under the
* GNU Lesser General Public License Version 2.1 (the "LGPL"), in
* which case the following provisions apply instead of the ones
* mentioned above:
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Library General Public
* License as published by the Free Software Foundation; either
* version 2 of the License, or (at your option) any later version.
*
* This library 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
* Library General Public License for more details.
*
* You should have received a copy of the GNU Library General Public
* License along with this library; if not, write to the
* Free Software Foundation, Inc., 59 Temple Place - Suite 330,
* Boston, MA 02111-1307, USA.
*/
#ifdef HAVE_CONFIG_H
#include <config.h>
#endif
#include "metadata_editor.h"
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <gst/gst.h>
#if !GTK_CHECK_VERSION (2, 17, 7)
static void
gtk_widget_get_allocation (GtkWidget * w, GtkAllocation * a)
{
*a = w->allocation;
}
#endif
/*
* Global constants
*/
enum
{
COL_TAG = 0,
COL_VALUE,
NUM_COLS
};
/* *INDENT-OFF* */
typedef enum _AppOptions {
APP_OPT_DEMUX_EXIF = (1 << 0),
APP_OPT_DEMUX_IPTC = (1 << 1),
APP_OPT_DEMUX_XMP = (1 << 2),
APP_OPT_MUX_EXIF = (1 << 3),
APP_OPT_MUX_IPTC = (1 << 4),
APP_OPT_MUX_XMP = (1 << 5),
APP_OPT_ALL = (1 << 6) - 1,
} AppOptions;
#define ENC_ERROR (-1)
#define ENC_DONE (0)
#define ENC_UNKNOWN (1)
/* *INDENT-OFF* */
/*
* functions prototypes
*/
/* gstreamer related functions */
static void me_gst_cleanup_elements ();
static int me_gst_setup_view_pipeline (const gchar * filename);
static int
me_gst_setup_capture_pipeline (const gchar * src_file, const gchar * dest_file,
gint * encode_status, gboolean use_v4l2);
static int
me_gst_setup_encode_pipeline (const gchar * src_file, const gchar * dest_file,
gint * encode_status);
/* ui related functions */
static void ui_refresh (void);
static void process_file(void);
/*
* Global Vars
*/
GstElement *gst_source = NULL;
GstElement *gst_metadata_demux = NULL;
GstElement *gst_metadata_mux = NULL;
GstElement *gst_image_dec = NULL;
GstElement *gst_image_enc = NULL;
GstElement *gst_video_scale = NULL;
GstElement *gst_video_convert = NULL;
GstElement *gst_video_sink = NULL;
GstElement *gst_file_sink = NULL;
GstElement *gst_pipeline = NULL;
GdkPixbuf *last_pixbuf = NULL; /* image as pixbuf at original size */
GdkPixbuf *draw_pixbuf = NULL; /* pixbuf resized for drawing */
AppOptions app_options = APP_OPT_ALL;
GstTagList *tag_list = NULL;
GtkBuilder *builder = NULL;
GtkWidget *ui_main_window = NULL;
GtkWidget *ui_drawing = NULL;
GtkWidget *ui_tree = NULL;
GtkEntry *ui_entry_insert_tag = NULL;
GtkEntry *ui_entry_insert_value = NULL;
GtkToggleButton *ui_chk_bnt_capture_v4l2 = NULL;
GtkToggleButton *ui_chk_bnt_capture_test = NULL;
GString *filename = NULL;
/*
* Helper functions
*/
static void
dump_tag_buffer(const char *tag, guint8 * buf, guint32 size)
{
guint32 i;
printf("\nDumping %s (size = %u)\n\n", tag, size);
for(i=0; i<size; ++i) {
if (i % 16 == 0)
printf("%04x:%04x | ", i >> 16, i & 0xFFFF);
printf("%02x", buf[i]);
if (i % 16 != 15)
printf(" ");
else
printf("\n");
}
printf("\n\n");
}
static void
insert_tag_on_tree (const GstTagList * list, const gchar * tag,
gpointer user_data)
{
gchar *str = NULL;
GtkTreeView *tree_view = NULL;
GtkTreeStore *tree_store = NULL;
GtkTreeIter iter;
tree_view = GTK_TREE_VIEW (user_data);
if (gst_tag_get_type (tag) == G_TYPE_STRING) {
if (!gst_tag_list_get_string_index (list, tag, 0, &str))
g_assert_not_reached ();
} else if ( gst_tag_get_type (tag) == GST_TYPE_BUFFER ) {
const GValue *val = NULL;
GstBuffer *buf;
val = gst_tag_list_get_value_index (list, tag, 0);
buf = gst_value_get_buffer (val);
dump_tag_buffer(tag, GST_BUFFER_DATA(buf), GST_BUFFER_SIZE(buf));
str = g_strdup("It has been printed to stdout");
} else {
str = g_strdup_value_contents (gst_tag_list_get_value_index (list, tag, 0));
}
tree_store = GTK_TREE_STORE (gtk_tree_view_get_model (tree_view));
gtk_tree_store_append (tree_store, &iter, NULL);
gtk_tree_store_set (tree_store, &iter, COL_TAG, tag, COL_VALUE, str, -1);
if (str)
g_free (str);
}
static gboolean
change_tag_list (GstTagList ** list, const gchar * tag, const gchar * value)
{
GType type;
gboolean ret = FALSE;
if (list == NULL || tag == NULL || value == NULL)
goto done;
if (!gst_tag_exists (tag)) {
fprintf (stderr, "%s is not a GStreamer registered tag\n", tag);
goto done;
}
if (*list == NULL)
*list = gst_tag_list_new ();
type = gst_tag_get_type (tag);
if (type == GST_TYPE_FRACTION) {
/* FIXME: Ask GStreamer guys to add GST_FRACTION support to TAGS */
/* Even better: ask GLib guys to add this type */
gint n, d;
sscanf (value, "%d/%d", &n, &d);
gst_tag_list_add (*list, GST_TAG_MERGE_REPLACE, tag, n, d, NULL);
ret = TRUE;
} else {
switch (type) {
case G_TYPE_STRING:
gst_tag_list_add (*list, GST_TAG_MERGE_REPLACE, tag, value, NULL);
ret = TRUE;
break;
case G_TYPE_FLOAT:
{
gfloat fv = (gfloat) g_strtod (value, NULL);
gst_tag_list_add (*list, GST_TAG_MERGE_REPLACE, tag, fv, NULL);
ret = TRUE;
}
break;
case G_TYPE_INT:
/* fall through */
case G_TYPE_UINT:
{
gint iv = atoi (value);
gst_tag_list_add (*list, GST_TAG_MERGE_REPLACE, tag, iv, NULL);
ret = TRUE;
}
break;
default:
g_printerr ("Tags of type '%s' are not supported yet for editing"
" by this application.\n",
g_type_name (type));
break;
}
}
done:
return ret;
}
static void
update_draw_pixbuf (guint max_width, guint max_height)
{
gdouble wratio, hratio;
gint w = 0, h = 0;
if (last_pixbuf) {
w = gdk_pixbuf_get_width (last_pixbuf);
h = gdk_pixbuf_get_height (last_pixbuf);
}
g_print ("Allocation: %dx%d, pixbuf: %dx%d\n", max_width, max_height, w, h);
if (last_pixbuf == NULL)
return;
g_return_if_fail (max_width > 0 && max_height > 0);
wratio = w * 1.0 / max_width * 1.0;
hratio = h * 1.0 / max_height;
g_print ("ratios = %.2f / %.2f\n", wratio, hratio);
if (hratio > wratio) {
w = (gint) (w * 1.0 / hratio);
h = (gint) (h * 1.0 / hratio);
} else {
w = (gint) (w * 1.0 / wratio);
h = (gint) (h * 1.0 / wratio);
}
if (draw_pixbuf != NULL && gdk_pixbuf_get_width (draw_pixbuf) == w &&
gdk_pixbuf_get_height (last_pixbuf) == h) {
return; /* nothing to do */
}
g_print ("drawing pixbuf at %dx%d\n", w, h);
if (draw_pixbuf)
g_object_unref (draw_pixbuf);
draw_pixbuf =
gdk_pixbuf_scale_simple (last_pixbuf, w, h, GDK_INTERP_BILINEAR);
}
static void
ui_drawing_size_allocate_cb (GtkWidget * drawing_area,
GtkAllocation * allocation, gpointer data)
{
update_draw_pixbuf (allocation->width, allocation->height);
}
/*
* UI handling functions (mapped by GtkBuilder)
*/
gboolean
on_drawingMain_expose_event (GtkWidget * widget, GdkEventExpose * event,
gpointer data)
{
GtkAllocation a;
cairo_t *cr;
GdkRectangle rect;
gtk_widget_get_allocation (widget, &a);
if (draw_pixbuf == NULL)
return FALSE;
rect.width = gdk_pixbuf_get_width (draw_pixbuf);
rect.height = gdk_pixbuf_get_height (draw_pixbuf);
/* center image */
rect.x = (a.width - rect.width) / 2;
rect.y = (a.height - rect.height) / 2;
/* sanity check, shouldn't happen */
if (rect.x < 0)
rect.x = 0;
if (rect.y < 0)
rect.y = 0;
cr = gdk_cairo_create (event->window);
gdk_cairo_set_source_pixbuf (cr, draw_pixbuf, 0, 0);
gdk_cairo_rectangle (cr, &rect);
cairo_fill (cr);
cairo_destroy (cr);
return TRUE; /* handled expose event */
}
void
on_windowMain_delete_event (GtkWidget * widget, GdkEvent * event,
gpointer user_data)
{
gst_element_set_state (gst_pipeline, GST_STATE_NULL);
gst_element_get_state (gst_pipeline, NULL, NULL, GST_CLOCK_TIME_NONE);
gtk_main_quit ();
}
void
on_buttonInsert_clicked (GtkButton * button, gpointer user_data)
{
GtkTreeStore *store = NULL;
GtkTreeIter iter;
const gchar *tag = gtk_entry_get_text (ui_entry_insert_tag);
const gchar *value = gtk_entry_get_text (ui_entry_insert_value);
if ( tag_list == NULL ) {
tag_list = gst_tag_list_new ();
}
if (tag && value && tag[0] != '\0') {
/* insert just new tags (the ones already in list should be modified) */
if (gst_tag_list_get_tag_size (tag_list, tag)) {
fprintf (stderr, "%s tag is already in the list try to modify it\n", tag);
} else {
if (change_tag_list (&tag_list, tag, value)) {
/* just add to ui_tree if it has been added to tag_list */
store =
GTK_TREE_STORE (gtk_tree_view_get_model (GTK_TREE_VIEW (ui_tree)));
gtk_tree_store_append (store, &iter, NULL);
gtk_tree_store_set (store, &iter, COL_TAG, tag, COL_VALUE, value, -1);
}
}
}
return;
}
static void
setup_new_filename (GString * str, const gchar * ext)
{
int i = 0;
for (i = str->len - 1; i > 0; --i) {
if (str->str[i] == '/') {
++i;
break;
}
}
g_string_insert (str, i, "_new_");
if (ext) {
int len = strlen (ext);
if (len > str->len)
g_string_append (str, ext);
else if (strcasecmp (ext, &str->str[str->len - len]))
g_string_append (str, ext);
}
}
void
on_buttonSaveFile_clicked (GtkButton * button, gpointer user_data)
{
GString *src_file = NULL;
gint enc_status = ENC_UNKNOWN;
const gboolean use_v4l2 =
gtk_toggle_button_get_active (ui_chk_bnt_capture_v4l2);
const gboolean use_test =
gtk_toggle_button_get_active (ui_chk_bnt_capture_test);
gst_element_set_state (gst_pipeline, GST_STATE_NULL);
gst_element_get_state (gst_pipeline, NULL, NULL, GST_CLOCK_TIME_NONE);
src_file = g_string_new (filename->str);
if (use_v4l2 || use_test) {
setup_new_filename (filename, ".jpg");
if (me_gst_setup_capture_pipeline (src_file->str, filename->str,
&enc_status, use_v4l2)) {
goto done;
}
} else {
setup_new_filename (filename, NULL);
if (me_gst_setup_encode_pipeline (src_file->str, filename->str,
&enc_status)) {
goto done;
}
}
ui_refresh ();
remove (filename->str);
if (tag_list && gst_metadata_mux) {
GstTagSetter *setter = GST_TAG_SETTER (gst_metadata_mux);
if (setter) {
gst_element_set_state (gst_pipeline, GST_STATE_READY);
gst_element_get_state (gst_pipeline, NULL, NULL, GST_CLOCK_TIME_NONE);
gst_tag_setter_merge_tags (setter, tag_list, GST_TAG_MERGE_REPLACE);
}
}
gst_element_set_state (gst_pipeline, GST_STATE_PLAYING);
gst_element_get_state (gst_pipeline, NULL, NULL, GST_CLOCK_TIME_NONE);
/* wait until finished */
gtk_main ();
gst_element_set_state (gst_pipeline, GST_STATE_NULL);
gst_element_get_state (gst_pipeline, NULL, NULL, GST_CLOCK_TIME_NONE);
if (enc_status == ENC_DONE) {
/* view new file */
if (tag_list) {
gst_tag_list_unref (tag_list);
tag_list = NULL;
}
me_gst_setup_view_pipeline (filename->str);
gst_element_set_state (gst_pipeline, GST_STATE_PLAYING);
gst_element_get_state (gst_pipeline, NULL, NULL, GST_CLOCK_TIME_NONE);
}
done:
if (src_file)
g_string_free (src_file, TRUE);
}
void
on_checkbuttonCaptureV4l2_toggled (GtkToggleButton * togglebutton,
gpointer user_data)
{
if (gtk_toggle_button_get_active (togglebutton))
gtk_toggle_button_set_active(ui_chk_bnt_capture_test, FALSE);
}
void
on_checkbuttonCaptureTest_toggled (GtkToggleButton * togglebutton,
gpointer user_data)
{
if (gtk_toggle_button_get_active (togglebutton))
gtk_toggle_button_set_active(ui_chk_bnt_capture_v4l2, FALSE);
}
void
on_checkbuttonOptionsDemuxExif_toggled (GtkToggleButton * togglebutton,
gpointer user_data)
{
if (gtk_toggle_button_get_active (togglebutton))
app_options |= APP_OPT_DEMUX_EXIF;
else
app_options &= ~APP_OPT_DEMUX_EXIF;
}
void
on_checkbuttonOptionsDemuxIptc_toggled (GtkToggleButton * togglebutton,
gpointer user_data)
{
if (gtk_toggle_button_get_active (togglebutton))
app_options |= APP_OPT_DEMUX_IPTC;
else
app_options &= ~APP_OPT_DEMUX_IPTC;
}
void
on_checkbuttonOptionsDemuxXmp_toggled (GtkToggleButton * togglebutton,
gpointer user_data)
{
if (gtk_toggle_button_get_active (togglebutton))
app_options |= APP_OPT_DEMUX_XMP;
else
app_options &= ~APP_OPT_DEMUX_XMP;
}
void
on_checkbuttonOptionsMuxExif_toggled (GtkToggleButton * togglebutton,
gpointer user_data)
{
if (gtk_toggle_button_get_active (togglebutton))
app_options |= APP_OPT_MUX_EXIF;
else
app_options &= ~APP_OPT_MUX_EXIF;
}
void
on_checkbuttonOptionsMuxIptc_toggled (GtkToggleButton * togglebutton,
gpointer user_data)
{
if (gtk_toggle_button_get_active (togglebutton))
app_options |= APP_OPT_MUX_IPTC;
else
app_options &= ~APP_OPT_MUX_IPTC;
}
void
on_checkbuttonOptionsMuxXmp_toggled (GtkToggleButton * togglebutton,
gpointer user_data)
{
if (gtk_toggle_button_get_active (togglebutton))
app_options |= APP_OPT_MUX_XMP;
else
app_options &= ~APP_OPT_MUX_XMP;
}
void
on_buttonOpenFile_clicked (GtkButton * button, gpointer user_data)
{
GtkWidget *dialog;
gboolean open = FALSE;
dialog = gtk_file_chooser_dialog_new ("Open File",
GTK_WINDOW(ui_main_window),
GTK_FILE_CHOOSER_ACTION_OPEN,
GTK_STOCK_CANCEL, GTK_RESPONSE_CANCEL,
GTK_STOCK_OPEN, GTK_RESPONSE_ACCEPT,
NULL);
if (filename) {
const char *p = filename->str;
char *q = filename->str + filename->len - 1;
for (;p != q; --q) {
if ( *q == '/' )
break;
}
if ( p != q )
*q = '\0';
gtk_file_chooser_set_current_folder(GTK_FILE_CHOOSER (dialog),
filename->str);
if ( p != q )
*q = '/';
}
open = gtk_dialog_run (GTK_DIALOG (dialog)) == GTK_RESPONSE_ACCEPT;
if (open) {
char *str;
str = gtk_file_chooser_get_filename (GTK_FILE_CHOOSER (dialog));
if (filename)
g_string_free (filename, TRUE);
filename = g_string_new(str);
g_free (str);
}
gtk_widget_destroy (dialog);
if (open) {
process_file();
}
}
/*
* UI handling functions
*/
void
on_cell_edited (GtkCellRendererText * renderer, gchar * str_path,
gchar * new_text, gpointer user_data)
{
GtkTreePath *path = NULL;
GtkTreeIter iter;
GtkTreeModel *model = NULL;
const gint col_index = GPOINTER_TO_INT (user_data);
const gchar *tag = gtk_entry_get_text (ui_entry_insert_tag);
path = gtk_tree_path_new_from_string (str_path);
model = gtk_tree_view_get_model (GTK_TREE_VIEW (ui_tree));
if (change_tag_list (&tag_list, tag, new_text)) {
if (gtk_tree_model_get_iter (model, &iter, path)) {
gtk_tree_store_set (GTK_TREE_STORE (model), &iter, col_index, new_text,
-1);
gtk_entry_set_text (ui_entry_insert_value, new_text);
}
}
if (path)
gtk_tree_path_free (path);
}
static void
on_tree_selection_changed (GtkTreeSelection * selection, gpointer data)
{
GtkTreeIter iter;
GtkTreeModel *model;
if (gtk_tree_selection_get_selected (selection, &model, &iter)) {
gchar *tag;
gchar *value;
gtk_tree_model_get (model, &iter, COL_TAG, &tag, -1);
gtk_tree_model_get (model, &iter, COL_VALUE, &value, -1);
gtk_entry_set_text (ui_entry_insert_tag, tag);
gtk_entry_set_text (ui_entry_insert_value, value);
g_free (value);
g_free (tag);
}
}
/*
* UI helper functions
*/
static int
ui_add_columns (GtkTreeView * tree_view, const gchar * title, gint col_index,
gboolean editable)
{
GtkCellRenderer *renderer;
GtkTreeViewColumn *tree_col;
int ret = 0;
renderer = gtk_cell_renderer_text_new ();
if (editable) {
g_object_set (renderer, "editable", TRUE, NULL);
g_signal_connect (G_OBJECT (renderer), "edited",
G_CALLBACK (on_cell_edited), GINT_TO_POINTER (col_index));
}
if ((tree_col = gtk_tree_view_column_new_with_attributes (title, renderer,
"text", col_index, NULL))) {
gtk_tree_view_append_column (tree_view, tree_col);
} else {
fprintf (stderr, "UI: could not create column %s\n", title);
ret = -201;
goto done;
}
done:
return ret;
}
static int
ui_setup_tree_view (GtkTreeView * tree_view)
{
int ret = 0;
GtkTreeStore *tree_store = NULL;
GtkTreeSelection *select;
if ((ret = ui_add_columns (tree_view, "tag", COL_TAG, FALSE)))
goto done;
if ((ret = ui_add_columns (tree_view, "value", COL_VALUE, TRUE)))
goto done;
tree_store = gtk_tree_store_new (NUM_COLS, G_TYPE_STRING, G_TYPE_STRING);
gtk_tree_view_set_model (tree_view, GTK_TREE_MODEL (tree_store));
select = gtk_tree_view_get_selection (tree_view);
gtk_tree_selection_set_mode (select, GTK_SELECTION_SINGLE);
g_signal_connect (G_OBJECT (select), "changed",
G_CALLBACK (on_tree_selection_changed), NULL);
done:
if (tree_store)
g_object_unref (tree_store);
return ret;
}
static void
ui_refresh (void)
{
GtkTreeStore *store =
GTK_TREE_STORE (gtk_tree_view_get_model (GTK_TREE_VIEW (ui_tree)));
gtk_tree_store_clear (store);
if (filename)
gtk_window_set_title (GTK_WINDOW (ui_main_window), filename->str);
}
static int
ui_create (void)
{
GError *error = NULL;
int ret = 0;
builder = gtk_builder_new ();
if (!gtk_builder_add_from_file (builder, "metadata_editor.ui", &error))
{
g_warning ("Couldn't load builder file: %s", error->message);
g_error_free (error);
ret = -101;
goto done;
}
ui_main_window = GTK_WIDGET (gtk_builder_get_object (builder, "windowMain"));
ui_drawing = GTK_WIDGET (gtk_builder_get_object (builder, "drawingMain"));
ui_tree = GTK_WIDGET (gtk_builder_get_object (builder, "treeMain"));
ui_entry_insert_tag =
GTK_ENTRY (gtk_builder_get_object (builder, "entryTag"));
ui_entry_insert_value =
GTK_ENTRY (gtk_builder_get_object (builder, "entryValue"));
ui_chk_bnt_capture_v4l2 =
GTK_TOGGLE_BUTTON (gtk_builder_get_object (builder,
"checkbuttonCaptureV4l2"));
ui_chk_bnt_capture_test =
GTK_TOGGLE_BUTTON (gtk_builder_get_object (builder,
"checkbuttonCaptureTest"));
if (!(ui_main_window && ui_drawing && ui_tree
&& ui_entry_insert_tag && ui_entry_insert_value
&& ui_chk_bnt_capture_v4l2 && ui_chk_bnt_capture_test)) {
fprintf (stderr, "Some widgets couldn't be created\n");
ret = -105;
goto done;
}
g_signal_connect_after (ui_drawing, "size-allocate",
G_CALLBACK (ui_drawing_size_allocate_cb), NULL);
gtk_builder_connect_signals (builder, NULL);
ui_setup_tree_view (GTK_TREE_VIEW (ui_tree));
ui_refresh ();
gtk_widget_show_all (ui_main_window);
done:
return ret;
}
/*
* GStreamer functions
*/
static gboolean
me_gst_bus_callback_encode (GstBus * bus, GstMessage * message, gpointer data)
{
gint *encode_status = (gint *) data;
fflush (stdout);
switch (GST_MESSAGE_TYPE (message)) {
case GST_MESSAGE_ERROR:
{
GError *err;
gchar *debug;
gst_message_parse_error (message, &err, &debug);
fprintf (stderr, "Error: %s\n", err->message);
g_error_free (err);
g_free (debug);
*encode_status = ENC_ERROR;
gtk_main_quit ();
}
break;
case GST_MESSAGE_TAG:
{
/* ignore, we alredy have the tag list */
}
break;
case GST_MESSAGE_EOS:
{
*encode_status = ENC_DONE;
gtk_main_quit ();
}
break;
default:
/* unhandled message */
break;
}
/* we want to be notified again the next time there is a message
* on the bus, so returning TRUE (FALSE means we want to stop watching
* for messages on the bus and our callback should not be called again)
*/
return TRUE;
}
static gboolean
me_gst_bus_callback_view (GstBus * bus, GstMessage * message, gpointer data)
{
switch (GST_MESSAGE_TYPE (message)) {
case GST_MESSAGE_ERROR:
{
GError *err;
gchar *debug;
gst_message_parse_error (message, &err, &debug);
fprintf (stderr, "Error: %s\n", err->message);
g_error_free (err);
g_free (debug);
gtk_main_quit ();
}
break;
case GST_MESSAGE_TAG:
{
if (tag_list == NULL)
gst_message_parse_tag (message, &tag_list);
else {
GstTagList *tl = NULL;
GstTagList *ntl = NULL;
gst_message_parse_tag (message, &tl);
if (tl) {
ntl = gst_tag_list_merge (tag_list, tl, GST_TAG_MERGE_PREPEND);
if (ntl) {
gst_tag_list_unref (tag_list);
tag_list = ntl;
}
gst_tag_list_unref (tl);
}
}
/* remove whole chunk tags */
gst_tag_list_remove_tag (tag_list, "exif");
gst_tag_list_remove_tag (tag_list, "iptc");
gst_tag_list_remove_tag (tag_list, "xmp");
}
break;
case GST_MESSAGE_EOS:
if (tag_list) {
gst_tag_list_foreach (tag_list, insert_tag_on_tree, ui_tree);
}
break;
case GST_MESSAGE_ELEMENT: {
const GValue *val;
GtkAllocation a;
/* only interested in element messages from our gdkpixbufsink */
if (message->src != GST_OBJECT_CAST (gst_video_sink))
break;
/* only interested in the first image (not any smaller previews) */
if (gst_structure_has_name (message->structure, "pixbuf"))
break;
if (!gst_structure_has_name (message->structure, "preroll-pixbuf"))
break;
val = gst_structure_get_value (message->structure, "pixbuf");
g_return_val_if_fail (val != NULL, TRUE);
if (last_pixbuf)
g_object_unref (last_pixbuf);
last_pixbuf = GDK_PIXBUF (g_value_dup_object (val));
g_print ("Got image pixbuf: %dx%d\n", gdk_pixbuf_get_width (last_pixbuf),
gdk_pixbuf_get_height (last_pixbuf));
gtk_widget_get_allocation (GTK_WIDGET (ui_drawing), &a);
update_draw_pixbuf (a.width, a.height);
gtk_widget_queue_draw (ui_drawing);
break;
}
default:
/* unhandled message */
break;
}
/* we want to be notified again the next time there is a message
* on the bus, so returning TRUE (FALSE means we want to stop watching
* for messages on the bus and our callback should not be called again)
*/
return TRUE;
}
static void
me_gst_cleanup_elements (void)
{
/* when adding an element to pipeline rember to set it to NULL or add extra ref */
if (gst_source) {
gst_object_unref (gst_source);
gst_source = NULL;
}
if (gst_metadata_demux) {
gst_object_unref (gst_metadata_demux);
gst_metadata_demux = NULL;
}
if (gst_metadata_mux) {
gst_object_unref (gst_metadata_mux);
gst_metadata_mux = NULL;
}
if (gst_image_dec) {
gst_object_unref (gst_image_dec);
gst_image_dec = NULL;
}
if (gst_image_enc) {
gst_object_unref (gst_image_enc);
gst_image_enc = NULL;
}
if (gst_video_scale) {
gst_object_unref (gst_video_scale);
gst_video_scale = NULL;
}
if (gst_video_convert) {
gst_object_unref (gst_video_convert);
gst_video_convert = NULL;
}
if (gst_video_sink) {
gst_object_unref (gst_video_sink);
gst_video_sink = NULL;
}
if (gst_file_sink) {
gst_object_unref (gst_file_sink);
gst_file_sink = NULL;
}
if (gst_pipeline) {
gst_element_set_state (gst_pipeline, GST_STATE_NULL);
gst_element_get_state (gst_pipeline, NULL, NULL, GST_CLOCK_TIME_NONE);
gst_object_unref (gst_pipeline);
gst_pipeline = NULL;
}
}
/* dummy function that looks the file extension */
static gboolean
is_png (const gchar * filename)
{
gboolean ret = FALSE;
guint32 len;
if (!filename)
goto done;
if ((len = strlen (filename)) < 4) /* at least ".png" */
goto done;
if (0 == strcasecmp (filename + (len - 4), ".png"))
ret = TRUE;
done:
return ret;
}
static int
me_gst_setup_capture_pipeline (const gchar * src_file, const gchar * dest_file,
gint * encode_status, gboolean use_v4l2)
{
int ret = 0;
GstBus *bus = NULL;
gboolean linked;
*encode_status = ENC_ERROR;
me_gst_cleanup_elements ();
/* create elements */
if ( use_v4l2 )
gst_source = gst_element_factory_make ("v4l2src", NULL);
else
gst_source = gst_element_factory_make ("videotestsrc", NULL);
gst_video_convert = gst_element_factory_make ("ffmpegcolorspace", NULL);
gst_image_enc = gst_element_factory_make ("jpegenc", NULL);
gst_metadata_mux = gst_element_factory_make ("metadatamux", NULL);
gst_file_sink = gst_element_factory_make ("filesink", NULL);
if (!(gst_source && gst_video_convert && gst_image_enc && gst_metadata_mux
&& gst_file_sink)) {
fprintf (stderr, "An element couldn't be created for ecoding\n");
ret = -300;
goto done;
}
/* create gst_pipeline */
gst_pipeline = gst_pipeline_new (NULL);
if (NULL == gst_pipeline) {
fprintf (stderr, "Pipeline couldn't be created\n");
ret = -305;
goto done;
}
/* set elements's properties */
g_object_set (gst_source, "num-buffers", 1, NULL);
g_object_set (gst_file_sink, "location", dest_file, NULL);
if ( app_options & APP_OPT_MUX_EXIF )
g_object_set (gst_metadata_mux, "exif", TRUE, NULL);
else
g_object_set (gst_metadata_mux, "exif", FALSE, NULL);
if ( app_options & APP_OPT_MUX_IPTC )
g_object_set (gst_metadata_mux, "iptc", TRUE, NULL);
else
g_object_set (gst_metadata_mux, "iptc", FALSE, NULL);
if ( app_options & APP_OPT_MUX_XMP )
g_object_set (gst_metadata_mux, "xmp", TRUE, NULL);
else
g_object_set (gst_metadata_mux, "xmp", FALSE, NULL);
/* adding and linking elements */
gst_bin_add_many (GST_BIN (gst_pipeline), gst_source, gst_video_convert,
gst_image_enc, gst_metadata_mux, gst_file_sink, NULL);
linked =
gst_element_link_many (gst_source, gst_video_convert, gst_image_enc,
gst_metadata_mux, gst_file_sink, NULL);
/* now element are owned by pipeline (for videosink we keep a extra ref) */
gst_source = gst_video_convert = gst_image_enc = gst_file_sink = NULL;
gst_object_ref (gst_metadata_mux);
if (!linked) {
fprintf (stderr, "Elements couldn't be linked\n");
ret = -310;
goto done;
}
*encode_status = ENC_UNKNOWN;
/* adding message bus */
bus = gst_pipeline_get_bus (GST_PIPELINE (gst_pipeline));
gst_bus_add_watch (bus, me_gst_bus_callback_encode, encode_status);
gst_object_unref (bus);
done:
return ret;
}
static int
me_gst_setup_encode_pipeline (const gchar * src_file, const gchar * dest_file,
gint * encode_status)
{
int ret = 0;
GstBus *bus = NULL;
gboolean linked;
*encode_status = ENC_ERROR;
me_gst_cleanup_elements ();
/* create elements */
gst_source = gst_element_factory_make ("filesrc", NULL);
gst_metadata_demux = gst_element_factory_make ("metadatademux", NULL);
gst_metadata_mux = gst_element_factory_make ("metadatamux", NULL);
gst_file_sink = gst_element_factory_make ("filesink", NULL);
if (!(gst_source && gst_metadata_demux && gst_metadata_mux && gst_file_sink)) {
fprintf (stderr, "An element couldn't be created for ecoding\n");
ret = -300;
goto done;
}
/* create gst_pipeline */
gst_pipeline = gst_pipeline_new (NULL);
if (NULL == gst_pipeline) {
fprintf (stderr, "Pipeline couldn't be created\n");
ret = -305;
goto done;
}
/* set elements's properties */
g_object_set (gst_source, "location", src_file, NULL);
g_object_set (gst_file_sink, "location", dest_file, NULL);
if ( app_options & APP_OPT_DEMUX_EXIF )
g_object_set (gst_metadata_demux, "exif", TRUE, NULL);
else
g_object_set (gst_metadata_demux, "exif", FALSE, NULL);
if ( app_options & APP_OPT_DEMUX_IPTC )
g_object_set (gst_metadata_demux, "iptc", TRUE, NULL);
else
g_object_set (gst_metadata_demux, "iptc", FALSE, NULL);
if ( app_options & APP_OPT_DEMUX_XMP )
g_object_set (gst_metadata_demux, "xmp", TRUE, NULL);
else
g_object_set (gst_metadata_demux, "xmp", FALSE, NULL);
if ( app_options & APP_OPT_MUX_EXIF )
g_object_set (gst_metadata_mux, "exif", TRUE, NULL);
else
g_object_set (gst_metadata_mux, "exif", FALSE, NULL);
if ( app_options & APP_OPT_MUX_IPTC )
g_object_set (gst_metadata_mux, "iptc", TRUE, NULL);
else
g_object_set (gst_metadata_mux, "iptc", FALSE, NULL);
if ( app_options & APP_OPT_MUX_XMP )
g_object_set (gst_metadata_mux, "xmp", TRUE, NULL);
else
g_object_set (gst_metadata_mux, "xmp", FALSE, NULL);
/* adding and linking elements */
gst_bin_add_many (GST_BIN (gst_pipeline), gst_source, gst_metadata_demux,
gst_metadata_mux, gst_file_sink, NULL);
linked =
gst_element_link_many (gst_source, gst_metadata_demux, gst_metadata_mux,
gst_file_sink, NULL);
/* now element are owned by pipeline (for videosink we keep a extra ref) */
gst_source = gst_metadata_demux = gst_file_sink = NULL;
gst_object_ref (gst_metadata_mux);
if (!linked) {
fprintf (stderr, "Elements couldn't be linked\n");
ret = -310;
goto done;
}
*encode_status = ENC_UNKNOWN;
/* adding message bus */
bus = gst_pipeline_get_bus (GST_PIPELINE (gst_pipeline));
gst_bus_add_watch (bus, me_gst_bus_callback_encode, encode_status);
gst_object_unref (bus);
done:
return ret;
}
static int
me_gst_setup_view_pipeline (const gchar * filename)
{
int ret = 0;
GstBus *bus = NULL;
gboolean linked;
me_gst_cleanup_elements ();
/* create elements */
gst_source = gst_element_factory_make ("filesrc", NULL);
gst_metadata_demux = gst_element_factory_make ("metadatademux", NULL);
/* let's do a dummy stuff to avoid decodebin */
if (is_png (filename))
gst_image_dec = gst_element_factory_make ("pngdec", NULL);
else
gst_image_dec = gst_element_factory_make ("jpegdec", NULL);
gst_video_scale = gst_element_factory_make ("videoscale", NULL);
gst_video_convert = gst_element_factory_make ("ffmpegcolorspace", NULL);
gst_video_sink = gst_element_factory_make ("gdkpixbufsink", NULL);
if (gst_video_sink == NULL) {
if (!gst_default_registry_check_feature_version ("gdkpixbufdec", 0, 10, 0))
g_warning ("Could not create 'gdkpixbufsink' element");
else {
g_warning ("Could not create 'gdkpixbufsink' element. "
"(May be your gst-plugins-good is too old?)");
ret = -400;
}
goto done;
}
if (!(gst_source && gst_metadata_demux && gst_image_dec && gst_video_scale
&& gst_video_convert && gst_video_sink)) {
fprintf (stderr, "An element couldn't be created for viewing\n");
ret = -400;
goto done;
}
/* create gst_pipeline */
gst_pipeline = gst_pipeline_new (NULL);
if (NULL == gst_pipeline) {
fprintf (stderr, "Pipeline couldn't be created\n");
ret = -405;
goto done;
}
/* set elements's properties */
g_object_set (gst_source, "location", filename, NULL);
g_object_set (gst_metadata_demux, "parse-only", TRUE, NULL);
/* adding and linking elements */
gst_bin_add_many (GST_BIN (gst_pipeline), gst_source, gst_metadata_demux,
gst_image_dec, gst_video_scale, gst_video_convert, gst_video_sink, NULL);
linked = gst_element_link_many (gst_source, gst_metadata_demux, gst_image_dec,
gst_video_scale, gst_video_convert, gst_video_sink, NULL);
/* now element are owned by pipeline (for videosink we keep a extra ref) */
gst_source = gst_metadata_demux = gst_image_dec = gst_video_scale =
gst_video_convert = NULL;
gst_object_ref (gst_video_sink);
if (last_pixbuf) {
g_object_unref (last_pixbuf);
last_pixbuf = NULL;
}
if (!linked) {
fprintf (stderr, "Elements couldn't be linked\n");
ret = -410;
goto done;
}
/* adding message bus */
bus = gst_pipeline_get_bus (GST_PIPELINE (gst_pipeline));
gst_bus_add_watch (bus, me_gst_bus_callback_view, NULL);
gst_object_unref (bus);
done:
return ret;
}
static void
process_file(void)
{
/* filename for future usage (title and file name to be created) */
me_gst_cleanup_elements ();
if (tag_list) {
gst_tag_list_unref (tag_list);
tag_list = NULL;
}
/* create pipeline */
me_gst_setup_view_pipeline (filename->str);
gst_element_set_state (gst_pipeline, GST_STATE_PLAYING);
ui_refresh ();
}
int
main (int argc, char *argv[])
{
int ret = 0;
if (argc >= 2) {
if (filename)
g_string_free (filename, TRUE);
filename = g_string_new (argv[1]);
}
gst_init (&argc, &argv);
gtk_init (&argc, &argv);
/* create UI */
if ((ret = ui_create ())) {
goto done;
}
if (argc >= 2) {
process_file();
}
gtk_main ();
done:
me_gst_cleanup_elements ();
if (tag_list) {
gst_tag_list_unref (tag_list);
tag_list = NULL;
}
if (filename) {
g_string_free (filename, TRUE);
filename = NULL;
}
return ret;
}