1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
|
/* SPDX-License-Identifier: GPL-2.0-or-later
* Copyright 2011 Blender Foundation. */
#include "COM_MovieClipAttributeOperation.h"
#include "BKE_movieclip.h"
#include "BKE_tracking.h"
namespace blender::compositor {
MovieClipAttributeOperation::MovieClipAttributeOperation()
{
this->add_output_socket(DataType::Value);
framenumber_ = 0;
attribute_ = MCA_X;
invert_ = false;
needs_canvas_to_get_constant_ = true;
is_value_calculated_ = false;
stabilization_resolution_socket_ = nullptr;
}
void MovieClipAttributeOperation::init_execution()
{
if (!is_value_calculated_) {
calc_value();
}
}
void MovieClipAttributeOperation::calc_value()
{
BLI_assert(this->get_flags().is_canvas_set);
is_value_calculated_ = true;
if (clip_ == nullptr) {
return;
}
float loc[2], scale, angle;
loc[0] = 0.0f;
loc[1] = 0.0f;
scale = 1.0f;
angle = 0.0f;
int clip_framenr = BKE_movieclip_remap_scene_to_clip_frame(clip_, framenumber_);
NodeOperation &stabilization_operation =
stabilization_resolution_socket_ ?
stabilization_resolution_socket_->get_link()->get_operation() :
*this;
BKE_tracking_stabilization_data_get(clip_,
clip_framenr,
stabilization_operation.get_width(),
stabilization_operation.get_height(),
loc,
&scale,
&angle);
switch (attribute_) {
case MCA_SCALE:
value_ = scale;
break;
case MCA_ANGLE:
value_ = angle;
break;
case MCA_X:
value_ = loc[0];
break;
case MCA_Y:
value_ = loc[1];
break;
}
if (invert_) {
if (attribute_ != MCA_SCALE) {
value_ = -value_;
}
else {
value_ = 1.0f / value_;
}
}
}
void MovieClipAttributeOperation::execute_pixel_sampled(float output[4],
float /*x*/,
float /*y*/,
PixelSampler /*sampler*/)
{
output[0] = value_;
}
void MovieClipAttributeOperation::determine_canvas(const rcti &preferred_area, rcti &r_area)
{
r_area = preferred_area;
}
const float *MovieClipAttributeOperation::get_constant_elem()
{
if (!is_value_calculated_) {
calc_value();
}
return &value_;
}
} // namespace blender::compositor
|