5569df927fb624e245400556f59b112cfe976b6d
[blender-staging.git] / intern / cycles / app / cycles_xml.cpp
1 /*
2  * Copyright 2011, Blender Foundation.
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
19 #include <stdio.h>
20
21 #include <sstream>
22 #include <algorithm>
23 #include <iterator>
24
25 #include "camera.h"
26 #include "film.h"
27 #include "graph.h"
28 #include "integrator.h"
29 #include "light.h"
30 #include "mesh.h"
31 #include "nodes.h"
32 #include "object.h"
33 #include "shader.h"
34 #include "scene.h"
35
36 #include "subd_mesh.h"
37 #include "subd_patch.h"
38 #include "subd_split.h"
39
40 #include "util_debug.h"
41 #include "util_foreach.h"
42 #include "util_path.h"
43 #include "util_transform.h"
44 #include "util_xml.h"
45
46 #include "cycles_xml.h"
47
48 CCL_NAMESPACE_BEGIN
49
50 /* XML reading state */
51
52 struct XMLReadState {
53         Scene *scene;           /* scene pointer */
54         Transform tfm;          /* current transform state */
55         bool smooth;            /* smooth normal state */
56         int shader;                     /* current shader */
57         string base;            /* base path to current file*/
58         float dicing_rate;      /* current dicing rate */
59         Mesh::DisplacementMethod displacement_method;
60 };
61
62 /* Attribute Reading */
63
64 static bool xml_read_bool(bool *value, pugi::xml_node node, const char *name)
65 {
66         pugi::xml_attribute attr = node.attribute(name);
67
68         if(attr) {
69                 *value = (string_iequals(attr.value(), "true")) || (atoi(attr.value()) != 0);
70                 return true;
71         }
72
73         return false;
74 }
75
76 static bool xml_read_int(int *value, pugi::xml_node node, const char *name)
77 {
78         pugi::xml_attribute attr = node.attribute(name);
79
80         if(attr) {
81                 *value = atoi(attr.value());
82                 return true;
83         }
84
85         return false;
86 }
87
88 static bool xml_read_int_array(vector<int>& value, pugi::xml_node node, const char *name)
89 {
90         pugi::xml_attribute attr = node.attribute(name);
91
92         if(attr) {
93                 vector<string> tokens;
94                 string_split(tokens, attr.value());
95
96                 foreach(const string& token, tokens)
97                         value.push_back(atoi(token.c_str()));
98
99                 return true;
100         }
101
102         return false;
103 }
104
105 static bool xml_read_float(float *value, pugi::xml_node node, const char *name)
106 {
107         pugi::xml_attribute attr = node.attribute(name);
108
109         if(attr) {
110                 *value = atof(attr.value());
111                 return true;
112         }
113
114         return false;
115 }
116
117 static bool xml_read_float_array(vector<float>& value, pugi::xml_node node, const char *name)
118 {
119         pugi::xml_attribute attr = node.attribute(name);
120
121         if(attr) {
122                 vector<string> tokens;
123                 string_split(tokens, attr.value());
124
125                 foreach(const string& token, tokens)
126                         value.push_back(atof(token.c_str()));
127
128                 return true;
129         }
130
131         return false;
132 }
133
134 static bool xml_read_float3(float3 *value, pugi::xml_node node, const char *name)
135 {
136         vector<float> array;
137
138         if(xml_read_float_array(array, node, name) && array.size() == 3) {
139                 *value = make_float3(array[0], array[1], array[2]);
140                 return true;
141         }
142
143         return false;
144 }
145
146 static bool xml_read_float3_array(vector<float3>& value, pugi::xml_node node, const char *name)
147 {
148         vector<float> array;
149
150         if(xml_read_float_array(array, node, name)) {
151                 for(size_t i = 0; i < array.size(); i += 3)
152                         value.push_back(make_float3(array[i+0], array[i+1], array[i+2]));
153
154                 return true;
155         }
156
157         return false;
158 }
159
160 static bool xml_read_float4(float4 *value, pugi::xml_node node, const char *name)
161 {
162         vector<float> array;
163
164         if(xml_read_float_array(array, node, name) && array.size() == 4) {
165                 *value = make_float4(array[0], array[1], array[2], array[3]);
166                 return true;
167         }
168
169         return false;
170 }
171
172 static bool xml_read_string(string *str, pugi::xml_node node, const char *name)
173 {
174         pugi::xml_attribute attr = node.attribute(name);
175
176         if(attr) {
177                 *str = attr.value();
178                 return true;
179         }
180
181         return false;
182 }
183
184 static bool xml_read_ustring(ustring *str, pugi::xml_node node, const char *name)
185 {
186         pugi::xml_attribute attr = node.attribute(name);
187
188         if(attr) {
189                 *str = ustring(attr.value());
190                 return true;
191         }
192
193         return false;
194 }
195
196 static bool xml_equal_string(pugi::xml_node node, const char *name, const char *value)
197 {
198         pugi::xml_attribute attr = node.attribute(name);
199
200         if(attr)
201                 return string_iequals(attr.value(), value);
202         
203         return false;
204 }
205
206 static bool xml_read_enum(ustring *str, ShaderEnum& enm, pugi::xml_node node, const char *name)
207 {
208         pugi::xml_attribute attr = node.attribute(name);
209
210         if(attr) {
211                 ustring ustr(attr.value());
212
213                 if(enm.exists(ustr)) {
214                         *str = ustr;
215                         return true;
216                 }
217                 else
218                         fprintf(stderr, "Unknown value \"%s\" for attribute \"%s\".\n", ustr.c_str(), name);
219         }
220
221         return false;
222 }
223
224 /* Film */
225
226 static void xml_read_film(const XMLReadState& state, pugi::xml_node node)
227 {
228         Camera *cam = state.scene->camera;
229
230         xml_read_int(&cam->width, node, "width");
231         xml_read_int(&cam->height, node, "height");
232
233         float aspect = (float)cam->width/(float)cam->height;
234
235         if(cam->width >= cam->height) {
236                 cam->left = -aspect;
237                 cam->right = aspect;
238                 cam->bottom = -1.0f;
239                 cam->top = 1.0f;
240         }
241         else {
242                 cam->left = -1.0f;
243                 cam->right = 1.0f;
244                 cam->bottom = -1.0f/aspect;
245                 cam->top = 1.0f/aspect;
246         }
247
248         cam->need_update = true;
249         cam->update();
250 }
251
252 /* Integrator */
253
254 static void xml_read_integrator(const XMLReadState& state, pugi::xml_node node)
255 {
256         Integrator *integrator = state.scene->integrator;
257         
258         xml_read_bool(&integrator->progressive, node, "progressive");
259         
260         if(!integrator->progressive) {
261                 xml_read_int(&integrator->diffuse_samples, node, "diffuse_samples");
262                 xml_read_int(&integrator->glossy_samples, node, "glossy_samples");
263                 xml_read_int(&integrator->transmission_samples, node, "transmission_samples");
264                 xml_read_int(&integrator->ao_samples, node, "ao_samples");
265                 xml_read_int(&integrator->mesh_light_samples, node, "mesh_light_samples");
266         }
267
268         xml_read_int(&integrator->min_bounce, node, "min_bounce");
269         xml_read_int(&integrator->max_bounce, node, "max_bounce");
270         
271         xml_read_int(&integrator->max_diffuse_bounce, node, "max_diffuse_bounce");
272         xml_read_int(&integrator->max_glossy_bounce, node, "max_glossy_bounce");
273         xml_read_int(&integrator->max_transmission_bounce, node, "max_transmission_bounce");
274         
275         xml_read_int(&integrator->transparent_min_bounce, node, "transparent_min_bounce");
276         xml_read_int(&integrator->transparent_max_bounce, node, "transparent_max_bounce");
277         
278         xml_read_bool(&integrator->transparent_shadows, node, "transparent_shadows");
279         xml_read_bool(&integrator->no_caustics, node, "no_caustics");
280         xml_read_float(&integrator->filter_glossy, node, "blur_glossy");
281         
282         xml_read_int(&integrator->seed, node, "seed");
283         xml_read_float(&integrator->sample_clamp, node, "sample_clamp");
284 }
285
286 /* Camera */
287
288 static void xml_read_camera(const XMLReadState& state, pugi::xml_node node)
289 {
290         Camera *cam = state.scene->camera;
291
292         if(xml_read_float(&cam->fov, node, "fov"))
293                 cam->fov *= M_PI/180.0f;
294
295         xml_read_float(&cam->nearclip, node, "nearclip");
296         xml_read_float(&cam->farclip, node, "farclip");
297         xml_read_float(&cam->aperturesize, node, "aperturesize"); // 0.5*focallength/fstop
298         xml_read_float(&cam->focaldistance, node, "focaldistance");
299         xml_read_float(&cam->shuttertime, node, "shuttertime");
300
301         if(xml_equal_string(node, "type", "orthographic"))
302                 cam->type = CAMERA_ORTHOGRAPHIC;
303         else if(xml_equal_string(node, "type", "perspective"))
304                 cam->type = CAMERA_PERSPECTIVE;
305         else if(xml_equal_string(node, "type", "panorama"))
306                 cam->type = CAMERA_PANORAMA;
307
308         if(xml_equal_string(node, "panorama_type", "equirectangular"))
309                 cam->panorama_type = PANORAMA_EQUIRECTANGULAR;
310         else if(xml_equal_string(node, "panorama_type", "fisheye_equidistant"))
311                 cam->panorama_type = PANORAMA_FISHEYE_EQUIDISTANT;
312         else if(xml_equal_string(node, "panorama_type", "fisheye_equisolid"))
313                 cam->panorama_type = PANORAMA_FISHEYE_EQUISOLID;
314
315         xml_read_float(&cam->fisheye_fov, node, "fisheye_fov");
316         xml_read_float(&cam->fisheye_lens, node, "fisheye_lens");
317
318         xml_read_float(&cam->sensorwidth, node, "sensorwidth");
319         xml_read_float(&cam->sensorheight, node, "sensorheight");
320
321
322         cam->matrix = state.tfm;
323
324         cam->need_update = true;
325         cam->update();
326 }
327
328 /* Shader */
329
330 static string xml_socket_name(const char *name)
331 {
332         string sname = name;
333         size_t i;
334
335         while((i = sname.find(" ")) != string::npos)
336                 sname.replace(i, 1, "");
337         
338         return sname;
339 }
340
341 static void xml_read_shader_graph(const XMLReadState& state, Shader *shader, pugi::xml_node graph_node)
342 {
343         ShaderGraph *graph = new ShaderGraph();
344
345         map<string, ShaderNode*> nodemap;
346
347         nodemap["output"] = graph->output();
348
349         for(pugi::xml_node node = graph_node.first_child(); node; node = node.next_sibling()) {
350                 ShaderNode *snode = NULL;
351
352                 if(string_iequals(node.name(), "image_texture")) {
353                         ImageTextureNode *img = new ImageTextureNode();
354
355                         xml_read_string(&img->filename, node, "src");
356                         img->filename = path_join(state.base, img->filename);
357
358                         snode = img;
359                 }
360                 else if(string_iequals(node.name(), "environment_texture")) {
361                         EnvironmentTextureNode *env = new EnvironmentTextureNode();
362
363                         xml_read_string(&env->filename, node, "src");
364                         env->filename = path_join(state.base, env->filename);
365
366                         snode = env;
367                 }
368                 else if(string_iequals(node.name(), "sky_texture")) {
369                         SkyTextureNode *sky = new SkyTextureNode();
370
371                         xml_read_float3(&sky->sun_direction, node, "sun_direction");
372                         xml_read_float(&sky->turbidity, node, "turbidity");
373                         
374                         snode = sky;
375                 }
376                 else if(string_iequals(node.name(), "noise_texture")) {
377                         snode = new NoiseTextureNode();
378                 }
379                 else if(string_iequals(node.name(), "checker_texture")) {
380                         snode = new CheckerTextureNode();
381                 }
382                 else if(string_iequals(node.name(), "gradient_texture")) {
383                         GradientTextureNode *blend = new GradientTextureNode();
384                         xml_read_enum(&blend->type, GradientTextureNode::type_enum, node, "type");
385                         snode = blend;
386                 }
387                 else if(string_iequals(node.name(), "voronoi_texture")) {
388                         VoronoiTextureNode *voronoi = new VoronoiTextureNode();
389                         xml_read_enum(&voronoi->coloring, VoronoiTextureNode::coloring_enum, node, "coloring");
390                         snode = voronoi;
391                 }
392                 else if(string_iequals(node.name(), "musgrave_texture")) {
393                         MusgraveTextureNode *musgrave = new MusgraveTextureNode();
394                         xml_read_enum(&musgrave->type, MusgraveTextureNode::type_enum, node, "type");
395                         snode = musgrave;
396                 }
397                 else if(string_iequals(node.name(), "magic_texture")) {
398                         MagicTextureNode *magic = new MagicTextureNode();
399                         xml_read_int(&magic->depth, node, "depth");
400                         snode = magic;
401                 }
402                 else if(string_iequals(node.name(), "noise_texture")) {
403                         NoiseTextureNode *dist = new NoiseTextureNode();
404                         snode = dist;
405                 }
406                 else if(string_iequals(node.name(), "wave_texture")) {
407                         WaveTextureNode *wave = new WaveTextureNode();
408                         xml_read_enum(&wave->type, WaveTextureNode::type_enum, node, "type");
409                         snode = wave;
410                 }
411                 else if(string_iequals(node.name(), "normal")) {
412                         snode = new NormalNode();
413                 }
414                 else if(string_iequals(node.name(), "mapping")) {
415                         snode = new MappingNode();
416                 }
417                 else if(string_iequals(node.name(), "ward_bsdf")) {
418                         snode = new WardBsdfNode();
419                 }
420                 else if(string_iequals(node.name(), "diffuse_bsdf")) {
421                         snode = new DiffuseBsdfNode();
422                 }
423                 else if(string_iequals(node.name(), "translucent_bsdf")) {
424                         snode = new TranslucentBsdfNode();
425                 }
426                 else if(string_iequals(node.name(), "transparent_bsdf")) {
427                         snode = new TransparentBsdfNode();
428                 }
429                 else if(string_iequals(node.name(), "velvet_bsdf")) {
430                         snode = new VelvetBsdfNode();
431                 }
432                 else if(string_iequals(node.name(), "glossy_bsdf")) {
433                         GlossyBsdfNode *glossy = new GlossyBsdfNode();
434                         xml_read_enum(&glossy->distribution, GlossyBsdfNode::distribution_enum, node, "distribution");
435                         snode = glossy;
436                 }
437                 else if(string_iequals(node.name(), "glass_bsdf")) {
438                         GlassBsdfNode *diel = new GlassBsdfNode();
439                         xml_read_enum(&diel->distribution, GlassBsdfNode::distribution_enum, node, "distribution");
440                         snode = diel;
441                 }
442                 else if(string_iequals(node.name(), "emission")) {
443                         EmissionNode *emission = new EmissionNode();
444                         xml_read_bool(&emission->total_power, node, "total_power");
445                         snode = emission;
446                 }
447                 else if(string_iequals(node.name(), "background")) {
448                         snode = new BackgroundNode();
449                 }
450                 else if(string_iequals(node.name(), "transparent_volume")) {
451                         snode = new TransparentVolumeNode();
452                 }
453                 else if(string_iequals(node.name(), "isotropic_volume")) {
454                         snode = new IsotropicVolumeNode();
455                 }
456                 else if(string_iequals(node.name(), "geometry")) {
457                         snode = new GeometryNode();
458                 }
459                 else if(string_iequals(node.name(), "texture_coordinate")) {
460                         snode = new TextureCoordinateNode();
461                 }
462                 else if(string_iequals(node.name(), "lightPath")) {
463                         snode = new LightPathNode();
464                 }
465                 else if(string_iequals(node.name(), "value")) {
466                         ValueNode *value = new ValueNode();
467                         xml_read_float(&value->value, node, "value");
468                         snode = value;
469                 }
470                 else if(string_iequals(node.name(), "color")) {
471                         ColorNode *color = new ColorNode();
472                         xml_read_float3(&color->value, node, "value");
473                         snode = color;
474                 }
475                 else if(string_iequals(node.name(), "mix_closure")) {
476                         snode = new MixClosureNode();
477                 }
478                 else if(string_iequals(node.name(), "add_closure")) {
479                         snode = new AddClosureNode();
480                 }
481                 else if(string_iequals(node.name(), "invert")) {
482                         snode = new InvertNode();
483                 }
484                 else if(string_iequals(node.name(), "mix")) {
485                         MixNode *mix = new MixNode();
486                         xml_read_enum(&mix->type, MixNode::type_enum, node, "type");
487                         xml_read_bool(&mix->use_clamp, node, "use_clamp");
488                         snode = mix;
489                 }
490                 else if(string_iequals(node.name(), "gamma")) {
491                         snode = new GammaNode();
492                 }
493                 else if(string_iequals(node.name(), "brightness")) {
494                         snode = new BrightContrastNode();
495                 }
496                 else if(string_iequals(node.name(), "combine_rgb")) {
497                         snode = new CombineRGBNode();
498                 }
499                 else if(string_iequals(node.name(), "separate_rgb")) {
500                         snode = new SeparateRGBNode();
501                 }
502                 else if(string_iequals(node.name(), "hsv")) {
503                         snode = new HSVNode();
504                 }
505                 else if(string_iequals(node.name(), "attribute")) {
506                         AttributeNode *attr = new AttributeNode();
507                         xml_read_ustring(&attr->attribute, node, "attribute");
508                         snode = attr;
509                 }
510                 else if(string_iequals(node.name(), "camera")) {
511                         snode = new CameraNode();
512                 }
513                 else if(string_iequals(node.name(), "fresnel")) {
514                         snode = new FresnelNode();
515                 }
516                 else if(string_iequals(node.name(), "math")) {
517                         MathNode *math = new MathNode();
518                         xml_read_enum(&math->type, MathNode::type_enum, node, "type");
519                         xml_read_bool(&math->use_clamp, node, "use_clamp");
520                         snode = math;
521                 }
522                 else if(string_iequals(node.name(), "vector_math")) {
523                         VectorMathNode *vmath = new VectorMathNode();
524                         xml_read_enum(&vmath->type, VectorMathNode::type_enum, node, "type");
525                         snode = vmath;
526                 }
527                 else if(string_iequals(node.name(), "connect")) {
528                         /* connect nodes */
529                         vector<string> from_tokens, to_tokens;
530
531                         string_split(from_tokens, node.attribute("from").value());
532                         string_split(to_tokens, node.attribute("to").value());
533
534                         if(from_tokens.size() == 2 && to_tokens.size() == 2) {
535                                 /* find nodes and sockets */
536                                 ShaderOutput *output = NULL;
537                                 ShaderInput *input = NULL;
538
539                                 if(nodemap.find(from_tokens[0]) != nodemap.end()) {
540                                         ShaderNode *fromnode = nodemap[from_tokens[0]];
541
542                                         foreach(ShaderOutput *out, fromnode->outputs)
543                                                 if(string_iequals(xml_socket_name(out->name), from_tokens[1]))
544                                                         output = out;
545
546                                         if(!output)
547                                                 fprintf(stderr, "Unknown output socket name \"%s\" on \"%s\".\n", from_tokens[1].c_str(), from_tokens[0].c_str());
548                                 }
549                                 else
550                                         fprintf(stderr, "Unknown shader node name \"%s\".\n", from_tokens[0].c_str());
551
552                                 if(nodemap.find(to_tokens[0]) != nodemap.end()) {
553                                         ShaderNode *tonode = nodemap[to_tokens[0]];
554
555                                         foreach(ShaderInput *in, tonode->inputs)
556                                                 if(string_iequals(xml_socket_name(in->name), to_tokens[1]))
557                                                         input = in;
558
559                                         if(!input)
560                                                 fprintf(stderr, "Unknown input socket name \"%s\" on \"%s\".\n", to_tokens[1].c_str(), to_tokens[0].c_str());
561                                 }
562                                 else
563                                         fprintf(stderr, "Unknown shader node name \"%s\".\n", to_tokens[0].c_str());
564
565                                 /* connect */
566                                 if(output && input)
567                                         graph->connect(output, input);
568                         }
569                         else
570                                 fprintf(stderr, "Invalid from or to value for connect node.\n");
571                 }
572                 else
573                         fprintf(stderr, "Unknown shader node \"%s\".\n", node.name());
574
575                 if(snode) {
576                         /* add to graph */
577                         graph->add(snode);
578
579                         /* add to map for name lookups */
580                         string name = "";
581                         xml_read_string(&name, node, "name");
582
583                         nodemap[name] = snode;
584
585                         /* read input values */
586                         for(pugi::xml_attribute attr = node.first_attribute(); attr; attr = attr.next_attribute()) {
587                                 foreach(ShaderInput *in, snode->inputs) {
588                                         if(string_iequals(in->name, attr.name())) {
589                                                 switch(in->type) {
590                                                         case SHADER_SOCKET_FLOAT:
591                                                                 xml_read_float(&in->value.x, node, attr.name());
592                                                                 break;
593                                                         case SHADER_SOCKET_COLOR:
594                                                         case SHADER_SOCKET_VECTOR:
595                                                         case SHADER_SOCKET_POINT:
596                                                         case SHADER_SOCKET_NORMAL:
597                                                                 xml_read_float3(&in->value, node, attr.name());
598                                                                 break;
599                                                         default:
600                                                                 break;
601                                                 }
602                                         }
603                                 }
604                         }
605                 }
606         }
607
608         shader->set_graph(graph);
609         shader->tag_update(state.scene);
610 }
611
612 static void xml_read_shader(const XMLReadState& state, pugi::xml_node node)
613 {
614         Shader *shader = new Shader();
615         xml_read_string(&shader->name, node, "name");
616         xml_read_shader_graph(state, shader, node);
617         state.scene->shaders.push_back(shader);
618 }
619
620 /* Background */
621
622 static void xml_read_background(const XMLReadState& state, pugi::xml_node node)
623 {
624         Shader *shader = state.scene->shaders[state.scene->default_background];
625
626         xml_read_shader_graph(state, shader, node);
627 }
628
629 /* Mesh */
630
631 static Mesh *xml_add_mesh(Scene *scene, const Transform& tfm)
632 {
633         /* create mesh */
634         Mesh *mesh = new Mesh();
635         scene->meshes.push_back(mesh);
636
637         /* create object*/
638         Object *object = new Object();
639         object->mesh = mesh;
640         object->tfm = tfm;
641         scene->objects.push_back(object);
642
643         return mesh;
644 }
645
646 static void xml_read_mesh(const XMLReadState& state, pugi::xml_node node)
647 {
648         /* add mesh */
649         Mesh *mesh = xml_add_mesh(state.scene, state.tfm);
650         mesh->used_shaders.push_back(state.shader);
651
652         /* read state */
653         int shader = state.shader;
654         bool smooth = state.smooth;
655
656         mesh->displacement_method = state.displacement_method;
657
658         /* read vertices and polygons, RIB style */
659         vector<float3> P;
660         vector<int> verts, nverts;
661
662         xml_read_float3_array(P, node, "P");
663         xml_read_int_array(verts, node, "verts");
664         xml_read_int_array(nverts, node, "nverts");
665
666         if(xml_equal_string(node, "subdivision", "catmull-clark")) {
667                 /* create subd mesh */
668                 SubdMesh sdmesh;
669
670                 /* create subd vertices */
671                 for(size_t i = 0; i < P.size(); i++)
672                         sdmesh.add_vert(P[i]);
673
674                 /* create subd faces */
675                 int index_offset = 0;
676
677                 for(size_t i = 0; i < nverts.size(); i++) {
678                         if(nverts[i] == 4) {
679                                 int v0 = verts[index_offset + 0];
680                                 int v1 = verts[index_offset + 1];
681                                 int v2 = verts[index_offset + 2];
682                                 int v3 = verts[index_offset + 3];
683
684                                 sdmesh.add_face(v0, v1, v2, v3);
685                         }
686                         else {
687                                 for(int j = 0; j < nverts[i]-2; j++) {
688                                         int v0 = verts[index_offset];
689                                         int v1 = verts[index_offset + j + 1];
690                                         int v2 = verts[index_offset + j + 2];
691
692                                         sdmesh.add_face(v0, v1, v2);
693                                 }
694                         }
695
696                         index_offset += nverts[i];
697                 }
698
699                 /* finalize subd mesh */
700                 sdmesh.link_boundary();
701
702                 /* subdivide */
703                 DiagSplit dsplit;
704                 //dsplit.camera = state.scene->camera;
705                 //dsplit.dicing_rate = 5.0f;
706                 dsplit.dicing_rate = state.dicing_rate;
707                 xml_read_float(&dsplit.dicing_rate, node, "dicing_rate");
708                 sdmesh.tessellate(&dsplit, false, mesh, shader, smooth);
709         }
710         else {
711                 /* create vertices */
712                 mesh->verts = P;
713
714                 /* create triangles */
715                 int index_offset = 0;
716
717                 for(size_t i = 0; i < nverts.size(); i++) {
718                         for(int j = 0; j < nverts[i]-2; j++) {
719                                 int v0 = verts[index_offset];
720                                 int v1 = verts[index_offset + j + 1];
721                                 int v2 = verts[index_offset + j + 2];
722
723                                 assert(v0 < (int)P.size());
724                                 assert(v1 < (int)P.size());
725                                 assert(v2 < (int)P.size());
726
727                                 mesh->add_triangle(v0, v1, v2, shader, smooth);
728                         }
729
730                         index_offset += nverts[i];
731                 }
732         }
733
734         /* temporary for test compatibility */
735         mesh->attributes.remove(ATTR_STD_VERTEX_NORMAL);
736 }
737
738 /* Patch */
739
740 static void xml_read_patch(const XMLReadState& state, pugi::xml_node node)
741 {
742         /* read patch */
743         Patch *patch = NULL;
744
745         vector<float3> P;
746         xml_read_float3_array(P, node, "P");
747
748         if(xml_equal_string(node, "type", "bilinear")) {
749                 /* bilinear patch */
750                 if(P.size() == 4) {
751                         LinearQuadPatch *bpatch = new LinearQuadPatch();
752
753                         for(int i = 0; i < 4; i++)
754                                 P[i] = transform_point(&state.tfm, P[i]);
755                         memcpy(bpatch->hull, &P[0], sizeof(bpatch->hull));
756
757                         patch = bpatch;
758                 }
759                 else
760                         fprintf(stderr, "Invalid number of control points for bilinear patch.\n");
761         }
762         else if(xml_equal_string(node, "type", "bicubic")) {
763                 /* bicubic patch */
764                 if(P.size() == 16) {
765                         BicubicPatch *bpatch = new BicubicPatch();
766
767                         for(int i = 0; i < 16; i++)
768                                 P[i] = transform_point(&state.tfm, P[i]);
769                         memcpy(bpatch->hull, &P[0], sizeof(bpatch->hull));
770
771                         patch = bpatch;
772                 }
773                 else
774                         fprintf(stderr, "Invalid number of control points for bicubic patch.\n");
775         }
776         else
777                 fprintf(stderr, "Unknown patch type.\n");
778
779         if(patch) {
780                 /* add mesh */
781                 Mesh *mesh = xml_add_mesh(state.scene, transform_identity());
782
783                 mesh->used_shaders.push_back(state.shader);
784
785                 /* split */
786                 DiagSplit dsplit;
787                 //dsplit.camera = state.scene->camera;
788                 //dsplit.dicing_rate = 5.0f;
789                 dsplit.dicing_rate = state.dicing_rate;
790                 xml_read_float(&dsplit.dicing_rate, node, "dicing_rate");
791                 dsplit.split_quad(mesh, patch, state.shader, state.smooth);
792
793                 delete patch;
794
795                 /* temporary for test compatibility */
796                 mesh->attributes.remove(ATTR_STD_VERTEX_NORMAL);
797         }
798 }
799
800 /* Light */
801
802 static void xml_read_light(const XMLReadState& state, pugi::xml_node node)
803 {
804         Light *light = new Light();
805         light->shader = state.shader;
806         xml_read_float3(&light->co, node, "P");
807         light->co = transform_point(&state.tfm, light->co);
808
809         state.scene->lights.push_back(light);
810 }
811
812 /* Transform */
813
814 static void xml_read_transform(pugi::xml_node node, Transform& tfm)
815 {
816         if(node.attribute("matrix")) {
817                 vector<float> matrix;
818                 if(xml_read_float_array(matrix, node, "matrix") && matrix.size() == 16)
819                         tfm = tfm * transform_transpose((*(Transform*)&matrix[0]));
820         }
821
822         if(node.attribute("translate")) {
823                 float3 translate = make_float3(0.0f, 0.0f, 0.0f);
824                 xml_read_float3(&translate, node, "translate");
825                 tfm = tfm * transform_translate(translate);
826         }
827
828         if(node.attribute("rotate")) {
829                 float4 rotate = make_float4(0.0f, 0.0f, 0.0f, 0.0f);
830                 xml_read_float4(&rotate, node, "rotate");
831                 tfm = tfm * transform_rotate(rotate.x*M_PI/180.0f, make_float3(rotate.y, rotate.z, rotate.w));
832         }
833
834         if(node.attribute("scale")) {
835                 float3 scale = make_float3(0.0f, 0.0f, 0.0f);
836                 xml_read_float3(&scale, node, "scale");
837                 tfm = tfm * transform_scale(scale);
838         }
839 }
840
841 /* State */
842
843 static void xml_read_state(XMLReadState& state, pugi::xml_node node)
844 {
845         /* read shader */
846         string shadername;
847
848         if(xml_read_string(&shadername, node, "shader")) {
849                 int i = 0;
850                 bool found = false;
851
852                 foreach(Shader *shader, state.scene->shaders) {
853                         if(shader->name == shadername) {
854                                 state.shader = i;
855                                 found = true;
856                                 break;
857                         }
858
859                         i++;
860                 }
861
862                 if(!found)
863                         fprintf(stderr, "Unknown shader \"%s\".\n", shadername.c_str());
864         }
865
866         xml_read_float(&state.dicing_rate, node, "dicing_rate");
867
868         /* read smooth/flat */
869         if(xml_equal_string(node, "interpolation", "smooth"))
870                 state.smooth = true;
871         else if(xml_equal_string(node, "interpolation", "flat"))
872                 state.smooth = false;
873
874         /* read displacement method */
875         if(xml_equal_string(node, "displacement_method", "true"))
876                 state.displacement_method = Mesh::DISPLACE_TRUE;
877         else if(xml_equal_string(node, "displacement_method", "bump"))
878                 state.displacement_method = Mesh::DISPLACE_BUMP;
879         else if(xml_equal_string(node, "displacement_method", "both"))
880                 state.displacement_method = Mesh::DISPLACE_BOTH;
881 }
882
883 /* Scene */
884
885 static void xml_read_include(const XMLReadState& state, const string& src);
886
887 static void xml_read_scene(const XMLReadState& state, pugi::xml_node scene_node)
888 {
889         for(pugi::xml_node node = scene_node.first_child(); node; node = node.next_sibling()) {
890                 if(string_iequals(node.name(), "film")) {
891                         xml_read_film(state, node);
892                 }
893                 else if(string_iequals(node.name(), "integrator")) {
894                         xml_read_integrator(state, node);
895                 }
896                 else if(string_iequals(node.name(), "camera")) {
897                         xml_read_camera(state, node);
898                 }
899                 else if(string_iequals(node.name(), "shader")) {
900                         xml_read_shader(state, node);
901                 }
902                 else if(string_iequals(node.name(), "background")) {
903                         xml_read_background(state, node);
904                 }
905                 else if(string_iequals(node.name(), "mesh")) {
906                         xml_read_mesh(state, node);
907                 }
908                 else if(string_iequals(node.name(), "patch")) {
909                         xml_read_patch(state, node);
910                 }
911                 else if(string_iequals(node.name(), "light")) {
912                         xml_read_light(state, node);
913                 }
914                 else if(string_iequals(node.name(), "transform")) {
915                         XMLReadState substate = state;
916
917                         xml_read_transform(node, substate.tfm);
918                         xml_read_scene(substate, node);
919                 }
920                 else if(string_iequals(node.name(), "state")) {
921                         XMLReadState substate = state;
922
923                         xml_read_state(substate, node);
924                         xml_read_scene(substate, node);
925                 }
926                 else if(string_iequals(node.name(), "include")) {
927                         string src;
928
929                         if(xml_read_string(&src, node, "src"))
930                                 xml_read_include(state, src);
931                 }
932                 else
933                         fprintf(stderr, "Unknown node \"%s\".\n", node.name());
934         }
935 }
936
937 /* Include */
938
939 static void xml_read_include(const XMLReadState& state, const string& src)
940 {
941         /* open XML document */
942         pugi::xml_document doc;
943         pugi::xml_parse_result parse_result;
944
945         string path = path_join(state.base, src);
946         parse_result = doc.load_file(path.c_str());
947
948         if(parse_result) {
949                 XMLReadState substate = state;
950                 substate.base = path_dirname(path);
951
952                 xml_read_scene(substate, doc);
953         }
954         else
955                 fprintf(stderr, "%s read error: %s\n", src.c_str(), parse_result.description());
956 }
957
958 /* File */
959
960 void xml_read_file(Scene *scene, const char *filepath)
961 {
962         XMLReadState state;
963
964         state.scene = scene;
965         state.tfm = transform_identity();
966         state.shader = scene->default_surface;
967         state.smooth = false;
968         state.dicing_rate = 0.1f;
969         state.base = path_dirname(filepath);
970
971         xml_read_include(state, path_filename(filepath));
972
973         scene->params.bvh_type = SceneParams::BVH_STATIC;
974 }
975
976 CCL_NAMESPACE_END
977