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