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
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
|
/* SPDX-License-Identifier: GPL-2.0-or-later
* Copyright 2011 Blender Foundation. */
#pragma once
#include "COM_MultiThreadedOperation.h"
#include "BLI_path_util.h"
#include "BLI_rect.h"
#include "DNA_color_types.h"
#include "IMB_openexr.h"
namespace blender::compositor {
/* Writes the image to a single-layer file. */
class OutputSingleLayerOperation : public MultiThreadedOperation {
protected:
const RenderData *rd_;
const bNodeTree *tree_;
ImageFormatData format_;
char path_[FILE_MAX];
float *output_buffer_;
DataType datatype_;
SocketReader *image_input_;
const char *view_name_;
bool save_as_render_;
public:
OutputSingleLayerOperation(const Scene *scene,
const RenderData *rd,
const bNodeTree *tree,
DataType datatype,
ImageFormatData *format,
const char *path,
const char *view_name,
bool save_as_render);
void execute_region(rcti *rect, unsigned int tile_number) override;
bool is_output_operation(bool /*rendering*/) const override
{
return true;
}
void init_execution() override;
void deinit_execution() override;
eCompositorPriority get_render_priority() const override
{
return eCompositorPriority::Low;
}
void update_memory_buffer_partial(MemoryBuffer *output,
const rcti &area,
Span<MemoryBuffer *> inputs) override;
};
/* extra info for OpenEXR layers */
struct OutputOpenExrLayer {
OutputOpenExrLayer(const char *name, DataType datatype, bool use_layer);
char name[EXR_TOT_MAXNAME - 2];
DataType datatype;
bool use_layer;
/* internals */
float *output_buffer;
SocketReader *image_input;
};
/* Writes inputs into OpenEXR multilayer channels. */
class OutputOpenExrMultiLayerOperation : public MultiThreadedOperation {
protected:
const Scene *scene_;
const RenderData *rd_;
const bNodeTree *tree_;
char path_[FILE_MAX];
char exr_codec_;
bool exr_half_float_;
Vector<OutputOpenExrLayer> layers_;
const char *view_name_;
StampData *create_stamp_data() const;
public:
OutputOpenExrMultiLayerOperation(const Scene *scene,
const RenderData *rd,
const bNodeTree *tree,
const char *path,
char exr_codec,
bool exr_half_float,
const char *view_name);
void add_layer(const char *name, DataType datatype, bool use_layer);
void execute_region(rcti *rect, unsigned int tile_number) override;
bool is_output_operation(bool /*rendering*/) const override
{
return true;
}
void init_execution() override;
void deinit_execution() override;
eCompositorPriority get_render_priority() const override
{
return eCompositorPriority::Low;
}
void update_memory_buffer_partial(MemoryBuffer *output,
const rcti &area,
Span<MemoryBuffer *> inputs) override;
};
void add_exr_channels(void *exrhandle,
const char *layer_name,
const DataType datatype,
const char *view_name,
size_t width,
bool use_half_float,
float *buf);
void free_exr_channels(void *exrhandle,
const RenderData *rd,
const char *layer_name,
const DataType datatype);
int get_datatype_size(DataType datatype);
} // namespace blender::compositor
|