Cleanup: style, use braces for blenkernel
[blender.git] / source / blender / blenkernel / intern / layer_utils.c
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 bke
19  */
20
21 #include <string.h>
22
23 #include "BLI_array.h"
24
25 #include "BKE_collection.h"
26 #include "BKE_editmesh.h"
27 #include "BKE_layer.h"
28
29 #include "DNA_ID.h"
30 #include "DNA_layer_types.h"
31 #include "DNA_object_types.h"
32 #include "DNA_mesh_types.h"
33 #include "DNA_scene_types.h"
34
35 #include "MEM_guardedalloc.h"
36
37 Base **BKE_view_layer_array_from_bases_in_mode_params(ViewLayer *view_layer,
38                                                       View3D *v3d,
39                                                       uint *r_len,
40                                                       const struct ObjectsInModeParams *params)
41 {
42   if (params->no_dup_data) {
43     FOREACH_BASE_IN_MODE_BEGIN (view_layer, v3d, -1, params->object_mode, base_iter) {
44       ID *id = base_iter->object->data;
45       if (id) {
46         id->tag |= LIB_TAG_DOIT;
47       }
48     }
49     FOREACH_BASE_IN_MODE_END;
50   }
51
52   Base **base_array = NULL;
53   BLI_array_declare(base_array);
54
55   FOREACH_BASE_IN_MODE_BEGIN (view_layer, v3d, -1, params->object_mode, base_iter) {
56     if (params->filter_fn) {
57       if (!params->filter_fn(base_iter->object, params->filter_userdata)) {
58         continue;
59       }
60     }
61     if (params->no_dup_data) {
62       ID *id = base_iter->object->data;
63       if (id) {
64         if (id->tag & LIB_TAG_DOIT) {
65           id->tag &= ~LIB_TAG_DOIT;
66         }
67         else {
68           continue;
69         }
70       }
71     }
72     BLI_array_append(base_array, base_iter);
73   }
74   FOREACH_BASE_IN_MODE_END;
75
76   base_array = MEM_reallocN(base_array, sizeof(*base_array) * BLI_array_len(base_array));
77   /* We always need a valid allocation (prevent crash on free). */
78   if (base_array == NULL) {
79     base_array = MEM_mallocN(0, __func__);
80   }
81   *r_len = BLI_array_len(base_array);
82   return base_array;
83 }
84
85 Object **BKE_view_layer_array_from_objects_in_mode_params(ViewLayer *view_layer,
86                                                           View3D *v3d,
87                                                           uint *r_len,
88                                                           const struct ObjectsInModeParams *params)
89 {
90   Base **base_array = BKE_view_layer_array_from_bases_in_mode_params(
91       view_layer, v3d, r_len, params);
92   if (base_array != NULL) {
93     for (uint i = 0; i < *r_len; i++) {
94       ((Object **)base_array)[i] = base_array[i]->object;
95     }
96   }
97   return (Object **)base_array;
98 }
99
100 bool BKE_view_layer_filter_edit_mesh_has_uvs(Object *ob, void *UNUSED(user_data))
101 {
102   if (ob->type == OB_MESH) {
103     Mesh *me = ob->data;
104     BMEditMesh *em = me->edit_mesh;
105     if (em != NULL) {
106       if (CustomData_get_offset(&em->bm->ldata, CD_MLOOPUV) != -1) {
107         return true;
108       }
109     }
110   }
111   return false;
112 }
113
114 bool BKE_view_layer_filter_edit_mesh_has_edges(Object *ob, void *UNUSED(user_data))
115 {
116   if (ob->type == OB_MESH) {
117     Mesh *me = ob->data;
118     BMEditMesh *em = me->edit_mesh;
119     if (em != NULL) {
120       if (em->bm->totedge != 0) {
121         return true;
122       }
123     }
124   }
125   return false;
126 }