fix for crash when a register script sets material colors, also made some changes...
[blender.git] / source / blender / windowmanager / intern / wm_files.c
1 /**
2  * $Id$
3  *
4  * ***** BEGIN GPL LICENSE BLOCK *****
5  *
6  * This program is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU General Public License
8  * as published by the Free Software Foundation; either version 2
9  * of the License, or (at your option) any later version. 
10  *
11  * This program is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  * GNU General Public License for more details.
15  *
16  * You should have received a copy of the GNU General Public License
17  * along with this program; if not, write to the Free Software Foundation,
18  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
19  *
20  * The Original Code is Copyright (C) 2001-2002 by NaN Holding BV.
21  * All rights reserved.
22  *
23  * Contributor(s): Blender Foundation 2007
24  *
25  * ***** END GPL LICENSE BLOCK *****
26  */
27
28         /* placed up here because of crappy
29          * winsock stuff.
30          */
31 #include <stdio.h>
32 #include <string.h>
33
34 #ifdef WIN32
35 #include <windows.h> /* need to include windows.h so _WIN32_IE is defined  */
36 #ifndef _WIN32_IE
37 #define _WIN32_IE 0x0400 /* minimal requirements for SHGetSpecialFolderPath on MINGW MSVC has this defined already */
38 #endif
39 #include <shlobj.h> /* for SHGetSpecialFolderPath, has to be done before BLI_winstuff because 'near' is disabled through BLI_windstuff */
40 #include <process.h> /* getpid */
41 #include "BLI_winstuff.h"
42 #else
43 #include <unistd.h> /* getpid */
44 #endif
45
46 #include "MEM_guardedalloc.h"
47 #include "MEM_CacheLimiterC-Api.h"
48
49 #include "BLI_blenlib.h"
50 #include "BLI_linklist.h"
51
52 #include "DNA_anim_types.h"
53 #include "DNA_ipo_types.h" // XXX old animation system
54 #include "DNA_object_types.h"
55 #include "DNA_space_types.h"
56 #include "DNA_userdef_types.h"
57 #include "DNA_scene_types.h"
58 #include "DNA_screen_types.h"
59 #include "DNA_windowmanager_types.h"
60
61 #include "BKE_blender.h"
62 #include "BKE_context.h"
63 #include "BKE_DerivedMesh.h"
64 #include "BKE_exotic.h"
65 #include "BKE_font.h"
66 #include "BKE_global.h"
67 #include "BKE_library.h"
68 #include "BKE_main.h"
69 #include "BKE_packedFile.h"
70 #include "BKE_report.h"
71 #include "BKE_sound.h"
72 #include "BKE_texture.h"
73 #include "BKE_utildefines.h"
74
75 #include "BLO_readfile.h"
76 #include "BLO_writefile.h"
77
78 #include "RNA_access.h"
79
80 #include "ED_datafiles.h"
81 #include "ED_object.h"
82 #include "ED_screen.h"
83 #include "ED_sculpt.h"
84 #include "ED_util.h"
85
86 #include "GHOST_C-api.h"
87
88 #include "UI_interface.h"
89
90 #include "GPU_draw.h"
91
92 #include "BPY_extern.h"
93
94 #include "WM_api.h"
95 #include "WM_types.h"
96 #include "wm.h"
97 #include "wm_window.h"
98 #include "wm_event_system.h"
99
100 static void writeBlog(void);
101
102 /* To be able to read files without windows closing, opening, moving 
103    we try to prepare for worst case:
104    - active window gets active screen from file 
105    - restoring the screens from non-active windows 
106    Best case is all screens match, in that case they get assigned to proper window  
107 */
108 static void wm_window_match_init(bContext *C, ListBase *wmlist)
109 {
110         wmWindowManager *wm= G.main->wm.first;
111         wmWindow *win, *active_win;
112         
113         *wmlist= G.main->wm;
114         G.main->wm.first= G.main->wm.last= NULL;
115         
116         active_win = CTX_wm_window(C);
117
118         /* first wrap up running stuff */
119         /* code copied from wm_init_exit.c */
120         for(wm= wmlist->first; wm; wm= wm->id.next) {
121                 
122                 WM_jobs_stop_all(wm);
123                 
124                 for(win= wm->windows.first; win; win= win->next) {
125                 
126                         CTX_wm_window_set(C, win);      /* needed by operator close callbacks */
127                         WM_event_remove_handlers(C, &win->handlers);
128                         WM_event_remove_handlers(C, &win->modalhandlers);
129                         ED_screen_exit(C, win, win->screen);
130                 }
131         }
132         
133         /* reset active window */
134         CTX_wm_window_set(C, active_win);
135
136         ED_editors_exit(C);
137         
138 return; 
139         if(wm==NULL) return;
140         if(G.fileflags & G_FILE_NO_UI) return;
141         
142         /* we take apart the used screens from non-active window */
143         for(win= wm->windows.first; win; win= win->next) {
144                 BLI_strncpy(win->screenname, win->screen->id.name, MAX_ID_NAME);
145                 if(win!=wm->winactive) {
146                         BLI_remlink(&G.main->screen, win->screen);
147                         //BLI_addtail(screenbase, win->screen);
148                 }
149         }
150 }
151
152 /* match old WM with new, 4 cases:
153   1- no current wm, no read wm: make new default
154   2- no current wm, but read wm: that's OK, do nothing
155   3- current wm, but not in file: try match screen names
156   4- current wm, and wm in file: try match ghostwin 
157 */
158
159 static void wm_window_match_do(bContext *C, ListBase *oldwmlist)
160 {
161         wmWindowManager *oldwm, *wm;
162         wmWindow *oldwin, *win;
163         
164         /* cases 1 and 2 */
165         if(oldwmlist->first==NULL) {
166                 if(G.main->wm.first); /* nothing todo */
167                 else
168                         wm_add_default(C);
169         }
170         else {
171                 /* cases 3 and 4 */
172                 
173                 /* we've read file without wm..., keep current one entirely alive */
174                 if(G.main->wm.first==NULL) {
175                         bScreen *screen= CTX_wm_screen(C);
176                         
177                         /* match oldwm to new dbase, only old files */
178                         
179                         for(wm= oldwmlist->first; wm; wm= wm->id.next) {
180                                 
181                                 for(win= wm->windows.first; win; win= win->next) {
182                                         /* all windows get active screen from file */
183                                         if(screen->winid==0)
184                                                 win->screen= screen;
185                                         else 
186                                                 win->screen= ED_screen_duplicate(win, screen);
187                                         
188                                         BLI_strncpy(win->screenname, win->screen->id.name+2, 21);
189                                         win->screen->winid= win->winid;
190                                 }
191                         }
192                         
193                         G.main->wm= *oldwmlist;
194                         
195                         /* screens were read from file! */
196                         ED_screens_initialize(G.main->wm.first);
197                 }
198                 else {
199                         /* what if old was 3, and loaded 1? */
200                         /* this code could move to setup_appdata */
201                         oldwm= oldwmlist->first;
202                         wm= G.main->wm.first;
203
204                         /* ensure making new keymaps and set space types */
205                         wm->initialized= 0;
206                         wm->winactive= NULL;
207                         
208                         /* only first wm in list has ghostwins */
209                         for(win= wm->windows.first; win; win= win->next) {
210                                 for(oldwin= oldwm->windows.first; oldwin; oldwin= oldwin->next) {
211                                         
212                                         if(oldwin->winid == win->winid ) {
213                                                 win->ghostwin= oldwin->ghostwin;
214                                                 win->active= oldwin->active;
215                                                 if(win->active)
216                                                         wm->winactive= win;
217
218                                                 GHOST_SetWindowUserData(win->ghostwin, win);    /* pointer back */
219                                                 oldwin->ghostwin= NULL;
220                                                 
221                                                 win->eventstate= oldwin->eventstate;
222                                                 oldwin->eventstate= NULL;
223                                                 
224                                                 /* ensure proper screen rescaling */
225                                                 win->sizex= oldwin->sizex;
226                                                 win->sizey= oldwin->sizey;
227                                                 win->posx= oldwin->posx;
228                                                 win->posy= oldwin->posy;
229                                         }
230                                 }
231                         }
232                         wm_close_and_free_all(C, oldwmlist);
233                 }
234         }
235 }
236
237 /* in case UserDef was read, we re-initialize all, and do versioning */
238 static void wm_init_userdef(bContext *C)
239 {
240         extern char btempdir[];
241
242         UI_init_userdef();
243         MEM_CacheLimiter_set_maximum(U.memcachelimit * 1024 * 1024);
244         sound_init(CTX_data_main(C));
245
246         /* set the python auto-execute setting from user prefs */
247         if (U.flag & USER_SCRIPT_AUTOEXEC_DISABLE)      G.f &= ~G_SCRIPT_AUTOEXEC;
248         else                                                                            G.f |=  G_SCRIPT_AUTOEXEC;
249
250         if(U.tempdir[0]) strncpy(btempdir, U.tempdir, FILE_MAXDIR+FILE_MAXFILE);
251 }
252
253 void WM_read_file(bContext *C, char *name, ReportList *reports)
254 {
255         int retval;
256
257         /* first try to append data from exotic file formats... */
258         /* it throws error box when file doesnt exist and returns -1 */
259         /* note; it should set some error message somewhere... (ton) */
260         retval= BKE_read_exotic(CTX_data_scene(C), name);
261         
262         /* we didn't succeed, now try to read Blender file */
263         if (retval== 0) {
264                 int G_f= G.f;
265                 ListBase wmbase;
266
267                 /* put aside screens to match with persistant windows later */
268                 /* also exit screens and editors */
269                 wm_window_match_init(C, &wmbase); 
270                 
271                 retval= BKE_read_file(C, name, NULL, reports);
272                 G.save_over = 1;
273
274                 /* this flag is initialized by the operator but overwritten on read.
275                  * need to re-enable it here else drivers + registered scripts wont work. */
276                 if(G_f & G_SCRIPT_AUTOEXEC) G.f |= G_SCRIPT_AUTOEXEC;
277                 else                                            G.f &= ~G_SCRIPT_AUTOEXEC;
278
279                 /* match the read WM with current WM */
280                 wm_window_match_do(C, &wmbase);
281                 WM_check(C); /* opens window(s), checks keymaps */
282                 
283 // XXX          mainwindow_set_filename_to_title(G.main->name);
284
285                 if(retval==2) wm_init_userdef(C);       // in case a userdef is read from regular .blend
286                 
287                 if (retval!=0) {
288                         G.relbase_valid = 1;
289                         writeBlog();
290                 }
291
292 // XXX          undo_editmode_clear();
293                 BKE_reset_undo();
294                 BKE_write_undo(C, "original");  /* save current state */
295
296                 WM_event_add_notifier(C, NC_WM|ND_FILEREAD, NULL);
297 //              refresh_interface_font();
298
299                 CTX_wm_window_set(C, CTX_wm_manager(C)->windows.first);
300                 ED_editors_init(C);
301
302 #ifndef DISABLE_PYTHON
303                 /* run any texts that were loaded in and flagged as modules */
304                 BPY_load_user_modules(C);
305 #endif
306                 CTX_wm_window_set(C, NULL); /* exits queues */
307         }
308         else if(retval==1)
309                 BKE_write_undo(C, "Import file");
310         else if(retval == -1) {
311                 if(reports)
312                         BKE_reportf(reports, RPT_ERROR, "Can't read file \"%s\".", name);
313         }
314 }
315
316
317 /* called on startup,  (context entirely filled with NULLs) */
318 /* or called for 'New File' */
319 /* op can be NULL */
320 int WM_read_homefile(bContext *C, wmOperator *op)
321 {
322         ListBase wmbase;
323         char tstr[FILE_MAXDIR+FILE_MAXFILE], scestr[FILE_MAXDIR];
324         char *home= BLI_gethome();
325         int from_memory= op?RNA_boolean_get(op->ptr, "factory"):0;
326         int success;
327                 
328         BLI_clean(home);
329         
330         free_ttfont(); /* still weird... what does it here? */
331                 
332         G.relbase_valid = 0;
333         if (!from_memory) {
334                 BLI_make_file_string(G.sce, tstr, home, ".B25.blend");
335         }
336         strcpy(scestr, G.sce);  /* temporary store */
337         
338         /* prevent loading no UI */
339         G.fileflags &= ~G_FILE_NO_UI;
340         
341         /* put aside screens to match with persistant windows later */
342         wm_window_match_init(C, &wmbase); 
343         
344         if (!from_memory && BLI_exists(tstr)) {
345                 success = BKE_read_file(C, tstr, NULL, NULL);
346         } else {
347                 success = BKE_read_file_from_memory(C, datatoc_B_blend, datatoc_B_blend_size, NULL, NULL);
348                 if (wmbase.first == NULL) wm_clear_default_size(C);
349         }
350         
351         /* match the read WM with current WM */
352         wm_window_match_do(C, &wmbase); 
353         WM_check(C); /* opens window(s), checks keymaps */
354
355         strcpy(G.sce, scestr); /* restore */
356         
357         wm_init_userdef(C);
358         
359         /* When loading factory settings, the reset solid OpenGL lights need to be applied. */
360         if (!G.background) GPU_default_lights();
361         
362         /* XXX */
363         G.save_over = 0;        // start with save preference untitled.blend
364         G.fileflags &= ~G_FILE_AUTOPLAY;        /*  disable autoplay in .B.blend... */
365 //      mainwindow_set_filename_to_title("");   // empty string re-initializes title to "Blender"
366         
367 //      refresh_interface_font();
368         
369 //      undo_editmode_clear();
370         BKE_reset_undo();
371         BKE_write_undo(C, "original");  /* save current state */
372         
373         WM_event_add_notifier(C, NC_WM|ND_FILEREAD, NULL);
374         CTX_wm_window_set(C, NULL); /* exits queues */
375
376         return OPERATOR_FINISHED;
377 }
378
379
380 void read_Blog(void)
381 {
382         char name[FILE_MAX];
383         LinkNode *l, *lines;
384         struct RecentFile *recent;
385         char *line;
386         int num;
387
388         BLI_make_file_string("/", name, BLI_gethome(), ".Blog");
389         lines= BLI_read_file_as_lines(name);
390
391         G.recent_files.first = G.recent_files.last = NULL;
392
393         /* read list of recent opend files from .Blog to memory */
394         for (l= lines, num= 0; l && (num<U.recent_files); l= l->next) {
395                 line = l->link;
396                 if (line[0] && BLI_exists(line)) {
397                         if (num==0) 
398                                 strcpy(G.sce, line); /* note: this seems highly dodgy since the file isnt actually read. please explain. - campbell */
399                         
400                         recent = (RecentFile*)MEM_mallocN(sizeof(RecentFile),"RecentFile");
401                         BLI_addtail(&(G.recent_files), recent);
402                         recent->filename = (char*)MEM_mallocN(sizeof(char)*(strlen(line)+1), "name of file");
403                         recent->filename[0] = '\0';
404                         
405                         strcpy(recent->filename, line);
406                         num++;
407                 }
408         }
409
410         if(G.sce[0] == 0)
411                 BLI_make_file_string("/", G.sce, BLI_gethome(), "untitled.blend");
412         
413         BLI_free_file_lines(lines);
414
415 }
416
417 static void writeBlog(void)
418 {
419         struct RecentFile *recent, *next_recent;
420         char name[FILE_MAXDIR+FILE_MAXFILE];
421         FILE *fp;
422         int i;
423
424         BLI_make_file_string("/", name, BLI_gethome(), ".Blog");
425
426         recent = G.recent_files.first;
427         /* refresh .Blog of recent opened files, when current file was changed */
428         if(!(recent) || (strcmp(recent->filename, G.sce)!=0)) {
429                 fp= fopen(name, "w");
430                 if (fp) {
431                         /* add current file to the beginning of list */
432                         recent = (RecentFile*)MEM_mallocN(sizeof(RecentFile),"RecentFile");
433                         recent->filename = (char*)MEM_mallocN(sizeof(char)*(strlen(G.sce)+1), "name of file");
434                         recent->filename[0] = '\0';
435                         strcpy(recent->filename, G.sce);
436                         BLI_addhead(&(G.recent_files), recent);
437                         /* write current file to .Blog */
438                         fprintf(fp, "%s\n", recent->filename);
439                         recent = recent->next;
440                         i=1;
441                         /* write rest of recent opened files to .Blog */
442                         while((i<U.recent_files) && (recent)){
443                                 /* this prevents to have duplicities in list */
444                                 if (strcmp(recent->filename, G.sce)!=0) {
445                                         fprintf(fp, "%s\n", recent->filename);
446                                         recent = recent->next;
447                                 }
448                                 else {
449                                         next_recent = recent->next;
450                                         MEM_freeN(recent->filename);
451                                         BLI_freelinkN(&(G.recent_files), recent);
452                                         recent = next_recent;
453                                 }
454                                 i++;
455                         }
456                         fclose(fp);
457                 }
458         }
459 }
460
461 static void do_history(char *name, ReportList *reports)
462 {
463         char tempname1[FILE_MAXDIR+FILE_MAXFILE], tempname2[FILE_MAXDIR+FILE_MAXFILE];
464         int hisnr= U.versions;
465         
466         if(U.versions==0) return;
467         if(strlen(name)<2) return;
468                 
469         while(hisnr > 1) {
470                 sprintf(tempname1, "%s%d", name, hisnr-1);
471                 sprintf(tempname2, "%s%d", name, hisnr);
472         
473                 if(BLI_rename(tempname1, tempname2))
474                         BKE_report(reports, RPT_ERROR, "Unable to make version backup");
475                         
476                 hisnr--;
477         }
478                 
479         /* is needed when hisnr==1 */
480         sprintf(tempname1, "%s%d", name, hisnr);
481         
482         if(BLI_rename(name, tempname1))
483                 BKE_report(reports, RPT_ERROR, "Unable to make version backup");
484 }
485
486 void WM_write_file(bContext *C, char *target, int fileflags, ReportList *reports)
487 {
488         Library *li;
489         int len;
490         char di[FILE_MAX];
491         
492         len = strlen(target);
493         
494         if (len == 0) return;
495         if (len >= FILE_MAX) {
496                 BKE_report(reports, RPT_ERROR, "Path too long, cannot save");
497                 return;
498         }
499  
500         /* send the OnSave event */
501         for (li= G.main->library.first; li; li= li->id.next) {
502                 if (BLI_streq(li->name, target)) {
503                         BKE_report(reports, RPT_ERROR, "Cannot overwrite used library");
504                         return;
505                 }
506         }
507         
508         if (!BLO_has_bfile_extension(target) && (len+6 < FILE_MAX)) {
509                 sprintf(di, "%s.blend", target);
510         } else {
511                 strcpy(di, target);
512         }
513
514 //      if (BLI_exists(di)) {
515 // XXX          if(!saveover(di))
516 // XXX                  return; 
517 //      }
518         
519         if (G.fileflags & G_AUTOPACK) {
520                 packAll(G.main, reports);
521         }
522         
523         ED_object_exit_editmode(C, EM_DO_UNDO);
524         ED_sculpt_force_update(C);
525
526         do_history(di, reports);
527         
528         if (BLO_write_file(CTX_data_main(C), di, fileflags, reports)) {
529                 strcpy(G.sce, di);
530                 G.relbase_valid = 1;
531                 strcpy(G.main->name, di);       /* is guaranteed current file */
532
533                 G.save_over = 1; /* disable untitled.blend convention */
534
535                 if(fileflags & G_FILE_COMPRESS) G.fileflags |= G_FILE_COMPRESS;
536                 else G.fileflags &= ~G_FILE_COMPRESS;
537                 
538                 if(fileflags & G_FILE_AUTOPLAY) G.fileflags |= G_FILE_AUTOPLAY;
539                 else G.fileflags &= ~G_FILE_AUTOPLAY;
540
541                 writeBlog();
542         }
543
544 // XXX  waitcursor(0);
545 }
546
547 /* operator entry */
548 int WM_write_homefile(bContext *C, wmOperator *op)
549 {
550         wmWindowManager *wm= CTX_wm_manager(C);
551         wmWindow *win= CTX_wm_window(C);
552         char tstr[FILE_MAXDIR+FILE_MAXFILE];
553         int fileflags;
554         
555         /* check current window and close it if temp */
556         if(win->screen->full == SCREENTEMP)
557                 wm_window_close(C, wm, win);
558         
559         BLI_make_file_string("/", tstr, BLI_gethome(), ".B25.blend");
560         
561         /*  force save as regular blend file */
562         fileflags = G.fileflags & ~(G_FILE_COMPRESS | G_FILE_AUTOPLAY | G_FILE_LOCK | G_FILE_SIGN);
563
564         BLO_write_file(CTX_data_main(C), tstr, fileflags, op->reports);
565         
566         G.save_over= 0;
567         
568         return OPERATOR_FINISHED;
569 }
570
571 /************************ autosave ****************************/
572
573 void wm_autosave_location(char *filename)
574 {
575         char pidstr[32];
576 #ifdef WIN32
577         char subdir[9];
578         char savedir[FILE_MAXDIR];
579 #endif
580
581         sprintf(pidstr, "%d.blend", abs(getpid()));
582         
583 #ifdef WIN32
584         if (!BLI_exists(U.tempdir)) {
585                 BLI_strncpy(subdir, "autosave", sizeof(subdir));
586                 BLI_make_file_string("/", savedir, BLI_gethome(), subdir);
587                 
588                 /* create a new autosave dir
589                  * function already checks for existence or not */
590                 BLI_recurdir_fileops(savedir);
591         
592                 BLI_make_file_string("/", filename, savedir, pidstr);
593                 return;
594         }
595 #endif
596         
597         BLI_make_file_string("/", filename, U.tempdir, pidstr);
598 }
599
600 void WM_autosave_init(wmWindowManager *wm)
601 {
602         wm_autosave_timer_ended(wm);
603
604         if(U.flag & USER_AUTOSAVE)
605                 wm->autosavetimer= WM_event_add_timer(wm, NULL, TIMERAUTOSAVE, U.savetime*60.0);
606 }
607
608 void wm_autosave_timer(const bContext *C, wmWindowManager *wm, wmTimer *wt)
609 {
610         wmWindow *win;
611         wmEventHandler *handler;
612         char filename[FILE_MAX];
613         int fileflags;
614
615         WM_event_remove_timer(wm, NULL, wm->autosavetimer);
616
617         /* if a modal operator is running, don't autosave, but try again in 10 seconds */
618         for(win=wm->windows.first; win; win=win->next) {
619                 for(handler=win->modalhandlers.first; handler; handler=handler->next) {
620                         if(handler->op) {
621                                 wm->autosavetimer= WM_event_add_timer(wm, NULL, TIMERAUTOSAVE, 10.0);
622                                 return;
623                         }
624                 }
625         }
626         
627         wm_autosave_location(filename);
628
629         /*  force save as regular blend file */
630         fileflags = G.fileflags & ~(G_FILE_COMPRESS|G_FILE_AUTOPLAY |G_FILE_LOCK|G_FILE_SIGN);
631
632         /* no error reporting to console */
633         BLO_write_file(CTX_data_main(C), filename, fileflags, NULL);
634
635         /* do timer after file write, just in case file write takes a long time */
636         wm->autosavetimer= WM_event_add_timer(wm, NULL, TIMERAUTOSAVE, U.savetime*60.0);
637 }
638
639 void wm_autosave_timer_ended(wmWindowManager *wm)
640 {
641         if(wm->autosavetimer) {
642                 WM_event_remove_timer(wm, NULL, wm->autosavetimer);
643                 wm->autosavetimer= NULL;
644         }
645 }
646
647 void wm_autosave_delete(void)
648 {
649         char filename[FILE_MAX];
650         
651         wm_autosave_location(filename);
652
653         if(BLI_exists(filename)) {
654                 char str[FILE_MAXDIR+FILE_MAXFILE];
655                 BLI_make_file_string("/", str, U.tempdir, "quit.blend");
656
657                 /* if global undo; remove tempsave, otherwise rename */
658                 if(U.uiflag & USER_GLOBALUNDO) BLI_delete(filename, 0, 0);
659                 else BLI_rename(filename, str);
660         }
661 }
662
663 void wm_autosave_read(bContext *C, ReportList *reports)
664 {
665         char filename[FILE_MAX];
666
667         wm_autosave_location(filename);
668         WM_read_file(C, filename, reports);
669 }
670