Version 0.4
[colorize.git] / gui / menu.c
1 #include <gtk/gtk.h>
2 #include <gdk/gdkkeysyms.h>
3 #include <string.h>
4 #include <stdlib.h>
5 #include <stdarg.h>
6 #include "../src/mark.h"
7 #ifdef WITH_OPENCV
8 #include "../src/opticalflow.h"
9 #endif
10 #include "main.h"
11 #include "menu.h"
12 #include "toolbar.h"
13 #include "image.h"
14 #include "palette.h"
15 #include "timeline.h"
16 #include "colorize.h"
17 #include "brightcontrast.h"
18 #include "level.h"
19 #include "fill.h"
20 #ifdef WITH_OPENCV
21 #include "flow.h"
22 #endif
23
24 /*
25  * menus
26  */
27
28 static void open_event(gpointer priv)
29 {
30         GtkWidget *dialog;
31         char *filename = NULL;
32         
33         if (anything_modified) {
34                 int ret;
35                 GtkWidget *dialog = gtk_message_dialog_new (GTK_WINDOW(main_window),
36                         GTK_DIALOG_DESTROY_WITH_PARENT,
37                         GTK_MESSAGE_ERROR,
38                         GTK_BUTTONS_OK_CANCEL,
39                         "Image has been changed, really open new image?");
40                 ret = gtk_dialog_run(GTK_DIALOG (dialog));
41                 gtk_widget_destroy(dialog);
42                 if (ret != GTK_RESPONSE_OK)
43                         return;
44         }
45
46         dialog = gtk_file_chooser_dialog_new("Select first grey image",
47                 GTK_WINDOW(main_window),
48                 GTK_FILE_CHOOSER_ACTION_OPEN,
49                 GTK_STOCK_CANCEL, GTK_RESPONSE_CANCEL,
50                 GTK_STOCK_OPEN, GTK_RESPONSE_ACCEPT,
51                 NULL);
52
53         if (frame_list)
54                 gtk_file_chooser_set_filename(GTK_FILE_CHOOSER(dialog), frame_list[timeline_selected].filename);
55         if (gtk_dialog_run(GTK_DIALOG(dialog)) == GTK_RESPONSE_ACCEPT) {
56                 filename = strdup(gtk_file_chooser_get_filename(GTK_FILE_CHOOSER(dialog)));
57         }
58
59         gtk_widget_destroy(dialog);
60
61         if (filename) {
62 // do not save          timeline_select_and_save(timeline_selected, timeline_selected);
63                 
64                 create_timeline(filename);
65                 if (frame_list)
66                         create_image(frame_list[timeline_selected].filename, 1);
67                 free(filename);
68         }
69
70 }
71
72 static void save_event(gpointer priv)
73 {
74         timeline_select_and_save(timeline_selected, timeline_selected);
75 }
76
77 static void exit_event(gpointer priv)
78 {
79         main_destroy();
80 }
81
82 static void color_changed(GtkWidget *widget, GtkColorSelection *colorsel)
83 {
84 }
85
86 void undo_event(gpointer priv)
87 {
88         copy_undo_to_mark(0);
89         draw_image(0, 0, -1, -1);
90 }
91
92 void redo_event(gpointer priv)
93 {
94         copy_undo_to_mark(1);
95         draw_image(0, 0, -1, -1);
96 }
97
98 void copy_event(gpointer priv)
99 {
100         copy_color(mark_selected + 1);
101 }
102
103 static void copy_all_event(gpointer priv)
104 {
105         copy_color(0);
106 }
107
108 void paste_event(gpointer priv)
109 {
110         copy_mark_to_undo();
111         paste_color();
112         draw_image(0, 0, -1, -1);
113 }
114
115 static void color_destroy(GtkWidget *widget, gpointer data)
116 {
117         colorseldlg = NULL;
118 }
119
120 void palette_event(gpointer priv)
121 {
122         GdkColor color;
123
124         if (colorseldlg) {
125                 gtk_widget_destroy(colorseldlg);
126                 colorseldlg = NULL;
127                 return;
128         }
129
130         color.red = mark_palette[mark_selected].r * 256 + 128;
131         color.green = mark_palette[mark_selected].g * 256 + 128;
132         color.blue = mark_palette[mark_selected].b * 256 + 128;
133
134         colorseldlg = gtk_color_selection_dialog_new("Select Color");
135         colorsel = GTK_COLOR_SELECTION (GTK_COLOR_SELECTION_DIALOG (colorseldlg)->colorsel);
136         gtk_color_selection_set_previous_color (colorsel, &color);
137         gtk_color_selection_set_current_color (colorsel, &color);
138         g_signal_connect(G_OBJECT(colorsel), "color_changed", G_CALLBACK(color_changed), (gpointer)colorsel);
139         g_signal_connect(G_OBJECT(colorseldlg), "response", G_CALLBACK(color_response), NULL);
140         g_signal_connect(G_OBJECT(colorseldlg), "destroy", G_CALLBACK(color_destroy), NULL);
141
142
143         gtk_widget_show(colorseldlg);
144 }
145
146 static void keyframe_event(gpointer priv)
147 {
148         toggle_keyframe();
149 }
150
151 static void erase_event(gpointer priv)
152 {
153         printerror("Use RIGHT mouse button to erase!");
154 }
155
156 static void clear_event(gpointer priv)
157 {
158         copy_mark_to_undo();
159         erase_mark(0);
160         draw_image(0, 0, -1, -1);
161 }
162
163 static void clear_color_event(gpointer priv)
164 {
165         copy_mark_to_undo();
166         erase_mark(mark_selected + 1);
167         draw_image(0, 0, -1, -1);
168 }
169
170 void zoom_in_event(gpointer priv)
171 {
172         if (img_scale_x < 64) {
173                 int aspect = img_scale_y / img_scale_x;
174                 img_scale_x += 4;
175                 img_scale_y = img_scale_x * aspect;
176                 create_or_reset_pixbuf(0);
177         }
178 }
179
180 void zoom_out_event(gpointer priv)
181 {
182         if (img_scale_x > 4) {
183                 int aspect = img_scale_y / img_scale_x;
184                 img_scale_x -= 4;
185                 img_scale_y = img_scale_x * aspect;
186                 create_or_reset_pixbuf(0);
187         }
188 }
189
190 static void zoom_100_event(gpointer priv)
191 {
192         img_scale_x = 16;
193         img_scale_y = 16;
194         create_or_reset_pixbuf(0);
195 }
196
197 static void zoom_field_event(gpointer priv)
198 {
199         if (gtk_check_menu_item_get_active((GtkCheckMenuItem *)priv))
200                 img_scale_y = img_scale_x*2;
201         else
202                 img_scale_y = img_scale_x;
203         create_or_reset_pixbuf(1);
204 }
205
206 static void show_highlighted_event(gpointer priv)
207 {
208         highlight = gtk_check_menu_item_get_active((GtkCheckMenuItem *)priv);
209         set_button_toggel_by_label(TOGGLE_LABEL_HIGHLIGHT, highlight);
210 }
211
212 static void show_preview_event(gpointer priv)
213 {
214         preview = gtk_check_menu_item_get_active((GtkCheckMenuItem *)priv);
215         set_button_toggel_by_label(TOGGLE_LABEL_PREVIEW, preview);
216 }
217
218 void colorize_event(gpointer priv)
219 {
220         if (rendered) {
221                 printerror("Disable colorized view first.");
222                 return;
223         }
224 #ifdef WITH_OPENCV
225         if (flowview) {
226                 printerror("Disable flow view first.");
227                 return;
228         }
229 #endif
230         if (img_mark_buffer)
231                 colorize_image();
232 }
233
234 static void show_colorized_event(gpointer priv)
235 {
236         if (timeline_frames > 1)
237                 rendered = gtk_check_menu_item_get_active((GtkCheckMenuItem *)priv);
238         else
239                 rendered = 0;
240         gtk_check_menu_item_set_active((GtkCheckMenuItem *)priv, rendered);
241         set_button_toggel_by_label(TOGGLE_LABEL_RENDERED, rendered);
242 }
243
244 #ifdef WITH_OPENCV
245 static void show_flow_event(gpointer priv)
246 {
247         if (flow_enable && timeline_frames > 1)
248                 flowview = gtk_check_menu_item_get_active((GtkCheckMenuItem *)priv);
249         else
250                 flowview = 0;
251         gtk_check_menu_item_set_active((GtkCheckMenuItem *)priv, flowview);
252         set_button_toggel_by_label(TOGGLE_LABEL_FLOWVIEW, flowview);
253 }
254 #endif
255
256 static void dummy_event(gpointer priv)
257 {
258 }
259
260
261
262 struct my_menu_item {
263         enum toggle_label label;
264         GtkWidget *widget;
265         const char *text;
266         void (*handler)(gpointer priv);
267         gchar *stock;
268 };
269
270 struct my_menu {
271         const char *text;
272         struct my_menu_item *items;
273 };
274
275 struct my_menu_item file_menu_items[] = {
276         { TOGGLE_LABEL_NONE, NULL, "Open", open_event, GTK_STOCK_OPEN },
277         { TOGGLE_LABEL_NONE, NULL, "Save", save_event, GTK_STOCK_SAVE },
278         { TOGGLE_LABEL_NONE, NULL, "", dummy_event, NULL },
279         { TOGGLE_LABEL_NONE, NULL, "Exit", exit_event, GTK_STOCK_QUIT },
280         { TOGGLE_LABEL_NONE, NULL, NULL, NULL }
281 };
282
283 struct my_menu_item edit_menu_items[] = {
284         { TOGGLE_LABEL_NONE, NULL, "Undo", undo_event, GTK_STOCK_UNDO },
285         { TOGGLE_LABEL_NONE, NULL, "Redo", redo_event, GTK_STOCK_REDO },
286         { TOGGLE_LABEL_NONE, NULL, "", dummy_event, NULL },
287         { TOGGLE_LABEL_NONE, NULL, "Copy", copy_event, GTK_STOCK_COPY },
288         { TOGGLE_LABEL_NONE, NULL, "Copy All", copy_all_event, NULL },
289         { TOGGLE_LABEL_NONE, NULL, "Paste", paste_event, GTK_STOCK_PASTE },
290         { TOGGLE_LABEL_NONE, NULL, "", dummy_event, NULL },
291         { TOGGLE_LABEL_NONE, NULL, "Palette", palette_event, NULL },
292         { TOGGLE_LABEL_NONE, NULL, "Bright/Contrast", bc_event, NULL },
293         { TOGGLE_LABEL_NONE, NULL, "Adjust Levels", level_event, NULL },
294 #ifdef WITH_OPENCV
295         { TOGGLE_LABEL_NONE, NULL, "Optical Flow", flow_event, NULL },
296 #endif
297         { TOGGLE_LABEL_NONE, NULL, "", dummy_event, NULL },
298         { TOGGLE_LABEL_NONE, NULL, "Keyframe", keyframe_event, NULL },
299         { TOGGLE_LABEL_NONE, NULL, "", dummy_event, NULL },
300         { TOGGLE_LABEL_NONE, NULL, "Erase Tool", erase_event, NULL },
301         { TOGGLE_LABEL_NONE, NULL, "Clear", clear_event, GTK_STOCK_CLEAR },
302         { TOGGLE_LABEL_NONE, NULL, "Clear Color", clear_color_event, NULL },
303         { TOGGLE_LABEL_NONE, NULL, NULL, NULL }
304 };
305
306 struct my_menu_item view_menu_items[] = {
307         { TOGGLE_LABEL_NONE, NULL, "Zoom In", zoom_in_event, GTK_STOCK_ZOOM_IN },
308         { TOGGLE_LABEL_NONE, NULL, "Zoom Out", zoom_out_event, GTK_STOCK_ZOOM_OUT },
309         { TOGGLE_LABEL_NONE, NULL, "Zoom 100", zoom_100_event, GTK_STOCK_ZOOM_100 },
310         { TOGGLE_LABEL_ZOOMFIELDS, NULL, "Zoom Field", zoom_field_event, NULL },
311         { TOGGLE_LABEL_NONE, NULL, "", dummy_event, NULL },
312         { TOGGLE_LABEL_HIGHLIGHT, NULL, "Highlight color", show_highlighted_event, NULL },
313         { TOGGLE_LABEL_PREVIEW, NULL, "Preview color", show_preview_event, NULL },
314         { TOGGLE_LABEL_RENDERED, NULL, "Show colorized", show_colorized_event, NULL },
315 #ifdef WITH_OPENCV
316         { TOGGLE_LABEL_FLOWVIEW, NULL, "Show flow", show_flow_event, NULL },
317 #endif
318         { TOGGLE_LABEL_NONE, NULL, NULL, NULL }
319 };
320
321 struct my_menu_item render_menu_items[] = {
322         { TOGGLE_LABEL_NONE, NULL, "Colorize Image", colorize_event, NULL },
323         { TOGGLE_LABEL_NONE, NULL, NULL, NULL }
324 };
325
326 struct my_menu menus[] = {
327         { "_File", file_menu_items },
328         { "_Edit", edit_menu_items },
329         { "_View", view_menu_items },
330         { "_Render", render_menu_items },
331         { NULL, NULL }
332 };
333
334 /* create menu bar */
335 void create_menus(GtkWidget *menu_bar)
336 {
337         GtkWidget *root_menu;
338         GtkWidget *menu;
339         GtkWidget *menu_item;
340         struct my_menu_item *items;
341         int i, j;
342
343         /* create vbox and add menu_bar */
344         for (i = 0; menus[i].text; i++) {
345                 /* create menu with items */
346                 items = menus[i].items;
347                 menu = gtk_menu_new();
348                 for (j = 0; items[j].text; j++) {
349                         if (!items[j].text[0])
350                                 menu_item = gtk_separator_menu_item_new();
351                         else if (items[j].stock)
352                                 menu_item = gtk_image_menu_item_new_from_stock(items[j].stock, NULL);
353                         else if (items[j].label != TOGGLE_LABEL_NONE)
354                                 menu_item = gtk_check_menu_item_new_with_label(items[j].text);
355                         else
356                                 menu_item = gtk_menu_item_new_with_label(items[j].text);
357                         items[j].widget = menu_item;
358                         gtk_menu_shell_append(GTK_MENU_SHELL(menu), menu_item);
359                         g_signal_connect_swapped(menu_item, "activate", G_CALLBACK(items[j].handler), menu_item);
360                         gtk_widget_show(menu_item);
361                 }
362                 /* create root menu and add to menu_bar */
363                 root_menu = gtk_menu_item_new_with_mnemonic(menus[i].text);
364                 gtk_widget_show(root_menu);
365                 gtk_menu_item_set_submenu(GTK_MENU_ITEM(root_menu), menu);
366                 gtk_menu_shell_append(GTK_MENU_SHELL(menu_bar), root_menu);
367         }
368
369 }
370
371 void set_menu_toggel_by_label(enum toggle_label label, gboolean active)
372 {
373         struct my_menu_item *items;
374         int i, j;
375
376         for (i = 0; menus[i].text; i++) {
377                 items = menus[i].items;
378                 for (j = 0; items[j].text; j++) {
379                         if (items[j].label == label)
380                                 gtk_check_menu_item_set_active((GtkCheckMenuItem *)items[j].widget, active);
381                 }
382         }
383 }