Merge remote-tracking branch 'origin/master' into blender2.8
[blender.git] / source / blender / blenkernel / intern / library.c
1 /*
2  * ***** BEGIN GPL LICENSE BLOCK *****
3  *
4  * This program is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU General Public License
6  * as published by the Free Software Foundation; either version 2
7  * of the License, or (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program; if not, write to the Free Software Foundation,
16  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
17  *
18  * The Original Code is Copyright (C) 2001-2002 by NaN Holding BV.
19  * All rights reserved.
20  *
21  * The Original Code is: all of this file.
22  *
23  * Contributor(s): none yet.
24  *
25  * ***** END GPL LICENSE BLOCK *****
26  */
27
28 /** \file blender/blenkernel/intern/library.c
29  *  \ingroup bke
30  *
31  * Contains management of ID's and libraries
32  * allocate and free of all library data
33  */
34
35 #include <stdio.h>
36 #include <ctype.h>
37 #include <string.h>
38 #include <stdlib.h>
39 #include <stddef.h>
40 #include <assert.h>
41
42 #include "MEM_guardedalloc.h"
43
44 /* all types are needed here, in order to do memory operations */
45 #include "DNA_anim_types.h"
46 #include "DNA_armature_types.h"
47 #include "DNA_brush_types.h"
48 #include "DNA_cachefile_types.h"
49 #include "DNA_camera_types.h"
50 #include "DNA_group_types.h"
51 #include "DNA_gpencil_types.h"
52 #include "DNA_ipo_types.h"
53 #include "DNA_key_types.h"
54 #include "DNA_lamp_types.h"
55 #include "DNA_lattice_types.h"
56 #include "DNA_linestyle_types.h"
57 #include "DNA_material_types.h"
58 #include "DNA_mesh_types.h"
59 #include "DNA_meta_types.h"
60 #include "DNA_modifier_types.h"
61 #include "DNA_movieclip_types.h"
62 #include "DNA_mask_types.h"
63 #include "DNA_node_types.h"
64 #include "DNA_object_types.h"
65 #include "DNA_scene_types.h"
66 #include "DNA_screen_types.h"
67 #include "DNA_speaker_types.h"
68 #include "DNA_sound_types.h"
69 #include "DNA_text_types.h"
70 #include "DNA_vfont_types.h"
71 #include "DNA_windowmanager_types.h"
72 #include "DNA_world_types.h"
73
74 #include "BLI_blenlib.h"
75 #include "BLI_utildefines.h"
76
77 #include "BLI_threads.h"
78 #include "BLT_translation.h"
79
80 #include "BKE_action.h"
81 #include "BKE_animsys.h"
82 #include "BKE_armature.h"
83 #include "BKE_bpath.h"
84 #include "BKE_brush.h"
85 #include "BKE_camera.h"
86 #include "BKE_cachefile.h"
87 #include "BKE_context.h"
88 #include "BKE_curve.h"
89 #include "BKE_depsgraph.h"
90 #include "BKE_font.h"
91 #include "BKE_global.h"
92 #include "BKE_group.h"
93 #include "BKE_gpencil.h"
94 #include "BKE_idcode.h"
95 #include "BKE_idprop.h"
96 #include "BKE_image.h"
97 #include "BKE_key.h"
98 #include "BKE_lamp.h"
99 #include "BKE_lattice.h"
100 #include "BKE_library.h"
101 #include "BKE_library_query.h"
102 #include "BKE_library_remap.h"
103 #include "BKE_linestyle.h"
104 #include "BKE_mesh.h"
105 #include "BKE_material.h"
106 #include "BKE_main.h"
107 #include "BKE_mball.h"
108 #include "BKE_mask.h"
109 #include "BKE_movieclip.h"
110 #include "BKE_node.h"
111 #include "BKE_object.h"
112 #include "BKE_paint.h"
113 #include "BKE_packedFile.h"
114 #include "BKE_sound.h"
115 #include "BKE_speaker.h"
116 #include "BKE_scene.h"
117 #include "BKE_text.h"
118 #include "BKE_texture.h"
119 #include "BKE_world.h"
120
121 #include "DEG_depsgraph.h"
122
123 #include "RNA_access.h"
124
125 #include "IMB_imbuf.h"
126 #include "IMB_imbuf_types.h"
127
128 /* GS reads the memory pointed at in a specific ordering. 
129  * only use this definition, makes little and big endian systems
130  * work fine, in conjunction with MAKE_ID */
131
132 /* ************* general ************************ */
133
134
135 /* this has to be called from each make_local_* func, we could call
136  * from id_make_local() but then the make local functions would not be self
137  * contained.
138  * also note that the id _must_ have a library - campbell */
139 void BKE_id_lib_local_paths(Main *bmain, Library *lib, ID *id)
140 {
141         const char *bpath_user_data[2] = {bmain->name, lib->filepath};
142
143         BKE_bpath_traverse_id(bmain, id,
144                               BKE_bpath_relocate_visitor,
145                               BKE_BPATH_TRAVERSE_SKIP_MULTIFILE,
146                               (void *)bpath_user_data);
147 }
148
149 void id_lib_extern(ID *id)
150 {
151         if (id && ID_IS_LINKED_DATABLOCK(id)) {
152                 BLI_assert(BKE_idcode_is_linkable(GS(id->name)));
153                 if (id->tag & LIB_TAG_INDIRECT) {
154                         id->tag -= LIB_TAG_INDIRECT;
155                         id->tag |= LIB_TAG_EXTERN;
156                 }
157         }
158 }
159
160 /* ensure we have a real user */
161 /* Note: Now that we have flags, we could get rid of the 'fake_user' special case, flags are enough to ensure
162  *       we always have a real user.
163  *       However, ID_REAL_USERS is used in several places outside of core library.c, so think we can wait later
164  *       to make this change... */
165 void id_us_ensure_real(ID *id)
166 {
167         if (id) {
168                 const int limit = ID_FAKE_USERS(id);
169                 id->tag |= LIB_TAG_EXTRAUSER;
170                 if (id->us <= limit) {
171                         if (id->us < limit || ((id->us == limit) && (id->tag & LIB_TAG_EXTRAUSER_SET))) {
172                                 printf("ID user count error: %s (from '%s')\n", id->name, id->lib ? id->lib->filepath : "[Main]");
173                                 BLI_assert(0);
174                         }
175                         id->us = limit + 1;
176                         id->tag |= LIB_TAG_EXTRAUSER_SET;
177                 }
178         }
179 }
180
181 void id_us_clear_real(ID *id)
182 {
183         if (id && (id->tag & LIB_TAG_EXTRAUSER)) {
184                 if (id->tag & LIB_TAG_EXTRAUSER_SET) {
185                         id->us--;
186                         BLI_assert(id->us >= ID_FAKE_USERS(id));
187                 }
188                 id->tag &= ~(LIB_TAG_EXTRAUSER | LIB_TAG_EXTRAUSER_SET);
189         }
190 }
191
192 /**
193  * Same as \a id_us_plus, but does not handle lib indirect -> extern.
194  * Only used by readfile.c so far, but simpler/safer to keep it here nonetheless.
195  */
196 void id_us_plus_no_lib(ID *id)
197 {
198         if (id) {
199                 if ((id->tag & LIB_TAG_EXTRAUSER) && (id->tag & LIB_TAG_EXTRAUSER_SET)) {
200                         BLI_assert(id->us >= 1);
201                         /* No need to increase count, just tag extra user as no more set.
202                          * Avoids annoying & inconsistent +1 in user count. */
203                         id->tag &= ~LIB_TAG_EXTRAUSER_SET;
204                 }
205                 else {
206                         BLI_assert(id->us >= 0);
207                         id->us++;
208                 }
209         }
210 }
211
212
213 void id_us_plus(ID *id)
214 {
215         if (id) {
216                 id_us_plus_no_lib(id);
217                 id_lib_extern(id);
218         }
219 }
220
221 /* decrements the user count for *id. */
222 void id_us_min(ID *id)
223 {
224         if (id) {
225                 const int limit = ID_FAKE_USERS(id);
226
227                 if (id->us <= limit) {
228                         printf("ID user decrement error: %s (from '%s'): %d <= %d\n",
229                                id->name, id->lib ? id->lib->filepath : "[Main]", id->us, limit);
230                         BLI_assert(0);
231                         id->us = limit;
232                 }
233                 else {
234                         id->us--;
235                 }
236
237                 if ((id->us == limit) && (id->tag & LIB_TAG_EXTRAUSER)) {
238                         /* We need an extra user here, but never actually incremented user count for it so far, do it now. */
239                         id_us_ensure_real(id);
240                 }
241         }
242 }
243
244 void id_fake_user_set(ID *id)
245 {
246         if (id && !(id->flag & LIB_FAKEUSER)) {
247                 id->flag |= LIB_FAKEUSER;
248                 id_us_plus(id);
249         }
250 }
251
252 void id_fake_user_clear(ID *id)
253 {
254         if (id && (id->flag & LIB_FAKEUSER)) {
255                 id->flag &= ~LIB_FAKEUSER;
256                 id_us_min(id);
257         }
258 }
259
260 static int id_expand_local_callback(
261         void *UNUSED(user_data), struct ID *UNUSED(id_self), struct ID **id_pointer, int UNUSED(cd_flag))
262 {
263         if (*id_pointer) {
264                 id_lib_extern(*id_pointer);
265         }
266
267         return IDWALK_RET_NOP;
268 }
269
270 /**
271  * Expand ID usages of given id as 'extern' (and no more indirect) linked data. Used by ID copy/make_local functions.
272  */
273 void BKE_id_expand_local(ID *id)
274 {
275         BKE_library_foreach_ID_link(id, id_expand_local_callback, NULL, 0);
276 }
277
278 /**
279  * Ensure new (copied) ID is fully made local.
280  */
281 void BKE_id_copy_ensure_local(Main *bmain, ID *old_id, ID *new_id)
282 {
283         if (ID_IS_LINKED_DATABLOCK(old_id)) {
284                 BKE_id_expand_local(new_id);
285                 BKE_id_lib_local_paths(bmain, old_id->lib, new_id);
286         }
287 }
288
289 /**
290  * Generic 'make local' function, works for most of datablock types...
291  */
292 void BKE_id_make_local_generic(Main *bmain, ID *id, const bool id_in_mainlist, const bool lib_local)
293 {
294         bool is_local = false, is_lib = false;
295
296         /* - only lib users: do nothing (unless force_local is set)
297          * - only local users: set flag
298          * - mixed: make copy
299          * In case we make a whole lib's content local, we always want to localize, and we skip remapping (done later).
300          */
301
302         if (!ID_IS_LINKED_DATABLOCK(id)) {
303                 return;
304         }
305
306         BKE_library_ID_test_usages(bmain, id, &is_local, &is_lib);
307
308         if (lib_local || is_local) {
309                 if (!is_lib) {
310                         id_clear_lib_data_ex(bmain, id, id_in_mainlist);
311                         BKE_id_expand_local(id);
312                 }
313                 else {
314                         ID *id_new;
315
316                         /* Should not fail in expected usecases, but id_copy does not copy Scene e.g. */
317                         if (id_copy(bmain, id, &id_new, false)) {
318                                 id_new->us = 0;
319
320                                 if (!lib_local) {
321                                         BKE_libblock_remap(bmain, id, id_new, ID_REMAP_SKIP_INDIRECT_USAGE);
322                                 }
323                         }
324                 }
325         }
326 }
327
328 /**
329  * Calls the appropriate make_local method for the block, unless test is set.
330  *
331  * \param lib_local Special flag used when making a whole library's content local, it needs specific handling.
332  *
333  * \return true if the block can be made local.
334  */
335 bool id_make_local(Main *bmain, ID *id, const bool test, const bool lib_local)
336 {
337         /* We don't care whether ID is directly or indirectly linked in case we are making a whole lib local... */
338         if (!lib_local && (id->tag & LIB_TAG_INDIRECT)) {
339                 return false;
340         }
341
342         switch ((ID_Type)GS(id->name)) {
343                 case ID_SCE:
344                         /* Partially implemented (has no copy...). */
345                         if (!test) BKE_scene_make_local(bmain, (Scene *)id, lib_local);
346                         return true;
347                 case ID_OB:
348                         if (!test) BKE_object_make_local(bmain, (Object *)id, lib_local);
349                         return true;
350                 case ID_ME:
351                         if (!test) BKE_mesh_make_local(bmain, (Mesh *)id, lib_local);
352                         return true;
353                 case ID_CU:
354                         if (!test) BKE_curve_make_local(bmain, (Curve *)id, lib_local);
355                         return true;
356                 case ID_MB:
357                         if (!test) BKE_mball_make_local(bmain, (MetaBall *)id, lib_local);
358                         return true;
359                 case ID_MA:
360                         if (!test) BKE_material_make_local(bmain, (Material *)id, lib_local);
361                         return true;
362                 case ID_TE:
363                         if (!test) BKE_texture_make_local(bmain, (Tex *)id, lib_local);
364                         return true;
365                 case ID_IM:
366                         if (!test) BKE_image_make_local(bmain, (Image *)id, lib_local);
367                         return true;
368                 case ID_LT:
369                         if (!test) BKE_lattice_make_local(bmain, (Lattice *)id, lib_local);
370                         return true;
371                 case ID_LA:
372                         if (!test) BKE_lamp_make_local(bmain, (Lamp *)id, lib_local);
373                         return true;
374                 case ID_CA:
375                         if (!test) BKE_camera_make_local(bmain, (Camera *)id, lib_local);
376                         return true;
377                 case ID_SPK:
378                         if (!test) BKE_speaker_make_local(bmain, (Speaker *)id, lib_local);
379                         return true;
380                 case ID_WO:
381                         if (!test) BKE_world_make_local(bmain, (World *)id, lib_local);
382                         return true;
383                 case ID_VF:
384                         /* Partially implemented (has no copy...). */
385                         if (!test) BKE_vfont_make_local(bmain, (VFont *)id, lib_local);
386                         return true;
387                 case ID_TXT:
388                         if (!test) BKE_text_make_local(bmain, (Text *)id, lib_local);
389                         return true;
390                 case ID_SO:
391                         /* Partially implemented (has no copy...). */
392                         if (!test) BKE_sound_make_local(bmain, (bSound *)id, lib_local);
393                         return true;
394                 case ID_GR:
395                         if (!test) BKE_group_make_local(bmain, (Group *)id, lib_local);
396                         return true;
397                 case ID_AR:
398                         if (!test) BKE_armature_make_local(bmain, (bArmature *)id, lib_local);
399                         return true;
400                 case ID_AC:
401                         if (!test) BKE_action_make_local(bmain, (bAction *)id, lib_local);
402                         return true;
403                 case ID_NT:
404                         if (!test) ntreeMakeLocal(bmain, (bNodeTree *)id, true, lib_local);
405                         return true;
406                 case ID_BR:
407                         if (!test) BKE_brush_make_local(bmain, (Brush *)id, lib_local);
408                         return true;
409                 case ID_GD:
410                         if (!test) BKE_gpencil_make_local(bmain, (bGPdata *)id, lib_local);
411                         return true;
412                 case ID_MC:
413                         if (!test) BKE_movieclip_make_local(bmain, (MovieClip *)id, lib_local);
414                         return true;
415                 case ID_MSK:
416                         if (!test) BKE_mask_make_local(bmain, (Mask *)id, lib_local);
417                         return true;
418                 case ID_LS:
419                         if (!test) BKE_linestyle_make_local(bmain, (FreestyleLineStyle *)id, lib_local);
420                         return true;
421                 case ID_PAL:
422                         if (!test) BKE_palette_make_local(bmain, (Palette *)id, lib_local);
423                         return true;
424                 case ID_PC:
425                         if (!test) BKE_paint_curve_make_local(bmain, (PaintCurve *)id, lib_local);
426                         return true;
427                 case ID_CF:
428                         if (!test) BKE_cachefile_make_local(bmain, (CacheFile *)id, lib_local);
429                         return true;
430                 case ID_SCR:
431                 case ID_LI:
432                 case ID_KE:
433                 case ID_WM:
434                         return false; /* can't be linked */
435                 case ID_IP:
436                         return false; /* deprecated */
437         }
438
439         return false;
440 }
441
442 /**
443  * Invokes the appropriate copy method for the block and returns the result in
444  * newid, unless test. Returns true if the block can be copied.
445  */
446 bool id_copy(Main *bmain, ID *id, ID **newid, bool test)
447 {
448         if (!test) {
449                 *newid = NULL;
450         }
451
452         /* conventions:
453          * - make shallow copy, only this ID block
454          * - id.us of the new ID is set to 1 */
455         switch ((ID_Type)GS(id->name)) {
456                 case ID_OB:
457                         if (!test) *newid = (ID *)BKE_object_copy(bmain, (Object *)id);
458                         return true;
459                 case ID_ME:
460                         if (!test) *newid = (ID *)BKE_mesh_copy(bmain, (Mesh *)id);
461                         return true;
462                 case ID_CU:
463                         if (!test) *newid = (ID *)BKE_curve_copy(bmain, (Curve *)id);
464                         return true;
465                 case ID_MB:
466                         if (!test) *newid = (ID *)BKE_mball_copy(bmain, (MetaBall *)id);
467                         return true;
468                 case ID_MA:
469                         if (!test) *newid = (ID *)BKE_material_copy(bmain, (Material *)id);
470                         return true;
471                 case ID_TE:
472                         if (!test) *newid = (ID *)BKE_texture_copy(bmain, (Tex *)id);
473                         return true;
474                 case ID_IM:
475                         if (!test) *newid = (ID *)BKE_image_copy(bmain, (Image *)id);
476                         return true;
477                 case ID_LT:
478                         if (!test) *newid = (ID *)BKE_lattice_copy(bmain, (Lattice *)id);
479                         return true;
480                 case ID_LA:
481                         if (!test) *newid = (ID *)BKE_lamp_copy(bmain, (Lamp *)id);
482                         return true;
483                 case ID_SPK:
484                         if (!test) *newid = (ID *)BKE_speaker_copy(bmain, (Speaker *)id);
485                         return true;
486                 case ID_CA:
487                         if (!test) *newid = (ID *)BKE_camera_copy(bmain, (Camera *)id);
488                         return true;
489                 case ID_KE:
490                         if (!test) *newid = (ID *)BKE_key_copy(bmain, (Key *)id);
491                         return true;
492                 case ID_WO:
493                         if (!test) *newid = (ID *)BKE_world_copy(bmain, (World *)id);
494                         return true;
495                 case ID_TXT:
496                         if (!test) *newid = (ID *)BKE_text_copy(bmain, (Text *)id);
497                         return true;
498                 case ID_GR:
499                         if (!test) *newid = (ID *)BKE_group_copy(bmain, (Group *)id);
500                         return true;
501                 case ID_AR:
502                         if (!test) *newid = (ID *)BKE_armature_copy(bmain, (bArmature *)id);
503                         return true;
504                 case ID_AC:
505                         if (!test) *newid = (ID *)BKE_action_copy(bmain, (bAction *)id);
506                         return true;
507                 case ID_NT:
508                         if (!test) *newid = (ID *)ntreeCopyTree(bmain, (bNodeTree *)id);
509                         return true;
510                 case ID_BR:
511                         if (!test) *newid = (ID *)BKE_brush_copy(bmain, (Brush *)id);
512                         return true;
513                 case ID_GD:
514                         if (!test) *newid = (ID *)BKE_gpencil_data_duplicate(bmain, (bGPdata *)id, false);
515                         return true;
516                 case ID_MC:
517                         if (!test) *newid = (ID *)BKE_movieclip_copy(bmain, (MovieClip *)id);
518                         return true;
519                 case ID_MSK:
520                         if (!test) *newid = (ID *)BKE_mask_copy(bmain, (Mask *)id);
521                         return true;
522                 case ID_LS:
523                         if (!test) *newid = (ID *)BKE_linestyle_copy(bmain, (FreestyleLineStyle *)id);
524                         return true;
525                 case ID_PAL:
526                         if (!test) *newid = (ID *)BKE_palette_copy(bmain, (Palette *)id);
527                         return true;
528                 case ID_PC:
529                         if (!test) *newid = (ID *)BKE_paint_curve_copy(bmain, (PaintCurve *)id);
530                         return true;
531                 case ID_CF:
532                         if (!test) *newid = (ID *)BKE_cachefile_copy(bmain, (CacheFile *)id);
533                         return true;
534                 case ID_SCE:
535                 case ID_LI:
536                 case ID_SCR:
537                 case ID_WM:
538                         return false;  /* can't be copied from here */
539                 case ID_VF:
540                 case ID_SO:
541                         return false;  /* not implemented */
542                 case ID_IP:
543                         return false;  /* deprecated */
544         }
545         
546         return false;
547 }
548
549 bool id_single_user(bContext *C, ID *id, PointerRNA *ptr, PropertyRNA *prop)
550 {
551         ID *newid = NULL;
552         PointerRNA idptr;
553         
554         if (id) {
555                 /* if property isn't editable, we're going to have an extra block hanging around until we save */
556                 if (RNA_property_editable(ptr, prop)) {
557                         if (id_copy(CTX_data_main(C), id, &newid, false) && newid) {
558                                 /* copy animation actions too */
559                                 BKE_animdata_copy_id_action(id);
560                                 /* us is 1 by convention, but RNA_property_pointer_set
561                                  * will also increment it, so set it to zero */
562                                 newid->us = 0;
563                                 
564                                 /* assign copy */
565                                 RNA_id_pointer_create(newid, &idptr);
566                                 RNA_property_pointer_set(ptr, prop, idptr);
567                                 RNA_property_update(C, ptr, prop);
568                                 
569                                 return true;
570                         }
571                 }
572         }
573         
574         return false;
575 }
576
577 ListBase *which_libbase(Main *mainlib, short type)
578 {
579         switch ((ID_Type)type) {
580                 case ID_SCE:
581                         return &(mainlib->scene);
582                 case ID_LI:
583                         return &(mainlib->library);
584                 case ID_OB:
585                         return &(mainlib->object);
586                 case ID_ME:
587                         return &(mainlib->mesh);
588                 case ID_CU:
589                         return &(mainlib->curve);
590                 case ID_MB:
591                         return &(mainlib->mball);
592                 case ID_MA:
593                         return &(mainlib->mat);
594                 case ID_TE:
595                         return &(mainlib->tex);
596                 case ID_IM:
597                         return &(mainlib->image);
598                 case ID_LT:
599                         return &(mainlib->latt);
600                 case ID_LA:
601                         return &(mainlib->lamp);
602                 case ID_CA:
603                         return &(mainlib->camera);
604                 case ID_IP:
605                         return &(mainlib->ipo);
606                 case ID_KE:
607                         return &(mainlib->key);
608                 case ID_WO:
609                         return &(mainlib->world);
610                 case ID_SCR:
611                         return &(mainlib->screen);
612                 case ID_VF:
613                         return &(mainlib->vfont);
614                 case ID_TXT:
615                         return &(mainlib->text);
616                 case ID_SPK:
617                         return &(mainlib->speaker);
618                 case ID_SO:
619                         return &(mainlib->sound);
620                 case ID_GR:
621                         return &(mainlib->group);
622                 case ID_AR:
623                         return &(mainlib->armature);
624                 case ID_AC:
625                         return &(mainlib->action);
626                 case ID_NT:
627                         return &(mainlib->nodetree);
628                 case ID_BR:
629                         return &(mainlib->brush);
630                 case ID_WM:
631                         return &(mainlib->wm);
632                 case ID_GD:
633                         return &(mainlib->gpencil);
634                 case ID_MC:
635                         return &(mainlib->movieclip);
636                 case ID_MSK:
637                         return &(mainlib->mask);
638                 case ID_LS:
639                         return &(mainlib->linestyle);
640                 case ID_PAL:
641                         return &(mainlib->palettes);
642                 case ID_PC:
643                         return &(mainlib->paintcurves);
644                 case ID_CF:
645                         return &(mainlib->cachefiles);
646         }
647         return NULL;
648 }
649
650 /**
651  * Clear or set given tags for all ids in listbase (runtime tags).
652  */
653 void BKE_main_id_tag_listbase(ListBase *lb, const int tag, const bool value)
654 {
655         ID *id;
656         if (value) {
657                 for (id = lb->first; id; id = id->next) {
658                         id->tag |= tag;
659                 }
660         }
661         else {
662                 const int ntag = ~tag;
663                 for (id = lb->first; id; id = id->next) {
664                         id->tag &= ntag;
665                 }
666         }
667 }
668
669 /**
670  * Clear or set given tags for all ids of given type in bmain (runtime tags).
671  */
672 void BKE_main_id_tag_idcode(struct Main *mainvar, const short type, const int tag, const bool value)
673 {
674         ListBase *lb = which_libbase(mainvar, type);
675
676         BKE_main_id_tag_listbase(lb, tag, value);
677 }
678
679 /**
680  * Clear or set given tags for all ids in bmain (runtime tags).
681  */
682 void BKE_main_id_tag_all(struct Main *mainvar, const int tag, const bool value)
683 {
684         ListBase *lbarray[MAX_LIBARRAY];
685         int a;
686
687         a = set_listbasepointers(mainvar, lbarray);
688         while (a--) {
689                 BKE_main_id_tag_listbase(lbarray[a], tag, value);
690         }
691 }
692
693
694 /**
695  * Clear or set given flags for all ids in listbase (persistent flags).
696  */
697 void BKE_main_id_flag_listbase(ListBase *lb, const int flag, const bool value)
698 {
699         ID *id;
700         if (value) {
701                 for (id = lb->first; id; id = id->next)
702                         id->tag |= flag;
703         }
704         else {
705                 const int nflag = ~flag;
706                 for (id = lb->first; id; id = id->next)
707                         id->tag &= nflag;
708         }
709 }
710
711 /**
712  * Clear or set given flags for all ids in bmain (persistent flags).
713  */
714 void BKE_main_id_flag_all(Main *bmain, const int flag, const bool value)
715 {
716         ListBase *lbarray[MAX_LIBARRAY];
717         int a;
718         a = set_listbasepointers(bmain, lbarray);
719         while (a--) {
720                 BKE_main_id_flag_listbase(lbarray[a], flag, value);
721         }
722 }
723
724 void BKE_main_lib_objects_recalc_all(Main *bmain)
725 {
726         Object *ob;
727
728         /* flag for full recalc */
729         for (ob = bmain->object.first; ob; ob = ob->id.next) {
730                 if (ID_IS_LINKED_DATABLOCK(ob)) {
731                         DAG_id_tag_update(&ob->id, OB_RECALC_OB | OB_RECALC_DATA | OB_RECALC_TIME);
732                 }
733         }
734
735         DAG_id_type_tag(bmain, ID_OB);
736 }
737
738 /**
739  * puts into array *lb pointers to all the ListBase structs in main,
740  * and returns the number of them as the function result. This is useful for
741  * generic traversal of all the blocks in a Main (by traversing all the
742  * lists in turn), without worrying about block types.
743  *
744  * \note MAX_LIBARRAY define should match this code */
745 int set_listbasepointers(Main *main, ListBase **lb)
746 {
747         /* BACKWARDS! also watch order of free-ing! (mesh<->mat), first items freed last.
748          * This is important because freeing data decreases usercounts of other datablocks,
749          * if this data is its self freed it can crash. */
750         lb[INDEX_ID_LI] = &(main->library);  /* Libraries may be accessed from pretty much any other ID... */
751         lb[INDEX_ID_IP] = &(main->ipo);
752         lb[INDEX_ID_AC] = &(main->action); /* moved here to avoid problems when freeing with animato (aligorith) */
753         lb[INDEX_ID_KE] = &(main->key);
754         lb[INDEX_ID_GD] = &(main->gpencil); /* referenced by nodes, objects, view, scene etc, before to free after. */
755         lb[INDEX_ID_NT] = &(main->nodetree);
756         lb[INDEX_ID_IM] = &(main->image);
757         lb[INDEX_ID_TE] = &(main->tex);
758         lb[INDEX_ID_MA] = &(main->mat);
759         lb[INDEX_ID_VF] = &(main->vfont);
760         
761         /* Important!: When adding a new object type,
762          * the specific data should be inserted here 
763          */
764
765         lb[INDEX_ID_AR] = &(main->armature);
766
767         lb[INDEX_ID_CF] = &(main->cachefiles);
768         lb[INDEX_ID_ME] = &(main->mesh);
769         lb[INDEX_ID_CU] = &(main->curve);
770         lb[INDEX_ID_MB] = &(main->mball);
771
772         lb[INDEX_ID_LT] = &(main->latt);
773         lb[INDEX_ID_LA] = &(main->lamp);
774         lb[INDEX_ID_CA] = &(main->camera);
775
776         lb[INDEX_ID_TXT] = &(main->text);
777         lb[INDEX_ID_SO]  = &(main->sound);
778         lb[INDEX_ID_GR]  = &(main->group);
779         lb[INDEX_ID_PAL] = &(main->palettes);
780         lb[INDEX_ID_PC]  = &(main->paintcurves);
781         lb[INDEX_ID_BR]  = &(main->brush);
782         lb[INDEX_ID_SPK] = &(main->speaker);
783
784         lb[INDEX_ID_WO]  = &(main->world);
785         lb[INDEX_ID_MC]  = &(main->movieclip);
786         lb[INDEX_ID_SCR] = &(main->screen);
787         lb[INDEX_ID_OB]  = &(main->object);
788         lb[INDEX_ID_LS]  = &(main->linestyle); /* referenced by scenes */
789         lb[INDEX_ID_SCE] = &(main->scene);
790         lb[INDEX_ID_WM]  = &(main->wm);
791         lb[INDEX_ID_MSK] = &(main->mask);
792         
793         lb[INDEX_ID_NULL] = NULL;
794
795         return (MAX_LIBARRAY - 1);
796 }
797
798 /* *********** ALLOC AND FREE *****************
799  *
800  * BKE_libblock_free(ListBase *lb, ID *id )
801  * provide a list-basis and datablock, but only ID is read
802  *
803  * void *BKE_libblock_alloc(ListBase *lb, type, name)
804  * inserts in list and returns a new ID
805  *
806  * **************************** */
807
808 /**
809  * Allocates and returns memory of the right size for the specified block type,
810  * initialized to zero.
811  */
812 void *BKE_libblock_alloc_notest(short type)
813 {
814         ID *id = NULL;
815         
816         switch ((ID_Type)type) {
817                 case ID_SCE:
818                         id = MEM_callocN(sizeof(Scene), "scene");
819                         break;
820                 case ID_LI:
821                         id = MEM_callocN(sizeof(Library), "library");
822                         break;
823                 case ID_OB:
824                         id = MEM_callocN(sizeof(Object), "object");
825                         break;
826                 case ID_ME:
827                         id = MEM_callocN(sizeof(Mesh), "mesh");
828                         break;
829                 case ID_CU:
830                         id = MEM_callocN(sizeof(Curve), "curve");
831                         break;
832                 case ID_MB:
833                         id = MEM_callocN(sizeof(MetaBall), "mball");
834                         break;
835                 case ID_MA:
836                         id = MEM_callocN(sizeof(Material), "mat");
837                         break;
838                 case ID_TE:
839                         id = MEM_callocN(sizeof(Tex), "tex");
840                         break;
841                 case ID_IM:
842                         id = MEM_callocN(sizeof(Image), "image");
843                         break;
844                 case ID_LT:
845                         id = MEM_callocN(sizeof(Lattice), "latt");
846                         break;
847                 case ID_LA:
848                         id = MEM_callocN(sizeof(Lamp), "lamp");
849                         break;
850                 case ID_CA:
851                         id = MEM_callocN(sizeof(Camera), "camera");
852                         break;
853                 case ID_IP:
854                         id = MEM_callocN(sizeof(Ipo), "ipo");
855                         break;
856                 case ID_KE:
857                         id = MEM_callocN(sizeof(Key), "key");
858                         break;
859                 case ID_WO:
860                         id = MEM_callocN(sizeof(World), "world");
861                         break;
862                 case ID_SCR:
863                         id = MEM_callocN(sizeof(bScreen), "screen");
864                         break;
865                 case ID_VF:
866                         id = MEM_callocN(sizeof(VFont), "vfont");
867                         break;
868                 case ID_TXT:
869                         id = MEM_callocN(sizeof(Text), "text");
870                         break;
871                 case ID_SPK:
872                         id = MEM_callocN(sizeof(Speaker), "speaker");
873                         break;
874                 case ID_SO:
875                         id = MEM_callocN(sizeof(bSound), "sound");
876                         break;
877                 case ID_GR:
878                         id = MEM_callocN(sizeof(Group), "group");
879                         break;
880                 case ID_AR:
881                         id = MEM_callocN(sizeof(bArmature), "armature");
882                         break;
883                 case ID_AC:
884                         id = MEM_callocN(sizeof(bAction), "action");
885                         break;
886                 case ID_NT:
887                         id = MEM_callocN(sizeof(bNodeTree), "nodetree");
888                         break;
889                 case ID_BR:
890                         id = MEM_callocN(sizeof(Brush), "brush");
891                         break;
892                 case ID_WM:
893                         id = MEM_callocN(sizeof(wmWindowManager), "Window manager");
894                         break;
895                 case ID_GD:
896                         id = MEM_callocN(sizeof(bGPdata), "Grease Pencil");
897                         break;
898                 case ID_MC:
899                         id = MEM_callocN(sizeof(MovieClip), "Movie Clip");
900                         break;
901                 case ID_MSK:
902                         id = MEM_callocN(sizeof(Mask), "Mask");
903                         break;
904                 case ID_LS:
905                         id = MEM_callocN(sizeof(FreestyleLineStyle), "Freestyle Line Style");
906                         break;
907                 case ID_PAL:
908                         id = MEM_callocN(sizeof(Palette), "Palette");
909                         break;
910                 case ID_PC:
911                         id = MEM_callocN(sizeof(PaintCurve), "Paint Curve");
912                         break;
913                 case ID_CF:
914                         id = MEM_callocN(sizeof(CacheFile), "Cache File");
915                         break;
916         }
917         return id;
918 }
919
920 /**
921  * Allocates and returns a block of the specified type, with the specified name
922  * (adjusted as necessary to ensure uniqueness), and appended to the specified list.
923  * The user count is set to 1, all other content (apart from name and links) being
924  * initialized to zero.
925  */
926 void *BKE_libblock_alloc(Main *bmain, short type, const char *name)
927 {
928         ID *id = NULL;
929         ListBase *lb = which_libbase(bmain, type);
930         
931         id = BKE_libblock_alloc_notest(type);
932         if (id) {
933                 BKE_main_lock(bmain);
934                 BLI_addtail(lb, id);
935                 id->us = 1;
936                 id->icon_id = 0;
937                 *( (short *)id->name) = type;
938                 new_id(lb, id, name);
939                 /* alphabetic insertion: is in new_id */
940                 BKE_main_unlock(bmain);
941         }
942         DAG_id_type_tag(bmain, type);
943         return id;
944 }
945
946 /**
947  * Initialize an ID of given type, such that it has valid 'empty' data.
948  * ID is assumed to be just calloc'ed.
949  */
950 void BKE_libblock_init_empty(ID *id)
951 {
952         /* Note that only ID types that are not valid when filled of zero should have a callback here. */
953         switch ((ID_Type)GS(id->name)) {
954                 case ID_SCE:
955                         BKE_scene_init((Scene *)id);
956                         break;
957                 case ID_LI:
958                         /* Nothing to do. */
959                         break;
960                 case ID_OB:
961                 {
962                         Object *ob = (Object *)id;
963                         ob->type = OB_EMPTY;
964                         BKE_object_init(ob);
965                         break;
966                 }
967                 case ID_ME:
968                         BKE_mesh_init((Mesh *)id);
969                         break;
970                 case ID_CU:
971                         BKE_curve_init((Curve *)id);
972                         break;
973                 case ID_MB:
974                         BKE_mball_init((MetaBall *)id);
975                         break;
976                 case ID_MA:
977                         BKE_material_init((Material *)id);
978                         break;
979                 case ID_TE:
980                         BKE_texture_default((Tex *)id);
981                         break;
982                 case ID_IM:
983                         BKE_image_init((Image *)id);
984                         break;
985                 case ID_LT:
986                         BKE_lattice_init((Lattice *)id);
987                         break;
988                 case ID_LA:
989                         BKE_lamp_init((Lamp *)id);
990                         break;
991                 case ID_SPK:
992                         BKE_speaker_init((Speaker *)id);
993                         break;
994                 case ID_CA:
995                         BKE_camera_init((Camera *)id);
996                         break;
997                 case ID_WO:
998                         BKE_world_init((World *)id);
999                         break;
1000                 case ID_SCR:
1001                         /* Nothing to do. */
1002                         break;
1003                 case ID_VF:
1004                         BKE_vfont_init((VFont *)id);
1005                         break;
1006                 case ID_TXT:
1007                         BKE_text_init((Text *)id);
1008                         break;
1009                 case ID_SO:
1010                         /* Another fuzzy case, think NULLified content is OK here... */
1011                         break;
1012                 case ID_GR:
1013                         /* Nothing to do. */
1014                         break;
1015                 case ID_AR:
1016                         /* Nothing to do. */
1017                         break;
1018                 case ID_AC:
1019                         /* Nothing to do. */
1020                         break;
1021                 case ID_NT:
1022                         ntreeInitDefault((bNodeTree *)id);
1023                         break;
1024                 case ID_BR:
1025                         BKE_brush_init((Brush *)id);
1026                         break;
1027                 case ID_PC:
1028                         /* Nothing to do. */
1029                         break;
1030                 case ID_GD:
1031                         /* Nothing to do. */
1032                         break;
1033                 case ID_MSK:
1034                         /* Nothing to do. */
1035                         break;
1036                 case ID_LS:
1037                         BKE_linestyle_init((FreestyleLineStyle *)id);
1038                         break;
1039                 case ID_CF:
1040                         BKE_cachefile_init((CacheFile *)id);
1041                         break;
1042                 case ID_KE:
1043                         /* Shapekeys are a complex topic too - they depend on their 'user' data type...
1044                          * They are not linkable, though, so it should never reach here anyway. */
1045                         BLI_assert(0);
1046                         break;
1047                 case ID_WM:
1048                         /* We should never reach this. */
1049                         BLI_assert(0);
1050                         break;
1051                 case ID_IP:
1052                         /* Should not be needed - animation from lib pre-2.5 is broken anyway. */
1053                         BLI_assert(0);
1054                         break;
1055                 default:
1056                         BLI_assert(0);  /* Should never reach this point... */
1057         }
1058 }
1059
1060 /* by spec, animdata is first item after ID */
1061 /* and, trust that BKE_animdata_from_id() will only find AnimData for valid ID-types */
1062 static void id_copy_animdata(ID *id, const bool do_action)
1063 {
1064         AnimData *adt = BKE_animdata_from_id(id);
1065         
1066         if (adt) {
1067                 IdAdtTemplate *iat = (IdAdtTemplate *)id;
1068                 iat->adt = BKE_animdata_copy(iat->adt, do_action); /* could be set to false, need to investigate */
1069         }
1070 }
1071
1072 /* material nodes use this since they are not treated as libdata */
1073 void BKE_libblock_copy_data(ID *id, const ID *id_from, const bool do_action)
1074 {
1075         if (id_from->properties)
1076                 id->properties = IDP_CopyProperty(id_from->properties);
1077
1078         /* the duplicate should get a copy of the animdata */
1079         id_copy_animdata(id, do_action);
1080 }
1081
1082 /* used everywhere in blenkernel */
1083 void *BKE_libblock_copy(Main *bmain, ID *id)
1084 {
1085         ID *idn;
1086         size_t idn_len;
1087
1088         idn = BKE_libblock_alloc(bmain, GS(id->name), id->name + 2);
1089
1090         assert(idn != NULL);
1091
1092         idn_len = MEM_allocN_len(idn);
1093         if ((int)idn_len - (int)sizeof(ID) > 0) { /* signed to allow neg result */
1094                 const char *cp = (const char *)id;
1095                 char *cpn = (char *)idn;
1096
1097                 memcpy(cpn + sizeof(ID), cp + sizeof(ID), idn_len - sizeof(ID));
1098         }
1099         
1100         id->newid = idn;
1101         idn->tag |= LIB_TAG_NEW;
1102
1103         BKE_libblock_copy_data(idn, id, false);
1104         
1105         return idn;
1106 }
1107
1108 void *BKE_libblock_copy_nolib(ID *id, const bool do_action)
1109 {
1110         ID *idn;
1111         size_t idn_len;
1112
1113         idn = BKE_libblock_alloc_notest(GS(id->name));
1114         assert(idn != NULL);
1115
1116         BLI_strncpy(idn->name, id->name, sizeof(idn->name));
1117
1118         idn_len = MEM_allocN_len(idn);
1119         if ((int)idn_len - (int)sizeof(ID) > 0) { /* signed to allow neg result */
1120                 const char *cp = (const char *)id;
1121                 char *cpn = (char *)idn;
1122
1123                 memcpy(cpn + sizeof(ID), cp + sizeof(ID), idn_len - sizeof(ID));
1124         }
1125
1126         id->newid = idn;
1127         idn->tag |= LIB_TAG_NEW;
1128         idn->us = 1;
1129
1130         BKE_libblock_copy_data(idn, id, do_action);
1131
1132         return idn;
1133 }
1134
1135 static int id_relink_looper(void *UNUSED(user_data), ID *UNUSED(self_id), ID **id_pointer, const int cd_flag)
1136 {
1137         ID *id = *id_pointer;
1138         if (id) {
1139                 /* See: NEW_ID macro */
1140                 if (id->newid) {
1141                         BKE_library_update_ID_link_user(id->newid, id, cd_flag);
1142                         *id_pointer = id->newid;
1143                 }
1144                 else if (id->tag & LIB_TAG_NEW) {
1145                         id->tag &= ~LIB_TAG_NEW;
1146                         BKE_libblock_relink(id);
1147                 }
1148         }
1149         return IDWALK_RET_NOP;
1150 }
1151
1152 void BKE_libblock_relink(ID *id)
1153 {
1154         if (ID_IS_LINKED_DATABLOCK(id))
1155                 return;
1156
1157         BKE_library_foreach_ID_link(id, id_relink_looper, NULL, 0);
1158 }
1159
1160 void BKE_library_free(Library *lib)
1161 {
1162         if (lib->packedfile)
1163                 freePackedFile(lib->packedfile);
1164 }
1165
1166 Main *BKE_main_new(void)
1167 {
1168         Main *bmain = MEM_callocN(sizeof(Main), "new main");
1169         bmain->eval_ctx = DEG_evaluation_context_new(DAG_EVAL_VIEWPORT);
1170         bmain->lock = MEM_mallocN(sizeof(SpinLock), "main lock");
1171         BLI_spin_init((SpinLock *)bmain->lock);
1172         return bmain;
1173 }
1174
1175 void BKE_main_free(Main *mainvar)
1176 {
1177         /* also call when reading a file, erase all, etc */
1178         ListBase *lbarray[MAX_LIBARRAY];
1179         int a;
1180
1181         MEM_SAFE_FREE(mainvar->blen_thumb);
1182
1183         a = set_listbasepointers(mainvar, lbarray);
1184         while (a--) {
1185                 ListBase *lb = lbarray[a];
1186                 ID *id;
1187                 
1188                 while ( (id = lb->first) ) {
1189 #if 1
1190                         BKE_libblock_free_ex(mainvar, id, false);
1191 #else
1192                         /* errors freeing ID's can be hard to track down,
1193                          * enable this so valgrind will give the line number in its error log */
1194                         switch (a) {
1195                                 case   0: BKE_libblock_free_ex(mainvar, id, false); break;
1196                                 case   1: BKE_libblock_free_ex(mainvar, id, false); break;
1197                                 case   2: BKE_libblock_free_ex(mainvar, id, false); break;
1198                                 case   3: BKE_libblock_free_ex(mainvar, id, false); break;
1199                                 case   4: BKE_libblock_free_ex(mainvar, id, false); break;
1200                                 case   5: BKE_libblock_free_ex(mainvar, id, false); break;
1201                                 case   6: BKE_libblock_free_ex(mainvar, id, false); break;
1202                                 case   7: BKE_libblock_free_ex(mainvar, id, false); break;
1203                                 case   8: BKE_libblock_free_ex(mainvar, id, false); break;
1204                                 case   9: BKE_libblock_free_ex(mainvar, id, false); break;
1205                                 case  10: BKE_libblock_free_ex(mainvar, id, false); break;
1206                                 case  11: BKE_libblock_free_ex(mainvar, id, false); break;
1207                                 case  12: BKE_libblock_free_ex(mainvar, id, false); break;
1208                                 case  13: BKE_libblock_free_ex(mainvar, id, false); break;
1209                                 case  14: BKE_libblock_free_ex(mainvar, id, false); break;
1210                                 case  15: BKE_libblock_free_ex(mainvar, id, false); break;
1211                                 case  16: BKE_libblock_free_ex(mainvar, id, false); break;
1212                                 case  17: BKE_libblock_free_ex(mainvar, id, false); break;
1213                                 case  18: BKE_libblock_free_ex(mainvar, id, false); break;
1214                                 case  19: BKE_libblock_free_ex(mainvar, id, false); break;
1215                                 case  20: BKE_libblock_free_ex(mainvar, id, false); break;
1216                                 case  21: BKE_libblock_free_ex(mainvar, id, false); break;
1217                                 case  22: BKE_libblock_free_ex(mainvar, id, false); break;
1218                                 case  23: BKE_libblock_free_ex(mainvar, id, false); break;
1219                                 case  24: BKE_libblock_free_ex(mainvar, id, false); break;
1220                                 case  25: BKE_libblock_free_ex(mainvar, id, false); break;
1221                                 case  26: BKE_libblock_free_ex(mainvar, id, false); break;
1222                                 case  27: BKE_libblock_free_ex(mainvar, id, false); break;
1223                                 case  28: BKE_libblock_free_ex(mainvar, id, false); break;
1224                                 case  29: BKE_libblock_free_ex(mainvar, id, false); break;
1225                                 case  30: BKE_libblock_free_ex(mainvar, id, false); break;
1226                                 case  31: BKE_libblock_free_ex(mainvar, id, false); break;
1227                                 case  32: BKE_libblock_free_ex(mainvar, id, false); break;
1228                                 case  33: BKE_libblock_free_ex(mainvar, id, false); break;
1229                                 default:
1230                                         BLI_assert(0);
1231                                         break;
1232                         }
1233 #endif
1234                 }
1235         }
1236
1237         BLI_spin_end((SpinLock *)mainvar->lock);
1238         MEM_freeN(mainvar->lock);
1239         DEG_evaluation_context_free(mainvar->eval_ctx);
1240         MEM_freeN(mainvar);
1241 }
1242
1243 void BKE_main_lock(struct Main *bmain)
1244 {
1245         BLI_spin_lock((SpinLock *) bmain->lock);
1246 }
1247
1248 void BKE_main_unlock(struct Main *bmain)
1249 {
1250         BLI_spin_unlock((SpinLock *) bmain->lock);
1251 }
1252
1253 /**
1254  * Generates a raw .blend file thumbnail data from given image.
1255  *
1256  * \param bmain If not NULL, also store generated data in this Main.
1257  * \param img ImBuf image to generate thumbnail data from.
1258  * \return The generated .blend file raw thumbnail data.
1259  */
1260 BlendThumbnail *BKE_main_thumbnail_from_imbuf(Main *bmain, ImBuf *img)
1261 {
1262         BlendThumbnail *data = NULL;
1263
1264         if (bmain) {
1265                 MEM_SAFE_FREE(bmain->blen_thumb);
1266         }
1267
1268         if (img) {
1269                 const size_t sz = BLEN_THUMB_MEMSIZE(img->x, img->y);
1270                 data = MEM_mallocN(sz, __func__);
1271
1272                 IMB_rect_from_float(img);  /* Just in case... */
1273                 data->width = img->x;
1274                 data->height = img->y;
1275                 memcpy(data->rect, img->rect, sz - sizeof(*data));
1276         }
1277
1278         if (bmain) {
1279                 bmain->blen_thumb = data;
1280         }
1281         return data;
1282 }
1283
1284 /**
1285  * Generates an image from raw .blend file thumbnail \a data.
1286  *
1287  * \param bmain Use this bmain->blen_thumb data if given \a data is NULL.
1288  * \param data Raw .blend file thumbnail data.
1289  * \return An ImBuf from given data, or NULL if invalid.
1290  */
1291 ImBuf *BKE_main_thumbnail_to_imbuf(Main *bmain, BlendThumbnail *data)
1292 {
1293         ImBuf *img = NULL;
1294
1295         if (!data && bmain) {
1296                 data = bmain->blen_thumb;
1297         }
1298
1299         if (data) {
1300                 /* Note: we cannot use IMB_allocFromBuffer(), since it tries to dupalloc passed buffer, which will fail
1301                  *       here (we do not want to pass the first two ints!). */
1302                 img = IMB_allocImBuf((unsigned int)data->width, (unsigned int)data->height, 32, IB_rect | IB_metadata);
1303                 memcpy(img->rect, data->rect, BLEN_THUMB_MEMSIZE(data->width, data->height) - sizeof(*data));
1304         }
1305
1306         return img;
1307 }
1308
1309 /**
1310  * Generates an empty (black) thumbnail for given Main.
1311  */
1312 void BKE_main_thumbnail_create(struct Main *bmain)
1313 {
1314         MEM_SAFE_FREE(bmain->blen_thumb);
1315
1316         bmain->blen_thumb = MEM_callocN(BLEN_THUMB_MEMSIZE(BLEN_THUMB_SIZE, BLEN_THUMB_SIZE), __func__);
1317         bmain->blen_thumb->width = BLEN_THUMB_SIZE;
1318         bmain->blen_thumb->height = BLEN_THUMB_SIZE;
1319 }
1320
1321 /* ***************** ID ************************ */
1322 ID *BKE_libblock_find_name_ex(struct Main *bmain, const short type, const char *name)
1323 {
1324         ListBase *lb = which_libbase(bmain, type);
1325         BLI_assert(lb != NULL);
1326         return BLI_findstring(lb, name, offsetof(ID, name) + 2);
1327 }
1328 ID *BKE_libblock_find_name(const short type, const char *name)
1329 {
1330         return BKE_libblock_find_name_ex(G.main, type, name);
1331 }
1332
1333
1334 void id_sort_by_name(ListBase *lb, ID *id)
1335 {
1336         ID *idtest;
1337         
1338         /* insert alphabetically */
1339         if (lb->first != lb->last) {
1340                 BLI_remlink(lb, id);
1341                 
1342                 idtest = lb->first;
1343                 while (idtest) {
1344                         if (BLI_strcasecmp(idtest->name, id->name) > 0 || (idtest->lib && !id->lib)) {
1345                                 BLI_insertlinkbefore(lb, idtest, id);
1346                                 break;
1347                         }
1348                         idtest = idtest->next;
1349                 }
1350                 /* as last */
1351                 if (idtest == NULL) {
1352                         BLI_addtail(lb, id);
1353                 }
1354         }
1355         
1356 }
1357
1358 /**
1359  * Check to see if there is an ID with the same name as 'name'.
1360  * Returns the ID if so, if not, returns NULL
1361  */
1362 static ID *is_dupid(ListBase *lb, ID *id, const char *name)
1363 {
1364         ID *idtest = NULL;
1365         
1366         for (idtest = lb->first; idtest; idtest = idtest->next) {
1367                 /* if idtest is not a lib */ 
1368                 if (id != idtest && !ID_IS_LINKED_DATABLOCK(idtest)) {
1369                         /* do not test alphabetic! */
1370                         /* optimized */
1371                         if (idtest->name[2] == name[0]) {
1372                                 if (STREQ(name, idtest->name + 2)) break;
1373                         }
1374                 }
1375         }
1376         
1377         return idtest;
1378 }
1379
1380 /**
1381  * Check to see if an ID name is already used, and find a new one if so.
1382  * Return true if created a new name (returned in name).
1383  *
1384  * Normally the ID that's being check is already in the ListBase, so ID *id
1385  * points at the new entry.  The Python Library module needs to know what
1386  * the name of a datablock will be before it is appended; in this case ID *id
1387  * id is NULL
1388  */
1389
1390 static bool check_for_dupid(ListBase *lb, ID *id, char *name)
1391 {
1392         ID *idtest;
1393         int nr = 0, a, left_len;
1394 #define MAX_IN_USE 64
1395         bool in_use[MAX_IN_USE];
1396         /* to speed up finding unused numbers within [1 .. MAX_IN_USE - 1] */
1397
1398         char left[MAX_ID_NAME + 8], leftest[MAX_ID_NAME + 8];
1399
1400         while (true) {
1401
1402                 /* phase 1: id already exists? */
1403                 idtest = is_dupid(lb, id, name);
1404
1405                 /* if there is no double, done */
1406                 if (idtest == NULL) return false;
1407
1408                 /* we have a dup; need to make a new name */
1409                 /* quick check so we can reuse one of first MAX_IN_USE - 1 ids if vacant */
1410                 memset(in_use, false, sizeof(in_use));
1411
1412                 /* get name portion, number portion ("name.number") */
1413                 left_len = BLI_split_name_num(left, &nr, name, '.');
1414
1415                 /* if new name will be too long, truncate it */
1416                 if (nr > 999 && left_len > (MAX_ID_NAME - 8)) {  /* assumption: won't go beyond 9999 */
1417                         left[MAX_ID_NAME - 8] = 0;
1418                         left_len = MAX_ID_NAME - 8;
1419                 }
1420                 else if (left_len > (MAX_ID_NAME - 7)) {
1421                         left[MAX_ID_NAME - 7] = 0;
1422                         left_len = MAX_ID_NAME - 7;
1423                 }
1424
1425                 for (idtest = lb->first; idtest; idtest = idtest->next) {
1426                         int nrtest;
1427                         if ( (id != idtest) &&
1428                              !ID_IS_LINKED_DATABLOCK(idtest) &&
1429                              (*name == *(idtest->name + 2)) &&
1430                              STREQLEN(name, idtest->name + 2, left_len) &&
1431                              (BLI_split_name_num(leftest, &nrtest, idtest->name + 2, '.') == left_len)
1432                              )
1433                         {
1434                                 /* will get here at least once, otherwise is_dupid call above would have returned NULL */
1435                                 if (nrtest < MAX_IN_USE)
1436                                         in_use[nrtest] = true;  /* mark as used */
1437                                 if (nr <= nrtest)
1438                                         nr = nrtest + 1;    /* track largest unused */
1439                         }
1440                 }
1441                 /* At this point, 'nr' will typically be at least 1. (but not always) */
1442                 // BLI_assert(nr >= 1);
1443
1444                 /* decide which value of nr to use */
1445                 for (a = 0; a < MAX_IN_USE; a++) {
1446                         if (a >= nr) break;  /* stop when we've checked up to biggest */  /* redundant check */
1447                         if (!in_use[a]) { /* found an unused value */
1448                                 nr = a;
1449                                 /* can only be zero if all potential duplicate names had
1450                                  * nonzero numeric suffixes, which means name itself has
1451                                  * nonzero numeric suffix (else no name conflict and wouldn't
1452                                  * have got here), which means name[left_len] is not a null */
1453                                 break;
1454                         }
1455                 }
1456                 /* At this point, nr is either the lowest unused number within [0 .. MAX_IN_USE - 1],
1457                  * or 1 greater than the largest used number if all those low ones are taken.
1458                  * We can't be bothered to look for the lowest unused number beyond (MAX_IN_USE - 1). */
1459
1460                 /* If the original name has no numeric suffix, 
1461                  * rather than just chopping and adding numbers, 
1462                  * shave off the end chars until we have a unique name.
1463                  * Check the null terminators match as well so we don't get Cube.000 -> Cube.00 */
1464                 if (nr == 0 && name[left_len] == '\0') {
1465                         int len;
1466                         /* FIXME: this code will never be executed, because either nr will be
1467                          * at least 1, or name will not end at left_len! */
1468                         BLI_assert(0);
1469
1470                         len = left_len - 1;
1471                         idtest = is_dupid(lb, id, name);
1472                         
1473                         while (idtest && len > 1) {
1474                                 name[len--] = '\0';
1475                                 idtest = is_dupid(lb, id, name);
1476                         }
1477                         if (idtest == NULL) return true;
1478                         /* otherwise just continue and use a number suffix */
1479                 }
1480                 
1481                 if (nr > 999 && left_len > (MAX_ID_NAME - 8)) {
1482                         /* this would overflow name buffer */
1483                         left[MAX_ID_NAME - 8] = 0;
1484                         /* left_len = MAX_ID_NAME - 8; */ /* for now this isn't used again */
1485                         memcpy(name, left, sizeof(char) * (MAX_ID_NAME - 7));
1486                         continue;
1487                 }
1488                 /* this format specifier is from hell... */
1489                 BLI_snprintf(name, sizeof(id->name) - 2, "%s.%.3d", left, nr);
1490
1491                 return true;
1492         }
1493
1494 #undef MAX_IN_USE
1495 }
1496
1497 /*
1498  * Only for local blocks: external en indirect blocks already have a
1499  * unique ID.
1500  *
1501  * return true: created a new name
1502  */
1503
1504 bool new_id(ListBase *lb, ID *id, const char *tname)
1505 {
1506         bool result;
1507         char name[MAX_ID_NAME - 2];
1508
1509         /* if library, don't rename */
1510         if (ID_IS_LINKED_DATABLOCK(id))
1511                 return false;
1512
1513         /* if no libdata given, look up based on ID */
1514         if (lb == NULL)
1515                 lb = which_libbase(G.main, GS(id->name));
1516
1517         /* if no name given, use name of current ID
1518          * else make a copy (tname args can be const) */
1519         if (tname == NULL)
1520                 tname = id->name + 2;
1521
1522         BLI_strncpy(name, tname, sizeof(name));
1523
1524         if (name[0] == '\0') {
1525                 /* disallow empty names */
1526                 BLI_strncpy(name, DATA_(ID_FALLBACK_NAME), sizeof(name));
1527         }
1528         else {
1529                 /* disallow non utf8 chars,
1530                  * the interface checks for this but new ID's based on file names don't */
1531                 BLI_utf8_invalid_strip(name, strlen(name));
1532         }
1533
1534         result = check_for_dupid(lb, id, name);
1535         strcpy(id->name + 2, name);
1536
1537         /* This was in 2.43 and previous releases
1538          * however all data in blender should be sorted, not just duplicate names
1539          * sorting should not hurt, but noting just incase it alters the way other
1540          * functions work, so sort every time */
1541 #if 0
1542         if (result)
1543                 id_sort_by_name(lb, id);
1544 #endif
1545
1546         id_sort_by_name(lb, id);
1547         
1548         return result;
1549 }
1550
1551 /**
1552  * Pull an ID out of a library (make it local). Only call this for IDs that
1553  * don't have other library users.
1554  */
1555 void id_clear_lib_data_ex(Main *bmain, ID *id, const bool id_in_mainlist)
1556 {
1557         bNodeTree *ntree = NULL;
1558         Key *key = NULL;
1559
1560         BKE_id_lib_local_paths(bmain, id->lib, id);
1561
1562         id_fake_user_clear(id);
1563
1564         id->lib = NULL;
1565         id->tag &= ~(LIB_TAG_INDIRECT | LIB_TAG_EXTERN);
1566         if (id_in_mainlist)
1567                 new_id(which_libbase(bmain, GS(id->name)), id, NULL);
1568
1569         /* Internal bNodeTree blocks inside datablocks also stores id->lib, make sure this stays in sync. */
1570         if ((ntree = ntreeFromID(id))) {
1571                 id_clear_lib_data_ex(bmain, &ntree->id, false);  /* Datablocks' nodetree is never in Main. */
1572         }
1573
1574         /* Same goes for shapekeys. */
1575         if ((key = BKE_key_from_id(id))) {
1576                 id_clear_lib_data_ex(bmain, &key->id, id_in_mainlist);  /* sigh, why are keys in Main? */
1577         }
1578 }
1579
1580 void id_clear_lib_data(Main *bmain, ID *id)
1581 {
1582         id_clear_lib_data_ex(bmain, id, true);
1583 }
1584
1585 /* next to indirect usage in read/writefile also in editobject.c scene.c */
1586 void BKE_main_id_clear_newpoins(Main *bmain)
1587 {
1588         ListBase *lbarray[MAX_LIBARRAY];
1589         ID *id;
1590         int a;
1591
1592         a = set_listbasepointers(bmain, lbarray);
1593         while (a--) {
1594                 id = lbarray[a]->first;
1595                 while (id) {
1596                         id->newid = NULL;
1597                         id->tag &= ~LIB_TAG_NEW;
1598                         id = id->next;
1599                 }
1600         }
1601 }
1602
1603 /** Make linked datablocks local.
1604  *
1605  * \param bmain Almost certainly G.main.
1606  * \param lib If not NULL, only make local datablocks from this library.
1607  * \param untagged_only If true, only make local datablocks not tagged with LIB_TAG_PRE_EXISTING.
1608  * \param set_fake If true, set fake user on all localized datablocks (except group and objects ones).
1609  */
1610 /* XXX TODO This function should probably be reworked.
1611  *
1612  * Old (2.77) version was simply making (tagging) datablocks as local, without actually making any check whether
1613  * they were also indirectly used or not...
1614  *
1615  * Current version uses regular id_make_local callback, but this is not super-efficient since this ends up
1616  * duplicating some IDs and then removing original ones (due to missing knowledge of which ID uses some other ID).
1617  *
1618  * We could first check all IDs and detect those to be made local that are only used by other local or future-local
1619  * datablocks, and directly tag those as local (instead of going through id_make_local) maybe...
1620  *
1621  * We'll probably need at some point a true dependency graph between datablocks, but for now this should work
1622  * good enough (performances is not a critical point here anyway).
1623  */
1624 void BKE_library_make_local(Main *bmain, const Library *lib, const bool untagged_only, const bool set_fake)
1625 {
1626         ListBase *lbarray[MAX_LIBARRAY];
1627         ID *id, *id_next;
1628         int a;
1629
1630         for (a = set_listbasepointers(bmain, lbarray); a--; ) {
1631                 id = lbarray[a]->first;
1632
1633                 /* Do not explicitly make local non-linkable IDs (shapekeys, in fact), they are assumed to be handled
1634                  * by real datablocks responsible of them. */
1635                 const bool do_skip = (id && !BKE_idcode_is_linkable(GS(id->name)));
1636
1637                 for (; id; id = id_next) {
1638                         id->newid = NULL;
1639                         id_next = id->next;  /* id is possibly being inserted again */
1640
1641                         /* The check on the second line (LIB_TAG_PRE_EXISTING) is done so its
1642                          * possible to tag data you don't want to be made local, used for
1643                          * appending data, so any libdata already linked wont become local
1644                          * (very nasty to discover all your links are lost after appending)  
1645                          * */
1646                         if (!do_skip && id->tag & (LIB_TAG_EXTERN | LIB_TAG_INDIRECT | LIB_TAG_NEW) &&
1647                             ((untagged_only == false) || !(id->tag & LIB_TAG_PRE_EXISTING)))
1648                         {
1649                                 if (lib == NULL || id->lib == lib) {
1650                                         if (id->lib) {
1651                                                 /* In this specific case, we do want to make ID local even if it has no local usage yet... */
1652                                                 if (GS(id->name) == ID_OB) {
1653                                                         /* Special case for objects because we don't want proxy pointers to be
1654                                                          * cleared yet. This will happen down the road in this function.
1655                                                          */
1656                                                         BKE_object_make_local_ex(bmain, (Object*)id, true, false);
1657                                                 }
1658                                                 else {
1659                                                         id_make_local(bmain, id, false, true);
1660                                                 }
1661                                         }
1662                                         else {
1663                                                 id->tag &= ~(LIB_TAG_EXTERN | LIB_TAG_INDIRECT | LIB_TAG_NEW);
1664                                         }
1665                                 }
1666
1667                                 if (set_fake) {
1668                                         if (!ELEM(GS(id->name), ID_OB, ID_GR)) {
1669                                                 /* do not set fake user on objects, groups (instancing) */
1670                                                 id_fake_user_set(id);
1671                                         }
1672                                 }
1673                         }
1674                 }
1675         }
1676
1677         /* We have to remap local usages of old (linked) ID to new (local) id in a second loop, as lbarray ordering is not
1678          * enough to ensure us we did catch all dependencies (e.g. if making local a parent object before its child...).
1679          * See T48907. */
1680         for (a = set_listbasepointers(bmain, lbarray); a--; ) {
1681                 for (id = lbarray[a]->first; id; id = id->next) {
1682                         if (id->newid) {
1683                                 BKE_libblock_remap(bmain, id, id->newid, ID_REMAP_SKIP_INDIRECT_USAGE);
1684                         }
1685                 }
1686         }
1687
1688         /* Third step: remove datablocks that have been copied to be localized and are no more used in the end...
1689          * Note that we may have to loop more than once here, to tackle dependencies between linked objects... */
1690         bool do_loop = true;
1691         while (do_loop) {
1692                 do_loop = false;
1693                 for (a = set_listbasepointers(bmain, lbarray); a--; ) {
1694                         for (id = lbarray[a]->first; id; id = id_next) {
1695                                 id_next = id->next;
1696                                 if (id->newid) {
1697                                         bool is_local = false, is_lib = false;
1698
1699                                         /* Attempt to re-link copied proxy objects. This allows appending of an entire scene
1700                                          * from another blend file into this one, even when that blend file contains proxified
1701                                          * armatures that have local references. Since the proxified object needs to be linked
1702                                          * (not local), this will only work when the "Localize all" checkbox is disabled.
1703                                          * TL;DR: this is a dirty hack on top of an already weak feature (proxies). */
1704                                         if (GS(id->name) == ID_OB && ((Object *)id)->proxy != NULL) {
1705                                                 Object *ob = (Object *)id;
1706                                                 Object *ob_new = (Object *)id->newid;
1707
1708                                                 /* Proxies only work when the proxified object is linked-in from a library. */
1709                                                 if (ob->proxy->id.lib == NULL) {
1710                                                         printf("Warning, proxy object %s will loose its link to %s, because the "
1711                                                                "proxified object is local.\n", id->newid->name, ob->proxy->id.name);
1712                                                 }
1713                                                 /* We can only switch the proxy'ing to a made-local proxy if it is no longer
1714                                                  * referred to from a library. Not checking for local use; if new local proxy
1715                                                  * was not used locally would be a nasty bug! */
1716                                                 else if (is_local || is_lib) {
1717                                                         printf("Warning, made-local proxy object %s will loose its link to %s, "
1718                                                                "because the linked-in proxy is referenced (is_local=%i, is_lib=%i).\n",
1719                                                                id->newid->name, ob->proxy->id.name, is_local, is_lib);
1720                                                 }
1721                                                 else {
1722                                                         /* we can switch the proxy'ing from the linked-in to the made-local proxy.
1723                                                          * BKE_object_make_proxy() shouldn't be used here, as it allocates memory that
1724                                                          * was already allocated by BKE_object_make_local_ex() (which called BKE_object_copy_ex). */
1725                                                         ob_new->proxy = ob->proxy;
1726                                                         ob_new->proxy_group = ob->proxy_group;
1727                                                         ob_new->proxy_from = ob->proxy_from;
1728                                                         ob_new->proxy->proxy_from = ob_new;
1729                                                         ob->proxy = ob->proxy_from = ob->proxy_group = NULL;
1730                                                 }
1731                                         }
1732
1733                                         BKE_library_ID_test_usages(bmain, id, &is_local, &is_lib);
1734                                         if (!is_local && !is_lib) {
1735                                                 BKE_libblock_free(bmain, id);
1736                                                 do_loop = true;
1737                                         }
1738                                 }
1739                         }
1740                 }
1741         }
1742 }
1743
1744 /**
1745  * Use after setting the ID's name
1746  * When name exists: call 'new_id'
1747  */
1748 void BLI_libblock_ensure_unique_name(Main *bmain, const char *name)
1749 {
1750         ListBase *lb;
1751         ID *idtest;
1752
1753
1754         lb = which_libbase(bmain, GS(name));
1755         if (lb == NULL) return;
1756         
1757         /* search for id */
1758         idtest = BLI_findstring(lb, name + 2, offsetof(ID, name) + 2);
1759
1760         if (idtest && !new_id(lb, idtest, idtest->name + 2)) {
1761                 id_sort_by_name(lb, idtest);
1762         }
1763 }
1764
1765 /**
1766  * Sets the name of a block to name, suitably adjusted for uniqueness.
1767  */
1768 void BKE_libblock_rename(Main *bmain, ID *id, const char *name)
1769 {
1770         ListBase *lb = which_libbase(bmain, GS(id->name));
1771         new_id(lb, id, name);
1772 }
1773
1774 /**
1775  * Returns in name the name of the block, with a 3-character prefix prepended
1776  * indicating whether it comes from a library, has a fake user, or no users.
1777  */
1778 void BKE_id_ui_prefix(char name[MAX_ID_NAME + 1], const ID *id)
1779 {
1780         name[0] = id->lib ? (ID_MISSING(id) ? 'M' : 'L') : ' ';
1781         name[1] = (id->flag & LIB_FAKEUSER) ? 'F' : ((id->us == 0) ? '0' : ' ');
1782         name[2] = ' ';
1783
1784         strcpy(name + 3, id->name + 2);
1785 }
1786
1787 void BKE_library_filepath_set(Library *lib, const char *filepath)
1788 {
1789         /* in some cases this is used to update the absolute path from the
1790          * relative */
1791         if (lib->name != filepath) {
1792                 BLI_strncpy(lib->name, filepath, sizeof(lib->name));
1793         }
1794
1795         BLI_strncpy(lib->filepath, filepath, sizeof(lib->filepath));
1796
1797         /* not essential but set filepath is an absolute copy of value which
1798          * is more useful if its kept in sync */
1799         if (BLI_path_is_rel(lib->filepath)) {
1800                 /* note that the file may be unsaved, in this case, setting the
1801                  * filepath on an indirectly linked path is not allowed from the
1802                  * outliner, and its not really supported but allow from here for now
1803                  * since making local could cause this to be directly linked - campbell
1804                  */
1805                 const char *basepath = lib->parent ? lib->parent->filepath : G.main->name;
1806                 BLI_path_abs(lib->filepath, basepath);
1807         }
1808 }