Camera tracking: tracks copy/paste operator
[blender.git] / source / blender / editors / space_clip / space_clip.c
index ef0457d92df759293dec63d1ac53e859edebe2f5..82da9b3b956545af24af0a21182d3a9fd061b183 100644 (file)
@@ -1,6 +1,4 @@
 /*
- * $Id$
- *
  * ***** BEGIN GPL LICENSE BLOCK *****
  *
  * This program is free software; you can redistribute it and/or
@@ -41,6 +39,7 @@
 
 #include "BLI_blenlib.h"
 #include "BLI_utildefines.h"
+#include "BLI_math.h"
 
 #include "BKE_main.h"
 #include "BKE_context.h"
 
 #include "clip_intern.h"       // own include
 
+static void init_preview_region(const bContext *C, ARegion *ar)
+{
+       Scene *scene= CTX_data_scene(C);
+
+       ar->regiontype= RGN_TYPE_PREVIEW;
+       ar->alignment= RGN_ALIGN_TOP;
+       ar->flag|= RGN_FLAG_HIDDEN;
+
+       ar->v2d.tot.xmin= 0.0f;
+       ar->v2d.tot.ymin= (float)scene->r.sfra - 10.0f;
+       ar->v2d.tot.xmax= (float)scene->r.efra;
+       ar->v2d.tot.ymax= 10.0f;
+
+       ar->v2d.cur= ar->v2d.tot;
+
+       ar->v2d.min[0]= FLT_MIN;
+       ar->v2d.min[1]= FLT_MIN;
+
+       ar->v2d.max[0]= MAXFRAMEF;
+       ar->v2d.max[1]= FLT_MAX;
+
+       ar->v2d.scroll= (V2D_SCROLL_BOTTOM|V2D_SCROLL_SCALE_HORIZONTAL);
+       ar->v2d.scroll |= (V2D_SCROLL_LEFT|V2D_SCROLL_SCALE_VERTICAL);
+
+       ar->v2d.keeptot= 0;
+}
+
+static ARegion *clip_has_preview_region(const bContext *C, ScrArea *sa)
+{
+       ARegion *ar, *arnew;
+
+       ar= BKE_area_find_region_type(sa, RGN_TYPE_PREVIEW);
+       if(ar)
+               return ar;
+
+       /* add subdiv level; after header */
+       ar= BKE_area_find_region_type(sa, RGN_TYPE_WINDOW);
+
+       /* is error! */
+       if(ar==NULL)
+               return NULL;
+
+       arnew= MEM_callocN(sizeof(ARegion), "clip preview region");
+
+       BLI_insertlinkbefore(&sa->regionbase, ar, arnew);
+       init_preview_region(C, arnew);
+
+       return arnew;
+}
+
 static void clip_scopes_tag_refresh(ScrArea *sa)
 {
        SpaceClip *sc= (SpaceClip *)sa->spacedata.first;
        ARegion *ar;
 
+       if(sc->mode!=SC_MODE_TRACKING)
+               return;
+
        /* only while proeprties are visible */
        for (ar=sa->regionbase.first; ar; ar=ar->next) {
                if (ar->regiontype == RGN_TYPE_UI && ar->flag & RGN_FLAG_HIDDEN)
@@ -98,14 +150,14 @@ static void clip_stabilization_tag_refresh(ScrArea *sa)
 
 /* ******************** default callbacks for clip space ***************** */
 
-static SpaceLink *clip_new(const bContext *UNUSED(C))
+static SpaceLink *clip_new(const bContext *C)
 {
        ARegion *ar;
        SpaceClip *sc;
 
        sc= MEM_callocN(sizeof(SpaceClip), "initclip");
        sc->spacetype= SPACE_CLIP;
-       sc->flag= SC_SHOW_MARKER_PATTERN|SC_SHOW_TRACK_PATH|SC_SHOW_GPENCIL;
+       sc->flag= SC_SHOW_MARKER_PATTERN|SC_SHOW_TRACK_PATH|SC_SHOW_GPENCIL|SC_MANUAL_CALIBRATION|SC_SHOW_GRAPH_TRACKS|SC_SHOW_GRAPH_FRAMES;
        sc->zoom= 1.0f;
        sc->path_length= 20;
        sc->scopes.track_preview_height= 120;
@@ -118,7 +170,7 @@ static SpaceLink *clip_new(const bContext *UNUSED(C))
        ar->alignment= RGN_ALIGN_BOTTOM;
 
        /* tools view */
-       ar= MEM_callocN(sizeof(ARegion), "tools for logic");
+       ar= MEM_callocN(sizeof(ARegion), "tools for clip");
 
        BLI_addtail(&sc->regionbase, ar);
        ar->regiontype= RGN_TYPE_TOOLS;
@@ -132,12 +184,18 @@ static SpaceLink *clip_new(const bContext *UNUSED(C))
        ar->alignment= RGN_ALIGN_BOTTOM|RGN_SPLIT_PREV;
 
        /* properties view */
-       ar= MEM_callocN(sizeof(ARegion), "properties for logic");
+       ar= MEM_callocN(sizeof(ARegion), "properties for clip");
 
        BLI_addtail(&sc->regionbase, ar);
        ar->regiontype= RGN_TYPE_UI;
        ar->alignment= RGN_ALIGN_RIGHT;
 
+       /* preview view */
+       ar= MEM_callocN(sizeof(ARegion), "preview for clip");
+
+       BLI_addtail(&sc->regionbase, ar);
+       init_preview_region(C, ar);
+
        /* main area */
        ar= MEM_callocN(sizeof(ARegion), "main area for clip");
 
@@ -240,7 +298,6 @@ static void clip_operatortypes(void)
        /* ** clip_ops.c ** */
        WM_operatortype_append(CLIP_OT_open);
        WM_operatortype_append(CLIP_OT_reload);
-       // WM_operatortype_append(CLIP_OT_unlink);
        WM_operatortype_append(CLIP_OT_view_pan);
        WM_operatortype_append(CLIP_OT_view_zoom);
        WM_operatortype_append(CLIP_OT_view_zoom_in);
@@ -250,6 +307,7 @@ static void clip_operatortypes(void)
        WM_operatortype_append(CLIP_OT_view_selected);
        WM_operatortype_append(CLIP_OT_change_frame);
        WM_operatortype_append(CLIP_OT_rebuild_proxy);
+       WM_operatortype_append(CLIP_OT_mode_set);
 
        /* ** clip_toolbar.c ** */
        WM_operatortype_append(CLIP_OT_tools);
@@ -281,7 +339,7 @@ static void clip_operatortypes(void)
 
        /* solving */
        WM_operatortype_append(CLIP_OT_solve_camera);
-       WM_operatortype_append(CLIP_OT_clear_reconstruction);
+       WM_operatortype_append(CLIP_OT_clear_solution);
 
        WM_operatortype_append(CLIP_OT_disable_markers);
        WM_operatortype_append(CLIP_OT_hide_tracks);
@@ -293,6 +351,7 @@ static void clip_operatortypes(void)
        WM_operatortype_append(CLIP_OT_set_floor);
        WM_operatortype_append(CLIP_OT_set_axis);
        WM_operatortype_append(CLIP_OT_set_scale);
+       WM_operatortype_append(CLIP_OT_set_solution_scale);
 
        /* detect */
        WM_operatortype_append(CLIP_OT_detect_features);
@@ -301,6 +360,7 @@ static void clip_operatortypes(void)
        WM_operatortype_append(CLIP_OT_stabilize_2d_add);
        WM_operatortype_append(CLIP_OT_stabilize_2d_remove);
        WM_operatortype_append(CLIP_OT_stabilize_2d_select);
+       WM_operatortype_append(CLIP_OT_stabilize_2d_set_rotation);
 
        /* clean-up */
        WM_operatortype_append(CLIP_OT_clear_track_path);
@@ -308,6 +368,19 @@ static void clip_operatortypes(void)
        WM_operatortype_append(CLIP_OT_track_copy_color);
 
        WM_operatortype_append(CLIP_OT_clean_tracks);
+
+       /* graph editing */
+       WM_operatortype_append(CLIP_OT_graph_select);
+       WM_operatortype_append(CLIP_OT_graph_delete_curve);
+       WM_operatortype_append(CLIP_OT_graph_delete_knot);
+
+       /* object tracking */
+       WM_operatortype_append(CLIP_OT_tracking_object_new);
+       WM_operatortype_append(CLIP_OT_tracking_object_remove);
+
+       /* clipboard */
+       WM_operatortype_append(CLIP_OT_copy_tracks);
+       WM_operatortype_append(CLIP_OT_paste_tracks);
 }
 
 static void clip_keymap(struct wmKeyConfig *keyconf)
@@ -317,7 +390,7 @@ static void clip_keymap(struct wmKeyConfig *keyconf)
 
        /* ******** Global hotkeys avalaible for all regions ******** */
 
-       keymap= WM_keymap_find(keyconf, "Clip Globals", SPACE_CLIP, 0);
+       keymap= WM_keymap_find(keyconf, "Clip", SPACE_CLIP, 0);
 
        WM_keymap_add_item(keymap, "CLIP_OT_open", OKEY, KM_PRESS, KM_ALT, 0);
 
@@ -334,6 +407,15 @@ static void clip_keymap(struct wmKeyConfig *keyconf)
        RNA_boolean_set(kmi->ptr, "backwards", 1);
        RNA_boolean_set(kmi->ptr, "sequence", 1);
 
+       /* mode */
+       kmi= WM_keymap_add_item(keymap, "CLIP_OT_mode_set", TABKEY, KM_PRESS, 0, 0);
+       RNA_enum_set(kmi->ptr, "mode", SC_MODE_RECONSTRUCTION);
+       RNA_boolean_set(kmi->ptr, "toggle", 1);
+
+       kmi= WM_keymap_add_item(keymap, "CLIP_OT_mode_set", TABKEY, KM_PRESS, KM_CTRL, 0);
+       RNA_enum_set(kmi->ptr, "mode", SC_MODE_DISTORTION);
+       RNA_boolean_set(kmi->ptr, "toggle", 1);
+
        /* ******** Hotkeys avalaible for main region only ******** */
 
        keymap= WM_keymap_find(keyconf, "Clip Editor", SPACE_CLIP, 0);
@@ -436,6 +518,24 @@ static void clip_keymap(struct wmKeyConfig *keyconf)
        RNA_string_set(kmi->ptr, "data_path", "space_data.use_mute_footage");
 
        transform_keymap_for_space(keyconf, keymap, SPACE_CLIP);
+
+       /* ******** Hotkeys avalaible for preview region only ******** */
+
+       keymap= WM_keymap_find(keyconf, "Clip Graph Editor", SPACE_CLIP, 0);
+
+       /* "timeline" */
+       WM_keymap_add_item(keymap, "CLIP_OT_change_frame", ACTIONMOUSE, KM_PRESS, 0, 0);
+
+       /* selection */
+       WM_keymap_add_item(keymap, "CLIP_OT_graph_select", SELECTMOUSE, KM_PRESS, 0, 0);
+       RNA_boolean_set(WM_keymap_add_item(keymap, "CLIP_OT_graph_select", SELECTMOUSE, KM_PRESS, KM_SHIFT, 0)->ptr, "extend", 1);
+
+       /* delete */
+       WM_keymap_add_item(keymap, "CLIP_OT_graph_delete_curve", DELKEY, KM_PRESS, 0, 0);
+       WM_keymap_add_item(keymap, "CLIP_OT_graph_delete_curve", XKEY, KM_PRESS, 0, 0);
+
+       WM_keymap_add_item(keymap, "CLIP_OT_graph_delete_knot", DELKEY, KM_PRESS, KM_SHIFT, 0);
+       WM_keymap_add_item(keymap, "CLIP_OT_graph_delete_knot", XKEY, KM_PRESS, KM_SHIFT, 0);
 }
 
 const char *clip_context_dir[]= {"edit_movieclip", NULL};
@@ -456,9 +556,54 @@ static int clip_context(const bContext *C, const char *member, bContextDataResul
        return 0;
 }
 
-static void clip_refresh(const bContext *C, ScrArea *UNUSED(sa))
+static void clip_refresh(const bContext *C, ScrArea *sa)
 {
-       SpaceClip *sc= CTX_wm_space_clip(C);
+       wmWindowManager *wm= CTX_wm_manager(C);
+       wmWindow *window= CTX_wm_window(C);
+       SpaceClip *sc= (SpaceClip *)sa->spacedata.first;
+       ARegion *ar_main= BKE_area_find_region_type(sa, RGN_TYPE_WINDOW);
+       ARegion *ar_preview= clip_has_preview_region(C, sa);
+       int view_changed= 0;
+
+       switch (sc->view) {
+               case SC_VIEW_CLIP:
+                       if (ar_preview && !(ar_preview->flag & RGN_FLAG_HIDDEN)) {
+                               ar_preview->flag |= RGN_FLAG_HIDDEN;
+                               ar_preview->v2d.flag &= ~V2D_IS_INITIALISED;
+                               WM_event_remove_handlers((bContext*)C, &ar_preview->handlers);
+                               view_changed= 1;
+                       }
+                       if (ar_main && ar_main->alignment != RGN_ALIGN_NONE) {
+                               ar_main->alignment= RGN_ALIGN_NONE;
+                               view_changed= 1;
+                       }
+                       if (ar_preview && ar_preview->alignment != RGN_ALIGN_NONE) {
+                               ar_preview->alignment= RGN_ALIGN_NONE;
+                               view_changed= 1;
+                       }
+                       break;
+               case SC_VIEW_GRAPH:
+                       if (ar_preview && (ar_preview->flag & RGN_FLAG_HIDDEN)) {
+                               ar_preview->flag &= ~RGN_FLAG_HIDDEN;
+                               ar_preview->v2d.flag &= ~V2D_IS_INITIALISED;
+                               ar_preview->v2d.cur = ar_preview->v2d.tot;
+                               view_changed= 1;
+                       }
+                       if (ar_main && ar_main->alignment != RGN_ALIGN_NONE) {
+                               ar_main->alignment= RGN_ALIGN_NONE;
+                               view_changed= 1;
+                       }
+                       if (ar_preview && ar_preview->alignment != RGN_ALIGN_TOP) {
+                               ar_preview->alignment= RGN_ALIGN_TOP;
+                               view_changed= 1;
+                       }
+                       break;
+       }
+
+       if(view_changed) {
+               ED_area_initialize(wm, window, sa);
+               ED_area_tag_redraw(sa);
+       }
 
        BKE_movieclip_user_set_frame(&sc->user, CTX_data_scene(C)->r.cfra);
 }
@@ -522,7 +667,7 @@ static void clip_main_area_init(wmWindowManager *wm, ARegion *ar)
        UI_view2d_region_reinit(&ar->v2d, V2D_COMMONVIEW_STANDARD, ar->winx, ar->winy);
 
        /* own keymap */
-       keymap= WM_keymap_find(wm->defaultconf, "Clip Globals", SPACE_CLIP, 0);
+       keymap= WM_keymap_find(wm->defaultconf, "Clip", SPACE_CLIP, 0);
        WM_event_add_keymap_handler_bb(&ar->handlers, keymap, &ar->v2d.mask, &ar->winrct);
 
        keymap= WM_keymap_find(wm->defaultconf, "Clip Editor", SPACE_CLIP, 0);
@@ -545,10 +690,13 @@ static void clip_main_area_draw(const bContext *C, ARegion *ar)
                ImBuf *tmpibuf= NULL;
 
                if(clip && clip->tracking.stabilization.flag&TRACKING_2D_STABILIZATION) {
-                       tmpibuf= ED_space_clip_acquire_stable_buffer(sc, NULL, NULL);
+                       tmpibuf= ED_space_clip_get_stable_buffer(sc, NULL, NULL, NULL);
                }
 
-               ED_clip_view_selection(sc, ar, 0);
+               if(ED_clip_view_selection(sc, ar, 0)) {
+                       sc->xof+= sc->xlockof;
+                       sc->yof+= sc->ylockof;
+               }
 
                if(tmpibuf)
                        IMB_freeImBuf(tmpibuf);
@@ -561,16 +709,16 @@ static void clip_main_area_draw(const bContext *C, ARegion *ar)
        /* data... */
        movieclip_main_area_set_view2d(sc, ar);
 
-       draw_clip_main(sc, ar, scene);
+       clip_draw_main(sc, ar, scene);
 
        /* Grease Pencil */
-       draw_clip_grease_pencil((bContext *)C, 1);
+       clip_draw_grease_pencil((bContext *)C, 1);
 
        /* reset view matrix */
        UI_view2d_view_restore(C);
 
        /* draw Grease Pencil - screen space only */
-       draw_clip_grease_pencil((bContext *)C, 0);
+       clip_draw_grease_pencil((bContext *)C, 0);
 }
 
 static void clip_main_area_listener(ARegion *ar, wmNotifier *wmn)
@@ -584,6 +732,52 @@ static void clip_main_area_listener(ARegion *ar, wmNotifier *wmn)
        }
 }
 
+/****************** preview region ******************/
+
+static void clip_preview_area_init(wmWindowManager *wm, ARegion *ar)
+{
+       wmKeyMap *keymap;
+
+       UI_view2d_region_reinit(&ar->v2d, V2D_COMMONVIEW_CUSTOM, ar->winx, ar->winy);
+
+       /* own keymap */
+       keymap= WM_keymap_find(wm->defaultconf, "Clip", SPACE_CLIP, 0);
+       WM_event_add_keymap_handler_bb(&ar->handlers, keymap, &ar->v2d.mask, &ar->winrct);
+
+       keymap= WM_keymap_find(wm->defaultconf, "Clip Graph Editor", SPACE_CLIP, 0);
+       WM_event_add_keymap_handler_bb(&ar->handlers, keymap, &ar->v2d.mask, &ar->winrct);
+}
+
+static void clip_preview_area_draw(const bContext *C, ARegion *ar)
+{
+       View2D *v2d= &ar->v2d;
+       View2DScrollers *scrollers;
+       SpaceClip *sc= CTX_wm_space_clip(C);
+       Scene *scene= CTX_data_scene(C);
+       short unitx= V2D_UNIT_FRAMESCALE, unity= V2D_UNIT_VALUES;
+
+       /* clear and setup matrix */
+       UI_ThemeClearColor(TH_BACK);
+       glClear(GL_COLOR_BUFFER_BIT);
+
+       UI_view2d_view_ortho(v2d);
+
+       /* data... */
+       clip_draw_graph(sc, ar, scene);
+
+       /* reset view matrix */
+       UI_view2d_view_restore(C);
+
+       /* scrollers */
+       scrollers= UI_view2d_scrollers_calc(C, v2d, unitx, V2D_GRID_NOCLAMP, unity, V2D_GRID_NOCLAMP);
+       UI_view2d_scrollers_draw(C, v2d, scrollers);
+       UI_view2d_scrollers_free(scrollers);
+}
+
+static void clip_preview_area_listener(ARegion *UNUSED(ar), wmNotifier *UNUSED(wmn))
+{
+}
+
 /****************** header region ******************/
 
 /* add handlers, stuff you only do once or on area/region changes */
@@ -625,7 +819,11 @@ static void clip_props_area_listener(ARegion *ar, wmNotifier *wmn)
                                ED_region_tag_redraw(ar);
                        break;
                case NC_SPACE:
-                       if(wmn->data == ND_SPACE_VIEW3D)
+                       if(wmn->data == ND_SPACE_CLIP)
+                               ED_region_tag_redraw(ar);
+                       break;
+               case NC_SCREEN:
+                       if(wmn->data == ND_GPENCIL)
                                ED_region_tag_redraw(ar);
                        break;
        }
@@ -640,7 +838,7 @@ static void clip_properties_area_init(wmWindowManager *wm, ARegion *ar)
 
        ED_region_panels_init(wm, ar);
 
-       keymap= WM_keymap_find(wm->defaultconf, "Clip Globals", SPACE_CLIP, 0);
+       keymap= WM_keymap_find(wm->defaultconf, "Clip", SPACE_CLIP, 0);
        WM_event_add_keymap_handler(&ar->handlers, keymap);
 }
 
@@ -699,6 +897,17 @@ void ED_spacetype_clip(void)
 
        BLI_addhead(&st->regiontypes, art);
 
+       /* preview */
+       art= MEM_callocN(sizeof(ARegionType), "spacetype clip region preview");
+       art->regionid = RGN_TYPE_PREVIEW;
+       art->prefsizey = 240;
+       art->init= clip_preview_area_init;
+       art->draw= clip_preview_area_draw;
+       art->listener= clip_preview_area_listener;
+       art->keymapflag= ED_KEYMAP_FRAMES|ED_KEYMAP_UI|ED_KEYMAP_VIEW2D;
+
+       BLI_addhead(&st->regiontypes, art);
+
        /* regions: properties */
        art= MEM_callocN(sizeof(ARegionType), "spacetype clip region properties");
        art->regionid= RGN_TYPE_UI;
@@ -715,6 +924,7 @@ void ED_spacetype_clip(void)
        art->regionid= RGN_TYPE_TOOLS;
        art->prefsizex= UI_COMPACT_PANEL_WIDTH;
        art->keymapflag= ED_KEYMAP_FRAMES|ED_KEYMAP_UI;
+       art->listener= clip_props_area_listener;
        art->init= clip_tools_area_init;
        art->draw= clip_tools_area_draw;