Blender  V3.3
BlenderContext.cpp
Go to the documentation of this file.
1 /* SPDX-License-Identifier: GPL-2.0-or-later */
2 
7 #include <vector>
8 
9 #include "BlenderContext.h"
10 #include "ExportSettings.h"
11 
12 #include "BKE_scene.h"
13 
14 bool bc_is_base_node(LinkNode *export_set, Object *ob, ViewLayer *view_layer)
15 {
16  Object *root = bc_get_highest_exported_ancestor_or_self(export_set, ob, view_layer);
17  return (root == ob);
18 }
19 
21  Object *ob,
22  ViewLayer *view_layer)
23 {
24  Object *ancestor = ob;
25  while (ob->parent) {
26  if (bc_is_in_Export_set(export_set, ob->parent, view_layer)) {
27  ancestor = ob->parent;
28  }
29  ob = ob->parent;
30  }
31  return ancestor;
32 }
33 
34 void bc_get_children(std::vector<Object *> &child_set, Object *ob, ViewLayer *view_layer)
35 {
36  Base *base;
37  for (base = (Base *)view_layer->object_bases.first; base; base = base->next) {
38  Object *cob = base->object;
39  if (cob->parent == ob) {
40  switch (ob->type) {
41  case OB_MESH:
42  case OB_CAMERA:
43  case OB_LAMP:
44  case OB_EMPTY:
45  case OB_ARMATURE:
46  child_set.push_back(cob);
47  default:
48  break;
49  }
50  }
51  }
52 }
53 
54 bool bc_is_in_Export_set(LinkNode *export_set, Object *ob, ViewLayer *view_layer)
55 {
56  bool to_export = (BLI_linklist_index(export_set, ob) != -1);
57 
58  if (!to_export) {
59  /* Mark this object as to_export even if it is not in the
60  * export list, but it contains children to export. */
61 
62  std::vector<Object *> children;
63  bc_get_children(children, ob, view_layer);
64  for (Object *child : children) {
65  if (bc_is_in_Export_set(export_set, child, view_layer)) {
66  to_export = true;
67  break;
68  }
69  }
70  }
71  return to_export;
72 }
73 
75 {
76  return ob && (ob->id.tag & LIB_TAG_DOIT);
77 }
78 
80 {
81  ob->id.tag &= ~LIB_TAG_DOIT;
82 }
83 
85 {
86  ob->id.tag |= LIB_TAG_DOIT;
87 }
88 
89 BlenderContext::BlenderContext(bContext *C)
90 {
91  context = C;
92  main = CTX_data_main(C);
94  view_layer = CTX_data_view_layer(C);
95  depsgraph = nullptr; /* create only when needed */
96 }
97 
98 bContext *BlenderContext::get_context()
99 {
100  return context;
101 }
102 
103 Depsgraph *BlenderContext::get_depsgraph()
104 {
105  if (!depsgraph) {
107  }
108  return depsgraph;
109 }
110 
111 Scene *BlenderContext::get_scene()
112 {
113  return scene;
114 }
115 
116 Scene *BlenderContext::get_evaluated_scene()
117 {
118  Scene *scene_eval = DEG_get_evaluated_scene(get_depsgraph());
119  return scene_eval;
120 }
121 
122 Object *BlenderContext::get_evaluated_object(Object *ob)
123 {
124  Object *ob_eval = DEG_get_evaluated_object(depsgraph, ob);
125  return ob_eval;
126 }
127 
128 ViewLayer *BlenderContext::get_view_layer()
129 {
130  return view_layer;
131 }
132 
133 Main *BlenderContext::get_main()
134 {
135  return main;
136 }
struct Scene * CTX_data_scene(const bContext *C)
Definition: context.c:1090
struct ViewLayer * CTX_data_view_layer(const bContext *C)
Definition: context.c:1100
struct Main * CTX_data_main(const bContext *C)
Definition: context.c:1074
struct Depsgraph * BKE_scene_ensure_depsgraph(struct Main *bmain, struct Scene *scene, struct ViewLayer *view_layer)
Definition: scene.cc:3456
void bc_remove_mark(Object *ob)
Object * bc_get_highest_exported_ancestor_or_self(LinkNode *export_set, Object *ob, ViewLayer *view_layer)
int bc_is_marked(Object *ob)
bool bc_is_in_Export_set(LinkNode *export_set, Object *ob, ViewLayer *view_layer)
void bc_set_mark(Object *ob)
void bc_get_children(std::vector< Object * > &child_set, Object *ob, ViewLayer *view_layer)
bool bc_is_base_node(LinkNode *export_set, Object *ob, ViewLayer *view_layer)
struct Depsgraph Depsgraph
Definition: DEG_depsgraph.h:35
struct Object * DEG_get_evaluated_object(const struct Depsgraph *depsgraph, struct Object *object)
struct Scene * DEG_get_evaluated_scene(const struct Depsgraph *graph)
@ LIB_TAG_DOIT
Definition: DNA_ID.h:707
@ OB_EMPTY
@ OB_CAMERA
@ OB_ARMATURE
@ OB_LAMP
@ OB_MESH
#define C
Definition: RandGen.cpp:25
int main(int argc, char *argv[])
Scene scene
const Depsgraph * depsgraph
struct Base * next
struct Object * object
int tag
Definition: DNA_ID.h:387
void * first
Definition: DNA_listBase.h:31
Definition: BKE_main.h:121
struct Object * parent
ListBase object_bases