35aaaad43fd55a0392e2db0a67851b1680049932
[blender.git] / source / blender / collada / MaterialExporter.h
1 /**
2  * $Id: DocumentExporter.cpp 32309 2010-10-05 00:05:14Z jesterking $
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  * Contributor(s): Chingiz Dyussenov, Arystanbek Dyussenov, Jan Diederich, Tod Liverseed,
21  *                 Nathan Letwory
22  *
23  * ***** END GPL LICENSE BLOCK *****
24  */
25
26 #ifndef __MATERIALEXPORTER_H__
27 #define __MATERIALEXPORTER_H__
28
29 #include "COLLADASWLibraryMaterials.h"
30 #include "COLLADASWStreamWriter.h"
31
32 #include "DNA_material_types.h"
33 #include "DNA_object_types.h"
34 #include "DNA_scene_types.h"
35
36 #include "GeometryExporter.h"
37
38 class MaterialsExporter: COLLADASW::LibraryMaterials
39 {
40 public:
41         MaterialsExporter(COLLADASW::StreamWriter *sw);
42         void exportMaterials(Scene *sce);
43         void operator()(Material *ma, Object *ob);
44 };
45
46 // used in forEachMaterialInScene
47 template <class Functor>
48 class ForEachMaterialFunctor
49 {
50         std::vector<std::string> mMat; // contains list of material names, to avoid duplicate calling of f
51         Functor *f;
52 public:
53         ForEachMaterialFunctor(Functor*f) : f(f) {}
54         
55         void operator ()(Object *ob)
56         {
57                 int a;
58                 for(a = 0; a < ob->totcol; a++) {
59
60                         Material *ma = give_current_material(ob, a+1);
61
62                         if (!ma) continue;
63
64                         std::string translated_id = translate_id(id_name(ma));
65                         if (find(mMat.begin(), mMat.end(), translated_id) == mMat.end()) {
66                                 (*this->f)(ma, ob);
67
68                                 mMat.push_back(translated_id);
69                         }
70                 }
71         }
72 };
73
74 struct MaterialFunctor {
75         // calls f for each unique material linked to each object in sce
76         // f should have
77         // void operator()(Material* ma)
78         template<class Functor>
79         void forEachMaterialInScene(Scene *sce, Functor &f)
80         {
81                 ForEachMaterialFunctor<Functor> matfunc(&f);
82                 GeometryFunctor gf;
83                 gf.forEachMeshObjectInScene<ForEachMaterialFunctor<Functor> >(sce, matfunc);
84         }
85 };
86
87
88 #endif