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 /** \file
18 * \ingroup collada
19 */
20
21 #include <vector>
22
23 #include "BlenderContext.h"
24 #include "ExportSettings.h"
25
26 #include "BKE_scene.h"
27
bc_is_base_node(LinkNode * export_set,Object * ob,ViewLayer * view_layer)28 bool bc_is_base_node(LinkNode *export_set, Object *ob, ViewLayer *view_layer)
29 {
30 Object *root = bc_get_highest_exported_ancestor_or_self(export_set, ob, view_layer);
31 return (root == ob);
32 }
33
34 /**
35 * Returns the highest selected ancestor
36 * returns NULL if no ancestor is selected
37 * IMPORTANT: This function expects that all exported objects have set:
38 * ob->id.tag & LIB_TAG_DOIT
39 */
bc_get_highest_exported_ancestor_or_self(LinkNode * export_set,Object * ob,ViewLayer * view_layer)40 Object *bc_get_highest_exported_ancestor_or_self(LinkNode *export_set,
41 Object *ob,
42 ViewLayer *view_layer)
43 {
44 Object *ancestor = ob;
45 while (ob->parent) {
46 if (bc_is_in_Export_set(export_set, ob->parent, view_layer)) {
47 ancestor = ob->parent;
48 }
49 ob = ob->parent;
50 }
51 return ancestor;
52 }
53
bc_get_children(std::vector<Object * > & child_set,Object * ob,ViewLayer * view_layer)54 void bc_get_children(std::vector<Object *> &child_set, Object *ob, ViewLayer *view_layer)
55 {
56 Base *base;
57 for (base = (Base *)view_layer->object_bases.first; base; base = base->next) {
58 Object *cob = base->object;
59 if (cob->parent == ob) {
60 switch (ob->type) {
61 case OB_MESH:
62 case OB_CAMERA:
63 case OB_LAMP:
64 case OB_EMPTY:
65 case OB_ARMATURE:
66 child_set.push_back(cob);
67 default:
68 break;
69 }
70 }
71 }
72 }
73
bc_is_in_Export_set(LinkNode * export_set,Object * ob,ViewLayer * view_layer)74 bool bc_is_in_Export_set(LinkNode *export_set, Object *ob, ViewLayer *view_layer)
75 {
76 bool to_export = (BLI_linklist_index(export_set, ob) != -1);
77
78 if (!to_export) {
79 /* Mark this object as to_export even if it is not in the
80 * export list, but it contains children to export. */
81
82 std::vector<Object *> children;
83 bc_get_children(children, ob, view_layer);
84 for (int i = 0; i < children.size(); i++) {
85 if (bc_is_in_Export_set(export_set, children[i], view_layer)) {
86 to_export = true;
87 break;
88 }
89 }
90 }
91 return to_export;
92 }
93
bc_is_marked(Object * ob)94 int bc_is_marked(Object *ob)
95 {
96 return ob && (ob->id.tag & LIB_TAG_DOIT);
97 }
98
bc_remove_mark(Object * ob)99 void bc_remove_mark(Object *ob)
100 {
101 ob->id.tag &= ~LIB_TAG_DOIT;
102 }
103
bc_set_mark(Object * ob)104 void bc_set_mark(Object *ob)
105 {
106 ob->id.tag |= LIB_TAG_DOIT;
107 }
108
BlenderContext(bContext * C)109 BlenderContext::BlenderContext(bContext *C)
110 {
111 context = C;
112 main = CTX_data_main(C);
113 scene = CTX_data_scene(C);
114 view_layer = CTX_data_view_layer(C);
115 depsgraph = nullptr; /* create only when needed */
116 }
117
get_context()118 bContext *BlenderContext::get_context()
119 {
120 return context;
121 }
122
get_depsgraph()123 Depsgraph *BlenderContext::get_depsgraph()
124 {
125 if (!depsgraph) {
126 depsgraph = BKE_scene_ensure_depsgraph(main, scene, view_layer);
127 }
128 return depsgraph;
129 }
130
get_scene()131 Scene *BlenderContext::get_scene()
132 {
133 return scene;
134 }
135
get_evaluated_scene()136 Scene *BlenderContext::get_evaluated_scene()
137 {
138 Scene *scene_eval = DEG_get_evaluated_scene(get_depsgraph());
139 return scene_eval;
140 }
141
get_evaluated_object(Object * ob)142 Object *BlenderContext::get_evaluated_object(Object *ob)
143 {
144 Object *ob_eval = DEG_get_evaluated_object(depsgraph, ob);
145 return ob_eval;
146 }
147
get_view_layer()148 ViewLayer *BlenderContext::get_view_layer()
149 {
150 return view_layer;
151 }
152
get_main()153 Main *BlenderContext::get_main()
154 {
155 return main;
156 }
157