Threaded object update and EvaluationContext
[blender.git] / source / blender / editors / object / object_modifier.c
1 /*
2  * ***** BEGIN GPL LICENSE BLOCK *****
3  *
4  * This program is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU General Public License
6  * as published by the Free Software Foundation; either version 2
7  * of the License, or (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program; if not, write to the Free Software Foundation,
16  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
17  *
18  * The Original Code is Copyright (C) 2001-2002 by NaN Holding BV.
19  * All rights reserved.
20  *
21  * Contributor(s): Blender Foundation, 2009
22  *
23  * ***** END GPL LICENSE BLOCK *****
24  */
25
26 /** \file blender/editors/object/object_modifier.c
27  *  \ingroup edobj
28  */
29
30
31 #include <math.h>
32 #include <stdio.h>
33 #include <stdlib.h>
34
35 #include "MEM_guardedalloc.h"
36
37 #include "DNA_anim_types.h"
38 #include "DNA_armature_types.h"
39 #include "DNA_curve_types.h"
40 #include "DNA_key_types.h"
41 #include "DNA_mesh_types.h"
42 #include "DNA_meshdata_types.h"
43 #include "DNA_object_force.h"
44 #include "DNA_scene_types.h"
45
46 #include "BLI_bitmap.h"
47 #include "BLI_math.h"
48 #include "BLI_listbase.h"
49 #include "BLI_string.h"
50 #include "BLI_string_utf8.h"
51 #include "BLI_path_util.h"
52 #include "BLI_utildefines.h"
53
54 #include "BKE_animsys.h"
55 #include "BKE_curve.h"
56 #include "BKE_context.h"
57 #include "BKE_depsgraph.h"
58 #include "BKE_displist.h"
59 #include "BKE_DerivedMesh.h"
60 #include "BKE_effect.h"
61 #include "BKE_global.h"
62 #include "BKE_key.h"
63 #include "BKE_lattice.h"
64 #include "BKE_main.h"
65 #include "BKE_mesh.h"
66 #include "BKE_mesh_mapping.h"
67 #include "BKE_modifier.h"
68 #include "BKE_multires.h"
69 #include "BKE_report.h"
70 #include "BKE_object.h"
71 #include "BKE_ocean.h"
72 #include "BKE_particle.h"
73 #include "BKE_softbody.h"
74 #include "BKE_editmesh.h"
75
76 #include "RNA_access.h"
77 #include "RNA_define.h"
78 #include "RNA_enum_types.h"
79
80 #include "ED_armature.h"
81 #include "ED_object.h"
82 #include "ED_screen.h"
83 #include "ED_sculpt.h"
84 #include "ED_mesh.h"
85
86 #include "WM_api.h"
87 #include "WM_types.h"
88
89 #include "object_intern.h"
90
91 static void modifier_skin_customdata_delete(struct Object *ob);
92
93 /******************************** API ****************************/
94
95 ModifierData *ED_object_modifier_add(ReportList *reports, Main *bmain, Scene *scene, Object *ob, const char *name, int type)
96 {
97         ModifierData *md = NULL, *new_md = NULL;
98         ModifierTypeInfo *mti = modifierType_getInfo(type);
99         
100         /* only geometry objects should be able to get modifiers [#25291] */
101         if (!ELEM5(ob->type, OB_MESH, OB_CURVE, OB_SURF, OB_FONT, OB_LATTICE)) {
102                 BKE_reportf(reports, RPT_WARNING, "Modifiers cannot be added to object '%s'", ob->id.name + 2);
103                 return NULL;
104         }
105         
106         if (mti->flags & eModifierTypeFlag_Single) {
107                 if (modifiers_findByType(ob, type)) {
108                         BKE_report(reports, RPT_WARNING, "Only one modifier of this type is allowed");
109                         return NULL;
110                 }
111         }
112         
113         if (type == eModifierType_ParticleSystem) {
114                 /* don't need to worry about the new modifier's name, since that is set to the number
115                  * of particle systems which shouldn't have too many duplicates 
116                  */
117                 new_md = object_add_particle_system(scene, ob, name);
118         }
119         else {
120                 /* get new modifier data to add */
121                 new_md = modifier_new(type);
122                 
123                 if (mti->flags & eModifierTypeFlag_RequiresOriginalData) {
124                         md = ob->modifiers.first;
125                         
126                         while (md && modifierType_getInfo(md->type)->type == eModifierTypeType_OnlyDeform)
127                                 md = md->next;
128                         
129                         BLI_insertlinkbefore(&ob->modifiers, md, new_md);
130                 }
131                 else
132                         BLI_addtail(&ob->modifiers, new_md);
133
134                 if (name) {
135                         BLI_strncpy_utf8(new_md->name, name, sizeof(new_md->name));
136                 }
137
138                 /* make sure modifier data has unique name */
139
140                 modifier_unique_name(&ob->modifiers, new_md);
141                 
142                 /* special cases */
143                 if (type == eModifierType_Softbody) {
144                         if (!ob->soft) {
145                                 ob->soft = sbNew(scene);
146                                 ob->softflag |= OB_SB_GOAL | OB_SB_EDGES;
147                         }
148                 }
149                 else if (type == eModifierType_Collision) {
150                         if (!ob->pd)
151                                 ob->pd = object_add_collision_fields(0);
152                         
153                         ob->pd->deflect = 1;
154                         DAG_relations_tag_update(bmain);
155                 }
156                 else if (type == eModifierType_Surface)
157                         DAG_relations_tag_update(bmain);
158                 else if (type == eModifierType_Multires) {
159                         /* set totlvl from existing MDISPS layer if object already had it */
160                         multiresModifier_set_levels_from_disps((MultiresModifierData *)new_md, ob);
161
162                         if (ob->mode & OB_MODE_SCULPT) {
163                                 /* ensure that grid paint mask layer is created */
164                                 ED_sculpt_mask_layers_ensure(ob, (MultiresModifierData *)new_md);
165                         }
166                 }
167                 else if (type == eModifierType_Skin) {
168                         /* ensure skin-node customdata exists */
169                         modifier_skin_customdata_ensure(ob);
170                 }
171         }
172
173         DAG_id_tag_update(&ob->id, OB_RECALC_DATA);
174
175         return new_md;
176 }
177
178 /* Return TRUE if the object has a modifier of type 'type' other than
179  * the modifier pointed to be 'exclude', otherwise returns FALSE. */
180 static int object_has_modifier(const Object *ob, const ModifierData *exclude,
181                                ModifierType type)
182 {
183         ModifierData *md;
184
185         for (md = ob->modifiers.first; md; md = md->next) {
186                 if ((md != exclude) && (md->type == type))
187                         return TRUE;
188         }
189
190         return FALSE;
191 }
192
193 /* If the object data of 'orig_ob' has other users, run 'callback' on
194  * each of them.
195  *
196  * If include_orig is TRUE, the callback will run on 'orig_ob' too.
197  * 
198  * If the callback ever returns TRUE, iteration will stop and the
199  * function value will be TRUE. Otherwise the function returns FALSE.
200  */
201 int ED_object_iter_other(Main *bmain, Object *orig_ob, const bool include_orig,
202                          int (*callback)(Object *ob, void *callback_data),
203                          void *callback_data)
204 {
205         ID *ob_data_id = orig_ob->data;
206         int users = ob_data_id->us;
207
208         if (ob_data_id->flag & LIB_FAKEUSER)
209                 users--;
210
211         /* First check that the object's data has multiple users */
212         if (users > 1) {
213                 Object *ob;
214                 int totfound = include_orig ? 0 : 1;
215
216                 for (ob = bmain->object.first; ob && totfound < users;
217                      ob = ob->id.next)
218                 {
219                         if (((ob != orig_ob) || include_orig) &&
220                             (ob->data == orig_ob->data))
221                         {
222                                 if (callback(ob, callback_data))
223                                         return TRUE;
224
225                                 totfound++;
226                         }
227                 }
228         }
229         else if (include_orig) {
230                 return callback(orig_ob, callback_data);
231         }
232
233         return FALSE;
234 }
235
236 static int object_has_modifier_cb(Object *ob, void *data)
237 {
238         ModifierType type = *((ModifierType *)data);
239
240         return object_has_modifier(ob, NULL, type);
241 }
242
243 /* Use with ED_object_iter_other(). Sets the total number of levels
244  * for any multires modifiers on the object to the int pointed to by
245  * callback_data. */
246 int ED_object_multires_update_totlevels_cb(Object *ob, void *totlevel_v)
247 {
248         ModifierData *md;
249         int totlevel = *((int *)totlevel_v);
250
251         for (md = ob->modifiers.first; md; md = md->next) {
252                 if (md->type == eModifierType_Multires) {
253                         multires_set_tot_level(ob, (MultiresModifierData *)md, totlevel);
254                         DAG_id_tag_update(&ob->id, OB_RECALC_DATA);
255                 }
256         }
257         return FALSE;
258 }
259
260 /* Return TRUE if no modifier of type 'type' other than 'exclude' */
261 static int object_modifier_safe_to_delete(Main *bmain, Object *ob,
262                                           ModifierData *exclude,
263                                           ModifierType type)
264 {
265         return (!object_has_modifier(ob, exclude, type) &&
266                 !ED_object_iter_other(bmain, ob, FALSE,
267                                       object_has_modifier_cb, &type));
268 }
269
270 static int object_modifier_remove(Main *bmain, Object *ob, ModifierData *md,
271                                   int *sort_depsgraph)
272 {
273         /* It seems on rapid delete it is possible to
274          * get called twice on same modifier, so make
275          * sure it is in list. */
276         if (BLI_findindex(&ob->modifiers, md) == -1) {
277                 return 0;
278         }
279
280         /* special cases */
281         if (md->type == eModifierType_ParticleSystem) {
282                 ParticleSystemModifierData *psmd = (ParticleSystemModifierData *)md;
283
284                 BLI_remlink(&ob->particlesystem, psmd->psys);
285                 psys_free(ob, psmd->psys);
286                 psmd->psys = NULL;
287         }
288         else if (md->type == eModifierType_Softbody) {
289                 if (ob->soft) {
290                         sbFree(ob->soft);
291                         ob->soft = NULL;
292                         ob->softflag = 0;
293                 }
294         }
295         else if (md->type == eModifierType_Collision) {
296                 if (ob->pd)
297                         ob->pd->deflect = 0;
298
299                 *sort_depsgraph = 1;
300         }
301         else if (md->type == eModifierType_Surface) {
302                 *sort_depsgraph = 1;
303         }
304         else if (md->type == eModifierType_Multires) {
305                 /* Delete MDisps layer if not used by another multires modifier */
306                 if (object_modifier_safe_to_delete(bmain, ob, md, eModifierType_Multires))
307                         multires_customdata_delete(ob->data);
308         }
309         else if (md->type == eModifierType_Skin) {
310                 /* Delete MVertSkin layer if not used by another skin modifier */
311                 if (object_modifier_safe_to_delete(bmain, ob, md, eModifierType_Skin))
312                         modifier_skin_customdata_delete(ob);
313         }
314
315         if (ELEM(md->type, eModifierType_Softbody, eModifierType_Cloth) &&
316             ob->particlesystem.first == NULL)
317         {
318                 ob->mode &= ~OB_MODE_PARTICLE_EDIT;
319         }
320
321         BLI_remlink(&ob->modifiers, md);
322         modifier_free(md);
323
324         return 1;
325 }
326
327 int ED_object_modifier_remove(ReportList *reports, Main *bmain, Object *ob, ModifierData *md)
328 {
329         int sort_depsgraph = 0;
330         int ok;
331
332         ok = object_modifier_remove(bmain, ob, md, &sort_depsgraph);
333
334         if (!ok) {
335                 BKE_reportf(reports, RPT_ERROR, "Modifier '%s' not in object '%s'", md->name, ob->id.name);
336                 return 0;
337         }
338
339         DAG_id_tag_update(&ob->id, OB_RECALC_DATA);
340         DAG_relations_tag_update(bmain);
341
342         return 1;
343 }
344
345 void ED_object_modifier_clear(Main *bmain, Object *ob)
346 {
347         ModifierData *md = ob->modifiers.first;
348         int sort_depsgraph = 0;
349
350         if (!md)
351                 return;
352
353         while (md) {
354                 ModifierData *next_md;
355
356                 next_md = md->next;
357
358                 object_modifier_remove(bmain, ob, md, &sort_depsgraph);
359
360                 md = next_md;
361         }
362
363         DAG_id_tag_update(&ob->id, OB_RECALC_DATA);
364         DAG_relations_tag_update(bmain);
365 }
366
367 int ED_object_modifier_move_up(ReportList *reports, Object *ob, ModifierData *md)
368 {
369         if (md->prev) {
370                 ModifierTypeInfo *mti = modifierType_getInfo(md->type);
371
372                 if (mti->type != eModifierTypeType_OnlyDeform) {
373                         ModifierTypeInfo *nmti = modifierType_getInfo(md->prev->type);
374
375                         if (nmti->flags & eModifierTypeFlag_RequiresOriginalData) {
376                                 BKE_report(reports, RPT_WARNING, "Cannot move above a modifier requiring original data");
377                                 return 0;
378                         }
379                 }
380
381                 BLI_remlink(&ob->modifiers, md);
382                 BLI_insertlinkbefore(&ob->modifiers, md->prev, md);
383         }
384
385         return 1;
386 }
387
388 int ED_object_modifier_move_down(ReportList *reports, Object *ob, ModifierData *md)
389 {
390         if (md->next) {
391                 ModifierTypeInfo *mti = modifierType_getInfo(md->type);
392
393                 if (mti->flags & eModifierTypeFlag_RequiresOriginalData) {
394                         ModifierTypeInfo *nmti = modifierType_getInfo(md->next->type);
395
396                         if (nmti->type != eModifierTypeType_OnlyDeform) {
397                                 BKE_report(reports, RPT_WARNING, "Cannot move beyond a non-deforming modifier");
398                                 return 0;
399                         }
400                 }
401
402                 BLI_remlink(&ob->modifiers, md);
403                 BLI_insertlinkafter(&ob->modifiers, md->next, md);
404         }
405
406         return 1;
407 }
408
409 int ED_object_modifier_convert(ReportList *UNUSED(reports), Main *bmain, Scene *scene, Object *ob, ModifierData *md)
410 {
411         Object *obn;
412         ParticleSystem *psys;
413         ParticleCacheKey *key, **cache;
414         ParticleSettings *part;
415         Mesh *me;
416         MVert *mvert;
417         MEdge *medge;
418         int a, k, kmax;
419         int totvert = 0, totedge = 0, cvert = 0;
420         int totpart = 0, totchild = 0;
421
422         if (md->type != eModifierType_ParticleSystem) return 0;
423         if (ob && ob->mode & OB_MODE_PARTICLE_EDIT) return 0;
424
425         psys = ((ParticleSystemModifierData *)md)->psys;
426         part = psys->part;
427
428         if (part->ren_as != PART_DRAW_PATH || psys->pathcache == NULL)
429                 return 0;
430
431         totpart = psys->totcached;
432         totchild = psys->totchildcache;
433
434         if (totchild && (part->draw & PART_DRAW_PARENT) == 0)
435                 totpart = 0;
436
437         /* count */
438         cache = psys->pathcache;
439         for (a = 0; a < totpart; a++) {
440                 key = cache[a];
441
442                 if (key->steps > 0) {
443                         totvert += key->steps + 1;
444                         totedge += key->steps;
445                 }
446         }
447
448         cache = psys->childcache;
449         for (a = 0; a < totchild; a++) {
450                 key = cache[a];
451
452                 if (key->steps > 0) {
453                         totvert += key->steps + 1;
454                         totedge += key->steps;
455                 }
456         }
457
458         if (totvert == 0) return 0;
459
460         /* add new mesh */
461         obn = BKE_object_add(bmain, scene, OB_MESH);
462         me = obn->data;
463         
464         me->totvert = totvert;
465         me->totedge = totedge;
466         
467         me->mvert = CustomData_add_layer(&me->vdata, CD_MVERT, CD_CALLOC, NULL, totvert);
468         me->medge = CustomData_add_layer(&me->edata, CD_MEDGE, CD_CALLOC, NULL, totedge);
469         me->mface = CustomData_add_layer(&me->fdata, CD_MFACE, CD_CALLOC, NULL, 0);
470         
471         mvert = me->mvert;
472         medge = me->medge;
473
474         /* copy coordinates */
475         cache = psys->pathcache;
476         for (a = 0; a < totpart; a++) {
477                 key = cache[a];
478                 kmax = key->steps;
479                 for (k = 0; k <= kmax; k++, key++, cvert++, mvert++) {
480                         copy_v3_v3(mvert->co, key->co);
481                         if (k) {
482                                 medge->v1 = cvert - 1;
483                                 medge->v2 = cvert;
484                                 medge->flag = ME_EDGEDRAW | ME_EDGERENDER | ME_LOOSEEDGE;
485                                 medge++;
486                         }
487                         else {
488                                 /* cheap trick to select the roots */
489                                 mvert->flag |= SELECT;
490                         }
491                 }
492         }
493
494         cache = psys->childcache;
495         for (a = 0; a < totchild; a++) {
496                 key = cache[a];
497                 kmax = key->steps;
498                 for (k = 0; k <= kmax; k++, key++, cvert++, mvert++) {
499                         copy_v3_v3(mvert->co, key->co);
500                         if (k) {
501                                 medge->v1 = cvert - 1;
502                                 medge->v2 = cvert;
503                                 medge->flag = ME_EDGEDRAW | ME_EDGERENDER | ME_LOOSEEDGE;
504                                 medge++;
505                         }
506                         else {
507                                 /* cheap trick to select the roots */
508                                 mvert->flag |= SELECT;
509                         }
510                 }
511         }
512
513         DAG_relations_tag_update(bmain);
514
515         return 1;
516 }
517
518 static int modifier_apply_shape(ReportList *reports, Scene *scene, Object *ob, ModifierData *md)
519 {
520         ModifierTypeInfo *mti = modifierType_getInfo(md->type);
521
522         md->scene = scene;
523
524         if (mti->isDisabled && mti->isDisabled(md, 0)) {
525                 BKE_report(reports, RPT_ERROR, "Modifier is disabled, skipping apply");
526                 return 0;
527         }
528
529         /*
530          * It should be ridiculously easy to extract the original verts that we want
531          * and form the shape data.  We can probably use the CD KEYINDEX layer (or
532          * whatever I ended up calling it, too tired to check now), though this would
533          * by necessity have to make some potentially ugly assumptions about the order
534          * of the mesh data :-/  you can probably assume in 99% of cases that the first
535          * element of a given index is the original, and any subsequent duplicates are
536          * copies/interpolates, but that's an assumption that would need to be tested
537          * and then predominantly stated in comments in a half dozen headers.
538          */
539
540         if (ob->type == OB_MESH) {
541                 DerivedMesh *dm;
542                 Mesh *me = ob->data;
543                 Key *key = me->key;
544                 KeyBlock *kb;
545                 
546                 if (!modifier_isSameTopology(md) || mti->type == eModifierTypeType_NonGeometrical) {
547                         BKE_report(reports, RPT_ERROR, "Only deforming modifiers can be applied to shapes");
548                         return 0;
549                 }
550                 
551                 dm = mesh_create_derived_for_modifier(scene, ob, md, 0);
552                 if (!dm) {
553                         BKE_report(reports, RPT_ERROR, "Modifier is disabled or returned error, skipping apply");
554                         return 0;
555                 }
556                 
557                 if (key == NULL) {
558                         key = me->key = BKE_key_add((ID *)me);
559                         key->type = KEY_RELATIVE;
560                         /* if that was the first key block added, then it was the basis.
561                          * Initialize it with the mesh, and add another for the modifier */
562                         kb = BKE_keyblock_add(key, NULL);
563                         BKE_key_convert_from_mesh(me, kb);
564                 }
565
566                 kb = BKE_keyblock_add(key, md->name);
567                 DM_to_meshkey(dm, me, kb);
568                 
569                 dm->release(dm);
570         }
571         else {
572                 BKE_report(reports, RPT_ERROR, "Cannot apply modifier for this object type");
573                 return 0;
574         }
575         return 1;
576 }
577
578 static int modifier_apply_obdata(ReportList *reports, Scene *scene, Object *ob, ModifierData *md)
579 {
580         ModifierTypeInfo *mti = modifierType_getInfo(md->type);
581
582         md->scene = scene;
583
584         if (mti->isDisabled && mti->isDisabled(md, 0)) {
585                 BKE_report(reports, RPT_ERROR, "Modifier is disabled, skipping apply");
586                 return 0;
587         }
588
589         if (ob->type == OB_MESH) {
590                 DerivedMesh *dm;
591                 Mesh *me = ob->data;
592                 MultiresModifierData *mmd = find_multires_modifier_before(scene, md);
593
594                 if (me->key && mti->type != eModifierTypeType_NonGeometrical) {
595                         BKE_report(reports, RPT_ERROR, "Modifier cannot be applied to a mesh with shape keys");
596                         return 0;
597                 }
598
599                 /* Multires: ensure that recent sculpting is applied */
600                 if (md->type == eModifierType_Multires)
601                         multires_force_update(ob);
602
603                 if (mmd && mmd->totlvl && mti->type == eModifierTypeType_OnlyDeform) {
604                         if (!multiresModifier_reshapeFromDeformMod(scene, mmd, ob, md)) {
605                                 BKE_report(reports, RPT_ERROR, "Multires modifier returned error, skipping apply");
606                                 return 0;
607                         }
608                 }
609                 else {
610                         dm = mesh_create_derived_for_modifier(scene, ob, md, 1);
611                         if (!dm) {
612                                 BKE_report(reports, RPT_ERROR, "Modifier returned error, skipping apply");
613                                 return 0;
614                         }
615
616                         DM_to_mesh(dm, me, ob, CD_MASK_MESH);
617
618                         dm->release(dm);
619
620                         if (md->type == eModifierType_Multires)
621                                 multires_customdata_delete(me);
622                 }
623         }
624         else if (ELEM(ob->type, OB_CURVE, OB_SURF)) {
625                 Curve *cu;
626                 int numVerts;
627                 float (*vertexCos)[3];
628
629                 if (ELEM(mti->type, eModifierTypeType_Constructive, eModifierTypeType_Nonconstructive)) {
630                         BKE_report(reports, RPT_ERROR, "Cannot apply constructive modifiers on curve");
631                         return 0;
632                 }
633
634                 cu = ob->data;
635                 BKE_report(reports, RPT_INFO, "Applied modifier only changed CV points, not tessellated/bevel vertices");
636
637                 vertexCos = BKE_curve_nurbs_vertexCos_get(&cu->nurb, &numVerts);
638                 mti->deformVerts(md, ob, NULL, vertexCos, numVerts, 0);
639                 BK_curve_nurbs_vertexCos_apply(&cu->nurb, vertexCos);
640
641                 MEM_freeN(vertexCos);
642
643                 DAG_id_tag_update(&ob->id, OB_RECALC_DATA);
644         }
645         else {
646                 BKE_report(reports, RPT_ERROR, "Cannot apply modifier for this object type");
647                 return 0;
648         }
649
650         /* lattice modifier can be applied to particle system too */
651         if (ob->particlesystem.first) {
652
653                 ParticleSystem *psys = ob->particlesystem.first;
654
655                 for (; psys; psys = psys->next) {
656                         
657                         if (psys->part->type != PART_HAIR)
658                                 continue;
659
660                         psys_apply_hair_lattice(scene, ob, psys);
661                 }
662         }
663
664         return 1;
665 }
666
667 int ED_object_modifier_apply(ReportList *reports, Scene *scene, Object *ob, ModifierData *md, int mode)
668 {
669         int prev_mode;
670
671         if (scene->obedit) {
672                 BKE_report(reports, RPT_ERROR, "Modifiers cannot be applied in edit mode");
673                 return 0;
674         }
675         else if (((ID *) ob->data)->us > 1) {
676                 BKE_report(reports, RPT_ERROR, "Modifiers cannot be applied to multi-user data");
677                 return 0;
678         }
679         else if ((ob->mode & OB_MODE_SCULPT) &&
680                  (find_multires_modifier_before(scene, md)) &&
681                  (modifier_isSameTopology(md) == FALSE))
682         {
683                 BKE_report(reports, RPT_ERROR, "Constructive modifier cannot be applied to multi-res data in sculpt mode");
684                 return 0;
685         }
686
687         if (md != ob->modifiers.first)
688                 BKE_report(reports, RPT_INFO, "Applied modifier was not first, result may not be as expected");
689
690         /* allow apply of a not-realtime modifier, by first re-enabling realtime. */
691         prev_mode = md->mode;
692         md->mode |= eModifierMode_Realtime;
693
694         if (mode == MODIFIER_APPLY_SHAPE) {
695                 if (!modifier_apply_shape(reports, scene, ob, md)) {
696                         md->mode = prev_mode;
697                         return 0;
698                 }
699         }
700         else {
701                 if (!modifier_apply_obdata(reports, scene, ob, md)) {
702                         md->mode = prev_mode;
703                         return 0;
704                 }
705         }
706
707         BLI_remlink(&ob->modifiers, md);
708         modifier_free(md);
709
710         return 1;
711 }
712
713 int ED_object_modifier_copy(ReportList *UNUSED(reports), Object *ob, ModifierData *md)
714 {
715         ModifierData *nmd;
716         
717         nmd = modifier_new(md->type);
718         modifier_copyData(md, nmd);
719         BLI_insertlinkafter(&ob->modifiers, md, nmd);
720         modifier_unique_name(&ob->modifiers, nmd);
721
722         return 1;
723 }
724
725 /************************ add modifier operator *********************/
726
727 static int modifier_add_exec(bContext *C, wmOperator *op)
728 {
729         Main *bmain = CTX_data_main(C);
730         Scene *scene = CTX_data_scene(C);
731         Object *ob = ED_object_active_context(C);
732         int type = RNA_enum_get(op->ptr, "type");
733
734         if (!ED_object_modifier_add(op->reports, bmain, scene, ob, NULL, type))
735                 return OPERATOR_CANCELLED;
736
737         WM_event_add_notifier(C, NC_OBJECT | ND_MODIFIER, ob);
738         
739         return OPERATOR_FINISHED;
740 }
741
742 static EnumPropertyItem *modifier_add_itemf(bContext *C, PointerRNA *UNUSED(ptr), PropertyRNA *UNUSED(prop), int *free)
743 {       
744         Object *ob = ED_object_active_context(C);
745         EnumPropertyItem *item = NULL, *md_item, *group_item = NULL;
746         ModifierTypeInfo *mti;
747         int totitem = 0, a;
748         
749         if (!ob)
750                 return modifier_type_items;
751
752         for (a = 0; modifier_type_items[a].identifier; a++) {
753                 md_item = &modifier_type_items[a];
754
755                 if (md_item->identifier[0]) {
756                         mti = modifierType_getInfo(md_item->value);
757
758                         if (mti->flags & eModifierTypeFlag_NoUserAdd)
759                                 continue;
760
761                         if (!BKE_object_support_modifier_type_check(ob, md_item->value))
762                                 continue;
763                 }
764                 else {
765                         group_item = md_item;
766                         md_item = NULL;
767
768                         continue;
769                 }
770
771                 if (group_item) {
772                         RNA_enum_item_add(&item, &totitem, group_item);
773                         group_item = NULL;
774                 }
775
776                 RNA_enum_item_add(&item, &totitem, md_item);
777         }
778
779         RNA_enum_item_end(&item, &totitem);
780         *free = 1;
781
782         return item;
783 }
784
785 void OBJECT_OT_modifier_add(wmOperatorType *ot)
786 {
787         PropertyRNA *prop;
788
789         /* identifiers */
790         ot->name = "Add Modifier";
791         ot->description = "Add a modifier to the active object";
792         ot->idname = "OBJECT_OT_modifier_add";
793         
794         /* api callbacks */
795         ot->invoke = WM_menu_invoke;
796         ot->exec = modifier_add_exec;
797         ot->poll = ED_operator_object_active_editable;
798         
799         /* flags */
800         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
801         
802         /* properties */
803         prop = RNA_def_enum(ot->srna, "type", modifier_type_items, eModifierType_Subsurf, "Type", "");
804         RNA_def_enum_funcs(prop, modifier_add_itemf);
805         ot->prop = prop;
806 }
807
808 /************************ generic functions for operators using mod names and data context *********************/
809
810 static int edit_modifier_poll_generic(bContext *C, StructRNA *rna_type, int obtype_flag)
811 {
812         PointerRNA ptr = CTX_data_pointer_get_type(C, "modifier", rna_type);
813         Object *ob = (ptr.id.data) ? ptr.id.data : ED_object_active_context(C);
814         
815         if (!ob || ob->id.lib) return 0;
816         if (obtype_flag && ((1 << ob->type) & obtype_flag) == 0) return 0;
817         if (ptr.id.data && ((ID *)ptr.id.data)->lib) return 0;
818         
819         return 1;
820 }
821
822 static int edit_modifier_poll(bContext *C)
823 {
824         return edit_modifier_poll_generic(C, &RNA_Modifier, 0);
825 }
826
827 static void edit_modifier_properties(wmOperatorType *ot)
828 {
829         RNA_def_string(ot->srna, "modifier", "", MAX_NAME, "Modifier", "Name of the modifier to edit");
830 }
831
832 static int edit_modifier_invoke_properties(bContext *C, wmOperator *op)
833 {
834         ModifierData *md;
835         
836         if (RNA_struct_property_is_set(op->ptr, "modifier")) {
837                 return TRUE;
838         }
839         else {
840                 PointerRNA ptr = CTX_data_pointer_get_type(C, "modifier", &RNA_Modifier);
841                 if (ptr.data) {
842                         md = ptr.data;
843                         RNA_string_set(op->ptr, "modifier", md->name);
844                         return TRUE;
845                 }
846         }
847
848         return FALSE;
849 }
850
851 static ModifierData *edit_modifier_property_get(wmOperator *op, Object *ob, int type)
852 {
853         char modifier_name[MAX_NAME];
854         ModifierData *md;
855         RNA_string_get(op->ptr, "modifier", modifier_name);
856         
857         md = modifiers_findByName(ob, modifier_name);
858         
859         if (md && type != 0 && md->type != type)
860                 md = NULL;
861
862         return md;
863 }
864
865 /************************ remove modifier operator *********************/
866
867 static int modifier_remove_exec(bContext *C, wmOperator *op)
868 {
869         Main *bmain = CTX_data_main(C);
870         Scene *scene = CTX_data_scene(C);
871         Object *ob = ED_object_active_context(C);
872         ModifierData *md = edit_modifier_property_get(op, ob, 0);
873         int mode_orig = ob->mode;
874         
875         if (!md || !ED_object_modifier_remove(op->reports, bmain, ob, md))
876                 return OPERATOR_CANCELLED;
877
878         WM_event_add_notifier(C, NC_OBJECT | ND_MODIFIER, ob);
879
880         /* if cloth/softbody was removed, particle mode could be cleared */
881         if (mode_orig & OB_MODE_PARTICLE_EDIT)
882                 if ((ob->mode & OB_MODE_PARTICLE_EDIT) == 0)
883                         if (scene->basact && scene->basact->object == ob)
884                                 WM_event_add_notifier(C, NC_SCENE | ND_MODE | NS_MODE_OBJECT, NULL);
885         
886         return OPERATOR_FINISHED;
887 }
888
889 static int modifier_remove_invoke(bContext *C, wmOperator *op, const wmEvent *UNUSED(event))
890 {
891         if (edit_modifier_invoke_properties(C, op))
892                 return modifier_remove_exec(C, op);
893         else
894                 return OPERATOR_CANCELLED;
895 }
896
897 void OBJECT_OT_modifier_remove(wmOperatorType *ot)
898 {
899         ot->name = "Remove Modifier";
900         ot->description = "Remove a modifier from the active object";
901         ot->idname = "OBJECT_OT_modifier_remove";
902
903         ot->invoke = modifier_remove_invoke;
904         ot->exec = modifier_remove_exec;
905         ot->poll = edit_modifier_poll;
906         
907         /* flags */
908         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO | OPTYPE_INTERNAL;
909         edit_modifier_properties(ot);
910 }
911
912 /************************ move up modifier operator *********************/
913
914 static int modifier_move_up_exec(bContext *C, wmOperator *op)
915 {
916         Object *ob = ED_object_active_context(C);
917         ModifierData *md = edit_modifier_property_get(op, ob, 0);
918
919         if (!md || !ED_object_modifier_move_up(op->reports, ob, md))
920                 return OPERATOR_CANCELLED;
921
922         DAG_id_tag_update(&ob->id, OB_RECALC_DATA);
923         WM_event_add_notifier(C, NC_OBJECT | ND_MODIFIER, ob);
924         
925         return OPERATOR_FINISHED;
926 }
927
928 static int modifier_move_up_invoke(bContext *C, wmOperator *op, const wmEvent *UNUSED(event))
929 {
930         if (edit_modifier_invoke_properties(C, op))
931                 return modifier_move_up_exec(C, op);
932         else
933                 return OPERATOR_CANCELLED;
934 }
935
936 void OBJECT_OT_modifier_move_up(wmOperatorType *ot)
937 {
938         ot->name = "Move Up Modifier";
939         ot->description = "Move modifier up in the stack";
940         ot->idname = "OBJECT_OT_modifier_move_up";
941
942         ot->invoke = modifier_move_up_invoke;
943         ot->exec = modifier_move_up_exec;
944         ot->poll = edit_modifier_poll;
945         
946         /* flags */
947         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO | OPTYPE_INTERNAL;
948         edit_modifier_properties(ot);
949 }
950
951 /************************ move down modifier operator *********************/
952
953 static int modifier_move_down_exec(bContext *C, wmOperator *op)
954 {
955         Object *ob = ED_object_active_context(C);
956         ModifierData *md = edit_modifier_property_get(op, ob, 0);
957
958         if (!md || !ED_object_modifier_move_down(op->reports, ob, md))
959                 return OPERATOR_CANCELLED;
960
961         DAG_id_tag_update(&ob->id, OB_RECALC_DATA);
962         WM_event_add_notifier(C, NC_OBJECT | ND_MODIFIER, ob);
963         
964         return OPERATOR_FINISHED;
965 }
966
967 static int modifier_move_down_invoke(bContext *C, wmOperator *op, const wmEvent *UNUSED(event))
968 {
969         if (edit_modifier_invoke_properties(C, op))
970                 return modifier_move_down_exec(C, op);
971         else
972                 return OPERATOR_CANCELLED;
973 }
974
975 void OBJECT_OT_modifier_move_down(wmOperatorType *ot)
976 {
977         ot->name = "Move Down Modifier";
978         ot->description = "Move modifier down in the stack";
979         ot->idname = "OBJECT_OT_modifier_move_down";
980
981         ot->invoke = modifier_move_down_invoke;
982         ot->exec = modifier_move_down_exec;
983         ot->poll = edit_modifier_poll;
984         
985         /* flags */
986         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO | OPTYPE_INTERNAL;
987         edit_modifier_properties(ot);
988 }
989
990 /************************ apply modifier operator *********************/
991
992 static int modifier_apply_exec(bContext *C, wmOperator *op)
993 {
994         Scene *scene = CTX_data_scene(C);
995         Object *ob = ED_object_active_context(C);
996         ModifierData *md = edit_modifier_property_get(op, ob, 0);
997         int apply_as = RNA_enum_get(op->ptr, "apply_as");
998
999         if (!md || !ED_object_modifier_apply(op->reports, scene, ob, md, apply_as)) {
1000                 return OPERATOR_CANCELLED;
1001         }
1002
1003         DAG_id_tag_update(&ob->id, OB_RECALC_DATA);
1004         WM_event_add_notifier(C, NC_OBJECT | ND_MODIFIER, ob);
1005         
1006         return OPERATOR_FINISHED;
1007 }
1008
1009 static int modifier_apply_invoke(bContext *C, wmOperator *op, const wmEvent *UNUSED(event))
1010 {
1011         if (edit_modifier_invoke_properties(C, op))
1012                 return modifier_apply_exec(C, op);
1013         else
1014                 return OPERATOR_CANCELLED;
1015 }
1016
1017 static EnumPropertyItem modifier_apply_as_items[] = {
1018         {MODIFIER_APPLY_DATA, "DATA", 0, "Object Data", "Apply modifier to the object's data"},
1019         {MODIFIER_APPLY_SHAPE, "SHAPE", 0, "New Shape", "Apply deform-only modifier to a new shape on this object"},
1020         {0, NULL, 0, NULL, NULL}
1021 };
1022
1023 void OBJECT_OT_modifier_apply(wmOperatorType *ot)
1024 {
1025         ot->name = "Apply Modifier";
1026         ot->description = "Apply modifier and remove from the stack";
1027         ot->idname = "OBJECT_OT_modifier_apply";
1028
1029         ot->invoke = modifier_apply_invoke;
1030         ot->exec = modifier_apply_exec;
1031         ot->poll = edit_modifier_poll;
1032         
1033         /* flags */
1034         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO | OPTYPE_INTERNAL;
1035         
1036         RNA_def_enum(ot->srna, "apply_as", modifier_apply_as_items, MODIFIER_APPLY_DATA, "Apply as", "How to apply the modifier to the geometry");
1037         edit_modifier_properties(ot);
1038 }
1039
1040 /************************ convert modifier operator *********************/
1041
1042 static int modifier_convert_exec(bContext *C, wmOperator *op)
1043 {
1044         Main *bmain = CTX_data_main(C);
1045         Scene *scene = CTX_data_scene(C);
1046         Object *ob = ED_object_active_context(C);
1047         ModifierData *md = edit_modifier_property_get(op, ob, 0);
1048         
1049         if (!md || !ED_object_modifier_convert(op->reports, bmain, scene, ob, md))
1050                 return OPERATOR_CANCELLED;
1051
1052         DAG_id_tag_update(&ob->id, OB_RECALC_DATA);
1053         WM_event_add_notifier(C, NC_OBJECT | ND_MODIFIER, ob);
1054         
1055         return OPERATOR_FINISHED;
1056 }
1057
1058 static int modifier_convert_invoke(bContext *C, wmOperator *op, const wmEvent *UNUSED(event))
1059 {
1060         if (edit_modifier_invoke_properties(C, op))
1061                 return modifier_convert_exec(C, op);
1062         else
1063                 return OPERATOR_CANCELLED;
1064 }
1065
1066 void OBJECT_OT_modifier_convert(wmOperatorType *ot)
1067 {
1068         ot->name = "Convert Modifier";
1069         ot->description = "Convert particles to a mesh object";
1070         ot->idname = "OBJECT_OT_modifier_convert";
1071
1072         ot->invoke = modifier_convert_invoke;
1073         ot->exec = modifier_convert_exec;
1074         ot->poll = edit_modifier_poll;
1075         
1076         /* flags */
1077         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO | OPTYPE_INTERNAL;
1078         edit_modifier_properties(ot);
1079 }
1080
1081 /************************ copy modifier operator *********************/
1082
1083 static int modifier_copy_exec(bContext *C, wmOperator *op)
1084 {
1085         Object *ob = ED_object_active_context(C);
1086         ModifierData *md = edit_modifier_property_get(op, ob, 0);
1087
1088         if (!md || !ED_object_modifier_copy(op->reports, ob, md))
1089                 return OPERATOR_CANCELLED;
1090
1091         DAG_id_tag_update(&ob->id, OB_RECALC_DATA);
1092         WM_event_add_notifier(C, NC_OBJECT | ND_MODIFIER, ob);
1093         
1094         return OPERATOR_FINISHED;
1095 }
1096
1097 static int modifier_copy_invoke(bContext *C, wmOperator *op, const wmEvent *UNUSED(event))
1098 {
1099         if (edit_modifier_invoke_properties(C, op))
1100                 return modifier_copy_exec(C, op);
1101         else
1102                 return OPERATOR_CANCELLED;
1103 }
1104
1105 void OBJECT_OT_modifier_copy(wmOperatorType *ot)
1106 {
1107         ot->name = "Copy Modifier";
1108         ot->description = "Duplicate modifier at the same position in the stack";
1109         ot->idname = "OBJECT_OT_modifier_copy";
1110
1111         ot->invoke = modifier_copy_invoke;
1112         ot->exec = modifier_copy_exec;
1113         ot->poll = edit_modifier_poll;
1114         
1115         /* flags */
1116         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO | OPTYPE_INTERNAL;
1117         edit_modifier_properties(ot);
1118 }
1119
1120 /************* multires delete higher levels operator ****************/
1121
1122 static int multires_poll(bContext *C)
1123 {
1124         return edit_modifier_poll_generic(C, &RNA_MultiresModifier, (1 << OB_MESH));
1125 }
1126
1127 static int multires_higher_levels_delete_exec(bContext *C, wmOperator *op)
1128 {
1129         Object *ob = ED_object_active_context(C);
1130         MultiresModifierData *mmd = (MultiresModifierData *)edit_modifier_property_get(op, ob, eModifierType_Multires);
1131         
1132         if (!mmd)
1133                 return OPERATOR_CANCELLED;
1134         
1135         multiresModifier_del_levels(mmd, ob, 1);
1136
1137         ED_object_iter_other(CTX_data_main(C), ob, TRUE,
1138                              ED_object_multires_update_totlevels_cb,
1139                              &mmd->totlvl);
1140         
1141         WM_event_add_notifier(C, NC_OBJECT | ND_MODIFIER, ob);
1142         
1143         return OPERATOR_FINISHED;
1144 }
1145
1146 static int multires_higher_levels_delete_invoke(bContext *C, wmOperator *op, const wmEvent *UNUSED(event))
1147 {
1148         if (edit_modifier_invoke_properties(C, op))
1149                 return multires_higher_levels_delete_exec(C, op);
1150         else
1151                 return OPERATOR_CANCELLED;
1152 }
1153
1154 void OBJECT_OT_multires_higher_levels_delete(wmOperatorType *ot)
1155 {
1156         ot->name = "Delete Higher Levels";
1157         ot->description = "Deletes the higher resolution mesh, potential loss of detail";
1158         ot->idname = "OBJECT_OT_multires_higher_levels_delete";
1159
1160         ot->poll = multires_poll;
1161         ot->invoke = multires_higher_levels_delete_invoke;
1162         ot->exec = multires_higher_levels_delete_exec;
1163         
1164         /* flags */
1165         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO | OPTYPE_INTERNAL;
1166         edit_modifier_properties(ot);
1167 }
1168
1169 /****************** multires subdivide operator *********************/
1170
1171 static int multires_subdivide_exec(bContext *C, wmOperator *op)
1172 {
1173         Object *ob = ED_object_active_context(C);
1174         MultiresModifierData *mmd = (MultiresModifierData *)edit_modifier_property_get(op, ob, eModifierType_Multires);
1175         
1176         if (!mmd)
1177                 return OPERATOR_CANCELLED;
1178         
1179         multiresModifier_subdivide(mmd, ob, 0, mmd->simple);
1180
1181         ED_object_iter_other(CTX_data_main(C), ob, TRUE,
1182                              ED_object_multires_update_totlevels_cb,
1183                              &mmd->totlvl);
1184
1185         DAG_id_tag_update(&ob->id, OB_RECALC_DATA);
1186         WM_event_add_notifier(C, NC_OBJECT | ND_MODIFIER, ob);
1187
1188         if (ob->mode & OB_MODE_SCULPT) {
1189                 /* ensure that grid paint mask layer is created */
1190                 ED_sculpt_mask_layers_ensure(ob, mmd);
1191         }
1192         
1193         return OPERATOR_FINISHED;
1194 }
1195
1196 static int multires_subdivide_invoke(bContext *C, wmOperator *op, const wmEvent *UNUSED(event))
1197 {
1198         if (edit_modifier_invoke_properties(C, op))
1199                 return multires_subdivide_exec(C, op);
1200         else
1201                 return OPERATOR_CANCELLED;
1202 }
1203
1204 void OBJECT_OT_multires_subdivide(wmOperatorType *ot)
1205 {
1206         ot->name = "Multires Subdivide";
1207         ot->description = "Add a new level of subdivision";
1208         ot->idname = "OBJECT_OT_multires_subdivide";
1209
1210         ot->poll = multires_poll;
1211         ot->invoke = multires_subdivide_invoke;
1212         ot->exec = multires_subdivide_exec;
1213         
1214         /* flags */
1215         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO | OPTYPE_INTERNAL;
1216         edit_modifier_properties(ot);
1217 }
1218
1219 /****************** multires reshape operator *********************/
1220
1221 static int multires_reshape_exec(bContext *C, wmOperator *op)
1222 {
1223         Object *ob = ED_object_active_context(C), *secondob = NULL;
1224         Scene *scene = CTX_data_scene(C);
1225         MultiresModifierData *mmd = (MultiresModifierData *)edit_modifier_property_get(op, ob, eModifierType_Multires);
1226
1227         if (!mmd)
1228                 return OPERATOR_CANCELLED;
1229
1230         if (mmd->lvl == 0) {
1231                 BKE_report(op->reports, RPT_ERROR, "Reshape can work only with higher levels of subdivisions");
1232                 return OPERATOR_CANCELLED;
1233         }
1234
1235         CTX_DATA_BEGIN(C, Object *, selob, selected_editable_objects)
1236         {
1237                 if (selob->type == OB_MESH && selob != ob) {
1238                         secondob = selob;
1239                         break;
1240                 }
1241         }
1242         CTX_DATA_END;
1243
1244         if (!secondob) {
1245                 BKE_report(op->reports, RPT_ERROR, "Second selected mesh object required to copy shape from");
1246                 return OPERATOR_CANCELLED;
1247         }
1248
1249         if (!multiresModifier_reshape(scene, mmd, ob, secondob)) {
1250                 BKE_report(op->reports, RPT_ERROR, "Objects do not have the same number of vertices");
1251                 return OPERATOR_CANCELLED;
1252         }
1253
1254         DAG_id_tag_update(&ob->id, OB_RECALC_DATA);
1255         WM_event_add_notifier(C, NC_OBJECT | ND_MODIFIER, ob);
1256
1257         return OPERATOR_FINISHED;
1258 }
1259
1260 static int multires_reshape_invoke(bContext *C, wmOperator *op, const wmEvent *UNUSED(event))
1261 {
1262         if (edit_modifier_invoke_properties(C, op))
1263                 return multires_reshape_exec(C, op);
1264         else
1265                 return OPERATOR_CANCELLED;
1266 }
1267
1268 void OBJECT_OT_multires_reshape(wmOperatorType *ot)
1269 {
1270         ot->name = "Multires Reshape";
1271         ot->description = "Copy vertex coordinates from other object";
1272         ot->idname = "OBJECT_OT_multires_reshape";
1273
1274         ot->poll = multires_poll;
1275         ot->invoke = multires_reshape_invoke;
1276         ot->exec = multires_reshape_exec;
1277         
1278         /* flags */
1279         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO | OPTYPE_INTERNAL;
1280         edit_modifier_properties(ot);
1281 }
1282
1283
1284                 
1285 /****************** multires save external operator *********************/
1286
1287 static int multires_external_save_exec(bContext *C, wmOperator *op)
1288 {
1289         Object *ob = ED_object_active_context(C);
1290         Mesh *me = (ob) ? ob->data : op->customdata;
1291         char path[FILE_MAX];
1292         int relative = RNA_boolean_get(op->ptr, "relative_path");
1293
1294         if (!me)
1295                 return OPERATOR_CANCELLED;
1296
1297         if (CustomData_external_test(&me->ldata, CD_MDISPS))
1298                 return OPERATOR_CANCELLED;
1299         
1300         RNA_string_get(op->ptr, "filepath", path);
1301
1302         if (relative)
1303                 BLI_path_rel(path, G.main->name);
1304
1305         CustomData_external_add(&me->ldata, &me->id, CD_MDISPS, me->totloop, path);
1306         CustomData_external_write(&me->ldata, &me->id, CD_MASK_MESH, me->totloop, 0);
1307         
1308         return OPERATOR_FINISHED;
1309 }
1310
1311 static int multires_external_save_invoke(bContext *C, wmOperator *op, const wmEvent *UNUSED(event))
1312 {
1313         Object *ob = ED_object_active_context(C);
1314         MultiresModifierData *mmd;
1315         Mesh *me = ob->data;
1316         char path[FILE_MAX];
1317
1318         if (!edit_modifier_invoke_properties(C, op))
1319                 return OPERATOR_CANCELLED;
1320         
1321         mmd = (MultiresModifierData *)edit_modifier_property_get(op, ob, eModifierType_Multires);
1322         
1323         if (!mmd)
1324                 return OPERATOR_CANCELLED;
1325         
1326         if (CustomData_external_test(&me->ldata, CD_MDISPS))
1327                 return OPERATOR_CANCELLED;
1328
1329         if (RNA_struct_property_is_set(op->ptr, "filepath"))
1330                 return multires_external_save_exec(C, op);
1331         
1332         op->customdata = me;
1333
1334         BLI_snprintf(path, sizeof(path), "//%s.btx", me->id.name + 2);
1335         RNA_string_set(op->ptr, "filepath", path);
1336         
1337         WM_event_add_fileselect(C, op);
1338
1339         return OPERATOR_RUNNING_MODAL;
1340 }
1341
1342 void OBJECT_OT_multires_external_save(wmOperatorType *ot)
1343 {
1344         ot->name = "Multires Save External";
1345         ot->description = "Save displacements to an external file";
1346         ot->idname = "OBJECT_OT_multires_external_save";
1347
1348         /* XXX modifier no longer in context after file browser .. ot->poll = multires_poll; */
1349         ot->exec = multires_external_save_exec;
1350         ot->invoke = multires_external_save_invoke;
1351         ot->poll = multires_poll;
1352         
1353         /* flags */
1354         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO | OPTYPE_INTERNAL;
1355
1356         WM_operator_properties_filesel(ot, FOLDERFILE | BTXFILE, FILE_SPECIAL, FILE_SAVE,
1357                                        WM_FILESEL_FILEPATH | WM_FILESEL_RELPATH, FILE_DEFAULTDISPLAY);
1358         edit_modifier_properties(ot);
1359 }
1360
1361 /****************** multires pack operator *********************/
1362
1363 static int multires_external_pack_exec(bContext *C, wmOperator *UNUSED(op))
1364 {
1365         Object *ob = ED_object_active_context(C);
1366         Mesh *me = ob->data;
1367
1368         if (!CustomData_external_test(&me->ldata, CD_MDISPS))
1369                 return OPERATOR_CANCELLED;
1370
1371         /* XXX don't remove.. */
1372         CustomData_external_remove(&me->ldata, &me->id, CD_MDISPS, me->totloop);
1373         
1374         return OPERATOR_FINISHED;
1375 }
1376
1377 void OBJECT_OT_multires_external_pack(wmOperatorType *ot)
1378 {
1379         ot->name = "Multires Pack External";
1380         ot->description = "Pack displacements from an external file";
1381         ot->idname = "OBJECT_OT_multires_external_pack";
1382
1383         ot->poll = multires_poll;
1384         ot->exec = multires_external_pack_exec;
1385         
1386         /* flags */
1387         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
1388 }
1389
1390 /********************* multires apply base ***********************/
1391 static int multires_base_apply_exec(bContext *C, wmOperator *op)
1392 {
1393         Object *ob = ED_object_active_context(C);
1394         MultiresModifierData *mmd = (MultiresModifierData *)edit_modifier_property_get(op, ob, eModifierType_Multires);
1395         
1396         if (!mmd)
1397                 return OPERATOR_CANCELLED;
1398         
1399         multiresModifier_base_apply(mmd, ob);
1400
1401         DAG_id_tag_update(&ob->id, OB_RECALC_DATA);
1402         WM_event_add_notifier(C, NC_OBJECT | ND_MODIFIER, ob);
1403         
1404         return OPERATOR_FINISHED;
1405 }
1406
1407 static int multires_base_apply_invoke(bContext *C, wmOperator *op, const wmEvent *UNUSED(event))
1408 {
1409         if (edit_modifier_invoke_properties(C, op))
1410                 return multires_base_apply_exec(C, op);
1411         else
1412                 return OPERATOR_CANCELLED;
1413 }
1414
1415
1416 void OBJECT_OT_multires_base_apply(wmOperatorType *ot)
1417 {
1418         ot->name = "Multires Apply Base";
1419         ot->description = "Modify the base mesh to conform to the displaced mesh";
1420         ot->idname = "OBJECT_OT_multires_base_apply";
1421
1422         ot->poll = multires_poll;
1423         ot->invoke = multires_base_apply_invoke;
1424         ot->exec = multires_base_apply_exec;
1425         
1426         /* flags */
1427         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO | OPTYPE_INTERNAL;
1428         edit_modifier_properties(ot);
1429 }
1430
1431
1432 /************************** skin modifier ***********************/
1433
1434 void modifier_skin_customdata_ensure(Object *ob)
1435 {
1436         Mesh *me = ob->data;
1437         BMesh *bm = me->edit_btmesh ? me->edit_btmesh->bm : NULL;
1438         MVertSkin *vs;
1439
1440         if (bm && !CustomData_has_layer(&bm->vdata, CD_MVERT_SKIN)) {
1441                 BMVert *v;
1442                 BMIter iter;
1443
1444                 BM_data_layer_add(bm, &bm->vdata, CD_MVERT_SKIN);
1445                 
1446                 /* Mark an arbitrary vertex as root */
1447                 BM_ITER_MESH (v, &iter, bm, BM_VERTS_OF_MESH) {
1448                         vs = CustomData_bmesh_get(&bm->vdata, v->head.data,
1449                                                   CD_MVERT_SKIN);
1450                         vs->flag |= MVERT_SKIN_ROOT;
1451                         break;
1452                 }
1453         }
1454         else if (!CustomData_has_layer(&me->vdata, CD_MVERT_SKIN)) {
1455                 vs = CustomData_add_layer(&me->vdata,
1456                                           CD_MVERT_SKIN,
1457                                           CD_DEFAULT,
1458                                           NULL,
1459                                           me->totvert);
1460
1461                 /* Mark an arbitrary vertex as root */
1462                 if (vs)
1463                         vs->flag |= MVERT_SKIN_ROOT;
1464         }
1465 }
1466
1467 static void modifier_skin_customdata_delete(Object *ob)
1468 {
1469         Mesh *me = ob->data;
1470         BMEditMesh *em = me->edit_btmesh;
1471         
1472         if (em)
1473                 BM_data_layer_free(em->bm, &em->bm->vdata, CD_MVERT_SKIN);
1474         else
1475                 CustomData_free_layer_active(&me->vdata, CD_MVERT_SKIN, me->totvert);
1476 }
1477
1478 static int skin_poll(bContext *C)
1479 {
1480         return (!CTX_data_edit_object(C) &&
1481                 edit_modifier_poll_generic(C, &RNA_SkinModifier, (1 << OB_MESH)));
1482 }
1483
1484 static int skin_edit_poll(bContext *C)
1485 {
1486         return (CTX_data_edit_object(C) &&
1487                 edit_modifier_poll_generic(C, &RNA_SkinModifier, (1 << OB_MESH)));
1488 }
1489
1490 static void skin_root_clear(BMesh *bm, BMVert *bm_vert, GHash *visited)
1491 {
1492         BMEdge *bm_edge;
1493         BMIter bm_iter;
1494         
1495         BM_ITER_ELEM (bm_edge, &bm_iter, bm_vert, BM_EDGES_OF_VERT) {
1496                 BMVert *v2 = BM_edge_other_vert(bm_edge, bm_vert);
1497
1498                 if (!BLI_ghash_lookup(visited, v2)) {
1499                         MVertSkin *vs = CustomData_bmesh_get(&bm->vdata,
1500                                                              v2->head.data,
1501                                                              CD_MVERT_SKIN);
1502
1503                         /* clear vertex root flag and add to visited set */
1504                         vs->flag &= ~MVERT_SKIN_ROOT;
1505                         BLI_ghash_insert(visited, v2, v2);
1506
1507                         skin_root_clear(bm, v2, visited);
1508                 }
1509         }
1510 }
1511
1512 static int skin_root_mark_exec(bContext *C, wmOperator *UNUSED(op))
1513 {
1514         Object *ob = CTX_data_edit_object(C);
1515         BMEditMesh *em = BKE_editmesh_from_object(ob);
1516         BMesh *bm = em->bm;
1517         BMVert *bm_vert;
1518         BMIter bm_iter;
1519         GHash *visited;
1520
1521         visited = BLI_ghash_ptr_new("skin_root_mark_exec visited");
1522
1523         modifier_skin_customdata_ensure(ob);
1524
1525         BM_ITER_MESH (bm_vert, &bm_iter, bm, BM_VERTS_OF_MESH) {
1526                 if (!BLI_ghash_lookup(visited, bm_vert) &&
1527                     BM_elem_flag_test(bm_vert, BM_ELEM_SELECT))
1528                 {
1529                         MVertSkin *vs = CustomData_bmesh_get(&bm->vdata,
1530                                                              bm_vert->head.data,
1531                                                              CD_MVERT_SKIN);
1532
1533                         /* mark vertex as root and add to visited set */
1534                         vs->flag |= MVERT_SKIN_ROOT;
1535                         BLI_ghash_insert(visited, bm_vert, bm_vert);
1536
1537                         /* clear root flag from all connected vertices (recursively) */
1538                         skin_root_clear(bm, bm_vert, visited);
1539                 }
1540         }
1541
1542         BLI_ghash_free(visited, NULL, NULL);
1543
1544         DAG_id_tag_update(&ob->id, OB_RECALC_DATA);
1545         WM_event_add_notifier(C, NC_OBJECT | ND_MODIFIER, ob);
1546         
1547         return OPERATOR_FINISHED;
1548 }
1549
1550 void OBJECT_OT_skin_root_mark(wmOperatorType *ot)
1551 {
1552         ot->name = "Skin Root Mark";
1553         ot->description = "Mark selected vertices as roots";
1554         ot->idname = "OBJECT_OT_skin_root_mark";
1555
1556         ot->poll = skin_edit_poll;
1557         ot->exec = skin_root_mark_exec;
1558         
1559         /* flags */
1560         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
1561 }
1562
1563 typedef enum {
1564         SKIN_LOOSE_MARK,
1565         SKIN_LOOSE_CLEAR
1566 } SkinLooseAction;
1567
1568 static int skin_loose_mark_clear_exec(bContext *C, wmOperator *op)
1569 {
1570         Object *ob = CTX_data_edit_object(C);
1571         BMEditMesh *em = BKE_editmesh_from_object(ob);
1572         BMesh *bm = em->bm;
1573         BMVert *bm_vert;
1574         BMIter bm_iter;
1575         SkinLooseAction action = RNA_enum_get(op->ptr, "action");
1576
1577         if (!CustomData_has_layer(&bm->vdata, CD_MVERT_SKIN)) {
1578                 return OPERATOR_CANCELLED;
1579         }
1580
1581         BM_ITER_MESH (bm_vert, &bm_iter, bm, BM_VERTS_OF_MESH) {
1582                 if (BM_elem_flag_test(bm_vert, BM_ELEM_SELECT)) {
1583                         MVertSkin *vs = CustomData_bmesh_get(&bm->vdata,
1584                                                              bm_vert->head.data,
1585                                                              CD_MVERT_SKIN);
1586
1587
1588                         switch (action) {
1589                                 case SKIN_LOOSE_MARK:
1590                                         vs->flag |= MVERT_SKIN_LOOSE;
1591                                         break;
1592                                 case SKIN_LOOSE_CLEAR:
1593                                         vs->flag &= ~MVERT_SKIN_LOOSE;
1594                                         break;
1595                         }
1596                 }
1597         }
1598
1599         DAG_id_tag_update(&ob->id, OB_RECALC_DATA);
1600         WM_event_add_notifier(C, NC_OBJECT | ND_MODIFIER, ob);
1601         
1602         return OPERATOR_FINISHED;
1603 }
1604
1605 void OBJECT_OT_skin_loose_mark_clear(wmOperatorType *ot)
1606 {
1607         static EnumPropertyItem action_items[] = {
1608                 {SKIN_LOOSE_MARK, "MARK", 0, "Mark", "Mark selected vertices as loose"},
1609                 {SKIN_LOOSE_CLEAR, "CLEAR", 0, "Clear", "Set selected vertices as not loose"},
1610                 {0, NULL, 0, NULL, NULL}
1611         };
1612
1613         ot->name = "Skin Mark/Clear Loose";
1614         ot->description = "Mark/clear selected vertices as loose";
1615         ot->idname = "OBJECT_OT_skin_loose_mark_clear";
1616
1617         ot->poll = skin_edit_poll;
1618         ot->exec = skin_loose_mark_clear_exec;
1619         
1620         /* flags */
1621         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
1622
1623         RNA_def_enum(ot->srna, "action", action_items, SKIN_LOOSE_MARK, "Action", NULL);
1624 }
1625
1626 static int skin_radii_equalize_exec(bContext *C, wmOperator *UNUSED(op))
1627 {
1628         Object *ob = CTX_data_edit_object(C);
1629         BMEditMesh *em = BKE_editmesh_from_object(ob);
1630         BMesh *bm = em->bm;
1631         BMVert *bm_vert;
1632         BMIter bm_iter;
1633
1634         if (!CustomData_has_layer(&bm->vdata, CD_MVERT_SKIN)) {
1635                 return OPERATOR_CANCELLED;
1636         }
1637
1638         BM_ITER_MESH (bm_vert, &bm_iter, bm, BM_VERTS_OF_MESH) {
1639                 if (BM_elem_flag_test(bm_vert, BM_ELEM_SELECT)) {
1640                         MVertSkin *vs = CustomData_bmesh_get(&bm->vdata,
1641                                                              bm_vert->head.data,
1642                                                              CD_MVERT_SKIN);
1643                         float avg = (vs->radius[0] + vs->radius[1]) * 0.5f;
1644
1645                         vs->radius[0] = vs->radius[1] = avg;
1646                 }
1647         }
1648
1649         DAG_id_tag_update(&ob->id, OB_RECALC_DATA);
1650         WM_event_add_notifier(C, NC_OBJECT | ND_MODIFIER, ob);
1651         
1652         return OPERATOR_FINISHED;
1653 }
1654
1655 void OBJECT_OT_skin_radii_equalize(wmOperatorType *ot)
1656 {
1657         ot->name = "Skin Radii Equalize";
1658         ot->description = "Make skin radii of selected vertices equal on each axis";
1659         ot->idname = "OBJECT_OT_skin_radii_equalize";
1660
1661         ot->poll = skin_edit_poll;
1662         ot->exec = skin_radii_equalize_exec;
1663         
1664         /* flags */
1665         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
1666 }
1667
1668 static void skin_armature_bone_create(Object *skin_ob,
1669                                       MVert *mvert, MEdge *medge,
1670                                       bArmature *arm,
1671                                       BLI_bitmap *edges_visited,
1672                                       const MeshElemMap *emap,
1673                                       EditBone *parent_bone,
1674                                       int parent_v)
1675 {
1676         int i;
1677
1678         for (i = 0; i < emap[parent_v].count; i++) {
1679                 int endx = emap[parent_v].indices[i];
1680                 const MEdge *e = &medge[endx];
1681                 EditBone *bone;
1682                 bDeformGroup *dg;
1683                 int v;
1684
1685                 /* ignore edge if already visited */
1686                 if (BLI_BITMAP_GET(edges_visited, endx))
1687                         continue;
1688                 BLI_BITMAP_SET(edges_visited, endx);
1689
1690                 v = (e->v1 == parent_v ? e->v2 : e->v1);
1691
1692                 bone = ED_armature_edit_bone_add(arm, "Bone");
1693
1694                 bone->parent = parent_bone;
1695                 bone->flag |= BONE_CONNECTED;
1696
1697                 copy_v3_v3(bone->head, mvert[parent_v].co);
1698                 copy_v3_v3(bone->tail, mvert[v].co);
1699                 bone->rad_head = bone->rad_tail = 0.25;
1700                 BLI_snprintf(bone->name, sizeof(bone->name), "Bone.%.2d", endx);
1701
1702                 /* add bDeformGroup */
1703                 if ((dg = ED_vgroup_add_name(skin_ob, bone->name))) {
1704                         ED_vgroup_vert_add(skin_ob, dg, parent_v, 1, WEIGHT_REPLACE);
1705                         ED_vgroup_vert_add(skin_ob, dg, v, 1, WEIGHT_REPLACE);
1706                 }
1707                 
1708                 skin_armature_bone_create(skin_ob,
1709                                           mvert, medge,
1710                                           arm,
1711                                           edges_visited,
1712                                           emap,
1713                                           bone,
1714                                           v);
1715         }
1716 }
1717
1718 static Object *modifier_skin_armature_create(Main *bmain, Scene *scene, Object *skin_ob)
1719 {
1720         BLI_bitmap *edges_visited;
1721         DerivedMesh *deform_dm;
1722         MVert *mvert;
1723         Mesh *me = skin_ob->data;
1724         Object *arm_ob;
1725         bArmature *arm;
1726         MVertSkin *mvert_skin;
1727         MeshElemMap *emap;
1728         int *emap_mem;
1729         int v;
1730
1731         deform_dm = mesh_get_derived_deform(scene, skin_ob, CD_MASK_BAREMESH);
1732         mvert = deform_dm->getVertArray(deform_dm);
1733
1734         /* add vertex weights to original mesh */
1735         CustomData_add_layer(&me->vdata,
1736                              CD_MDEFORMVERT,
1737                              CD_CALLOC,
1738                              NULL,
1739                              me->totvert);
1740         
1741         arm_ob = BKE_object_add(bmain, scene, OB_ARMATURE);
1742         BKE_object_transform_copy(arm_ob, skin_ob);
1743         arm = arm_ob->data;
1744         arm->layer = 1;
1745         arm_ob->dtx |= OB_DRAWXRAY;
1746         arm->drawtype = ARM_LINE;
1747         arm->edbo = MEM_callocN(sizeof(ListBase), "edbo armature");
1748
1749         mvert_skin = CustomData_get_layer(&me->vdata, CD_MVERT_SKIN);
1750         BKE_mesh_vert_edge_map_create(&emap, &emap_mem,
1751                              me->medge, me->totvert, me->totedge);
1752
1753         edges_visited = BLI_BITMAP_NEW(me->totedge, "edge_visited");
1754
1755         /* note: we use EditBones here, easier to set them up and use
1756          * edit-armature functions to convert back to regular bones */
1757         for (v = 0; v < me->totvert; v++) {
1758                 if (mvert_skin[v].flag & MVERT_SKIN_ROOT) {
1759                         EditBone *bone = NULL;
1760
1761                         /* Unless the skin root has just one adjacent edge, create
1762                          * a fake root bone (have it going off in the Y direction
1763                          * (arbitrary) */
1764                         if (emap[v].count > 1) {
1765                                 bone = ED_armature_edit_bone_add(arm, "Bone");
1766
1767                                 copy_v3_v3(bone->head, me->mvert[v].co);
1768                                 copy_v3_v3(bone->tail, me->mvert[v].co);
1769
1770                                 bone->head[1] = 1.0f;
1771                                 bone->rad_head = bone->rad_tail = 0.25;
1772                         }
1773                         
1774                         if (emap[v].count >= 1) {
1775                                 skin_armature_bone_create(skin_ob,
1776                                                           mvert, me->medge,
1777                                                           arm,
1778                                                           edges_visited,
1779                                                           emap,
1780                                                           bone,
1781                                                           v);
1782                         }
1783                 }
1784         }
1785
1786         MEM_freeN(edges_visited);
1787         MEM_freeN(emap);
1788         MEM_freeN(emap_mem);
1789
1790         ED_armature_from_edit(arm);
1791         ED_armature_edit_free(arm);
1792
1793         return arm_ob;
1794 }
1795
1796 static int skin_armature_create_exec(bContext *C, wmOperator *op)
1797 {
1798         Main *bmain = CTX_data_main(C);
1799         Scene *scene = CTX_data_scene(C);
1800         Object *ob = CTX_data_active_object(C), *arm_ob;
1801         Mesh *me = ob->data;
1802         ModifierData *skin_md;
1803         ArmatureModifierData *arm_md;
1804
1805         if (!CustomData_has_layer(&me->vdata, CD_MVERT_SKIN)) {
1806                 BKE_reportf(op->reports, RPT_WARNING, "Mesh '%s' has no skin vertex data", me->id.name + 2);
1807                 return OPERATOR_CANCELLED;
1808         }
1809
1810         /* create new armature */
1811         arm_ob = modifier_skin_armature_create(bmain, scene, ob);
1812
1813         /* add a modifier to connect the new armature to the mesh */
1814         arm_md = (ArmatureModifierData *)modifier_new(eModifierType_Armature);
1815         if (arm_md) {
1816                 skin_md = edit_modifier_property_get(op, ob, eModifierType_Skin);
1817                 BLI_insertlinkafter(&ob->modifiers, skin_md, arm_md);
1818
1819                 arm_md->object = arm_ob;
1820                 arm_md->deformflag = ARM_DEF_VGROUP | ARM_DEF_QUATERNION;
1821                 DAG_relations_tag_update(bmain);
1822                 DAG_id_tag_update(&ob->id, OB_RECALC_DATA);
1823         }
1824
1825         WM_event_add_notifier(C, NC_OBJECT | ND_MODIFIER, ob);
1826
1827         return OPERATOR_FINISHED;
1828 }
1829
1830 static int skin_armature_create_invoke(bContext *C, wmOperator *op, const wmEvent *UNUSED(event))
1831 {
1832         if (edit_modifier_invoke_properties(C, op))
1833                 return skin_armature_create_exec(C, op);
1834         else
1835                 return OPERATOR_CANCELLED;
1836 }
1837
1838 void OBJECT_OT_skin_armature_create(wmOperatorType *ot)
1839 {
1840         ot->name = "Skin Armature Create";
1841         ot->description = "Create an armature that parallels the skin layout";
1842         ot->idname = "OBJECT_OT_skin_armature_create";
1843
1844         ot->poll = skin_poll;
1845         ot->invoke = skin_armature_create_invoke;
1846         ot->exec = skin_armature_create_exec;
1847         
1848         /* flags */
1849         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO | OPTYPE_INTERNAL;
1850         edit_modifier_properties(ot);
1851 }
1852
1853 /************************ mdef bind operator *********************/
1854
1855 static int meshdeform_poll(bContext *C)
1856 {
1857         return edit_modifier_poll_generic(C, &RNA_MeshDeformModifier, 0);
1858 }
1859
1860 static int meshdeform_bind_exec(bContext *C, wmOperator *op)
1861 {
1862         Scene *scene = CTX_data_scene(C);
1863         Object *ob = ED_object_active_context(C);
1864         MeshDeformModifierData *mmd = (MeshDeformModifierData *)edit_modifier_property_get(op, ob, eModifierType_MeshDeform);
1865         
1866         if (!mmd)
1867                 return OPERATOR_CANCELLED;
1868
1869         if (mmd->bindcagecos) {
1870                 MEM_freeN(mmd->bindcagecos);
1871                 if (mmd->dyngrid) MEM_freeN(mmd->dyngrid);
1872                 if (mmd->dyninfluences) MEM_freeN(mmd->dyninfluences);
1873                 if (mmd->bindinfluences) MEM_freeN(mmd->bindinfluences);
1874                 if (mmd->bindoffsets) MEM_freeN(mmd->bindoffsets);
1875                 if (mmd->dynverts) MEM_freeN(mmd->dynverts);
1876                 if (mmd->bindweights) MEM_freeN(mmd->bindweights);  /* deprecated */
1877                 if (mmd->bindcos) MEM_freeN(mmd->bindcos);  /* deprecated */
1878
1879                 mmd->bindcagecos = NULL;
1880                 mmd->dyngrid = NULL;
1881                 mmd->dyninfluences = NULL;
1882                 mmd->bindinfluences = NULL;
1883                 mmd->bindoffsets = NULL;
1884                 mmd->dynverts = NULL;
1885                 mmd->bindweights = NULL; /* deprecated */
1886                 mmd->bindcos = NULL; /* deprecated */
1887                 mmd->totvert = 0;
1888                 mmd->totcagevert = 0;
1889                 mmd->totinfluence = 0;
1890                 
1891                 DAG_id_tag_update(&ob->id, OB_RECALC_DATA);
1892                 WM_event_add_notifier(C, NC_OBJECT | ND_MODIFIER, ob);
1893         }
1894         else {
1895                 DerivedMesh *dm;
1896                 int mode = mmd->modifier.mode;
1897
1898                 /* force modifier to run, it will call binding routine */
1899                 mmd->bindfunc = mesh_deform_bind;
1900                 mmd->modifier.mode |= eModifierMode_Realtime;
1901
1902                 if (ob->type == OB_MESH) {
1903                         dm = mesh_create_derived_view(scene, ob, 0);
1904                         dm->release(dm);
1905                 }
1906                 else if (ob->type == OB_LATTICE) {
1907                         BKE_lattice_modifiers_calc(scene, ob);
1908                 }
1909                 else if (ob->type == OB_MBALL) {
1910                         BKE_displist_make_mball(CTX_data_main(C)->eval_ctx, scene, ob);
1911                 }
1912                 else if (ELEM3(ob->type, OB_CURVE, OB_SURF, OB_FONT)) {
1913                         BKE_displist_make_curveTypes(scene, ob, 0);
1914                 }
1915
1916                 mmd->bindfunc = NULL;
1917                 mmd->modifier.mode = mode;
1918         }
1919         
1920         return OPERATOR_FINISHED;
1921 }
1922
1923 static int meshdeform_bind_invoke(bContext *C, wmOperator *op, const wmEvent *UNUSED(event))
1924 {
1925         if (edit_modifier_invoke_properties(C, op))
1926                 return meshdeform_bind_exec(C, op);
1927         else 
1928                 return OPERATOR_CANCELLED;
1929 }
1930
1931 void OBJECT_OT_meshdeform_bind(wmOperatorType *ot)
1932 {
1933         /* identifiers */
1934         ot->name = "Mesh Deform Bind";
1935         ot->description = "Bind mesh to cage in mesh deform modifier";
1936         ot->idname = "OBJECT_OT_meshdeform_bind";
1937         
1938         /* api callbacks */
1939         ot->poll = meshdeform_poll;
1940         ot->invoke = meshdeform_bind_invoke;
1941         ot->exec = meshdeform_bind_exec;
1942         
1943         /* flags */
1944         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO | OPTYPE_INTERNAL;
1945         edit_modifier_properties(ot);
1946 }
1947
1948 /****************** explode refresh operator *********************/
1949
1950 static int explode_poll(bContext *C)
1951 {
1952         return edit_modifier_poll_generic(C, &RNA_ExplodeModifier, 0);
1953 }
1954
1955 static int explode_refresh_exec(bContext *C, wmOperator *op)
1956 {
1957         Object *ob = ED_object_active_context(C);
1958         ExplodeModifierData *emd = (ExplodeModifierData *)edit_modifier_property_get(op, ob, eModifierType_Explode);
1959         
1960         if (!emd)
1961                 return OPERATOR_CANCELLED;
1962
1963         emd->flag |= eExplodeFlag_CalcFaces;
1964
1965         DAG_id_tag_update(&ob->id, OB_RECALC_DATA);
1966         WM_event_add_notifier(C, NC_OBJECT | ND_MODIFIER, ob);
1967         
1968         return OPERATOR_FINISHED;
1969 }
1970
1971 static int explode_refresh_invoke(bContext *C, wmOperator *op, const wmEvent *UNUSED(event))
1972 {
1973         if (edit_modifier_invoke_properties(C, op))
1974                 return explode_refresh_exec(C, op);
1975         else
1976                 return OPERATOR_CANCELLED;
1977 }
1978
1979
1980 void OBJECT_OT_explode_refresh(wmOperatorType *ot)
1981 {
1982         ot->name = "Explode Refresh";
1983         ot->description = "Refresh data in the Explode modifier";
1984         ot->idname = "OBJECT_OT_explode_refresh";
1985
1986         ot->poll = explode_poll;
1987         ot->invoke = explode_refresh_invoke;
1988         ot->exec = explode_refresh_exec;
1989         
1990         /* flags */
1991         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO | OPTYPE_INTERNAL;
1992         edit_modifier_properties(ot);
1993 }
1994
1995
1996 /****************** ocean bake operator *********************/
1997
1998 static int ocean_bake_poll(bContext *C)
1999 {
2000         return edit_modifier_poll_generic(C, &RNA_OceanModifier, 0);
2001 }
2002
2003 /* copied from init_ocean_modifier, MOD_ocean.c */
2004 static void init_ocean_modifier_bake(struct Ocean *oc, struct OceanModifierData *omd)
2005 {
2006         int do_heightfield, do_chop, do_normals, do_jacobian;
2007         
2008         if (!omd || !oc) return; 
2009         
2010         do_heightfield = TRUE;
2011         do_chop = (omd->chop_amount > 0);
2012         do_normals = (omd->flag & MOD_OCEAN_GENERATE_NORMALS);
2013         do_jacobian = (omd->flag & MOD_OCEAN_GENERATE_FOAM);
2014         
2015         BKE_init_ocean(oc, omd->resolution * omd->resolution, omd->resolution * omd->resolution, omd->spatial_size, omd->spatial_size,
2016                        omd->wind_velocity, omd->smallest_wave, 1.0, omd->wave_direction, omd->damp, omd->wave_alignment,
2017                        omd->depth, omd->time,
2018                        do_heightfield, do_chop, do_normals, do_jacobian,
2019                        omd->seed);
2020 }
2021
2022 typedef struct OceanBakeJob {
2023         /* from wmJob */
2024         void *owner;
2025         short *stop, *do_update;
2026         float *progress;
2027         int current_frame;
2028         struct OceanCache *och;
2029         struct Ocean *ocean;
2030         struct OceanModifierData *omd;
2031 } OceanBakeJob;
2032
2033 static void oceanbake_free(void *customdata)
2034 {
2035         OceanBakeJob *oj = customdata;
2036         MEM_freeN(oj);
2037 }
2038
2039 /* called by oceanbake, only to check job 'stop' value */
2040 static int oceanbake_breakjob(void *UNUSED(customdata))
2041 {
2042         //OceanBakeJob *ob = (OceanBakeJob *)customdata;
2043         //return *(ob->stop);
2044         
2045         /* this is not nice yet, need to make the jobs list template better 
2046          * for identifying/acting upon various different jobs */
2047         /* but for now we'll reuse the render break... */
2048         return (G.is_break);
2049 }
2050
2051 /* called by oceanbake, wmJob sends notifier */
2052 static void oceanbake_update(void *customdata, float progress, int *cancel)
2053 {
2054         OceanBakeJob *oj = customdata;
2055         
2056         if (oceanbake_breakjob(oj))
2057                 *cancel = 1;
2058         
2059         *(oj->do_update) = TRUE;
2060         *(oj->progress) = progress;
2061 }
2062
2063 static void oceanbake_startjob(void *customdata, short *stop, short *do_update, float *progress)
2064 {
2065         OceanBakeJob *oj = customdata;
2066         
2067         oj->stop = stop;
2068         oj->do_update = do_update;
2069         oj->progress = progress;
2070         
2071         G.is_break = FALSE;   /* XXX shared with render - replace with job 'stop' switch */
2072         
2073         BKE_bake_ocean(oj->ocean, oj->och, oceanbake_update, (void *)oj);
2074         
2075         *do_update = TRUE;
2076         *stop = 0;
2077 }
2078
2079 static void oceanbake_endjob(void *customdata)
2080 {
2081         OceanBakeJob *oj = customdata;
2082         
2083         if (oj->ocean) {
2084                 BKE_free_ocean(oj->ocean);
2085                 oj->ocean = NULL;
2086         }
2087         
2088         oj->omd->oceancache = oj->och;
2089         oj->omd->cached = TRUE;
2090 }
2091
2092 static int ocean_bake_exec(bContext *C, wmOperator *op)
2093 {
2094         Object *ob = ED_object_active_context(C);
2095         OceanModifierData *omd = (OceanModifierData *)edit_modifier_property_get(op, ob, eModifierType_Ocean);
2096         Scene *scene = CTX_data_scene(C);
2097         OceanCache *och;
2098         struct Ocean *ocean;
2099         int f, cfra, i = 0;
2100         int free = RNA_boolean_get(op->ptr, "free");
2101         
2102         wmJob *wm_job;
2103         OceanBakeJob *oj;
2104         
2105         if (!omd)
2106                 return OPERATOR_CANCELLED;
2107         
2108         if (free) {
2109                 omd->refresh |= MOD_OCEAN_REFRESH_CLEAR_CACHE;
2110                 DAG_id_tag_update(&ob->id, OB_RECALC_DATA);
2111                 WM_event_add_notifier(C, NC_OBJECT | ND_MODIFIER, ob);
2112                 return OPERATOR_FINISHED;
2113         }
2114
2115         och = BKE_init_ocean_cache(omd->cachepath, modifier_path_relbase(ob),
2116                                    omd->bakestart, omd->bakeend, omd->wave_scale,
2117                                    omd->chop_amount, omd->foam_coverage, omd->foam_fade, omd->resolution);
2118         
2119         och->time = MEM_mallocN(och->duration * sizeof(float), "foam bake time");
2120         
2121         cfra = scene->r.cfra;
2122         
2123         /* precalculate time variable before baking */
2124         for (f = omd->bakestart; f <= omd->bakeend; f++) {
2125                 /* from physics_fluid.c:
2126                  *
2127                  * XXX: This can't be used due to an anim sys optimization that ignores recalc object animation,
2128                  * leaving it for the depgraph (this ignores object animation such as modifier properties though... :/ )
2129                  * --> BKE_animsys_evaluate_all_animation(G.main, eval_time);
2130                  * This doesn't work with drivers:
2131                  * --> BKE_animsys_evaluate_animdata(&fsDomain->id, fsDomain->adt, eval_time, ADT_RECALC_ALL);
2132                  */
2133                 
2134                 /* Modifying the global scene isn't nice, but we can do it in 
2135                  * this part of the process before a threaded job is created */
2136                 
2137                 //scene->r.cfra = f;
2138                 //ED_update_for_newframe(CTX_data_main(C), scene, 1);
2139                 
2140                 /* ok, this doesn't work with drivers, but is way faster. 
2141                  * let's use this for now and hope nobody wants to drive the time value... */
2142                 BKE_animsys_evaluate_animdata(scene, (ID *)ob, ob->adt, f, ADT_RECALC_ANIM);
2143                 
2144                 och->time[i] = omd->time;
2145                 i++;
2146         }
2147         
2148         /* make a copy of ocean to use for baking - threadsafety */
2149         ocean = BKE_add_ocean();
2150         init_ocean_modifier_bake(ocean, omd);
2151         
2152 #if 0
2153         BKE_bake_ocean(ocean, och);
2154         
2155         omd->oceancache = och;
2156         omd->cached = TRUE;
2157         
2158         scene->r.cfra = cfra;
2159         
2160         DAG_id_tag_update(&ob->id, OB_RECALC_DATA);
2161         WM_event_add_notifier(C, NC_OBJECT | ND_MODIFIER, ob);
2162 #endif
2163         
2164         /* job stuff */
2165         
2166         scene->r.cfra = cfra;
2167         
2168         /* setup job */
2169         wm_job = WM_jobs_get(CTX_wm_manager(C), CTX_wm_window(C), scene, "Ocean Simulation",
2170                              WM_JOB_PROGRESS, WM_JOB_TYPE_OBJECT_SIM_OCEAN);
2171         oj = MEM_callocN(sizeof(OceanBakeJob), "ocean bake job");
2172         oj->ocean = ocean;
2173         oj->och = och;
2174         oj->omd = omd;
2175         
2176         WM_jobs_customdata_set(wm_job, oj, oceanbake_free);
2177         WM_jobs_timer(wm_job, 0.1, NC_OBJECT | ND_MODIFIER, NC_OBJECT | ND_MODIFIER);
2178         WM_jobs_callbacks(wm_job, oceanbake_startjob, NULL, NULL, oceanbake_endjob);
2179         
2180         WM_jobs_start(CTX_wm_manager(C), wm_job);
2181         
2182         
2183         
2184         return OPERATOR_FINISHED;
2185 }
2186
2187 static int ocean_bake_invoke(bContext *C, wmOperator *op, const wmEvent *UNUSED(event))
2188 {
2189         if (edit_modifier_invoke_properties(C, op))
2190                 return ocean_bake_exec(C, op);
2191         else
2192                 return OPERATOR_CANCELLED;
2193 }
2194
2195
2196 void OBJECT_OT_ocean_bake(wmOperatorType *ot)
2197 {
2198         ot->name = "Bake Ocean";
2199         ot->description = "Bake an image sequence of ocean data";
2200         ot->idname = "OBJECT_OT_ocean_bake";
2201         
2202         ot->poll = ocean_bake_poll;
2203         ot->invoke = ocean_bake_invoke;
2204         ot->exec = ocean_bake_exec;
2205         
2206         /* flags */
2207         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO | OPTYPE_INTERNAL;
2208         edit_modifier_properties(ot);
2209         
2210         RNA_def_boolean(ot->srna, "free", FALSE, "Free", "Free the bake, rather than generating it");
2211 }
2212
2213 /************************ LaplacianDeform bind operator *********************/
2214
2215 static int laplaciandeform_poll(bContext *C)
2216 {
2217         return edit_modifier_poll_generic(C, &RNA_LaplacianDeformModifier, 0);
2218 }
2219
2220 static int laplaciandeform_bind_exec(bContext *C, wmOperator *op)
2221 {
2222         Object *ob = ED_object_active_context(C);
2223         LaplacianDeformModifierData *lmd = (LaplacianDeformModifierData *)edit_modifier_property_get(op, ob, eModifierType_LaplacianDeform);
2224
2225         if (!lmd)
2226                 return OPERATOR_CANCELLED;
2227         if (lmd->flag & MOD_LAPLACIANDEFORM_BIND) {
2228                 lmd->flag &= ~MOD_LAPLACIANDEFORM_BIND;
2229         }
2230         else {
2231                 lmd->flag |= MOD_LAPLACIANDEFORM_BIND;
2232         }
2233         DAG_id_tag_update(&ob->id, OB_RECALC_DATA);
2234         WM_event_add_notifier(C, NC_OBJECT | ND_MODIFIER, ob);
2235         return OPERATOR_FINISHED;
2236 }
2237
2238 static int laplaciandeform_bind_invoke(bContext *C, wmOperator *op, const wmEvent *UNUSED(event))
2239 {
2240         if (edit_modifier_invoke_properties(C, op))
2241                 return laplaciandeform_bind_exec(C, op);
2242         else 
2243                 return OPERATOR_CANCELLED;
2244 }
2245
2246 void OBJECT_OT_laplaciandeform_bind(wmOperatorType *ot)
2247 {
2248         /* identifiers */
2249         ot->name = "Laplacian Deform Bind";
2250         ot->description = "Bind mesh to system in laplacian deform modifier";
2251         ot->idname = "OBJECT_OT_laplaciandeform_bind";
2252         
2253         /* api callbacks */
2254         ot->poll = laplaciandeform_poll;
2255         ot->invoke = laplaciandeform_bind_invoke;
2256         ot->exec = laplaciandeform_bind_exec;
2257         
2258         /* flags */
2259         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO | OPTYPE_INTERNAL;
2260         edit_modifier_properties(ot);
2261 }