Internal node links are now cached in a per-node list, instead of being generated...
[blender.git] / source / blender / blenkernel / intern / world.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/world.c
29  *  \ingroup bke
30  */
31
32
33 #include <string.h>
34 #include <math.h>
35 #include "MEM_guardedalloc.h"
36
37 #include "DNA_world_types.h"
38 #include "DNA_scene_types.h"
39 #include "DNA_texture_types.h"
40
41 #include "BLI_listbase.h"
42 #include "BLI_utildefines.h"
43 #include "BLI_bpath.h"
44
45 #include "BKE_animsys.h"
46 #include "BKE_global.h"
47 #include "BKE_icons.h"
48 #include "BKE_library.h"
49 #include "BKE_library.h"
50 #include "BKE_main.h"
51 #include "BKE_node.h"
52 #include "BKE_world.h"
53
54 void BKE_world_free_ex(World *wrld, int do_id_user)
55 {
56         MTex *mtex;
57         int a;
58         
59         for (a = 0; a < MAX_MTEX; a++) {
60                 mtex = wrld->mtex[a];
61                 if (do_id_user && mtex && mtex->tex) mtex->tex->id.us--;
62                 if (mtex) MEM_freeN(mtex);
63         }
64         BKE_previewimg_free(&wrld->preview);
65
66         BKE_free_animdata((ID *)wrld);
67
68         /* is no lib link block, but world extension */
69         if (wrld->nodetree) {
70                 ntreeFreeTree_ex(wrld->nodetree, do_id_user);
71                 MEM_freeN(wrld->nodetree);
72         }
73
74         BKE_icon_delete((struct ID *)wrld);
75         wrld->id.icon_id = 0;
76 }
77
78 void BKE_world_free(World *wrld)
79 {
80         BKE_world_free_ex(wrld, TRUE);
81 }
82
83 World *add_world(const char *name)
84 {
85         Main *bmain = G.main;
86         World *wrld;
87
88         wrld = BKE_libblock_alloc(&bmain->world, ID_WO, name);
89         
90         wrld->horr = 0.05f;
91         wrld->horg = 0.05f;
92         wrld->horb = 0.05f;
93         wrld->zenr = 0.01f;
94         wrld->zeng = 0.01f;
95         wrld->zenb = 0.01f;
96         wrld->skytype = 0;
97         wrld->stardist = 15.0f;
98         wrld->starsize = 2.0f;
99
100         wrld->exp = 0.0f;
101         wrld->exposure = wrld->range = 1.0f;
102
103         wrld->aodist = 10.0f;
104         wrld->aosamp = 5;
105         wrld->aoenergy = 1.0f;
106         wrld->ao_env_energy = 1.0f;
107         wrld->ao_indirect_energy = 1.0f;
108         wrld->ao_indirect_bounces = 1;
109         wrld->aobias = 0.05f;
110         wrld->ao_samp_method = WO_AOSAMP_HAMMERSLEY;
111         wrld->ao_approx_error = 0.25f;
112         
113         wrld->preview = NULL;
114         wrld->miststa = 5.0f;
115         wrld->mistdist = 25.0f;
116
117         return wrld;
118 }
119
120 World *BKE_world_copy(World *wrld)
121 {
122         World *wrldn;
123         int a;
124         
125         wrldn = BKE_libblock_copy(&wrld->id);
126         
127         for (a = 0; a < MAX_MTEX; a++) {
128                 if (wrld->mtex[a]) {
129                         wrldn->mtex[a] = MEM_mallocN(sizeof(MTex), "BKE_world_copy");
130                         memcpy(wrldn->mtex[a], wrld->mtex[a], sizeof(MTex));
131                         id_us_plus((ID *)wrldn->mtex[a]->tex);
132                 }
133         }
134
135         if (wrld->nodetree) {
136                 wrldn->nodetree = ntreeCopyTree(wrld->nodetree);
137         }
138         
139         if (wrld->preview)
140                 wrldn->preview = BKE_previewimg_copy(wrld->preview);
141
142         return wrldn;
143 }
144
145 World *localize_world(World *wrld)
146 {
147         World *wrldn;
148         int a;
149         
150         wrldn = BKE_libblock_copy(&wrld->id);
151         BLI_remlink(&G.main->world, wrldn);
152         
153         for (a = 0; a < MAX_MTEX; a++) {
154                 if (wrld->mtex[a]) {
155                         wrldn->mtex[a] = MEM_mallocN(sizeof(MTex), "localize_world");
156                         memcpy(wrldn->mtex[a], wrld->mtex[a], sizeof(MTex));
157                         /* free world decrements */
158                         id_us_plus((ID *)wrldn->mtex[a]->tex);
159                 }
160         }
161
162         if (wrld->nodetree)
163                 wrldn->nodetree = ntreeLocalize(wrld->nodetree);
164         
165         wrldn->preview = NULL;
166         
167         return wrldn;
168 }
169
170 void BKE_world_make_local(World *wrld)
171 {
172         Main *bmain = G.main;
173         Scene *sce;
174         int is_local = FALSE, is_lib = FALSE;
175
176         /* - only lib users: do nothing
177          * - only local users: set flag
178          * - mixed: make copy
179          */
180         
181         if (wrld->id.lib == NULL) return;
182         if (wrld->id.us == 1) {
183                 id_clear_lib_data(bmain, &wrld->id);
184                 return;
185         }
186         
187         for (sce = bmain->scene.first; sce && ELEM(FALSE, is_lib, is_local); sce = sce->id.next) {
188                 if (sce->world == wrld) {
189                         if (sce->id.lib) is_lib = TRUE;
190                         else is_local = TRUE;
191                 }
192         }
193
194         if (is_local && is_lib == FALSE) {
195                 id_clear_lib_data(bmain, &wrld->id);
196         }
197         else if (is_local && is_lib) {
198                 World *wrld_new = BKE_world_copy(wrld);
199                 wrld_new->id.us = 0;
200
201                 /* Remap paths of new ID using old library as base. */
202                 BKE_id_lib_local_paths(bmain, wrld->id.lib, &wrld_new->id);
203
204                 for (sce = bmain->scene.first; sce; sce = sce->id.next) {
205                         if (sce->world == wrld) {
206                                 if (sce->id.lib == NULL) {
207                                         sce->world = wrld_new;
208                                         wrld_new->id.us++;
209                                         wrld->id.us--;
210                                 }
211                         }
212                 }
213         }
214 }