/* * GStreamer * Copyright 2007 Edgard Lima * * 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 #endif #include "metadata_editor.h" #include #include #include #include #include #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> 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; gint w, h, x, y; gtk_widget_get_allocation (widget, &a); if (draw_pixbuf == NULL) return FALSE; w = gdk_pixbuf_get_width (draw_pixbuf); h = gdk_pixbuf_get_height (draw_pixbuf); /* center image */ x = (a.width - w) / 2; y = (a.height - h) / 2; /* sanity check, shouldn't happen */ if (x < 0) x = 0; if (y < 0) y = 0; gdk_draw_pixbuf (GDK_DRAWABLE (gtk_widget_get_window (widget)), gtk_widget_get_style (widget)->black_gc, draw_pixbuf, 0, 0, x, y, w, h, GDK_RGB_DITHER_NONE, 0, 0); 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_free (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_free (tag_list); tag_list = ntl; } gst_tag_list_free (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_free (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_free (tag_list); tag_list = NULL; } if (filename) { g_string_free (filename, TRUE); filename = NULL; } return ret; }