Merge branch 'blender-v2.81-release'
[blender.git] / source / blender / freestyle / intern / blender_interface / BlenderStrokeRenderer.h
1 /*
2  * This program is free software; you can redistribute it and/or
3  * modify it under the terms of the GNU General Public License
4  * as published by the Free Software Foundation; either version 2
5  * of the License, or (at your option) any later version.
6  *
7  * This program is distributed in the hope that it will be useful,
8  * but WITHOUT ANY WARRANTY; without even the implied warranty of
9  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
10  * GNU General Public License for more details.
11  *
12  * You should have received a copy of the GNU General Public License
13  * along with this program; if not, write to the Free Software Foundation,
14  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
15  */
16
17 #ifndef __BLENDER_STROKE_RENDERER_H__
18 #define __BLENDER_STROKE_RENDERER_H__
19
20 /** \file
21  * \ingroup freestyle
22  */
23
24 #include "../stroke/StrokeRenderer.h"
25 #include "../system/FreestyleConfig.h"
26
27 extern "C" {
28 struct Depsgraph;
29 struct GHash;
30 struct Main;
31 struct Material;
32 struct Object;
33 struct Render;
34 struct Scene;
35 struct bContext;
36 struct bNodeTree;
37 }
38
39 namespace Freestyle {
40
41 class BlenderStrokeRenderer : public StrokeRenderer {
42  public:
43   BlenderStrokeRenderer(Render *re, int render_count);
44   virtual ~BlenderStrokeRenderer();
45
46   /*! Renders a stroke rep */
47   virtual void RenderStrokeRep(StrokeRep *iStrokeRep) const;
48   virtual void RenderStrokeRepBasic(StrokeRep *iStrokeRep) const;
49
50   Object *NewMesh() const;
51
52   struct StrokeGroup {
53     explicit StrokeGroup() : totvert(0), totedge(0), totpoly(0), totloop(0), totcol(0)
54     {
55     }
56     vector<StrokeRep *> strokes;
57     int totvert;
58     int totedge;
59     int totpoly;
60     int totloop;
61     int totcol;
62   };
63   vector<StrokeGroup *> strokeGroups, texturedStrokeGroups;
64
65   int GenerateScene();
66   void GenerateStrokeMesh(StrokeGroup *group, bool hasTex);
67   void FreeStrokeGroups();
68
69   Render *RenderScene(Render *re, bool render);
70
71   static Material *GetStrokeShader(Main *bmain, bNodeTree *iNodeTree, bool do_id_user);
72
73  protected:
74   Main *freestyle_bmain;
75   Scene *old_scene;
76   Scene *freestyle_scene;
77   Depsgraph *freestyle_depsgraph;
78   bContext *_context;
79   float _width, _height;
80   float _z, _z_delta;
81   unsigned int _mesh_id;
82   bool _use_shading_nodes;
83   struct GHash *_nodetree_hash;
84
85   static const char *uvNames[];
86
87   int get_stroke_count() const;
88   float get_stroke_vertex_z(void) const;
89   unsigned int get_stroke_mesh_id(void) const;
90   bool test_triangle_visibility(StrokeVertexRep *svRep[3]) const;
91   void test_strip_visibility(Strip::vertex_container &strip_vertices,
92                              int *visible_faces,
93                              int *visible_segments) const;
94
95   vector<StrokeRep *> _strokeReps;
96
97 #ifdef WITH_CXX_GUARDEDALLOC
98   MEM_CXX_CLASS_ALLOC_FUNCS("Freestyle:BlenderStrokeRenderer")
99 #endif
100 };
101
102 } /* namespace Freestyle */
103
104 #endif  // __BLENDER_STROKE_RENDERER_H__