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
|
/* SPDX-License-Identifier: GPL-2.0-or-later
* Copyright 2013 Blender Foundation. */
#pragma once
#include <string.h>
#include "COM_MultiThreadedOperation.h"
#include "DNA_movieclip_types.h"
#include "DNA_tracking_types.h"
#include "BLI_listbase.h"
#include "BLI_string.h"
namespace blender::compositor {
#define PLANE_DISTORT_MAX_SAMPLES 64
class PlaneDistortBaseOperation : public MultiThreadedOperation {
protected:
struct MotionSample {
float frame_space_corners[4][2]; /* Corners coordinates in pixel space. */
float perspective_matrix[3][3];
};
MotionSample samples_[PLANE_DISTORT_MAX_SAMPLES];
int motion_blur_samples_;
float motion_blur_shutter_;
public:
PlaneDistortBaseOperation();
void set_motion_blur_samples(int samples)
{
BLI_assert(samples <= PLANE_DISTORT_MAX_SAMPLES);
motion_blur_samples_ = samples;
}
void set_motion_blur_shutter(float shutter)
{
motion_blur_shutter_ = shutter;
}
virtual void calculate_corners(const float corners[4][2], bool normalized, int sample);
private:
friend class PlaneTrackCommon;
};
class PlaneDistortWarpImageOperation : public PlaneDistortBaseOperation {
protected:
SocketReader *pixel_reader_;
public:
PlaneDistortWarpImageOperation();
void calculate_corners(const float corners[4][2], bool normalized, int sample) override;
void init_execution() override;
void deinit_execution() override;
void execute_pixel_sampled(float output[4], float x, float y, PixelSampler sampler) override;
bool determine_depending_area_of_interest(rcti *input,
ReadBufferOperation *read_operation,
rcti *output) override;
void get_area_of_interest(int input_idx, const rcti &output_area, rcti &r_input_area) override;
void update_memory_buffer_partial(MemoryBuffer *output,
const rcti &area,
Span<MemoryBuffer *> inputs) override;
};
class PlaneDistortMaskOperation : public PlaneDistortBaseOperation {
protected:
int osa_;
float jitter_[32][2];
public:
PlaneDistortMaskOperation();
void init_execution() override;
void execute_pixel_sampled(float output[4], float x, float y, PixelSampler sampler) override;
void update_memory_buffer_partial(MemoryBuffer *output,
const rcti &area,
Span<MemoryBuffer *> inputs) override;
private:
int get_jitter_samples_inside_count(int x, int y, MotionSample &sample_data);
};
} // namespace blender::compositor
|