b9bdabcdbfdde3b576a90bbc86942abc2fe3a8a8
[blender-staging.git] / source / blender / windowmanager / intern / wm_operators.c
1 /**
2  * $Id:
3  *
4  * ***** BEGIN GPL LICENSE BLOCK *****
5  *
6  * This program is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU General Public License
8  * as published by the Free Software Foundation; either version 2
9  * of the License, or (at your option) any later version. 
10  *
11  * This program is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  * GNU General Public License for more details.
15  *
16  * You should have received a copy of the GNU General Public License
17  * along with this program; if not, write to the Free Software Foundation,
18  * Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
19  *
20  * The Original Code is Copyright (C) 2007 Blender Foundation.
21  * All rights reserved.
22  *
23  * 
24  * Contributor(s): Blender Foundation
25  *
26  * ***** END GPL LICENSE BLOCK *****
27  */
28
29 #include <float.h>
30 #define _USE_MATH_DEFINES
31 #include <math.h>
32 #include <string.h>
33
34 #include "DNA_ID.h"
35 #include "DNA_screen_types.h"
36 #include "DNA_userdef_types.h"
37 #include "DNA_windowmanager_types.h"
38
39 #include "MEM_guardedalloc.h"
40
41 #include "BLI_blenlib.h"
42 #include "BLI_dynstr.h" /*for WM_operator_pystring */
43
44 #include "BKE_blender.h"
45 #include "BKE_context.h"
46 #include "BKE_idprop.h"
47 #include "BKE_library.h"
48 #include "BKE_global.h"
49 #include "BKE_main.h"
50 #include "BKE_utildefines.h"
51
52 #include "BIF_gl.h"
53 #include "BIF_glutil.h" /* for paint cursor */
54 #include "IMB_imbuf_types.h"
55
56 #include "ED_fileselect.h"
57 #include "ED_screen.h"
58
59 #include "RNA_access.h"
60 #include "RNA_define.h"
61
62 #include "UI_interface.h"
63 #include "UI_resources.h"
64
65 #include "WM_api.h"
66 #include "WM_types.h"
67
68 #include "wm.h"
69 #include "wm_window.h"
70 #include "wm_subwindow.h"
71 #include "wm_event_system.h"
72
73
74
75 static ListBase global_ops= {NULL, NULL};
76
77 /* ************ operator API, exported ********** */
78
79 wmOperatorType *WM_operatortype_find(const char *idname)
80 {
81         wmOperatorType *ot;
82         
83         for(ot= global_ops.first; ot; ot= ot->next) {
84                 if(strncmp(ot->idname, idname, OP_MAX_TYPENAME)==0)
85                    return ot;
86         }
87         printf("search for unknown operator %s\n", idname);
88         return NULL;
89 }
90
91 wmOperatorType *WM_operatortype_first(void)
92 {
93         return global_ops.first;
94 }
95
96 /* all ops in 1 list (for time being... needs evaluation later) */
97 void WM_operatortype_append(void (*opfunc)(wmOperatorType*))
98 {
99         wmOperatorType *ot;
100         
101         ot= MEM_callocN(sizeof(wmOperatorType), "operatortype");
102         ot->srna= RNA_def_struct(&BLENDER_RNA, "", "OperatorProperties");
103         opfunc(ot);
104         RNA_def_struct_ui_text(ot->srna, ot->name, "DOC_BROKEN"); /* TODO - add a discription to wmOperatorType? */
105         RNA_def_struct_identifier(ot->srna, ot->idname);
106         BLI_addtail(&global_ops, ot);
107 }
108
109 void WM_operatortype_append_ptr(void (*opfunc)(wmOperatorType*, void*), void *userdata)
110 {
111         wmOperatorType *ot;
112
113         ot= MEM_callocN(sizeof(wmOperatorType), "operatortype");
114         ot->srna= RNA_def_struct(&BLENDER_RNA, "", "OperatorProperties");
115         opfunc(ot, userdata);
116         RNA_def_struct_ui_text(ot->srna, ot->name, "DOC_BROKEN"); /* TODO - add a discription to wmOperatorType? */
117         RNA_def_struct_identifier(ot->srna, ot->idname);
118         BLI_addtail(&global_ops, ot);
119 }
120
121 int WM_operatortype_remove(const char *idname)
122 {
123         wmOperatorType *ot = WM_operatortype_find(idname);
124
125         if (ot==NULL)
126                 return 0;
127         
128         BLI_remlink(&global_ops, ot);
129         RNA_struct_free(&BLENDER_RNA, ot->srna);
130         MEM_freeN(ot);
131
132         return 1;
133 }
134
135 /* print a string representation of the operator, with the args that it runs 
136  * so python can run it again */
137 char *WM_operator_pystring(wmOperator *op)
138 {
139         const char *arg_name= NULL;
140
141         PropertyRNA *prop, *iterprop;
142         CollectionPropertyIterator iter;
143
144         /* for building the string */
145         DynStr *dynstr= BLI_dynstr_new();
146         char *cstring, *buf;
147         int first_iter=1;
148
149         BLI_dynstr_appendf(dynstr, "%s(", op->idname);
150
151         iterprop= RNA_struct_iterator_property(op->ptr);
152         RNA_property_collection_begin(op->ptr, iterprop, &iter);
153
154         for(; iter.valid; RNA_property_collection_next(&iter)) {
155                 prop= iter.ptr.data;
156                 arg_name= RNA_property_identifier(&iter.ptr, prop);
157
158                 if (strcmp(arg_name, "rna_type")==0) continue;
159
160                 buf= RNA_property_as_string(op->ptr, prop);
161                 
162                 BLI_dynstr_appendf(dynstr, first_iter?"%s=%s":", %s=%s", arg_name, buf);
163
164                 MEM_freeN(buf);
165                 first_iter = 0;
166         }
167
168         RNA_property_collection_end(&iter);
169
170         BLI_dynstr_append(dynstr, ")");
171
172         cstring = BLI_dynstr_get_cstring(dynstr);
173         BLI_dynstr_free(dynstr);
174         return cstring;
175 }
176
177 void WM_operator_properties_create(PointerRNA *ptr, const char *opstring)
178 {
179         wmOperatorType *ot= WM_operatortype_find(opstring);
180
181         if(ot)
182                 RNA_pointer_create(NULL, ot->srna, NULL, ptr);
183         else
184                 RNA_pointer_create(NULL, &RNA_OperatorProperties, NULL, ptr);
185 }
186
187 void WM_operator_properties_free(PointerRNA *ptr)
188 {
189         IDProperty *properties= ptr->data;
190
191         if(properties) {
192                 IDP_FreeProperty(properties);
193                 MEM_freeN(properties);
194         }
195 }
196
197 /* ************ default op callbacks, exported *********** */
198
199 /* invoke callback, uses enum property named "type" */
200 /* only weak thing is the fixed property name... */
201 int WM_menu_invoke(bContext *C, wmOperator *op, wmEvent *event)
202 {
203         PropertyRNA *prop= RNA_struct_find_property(op->ptr, "type");
204         uiMenuItem *head;
205
206         if(prop==NULL) {
207                 printf("WM_menu_invoke: %s has no \"type\" enum property\n", op->type->idname);
208         }
209         else if (RNA_property_type(op->ptr, prop) != PROP_ENUM) {
210                 printf("WM_menu_invoke: %s \"type\" is not an enum property\n", op->type->idname);
211         }
212         else {
213                 head= uiPupMenuBegin(op->type->name, 0);
214                 uiMenuItemsEnumO(head, op->type->idname, "type");
215                 uiPupMenuEnd(C, head);
216         }
217
218         return OPERATOR_CANCELLED;
219 }
220
221 /* op->invoke */
222 int WM_operator_confirm(bContext *C, wmOperator *op, wmEvent *event)
223 {
224         uiMenuItem *head;
225
226         head= uiPupMenuBegin("OK?", ICON_HELP);
227         uiMenuItemO(head, 0, op->type->idname);
228         uiPupMenuEnd(C, head);
229         
230         return OPERATOR_CANCELLED;
231 }
232
233 /* op->invoke, opens fileselect if filename property not set, otherwise executes */
234 int WM_operator_filesel(bContext *C, wmOperator *op, wmEvent *event)
235 {
236         if (RNA_property_is_set(op->ptr, "filename")) {
237                 return WM_operator_call(C, op);
238         } else {
239                 SpaceFile *sfile;
240
241                 ED_screen_full_newspace(C, CTX_wm_area(C), SPACE_FILE);
242                 
243                 /* settings for filebrowser */
244                 sfile= (SpaceFile*)CTX_wm_space_data(C);
245                 sfile->op = op;
246                 ED_fileselect_set_params(sfile, FILE_BLENDER, op->type->name, "", 0, 0, 0);
247
248                 /* screen and area have been reset already in ED_screen_full_newspace */
249                 return OPERATOR_RUNNING_MODAL;
250         }
251 }
252
253 /* op->poll */
254 int WM_operator_winactive(bContext *C)
255 {
256         if(CTX_wm_window(C)==NULL) return 0;
257         return 1;
258 }
259
260 /* ************ window / screen operator definitions ************** */
261
262 static void WM_OT_window_duplicate(wmOperatorType *ot)
263 {
264         ot->name= "Duplicate Window";
265         ot->idname= "WM_OT_window_duplicate";
266         
267         ot->invoke= WM_operator_confirm;
268         ot->exec= wm_window_duplicate_op;
269         ot->poll= WM_operator_winactive;
270 }
271
272 static void WM_OT_save_homefile(wmOperatorType *ot)
273 {
274         ot->name= "Save User Settings";
275         ot->idname= "WM_OT_save_homefile";
276         
277         ot->invoke= WM_operator_confirm;
278         ot->exec= WM_write_homefile;
279         ot->poll= WM_operator_winactive;
280 }
281
282 static void WM_OT_read_homefile(wmOperatorType *ot)
283 {
284         ot->name= "Reload Start-Up File";
285         ot->idname= "WM_OT_read_homefile";
286         
287         ot->invoke= WM_operator_confirm;
288         ot->exec= WM_read_homefile;
289         ot->poll= WM_operator_winactive;
290         
291         RNA_def_boolean(ot->srna, "factory", 0, "Factory Settings", "");
292 }
293
294
295 /* ********* recent file *********** */
296
297 static int recentfile_exec(bContext *C, wmOperator *op)
298 {
299         int event= RNA_enum_get(op->ptr, "nr");
300
301         // XXX wm in context is not set correctly after WM_read_file -> crash
302         // do it before for now, but is this correct with multiple windows?
303
304         if(event>0) {
305                 if (G.sce[0] && (event==1)) {
306                         WM_event_add_notifier(C, NC_WINDOW, NULL);
307                         WM_read_file(C, G.sce, op->reports);
308                 }
309                 else {
310                         struct RecentFile *recent = BLI_findlink(&(G.recent_files), event-2);
311                         if(recent) {
312                                 WM_event_add_notifier(C, NC_WINDOW, NULL);
313                                 WM_read_file(C, recent->filename, op->reports);
314                         }
315                 }
316         }
317         return 0;
318 }
319
320 static int wm_recentfile_invoke(bContext *C, wmOperator *op, wmEvent *event)
321 {
322         struct RecentFile *recent;
323         uiMenuItem *head;
324         int i, ofs= 0;
325
326         head= uiPupMenuBegin("Open Recent", 0);
327
328         if(G.sce[0]) {
329                 uiMenuItemIntO(head, G.sce, 0, op->type->idname, "nr", 1);
330                 ofs = 1;
331         }
332         
333         for(recent = G.recent_files.first, i=0; (i<U.recent_files) && (recent); recent = recent->next, i++)
334                 if(strcmp(recent->filename, G.sce))
335                         uiMenuItemIntO(head, recent->filename, 0, op->type->idname, "nr", i+ofs+1);
336
337         uiPupMenuEnd(C, head);
338         
339         return OPERATOR_CANCELLED;
340 }
341
342 static void WM_OT_open_recentfile(wmOperatorType *ot)
343 {
344         ot->name= "Open Recent File";
345         ot->idname= "WM_OT_open_recentfile";
346         
347         ot->invoke= wm_recentfile_invoke;
348         ot->exec= recentfile_exec;
349         ot->poll= WM_operator_winactive;
350         
351         RNA_def_property(ot->srna, "nr", PROP_ENUM, PROP_NONE);
352 }
353
354 /* ********* main file *********** */
355
356 static int wm_mainfile_invoke(bContext *C, wmOperator *op, wmEvent *event)
357 {
358         SpaceFile *sfile;
359         
360         ED_screen_full_newspace(C, CTX_wm_area(C), SPACE_FILE);
361
362         /* settings for filebrowser */
363         sfile= (SpaceFile*)CTX_wm_space_data(C);
364         sfile->op = op;
365         // XXX replace G.sce
366         ED_fileselect_set_params(sfile, FILE_BLENDER, "Load", G.sce, 0, 0, 0);
367
368         /* screen and area have been reset already in ED_screen_full_newspace */
369
370         return OPERATOR_RUNNING_MODAL;
371 }
372
373 static int wm_mainfile_exec(bContext *C, wmOperator *op)
374 {
375         char filename[FILE_MAX];
376         RNA_string_get(op->ptr, "filename", filename);
377         
378         // XXX wm in context is not set correctly after WM_read_file -> crash
379         // do it before for now, but is this correct with multiple windows?
380         WM_event_add_notifier(C, NC_WINDOW, NULL);
381
382         WM_read_file(C, filename, op->reports);
383         
384         return 0;
385 }
386
387 static void WM_OT_open_mainfile(wmOperatorType *ot)
388 {
389         ot->name= "Open Blender File";
390         ot->idname= "WM_OT_open_mainfile";
391         
392         ot->invoke= wm_mainfile_invoke;
393         ot->exec= wm_mainfile_exec;
394         ot->poll= WM_operator_winactive;
395         
396         ot->flag= 0;
397         
398         RNA_def_property(ot->srna, "filename", PROP_STRING, PROP_FILEPATH);
399
400 }
401
402 static int wm_save_as_mainfile_invoke(bContext *C, wmOperator *op, wmEvent *event)
403 {
404         SpaceFile *sfile;
405         
406         ED_screen_full_newspace(C, CTX_wm_area(C), SPACE_FILE);
407
408         /* settings for filebrowser */
409         sfile= (SpaceFile*)CTX_wm_space_data(C);
410         sfile->op = op;
411         // XXX replace G.sce
412         ED_fileselect_set_params(sfile, FILE_BLENDER, "Save As", G.sce, 0, 0, 0);
413
414         /* screen and area have been reset already in ED_screen_full_newspace */
415
416         return OPERATOR_RUNNING_MODAL;
417 }
418
419 static int wm_save_as_mainfile_exec(bContext *C, wmOperator *op)
420 {
421         char filename[FILE_MAX];
422         RNA_string_get(op->ptr, "filename", filename);
423         
424         WM_write_file(C, filename, op->reports);
425         
426         WM_event_add_notifier(C, NC_WINDOW, NULL);
427
428         return 0;
429 }
430
431 static void WM_OT_save_as_mainfile(wmOperatorType *ot)
432 {
433         ot->name= "Open Blender File";
434         ot->idname= "WM_OT_save_as_mainfile";
435         
436         ot->invoke= wm_save_as_mainfile_invoke;
437         ot->exec= wm_save_as_mainfile_exec;
438         ot->poll= WM_operator_winactive;
439         
440         ot->flag= 0;
441         
442         RNA_def_property(ot->srna, "filename", PROP_STRING, PROP_FILEPATH);
443
444 }
445
446 /* *********************** */
447
448 static void WM_OT_window_fullscreen_toggle(wmOperatorType *ot)
449 {
450     ot->name= "Toggle Fullscreen";
451     ot->idname= "WM_OT_window_fullscreen_toggle";
452
453     ot->invoke= WM_operator_confirm;
454     ot->exec= wm_window_fullscreen_toggle_op;
455     ot->poll= WM_operator_winactive;
456 }
457
458 static int wm_exit_blender_op(bContext *C, wmOperator *op)
459 {
460         WM_operator_free(op);
461         
462         WM_exit(C);     
463         
464         return OPERATOR_FINISHED;
465 }
466
467 static void WM_OT_exit_blender(wmOperatorType *ot)
468 {
469         ot->name= "Exit Blender";
470         ot->idname= "WM_OT_exit_blender";
471
472         ot->invoke= WM_operator_confirm;
473         ot->exec= wm_exit_blender_op;
474         ot->poll= WM_operator_winactive;
475 }
476
477 /* ************ default paint cursors, draw always around cursor *********** */
478 /*
479  - returns handler to free 
480  - poll(bContext): returns 1 if draw should happen
481  - draw(bContext): drawing callback for paint cursor
482 */
483
484 void *WM_paint_cursor_activate(wmWindowManager *wm, int (*poll)(bContext *C),
485                                void (*draw)(bContext *C, int, int, void *customdata), void *customdata)
486 {
487         wmPaintCursor *pc= MEM_callocN(sizeof(wmPaintCursor), "paint cursor");
488         
489         BLI_addtail(&wm->paintcursors, pc);
490         
491         pc->customdata = customdata;
492         pc->poll= poll;
493         pc->draw= draw;
494         
495         return pc;
496 }
497
498 void WM_paint_cursor_end(wmWindowManager *wm, void *handle)
499 {
500         wmPaintCursor *pc;
501         
502         for(pc= wm->paintcursors.first; pc; pc= pc->next) {
503                 if(pc == (wmPaintCursor *)handle) {
504                         BLI_remlink(&wm->paintcursors, pc);
505                         MEM_freeN(pc);
506                         return;
507                 }
508         }
509 }
510
511 /* ************ window gesture operator-callback definitions ************** */
512 /*
513  * These are default callbacks for use in operators requiring gesture input
514  */
515
516 /* **************** Border gesture *************** */
517
518 /* Border gesture has two types:
519    1) WM_GESTURE_CROSS_RECT: starts a cross, on mouse click it changes to border 
520    2) WM_GESTURE_RECT: starts immediate as a border, on mouse click or release it ends
521
522    It stores 4 values (xmin, xmax, ymin, ymax) and event it ended with (event_type)
523 */
524
525 static int border_apply(bContext *C, wmOperator *op, int event_type)
526 {
527         wmGesture *gesture= op->customdata;
528         rcti *rect= gesture->customdata;
529         
530         if(rect->xmin > rect->xmax)
531                 SWAP(int, rect->xmin, rect->xmax);
532         if(rect->ymin > rect->ymax)
533                 SWAP(int, rect->ymin, rect->ymax);
534         
535         if(rect->xmin==rect->xmax || rect->ymin==rect->ymax)
536                 return 0;
537                 
538         /* operator arguments and storage. */
539         RNA_int_set(op->ptr, "xmin", rect->xmin);
540         RNA_int_set(op->ptr, "ymin", rect->ymin);
541         RNA_int_set(op->ptr, "xmax", rect->xmax);
542         RNA_int_set(op->ptr, "ymax", rect->ymax);
543         if( RNA_struct_find_property(op->ptr, "event_type") )
544                 RNA_int_set(op->ptr, "event_type", event_type);
545         
546         op->type->exec(C, op);
547         
548         return 1;
549 }
550
551 static void wm_gesture_end(bContext *C, wmOperator *op)
552 {
553         wmGesture *gesture= op->customdata;
554         
555         WM_gesture_end(C, gesture);     /* frees gesture itself, and unregisters from window */
556         op->customdata= NULL;
557
558         ED_area_tag_redraw(CTX_wm_area(C));
559         
560         if( RNA_struct_find_property(op->ptr, "cursor") )
561                 WM_cursor_restore(CTX_wm_window(C));
562 }
563
564 int WM_border_select_invoke(bContext *C, wmOperator *op, wmEvent *event)
565 {
566         op->customdata= WM_gesture_new(C, event, WM_GESTURE_CROSS_RECT);
567
568         /* add modal handler */
569         WM_event_add_modal_handler(C, &CTX_wm_window(C)->handlers, op);
570         
571         wm_gesture_tag_redraw(C);
572
573         return OPERATOR_RUNNING_MODAL;
574 }
575
576 int WM_border_select_modal(bContext *C, wmOperator *op, wmEvent *event)
577 {
578         wmGesture *gesture= op->customdata;
579         rcti *rect= gesture->customdata;
580         int sx, sy;
581         
582         switch(event->type) {
583                 case MOUSEMOVE:
584                         
585                         wm_subwindow_getorigin(CTX_wm_window(C), gesture->swinid, &sx, &sy);
586                         
587                         if(gesture->type==WM_GESTURE_CROSS_RECT && gesture->mode==0) {
588                                 rect->xmin= rect->xmax= event->x - sx;
589                                 rect->ymin= rect->ymax= event->y - sy;
590                         }
591                         else {
592                                 rect->xmax= event->x - sx;
593                                 rect->ymax= event->y - sy;
594                         }
595                         
596                         wm_gesture_tag_redraw(C);
597
598                         break;
599                         
600                 case LEFTMOUSE:
601                 case MIDDLEMOUSE:
602                 case RIGHTMOUSE:
603                         if(event->val==1) {
604                                 if(gesture->type==WM_GESTURE_CROSS_RECT && gesture->mode==0) {
605                                         gesture->mode= 1;
606                                         wm_gesture_tag_redraw(C);
607                                 }
608                         }
609                         else {
610                                 if(border_apply(C, op, event->type)) {
611                                         wm_gesture_end(C, op);
612                                         return OPERATOR_FINISHED;
613                                 }
614                                 wm_gesture_end(C, op);
615                                 return OPERATOR_CANCELLED;
616                         }
617                         break;
618                 case ESCKEY:
619                         wm_gesture_end(C, op);
620                         return OPERATOR_CANCELLED;
621         }
622         return OPERATOR_RUNNING_MODAL;
623 }
624
625 /* **************** circle gesture *************** */
626 /* works now only for selection or modal paint stuff, calls exec while hold mouse, exit on release */
627
628 int WM_gesture_circle_invoke(bContext *C, wmOperator *op, wmEvent *event)
629 {
630         op->customdata= WM_gesture_new(C, event, WM_GESTURE_CIRCLE);
631         
632         /* add modal handler */
633         WM_event_add_modal_handler(C, &CTX_wm_window(C)->handlers, op);
634         
635         wm_gesture_tag_redraw(C);
636         
637         return OPERATOR_RUNNING_MODAL;
638 }
639
640 static void gesture_circle_apply(bContext *C, wmOperator *op)
641 {
642         wmGesture *gesture= op->customdata;
643         rcti *rect= gesture->customdata;
644         
645         /* operator arguments and storage. */
646         RNA_int_set(op->ptr, "x", rect->xmin);
647         RNA_int_set(op->ptr, "y", rect->ymin);
648         RNA_int_set(op->ptr, "radius", rect->xmax);
649         
650         if(op->type->exec)
651                 op->type->exec(C, op);
652 }
653
654 int WM_gesture_circle_modal(bContext *C, wmOperator *op, wmEvent *event)
655 {
656         wmGesture *gesture= op->customdata;
657         rcti *rect= gesture->customdata;
658         int sx, sy;
659         
660         switch(event->type) {
661                 case MOUSEMOVE:
662                         
663                         wm_subwindow_getorigin(CTX_wm_window(C), gesture->swinid, &sx, &sy);
664                         
665                         rect->xmin= event->x - sx;
666                         rect->ymin= event->y - sy;
667                         
668                         wm_gesture_tag_redraw(C);
669                         
670                         if(gesture->mode)
671                                 gesture_circle_apply(C, op);
672
673                         break;
674                 case WHEELUPMOUSE:
675                         rect->xmax += 2 + rect->xmax/10;
676                         wm_gesture_tag_redraw(C);
677                         break;
678                 case WHEELDOWNMOUSE:
679                         rect->xmax -= 2 + rect->xmax/10;
680                         if(rect->xmax < 1) rect->xmax= 1;
681                         wm_gesture_tag_redraw(C);
682                         break;
683                 case LEFTMOUSE:
684                 case MIDDLEMOUSE:
685                 case RIGHTMOUSE:
686                         if(event->val==0) {     /* key release */
687                                 wm_gesture_end(C, op);
688                                 return OPERATOR_FINISHED;
689                         }
690                         else {
691                                 if( RNA_struct_find_property(op->ptr, "event_type") )
692                                         RNA_int_set(op->ptr, "event_type", event->type);
693                                 
694                                 /* apply first click */
695                                 gesture_circle_apply(C, op);
696                                 gesture->mode= 1;
697                         }
698                         break;
699                 case ESCKEY:
700                         wm_gesture_end(C, op);
701                         return OPERATOR_CANCELLED;
702         }
703         return OPERATOR_RUNNING_MODAL;
704 }
705
706 #if 0
707 /* template to copy from */
708 void WM_OT_circle_gesture(wmOperatorType *ot)
709 {
710         ot->name= "Circle Gesture";
711         ot->idname= "WM_OT_circle_gesture";
712         
713         ot->invoke= WM_gesture_circle_invoke;
714         ot->modal= WM_gesture_circle_modal;
715         
716         ot->poll= WM_operator_winactive;
717         
718         RNA_def_property(ot->srna, "x", PROP_INT, PROP_NONE);
719         RNA_def_property(ot->srna, "y", PROP_INT, PROP_NONE);
720         RNA_def_property(ot->srna, "radius", PROP_INT, PROP_NONE);
721
722 }
723 #endif
724
725 /* **************** Tweak gesture *************** */
726
727 static void tweak_gesture_modal(bContext *C, wmEvent *event)
728 {
729         wmWindow *window= CTX_wm_window(C);
730         wmGesture *gesture= window->tweak;
731         rcti *rect= gesture->customdata;
732         int sx, sy, val;
733         
734         switch(event->type) {
735                 case MOUSEMOVE:
736                         
737                         wm_subwindow_getorigin(window, gesture->swinid, &sx, &sy);
738                         
739                         rect->xmax= event->x - sx;
740                         rect->ymax= event->y - sy;
741                         
742                         if((val= wm_gesture_evaluate(C, gesture))) {
743                                 wmEvent event;
744
745                                 event= *(window->eventstate);
746                                 if(gesture->event_type==LEFTMOUSE)
747                                         event.type= EVT_TWEAK_L;
748                                 else if(gesture->event_type==RIGHTMOUSE)
749                                         event.type= EVT_TWEAK_R;
750                                 else
751                                         event.type= EVT_TWEAK_M;
752                                 event.val= val;
753                                 /* mouse coords! */
754                                 wm_event_add(window, &event);
755                                 
756                                 WM_gesture_end(C, gesture);     /* frees gesture itself, and unregisters from window */
757                                 window->tweak= NULL;
758                         }
759                         
760                         break;
761                         
762                 case LEFTMOUSE:
763                 case RIGHTMOUSE:
764                 case MIDDLEMOUSE:
765                         if(gesture->event_type==event->type) {
766                                 WM_gesture_end(C, gesture);
767                                 window->tweak= NULL;
768                                 
769                                 /* when tweak fails we should give the other keymap entries a chance */
770                                 event->val= KM_RELEASE;
771                         }
772                         break;
773                 default:
774                         WM_gesture_end(C, gesture);
775                         window->tweak= NULL;
776         }
777 }
778
779 /* standard tweak, called after window handlers passed on event */
780 void wm_tweakevent_test(bContext *C, wmEvent *event, int action)
781 {
782         wmWindow *win= CTX_wm_window(C);
783         
784         if(win->tweak==NULL) {
785                 if(CTX_wm_region(C)) {
786                         if(event->val) { // pressed
787                                 if( ELEM3(event->type, LEFTMOUSE, MIDDLEMOUSE, RIGHTMOUSE) )
788                                         win->tweak= WM_gesture_new(C, event, WM_GESTURE_TWEAK);
789                         }
790                 }
791         }
792         else {
793                 if(action==WM_HANDLER_BREAK) {
794                         WM_gesture_end(C, win->tweak);
795                         win->tweak= NULL;
796                 }
797                 else
798                         tweak_gesture_modal(C, event);
799         }
800 }
801
802 /* *********************** lasso gesture ****************** */
803
804 int WM_gesture_lasso_invoke(bContext *C, wmOperator *op, wmEvent *event)
805 {
806         op->customdata= WM_gesture_new(C, event, WM_GESTURE_LASSO);
807         
808         /* add modal handler */
809         WM_event_add_modal_handler(C, &CTX_wm_window(C)->handlers, op);
810         
811         wm_gesture_tag_redraw(C);
812         
813         if( RNA_struct_find_property(op->ptr, "cursor") )
814                 WM_cursor_modal(CTX_wm_window(C), RNA_int_get(op->ptr, "cursor"));
815         
816         return OPERATOR_RUNNING_MODAL;
817 }
818
819 int WM_gesture_lines_invoke(bContext *C, wmOperator *op, wmEvent *event)
820 {
821         op->customdata= WM_gesture_new(C, event, WM_GESTURE_LINES);
822         
823         /* add modal handler */
824         WM_event_add_modal_handler(C, &CTX_wm_window(C)->handlers, op);
825         
826         wm_gesture_tag_redraw(C);
827         
828         if( RNA_struct_find_property(op->ptr, "cursor") )
829                 WM_cursor_modal(CTX_wm_window(C), RNA_int_get(op->ptr, "cursor"));
830         
831         return OPERATOR_RUNNING_MODAL;
832 }
833
834
835 static void gesture_lasso_apply(bContext *C, wmOperator *op, int event_type)
836 {
837         wmGesture *gesture= op->customdata;
838         PointerRNA itemptr;
839         float loc[2];
840         int i;
841         short *lasso= gesture->customdata;
842         
843         /* operator storage as path. */
844
845         for(i=0; i<gesture->points; i++, lasso+=2) {
846                 loc[0]= lasso[0];
847                 loc[1]= lasso[1];
848                 RNA_collection_add(op->ptr, "path", &itemptr);
849                 RNA_float_set_array(&itemptr, "loc", loc);
850         }
851         
852         wm_gesture_end(C, op);
853                 
854         if(op->type->exec)
855                 op->type->exec(C, op);
856         
857 }
858
859 int WM_gesture_lasso_modal(bContext *C, wmOperator *op, wmEvent *event)
860 {
861         wmGesture *gesture= op->customdata;
862         int sx, sy;
863         
864         switch(event->type) {
865                 case MOUSEMOVE:
866                         
867                         wm_gesture_tag_redraw(C);
868                         
869                         wm_subwindow_getorigin(CTX_wm_window(C), gesture->swinid, &sx, &sy);
870                         if(gesture->points < WM_LASSO_MAX_POINTS) {
871                                 short *lasso= gesture->customdata;
872                                 lasso += 2 * gesture->points;
873                                 lasso[0] = event->x - sx;
874                                 lasso[1] = event->y - sy;
875                                 gesture->points++;
876                         }
877                         else {
878                                 gesture_lasso_apply(C, op, event->type);
879                                 return OPERATOR_FINISHED;
880                         }
881                         break;
882                         
883                 case LEFTMOUSE:
884                 case MIDDLEMOUSE:
885                 case RIGHTMOUSE:
886                         if(event->val==0) {     /* key release */
887                                 gesture_lasso_apply(C, op, event->type);
888                                 return OPERATOR_FINISHED;
889                         }
890                         break;
891                 case ESCKEY:
892                         wm_gesture_end(C, op);
893                         return OPERATOR_CANCELLED;
894         }
895         return OPERATOR_RUNNING_MODAL;
896 }
897
898 int WM_gesture_lines_modal(bContext *C, wmOperator *op, wmEvent *event)
899 {
900         return WM_gesture_lasso_modal(C, op, event);
901 }
902
903 #if 0
904 /* template to copy from */
905
906 static int gesture_lasso_exec(bContext *C, wmOperator *op)
907 {
908         RNA_BEGIN(op->ptr, itemptr, "path") {
909                 float loc[2];
910                 
911                 RNA_float_get_array(&itemptr, "loc", loc);
912                 printf("Location: %f %f\n", loc[0], loc[1]);
913         }
914         RNA_END;
915         
916         return OPERATOR_FINISHED;
917 }
918
919 void WM_OT_lasso_gesture(wmOperatorType *ot)
920 {
921         PropertyRNA *prop;
922         
923         ot->name= "Lasso Gesture";
924         ot->idname= "WM_OT_lasso_gesture";
925         
926         ot->invoke= WM_gesture_lasso_invoke;
927         ot->modal= WM_gesture_lasso_modal;
928         ot->exec= gesture_lasso_exec;
929         
930         ot->poll= WM_operator_winactive;
931         
932         prop= RNA_def_property(ot->srna, "path", PROP_COLLECTION, PROP_NONE);
933         RNA_def_property_struct_runtime(prop, &RNA_OperatorMousePath);
934 }
935 #endif
936
937 /* *********************** radial control ****************** */
938
939 const int WM_RADIAL_CONTROL_DISPLAY_SIZE = 200;
940
941 typedef struct wmRadialControl {
942         int mode;
943         float initial_value, value, max_value;
944         int initial_mouse[2];
945         void *cursor;
946         GLuint tex;
947 } wmRadialControl;
948
949 static void wm_radial_control_paint(bContext *C, int x, int y, void *customdata)
950 {
951         wmRadialControl *rc = (wmRadialControl*)customdata;
952         ARegion *ar = CTX_wm_region(C);
953         float r1=0.0f, r2=0.0f, r3=0.0f, angle=0.0f;
954
955         /* Keep cursor in the original place */
956         x = rc->initial_mouse[0] - ar->winrct.xmin;
957         y = rc->initial_mouse[1] - ar->winrct.ymin;
958
959         glPushMatrix();
960         
961         glTranslatef((float)x, (float)y, 0.0f);
962
963         if(rc->mode == WM_RADIALCONTROL_SIZE) {
964                 r1= rc->value;
965                 r2= rc->initial_value;
966                 r3= r1;
967         } else if(rc->mode == WM_RADIALCONTROL_STRENGTH) {
968                 r1= (1 - rc->value) * WM_RADIAL_CONTROL_DISPLAY_SIZE;
969                 r2= WM_RADIAL_CONTROL_DISPLAY_SIZE;
970                 r3= WM_RADIAL_CONTROL_DISPLAY_SIZE;
971         } else if(rc->mode == WM_RADIALCONTROL_ANGLE) {
972                 r1= r2= WM_RADIAL_CONTROL_DISPLAY_SIZE;
973                 r3= WM_RADIAL_CONTROL_DISPLAY_SIZE;
974                 angle = rc->value;
975         }
976
977         glColor4ub(255, 255, 255, 128);
978         glEnable( GL_LINE_SMOOTH );
979         glEnable(GL_BLEND);
980
981         if(rc->mode == WM_RADIALCONTROL_ANGLE)
982                 fdrawline(0, 0, WM_RADIAL_CONTROL_DISPLAY_SIZE, 0);
983
984         if(rc->tex) {
985                 const float str = rc->mode == WM_RADIALCONTROL_STRENGTH ? (rc->value + 0.5) : 1;
986
987                 if(rc->mode == WM_RADIALCONTROL_ANGLE) {
988                         glRotatef(angle, 0, 0, 1);
989                         fdrawline(0, 0, WM_RADIAL_CONTROL_DISPLAY_SIZE, 0);
990                 }
991
992                 glBindTexture(GL_TEXTURE_2D, rc->tex);
993
994                 glTexParameterf(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_LINEAR);
995                 glTexParameterf(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_LINEAR);
996
997                 glEnable(GL_TEXTURE_2D);
998                 glBegin(GL_QUADS);
999                 glColor4f(0,0,0, str);
1000                 glTexCoord2f(0,0);
1001                 glVertex2f(-r3, -r3);
1002                 glTexCoord2f(1,0);
1003                 glVertex2f(r3, -r3);
1004                 glTexCoord2f(1,1);
1005                 glVertex2f(r3, r3);
1006                 glTexCoord2f(0,1);
1007                 glVertex2f(-r3, r3);
1008                 glEnd();
1009                 glDisable(GL_TEXTURE_2D);
1010         }
1011
1012         glColor4ub(255, 255, 255, 128); 
1013         glutil_draw_lined_arc(0.0, M_PI*2.0, r1, 40);
1014         glutil_draw_lined_arc(0.0, M_PI*2.0, r2, 40);
1015         glDisable(GL_BLEND);
1016         glDisable( GL_LINE_SMOOTH );
1017         
1018         glPopMatrix();
1019 }
1020
1021 int WM_radial_control_modal(bContext *C, wmOperator *op, wmEvent *event)
1022 {
1023         wmRadialControl *rc = (wmRadialControl*)op->customdata;
1024         int mode, initial_mouse[2], delta[2];
1025         float dist;
1026         double new_value = RNA_float_get(op->ptr, "new_value");
1027         int ret = OPERATOR_RUNNING_MODAL;
1028
1029         mode = RNA_int_get(op->ptr, "mode");
1030         RNA_int_get_array(op->ptr, "initial_mouse", initial_mouse);
1031
1032         switch(event->type) {
1033         case MOUSEMOVE:
1034                 delta[0]= initial_mouse[0] - event->x;
1035                 delta[1]= initial_mouse[1] - event->y;
1036                 dist= sqrt(delta[0]*delta[0]+delta[1]*delta[1]);
1037
1038                 if(mode == WM_RADIALCONTROL_SIZE)
1039                         new_value = dist;
1040                 else if(mode == WM_RADIALCONTROL_STRENGTH) {
1041                         new_value = 1 - dist / WM_RADIAL_CONTROL_DISPLAY_SIZE;
1042                 } else if(mode == WM_RADIALCONTROL_ANGLE)
1043                         new_value = ((int)(atan2(delta[1], delta[0]) * (180.0 / M_PI)) + 180);
1044                 
1045                 if(event->ctrl) {
1046                         if(mode == WM_RADIALCONTROL_STRENGTH)
1047                                 new_value = ((int)(new_value * 100) / 10*10) / 100.0f;
1048                         else
1049                                 new_value = ((int)new_value + 5) / 10*10;
1050                 }
1051                 
1052                 break;
1053         case ESCKEY:
1054         case RIGHTMOUSE:
1055                 ret = OPERATOR_CANCELLED;
1056                 break;
1057         case LEFTMOUSE:
1058         case PADENTER:
1059                 op->type->exec(C, op);
1060                 ret = OPERATOR_FINISHED;
1061                 break;
1062         }
1063
1064         /* Clamp */
1065         if(new_value > rc->max_value)
1066                 new_value = rc->max_value;
1067         else if(new_value < 0)
1068                 new_value = 0;
1069
1070         /* Update paint data */
1071         rc->value = new_value;
1072
1073         RNA_float_set(op->ptr, "new_value", new_value);
1074
1075         if(ret != OPERATOR_RUNNING_MODAL) {
1076                 WM_paint_cursor_end(CTX_wm_manager(C), rc->cursor);
1077                 MEM_freeN(rc);
1078         }
1079         
1080         ED_region_tag_redraw(CTX_wm_region(C));
1081
1082         return ret;
1083 }
1084
1085 /* Expects the operator customdata to be an ImBuf (or NULL) */
1086 int WM_radial_control_invoke(bContext *C, wmOperator *op, wmEvent *event)
1087 {
1088         wmRadialControl *rc = MEM_callocN(sizeof(wmRadialControl), "radial control");
1089         int mode = RNA_int_get(op->ptr, "mode");
1090         float initial_value = RNA_float_get(op->ptr, "initial_value");
1091         int mouse[2] = {event->x, event->y};
1092
1093         if(mode == WM_RADIALCONTROL_SIZE) {
1094                 rc->max_value = 200;
1095                 mouse[0]-= initial_value;
1096         }
1097         else if(mode == WM_RADIALCONTROL_STRENGTH) {
1098                 rc->max_value = 1;
1099                 mouse[0]-= WM_RADIAL_CONTROL_DISPLAY_SIZE * (1 - initial_value);
1100         }
1101         else if(mode == WM_RADIALCONTROL_ANGLE) {
1102                 rc->max_value = 360;
1103                 mouse[0]-= WM_RADIAL_CONTROL_DISPLAY_SIZE * cos(initial_value);
1104                 mouse[1]-= WM_RADIAL_CONTROL_DISPLAY_SIZE * sin(initial_value);
1105                 initial_value *= 180.0f/M_PI;
1106         }
1107
1108         if(op->customdata) {
1109                 ImBuf *im = (ImBuf*)op->customdata;
1110                 /* Build GL texture */
1111                 glGenTextures(1, &rc->tex);
1112                 glBindTexture(GL_TEXTURE_2D, rc->tex);
1113                 glTexImage2D(GL_TEXTURE_2D, 0, GL_ALPHA, im->x, im->y, 0, GL_ALPHA, GL_FLOAT, im->rect_float);
1114                 MEM_freeN(im->rect_float);
1115                 MEM_freeN(im);
1116         }
1117
1118         RNA_int_set_array(op->ptr, "initial_mouse", mouse);
1119         RNA_float_set(op->ptr, "new_value", initial_value);
1120                 
1121         op->customdata = rc;
1122         rc->mode = mode;
1123         rc->initial_value = initial_value;
1124         rc->initial_mouse[0] = mouse[0];
1125         rc->initial_mouse[1] = mouse[1];
1126         rc->cursor = WM_paint_cursor_activate(CTX_wm_manager(C), op->type->poll,
1127                                               wm_radial_control_paint, op->customdata);
1128
1129         /* add modal handler */
1130         WM_event_add_modal_handler(C, &CTX_wm_window(C)->handlers, op);
1131         
1132         WM_radial_control_modal(C, op, event);
1133         
1134         return OPERATOR_RUNNING_MODAL;
1135 }
1136
1137 /* Gets a descriptive string of the operation */
1138 void WM_radial_control_string(wmOperator *op, char str[], int maxlen)
1139 {
1140         int mode = RNA_int_get(op->ptr, "mode");
1141         float v = RNA_float_get(op->ptr, "new_value");
1142
1143         if(mode == WM_RADIALCONTROL_SIZE)
1144                 sprintf(str, "Size: %d", (int)v);
1145         else if(mode == WM_RADIALCONTROL_STRENGTH)
1146                 sprintf(str, "Strength: %d", (int)v);
1147         else if(mode == WM_RADIALCONTROL_ANGLE)
1148                 sprintf(str, "Angle: %d", (int)(v * 180.0f/M_PI));
1149 }
1150
1151 /** Important: this doesn't define an actual operator, it
1152     just sets up the common parts of the radial control op. **/
1153 void WM_OT_radial_control_partial(wmOperatorType *ot)
1154 {
1155         static EnumPropertyItem prop_mode_items[] = {
1156                 {WM_RADIALCONTROL_SIZE, "SIZE", "Size", ""},
1157                 {WM_RADIALCONTROL_STRENGTH, "STRENGTH", "Strength", ""},
1158                 {WM_RADIALCONTROL_ANGLE, "ANGLE", "Angle", ""},
1159                 {0, NULL, NULL, NULL}};
1160
1161         /* Should be set in custom invoke() */
1162         RNA_def_float(ot->srna, "initial_value", 0, 0, FLT_MAX, "Initial Value", "", 0, FLT_MAX);
1163
1164         /* Set internally, should be used in custom exec() to get final value */
1165         RNA_def_float(ot->srna, "new_value", 0, 0, FLT_MAX, "New Value", "", 0, FLT_MAX);
1166
1167         /* Should be set before calling operator */
1168         RNA_def_enum(ot->srna, "mode", prop_mode_items, 0, "Mode", "");
1169
1170         /* Internal */
1171         RNA_def_int_vector(ot->srna, "initial_mouse", 2, NULL, INT_MIN, INT_MAX, "initial_mouse", "", INT_MIN, INT_MAX);
1172 }
1173
1174 /* ******************************************************* */
1175  
1176 /* called on initialize WM_exit() */
1177 void wm_operatortype_free(void)
1178 {
1179         BLI_freelistN(&global_ops);
1180 }
1181
1182 /* called on initialize WM_init() */
1183 void wm_operatortype_init(void)
1184 {
1185         WM_operatortype_append(WM_OT_window_duplicate);
1186         WM_operatortype_append(WM_OT_read_homefile);
1187         WM_operatortype_append(WM_OT_save_homefile);
1188         WM_operatortype_append(WM_OT_window_fullscreen_toggle);
1189         WM_operatortype_append(WM_OT_exit_blender);
1190         WM_operatortype_append(WM_OT_open_recentfile);
1191         WM_operatortype_append(WM_OT_open_mainfile);
1192         WM_operatortype_append(WM_OT_jobs_timer);
1193         WM_operatortype_append(WM_OT_save_as_mainfile);
1194 }
1195
1196 /* default keymap for windows and screens, only call once per WM */
1197 void wm_window_keymap(wmWindowManager *wm)
1198 {
1199         ListBase *keymap= WM_keymap_listbase(wm, "Window", 0, 0);
1200         
1201         /* items to make WM work */
1202         WM_keymap_verify_item(keymap, "WM_OT_jobs_timer", TIMERJOBS, KM_ANY, KM_ANY, 0);
1203         
1204         /* note, this doesn't replace existing keymap items */
1205         WM_keymap_verify_item(keymap, "WM_OT_window_duplicate", WKEY, KM_PRESS, KM_CTRL|KM_ALT, 0);
1206         WM_keymap_verify_item(keymap, "WM_OT_read_homefile", XKEY, KM_PRESS, KM_CTRL, 0);
1207         WM_keymap_verify_item(keymap, "WM_OT_save_homefile", UKEY, KM_PRESS, KM_CTRL, 0);
1208         WM_keymap_verify_item(keymap, "WM_OT_open_recentfile", OKEY, KM_PRESS, KM_CTRL, 0);
1209         WM_keymap_verify_item(keymap, "WM_OT_open_mainfile", F1KEY, KM_PRESS, 0, 0);
1210         WM_keymap_verify_item(keymap, "WM_OT_save_as_mainfile", F2KEY, KM_PRESS, 0, 0);
1211         WM_keymap_verify_item(keymap, "WM_OT_window_fullscreen_toggle", F11KEY, KM_PRESS, 0, 0);
1212         WM_keymap_verify_item(keymap, "WM_OT_exit_blender", QKEY, KM_PRESS, KM_CTRL, 0);
1213
1214 }
1215