c0bf82becdaca2feb63ac74cf06453a370f50dbe
[blender-staging.git] / source / blender / depsgraph / intern / builder / deg_builder_relations.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  * The Original Code is Copyright (C) 2013 Blender Foundation.
19  * All rights reserved.
20  *
21  * Original Author: Lukas Toenne
22  * Contributor(s): None Yet
23  *
24  * ***** END GPL LICENSE BLOCK *****
25  */
26
27 /** \file blender/depsgraph/intern/builder/deg_builder_relations.h
28  *  \ingroup depsgraph
29  */
30
31 #pragma once
32
33 #include <cstdio>
34
35 #include "intern/depsgraph_types.h"
36
37 #include "DNA_ID.h"
38
39 #include "RNA_access.h"
40 #include "RNA_types.h"
41
42 #include "BLI_utildefines.h"
43 #include "BLI_string.h"
44
45 #include "intern/depsgraph_types.h"
46 #include "intern/nodes/deg_node.h"
47 #include "intern/nodes/deg_node_operation.h"
48
49 struct Base;
50 struct bGPdata;
51 struct ListBase;
52 struct GHash;
53 struct ID;
54 struct FCurve;
55 struct Group;
56 struct Key;
57 struct Main;
58 struct Material;
59 struct MTex;
60 struct bNodeTree;
61 struct Object;
62 struct bPoseChannel;
63 struct bConstraint;
64 struct Scene;
65 struct Tex;
66 struct World;
67
68 struct PropertyRNA;
69
70 namespace DEG {
71
72 struct Depsgraph;
73 struct DepsNode;
74 struct DepsNodeHandle;
75 struct RootDepsNode;
76 struct SubgraphDepsNode;
77 struct IDDepsNode;
78 struct TimeSourceDepsNode;
79 struct ComponentDepsNode;
80 struct OperationDepsNode;
81 struct RootPChanMap;
82
83 struct RootKey
84 {
85         RootKey() {}
86 };
87
88 struct TimeSourceKey
89 {
90         TimeSourceKey() : id(NULL) {}
91         TimeSourceKey(ID *id) : id(id) {}
92
93         string identifier() const
94         {
95                 return string("TimeSourceKey");
96         }
97
98         ID *id;
99 };
100
101 struct ComponentKey
102 {
103         ComponentKey() :
104             id(NULL), type(DEPSNODE_TYPE_UNDEFINED), name("")
105         {}
106         ComponentKey(ID *id, eDepsNode_Type type, const string &name = "") :
107             id(id), type(type), name(name)
108         {}
109
110         string identifier() const
111         {
112                 const char *idname = (id) ? id->name : "<None>";
113
114                 char typebuf[5];
115                 BLI_snprintf(typebuf, sizeof(typebuf), "%d", type);
116
117                 return string("ComponentKey(") + idname + ", " + typebuf + ", '" + name + "')";
118         }
119
120         ID *id;
121         eDepsNode_Type type;
122         string name;
123 };
124
125 struct OperationKey
126 {
127         OperationKey() :
128             id(NULL), component_type(DEPSNODE_TYPE_UNDEFINED), component_name(""), opcode(DEG_OPCODE_OPERATION), name("")
129         {}
130
131         OperationKey(ID *id, eDepsNode_Type component_type, const string &name) :
132             id(id), component_type(component_type), component_name(""), opcode(DEG_OPCODE_OPERATION), name(name)
133         {}
134         OperationKey(ID *id, eDepsNode_Type component_type, const string &component_name, const string &name) :
135             id(id), component_type(component_type), component_name(component_name), opcode(DEG_OPCODE_OPERATION), name(name)
136         {}
137
138         OperationKey(ID *id, eDepsNode_Type component_type, eDepsOperation_Code opcode) :
139             id(id), component_type(component_type), component_name(""), opcode(opcode), name("")
140         {}
141         OperationKey(ID *id, eDepsNode_Type component_type, const string &component_name, eDepsOperation_Code opcode) :
142             id(id), component_type(component_type), component_name(component_name), opcode(opcode), name("")
143         {}
144
145         OperationKey(ID *id, eDepsNode_Type component_type, eDepsOperation_Code opcode, const string &name) :
146             id(id), component_type(component_type), component_name(""), opcode(opcode), name(name)
147         {}
148         OperationKey(ID *id, eDepsNode_Type component_type, const string &component_name, eDepsOperation_Code opcode, const string &name) :
149             id(id), component_type(component_type), component_name(component_name), opcode(opcode), name(name)
150         {}
151
152         string identifier() const
153         {
154                 char typebuf[5];
155                 BLI_snprintf(typebuf, sizeof(typebuf), "%d", component_type);
156
157                 return string("OperationKey(") + "t: " + typebuf + ", cn: '" + component_name + "', c: " + DEG_OPNAMES[opcode] + ", n: '" + name + "')";
158         }
159
160
161         ID *id;
162         eDepsNode_Type component_type;
163         string component_name;
164         eDepsOperation_Code opcode;
165         string name;
166 };
167
168 struct RNAPathKey
169 {
170         // Note: see depsgraph_build.cpp for implementation
171         RNAPathKey(ID *id, const char *path);
172
173         RNAPathKey(ID *id, const PointerRNA &ptr, PropertyRNA *prop) :
174             id(id), ptr(ptr), prop(prop)
175         {}
176
177         string identifier() const
178         {
179                 const char *id_name   = (id) ?  id->name : "<No ID>";
180                 const char *prop_name = (prop) ? RNA_property_identifier(prop) : "<No Prop>";
181
182                 return string("RnaPathKey(") + "id: " + id_name + ", prop: " + prop_name +  "')";
183         }
184
185
186         ID *id;
187         PointerRNA ptr;
188         PropertyRNA *prop;
189 };
190
191 struct DepsgraphRelationBuilder
192 {
193         DepsgraphRelationBuilder(Depsgraph *graph);
194
195         template <typename KeyFrom, typename KeyTo>
196         void add_relation(const KeyFrom& key_from,
197                           const KeyTo& key_to,
198                           eDepsRelation_Type type,
199                           const char *description);
200
201         template <typename KeyTo>
202         void add_relation(const TimeSourceKey& key_from,
203                           const KeyTo& key_to,
204                           eDepsRelation_Type type,
205                           const char *description);
206
207         template <typename KeyType>
208         void add_node_handle_relation(const KeyType& key_from,
209                                       const DepsNodeHandle *handle,
210                                       eDepsRelation_Type type,
211                                       const char *description);
212
213         void build_scene(Main *bmain, Scene *scene);
214         void build_group(Main *bmain, Scene *scene, Object *object, Group *group);
215         void build_object(Main *bmain, Scene *scene, Object *ob);
216         void build_object_parent(Object *ob);
217         void build_constraints(Scene *scene, ID *id,
218                                eDepsNode_Type component_type,
219                                const char *component_subdata,
220                                ListBase *constraints,
221                                RootPChanMap *root_map);
222         void build_animdata(ID *id);
223         void build_driver(ID *id, FCurve *fcurve);
224         void build_world(World *world);
225         void build_rigidbody(Scene *scene);
226         void build_particles(Scene *scene, Object *ob);
227         void build_ik_pose(Object *ob,
228                            bPoseChannel *pchan,
229                            bConstraint *con,
230                            RootPChanMap *root_map);
231         void build_splineik_pose(Object *ob,
232                                  bPoseChannel *pchan,
233                                  bConstraint *con,
234                                  RootPChanMap *root_map);
235         void build_rig(Scene *scene, Object *ob);
236         void build_proxy_rig(Object *ob);
237         void build_shapekeys(ID *obdata, Key *key);
238         void build_obdata_geom(Main *bmain, Scene *scene, Object *ob);
239         void build_camera(Object *ob);
240         void build_lamp(Object *ob);
241         void build_nodetree(ID *owner, bNodeTree *ntree);
242         void build_material(ID *owner, Material *ma);
243         void build_texture(ID *owner, Tex *tex);
244         void build_texture_stack(ID *owner, MTex **texture_stack);
245         void build_compositor(Scene *scene);
246         void build_gpencil(ID *owner, bGPdata *gpd);
247
248         template <typename KeyType>
249         OperationDepsNode *find_operation_node(const KeyType &key);
250
251 protected:
252         RootDepsNode *find_node(const RootKey &key) const;
253         TimeSourceDepsNode *find_node(const TimeSourceKey &key) const;
254         ComponentDepsNode *find_node(const ComponentKey &key) const;
255         OperationDepsNode *find_node(const OperationKey &key) const;
256         DepsNode *find_node(const RNAPathKey &key) const;
257         OperationDepsNode *has_node(const OperationKey &key) const;
258
259         void add_time_relation(TimeSourceDepsNode *timesrc,
260                                DepsNode *node_to,
261                                const char *description);
262         void add_operation_relation(OperationDepsNode *node_from,
263                                     OperationDepsNode *node_to,
264                                     eDepsRelation_Type type,
265                                     const char *description);
266
267         template <typename KeyType>
268         DepsNodeHandle create_node_handle(const KeyType& key,
269                                           const string& default_name = "");
270
271         bool needs_animdata_node(ID *id);
272
273 private:
274         Depsgraph *m_graph;
275 };
276
277 struct DepsNodeHandle
278 {
279         DepsNodeHandle(DepsgraphRelationBuilder *builder, OperationDepsNode *node, const string &default_name = "") :
280             builder(builder),
281             node(node),
282             default_name(default_name)
283         {
284                 BLI_assert(node != NULL);
285         }
286
287         DepsgraphRelationBuilder *builder;
288         OperationDepsNode *node;
289         const string &default_name;
290 };
291
292 /* Utilities for Builders ----------------------------------------------------- */
293
294 template <typename KeyType>
295 OperationDepsNode *DepsgraphRelationBuilder::find_operation_node(const KeyType& key) {
296         DepsNode *node = find_node(key);
297         return node != NULL ? node->get_exit_operation() : NULL;
298 }
299
300 template <typename KeyFrom, typename KeyTo>
301 void DepsgraphRelationBuilder::add_relation(const KeyFrom &key_from,
302                                             const KeyTo &key_to,
303                                             eDepsRelation_Type type,
304                                             const char *description)
305 {
306         DepsNode *node_from = find_node(key_from);
307         DepsNode *node_to = find_node(key_to);
308         OperationDepsNode *op_from = node_from ? node_from->get_exit_operation() : NULL;
309         OperationDepsNode *op_to = node_to ? node_to->get_entry_operation() : NULL;
310         if (op_from && op_to) {
311                 add_operation_relation(op_from, op_to, type, description);
312         }
313         else {
314                 if (!op_from) {
315                         /* XXX TODO handle as error or report if needed */
316                         fprintf(stderr, "add_relation(%d, %s) - Could not find op_from (%s)\n",
317                                 type, description, key_from.identifier().c_str());
318                 }
319                 else {
320                         fprintf(stderr, "add_relation(%d, %s) - Failed, but op_from (%s) was ok\n",
321                                 type, description, key_from.identifier().c_str());
322                 }
323                 if (!op_to) {
324                         /* XXX TODO handle as error or report if needed */
325                         fprintf(stderr, "add_relation(%d, %s) - Could not find op_to (%s)\n",
326                                 type, description, key_to.identifier().c_str());
327                 }
328                 else {
329                         fprintf(stderr, "add_relation(%d, %s) - Failed, but op_to (%s) was ok\n",
330                                 type, description, key_to.identifier().c_str());
331                 }
332         }
333 }
334
335 template <typename KeyTo>
336 void DepsgraphRelationBuilder::add_relation(const TimeSourceKey &key_from,
337                                             const KeyTo &key_to,
338                                             eDepsRelation_Type type,
339                                             const char *description)
340 {
341         (void)type;  /* Ignored in release builds. */
342         BLI_assert(type == DEPSREL_TYPE_TIME);
343         TimeSourceDepsNode *time_from = find_node(key_from);
344         DepsNode *node_to = find_node(key_to);
345         OperationDepsNode *op_to = node_to ? node_to->get_entry_operation() : NULL;
346         if (time_from && op_to) {
347                 add_time_relation(time_from, op_to, description);
348         }
349         else {
350         }
351 }
352
353 template <typename KeyType>
354 void DepsgraphRelationBuilder::add_node_handle_relation(
355         const KeyType &key_from,
356         const DepsNodeHandle *handle,
357         eDepsRelation_Type type,
358         const char *description)
359 {
360         DepsNode *node_from = find_node(key_from);
361         OperationDepsNode *op_from = node_from ? node_from->get_exit_operation() : NULL;
362         OperationDepsNode *op_to = handle->node->get_entry_operation();
363         if (op_from && op_to) {
364                 add_operation_relation(op_from, op_to, type, description);
365         }
366         else {
367                 if (!op_from) {
368                         /* XXX TODO handle as error or report if needed */
369                 }
370                 if (!op_to) {
371                         /* XXX TODO handle as error or report if needed */
372                 }
373         }
374 }
375
376 template <typename KeyType>
377 DepsNodeHandle DepsgraphRelationBuilder::create_node_handle(
378         const KeyType &key,
379         const string &default_name)
380 {
381         return DepsNodeHandle(this, find_node(key), default_name);
382 }
383
384 }  // namespace DEG