Fix BKE_image_walk_all_users not including all image users.
[blender.git] / source / blender / blenkernel / BKE_image.h
1 /*
2  * This program is free software; you can redistribute it and/or
3  * modify it under the terms of the GNU General Public License
4  * as published by the Free Software Foundation; either version 2
5  * of the License, or (at your option) any later version.
6  *
7  * This program is distributed in the hope that it will be useful,
8  * but WITHOUT ANY WARRANTY; without even the implied warranty of
9  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
10  * GNU General Public License for more details.
11  *
12  * You should have received a copy of the GNU General Public License
13  * along with this program; if not, write to the Free Software Foundation,
14  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
15  *
16  * The Original Code is Copyright (C) 2001-2002 by NaN Holding BV.
17  * All rights reserved.
18  */
19 #ifndef __BKE_IMAGE_H__
20 #define __BKE_IMAGE_H__
21
22 /** \file
23  * \ingroup bke
24  */
25
26 #ifdef __cplusplus
27 extern "C" {
28 #endif
29
30 struct Depsgraph;
31 struct ID;
32 struct ImBuf;
33 struct Image;
34 struct ImageFormatData;
35 struct ImagePool;
36 struct ImbFormatOptions;
37 struct Main;
38 struct Object;
39 struct RenderResult;
40 struct ReportList;
41 struct Scene;
42 struct StampData;
43 struct anim;
44
45 #define IMA_MAX_SPACE       64
46
47 void   BKE_images_init(void);
48 void   BKE_images_exit(void);
49
50 void    BKE_image_free_packedfiles(struct Image *image);
51 void    BKE_image_free_views(struct Image *image);
52 void    BKE_image_free_buffers(struct Image *image);
53 void    BKE_image_free_buffers_ex(struct Image *image, bool do_lock);
54 /* call from library */
55 void    BKE_image_free(struct Image *image);
56
57 void    BKE_image_init(struct Image *image);
58
59 typedef void (StampCallback)(void *data, const char *propname, char *propvalue, int len);
60
61 void    BKE_render_result_stamp_info(struct Scene *scene, struct Object *camera, struct RenderResult *rr, bool allocate_only);
62 /**
63  * Fills in the static stamp data (i.e. everything except things that can change per frame).
64  * The caller is responsible for freeing the allocated memory.
65  */
66 struct StampData *BKE_stamp_info_from_scene_static(struct Scene *scene);
67 bool    BKE_stamp_is_known_field(const char *field_name);
68 void    BKE_imbuf_stamp_info(struct RenderResult *rr, struct ImBuf *ibuf);
69 void    BKE_stamp_info_from_imbuf(struct RenderResult *rr, struct ImBuf *ibuf);
70 void    BKE_stamp_info_callback(void *data, struct StampData *stamp_data, StampCallback callback, bool noskip);
71 void    BKE_render_result_stamp_data(struct RenderResult *rr, const char *key, const char *value);
72 void    BKE_stamp_data_free(struct StampData *stamp_data);
73 void    BKE_image_stamp_buf(
74         struct Scene *scene, struct Object *camera, const struct StampData *stamp_data_template,
75         unsigned char *rect, float *rectf, int width, int height, int channels);
76 bool    BKE_imbuf_alpha_test(struct ImBuf *ibuf);
77 int     BKE_imbuf_write_stamp(
78         struct Scene *scene, struct RenderResult *rr, struct ImBuf *ibuf, const char *name,
79         const struct ImageFormatData *imf);
80 void    BKE_imbuf_write_prepare(struct ImBuf *ibuf, const struct ImageFormatData *imf);
81 int     BKE_imbuf_write(struct ImBuf *ibuf, const char *name, const struct ImageFormatData *imf);
82 int     BKE_imbuf_write_as(struct ImBuf *ibuf, const char *name, struct ImageFormatData *imf, const bool is_copy);
83 void    BKE_image_path_from_imformat(
84         char *string, const char *base, const char *relbase, int frame,
85         const struct ImageFormatData *im_format, const bool use_ext, const bool use_frames, const char *suffix);
86 void    BKE_image_path_from_imtype(
87         char *string, const char *base, const char *relbase, int frame,
88         const char imtype, const bool use_ext, const bool use_frames, const char *suffix);
89 int     BKE_image_path_ensure_ext_from_imformat(char *string, const struct ImageFormatData *im_format);
90 int     BKE_image_path_ensure_ext_from_imtype(char *string, const char imtype);
91 char    BKE_image_ftype_to_imtype(const int ftype, const struct ImbFormatOptions *options);
92 int     BKE_image_imtype_to_ftype(const char imtype, struct ImbFormatOptions *r_options);
93
94 bool    BKE_imtype_is_movie(const char imtype);
95 int     BKE_imtype_supports_zbuf(const char imtype);
96 int     BKE_imtype_supports_compress(const char imtype);
97 int     BKE_imtype_supports_quality(const char imtype);
98 int     BKE_imtype_requires_linear_float(const char imtype);
99 char    BKE_imtype_valid_channels(const char imtype, bool write_file);
100 char    BKE_imtype_valid_depths(const char imtype);
101
102 char    BKE_imtype_from_arg(const char *arg);
103
104 void    BKE_imformat_defaults(struct ImageFormatData *im_format);
105 void    BKE_imbuf_to_image_format(struct ImageFormatData *im_format, const struct ImBuf *imbuf);
106
107 struct anim *openanim(const char *name, int flags, int streamindex, char colorspace[IMA_MAX_SPACE]);
108 struct anim *openanim_noload(const char *name, int flags, int streamindex, char colorspace[IMA_MAX_SPACE]);
109
110 void    BKE_image_make_local(struct Main *bmain, struct Image *ima, const bool lib_local);
111
112 void    BKE_image_tag_time(struct Image *ima);
113
114 /* ********************************** NEW IMAGE API *********************** */
115
116 /* ImageUser is in Texture, in Nodes, Background Image, Image Window, .... */
117 /* should be used in conjunction with an ID * to Image. */
118 struct ImageUser;
119 struct RenderData;
120 struct RenderPass;
121 struct RenderResult;
122
123 /* ima->source; where image comes from */
124 #define IMA_SRC_CHECK       0
125 #define IMA_SRC_FILE        1
126 #define IMA_SRC_SEQUENCE    2
127 #define IMA_SRC_MOVIE       3
128 #define IMA_SRC_GENERATED   4
129 #define IMA_SRC_VIEWER      5
130
131 /* ima->type, how to handle/generate it */
132 #define IMA_TYPE_IMAGE      0
133 #define IMA_TYPE_MULTILAYER 1
134 /* generated */
135 #define IMA_TYPE_UV_TEST    2
136 /* viewers */
137 #define IMA_TYPE_R_RESULT   4
138 #define IMA_TYPE_COMPOSITE  5
139
140 enum {
141         IMA_GENTYPE_BLANK = 0,
142         IMA_GENTYPE_GRID = 1,
143         IMA_GENTYPE_GRID_COLOR = 2,
144 };
145
146 /* ima->ok */
147 #define IMA_OK              1
148 #define IMA_OK_LOADED       2
149
150 /* signals */
151 /* reload only frees, doesn't read until image_get_ibuf() called */
152 #define IMA_SIGNAL_RELOAD           0
153 #define IMA_SIGNAL_FREE             1
154 /* source changes, from image to sequence or movie, etc */
155 #define IMA_SIGNAL_SRC_CHANGE       5
156 /* image-user gets a new image, check settings */
157 #define IMA_SIGNAL_USER_NEW_IMAGE   6
158 #define IMA_SIGNAL_COLORMANAGE      7
159
160 #define IMA_CHAN_FLAG_BW    1
161 #define IMA_CHAN_FLAG_RGB   2
162 #define IMA_CHAN_FLAG_ALPHA 4
163
164 /* checks whether there's an image buffer for given image and user */
165 bool BKE_image_has_ibuf(struct Image *ima, struct ImageUser *iuser);
166
167 /* same as above, but can be used to retrieve images being rendered in
168  * a thread safe way, always call both acquire and release */
169 struct ImBuf *BKE_image_acquire_ibuf(struct Image *ima, struct ImageUser *iuser, void **r_lock);
170 void BKE_image_release_ibuf(struct Image *ima, struct ImBuf *ibuf, void *lock);
171
172 struct ImagePool *BKE_image_pool_new(void);
173 void BKE_image_pool_free(struct ImagePool *pool);
174 struct ImBuf *BKE_image_pool_acquire_ibuf(struct Image *ima, struct ImageUser *iuser, struct ImagePool *pool);
175 void BKE_image_pool_release_ibuf(struct Image *ima, struct ImBuf *ibuf, struct ImagePool *pool);
176
177 /* set an alpha mode based on file extension */
178 char  BKE_image_alpha_mode_from_extension_ex(const char *filepath);
179 void BKE_image_alpha_mode_from_extension(struct Image *image);
180
181 /* returns a new image or NULL if it can't load */
182 struct Image *BKE_image_load(struct Main *bmain, const char *filepath);
183 /* returns existing Image when filename/type is same (frame optional) */
184 struct Image *BKE_image_load_exists_ex(struct Main *bmain, const char *filepath, bool *r_exists);
185 struct Image *BKE_image_load_exists(struct Main *bmain, const char *filepath);
186
187 /* adds image, adds ibuf, generates color or pattern */
188 struct Image *BKE_image_add_generated(
189         struct Main *bmain, unsigned int width, unsigned int height, const char *name,
190         int depth, int floatbuf, short gen_type, const float color[4], const bool stereo3d);
191 /* adds image from imbuf, owns imbuf */
192 struct Image *BKE_image_add_from_imbuf(struct Main *bmain, struct ImBuf *ibuf, const char *name);
193
194 /* for reload, refresh, pack */
195 void BKE_image_init_imageuser(struct Image *ima, struct ImageUser *iuser);
196 void BKE_image_signal(struct Main *bmain, struct Image *ima, struct ImageUser *iuser, int signal);
197
198 void BKE_image_walk_id_all_users(struct ID *id, void *customdata,
199                                  void callback(struct Image *ima, struct ImageUser *iuser, void *customdata));
200 void BKE_image_walk_all_users(const struct Main *mainp, void *customdata,
201                               void callback(struct Image *ima, struct ImageUser *iuser, void *customdata));
202
203 /* ensures an Image exists for viewing nodes or render */
204 struct Image *BKE_image_verify_viewer(struct Main *bmain, int type, const char *name);
205 /* ensures the view node cache is compatible with the scene views */
206 void BKE_image_verify_viewer_views(const struct RenderData *rd, struct Image *ima, struct ImageUser *iuser);
207
208 /* called on frame change or before render */
209 void BKE_image_user_frame_calc(struct ImageUser *iuser, int cfra);
210 void BKE_image_user_check_frame_calc(struct ImageUser *iuser, int cfra);
211 int  BKE_image_user_frame_get(const struct ImageUser *iuser, int cfra, bool *r_is_in_range);
212 void BKE_image_user_file_path(struct ImageUser *iuser, struct Image *ima, char *path);
213 void BKE_image_update_frame(const struct Main *bmain, int cfra);
214
215 /* sets index offset for multilayer files */
216 struct RenderPass *BKE_image_multilayer_index(struct RenderResult *rr, struct ImageUser *iuser);
217
218 /* sets index offset for multiview files */
219 void BKE_image_multiview_index(struct Image *ima, struct ImageUser *iuser);
220
221 /* for multilayer images as well as for render-viewer */
222 bool BKE_image_is_multilayer(struct Image *ima);
223 bool BKE_image_is_multiview(struct Image *ima);
224 bool BKE_image_is_stereo(struct Image *ima);
225 struct RenderResult *BKE_image_acquire_renderresult(struct Scene *scene, struct Image *ima);
226 void BKE_image_release_renderresult(struct Scene *scene, struct Image *ima);
227
228 /* for multilayer images as well as for singlelayer */
229 bool BKE_image_is_openexr(struct Image *ima);
230
231 /* for multiple slot render, call this before render */
232 void BKE_image_backup_render(struct Scene *scene, struct Image *ima, bool free_current_slot);
233
234 /* for singlelayer openexr saving */
235 bool BKE_image_save_openexr_multiview(struct Image *ima, struct ImBuf *ibuf, const char *filepath, const int flags);
236
237 /* goes over all textures that use images */
238 void    BKE_image_free_all_textures(struct Main *bmain);
239
240 /* does one image! */
241 void    BKE_image_free_anim_ibufs(struct Image *ima, int except_frame);
242
243 /* does all images with type MOVIE or SEQUENCE */
244 void BKE_image_all_free_anim_ibufs(struct Main *bmain, int except_frame);
245
246 void BKE_image_memorypack(struct Image *ima);
247 void BKE_image_packfiles(struct ReportList *reports, struct Image *ima, const char *basepath);
248 void BKE_image_packfiles_from_mem(struct ReportList *reports, struct Image *ima, char *data, const size_t data_len);
249
250 /* prints memory statistics for images */
251 void BKE_image_print_memlist(struct Main *bmain);
252
253 /* empty image block, of similar type and filename */
254 void BKE_image_copy_data(struct Main *bmain, struct Image *ima_dst, const struct Image *ima_src, const int flag);
255 struct Image *BKE_image_copy(struct Main *bmain, const struct Image *ima);
256
257 /* merge source into dest, and free source */
258 void BKE_image_merge(struct Main *bmain, struct Image *dest, struct Image *source);
259
260 /* scale the image */
261 bool BKE_image_scale(struct Image *image, int width, int height);
262
263 /* check if texture has alpha (depth=32) */
264 bool BKE_image_has_alpha(struct Image *image);
265
266 /* check if texture has gpu texture code */
267 bool BKE_image_has_opengl_texture(struct Image *ima);
268
269 void BKE_image_get_size(struct Image *image, struct ImageUser *iuser, int *width, int *height);
270 void BKE_image_get_size_fl(struct Image *image, struct ImageUser *iuser, float size[2]);
271 void BKE_image_get_aspect(struct Image *image, float *aspx, float *aspy);
272
273 /* image_gen.c */
274 void BKE_image_buf_fill_color(unsigned char *rect, float *rect_float, int width, int height, const float color[4]);
275 void BKE_image_buf_fill_checker(unsigned char *rect, float *rect_float, int height, int width);
276 void BKE_image_buf_fill_checker_color(unsigned char *rect, float *rect_float, int height, int width);
277
278 /* Cycles hookup */
279 unsigned char *BKE_image_get_pixels_for_frame(struct Image *image, int frame);
280 float *BKE_image_get_float_pixels_for_frame(struct Image *image, int frame);
281
282 /* Guess offset for the first frame in the sequence */
283 int BKE_image_sequence_guess_offset(struct Image *image);
284 bool BKE_image_has_anim(struct Image *image);
285 bool BKE_image_has_packedfile(struct Image *image);
286 bool BKE_image_is_animated(struct Image *image);
287 bool BKE_image_is_dirty(struct Image *image);
288 void BKE_image_file_format_set(struct Image *image, int ftype, const struct ImbFormatOptions *options);
289 bool BKE_image_has_loaded_ibuf(struct Image *image);
290 struct ImBuf *BKE_image_get_ibuf_with_name(struct Image *image, const char *name);
291 struct ImBuf *BKE_image_get_first_ibuf(struct Image *image);
292
293 struct RenderSlot *BKE_image_add_renderslot(struct Image *ima, const char *name);
294 bool BKE_image_remove_renderslot(struct Image *ima, struct ImageUser *iuser, int slot);
295 struct RenderSlot *BKE_image_get_renderslot(struct Image *ima, int slot);
296 bool BKE_image_clear_renderslot(struct Image *ima, struct ImageUser *iuser, int slot);
297
298 #ifdef __cplusplus
299 }
300 #endif
301
302 #endif