Fixed:
[blender.git] / source / blender / yafray / intern / yafray_Render.cpp
1 //----------------------------------------------------------------------------------------------------
2 // YafRay XML export
3 //
4 // For anyone else looking at this, this was designed for a tabspacing of 2 (YafRay/Jandro standard :)
5 //----------------------------------------------------------------------------------------------------
6
7 #include "yafray_Render.h"
8
9 #include <math.h>
10
11 using namespace std;
12
13 void yafrayRender_t::clearAll()
14 {
15         all_objects.clear();
16         used_materials.clear();
17         used_textures.clear();
18         dupliMtx_list.clear();
19         dup_srcob.clear();
20         objectData.clear();
21         imagetex.clear();
22         imgtex_shader.clear();
23 }
24
25 bool yafrayRender_t::exportScene()
26 {
27
28   // get camera first, no checking should be necessary, all done by Blender
29         maincam_obj = G.scene->camera;
30
31         // use fixed lens for objects functioning as temporary camera (ctrl-0)
32         mainCamLens = 35.0;
33         if (maincam_obj->type==OB_CAMERA) mainCamLens=((Camera*)maincam_obj->data)->lens;
34
35         maxraydepth = 5;        // will be set to maximum depth used in blender materials
36
37         // recreate the scene as object data, as well as sorting the material & textures, ignoring duplicates
38         if (!getAllMatTexObs())
39         {
40                 // error found, clear for next call
41                 clearAll();
42                 return false;
43         }
44
45         if (!initExport())
46         {
47                 G.afbreek = 1;
48                 clearAll();
49                 return false;
50         }
51
52         // start actual data export
53         writeTextures();
54         writeMaterialsAndModulators();
55         writeAllObjects();
56         writeLamps();
57         hasworld = writeWorld();
58         writeCamera();
59         writeRender();
60         
61         // clear for next call, before render to free some memory
62         clearAll();
63
64         if (!finishExport())
65         {
66                 G.afbreek = 1;  //stop render and anim if doing so
67                 return false;
68         }
69         else return true;
70 }
71
72
73 // gets all unique face materials & textures,
74 // and sorts the facelist rejecting anything that is not a quad or tri,
75 // as well as associating them again with the original Object.
76 bool yafrayRender_t::getAllMatTexObs()
77 {
78
79         VlakRen* vlr;
80
81         for (int i=0;i<R.totvlak;i++) {
82
83                 if ((i & 255)==0) vlr=R.blovl[i>>8]; else vlr++;
84
85                 // ---- The materials & textures
86                 // in this case, probably every face has a material assigned, which can be the default material,
87                 // so checking that this is !0 is probably not necessary, but just in case...
88                 Material* matr = vlr->mat;
89                 if (matr) {
90                         // The default assigned material seems to be nameless, no MA id, an empty string.
91                         // Since this name is needed in yafray, make it 'blender_default'
92                         if (strlen(matr->id.name)==0)
93                                 used_materials["blender_default"] = matr;
94                         else
95                                 used_materials[matr->id.name] = matr;
96                         // textures, all active channels
97                         for (int m=0;m<8;m++) {
98                                 if (matr->septex & (1<<m)) continue;    // only active channels
99                                 MTex* mx = matr->mtex[m];
100                                 // if no mtex, ignore
101                                 if (mx==NULL) continue;
102                                 // if no tex, ignore
103                                 Tex* tx = mx->tex;
104                                 if (tx==NULL) continue;
105                                 short txtp = tx->type;
106                                 // if texture type not available in yafray, ignore
107                                 if ((txtp==0) ||
108                                                 (txtp==TEX_MAGIC) ||
109                                                 (txtp==TEX_BLEND) ||
110                                                 (txtp==TEX_NOISE) ||
111                                                 (txtp==TEX_PLUGIN) ||
112                                                 (txtp==TEX_ENVMAP)) continue;
113                                 // In the case of an image texture, check that there is an actual image, otherwise ignore.
114                                 // Stupid error was here (...if (txtp & TEX_IMAGE)...),
115                                 // which happened to work sofar, but not anymore with the extended texture support..
116                                 if ((txtp==TEX_IMAGE) && (!tx->ima)) continue;
117                                 used_textures[tx->id.name] = mx;
118                         }
119                 }
120
121                 // Make list of faces per object, ignore <3 vert faces, duplicate vertex sorting done later.
122                 // ignore null object pointers.
123                 // Also make list of facetexture images (material 'TexFace').
124                 if (vlr->ob) {
125                         int nv = 0;     // number of vertices
126                         if (vlr->v4) nv=4; else if (vlr->v3) nv=3;
127                         if (nv) all_objects[vlr->ob].push_back(vlr);
128                         if (vlr->tface) {
129                                 Image* fc_img = (Image*)vlr->tface->tpage;
130                                 if (fc_img) {
131                                         Material* fmat = vlr->mat;
132                                         // only save if TexFace enabled
133                                         if (fmat && (fmat->mode & MA_FACETEXTURE)) imagetex[fc_img] = fmat;
134                                 }
135                         }
136                 }
137
138         }
139
140         // in case dupliMtx_list not empty, make sure that there is at least one source object
141         // in all_objects with the name given in dupliMtx_list
142         if (!dupliMtx_list.empty()) {
143
144                 for (map<Object*, vector<VlakRen*> >::const_iterator obn=all_objects.begin();
145                         obn!=all_objects.end();++obn)
146                 {
147                         Object* obj = obn->first;
148                         string obname = obj->id.name;
149                         if (dupliMtx_list.find(obname)!=dupliMtx_list.end()) dup_srcob[obname] = obj;
150                 }
151
152                 // if the name reference list is empty, return now, something was seriously wrong
153                 if (dup_srcob.empty()) {
154                   // error() doesn't work to well, when switching from Blender to console at least, so use stdout instead
155                         cout << "ERROR: Duplilist non_empty, but no srcobs\n";
156                         return false;
157                 }
158         }
159
160         return true;
161 }
162
163
164
165 void yafrayRender_t::addDupliMtx(Object* obj)
166 {
167         for (int i=0;i<4;i++)
168                 for (int j=0;j<4;j++)
169                         dupliMtx_list[obj->id.name].push_back(obj->obmat[i][j]);
170 }
171
172
173 bool yafrayRender_t::objectKnownData(Object* obj)
174 {
175         // if object data already known, no need to include in renderlist, otherwise save object datapointer
176         if (objectData.find(obj->data)!=objectData.end()) {
177                 Object* orgob = objectData[obj->data];
178                 // first save original object matrix in dupliMtx_list, if not added yet
179                 if (dupliMtx_list.find(orgob->id.name)==dupliMtx_list.end()) {
180                         cout << "Added original matrix\n";
181                         addDupliMtx(orgob);
182                 }
183                 // then save matrix of linked object in dupliMtx_list, using name of ORIGINAL object
184                 for (int i=0;i<4;i++)
185                         for (int j=0;j<4;j++)
186                                 dupliMtx_list[orgob->id.name].push_back(obj->obmat[i][j]);
187                 return true;
188         }
189         // object not known yet
190         objectData[obj->data] = obj;
191         return false;
192 }