Merge branch 'blender2.7'
[blender.git] / source / blender / freestyle / intern / blender_interface / BlenderStrokeRenderer.h
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  * ***** END GPL LICENSE BLOCK *****
19  */
20
21 #ifndef __BLENDER_STROKE_RENDERER_H__
22 #define __BLENDER_STROKE_RENDERER_H__
23
24 /** \file blender/freestyle/intern/blender_interface/BlenderStrokeRenderer.h
25  *  \ingroup freestyle
26  */
27
28 #include "../stroke/StrokeRenderer.h"
29 #include "../system/FreestyleConfig.h"
30
31 extern "C" {
32 struct Depsgraph;
33 struct GHash;
34 struct Main;
35 struct Material;
36 struct Object;
37 struct Render;
38 struct Scene;
39 struct bContext;
40 struct bNodeTree;
41 }
42
43 namespace Freestyle {
44
45 class BlenderStrokeRenderer : public StrokeRenderer
46 {
47 public:
48         BlenderStrokeRenderer(Render *re, int render_count);
49         virtual ~BlenderStrokeRenderer();
50
51         /*! Renders a stroke rep */
52         virtual void RenderStrokeRep(StrokeRep *iStrokeRep) const;
53         virtual void RenderStrokeRepBasic(StrokeRep *iStrokeRep) const;
54
55         Object *NewMesh() const;
56
57         struct StrokeGroup {
58                 explicit StrokeGroup() : totvert(0), totedge(0), totpoly(0), totloop(0), totcol(0) {}
59                 vector<StrokeRep*> strokes;
60                 int totvert;
61                 int totedge;
62                 int totpoly;
63                 int totloop;
64                 int totcol;
65         };
66         vector<StrokeGroup*> strokeGroups, texturedStrokeGroups;
67
68         int GenerateScene();
69         void GenerateStrokeMesh(StrokeGroup *group, bool hasTex);
70         void FreeStrokeGroups();
71
72         Render *RenderScene(Render *re, bool render);
73
74         static Material* GetStrokeShader(Main *bmain, bNodeTree *iNodeTree, bool do_id_user);
75
76 protected:
77         Main *freestyle_bmain;
78         Scene *old_scene;
79         Scene *freestyle_scene;
80         Depsgraph *freestyle_depsgraph;
81         bContext *_context;
82         float _width, _height;
83         float _z, _z_delta;
84         unsigned int _mesh_id;
85         bool _use_shading_nodes;
86         struct GHash *_nodetree_hash;
87
88         static const char *uvNames[];
89
90         int get_stroke_count() const;
91         float get_stroke_vertex_z(void) const;
92         unsigned int get_stroke_mesh_id(void) const;
93         bool test_triangle_visibility(StrokeVertexRep *svRep[3]) const;
94         void test_strip_visibility(Strip::vertex_container& strip_vertices,
95                 int *visible_faces, int *visible_segments) const;
96
97         vector<StrokeRep *> _strokeReps;
98
99 #ifdef WITH_CXX_GUARDEDALLOC
100         MEM_CXX_CLASS_ALLOC_FUNCS("Freestyle:BlenderStrokeRenderer")
101 #endif
102 };
103
104 } /* namespace Freestyle */
105
106 #endif // __BLENDER_STROKE_RENDERER_H__