Fix T52134: New depsgraph crashes when evaluating several psys on from object
[blender.git] / source / blender / depsgraph / intern / depsgraph_type_defines.cc
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: Joshua Leung
22  * Contributor(s): None Yet
23  *
24  * ***** END GPL LICENSE BLOCK *****
25  */
26
27 /** \file blender/depsgraph/intern/depsgraph_type_defines.cc
28  *  \ingroup depsgraph
29  *
30  * Defines and code for core node types.
31  */
32
33 #include <cstdlib>  // for BLI_assert()
34
35
36 #include "BLI_utildefines.h"
37 #include "BLI_ghash.h"
38
39 #include "DEG_depsgraph.h"
40
41 #include "intern/nodes/deg_node.h"
42 #include "intern/nodes/deg_node_component.h"
43 #include "intern/nodes/deg_node_operation.h"
44
45 #include "intern/depsgraph_intern.h"
46
47 namespace DEG {
48
49 /* ************ */
50 /* External API */
51
52 /* Global type registry */
53
54 /**
55  * \note For now, this is a hashtable not array, since the core node types
56  * currently do not have contiguous ID values. Using a hash here gives us
57  * more flexibility, albeit using more memory and also sacrificing a little
58  * speed. Later on, when things stabilise we may turn this back to an array
59  * since there are only just a few node types that an array would cope fine...
60  */
61 static GHash *_depsnode_typeinfo_registry = NULL;
62
63 /* Registration ------------------------------------------- */
64
65 /* Register node type */
66 void deg_register_node_typeinfo(DepsNodeFactory *factory)
67 {
68         BLI_assert(factory != NULL);
69         BLI_ghash_insert(_depsnode_typeinfo_registry, SET_INT_IN_POINTER(factory->type()), factory);
70 }
71
72 /* Getters ------------------------------------------------- */
73
74 /* Get typeinfo for specified type */
75 DepsNodeFactory *deg_get_node_factory(const eDepsNode_Type type)
76 {
77         /* look up type - at worst, it doesn't exist in table yet, and we fail */
78         return (DepsNodeFactory *)BLI_ghash_lookup(_depsnode_typeinfo_registry, SET_INT_IN_POINTER(type));
79 }
80
81 /* Get typeinfo for provided node */
82 DepsNodeFactory *deg_node_get_factory(const DepsNode *node)
83 {
84         if (node != NULL) {
85                 return NULL;
86         }
87         return deg_get_node_factory(node->type);
88 }
89
90 /* Stringified opcodes ------------------------------------- */
91
92 DepsOperationStringifier DEG_OPNAMES;
93
94 static const char *stringify_opcode(eDepsOperation_Code opcode)
95 {
96         switch (opcode) {
97 #define STRINGIFY_OPCODE(name) case DEG_OPCODE_##name: return #name
98                 STRINGIFY_OPCODE(OPERATION);
99                 STRINGIFY_OPCODE(PLACEHOLDER);
100                 STRINGIFY_OPCODE(ANIMATION);
101                 STRINGIFY_OPCODE(DRIVER);
102                 STRINGIFY_OPCODE(TRANSFORM_LOCAL);
103                 STRINGIFY_OPCODE(TRANSFORM_PARENT);
104                 STRINGIFY_OPCODE(TRANSFORM_CONSTRAINTS);
105                 STRINGIFY_OPCODE(RIGIDBODY_REBUILD);
106                 STRINGIFY_OPCODE(RIGIDBODY_SIM);
107                 STRINGIFY_OPCODE(TRANSFORM_RIGIDBODY);
108                 STRINGIFY_OPCODE(TRANSFORM_FINAL);
109                 STRINGIFY_OPCODE(OBJECT_UBEREVAL);
110                 STRINGIFY_OPCODE(GEOMETRY_UBEREVAL);
111                 STRINGIFY_OPCODE(GEOMETRY_PATH);
112                 STRINGIFY_OPCODE(POSE_INIT);
113                 STRINGIFY_OPCODE(POSE_DONE);
114                 STRINGIFY_OPCODE(POSE_IK_SOLVER);
115                 STRINGIFY_OPCODE(POSE_SPLINE_IK_SOLVER);
116                 STRINGIFY_OPCODE(BONE_LOCAL);
117                 STRINGIFY_OPCODE(BONE_POSE_PARENT);
118                 STRINGIFY_OPCODE(BONE_CONSTRAINTS);
119                 STRINGIFY_OPCODE(BONE_READY);
120                 STRINGIFY_OPCODE(BONE_DONE);
121                 STRINGIFY_OPCODE(PSYS_EVAL);
122                 STRINGIFY_OPCODE(PSYS_EVAL_INIT);
123
124                 case DEG_NUM_OPCODES: return "SpecialCase";
125 #undef STRINGIFY_OPCODE
126         }
127         return "UNKNOWN";
128 }
129
130 DepsOperationStringifier::DepsOperationStringifier()
131 {
132         for (int i = 0; i < DEG_NUM_OPCODES; ++i) {
133                 names_[i] = stringify_opcode((eDepsOperation_Code)i);
134         }
135 }
136
137 const char *DepsOperationStringifier::operator[](eDepsOperation_Code opcode)
138 {
139         BLI_assert((opcode >= 0) && (opcode < DEG_NUM_OPCODES));
140         if (opcode >= 0 && opcode < DEG_NUM_OPCODES) {
141                 return names_[opcode];
142         }
143         return "UnknownOpcode";
144 }
145
146 }  // namespace DEG
147
148 /* Register all node types */
149 void DEG_register_node_types(void)
150 {
151         /* initialise registry */
152         DEG::_depsnode_typeinfo_registry = BLI_ghash_int_new("Depsgraph Node Type Registry");
153
154         /* register node types */
155         DEG::deg_register_base_depsnodes();
156         DEG::deg_register_component_depsnodes();
157         DEG::deg_register_operation_depsnodes();
158 }
159
160 /* Free registry on exit */
161 void DEG_free_node_types(void)
162 {
163         BLI_ghash_free(DEG::_depsnode_typeinfo_registry, NULL, NULL);
164 }