Sergey Sharybin
a260d1cd69
It didn't work correctly when there in no continuously solved camera segment (aka there is a single isolated frame with solved camera). Basically, don't start drawing curve segment until it's known there is enough points for at least one segment. On user level it seemed to be fine, but it was assert failure in debug builds.
296 lines
8.7 KiB
C
296 lines
8.7 KiB
C
/*
|
|
* This program is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU General Public License
|
|
* as published by the Free Software Foundation; either version 2
|
|
* of the License, or (at your option) any later version.
|
|
*
|
|
* This program 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 General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; if not, write to the Free Software Foundation,
|
|
* Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
*
|
|
* The Original Code is Copyright (C) 2011 Blender Foundation.
|
|
* All rights reserved.
|
|
*/
|
|
|
|
/** \file
|
|
* \ingroup spclip
|
|
*/
|
|
|
|
#include "DNA_movieclip_types.h"
|
|
#include "DNA_scene_types.h"
|
|
|
|
#include "BLI_math.h"
|
|
#include "BLI_utildefines.h"
|
|
|
|
#include "BKE_context.h"
|
|
#include "BKE_movieclip.h"
|
|
#include "BKE_tracking.h"
|
|
|
|
#include "ED_clip.h"
|
|
#include "ED_screen.h"
|
|
|
|
#include "GPU_immediate.h"
|
|
#include "GPU_immediate_util.h"
|
|
#include "GPU_matrix.h"
|
|
#include "GPU_state.h"
|
|
|
|
#include "WM_types.h"
|
|
|
|
#include "UI_interface.h"
|
|
#include "UI_resources.h"
|
|
#include "UI_view2d.h"
|
|
|
|
#include "clip_intern.h" // own include
|
|
|
|
typedef struct TrackMotionCurveUserData {
|
|
SpaceClip *sc;
|
|
MovieTrackingTrack *act_track;
|
|
bool sel;
|
|
float xscale, yscale, hsize;
|
|
uint pos;
|
|
} TrackMotionCurveUserData;
|
|
|
|
static void tracking_segment_point_cb(void *userdata,
|
|
MovieTrackingTrack *UNUSED(track),
|
|
MovieTrackingMarker *UNUSED(marker),
|
|
eClipCurveValueSource value_source,
|
|
int scene_framenr,
|
|
float val)
|
|
{
|
|
TrackMotionCurveUserData *data = (TrackMotionCurveUserData *)userdata;
|
|
|
|
if (!clip_graph_value_visible(data->sc, value_source)) {
|
|
return;
|
|
}
|
|
|
|
immVertex2f(data->pos, scene_framenr, val);
|
|
}
|
|
|
|
static void tracking_segment_start_cb(void *userdata,
|
|
MovieTrackingTrack *track,
|
|
eClipCurveValueSource value_source,
|
|
bool is_point)
|
|
{
|
|
TrackMotionCurveUserData *data = (TrackMotionCurveUserData *)userdata;
|
|
SpaceClip *sc = data->sc;
|
|
float col[4] = {0.0f, 0.0f, 0.0f, 0.0f};
|
|
|
|
if (!clip_graph_value_visible(sc, value_source)) {
|
|
return;
|
|
}
|
|
|
|
switch (value_source) {
|
|
case CLIP_VALUE_SOURCE_SPEED_X:
|
|
col[0] = 1.0f;
|
|
break;
|
|
case CLIP_VALUE_SOURCE_SPEED_Y:
|
|
col[1] = 1.0f;
|
|
break;
|
|
case CLIP_VALUE_SOURCE_REPROJECTION_ERROR:
|
|
col[2] = 1.0f;
|
|
break;
|
|
}
|
|
|
|
if (track == data->act_track) {
|
|
col[3] = 1.0f;
|
|
GPU_line_width(2.0f);
|
|
}
|
|
else {
|
|
col[3] = 0.5f;
|
|
GPU_line_width(1.0f);
|
|
}
|
|
|
|
immUniformColor4fv(col);
|
|
|
|
if (is_point) {
|
|
immBeginAtMost(GPU_PRIM_POINTS, 1);
|
|
}
|
|
else {
|
|
/* Graph can be composed of smaller segments, if any marker is disabled */
|
|
immBeginAtMost(GPU_PRIM_LINE_STRIP, track->markersnr);
|
|
}
|
|
}
|
|
|
|
static void tracking_segment_end_cb(void *userdata, eClipCurveValueSource value_source)
|
|
{
|
|
TrackMotionCurveUserData *data = (TrackMotionCurveUserData *)userdata;
|
|
SpaceClip *sc = data->sc;
|
|
if (!clip_graph_value_visible(sc, value_source)) {
|
|
return;
|
|
}
|
|
immEnd();
|
|
}
|
|
|
|
static void tracking_segment_knot_cb(void *userdata,
|
|
MovieTrackingTrack *track,
|
|
MovieTrackingMarker *marker,
|
|
eClipCurveValueSource value_source,
|
|
int scene_framenr,
|
|
float val)
|
|
{
|
|
TrackMotionCurveUserData *data = (TrackMotionCurveUserData *)userdata;
|
|
int sel = 0, sel_flag;
|
|
|
|
if (track != data->act_track) {
|
|
return;
|
|
}
|
|
if (!ELEM(value_source, CLIP_VALUE_SOURCE_SPEED_X, CLIP_VALUE_SOURCE_SPEED_Y)) {
|
|
return;
|
|
}
|
|
|
|
sel_flag = value_source == CLIP_VALUE_SOURCE_SPEED_X ? MARKER_GRAPH_SEL_X : MARKER_GRAPH_SEL_Y;
|
|
sel = (marker->flag & sel_flag) ? 1 : 0;
|
|
|
|
if (sel == data->sel) {
|
|
immUniformThemeColor(sel ? TH_HANDLE_VERTEX_SELECT : TH_HANDLE_VERTEX);
|
|
|
|
GPU_matrix_push();
|
|
GPU_matrix_translate_2f(scene_framenr, val);
|
|
GPU_matrix_scale_2f(1.0f / data->xscale * data->hsize, 1.0f / data->yscale * data->hsize);
|
|
|
|
imm_draw_circle_wire_2d(data->pos, 0, 0, 0.7, 8);
|
|
|
|
GPU_matrix_pop();
|
|
}
|
|
}
|
|
|
|
static void draw_tracks_motion_and_error_curves(View2D *v2d, SpaceClip *sc, uint pos)
|
|
{
|
|
MovieClip *clip = ED_space_clip_get_clip(sc);
|
|
MovieTracking *tracking = &clip->tracking;
|
|
MovieTrackingTrack *act_track = BKE_tracking_track_get_active(tracking);
|
|
const bool draw_knots = (sc->flag & SC_SHOW_GRAPH_TRACKS_MOTION) != 0;
|
|
|
|
int width, height;
|
|
BKE_movieclip_get_size(clip, &sc->user, &width, &height);
|
|
if (!width || !height) {
|
|
return;
|
|
}
|
|
|
|
TrackMotionCurveUserData userdata;
|
|
userdata.sc = sc;
|
|
userdata.hsize = UI_GetThemeValuef(TH_HANDLE_VERTEX_SIZE);
|
|
userdata.sel = false;
|
|
userdata.act_track = act_track;
|
|
userdata.pos = pos;
|
|
|
|
/* Non-selected knot handles. */
|
|
if (draw_knots) {
|
|
UI_view2d_scale_get(v2d, &userdata.xscale, &userdata.yscale);
|
|
clip_graph_tracking_values_iterate(sc,
|
|
(sc->flag & SC_SHOW_GRAPH_SEL_ONLY) != 0,
|
|
(sc->flag & SC_SHOW_GRAPH_HIDDEN) != 0,
|
|
&userdata,
|
|
tracking_segment_knot_cb,
|
|
NULL,
|
|
NULL);
|
|
}
|
|
|
|
/* Draw graph lines. */
|
|
GPU_blend(true);
|
|
clip_graph_tracking_values_iterate(sc,
|
|
(sc->flag & SC_SHOW_GRAPH_SEL_ONLY) != 0,
|
|
(sc->flag & SC_SHOW_GRAPH_HIDDEN) != 0,
|
|
&userdata,
|
|
tracking_segment_point_cb,
|
|
tracking_segment_start_cb,
|
|
tracking_segment_end_cb);
|
|
GPU_blend(false);
|
|
|
|
/* Selected knot handles on top of curves. */
|
|
if (draw_knots) {
|
|
userdata.sel = true;
|
|
clip_graph_tracking_values_iterate(sc,
|
|
(sc->flag & SC_SHOW_GRAPH_SEL_ONLY) != 0,
|
|
(sc->flag & SC_SHOW_GRAPH_HIDDEN) != 0,
|
|
&userdata,
|
|
tracking_segment_knot_cb,
|
|
NULL,
|
|
NULL);
|
|
}
|
|
}
|
|
|
|
static void draw_frame_curves(SpaceClip *sc, uint pos)
|
|
{
|
|
MovieClip *clip = ED_space_clip_get_clip(sc);
|
|
MovieTracking *tracking = &clip->tracking;
|
|
MovieTrackingReconstruction *reconstruction = BKE_tracking_get_active_reconstruction(tracking);
|
|
|
|
int previous_frame;
|
|
float previous_error;
|
|
bool have_previous_point = false;
|
|
|
|
/* Indicates whether immBegin() was called. */
|
|
bool is_lines_segment_open = false;
|
|
|
|
immUniformColor3f(0.0f, 0.0f, 1.0f);
|
|
|
|
for (int i = 0; i < reconstruction->camnr; i++) {
|
|
MovieReconstructedCamera *camera = &reconstruction->cameras[i];
|
|
|
|
const int current_frame = BKE_movieclip_remap_clip_to_scene_frame(clip, camera->framenr);
|
|
const float current_error = camera->error;
|
|
|
|
if (have_previous_point && current_frame != previous_frame + 1) {
|
|
if (is_lines_segment_open) {
|
|
immEnd();
|
|
is_lines_segment_open = false;
|
|
}
|
|
have_previous_point = false;
|
|
}
|
|
|
|
if (have_previous_point) {
|
|
if (!is_lines_segment_open) {
|
|
immBeginAtMost(GPU_PRIM_LINE_STRIP, reconstruction->camnr);
|
|
is_lines_segment_open = true;
|
|
|
|
immVertex2f(pos, previous_frame, previous_error);
|
|
}
|
|
immVertex2f(pos, current_frame, current_error);
|
|
}
|
|
|
|
previous_frame = current_frame;
|
|
previous_error = current_error;
|
|
have_previous_point = true;
|
|
}
|
|
|
|
if (is_lines_segment_open) {
|
|
immEnd();
|
|
}
|
|
}
|
|
|
|
void clip_draw_graph(SpaceClip *sc, ARegion *region, Scene *scene)
|
|
{
|
|
MovieClip *clip = ED_space_clip_get_clip(sc);
|
|
View2D *v2d = ®ion->v2d;
|
|
|
|
/* grid */
|
|
UI_view2d_draw_lines_x__values(v2d);
|
|
UI_view2d_draw_lines_y__values(v2d);
|
|
|
|
if (clip) {
|
|
uint pos = GPU_vertformat_attr_add(immVertexFormat(), "pos", GPU_COMP_F32, 2, GPU_FETCH_FLOAT);
|
|
immBindBuiltinProgram(GPU_SHADER_2D_UNIFORM_COLOR);
|
|
|
|
GPU_point_size(3.0f);
|
|
|
|
if (sc->flag & (SC_SHOW_GRAPH_TRACKS_MOTION | SC_SHOW_GRAPH_TRACKS_ERROR)) {
|
|
draw_tracks_motion_and_error_curves(v2d, sc, pos);
|
|
}
|
|
|
|
if (sc->flag & SC_SHOW_GRAPH_FRAMES) {
|
|
draw_frame_curves(sc, pos);
|
|
}
|
|
|
|
immUnbindProgram();
|
|
}
|
|
|
|
/* frame range */
|
|
clip_draw_sfra_efra(v2d, scene);
|
|
}
|