diff options
Diffstat (limited to 'source/blender/depsgraph/intern/depsgraph_query.cc')
-rw-r--r-- | source/blender/depsgraph/intern/depsgraph_query.cc | 186 |
1 files changed, 180 insertions, 6 deletions
diff --git a/source/blender/depsgraph/intern/depsgraph_query.cc b/source/blender/depsgraph/intern/depsgraph_query.cc index 9b1961baa48..ca9f32d4d8c 100644 --- a/source/blender/depsgraph/intern/depsgraph_query.cc +++ b/source/blender/depsgraph/intern/depsgraph_query.cc @@ -33,21 +33,74 @@ #include "MEM_guardedalloc.h" extern "C" { +#include <string.h> // XXX: memcpy + +#include "BLI_utildefines.h" #include "BKE_idcode.h" #include "BKE_main.h" +#include "BLI_listbase.h" + +#include "BKE_action.h" // XXX: BKE_pose_channel_from_name } /* extern "C" */ +#include "DNA_object_types.h" +#include "DNA_scene_types.h" + +#include "RNA_access.h" + +#include "DEG_depsgraph.h" #include "DEG_depsgraph_query.h" +#include "intern/eval/deg_eval_copy_on_write.h" #include "intern/depsgraph_intern.h" #include "intern/nodes/deg_node_id.h" -bool DEG_id_type_tagged(Main *bmain, short idtype) +struct Scene *DEG_get_input_scene(const Depsgraph *graph) { - return bmain->id_tag_update[BKE_idcode_to_index(idtype)] != 0; + const DEG::Depsgraph *deg_graph = reinterpret_cast<const DEG::Depsgraph *>(graph); + return deg_graph->scene; } -short DEG_get_eval_flags_for_id(Depsgraph *graph, ID *id) +struct ViewLayer *DEG_get_input_view_layer(const Depsgraph *graph) +{ + const DEG::Depsgraph *deg_graph = reinterpret_cast<const DEG::Depsgraph *>(graph); + return deg_graph->view_layer; +} + +eEvaluationMode DEG_get_mode(const Depsgraph *graph) +{ + const DEG::Depsgraph *deg_graph = reinterpret_cast<const DEG::Depsgraph *>(graph); + return deg_graph->mode; +} + +float DEG_get_ctime(const Depsgraph *graph) +{ + const DEG::Depsgraph *deg_graph = reinterpret_cast<const DEG::Depsgraph *>(graph); + return deg_graph->ctime; +} + + +bool DEG_id_type_updated(const Depsgraph *graph, short id_type) +{ + const DEG::Depsgraph *deg_graph = reinterpret_cast<const DEG::Depsgraph *>(graph); + return deg_graph->id_type_updated[BKE_idcode_to_index(id_type)] != 0; +} + +bool DEG_id_type_any_updated(const Depsgraph *graph) +{ + const DEG::Depsgraph *deg_graph = reinterpret_cast<const DEG::Depsgraph *>(graph); + + /* Loop over all ID types. */ + for (int id_type_index = 0; id_type_index < MAX_LIBARRAY; id_type_index++) { + if (deg_graph->id_type_updated[id_type_index]) { + return true; + } + } + + return false; +} + +short DEG_get_eval_flags_for_id(const Depsgraph *graph, ID *id) { if (graph == NULL) { /* Happens when converting objects to mesh from a python script @@ -59,9 +112,8 @@ short DEG_get_eval_flags_for_id(Depsgraph *graph, ID *id) return 0; } - DEG::Depsgraph *deg_graph = reinterpret_cast<DEG::Depsgraph *>(graph); - - DEG::IDDepsNode *id_node = deg_graph->find_id_node(id); + const DEG::Depsgraph *deg_graph = reinterpret_cast<const DEG::Depsgraph *>(graph); + const DEG::IDDepsNode *id_node = deg_graph->find_id_node(id); if (id_node == NULL) { /* TODO(sergey): Does it mean we need to check set scene? */ return 0; @@ -69,3 +121,125 @@ short DEG_get_eval_flags_for_id(Depsgraph *graph, ID *id) return id_node->eval_flags; } + +Scene *DEG_get_evaluated_scene(const Depsgraph *graph) +{ + const DEG::Depsgraph *deg_graph = + reinterpret_cast<const DEG::Depsgraph *>(graph); + Scene *scene_cow = deg_graph->scene_cow; + /* TODO(sergey): Shall we expand datablock here? Or is it OK to assume + * that calleer is OK with just a pointer in case scene is not up[dated + * yet? + */ + BLI_assert(DEG::deg_copy_on_write_is_expanded(&scene_cow->id)); + return scene_cow; +} + +ViewLayer *DEG_get_evaluated_view_layer(const Depsgraph *graph) +{ + const DEG::Depsgraph *deg_graph = + reinterpret_cast<const DEG::Depsgraph *>(graph); + Scene *scene_cow = DEG_get_evaluated_scene(graph); + /* Do name-based lookup. */ + /* TODO(sergey): Can this be optimized? */ + ViewLayer *view_layer_orig = deg_graph->view_layer; + ViewLayer *view_layer_cow = + (ViewLayer *)BLI_findstring(&scene_cow->view_layers, + view_layer_orig->name, + offsetof(ViewLayer, name)); + BLI_assert(view_layer_cow != NULL); + return view_layer_cow; +} + +Object *DEG_get_evaluated_object(const Depsgraph *depsgraph, Object *object) +{ + return (Object *)DEG_get_evaluated_id(depsgraph, &object->id); +} + +ID *DEG_get_evaluated_id(const Depsgraph *depsgraph, ID *id) +{ + if (id == NULL) { + return NULL; + } + /* TODO(sergey): This is a duplicate of Depsgraph::get_cow_id(), + * but here we never do assert, since we don't know nature of the + * incoming ID datablock. + */ + const DEG::Depsgraph *deg_graph = (const DEG::Depsgraph *)depsgraph; + const DEG::IDDepsNode *id_node = deg_graph->find_id_node(id); + if (id_node == NULL) { + return id; + } + return id_node->id_cow; +} + +/* Get evaluated version of data pointed to by RNA pointer */ +void DEG_get_evaluated_rna_pointer(const Depsgraph *depsgraph, PointerRNA *ptr, PointerRNA *r_ptr_eval) +{ + if ((ptr == NULL) || (r_ptr_eval == NULL)) { + return; + } + ID *orig_id = (ID *)ptr->id.data; + ID *cow_id = DEG_get_evaluated_id(depsgraph, orig_id); + if (ptr->id.data == ptr->data) { + /* For ID pointers, it's easy... */ + r_ptr_eval->id.data = (void *)cow_id; + r_ptr_eval->data = (void *)cow_id; + r_ptr_eval->type = ptr->type; + } + else if (ptr->type == &RNA_PoseBone) { + /* HACK: Since bone keyframing is quite commonly used, + * speed things up for this case by doing a special lookup + * for bones + */ + const Object *ob_eval = (Object *)cow_id; + bPoseChannel *pchan = (bPoseChannel *)ptr->data; + const bPoseChannel *pchan_eval = BKE_pose_channel_find_name(ob_eval->pose, pchan->name); + r_ptr_eval->id.data = (void *)cow_id; + r_ptr_eval->data = (void *)pchan_eval; + r_ptr_eval->type = ptr->type; + } + else { + /* For everything else, try to get RNA Path of the BMain-pointer, + * then use that to look up what the COW-domain one should be + * given the COW ID pointer as the new lookup point + */ + /* TODO: Find a faster alternative, or implement support for other + * common types too above (e.g. modifiers) + */ + char *path = RNA_path_from_ID_to_struct(ptr); + if (path) { + PointerRNA cow_id_ptr; + RNA_id_pointer_create(cow_id, &cow_id_ptr); + if (!RNA_path_resolve(&cow_id_ptr, path, r_ptr_eval, NULL)) { + /* Couldn't find COW copy of data */ + fprintf(stderr, + "%s: Couldn't resolve RNA path ('%s') relative to COW ID (%p) for '%s'\n", + __func__, path, (void *)cow_id, orig_id->name); + } + } + else { + /* Path resolution failed - XXX: Hide this behind a debug flag */ + fprintf(stderr, + "%s: Couldn't get RNA path for %s relative to %s\n", + __func__, RNA_struct_identifier(ptr->type), orig_id->name); + } + } +} + +Object *DEG_get_original_object(Object *object) +{ + return (Object *)DEG_get_original_id(&object->id); +} + +ID *DEG_get_original_id(ID *id) +{ + if (id == NULL) { + return NULL; + } + if (id->orig_id == NULL) { + return id; + } + BLI_assert((id->tag & LIB_TAG_COPIED_ON_WRITE) != 0); + return (ID *)id->orig_id; +} |