a7e7256757db0dba489aa289a68680fe2671b75e
[blender.git] / source / blender / editors / object / object_remesh.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  * The Original Code is Copyright (C) 2019 by Blender Foundation
17  * All rights reserved.
18  */
19
20 /** \file
21  * \ingroup edobj
22  */
23
24 #include <stdlib.h>
25 #include <string.h>
26 #include <math.h>
27 #include <float.h>
28 #include <ctype.h>
29
30 #include "MEM_guardedalloc.h"
31
32 #include "BLI_blenlib.h"
33 #include "BLI_math.h"
34 #include "BLI_utildefines.h"
35
36 #include "DNA_scene_types.h"
37 #include "DNA_object_types.h"
38 #include "DNA_meshdata_types.h"
39 #include "DNA_mesh_types.h"
40
41 #include "BKE_context.h"
42 #include "BKE_global.h"
43 #include "BKE_main.h"
44 #include "BKE_mesh.h"
45 #include "BKE_object.h"
46 #include "BKE_paint.h"
47 #include "BKE_report.h"
48 #include "BKE_scene.h"
49 #include "BKE_customdata.h"
50 #include "BKE_remesh.h"
51
52 #include "DEG_depsgraph.h"
53 #include "DEG_depsgraph_build.h"
54
55 #include "ED_mesh.h"
56 #include "ED_object.h"
57 #include "ED_screen.h"
58 #include "ED_sculpt.h"
59 #include "ED_undo.h"
60
61 #include "RNA_access.h"
62 #include "RNA_define.h"
63 #include "RNA_enum_types.h"
64
65 #include "WM_api.h"
66 #include "WM_types.h"
67 #include "WM_message.h"
68 #include "WM_toolsystem.h"
69
70 #include "object_intern.h"  // own include
71
72 static bool object_remesh_poll(bContext *C)
73 {
74   Object *ob = CTX_data_active_object(C);
75
76   if (BKE_object_is_in_editmode(ob)) {
77     CTX_wm_operator_poll_msg_set(C, "The voxel remesher cannot run from edit mode.");
78     return false;
79   }
80
81   if (ob->mode == OB_MODE_SCULPT && ob->sculpt->bm) {
82     CTX_wm_operator_poll_msg_set(C, "The voxel remesher cannot run with dyntopo activated.");
83   }
84
85   return ED_operator_object_active_editable_mesh(C);
86 }
87
88 static int voxel_remesh_exec(bContext *C, wmOperator *op)
89 {
90   Object *ob = CTX_data_active_object(C);
91   Main *bmain = CTX_data_main(C);
92
93   Mesh *mesh = ob->data;
94   Mesh *new_mesh;
95
96   if (mesh->remesh_voxel_size <= 0.0f) {
97     BKE_report(op->reports, RPT_ERROR, "Voxel remesher cannot run with a voxel size of 0.0.");
98     return OPERATOR_CANCELLED;
99   }
100
101   if (ob->mode == OB_MODE_SCULPT) {
102     ED_sculpt_undo_geometry_begin(ob);
103   }
104
105   new_mesh = BKE_remesh_voxel_to_mesh_nomain(mesh, mesh->remesh_voxel_size);
106
107   if (!new_mesh) {
108     return OPERATOR_CANCELLED;
109   }
110
111   Mesh *obj_mesh_copy;
112   if (mesh->flag & ME_REMESH_REPROJECT_PAINT_MASK) {
113     obj_mesh_copy = BKE_mesh_new_nomain_from_template(mesh, mesh->totvert, 0, 0, 0, 0);
114     CustomData_copy(
115         &mesh->vdata, &obj_mesh_copy->vdata, CD_MASK_MESH.vmask, CD_DUPLICATE, mesh->totvert);
116     for (int i = 0; i < mesh->totvert; i++) {
117       copy_v3_v3(obj_mesh_copy->mvert[i].co, mesh->mvert[i].co);
118     }
119   }
120
121   BKE_mesh_nomain_to_mesh(new_mesh, mesh, ob, &CD_MASK_MESH, true);
122   BKE_mesh_free(new_mesh);
123
124   if (mesh->flag & ME_REMESH_REPROJECT_PAINT_MASK) {
125     BKE_remesh_reproject_paint_mask(mesh, obj_mesh_copy);
126     BKE_mesh_free(obj_mesh_copy);
127   }
128
129   if (mesh->flag & ME_REMESH_SMOOTH_NORMALS) {
130     BKE_mesh_smooth_flag_set(ob, true);
131   }
132
133   if (ob->mode == OB_MODE_SCULPT) {
134     ED_sculpt_undo_geometry_end(ob);
135   }
136
137   BKE_mesh_batch_cache_dirty_tag(ob->data, BKE_MESH_BATCH_DIRTY_ALL);
138   DEG_relations_tag_update(bmain);
139   DEG_id_tag_update(&ob->id, ID_RECALC_GEOMETRY);
140   WM_event_add_notifier(C, NC_GEOM | ND_DATA, ob->data);
141
142   return OPERATOR_FINISHED;
143 }
144
145 void OBJECT_OT_voxel_remesh(wmOperatorType *ot)
146 {
147   /* identifiers */
148   ot->name = "Voxel Remesh";
149   ot->description =
150       "Calculates a new manifold mesh based on the volume of the current mesh. All data layers "
151       "will be lost";
152   ot->idname = "OBJECT_OT_voxel_remesh";
153
154   /* api callbacks */
155   ot->poll = object_remesh_poll;
156   ot->exec = voxel_remesh_exec;
157
158   ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
159 }