c946f749cb59d182d384bd37de412dec3c7da1d9
[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 "diag_bc.h"
18 #include "level.h"
19 #include "fill.h"
20 #ifdef WITH_OPENCV
21 #include "diag_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         if (!mark && !highlight)
89                 return;
90         copy_undo_to_mark(0);
91         draw_image(0, 0, -1, -1);
92 }
93
94 void redo_event(gpointer priv)
95 {
96         if (!mark && !highlight)
97                 return;
98         copy_undo_to_mark(1);
99         draw_image(0, 0, -1, -1);
100 }
101
102 void copy_event(gpointer priv)
103 {
104         if (!mark && !highlight)
105                 return;
106         copy_color(mark_selected + 1);
107 }
108
109 static void copy_all_event(gpointer priv)
110 {
111         if (!mark && !highlight)
112                 return;
113         copy_color(0);
114 }
115
116 void paste_event(gpointer priv)
117 {
118         if (!mark && !highlight)
119                 return;
120         copy_mark_to_undo();
121         paste_color();
122         draw_image(0, 0, -1, -1);
123 }
124
125 static void color_destroy(GtkWidget *widget, gpointer data)
126 {
127         colorseldlg = NULL;
128 }
129
130 void palette_event(gpointer priv)
131 {
132         GdkColor color;
133
134         if (colorseldlg) {
135                 gtk_widget_destroy(colorseldlg);
136                 colorseldlg = NULL;
137                 return;
138         }
139
140         color.red = mark_palette[mark_selected].r * 256 + 128;
141         color.green = mark_palette[mark_selected].g * 256 + 128;
142         color.blue = mark_palette[mark_selected].b * 256 + 128;
143
144         colorseldlg = gtk_color_selection_dialog_new("Select Color");
145         colorsel = GTK_COLOR_SELECTION (GTK_COLOR_SELECTION_DIALOG (colorseldlg)->colorsel);
146         gtk_color_selection_set_previous_color (colorsel, &color);
147         gtk_color_selection_set_current_color (colorsel, &color);
148         g_signal_connect(G_OBJECT(colorsel), "color_changed", G_CALLBACK(color_changed), (gpointer)colorsel);
149         g_signal_connect(G_OBJECT(colorseldlg), "response", G_CALLBACK(color_response), NULL);
150         g_signal_connect(G_OBJECT(colorseldlg), "destroy", G_CALLBACK(color_destroy), NULL);
151
152
153         gtk_widget_show(colorseldlg);
154 }
155
156 static void keyframe_event(gpointer priv)
157 {
158         toggle_keyframe();
159 }
160
161 static void erase_event(gpointer priv)
162 {
163         printerror("Use RIGHT mouse button to erase!");
164 }
165
166 static void clear_event(gpointer priv)
167 {
168         copy_mark_to_undo();
169         erase_mark(0);
170         draw_image(0, 0, -1, -1);
171 }
172
173 static void clear_color_event(gpointer priv)
174 {
175         copy_mark_to_undo();
176         erase_mark(mark_selected + 1);
177         draw_image(0, 0, -1, -1);
178 }
179
180 void zoom_in_event(gpointer priv)
181 {
182         if (img_scale_x < 64) {
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 void zoom_out_event(gpointer priv)
191 {
192         if (img_scale_x > 4) {
193                 int aspect = img_scale_y / img_scale_x;
194                 img_scale_x -= 4;
195                 img_scale_y = img_scale_x * aspect;
196                 create_or_reset_pixbuf(0);
197         }
198 }
199
200 static void zoom_100_event(gpointer priv)
201 {
202         img_scale_x = 16;
203         img_scale_y = 16;
204         create_or_reset_pixbuf(0);
205 }
206
207 static void zoom_field_event(gpointer priv)
208 {
209         if (gtk_check_menu_item_get_active((GtkCheckMenuItem *)priv))
210                 img_scale_y = img_scale_x*2;
211         else
212                 img_scale_y = img_scale_x;
213         create_or_reset_pixbuf(1);
214 }
215
216 static void show_highlighted_event(gpointer priv)
217 {
218         highlight = gtk_check_menu_item_get_active((GtkCheckMenuItem *)priv);
219         set_button_toggel_by_label(TOGGLE_LABEL_HIGHLIGHT, highlight);
220         if (mark == 1 && highlight == 1) {
221                 mark = 0;
222                 set_button_toggel_by_label(TOGGLE_LABEL_MARK, mark);
223                 set_menu_toggel_by_label(TOGGLE_LABEL_MARK, mark);
224         }
225 }
226
227 static void show_marked_event(gpointer priv)
228 {
229         mark = gtk_check_menu_item_get_active((GtkCheckMenuItem *)priv);
230         set_button_toggel_by_label(TOGGLE_LABEL_MARK, mark);
231         if (mark == 1 && highlight == 1) {
232                 highlight = 0;
233                 set_button_toggel_by_label(TOGGLE_LABEL_HIGHLIGHT, highlight);
234                 set_menu_toggel_by_label(TOGGLE_LABEL_HIGHLIGHT, highlight);
235         }
236 }
237
238 static void show_preview_event(gpointer priv)
239 {
240         preview = gtk_check_menu_item_get_active((GtkCheckMenuItem *)priv);
241         set_button_toggel_by_label(TOGGLE_LABEL_PREVIEW, preview);
242 }
243
244 void colorize_event(gpointer priv)
245 {
246 #ifdef WITH_OPENCV
247         if (flowview) {
248                 printerror("Disable flow view first.");
249                 return;
250         }
251 #endif
252         if (img_mark_buffer)
253                 colorize_image();
254 }
255
256 static void show_colorized_event(gpointer priv)
257 {
258         if (timeline_frames > 1)
259                 rendered = gtk_check_menu_item_get_active((GtkCheckMenuItem *)priv);
260         else
261                 rendered = 0;
262         gtk_check_menu_item_set_active((GtkCheckMenuItem *)priv, rendered);
263         set_button_toggel_by_label(TOGGLE_LABEL_RENDERED, rendered);
264 }
265
266 #ifdef WITH_OPENCV
267 static void show_flow_event(gpointer priv)
268 {
269         if (flow_enable && timeline_frames > 1)
270                 flowview = gtk_check_menu_item_get_active((GtkCheckMenuItem *)priv);
271         else
272                 flowview = 0;
273         gtk_check_menu_item_set_active((GtkCheckMenuItem *)priv, flowview);
274         set_button_toggel_by_label(TOGGLE_LABEL_FLOWVIEW, flowview);
275 }
276 #endif
277
278 static void dummy_event(gpointer priv)
279 {
280 }
281
282
283
284 struct my_menu_item {
285         enum toggle_label label;
286         GtkWidget *widget;
287         const char *text;
288         void (*handler)(gpointer priv);
289         gchar *stock;
290 };
291
292 struct my_menu {
293         const char *text;
294         struct my_menu_item *items;
295 };
296
297 struct my_menu_item file_menu_items[] = {
298         { TOGGLE_LABEL_NONE, NULL, "Open", open_event, GTK_STOCK_OPEN },
299         { TOGGLE_LABEL_NONE, NULL, "Save", save_event, GTK_STOCK_SAVE },
300         { TOGGLE_LABEL_NONE, NULL, "", dummy_event, NULL },
301         { TOGGLE_LABEL_NONE, NULL, "Exit", exit_event, GTK_STOCK_QUIT },
302         { TOGGLE_LABEL_NONE, NULL, NULL, NULL }
303 };
304
305 struct my_menu_item edit_menu_items[] = {
306         { TOGGLE_LABEL_NONE, NULL, "Undo", undo_event, GTK_STOCK_UNDO },
307         { TOGGLE_LABEL_NONE, NULL, "Redo", redo_event, GTK_STOCK_REDO },
308         { TOGGLE_LABEL_NONE, NULL, "", dummy_event, NULL },
309         { TOGGLE_LABEL_NONE, NULL, "Copy", copy_event, GTK_STOCK_COPY },
310         { TOGGLE_LABEL_NONE, NULL, "Copy All", copy_all_event, NULL },
311         { TOGGLE_LABEL_NONE, NULL, "Paste", paste_event, GTK_STOCK_PASTE },
312         { TOGGLE_LABEL_NONE, NULL, "", dummy_event, NULL },
313         { TOGGLE_LABEL_NONE, NULL, "Palette", palette_event, NULL },
314         { TOGGLE_LABEL_NONE, NULL, "Bright/Contrast", bc_event, NULL },
315         { TOGGLE_LABEL_NONE, NULL, "Adjust Levels", level_event, NULL },
316 #ifdef WITH_OPENCV
317         { TOGGLE_LABEL_NONE, NULL, "Optical Flow", flow_event, NULL },
318 #endif
319         { TOGGLE_LABEL_NONE, NULL, "", dummy_event, NULL },
320         { TOGGLE_LABEL_NONE, NULL, "Keyframe", keyframe_event, NULL },
321         { TOGGLE_LABEL_NONE, NULL, "", dummy_event, NULL },
322         { TOGGLE_LABEL_NONE, NULL, "Erase Tool", erase_event, NULL },
323         { TOGGLE_LABEL_NONE, NULL, "Clear", clear_event, GTK_STOCK_CLEAR },
324         { TOGGLE_LABEL_NONE, NULL, "Clear Color", clear_color_event, NULL },
325         { TOGGLE_LABEL_NONE, NULL, NULL, NULL }
326 };
327
328 struct my_menu_item view_menu_items[] = {
329         { TOGGLE_LABEL_NONE, NULL, "Zoom In", zoom_in_event, GTK_STOCK_ZOOM_IN },
330         { TOGGLE_LABEL_NONE, NULL, "Zoom Out", zoom_out_event, GTK_STOCK_ZOOM_OUT },
331         { TOGGLE_LABEL_NONE, NULL, "Zoom 100", zoom_100_event, GTK_STOCK_ZOOM_100 },
332         { TOGGLE_LABEL_ZOOMFIELDS, NULL, "Zoom Field", zoom_field_event, NULL },
333         { TOGGLE_LABEL_NONE, NULL, "", dummy_event, NULL },
334         { TOGGLE_LABEL_MARK, NULL, "Mark color", show_marked_event, NULL },
335         { TOGGLE_LABEL_HIGHLIGHT, NULL, "Highlight color", show_highlighted_event, NULL },
336         { TOGGLE_LABEL_PREVIEW, NULL, "Preview color", show_preview_event, NULL },
337         { TOGGLE_LABEL_RENDERED, NULL, "Show colorized", show_colorized_event, NULL },
338 #ifdef WITH_OPENCV
339         { TOGGLE_LABEL_FLOWVIEW, NULL, "Show flow", show_flow_event, NULL },
340 #endif
341         { TOGGLE_LABEL_NONE, NULL, NULL, NULL }
342 };
343
344 struct my_menu_item render_menu_items[] = {
345         { TOGGLE_LABEL_NONE, NULL, "Colorize Image", colorize_event, NULL },
346         { TOGGLE_LABEL_NONE, NULL, NULL, NULL }
347 };
348
349 struct my_menu menus[] = {
350         { "_File", file_menu_items },
351         { "_Edit", edit_menu_items },
352         { "_View", view_menu_items },
353         { "_Render", render_menu_items },
354         { NULL, NULL }
355 };
356
357 /* create menu bar */
358 void create_menus(GtkWidget *menu_bar)
359 {
360         GtkWidget *root_menu;
361         GtkWidget *menu;
362         GtkWidget *menu_item;
363         struct my_menu_item *items;
364         int i, j;
365
366         /* create vbox and add menu_bar */
367         for (i = 0; menus[i].text; i++) {
368                 /* create menu with items */
369                 items = menus[i].items;
370                 menu = gtk_menu_new();
371                 for (j = 0; items[j].text; j++) {
372                         if (!items[j].text[0])
373                                 menu_item = gtk_separator_menu_item_new();
374                         else if (items[j].stock)
375                                 menu_item = gtk_image_menu_item_new_from_stock(items[j].stock, NULL);
376                         else if (items[j].label != TOGGLE_LABEL_NONE)
377                                 menu_item = gtk_check_menu_item_new_with_label(items[j].text);
378                         else
379                                 menu_item = gtk_menu_item_new_with_label(items[j].text);
380                         items[j].widget = menu_item;
381                         gtk_menu_shell_append(GTK_MENU_SHELL(menu), menu_item);
382                         g_signal_connect_swapped(menu_item, "activate", G_CALLBACK(items[j].handler), menu_item);
383                         gtk_widget_show(menu_item);
384                 }
385                 /* create root menu and add to menu_bar */
386                 root_menu = gtk_menu_item_new_with_mnemonic(menus[i].text);
387                 gtk_widget_show(root_menu);
388                 gtk_menu_item_set_submenu(GTK_MENU_ITEM(root_menu), menu);
389                 gtk_menu_shell_append(GTK_MENU_SHELL(menu_bar), root_menu);
390         }
391
392 }
393
394 void set_menu_toggel_by_label(enum toggle_label label, gboolean active)
395 {
396         struct my_menu_item *items;
397         int i, j;
398
399         for (i = 0; menus[i].text; i++) {
400                 items = menus[i].items;
401                 for (j = 0; items[j].text; j++) {
402                         if (items[j].label == label)
403                                 gtk_check_menu_item_set_active((GtkCheckMenuItem *)items[j].widget, active);
404                 }
405         }
406 }