Patch #34204: [Render Animation] Fails with "Error: Specified sample_fmt is not suppo...
[blender.git] / source / blender / blenkernel / intern / blender.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/blender.c
29  *  \ingroup bke
30  */
31
32
33 #ifndef _WIN32 
34 #  include <unistd.h> // for read close
35 #else
36 #  include <io.h> // for open close read
37 #  define open _open
38 #  define read _read
39 #  define close _close
40 #  define write _write
41 #endif
42
43 #include <stdlib.h>
44 #include <stdio.h>
45 #include <stddef.h>
46 #include <string.h>
47 #include <fcntl.h>  /* for open */
48 #include <errno.h>
49
50 #include "MEM_guardedalloc.h"
51
52 #include "DNA_userdef_types.h"
53 #include "DNA_scene_types.h"
54 #include "DNA_screen_types.h"
55 #include "DNA_sequence_types.h"
56 #include "DNA_sound_types.h"
57 #include "DNA_windowmanager_types.h"
58
59 #include "BLI_blenlib.h"
60 #include "BLI_dynstr.h"
61 #include "BLI_utildefines.h"
62 #include "BLI_callbacks.h"
63
64 #include "IMB_imbuf.h"
65 #include "IMB_moviecache.h"
66
67 #include "BKE_blender.h"
68 #include "BKE_bpath.h"
69 #include "BKE_context.h"
70 #include "BKE_depsgraph.h"
71 #include "BKE_displist.h"
72 #include "BKE_global.h"
73 #include "BKE_idprop.h"
74 #include "BKE_image.h"
75 #include "BKE_ipo.h"
76 #include "BKE_library.h"
77 #include "BKE_main.h"
78 #include "BKE_node.h"
79 #include "BKE_report.h"
80 #include "BKE_scene.h"
81 #include "BKE_screen.h"
82 #include "BKE_sequencer.h"
83 #include "BKE_sound.h"
84
85 #include "RE_pipeline.h"
86
87 #include "BLF_api.h"
88
89 #include "BLO_undofile.h"
90 #include "BLO_readfile.h" 
91 #include "BLO_writefile.h" 
92
93 #include "RNA_access.h"
94
95 #include "WM_api.h" // XXXXX BAD, very BAD dependency (bad level call) - remove asap, elubie
96
97 #include "IMB_colormanagement.h"
98
99 #ifdef WITH_PYTHON
100 #  include "BPY_extern.h"
101 #endif
102
103 Global G;
104 UserDef U;
105 /* ListBase = {NULL, NULL}; */
106
107 char versionstr[48] = "";
108
109 /* ********** free ********** */
110
111 /* only to be called on exit blender */
112 void free_blender(void)
113 {
114         /* samples are in a global list..., also sets G.main->sound->sample NULL */
115         free_main(G.main);
116         G.main = NULL;
117
118         BKE_spacetypes_free();      /* after free main, it uses space callbacks */
119         
120         IMB_exit();
121         BKE_images_exit();
122
123         BLI_callback_global_finalize();
124
125         BKE_sequencer_cache_destruct();
126         IMB_moviecache_destruct();
127         
128         free_nodesystem();
129 }
130
131 void initglobals(void)
132 {
133         memset(&G, 0, sizeof(Global));
134         
135         U.savetime = 1;
136
137         G.main = MEM_callocN(sizeof(Main), "initglobals");
138
139         strcpy(G.ima, "//");
140
141         if (BLENDER_SUBVERSION)
142                 BLI_snprintf(versionstr, sizeof(versionstr), "v%d.%02d.%d", BLENDER_VERSION / 100, BLENDER_VERSION % 100, BLENDER_SUBVERSION);
143         else
144                 BLI_snprintf(versionstr, sizeof(versionstr), "v%d.%02d", BLENDER_VERSION / 100, BLENDER_VERSION % 100);
145
146 #ifdef _WIN32   // FULLSCREEN
147         G.windowstate = G_WINDOWSTATE_USERDEF;
148 #endif
149
150         G.charstart = 0x0000;
151         G.charmin = 0x0000;
152         G.charmax = 0xffff;
153
154 #ifndef WITH_PYTHON_SECURITY /* default */
155         G.f |= G_SCRIPT_AUTOEXEC;
156 #else
157         G.f &= ~G_SCRIPT_AUTOEXEC;
158 #endif
159 }
160
161 /***/
162
163 static void clear_global(void) 
164 {
165 //      extern short winqueue_break;    /* screen.c */
166
167         free_main(G.main);          /* free all lib data */
168         
169 //      free_vertexpaint();
170
171         G.main = NULL;
172 }
173
174 static int clean_paths_visit_cb(void *UNUSED(userdata), char *path_dst, const char *path_src)
175 {
176         strcpy(path_dst, path_src);
177         BLI_clean(path_dst);
178         return (strcmp(path_dst, path_src) == 0) ? FALSE : TRUE;
179 }
180
181 /* make sure path names are correct for OS */
182 static void clean_paths(Main *main)
183 {
184         Scene *scene;
185
186         BKE_bpath_traverse_main(main, clean_paths_visit_cb, BKE_BPATH_TRAVERSE_SKIP_MULTIFILE, NULL);
187
188         for (scene = main->scene.first; scene; scene = scene->id.next) {
189                 BLI_clean(scene->r.pic);
190         }
191 }
192
193 /* context matching */
194 /* handle no-ui case */
195
196 /* note, this is called on Undo so any slow conversion functions here
197  * should be avoided or check (mode!='u') */
198
199 static void setup_app_data(bContext *C, BlendFileData *bfd, const char *filepath)
200 {
201         bScreen *curscreen = NULL;
202         Scene *curscene = NULL;
203         int recover;
204         char mode;
205
206         /* 'u' = undo save, 'n' = no UI load */
207         if (bfd->main->screen.first == NULL) mode = 'u';
208         else if (G.fileflags & G_FILE_NO_UI) mode = 'n';
209         else mode = 0;
210
211         recover = (G.fileflags & G_FILE_RECOVER);
212
213         /* Free all render results, without this stale data gets displayed after loading files */
214         if (mode != 'u') {
215                 RE_FreeAllRenderResults();
216         }
217
218         /* Only make filepaths compatible when loading for real (not undo) */
219         if (mode != 'u') {
220                 clean_paths(bfd->main);
221         }
222
223         /* XXX here the complex windowmanager matching */
224         
225         /* no load screens? */
226         if (mode) {
227                 /* comes from readfile.c */
228                 SWAP(ListBase, G.main->wm, bfd->main->wm);
229                 SWAP(ListBase, G.main->screen, bfd->main->screen);
230                 SWAP(ListBase, G.main->script, bfd->main->script);
231                 
232                 /* we re-use current screen */
233                 curscreen = CTX_wm_screen(C);
234                 /* but use new Scene pointer */
235                 curscene = bfd->curscene;
236                 if (curscene == NULL) curscene = bfd->main->scene.first;
237                 /* empty file, we add a scene to make Blender work */
238                 if (curscene == NULL) curscene = BKE_scene_add(bfd->main, "Empty");
239                 
240                 /* and we enforce curscene to be in current screen */
241                 if (curscreen) curscreen->scene = curscene;  /* can run in bgmode */
242
243                 /* clear_global will free G.main, here we can still restore pointers */
244                 blo_lib_link_screen_restore(bfd->main, curscreen, curscene);
245         }
246         
247         /* free G.main Main database */
248 //      CTX_wm_manager_set(C, NULL);
249         clear_global();
250         
251         /* clear old property update cache, in case some old references are left dangling */
252         RNA_property_update_cache_free();
253         
254         G.main = bfd->main;
255
256         CTX_data_main_set(C, G.main);
257
258         sound_init_main(G.main);
259         
260         if (bfd->user) {
261                 
262                 /* only here free userdef themes... */
263                 BKE_userdef_free();
264                 
265                 U = *bfd->user;
266                 MEM_freeN(bfd->user);
267         }
268         
269         /* case G_FILE_NO_UI or no screens in file */
270         if (mode) {
271                 /* leave entire context further unaltered? */
272                 CTX_data_scene_set(C, curscene);
273         }
274         else {
275                 G.winpos = bfd->winpos;
276                 G.displaymode = bfd->displaymode;
277                 G.fileflags = bfd->fileflags;
278                 CTX_wm_manager_set(C, G.main->wm.first);
279                 CTX_wm_screen_set(C, bfd->curscreen);
280                 CTX_data_scene_set(C, bfd->curscene);
281                 CTX_wm_area_set(C, NULL);
282                 CTX_wm_region_set(C, NULL);
283                 CTX_wm_menu_set(C, NULL);
284         }
285         
286         /* this can happen when active scene was lib-linked, and doesn't exist anymore */
287         if (CTX_data_scene(C) == NULL) {
288                 /* in case we don't even have a local scene, add one */
289                 if (!G.main->scene.first)
290                         BKE_scene_add(G.main, "Scene");
291
292                 CTX_data_scene_set(C, G.main->scene.first);
293                 CTX_wm_screen(C)->scene = CTX_data_scene(C);
294                 curscene = CTX_data_scene(C);
295         }
296
297         /* special cases, override loaded flags: */
298         if (G.f != bfd->globalf) {
299                 const int flags_keep = (G_SWAP_EXCHANGE | G_SCRIPT_AUTOEXEC | G_SCRIPT_OVERRIDE_PREF);
300                 bfd->globalf = (bfd->globalf & ~flags_keep) | (G.f & flags_keep);
301         }
302
303
304         G.f = bfd->globalf;
305
306 #ifdef WITH_PYTHON
307         /* let python know about new main */
308         BPY_context_update(C);
309 #endif
310
311         if (!G.background) {
312                 //setscreen(G.curscreen);
313         }
314         
315         /* FIXME: this version patching should really be part of the file-reading code,
316          * but we still get too many unrelated data-corruption crashes otherwise... */
317         if (G.main->versionfile < 250)
318                 do_versions_ipos_to_animato(G.main);
319         
320         G.main->recovered = 0;
321         
322         /* startup.blend or recovered startup */
323         if (bfd->filename[0] == 0) {
324                 G.main->name[0] = 0;
325         }
326         else if (recover && G.relbase_valid) {
327                 /* in case of autosave or quit.blend, use original filename instead
328                  * use relbase_valid to make sure the file is saved, else we get <memory2> in the filename */
329                 filepath = bfd->filename;
330                 G.main->recovered = 1;
331         
332                 /* these are the same at times, should never copy to the same location */
333                 if (G.main->name != filepath)
334                         BLI_strncpy(G.main->name, filepath, FILE_MAX);
335         }
336         
337         /* baseflags, groups, make depsgraph, etc */
338         /* first handle case if other windows have different scenes visible */
339         if (mode == 0) {
340                 wmWindowManager *wm = G.main->wm.first;
341                 
342                 if (wm) {
343                         wmWindow *win;
344                         
345                         for (win = wm->windows.first; win; win = win->next) {
346                                 if (win->screen && win->screen->scene) /* zealous check... */
347                                         if (win->screen->scene != CTX_data_scene(C))
348                                                 BKE_scene_set_background(G.main, win->screen->scene);
349                         }
350                 }
351         }
352         BKE_scene_set_background(G.main, CTX_data_scene(C));
353
354         if (mode != 'u') {
355                 IMB_colormanagement_check_file_config(G.main);
356         }
357
358         MEM_freeN(bfd);
359
360 }
361
362 static int handle_subversion_warning(Main *main, ReportList *reports)
363 {
364         if (main->minversionfile > BLENDER_VERSION ||
365             (main->minversionfile == BLENDER_VERSION &&
366              main->minsubversionfile > BLENDER_SUBVERSION))
367         {
368                 BKE_reportf(reports, RPT_ERROR, "File written by newer Blender binary (%d.%d), expect loss of data!",
369                             main->minversionfile, main->minsubversionfile);
370         }
371
372         return 1;
373 }
374
375 static void keymap_item_free(wmKeyMapItem *kmi)
376 {
377         if (kmi->properties) {
378                 IDP_FreeProperty(kmi->properties);
379                 MEM_freeN(kmi->properties);
380         }
381         if (kmi->ptr)
382                 MEM_freeN(kmi->ptr);
383 }
384
385 void BKE_userdef_free(void)
386 {
387         wmKeyMap *km;
388         wmKeyMapItem *kmi;
389         wmKeyMapDiffItem *kmdi;
390         bAddon *addon, *addon_next;
391
392         for (km = U.user_keymaps.first; km; km = km->next) {
393                 for (kmdi = km->diff_items.first; kmdi; kmdi = kmdi->next) {
394                         if (kmdi->add_item) {
395                                 keymap_item_free(kmdi->add_item);
396                                 MEM_freeN(kmdi->add_item);
397                         }
398                         if (kmdi->remove_item) {
399                                 keymap_item_free(kmdi->remove_item);
400                                 MEM_freeN(kmdi->remove_item);
401                         }
402                 }
403
404                 for (kmi = km->items.first; kmi; kmi = kmi->next)
405                         keymap_item_free(kmi);
406
407                 BLI_freelistN(&km->diff_items);
408                 BLI_freelistN(&km->items);
409         }
410         
411         for (addon = U.addons.first; addon; addon = addon_next) {
412                 addon_next = addon->next;
413                 if (addon->prop) {
414                         IDP_FreeProperty(addon->prop);
415                         MEM_freeN(addon->prop);
416                 }
417                 MEM_freeN(addon);
418         }
419
420         BLI_freelistN(&U.uistyles);
421         BLI_freelistN(&U.uifonts);
422         BLI_freelistN(&U.themes);
423         BLI_freelistN(&U.user_keymaps);
424 }
425
426 /* handle changes in settings that need recalc */
427 void BKE_userdef_state(void)
428 {
429         /* prevent accidents */
430         if (U.pixelsize == 0) U.pixelsize = 1;
431         
432         BLF_default_dpi(U.pixelsize * U.dpi);
433         U.widget_unit = (U.pixelsize * U.dpi * 20 + 36) / 72;
434
435 }
436
437 int BKE_read_file(bContext *C, const char *filepath, ReportList *reports)
438 {
439         BlendFileData *bfd;
440         int retval = BKE_READ_FILE_OK;
441
442         if (strstr(filepath, BLENDER_STARTUP_FILE) == NULL) /* don't print user-pref loading */
443                 printf("read blend: %s\n", filepath);
444
445         bfd = BLO_read_from_file(filepath, reports);
446         if (bfd) {
447                 if (bfd->user) retval = BKE_READ_FILE_OK_USERPREFS;
448                 
449                 if (0 == handle_subversion_warning(bfd->main, reports)) {
450                         free_main(bfd->main);
451                         MEM_freeN(bfd);
452                         bfd = NULL;
453                         retval = BKE_READ_FILE_FAIL;
454                 }
455                 else
456                         setup_app_data(C, bfd, filepath);  // frees BFD
457         }
458         else
459                 BKE_reports_prependf(reports, "Loading '%s' failed: ", filepath);
460                 
461         return (bfd ? retval : BKE_READ_FILE_FAIL);
462 }
463
464 int BKE_read_file_from_memory(bContext *C, char *filebuf, int filelength, ReportList *reports)
465 {
466         BlendFileData *bfd;
467
468         bfd = BLO_read_from_memory(filebuf, filelength, reports);
469         if (bfd)
470                 setup_app_data(C, bfd, "<memory2>");
471         else
472                 BKE_reports_prepend(reports, "Loading failed: ");
473
474         return (bfd ? 1 : 0);
475 }
476
477 /* memfile is the undo buffer */
478 int BKE_read_file_from_memfile(bContext *C, MemFile *memfile, ReportList *reports)
479 {
480         BlendFileData *bfd;
481
482         bfd = BLO_read_from_memfile(CTX_data_main(C), G.main->name, memfile, reports);
483         if (bfd) {
484                 /* remove the unused screens and wm */
485                 while (bfd->main->wm.first)
486                         BKE_libblock_free(&bfd->main->wm, bfd->main->wm.first);
487                 while (bfd->main->screen.first)
488                         BKE_libblock_free(&bfd->main->screen, bfd->main->screen.first);
489                 
490                 setup_app_data(C, bfd, "<memory1>");
491         }
492         else
493                 BKE_reports_prepend(reports, "Loading failed: ");
494
495         return (bfd ? 1 : 0);
496 }
497
498 /* only read the userdef from a .blend */
499 int BKE_read_file_userdef(const char *filepath, ReportList *reports)
500 {
501         BlendFileData *bfd;
502         int retval = 0;
503         
504         bfd = BLO_read_from_file(filepath, reports);
505         if (bfd->user) {
506                 retval = BKE_READ_FILE_OK_USERPREFS;
507                 
508                 /* only here free userdef themes... */
509                 BKE_userdef_free();
510                 
511                 U = *bfd->user;
512                 MEM_freeN(bfd->user);
513         }
514         free_main(bfd->main);
515         MEM_freeN(bfd);
516         
517         return retval;
518 }
519
520 /* only write the userdef in a .blend */
521 int BKE_write_file_userdef(const char *filepath, ReportList *reports)
522 {
523         Main *mainb = MEM_callocN(sizeof(Main), "empty main");
524         int retval = 0;
525         
526         if (BLO_write_file(mainb, filepath, G_FILE_USERPREFS, reports, NULL)) {
527                 retval = 1;
528         }
529         
530         MEM_freeN(mainb);
531         
532         return retval;
533 }
534
535 /* *****************  testing for break ************* */
536
537 static void (*blender_test_break_cb)(void) = NULL;
538
539 void set_blender_test_break_cb(void (*func)(void) )
540 {
541         blender_test_break_cb = func;
542 }
543
544
545 int blender_test_break(void)
546 {
547         if (!G.background) {
548                 if (blender_test_break_cb)
549                         blender_test_break_cb();
550         }
551         
552         return (G.is_break == TRUE);
553 }
554
555
556 /* ***************** GLOBAL UNDO *************** */
557
558 #define UNDO_DISK   0
559
560 typedef struct UndoElem {
561         struct UndoElem *next, *prev;
562         char str[FILE_MAX];
563         char name[BKE_UNDO_STR_MAX];
564         MemFile memfile;
565         uintptr_t undosize;
566 } UndoElem;
567
568 static ListBase undobase = {NULL, NULL};
569 static UndoElem *curundo = NULL;
570
571
572 static int read_undosave(bContext *C, UndoElem *uel)
573 {
574         char mainstr[sizeof(G.main->name)];
575         int success = 0, fileflags;
576         
577         /* This is needed so undoing/redoing doesn't crash with threaded previews going */
578         WM_jobs_kill_all_except(CTX_wm_manager(C), CTX_wm_screen(C));
579
580         BLI_strncpy(mainstr, G.main->name, sizeof(mainstr));    /* temporal store */
581
582         fileflags = G.fileflags;
583         G.fileflags |= G_FILE_NO_UI;
584
585         if (UNDO_DISK) 
586                 success = (BKE_read_file(C, uel->str, NULL) != BKE_READ_FILE_FAIL);
587         else
588                 success = BKE_read_file_from_memfile(C, &uel->memfile, NULL);
589
590         /* restore */
591         BLI_strncpy(G.main->name, mainstr, sizeof(G.main->name)); /* restore */
592         G.fileflags = fileflags;
593
594         if (success) {
595                 /* important not to update time here, else non keyed tranforms are lost */
596                 DAG_on_visible_update(G.main, FALSE);
597         }
598
599         return success;
600 }
601
602 /* name can be a dynamic string */
603 void BKE_write_undo(bContext *C, const char *name)
604 {
605         uintptr_t maxmem, totmem, memused;
606         int nr /*, success */ /* UNUSED */;
607         UndoElem *uel;
608         
609         if ((U.uiflag & USER_GLOBALUNDO) == 0) {
610                 return;
611         }
612
613         if (U.undosteps == 0) {
614                 return;
615         }
616         
617         /* remove all undos after (also when curundo == NULL) */
618         while (undobase.last != curundo) {
619                 uel = undobase.last;
620                 BLI_remlink(&undobase, uel);
621                 BLO_free_memfile(&uel->memfile);
622                 MEM_freeN(uel);
623         }
624         
625         /* make new */
626         curundo = uel = MEM_callocN(sizeof(UndoElem), "undo file");
627         BLI_strncpy(uel->name, name, sizeof(uel->name));
628         BLI_addtail(&undobase, uel);
629         
630         /* and limit amount to the maximum */
631         nr = 0;
632         uel = undobase.last;
633         while (uel) {
634                 nr++;
635                 if (nr == U.undosteps) break;
636                 uel = uel->prev;
637         }
638         if (uel) {
639                 while (undobase.first != uel) {
640                         UndoElem *first = undobase.first;
641                         BLI_remlink(&undobase, first);
642                         /* the merge is because of compression */
643                         BLO_merge_memfile(&first->memfile, &first->next->memfile);
644                         MEM_freeN(first);
645                 }
646         }
647
648
649         /* disk save version */
650         if (UNDO_DISK) {
651                 static int counter = 0;
652                 char filepath[FILE_MAX];
653                 char numstr[32];
654                 int fileflags = G.fileflags & ~(G_FILE_HISTORY); /* don't do file history on undo */
655
656                 /* calculate current filepath */
657                 counter++;
658                 counter = counter % U.undosteps;
659         
660                 BLI_snprintf(numstr, sizeof(numstr), "%d.blend", counter);
661                 BLI_make_file_string("/", filepath, BLI_temporary_dir(), numstr);
662         
663                 /* success = */ /* UNUSED */ BLO_write_file(CTX_data_main(C), filepath, fileflags, NULL, NULL);
664                 
665                 BLI_strncpy(curundo->str, filepath, sizeof(curundo->str));
666         }
667         else {
668                 MemFile *prevfile = NULL;
669                 
670                 if (curundo->prev) prevfile = &(curundo->prev->memfile);
671                 
672                 memused = MEM_get_memory_in_use();
673                 /* success = */ /* UNUSED */ BLO_write_file_mem(CTX_data_main(C), prevfile, &curundo->memfile, G.fileflags);
674                 curundo->undosize = MEM_get_memory_in_use() - memused;
675         }
676
677         if (U.undomemory != 0) {
678                 /* limit to maximum memory (afterwards, we can't know in advance) */
679                 totmem = 0;
680                 maxmem = ((uintptr_t)U.undomemory) * 1024 * 1024;
681
682                 /* keep at least two (original + other) */
683                 uel = undobase.last;
684                 while (uel && uel->prev) {
685                         totmem += uel->undosize;
686                         if (totmem > maxmem) break;
687                         uel = uel->prev;
688                 }
689
690                 if (uel) {
691                         if (uel->prev && uel->prev->prev)
692                                 uel = uel->prev;
693
694                         while (undobase.first != uel) {
695                                 UndoElem *first = undobase.first;
696                                 BLI_remlink(&undobase, first);
697                                 /* the merge is because of compression */
698                                 BLO_merge_memfile(&first->memfile, &first->next->memfile);
699                                 MEM_freeN(first);
700                         }
701                 }
702         }
703 }
704
705 /* 1 = an undo, -1 is a redo. we have to make sure 'curundo' remains at current situation */
706 void BKE_undo_step(bContext *C, int step)
707 {
708         
709         if (step == 0) {
710                 read_undosave(C, curundo);
711         }
712         else if (step == 1) {
713                 /* curundo should never be NULL, after restart or load file it should call undo_save */
714                 if (curundo == NULL || curundo->prev == NULL) {
715                         // XXX error("No undo available");
716                 }
717                 else {
718                         if (G.debug & G_DEBUG) printf("undo %s\n", curundo->name);
719                         curundo = curundo->prev;
720                         read_undosave(C, curundo);
721                 }
722         }
723         else {
724                 /* curundo has to remain current situation! */
725                 
726                 if (curundo == NULL || curundo->next == NULL) {
727                         // XXX error("No redo available");
728                 }
729                 else {
730                         read_undosave(C, curundo->next);
731                         curundo = curundo->next;
732                         if (G.debug & G_DEBUG) printf("redo %s\n", curundo->name);
733                 }
734         }
735 }
736
737 void BKE_reset_undo(void)
738 {
739         UndoElem *uel;
740         
741         uel = undobase.first;
742         while (uel) {
743                 BLO_free_memfile(&uel->memfile);
744                 uel = uel->next;
745         }
746         
747         BLI_freelistN(&undobase);
748         curundo = NULL;
749 }
750
751 /* based on index nr it does a restore */
752 void BKE_undo_number(bContext *C, int nr)
753 {
754         curundo = BLI_findlink(&undobase, nr);
755         BKE_undo_step(C, 0);
756 }
757
758 /* go back to the last occurance of name in stack */
759 void BKE_undo_name(bContext *C, const char *name)
760 {
761         UndoElem *uel = BLI_rfindstring(&undobase, name, offsetof(UndoElem, name));
762
763         if (uel && uel->prev) {
764                 curundo = uel->prev;
765                 BKE_undo_step(C, 0);
766         }
767 }
768
769 /* name optional */
770 int BKE_undo_valid(const char *name)
771 {
772         if (name) {
773                 UndoElem *uel = BLI_rfindstring(&undobase, name, offsetof(UndoElem, name));
774                 return uel && uel->prev;
775         }
776         
777         return undobase.last != undobase.first;
778 }
779
780 /* get name of undo item, return null if no item with this index */
781 /* if active pointer, set it to 1 if true */
782 const char *BKE_undo_get_name(int nr, int *active)
783 {
784         UndoElem *uel = BLI_findlink(&undobase, nr);
785         
786         if (active) *active = 0;
787         
788         if (uel) {
789                 if (active && uel == curundo)
790                         *active = 1;
791                 return uel->name;
792         }
793         return NULL;
794 }
795
796 char *BKE_undo_menu_string(void)
797 {
798         UndoElem *uel;
799         DynStr *ds = BLI_dynstr_new();
800         char *menu;
801
802         BLI_dynstr_append(ds, "Global Undo History %t");
803         
804         for (uel = undobase.first; uel; uel = uel->next) {
805                 BLI_dynstr_append(ds, "|");
806                 BLI_dynstr_append(ds, uel->name);
807         }
808
809         menu = BLI_dynstr_get_cstring(ds);
810         BLI_dynstr_free(ds);
811
812         return menu;
813 }
814
815 /* saves .blend using undo buffer, returns 1 == success */
816 int BKE_undo_save_file(const char *filename)
817 {
818         UndoElem *uel;
819         MemFileChunk *chunk;
820         const int flag = O_BINARY + O_WRONLY + O_CREAT + O_TRUNC + O_EXCL;
821         int file;
822
823         if ((U.uiflag & USER_GLOBALUNDO) == 0) {
824                 return 0;
825         }
826
827         uel = curundo;
828         if (uel == NULL) {
829                 fprintf(stderr, "No undo buffer to save recovery file\n");
830                 return 0;
831         }
832
833         /* first try create the file, if it exists call without 'O_CREAT',
834          * to avoid writing to a symlink - use 'O_EXCL' (CVE-2008-1103) */
835         errno = 0;
836         file = BLI_open(filename, flag, 0666);
837         if (file == -1) {
838                 if (errno == EEXIST) {
839                         errno = 0;
840                         file = BLI_open(filename, flag & ~O_CREAT, 0666);
841                 }
842         }
843
844         if (file == -1) {
845                 fprintf(stderr, "Unable to save '%s': %s\n",
846                         filename, errno ? strerror(errno) : "Unknown error opening file");
847                 return 0;
848         }
849
850         for (chunk = uel->memfile.chunks.first; chunk; chunk = chunk->next) {
851                 if (write(file, chunk->buf, chunk->size) != chunk->size) {
852                         break;
853                 }
854         }
855         
856         close(file);
857         
858         if (chunk) {
859                 fprintf(stderr, "Unable to save '%s': %s\n",
860                         filename, errno ? strerror(errno) : "Unknown error writing file");
861                 return 0;
862         }
863         return 1;
864 }
865
866 /* sets curscene */
867 Main *BKE_undo_get_main(Scene **scene)
868 {
869         Main *mainp = NULL;
870         BlendFileData *bfd = BLO_read_from_memfile(G.main, G.main->name, &curundo->memfile, NULL);
871         
872         if (bfd) {
873                 mainp = bfd->main;
874                 if (scene)
875                         *scene = bfd->curscene;
876                 
877                 MEM_freeN(bfd);
878         }
879         
880         return mainp;
881 }
882
883 /* ************** copy paste .blend, partial saves ********** */
884
885 /* assumes data is in G.main */
886
887 void BKE_copybuffer_begin(void)
888 {
889         /* set all id flags to zero; */
890         flag_all_listbases_ids(LIB_NEED_EXPAND | LIB_DOIT, 0);
891 }
892
893 void BKE_copybuffer_tag_ID(ID *id)
894 {
895         id->flag |= LIB_NEED_EXPAND | LIB_DOIT;
896 }
897
898 static void copybuffer_doit(void *UNUSED(handle), Main *UNUSED(bmain), void *vid)
899 {
900         if (vid) {
901                 ID *id = vid;
902                 /* only tag for need-expand if not done, prevents eternal loops */
903                 if ((id->flag & LIB_DOIT) == 0)
904                         id->flag |= LIB_NEED_EXPAND | LIB_DOIT;
905         }
906 }
907
908 /* frees main in end */
909 int BKE_copybuffer_save(char *filename, ReportList *reports)
910 {
911         Main *mainb = MEM_callocN(sizeof(Main), "copybuffer");
912         ListBase *lbarray[MAX_LIBARRAY], *fromarray[MAX_LIBARRAY];
913         int a, retval;
914         
915         BLO_main_expander(copybuffer_doit);
916         BLO_expand_main(NULL, G.main);
917         
918         /* move over all tagged blocks */
919         set_listbasepointers(G.main, fromarray);
920         a = set_listbasepointers(mainb, lbarray);
921         while (a--) {
922                 ID *id, *nextid;
923                 ListBase *lb1 = lbarray[a], *lb2 = fromarray[a];
924                 
925                 for (id = lb2->first; id; id = nextid) {
926                         nextid = id->next;
927                         if (id->flag & LIB_DOIT) {
928                                 BLI_remlink(lb2, id);
929                                 BLI_addtail(lb1, id);
930                         }
931                 }
932         }
933         
934         
935         /* save the buffer */
936         retval = BLO_write_file(mainb, filename, 0, reports, NULL);
937         
938         /* move back the main, now sorted again */
939         set_listbasepointers(G.main, lbarray);
940         a = set_listbasepointers(mainb, fromarray);
941         while (a--) {
942                 ID *id;
943                 ListBase *lb1 = lbarray[a], *lb2 = fromarray[a];
944                 
945                 while (lb2->first) {
946                         id = lb2->first;
947                         BLI_remlink(lb2, id);
948                         BLI_addtail(lb1, id);
949                         id_sort_by_name(lb1, id);
950                 }
951         }
952         
953         MEM_freeN(mainb);
954         
955         /* set id flag to zero; */
956         flag_all_listbases_ids(LIB_NEED_EXPAND | LIB_DOIT, 0);
957         
958         return retval;
959 }
960
961 /* return success (1) */
962 int BKE_copybuffer_paste(bContext *C, char *libname, ReportList *reports)
963 {
964         Main *bmain = CTX_data_main(C);
965         Scene *scene = CTX_data_scene(C);
966         Main *mainl = NULL;
967         Library *lib;
968         BlendHandle *bh;
969                 
970         bh = BLO_blendhandle_from_file(libname, reports);
971         
972         if (bh == NULL) {
973                 /* error reports will have been made by BLO_blendhandle_from_file() */
974                 return 0;
975         }
976
977         BKE_scene_base_deselect_all(scene);
978         
979         /* tag everything, all untagged data can be made local
980          * its also generally useful to know what is new
981          *
982          * take extra care flag_all_listbases_ids(LIB_LINK_TAG, 0) is called after! */
983         flag_all_listbases_ids(LIB_PRE_EXISTING, 1);
984         
985         /* here appending/linking starts */
986         mainl = BLO_library_append_begin(bmain, &bh, libname);
987         
988         BLO_library_append_all(mainl, bh);
989
990         BLO_library_append_end(C, mainl, &bh, 0, 0);
991         
992         /* mark all library linked objects to be updated */
993         recalc_all_library_objects(bmain);
994         IMB_colormanagement_check_file_config(bmain);
995         
996         /* append, rather than linking */
997         lib = BLI_findstring(&bmain->library, libname, offsetof(Library, filepath));
998         BKE_library_make_local(bmain, lib, true);
999         
1000         /* important we unset, otherwise these object wont
1001          * link into other scenes from this blend file */
1002         flag_all_listbases_ids(LIB_PRE_EXISTING, 0);
1003         
1004         /* recreate dependency graph to include new objects */
1005         DAG_relations_tag_update(bmain);
1006         
1007         BLO_blendhandle_close(bh);
1008         /* remove library... */
1009         
1010         return 1;
1011 }