Fix #30496: Bugs and crashes about "make links modifers" function.
[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_curve_types.h"
39 #include "DNA_key_types.h"
40 #include "DNA_mesh_types.h"
41 #include "DNA_meshdata_types.h"
42 #include "DNA_object_force.h"
43 #include "DNA_scene_types.h"
44
45 #include "BLI_math.h"
46 #include "BLI_listbase.h"
47 #include "BLI_string.h"
48 #include "BLI_path_util.h"
49 #include "BLI_utildefines.h"
50
51 #include "BKE_animsys.h"
52 #include "BKE_curve.h"
53 #include "BKE_context.h"
54 #include "BKE_depsgraph.h"
55 #include "BKE_displist.h"
56 #include "BKE_DerivedMesh.h"
57 #include "BKE_effect.h"
58 #include "BKE_global.h"
59 #include "BKE_key.h"
60 #include "BKE_lattice.h"
61 #include "BKE_main.h"
62 #include "BKE_mesh.h"
63 #include "BKE_modifier.h"
64 #include "BKE_multires.h"
65 #include "BKE_report.h"
66 #include "BKE_object.h"
67 #include "BKE_ocean.h"
68 #include "BKE_particle.h"
69 #include "BKE_softbody.h"
70 #include "BKE_tessmesh.h"
71
72 #include "RNA_access.h"
73 #include "RNA_define.h"
74 #include "RNA_enum_types.h"
75
76 #include "ED_armature.h"
77 #include "ED_object.h"
78 #include "ED_screen.h"
79 #include "ED_mesh.h"
80
81 #include "WM_api.h"
82 #include "WM_types.h"
83
84 #include "object_intern.h"
85
86 /******************************** API ****************************/
87
88 ModifierData *ED_object_modifier_add(ReportList *reports, Main *bmain, Scene *scene, Object *ob, const char *name, int type)
89 {
90         ModifierData *md=NULL, *new_md=NULL;
91         ModifierTypeInfo *mti = modifierType_getInfo(type);
92         
93         /* only geometry objects should be able to get modifiers [#25291] */
94         if(!ELEM5(ob->type, OB_MESH, OB_CURVE, OB_SURF, OB_FONT, OB_LATTICE)) {
95                 BKE_reportf(reports, RPT_WARNING, "Modifiers cannot be added to Object '%s'", ob->id.name+2);
96                 return NULL;
97         }
98         
99         if(mti->flags&eModifierTypeFlag_Single) {
100                 if(modifiers_findByType(ob, type)) {
101                         BKE_report(reports, RPT_WARNING, "Only one modifier of this type allowed");
102                         return NULL;
103                 }
104         }
105         
106         if(type == eModifierType_ParticleSystem) {
107                 /* don't need to worry about the new modifier's name, since that is set to the number
108                  * of particle systems which shouldn't have too many duplicates 
109                  */
110                 new_md = object_add_particle_system(scene, ob, name);
111         }
112         else {
113                 /* get new modifier data to add */
114                 new_md= modifier_new(type);
115                 
116                 if(mti->flags&eModifierTypeFlag_RequiresOriginalData) {
117                         md = ob->modifiers.first;
118                         
119                         while(md && modifierType_getInfo(md->type)->type==eModifierTypeType_OnlyDeform)
120                                 md = md->next;
121                         
122                         BLI_insertlinkbefore(&ob->modifiers, md, new_md);
123                 }
124                 else
125                         BLI_addtail(&ob->modifiers, new_md);
126
127                 if(name)
128                         BLI_strncpy(new_md->name, name, sizeof(new_md->name));
129
130                 /* make sure modifier data has unique name */
131
132                 modifier_unique_name(&ob->modifiers, new_md);
133                 
134                 /* special cases */
135                 if(type == eModifierType_Softbody) {
136                         if(!ob->soft) {
137                                 ob->soft= sbNew(scene);
138                                 ob->softflag |= OB_SB_GOAL|OB_SB_EDGES;
139                         }
140                 }
141                 else if(type == eModifierType_Collision) {
142                         if(!ob->pd)
143                                 ob->pd= object_add_collision_fields(0);
144                         
145                         ob->pd->deflect= 1;
146                         DAG_scene_sort(bmain, scene);
147                 }
148                 else if(type == eModifierType_Surface)
149                         DAG_scene_sort(bmain, scene);
150                 else if(type == eModifierType_Multires)
151                         /* set totlvl from existing MDISPS layer if object already had it */
152                         multiresModifier_set_levels_from_disps((MultiresModifierData *)new_md, ob);
153         }
154
155         DAG_id_tag_update(&ob->id, OB_RECALC_DATA);
156
157         return new_md;
158 }
159
160 static int object_modifier_remove(Object *ob, ModifierData *md, int *sort_depsgraph)
161 {
162         ModifierData *obmd;
163
164         /* It seems on rapid delete it is possible to
165          * get called twice on same modifier, so make
166          * sure it is in list. */
167         for(obmd=ob->modifiers.first; obmd; obmd=obmd->next)
168                 if(obmd==md)
169                         break;
170
171         if(!obmd)
172                 return 0;
173
174         /* special cases */
175         if(md->type == eModifierType_ParticleSystem) {
176                 ParticleSystemModifierData *psmd=(ParticleSystemModifierData*)md;
177
178                 BLI_remlink(&ob->particlesystem, psmd->psys);
179                 psys_free(ob, psmd->psys);
180                 psmd->psys= NULL;
181         }
182         else if(md->type == eModifierType_Softbody) {
183                 if(ob->soft) {
184                         sbFree(ob->soft);
185                         ob->soft= NULL;
186                         ob->softflag= 0;
187                 }
188         }
189         else if(md->type == eModifierType_Collision) {
190                 if(ob->pd)
191                         ob->pd->deflect= 0;
192
193                 *sort_depsgraph = 1;
194         }
195         else if(md->type == eModifierType_Surface) {
196                 if(ob->pd && ob->pd->shape == PFIELD_SHAPE_SURFACE)
197                         ob->pd->shape = PFIELD_SHAPE_PLANE;
198
199                 *sort_depsgraph = 1;
200         }
201         else if(md->type == eModifierType_Smoke) {
202                 ob->dt = OB_TEXTURE;
203         }
204         else if(md->type == eModifierType_Multires) {
205                 int ok= 1;
206                 Mesh *me= ob->data;
207                 ModifierData *tmpmd;
208
209                 /* ensure MDISPS CustomData layer is't used by another multires modifiers */
210                 for(tmpmd= ob->modifiers.first; tmpmd; tmpmd= tmpmd->next)
211                         if(tmpmd!=md && tmpmd->type == eModifierType_Multires) {
212                                 ok= 0;
213                                 break;
214                         }
215
216                 if(ok) {
217                         if(me->edit_btmesh) {
218                                 BMEditMesh *em= me->edit_btmesh;
219                                 /* CustomData_external_remove is used here only to mark layer as non-external
220                                  * for further free-ing, so zero element count looks safer than em->totface */
221                                 CustomData_external_remove(&em->bm->ldata, &me->id, CD_MDISPS, 0);
222                                 BM_data_layer_free(em->bm, &em->bm->ldata, CD_MDISPS);
223                         } else {
224                                 CustomData_external_remove(&me->ldata, &me->id, CD_MDISPS, me->totloop);
225                                 CustomData_free_layer_active(&me->ldata, CD_MDISPS, me->totloop);
226                         }
227                 }
228         }
229
230         if(ELEM(md->type, eModifierType_Softbody, eModifierType_Cloth) &&
231                 ob->particlesystem.first == NULL) {
232                 ob->mode &= ~OB_MODE_PARTICLE_EDIT;
233         }
234
235         BLI_remlink(&ob->modifiers, md);
236         modifier_free(md);
237
238         return 1;
239 }
240
241 int ED_object_modifier_remove(ReportList *reports, Main *bmain, Scene *scene, Object *ob, ModifierData *md)
242 {
243         int sort_depsgraph = 0;
244         int ok;
245
246         ok= object_modifier_remove(ob, md, &sort_depsgraph);
247
248         if(!ok) {
249                 BKE_reportf(reports, RPT_ERROR, "Modifier '%s' not in object '%s'", ob->id.name, md->name);
250                 return 0;
251         }
252
253         DAG_id_tag_update(&ob->id, OB_RECALC_DATA);
254
255         /* sorting has to be done after the update so that dynamic systems can react properly */
256         if(sort_depsgraph)
257                 DAG_scene_sort(bmain, scene);
258
259         return 1;
260 }
261
262 void ED_object_modifier_clear(Main *bmain, Scene *scene, Object *ob)
263 {
264         ModifierData *md = ob->modifiers.first;
265         int sort_depsgraph = 0;
266
267         if(!md)
268                 return;
269
270         while(md) {
271                 ModifierData *next_md;
272
273                 next_md= md->next;
274
275                 object_modifier_remove(ob, md, &sort_depsgraph);
276
277                 md= next_md;
278         }
279
280         DAG_id_tag_update(&ob->id, OB_RECALC_DATA);
281
282         /* sorting has to be done after the update so that dynamic systems can react properly */
283         if(sort_depsgraph)
284                 DAG_scene_sort(bmain, scene);
285 }
286
287 int ED_object_modifier_move_up(ReportList *reports, Object *ob, ModifierData *md)
288 {
289         if(md->prev) {
290                 ModifierTypeInfo *mti = modifierType_getInfo(md->type);
291
292                 if(mti->type!=eModifierTypeType_OnlyDeform) {
293                         ModifierTypeInfo *nmti = modifierType_getInfo(md->prev->type);
294
295                         if(nmti->flags&eModifierTypeFlag_RequiresOriginalData) {
296                                 BKE_report(reports, RPT_WARNING, "Cannot move above a modifier requiring original data");
297                                 return 0;
298                         }
299                 }
300
301                 BLI_remlink(&ob->modifiers, md);
302                 BLI_insertlink(&ob->modifiers, md->prev->prev, md);
303         }
304
305         return 1;
306 }
307
308 int ED_object_modifier_move_down(ReportList *reports, Object *ob, ModifierData *md)
309 {
310         if(md->next) {
311                 ModifierTypeInfo *mti = modifierType_getInfo(md->type);
312
313                 if(mti->flags&eModifierTypeFlag_RequiresOriginalData) {
314                         ModifierTypeInfo *nmti = modifierType_getInfo(md->next->type);
315
316                         if(nmti->type!=eModifierTypeType_OnlyDeform) {
317                                 BKE_report(reports, RPT_WARNING, "Cannot move beyond a non-deforming modifier");
318                                 return 0;
319                         }
320                 }
321
322                 BLI_remlink(&ob->modifiers, md);
323                 BLI_insertlink(&ob->modifiers, md->next, md);
324         }
325
326         return 1;
327 }
328
329 int ED_object_modifier_convert(ReportList *UNUSED(reports), Main *bmain, Scene *scene, Object *ob, ModifierData *md)
330 {
331         Object *obn;
332         ParticleSystem *psys;
333         ParticleCacheKey *key, **cache;
334         ParticleSettings *part;
335         Mesh *me;
336         MVert *mvert;
337         MEdge *medge;
338         int a, k, kmax;
339         int totvert=0, totedge=0, cvert=0;
340         int totpart=0, totchild=0;
341
342         if(md->type != eModifierType_ParticleSystem) return 0;
343         if(ob && ob->mode & OB_MODE_PARTICLE_EDIT) return 0;
344
345         psys=((ParticleSystemModifierData *)md)->psys;
346         part= psys->part;
347
348         if(part->ren_as != PART_DRAW_PATH || psys->pathcache == NULL)
349                 return 0;
350
351         totpart= psys->totcached;
352         totchild= psys->totchildcache;
353
354         if(totchild && (part->draw&PART_DRAW_PARENT)==0)
355                 totpart= 0;
356
357         /* count */
358         cache= psys->pathcache;
359         for(a=0; a<totpart; a++) {
360                 key= cache[a];
361
362                 if(key->steps > 0) {
363                         totvert+= key->steps+1;
364                         totedge+= key->steps;
365                 }
366         }
367
368         cache= psys->childcache;
369         for(a=0; a<totchild; a++) {
370                 key= cache[a];
371
372                 if(key->steps > 0) {
373                         totvert+= key->steps+1;
374                         totedge+= key->steps;
375                 }
376         }
377
378         if(totvert==0) return 0;
379
380         /* add new mesh */
381         obn= add_object(scene, OB_MESH);
382         me= obn->data;
383         
384         me->totvert= totvert;
385         me->totedge= totedge;
386         
387         me->mvert= CustomData_add_layer(&me->vdata, CD_MVERT, CD_CALLOC, NULL, totvert);
388         me->medge= CustomData_add_layer(&me->edata, CD_MEDGE, CD_CALLOC, NULL, totedge);
389         me->mface= CustomData_add_layer(&me->fdata, CD_MFACE, CD_CALLOC, NULL, 0);
390         
391         mvert= me->mvert;
392         medge= me->medge;
393
394         /* copy coordinates */
395         cache= psys->pathcache;
396         for(a=0; a<totpart; a++) {
397                 key= cache[a];
398                 kmax= key->steps;
399                 for(k=0; k<=kmax; k++,key++,cvert++,mvert++) {
400                         copy_v3_v3(mvert->co,key->co);
401                         if(k) {
402                                 medge->v1= cvert-1;
403                                 medge->v2= cvert;
404                                 medge->flag= ME_EDGEDRAW|ME_EDGERENDER|ME_LOOSEEDGE;
405                                 medge++;
406                         }
407                         else {
408                                 /* cheap trick to select the roots */
409                                 mvert->flag |= SELECT;
410                         }
411                 }
412         }
413
414         cache=psys->childcache;
415         for(a=0; a<totchild; a++) {
416                 key=cache[a];
417                 kmax=key->steps;
418                 for(k=0; k<=kmax; k++,key++,cvert++,mvert++) {
419                         copy_v3_v3(mvert->co,key->co);
420                         if(k) {
421                                 medge->v1=cvert-1;
422                                 medge->v2=cvert;
423                                 medge->flag= ME_EDGEDRAW|ME_EDGERENDER|ME_LOOSEEDGE;
424                                 medge++;
425                         }
426                         else {
427                                 /* cheap trick to select the roots */
428                                 mvert->flag |= SELECT;
429                         }
430                 }
431         }
432
433         DAG_scene_sort(bmain, scene);
434
435         return 1;
436 }
437
438 static int modifier_apply_shape(ReportList *reports, Scene *scene, Object *ob, ModifierData *md)
439 {
440         ModifierTypeInfo *mti= modifierType_getInfo(md->type);
441
442         md->scene= scene;
443
444         if (mti->isDisabled && mti->isDisabled(md, 0)) {
445                 BKE_report(reports, RPT_ERROR, "Modifier is disabled, skipping apply");
446                 return 0;
447         }
448
449         /*
450          * It should be ridiculously easy to extract the original verts that we want
451          * and form the shape data.  We can probably use the CD KEYINDEX layer (or
452          * whatever I ended up calling it, too tired to check now), though this would
453          * by necessity have to make some potentially ugly assumptions about the order
454          * of the mesh data :-/  you can probably assume in 99% of cases that the first
455          * element of a given index is the original, and any subsequent duplicates are
456          * copies/interpolates, but that's an assumption that would need to be tested
457          * and then predominantly stated in comments in a half dozen headers.
458          */
459
460         if (ob->type==OB_MESH) {
461                 DerivedMesh *dm;
462                 Mesh *me= ob->data;
463                 Key *key=me->key;
464                 KeyBlock *kb;
465                 
466                 if(!modifier_sameTopology(md) || mti->type == eModifierTypeType_NonGeometrical) {
467                         BKE_report(reports, RPT_ERROR, "Only deforming modifiers can be applied to Shapes");
468                         return 0;
469                 }
470                 
471                 dm = mesh_create_derived_for_modifier(scene, ob, md, 0);
472                 if (!dm) {
473                         BKE_report(reports, RPT_ERROR, "Modifier is disabled or returned error, skipping apply");
474                         return 0;
475                 }
476                 
477                 if(key == NULL) {
478                         key= me->key= add_key((ID *)me);
479                         key->type= KEY_RELATIVE;
480                         /* if that was the first key block added, then it was the basis.
481                          * Initialize it with the mesh, and add another for the modifier */
482                         kb= add_keyblock(key, NULL);
483                         mesh_to_key(me, kb);
484                 }
485
486                 kb= add_keyblock(key, md->name);
487                 DM_to_meshkey(dm, me, kb);
488                 
489                 dm->release(dm);
490         }
491         else {
492                 BKE_report(reports, RPT_ERROR, "Cannot apply modifier for this object type");
493                 return 0;
494         }
495         return 1;
496 }
497
498 static int modifier_apply_obdata(ReportList *reports, Scene *scene, Object *ob, ModifierData *md)
499 {
500         ModifierTypeInfo *mti= modifierType_getInfo(md->type);
501
502         md->scene= scene;
503
504         if (mti->isDisabled && mti->isDisabled(md, 0)) {
505                 BKE_report(reports, RPT_ERROR, "Modifier is disabled, skipping apply");
506                 return 0;
507         }
508
509         if (ob->type==OB_MESH) {
510                 DerivedMesh *dm;
511                 Mesh *me = ob->data;
512                 MultiresModifierData *mmd= find_multires_modifier_before(scene, md);
513
514                 if(me->key && mti->type != eModifierTypeType_NonGeometrical) {
515                         BKE_report(reports, RPT_ERROR, "Modifier cannot be applied to Mesh with Shape Keys");
516                         return 0;
517                 }
518
519                 /* Multires: ensure that recent sculpting is applied */
520                 if(md->type == eModifierType_Multires)
521                         multires_force_update(ob);
522
523                 if (mmd && mmd->totlvl && mti->type==eModifierTypeType_OnlyDeform) {
524                         if(!multiresModifier_reshapeFromDeformMod (scene, mmd, ob, md)) {
525                                 BKE_report(reports, RPT_ERROR, "Multires modifier returned error, skipping apply");
526                                 return 0;
527                         }
528                 } else {
529                         dm = mesh_create_derived_for_modifier(scene, ob, md, 1);
530                         if (!dm) {
531                                 BKE_report(reports, RPT_ERROR, "Modifier returned error, skipping apply");
532                                 return 0;
533                         }
534
535                         DM_to_mesh(dm, me, ob);
536
537                         dm->release(dm);
538
539                         if(md->type == eModifierType_Multires) {
540                                 CustomData_external_remove(&me->ldata, &me->id, CD_MDISPS, me->totloop);
541                                 CustomData_free_layer_active(&me->ldata, CD_MDISPS, me->totloop);
542                         }
543                 }
544         }
545         else if (ELEM(ob->type, OB_CURVE, OB_SURF)) {
546                 Curve *cu;
547                 int numVerts;
548                 float (*vertexCos)[3];
549
550                 if (ELEM(mti->type, eModifierTypeType_Constructive, eModifierTypeType_Nonconstructive)) {
551                         BKE_report(reports, RPT_ERROR, "Cannot apply constructive modifiers on curve");
552                         return 0;
553                 }
554
555                 cu = ob->data;
556                 BKE_report(reports, RPT_INFO, "Applied modifier only changed CV points, not tesselated/bevel vertices");
557
558                 vertexCos = curve_getVertexCos(cu, &cu->nurb, &numVerts);
559                 mti->deformVerts(md, ob, NULL, vertexCos, numVerts, 0, 0);
560                 curve_applyVertexCos(cu, &cu->nurb, vertexCos);
561
562                 MEM_freeN(vertexCos);
563
564                 DAG_id_tag_update(&ob->id, OB_RECALC_DATA);
565         }
566         else {
567                 BKE_report(reports, RPT_ERROR, "Cannot apply modifier for this object type");
568                 return 0;
569         }
570
571         /* lattice modifier can be applied to particle system too */
572         if(ob->particlesystem.first) {
573
574                 ParticleSystem *psys = ob->particlesystem.first;
575
576                 for(; psys; psys=psys->next) {
577                         
578                         if(psys->part->type != PART_HAIR)
579                                 continue;
580
581                         psys_apply_hair_lattice(scene, ob, psys);
582                 }
583         }
584
585         return 1;
586 }
587
588 int ED_object_modifier_apply(ReportList *reports, Scene *scene, Object *ob, ModifierData *md, int mode)
589 {
590         int prev_mode;
591
592         if (scene->obedit) {
593                 BKE_report(reports, RPT_ERROR, "Modifiers cannot be applied in editmode");
594                 return 0;
595         } else if (((ID*) ob->data)->us>1) {
596                 BKE_report(reports, RPT_ERROR, "Modifiers cannot be applied to multi-user data");
597                 return 0;
598         }
599
600         if (md!=ob->modifiers.first)
601                 BKE_report(reports, RPT_INFO, "Applied modifier was not first, result may not be as expected");
602
603         /* allow apply of a not-realtime modifier, by first re-enabling realtime. */
604         prev_mode= md->mode;
605         md->mode |= eModifierMode_Realtime;
606
607         if (mode == MODIFIER_APPLY_SHAPE) {
608                 if (!modifier_apply_shape(reports, scene, ob, md)) {
609                         md->mode= prev_mode;
610                         return 0;
611                 }
612         } else {
613                 if (!modifier_apply_obdata(reports, scene, ob, md)) {
614                         md->mode= prev_mode;
615                         return 0;
616                 }
617         }
618
619         BLI_remlink(&ob->modifiers, md);
620         modifier_free(md);
621
622         return 1;
623 }
624
625 int ED_object_modifier_copy(ReportList *UNUSED(reports), Object *ob, ModifierData *md)
626 {
627         ModifierData *nmd;
628         
629         nmd = modifier_new(md->type);
630         modifier_copyData(md, nmd);
631         BLI_insertlink(&ob->modifiers, md, nmd);
632         modifier_unique_name(&ob->modifiers, nmd);
633
634         return 1;
635 }
636
637 /************************ add modifier operator *********************/
638
639 static int modifier_add_exec(bContext *C, wmOperator *op)
640 {
641         Main *bmain= CTX_data_main(C);
642         Scene *scene= CTX_data_scene(C);
643         Object *ob = ED_object_active_context(C);
644         int type= RNA_enum_get(op->ptr, "type");
645
646         if(!ED_object_modifier_add(op->reports, bmain, scene, ob, NULL, type))
647                 return OPERATOR_CANCELLED;
648
649         WM_event_add_notifier(C, NC_OBJECT|ND_MODIFIER, ob);
650         
651         return OPERATOR_FINISHED;
652 }
653
654 static EnumPropertyItem *modifier_add_itemf(bContext *C, PointerRNA *UNUSED(ptr), PropertyRNA *UNUSED(prop), int *free)
655 {       
656         Object *ob= ED_object_active_context(C);
657         EnumPropertyItem *item= NULL, *md_item, *group_item= NULL;
658         ModifierTypeInfo *mti;
659         int totitem= 0, a;
660         
661         if(!ob)
662                 return modifier_type_items;
663
664         for(a=0; modifier_type_items[a].identifier; a++) {
665                 md_item= &modifier_type_items[a];
666
667                 if(md_item->identifier[0]) {
668                         mti= modifierType_getInfo(md_item->value);
669
670                         if(mti->flags & eModifierTypeFlag_NoUserAdd)
671                                 continue;
672
673                         if(!object_support_modifier_type(ob, md_item->value))
674                                 continue;
675                 }
676                 else {
677                         group_item= md_item;
678                         md_item= NULL;
679
680                         continue;
681                 }
682
683                 if(group_item) {
684                         RNA_enum_item_add(&item, &totitem, group_item);
685                         group_item= NULL;
686                 }
687
688                 RNA_enum_item_add(&item, &totitem, md_item);
689         }
690
691         RNA_enum_item_end(&item, &totitem);
692         *free= 1;
693
694         return item;
695 }
696
697 void OBJECT_OT_modifier_add(wmOperatorType *ot)
698 {
699         PropertyRNA *prop;
700
701         /* identifiers */
702         ot->name= "Add Modifier";
703         ot->description = "Add a modifier to the active object";
704         ot->idname= "OBJECT_OT_modifier_add";
705         
706         /* api callbacks */
707         ot->invoke= WM_menu_invoke;
708         ot->exec= modifier_add_exec;
709         ot->poll= ED_operator_object_active_editable;
710         
711         /* flags */
712         ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
713         
714         /* properties */
715         prop= RNA_def_enum(ot->srna, "type", modifier_type_items, eModifierType_Subsurf, "Type", "");
716         RNA_def_enum_funcs(prop, modifier_add_itemf);
717         ot->prop= prop;
718 }
719
720 /************************ generic functions for operators using mod names and data context *********************/
721
722 static int edit_modifier_poll_generic(bContext *C, StructRNA *rna_type, int obtype_flag)
723 {
724         PointerRNA ptr= CTX_data_pointer_get_type(C, "modifier", rna_type);
725         Object *ob= (ptr.id.data)?ptr.id.data:ED_object_active_context(C);
726         
727         if (!ob || ob->id.lib) return 0;
728         if (obtype_flag && ((1<<ob->type) & obtype_flag)==0) return 0;
729         if (ptr.id.data && ((ID*)ptr.id.data)->lib) return 0;
730         
731         return 1;
732 }
733
734 static int edit_modifier_poll(bContext *C)
735 {
736         return edit_modifier_poll_generic(C, &RNA_Modifier, 0);
737 }
738
739 static void edit_modifier_properties(wmOperatorType *ot)
740 {
741         RNA_def_string(ot->srna, "modifier", "", MAX_NAME, "Modifier", "Name of the modifier to edit");
742 }
743
744 static int edit_modifier_invoke_properties(bContext *C, wmOperator *op)
745 {
746         PointerRNA ptr= CTX_data_pointer_get_type(C, "modifier", &RNA_Modifier);
747         ModifierData *md;
748         
749         if (RNA_struct_property_is_set(op->ptr, "modifier"))
750                 return 1;
751         
752         if (ptr.data) {
753                 md = ptr.data;
754                 RNA_string_set(op->ptr, "modifier", md->name);
755                 return 1;
756         }
757         
758         return 0;
759 }
760
761 static ModifierData *edit_modifier_property_get(wmOperator *op, Object *ob, int type)
762 {
763         char modifier_name[MAX_NAME];
764         ModifierData *md;
765         RNA_string_get(op->ptr, "modifier", modifier_name);
766         
767         md = modifiers_findByName(ob, modifier_name);
768         
769         if (md && type != 0 && md->type != type)
770                 md = NULL;
771
772         return md;
773 }
774
775 /************************ remove modifier operator *********************/
776
777 static int modifier_remove_exec(bContext *C, wmOperator *op)
778 {
779         Main *bmain= CTX_data_main(C);
780         Scene *scene= CTX_data_scene(C);
781         Object *ob = ED_object_active_context(C);
782         ModifierData *md = edit_modifier_property_get(op, ob, 0);
783         int mode_orig = ob ? ob->mode : 0;
784         
785         if(!ob || !md || !ED_object_modifier_remove(op->reports, bmain, scene, ob, md))
786                 return OPERATOR_CANCELLED;
787
788         WM_event_add_notifier(C, NC_OBJECT|ND_MODIFIER, ob);
789
790         /* if cloth/softbody was removed, particle mode could be cleared */
791         if(mode_orig & OB_MODE_PARTICLE_EDIT)
792                 if((ob->mode & OB_MODE_PARTICLE_EDIT)==0)
793                         if(scene->basact && scene->basact->object==ob)
794                                 WM_event_add_notifier(C, NC_SCENE|ND_MODE|NS_MODE_OBJECT, NULL);
795         
796         return OPERATOR_FINISHED;
797 }
798
799 static int modifier_remove_invoke(bContext *C, wmOperator *op, wmEvent *UNUSED(event))
800 {
801         if (edit_modifier_invoke_properties(C, op))
802                 return modifier_remove_exec(C, op);
803         else
804                 return OPERATOR_CANCELLED;
805 }
806
807 void OBJECT_OT_modifier_remove(wmOperatorType *ot)
808 {
809         ot->name= "Remove Modifier";
810         ot->description= "Remove a modifier from the active object";
811         ot->idname= "OBJECT_OT_modifier_remove";
812
813         ot->invoke= modifier_remove_invoke;
814         ot->exec= modifier_remove_exec;
815         ot->poll= edit_modifier_poll;
816         
817         /* flags */
818         ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
819         edit_modifier_properties(ot);
820 }
821
822 /************************ move up modifier operator *********************/
823
824 static int modifier_move_up_exec(bContext *C, wmOperator *op)
825 {
826         Object *ob = ED_object_active_context(C);
827         ModifierData *md = edit_modifier_property_get(op, ob, 0);
828
829         if(!ob || !md || !ED_object_modifier_move_up(op->reports, ob, md))
830                 return OPERATOR_CANCELLED;
831
832         DAG_id_tag_update(&ob->id, OB_RECALC_DATA);
833         WM_event_add_notifier(C, NC_OBJECT|ND_MODIFIER, ob);
834         
835         return OPERATOR_FINISHED;
836 }
837
838 static int modifier_move_up_invoke(bContext *C, wmOperator *op, wmEvent *UNUSED(event))
839 {
840         if (edit_modifier_invoke_properties(C, op))
841                 return modifier_move_up_exec(C, op);
842         else
843                 return OPERATOR_CANCELLED;
844 }
845
846 void OBJECT_OT_modifier_move_up(wmOperatorType *ot)
847 {
848         ot->name= "Move Up Modifier";
849         ot->description= "Move modifier up in the stack";
850         ot->idname= "OBJECT_OT_modifier_move_up";
851
852         ot->invoke= modifier_move_up_invoke;
853         ot->exec= modifier_move_up_exec;
854         ot->poll= edit_modifier_poll;
855         
856         /* flags */
857         ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
858         edit_modifier_properties(ot);
859 }
860
861 /************************ move down modifier operator *********************/
862
863 static int modifier_move_down_exec(bContext *C, wmOperator *op)
864 {
865         Object *ob = ED_object_active_context(C);
866         ModifierData *md = edit_modifier_property_get(op, ob, 0);
867
868         if(!ob || !md || !ED_object_modifier_move_down(op->reports, ob, md))
869                 return OPERATOR_CANCELLED;
870
871         DAG_id_tag_update(&ob->id, OB_RECALC_DATA);
872         WM_event_add_notifier(C, NC_OBJECT|ND_MODIFIER, ob);
873         
874         return OPERATOR_FINISHED;
875 }
876
877 static int modifier_move_down_invoke(bContext *C, wmOperator *op, wmEvent *UNUSED(event))
878 {
879         if (edit_modifier_invoke_properties(C, op))
880                 return modifier_move_down_exec(C, op);
881         else
882                 return OPERATOR_CANCELLED;
883 }
884
885 void OBJECT_OT_modifier_move_down(wmOperatorType *ot)
886 {
887         ot->name= "Move Down Modifier";
888         ot->description= "Move modifier down in the stack";
889         ot->idname= "OBJECT_OT_modifier_move_down";
890
891         ot->invoke= modifier_move_down_invoke;
892         ot->exec= modifier_move_down_exec;
893         ot->poll= edit_modifier_poll;
894         
895         /* flags */
896         ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
897         edit_modifier_properties(ot);
898 }
899
900 /************************ apply modifier operator *********************/
901
902 static int modifier_apply_exec(bContext *C, wmOperator *op)
903 {
904         Scene *scene= CTX_data_scene(C);
905         Object *ob = ED_object_active_context(C);
906         ModifierData *md = edit_modifier_property_get(op, ob, 0);
907         int apply_as= RNA_enum_get(op->ptr, "apply_as");
908         
909         if(!ob || !md || !ED_object_modifier_apply(op->reports, scene, ob, md, apply_as)) {
910                 return OPERATOR_CANCELLED;
911         }
912
913         DAG_id_tag_update(&ob->id, OB_RECALC_DATA);
914         WM_event_add_notifier(C, NC_OBJECT|ND_MODIFIER, ob);
915         
916         return OPERATOR_FINISHED;
917 }
918
919 static int modifier_apply_invoke(bContext *C, wmOperator *op, wmEvent *UNUSED(event))
920 {
921         if (edit_modifier_invoke_properties(C, op))
922                 return modifier_apply_exec(C, op);
923         else
924                 return OPERATOR_CANCELLED;
925 }
926
927 static EnumPropertyItem modifier_apply_as_items[] = {
928         {MODIFIER_APPLY_DATA, "DATA", 0, "Object Data", "Apply modifier to the object's data"},
929         {MODIFIER_APPLY_SHAPE, "SHAPE", 0, "New Shape", "Apply deform-only modifier to a new shape on this object"},
930         {0, NULL, 0, NULL, NULL}};
931
932 void OBJECT_OT_modifier_apply(wmOperatorType *ot)
933 {
934         ot->name= "Apply Modifier";
935         ot->description= "Apply modifier and remove from the stack";
936         ot->idname= "OBJECT_OT_modifier_apply";
937
938         ot->invoke= modifier_apply_invoke;
939         ot->exec= modifier_apply_exec;
940         ot->poll= edit_modifier_poll;
941         
942         /* flags */
943         ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
944         
945         RNA_def_enum(ot->srna, "apply_as", modifier_apply_as_items, MODIFIER_APPLY_DATA, "Apply as", "How to apply the modifier to the geometry");
946         edit_modifier_properties(ot);
947 }
948
949 /************************ convert modifier operator *********************/
950
951 static int modifier_convert_exec(bContext *C, wmOperator *op)
952 {
953         Main *bmain= CTX_data_main(C);
954         Scene *scene= CTX_data_scene(C);
955         Object *ob = ED_object_active_context(C);
956         ModifierData *md = edit_modifier_property_get(op, ob, 0);
957         
958         if(!ob || !md || !ED_object_modifier_convert(op->reports, bmain, scene, 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_convert_invoke(bContext *C, wmOperator *op, wmEvent *UNUSED(event))
968 {
969         if (edit_modifier_invoke_properties(C, op))
970                 return modifier_convert_exec(C, op);
971         else
972                 return OPERATOR_CANCELLED;
973 }
974
975 void OBJECT_OT_modifier_convert(wmOperatorType *ot)
976 {
977         ot->name= "Convert Modifier";
978         ot->description= "Convert particles to a mesh object";
979         ot->idname= "OBJECT_OT_modifier_convert";
980
981         ot->invoke= modifier_convert_invoke;
982         ot->exec= modifier_convert_exec;
983         ot->poll= edit_modifier_poll;
984         
985         /* flags */
986         ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
987         edit_modifier_properties(ot);
988 }
989
990 /************************ copy modifier operator *********************/
991
992 static int modifier_copy_exec(bContext *C, wmOperator *op)
993 {
994         Object *ob = ED_object_active_context(C);
995         ModifierData *md = edit_modifier_property_get(op, ob, 0);
996
997         if(!ob || !md || !ED_object_modifier_copy(op->reports, ob, md))
998                 return OPERATOR_CANCELLED;
999
1000         DAG_id_tag_update(&ob->id, OB_RECALC_DATA);
1001         WM_event_add_notifier(C, NC_OBJECT|ND_MODIFIER, ob);
1002         
1003         return OPERATOR_FINISHED;
1004 }
1005
1006 static int modifier_copy_invoke(bContext *C, wmOperator *op, wmEvent *UNUSED(event))
1007 {
1008         if (edit_modifier_invoke_properties(C, op))
1009                 return modifier_copy_exec(C, op);
1010         else
1011                 return OPERATOR_CANCELLED;
1012 }
1013
1014 void OBJECT_OT_modifier_copy(wmOperatorType *ot)
1015 {
1016         ot->name= "Copy Modifier";
1017         ot->description= "Duplicate modifier at the same position in the stack";
1018         ot->idname= "OBJECT_OT_modifier_copy";
1019
1020         ot->invoke= modifier_copy_invoke;
1021         ot->exec= modifier_copy_exec;
1022         ot->poll= edit_modifier_poll;
1023         
1024         /* flags */
1025         ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
1026         edit_modifier_properties(ot);
1027 }
1028
1029 /************* multires delete higher levels operator ****************/
1030
1031 static int multires_poll(bContext *C)
1032 {
1033         return edit_modifier_poll_generic(C, &RNA_MultiresModifier, (1<<OB_MESH));
1034 }
1035
1036 static int multires_higher_levels_delete_exec(bContext *C, wmOperator *op)
1037 {
1038         Object *ob = ED_object_active_context(C);
1039         MultiresModifierData *mmd = (MultiresModifierData *)edit_modifier_property_get(op, ob, eModifierType_Multires);
1040         
1041         if (!mmd)
1042                 return OPERATOR_CANCELLED;
1043         
1044         multiresModifier_del_levels(mmd, ob, 1);
1045         
1046         WM_event_add_notifier(C, NC_OBJECT|ND_MODIFIER, ob);
1047         
1048         return OPERATOR_FINISHED;
1049 }
1050
1051 static int multires_higher_levels_delete_invoke(bContext *C, wmOperator *op, wmEvent *UNUSED(event))
1052 {
1053         if (edit_modifier_invoke_properties(C, op))
1054                 return multires_higher_levels_delete_exec(C, op);
1055         else
1056                 return OPERATOR_CANCELLED;
1057 }
1058
1059 void OBJECT_OT_multires_higher_levels_delete(wmOperatorType *ot)
1060 {
1061         ot->name= "Delete Higher Levels";
1062         ot->description= "Deletes the higher resolution mesh, potential loss of detail";
1063         ot->idname= "OBJECT_OT_multires_higher_levels_delete";
1064
1065         ot->poll= multires_poll;
1066         ot->invoke= multires_higher_levels_delete_invoke;
1067         ot->exec= multires_higher_levels_delete_exec;
1068         
1069         /* flags */
1070         ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
1071         edit_modifier_properties(ot);
1072 }
1073
1074 /****************** multires subdivide operator *********************/
1075
1076 static int multires_subdivide_exec(bContext *C, wmOperator *op)
1077 {
1078         Object *ob = ED_object_active_context(C);
1079         MultiresModifierData *mmd = (MultiresModifierData *)edit_modifier_property_get(op, ob, eModifierType_Multires);
1080         
1081         if (!mmd)
1082                 return OPERATOR_CANCELLED;
1083         
1084         multiresModifier_subdivide(mmd, ob, 0, mmd->simple);
1085
1086         DAG_id_tag_update(&ob->id, OB_RECALC_DATA);
1087         WM_event_add_notifier(C, NC_OBJECT|ND_MODIFIER, ob);
1088         
1089         return OPERATOR_FINISHED;
1090 }
1091
1092 static int multires_subdivide_invoke(bContext *C, wmOperator *op, wmEvent *UNUSED(event))
1093 {
1094         if (edit_modifier_invoke_properties(C, op))
1095                 return multires_subdivide_exec(C, op);
1096         else
1097                 return OPERATOR_CANCELLED;
1098 }
1099
1100 void OBJECT_OT_multires_subdivide(wmOperatorType *ot)
1101 {
1102         ot->name= "Multires Subdivide";
1103         ot->description= "Add a new level of subdivision";
1104         ot->idname= "OBJECT_OT_multires_subdivide";
1105
1106         ot->poll= multires_poll;
1107         ot->invoke= multires_subdivide_invoke;
1108         ot->exec= multires_subdivide_exec;
1109         
1110         /* flags */
1111         ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
1112         edit_modifier_properties(ot);
1113 }
1114
1115 /****************** multires reshape operator *********************/
1116
1117 static int multires_reshape_exec(bContext *C, wmOperator *op)
1118 {
1119         Object *ob= ED_object_active_context(C), *secondob= NULL;
1120         Scene *scene= CTX_data_scene(C);
1121         MultiresModifierData *mmd = (MultiresModifierData *)edit_modifier_property_get(op, ob, eModifierType_Multires);
1122
1123         if (!mmd)
1124                 return OPERATOR_CANCELLED;
1125
1126         if(mmd->lvl==0) {
1127                 BKE_report(op->reports, RPT_ERROR, "Reshape can work only with higher levels of subdivisions");
1128                 return OPERATOR_CANCELLED;
1129         }
1130
1131         CTX_DATA_BEGIN(C, Object*, selob, selected_editable_objects) {
1132                 if(selob->type == OB_MESH && selob != ob) {
1133                         secondob= selob;
1134                         break;
1135                 }
1136         }
1137         CTX_DATA_END;
1138
1139         if(!secondob) {
1140                 BKE_report(op->reports, RPT_ERROR, "Second selected mesh object require to copy shape from");
1141                 return OPERATOR_CANCELLED;
1142         }
1143
1144         if(!multiresModifier_reshape(scene, mmd, ob, secondob)) {
1145                 BKE_report(op->reports, RPT_ERROR, "Objects do not have the same number of vertices");
1146                 return OPERATOR_CANCELLED;
1147         }
1148
1149         DAG_id_tag_update(&ob->id, OB_RECALC_DATA);
1150         WM_event_add_notifier(C, NC_OBJECT|ND_MODIFIER, ob);
1151
1152         return OPERATOR_FINISHED;
1153 }
1154
1155 static int multires_reshape_invoke(bContext *C, wmOperator *op, wmEvent *UNUSED(event))
1156 {
1157         if (edit_modifier_invoke_properties(C, op))
1158                 return multires_reshape_exec(C, op);
1159         else
1160                 return OPERATOR_CANCELLED;
1161 }
1162
1163 void OBJECT_OT_multires_reshape(wmOperatorType *ot)
1164 {
1165         ot->name= "Multires Reshape";
1166         ot->description= "Copy vertex coordinates from other object";
1167         ot->idname= "OBJECT_OT_multires_reshape";
1168
1169         ot->poll= multires_poll;
1170         ot->invoke= multires_reshape_invoke;
1171         ot->exec= multires_reshape_exec;
1172         
1173         /* flags */
1174         ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
1175         edit_modifier_properties(ot);
1176 }
1177
1178
1179                 
1180 /****************** multires save external operator *********************/
1181
1182 static int multires_external_save_exec(bContext *C, wmOperator *op)
1183 {
1184         Object *ob = ED_object_active_context(C);
1185         Mesh *me= (ob)? ob->data: op->customdata;
1186         char path[FILE_MAX];
1187         int relative= RNA_boolean_get(op->ptr, "relative_path");
1188
1189         if(!me)
1190                 return OPERATOR_CANCELLED;
1191
1192         if(CustomData_external_test(&me->ldata, CD_MDISPS))
1193                 return OPERATOR_CANCELLED;
1194         
1195         RNA_string_get(op->ptr, "filepath", path);
1196
1197         if(relative)
1198                 BLI_path_rel(path, G.main->name);
1199
1200         CustomData_external_add(&me->ldata, &me->id, CD_MDISPS, me->totloop, path);
1201         CustomData_external_write(&me->ldata, &me->id, CD_MASK_MESH, me->totloop, 0);
1202         
1203         return OPERATOR_FINISHED;
1204 }
1205
1206 static int multires_external_save_invoke(bContext *C, wmOperator *op, wmEvent *UNUSED(event))
1207 {
1208         Object *ob = ED_object_active_context(C);
1209         MultiresModifierData *mmd;
1210         Mesh *me= ob->data;
1211         char path[FILE_MAX];
1212
1213         if (!edit_modifier_invoke_properties(C, op))
1214                 return OPERATOR_CANCELLED;
1215         
1216         mmd = (MultiresModifierData *)edit_modifier_property_get(op, ob, eModifierType_Multires);
1217         
1218         if (!mmd)
1219                 return OPERATOR_CANCELLED;
1220         
1221         if(CustomData_external_test(&me->ldata, CD_MDISPS))
1222                 return OPERATOR_CANCELLED;
1223
1224         if(RNA_struct_property_is_set(op->ptr, "filepath"))
1225                 return multires_external_save_exec(C, op);
1226         
1227         op->customdata= me;
1228
1229         BLI_snprintf(path, sizeof(path), "//%s.btx", me->id.name+2);
1230         RNA_string_set(op->ptr, "filepath", path);
1231         
1232         WM_event_add_fileselect(C, op);
1233
1234         return OPERATOR_RUNNING_MODAL;
1235 }
1236
1237 void OBJECT_OT_multires_external_save(wmOperatorType *ot)
1238 {
1239         ot->name= "Multires Save External";
1240         ot->description= "Save displacements to an external file";
1241         ot->idname= "OBJECT_OT_multires_external_save";
1242
1243         // XXX modifier no longer in context after file browser .. ot->poll= multires_poll;
1244         ot->exec= multires_external_save_exec;
1245         ot->invoke= multires_external_save_invoke;
1246         ot->poll= multires_poll;
1247         
1248         /* flags */
1249         ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
1250
1251         WM_operator_properties_filesel(ot, FOLDERFILE|BTXFILE, FILE_SPECIAL, FILE_SAVE, WM_FILESEL_FILEPATH|WM_FILESEL_RELPATH, FILE_DEFAULTDISPLAY);
1252         edit_modifier_properties(ot);
1253 }
1254
1255 /****************** multires pack operator *********************/
1256
1257 static int multires_external_pack_exec(bContext *C, wmOperator *UNUSED(op))
1258 {
1259         Object *ob = ED_object_active_context(C);
1260         Mesh *me= ob->data;
1261
1262         if(!CustomData_external_test(&me->ldata, CD_MDISPS))
1263                 return OPERATOR_CANCELLED;
1264
1265         // XXX don't remove..
1266         CustomData_external_remove(&me->ldata, &me->id, CD_MDISPS, me->totloop);
1267         
1268         return OPERATOR_FINISHED;
1269 }
1270
1271 void OBJECT_OT_multires_external_pack(wmOperatorType *ot)
1272 {
1273         ot->name= "Multires Pack External";
1274         ot->description= "Pack displacements from an external file";
1275         ot->idname= "OBJECT_OT_multires_external_pack";
1276
1277         ot->poll= multires_poll;
1278         ot->exec= multires_external_pack_exec;
1279         
1280         /* flags */
1281         ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
1282 }
1283
1284 /********************* multires apply base ***********************/
1285 static int multires_base_apply_exec(bContext *C, wmOperator *op)
1286 {
1287         Object *ob = ED_object_active_context(C);
1288         MultiresModifierData *mmd = (MultiresModifierData *)edit_modifier_property_get(op, ob, eModifierType_Multires);
1289         
1290         if (!mmd)
1291                 return OPERATOR_CANCELLED;
1292         
1293         multiresModifier_base_apply(mmd, ob);
1294
1295         DAG_id_tag_update(&ob->id, OB_RECALC_DATA);
1296         WM_event_add_notifier(C, NC_OBJECT|ND_MODIFIER, ob);
1297         
1298         return OPERATOR_FINISHED;
1299 }
1300
1301 static int multires_base_apply_invoke(bContext *C, wmOperator *op, wmEvent *UNUSED(event))
1302 {
1303         if (edit_modifier_invoke_properties(C, op))
1304                 return multires_base_apply_exec(C, op);
1305         else
1306                 return OPERATOR_CANCELLED;
1307 }
1308
1309
1310 void OBJECT_OT_multires_base_apply(wmOperatorType *ot)
1311 {
1312         ot->name= "Multires Apply Base";
1313         ot->description= "Modify the base mesh to conform to the displaced mesh";
1314         ot->idname= "OBJECT_OT_multires_base_apply";
1315
1316         ot->poll= multires_poll;
1317         ot->invoke= multires_base_apply_invoke;
1318         ot->exec= multires_base_apply_exec;
1319         
1320         /* flags */
1321         ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
1322         edit_modifier_properties(ot);
1323 }
1324
1325
1326 /************************ mdef bind operator *********************/
1327
1328 static int meshdeform_poll(bContext *C)
1329 {
1330         return edit_modifier_poll_generic(C, &RNA_MeshDeformModifier, (1<<OB_MESH));
1331 }
1332
1333 static int meshdeform_bind_exec(bContext *C, wmOperator *op)
1334 {
1335         Scene *scene= CTX_data_scene(C);
1336         Object *ob = ED_object_active_context(C);
1337         MeshDeformModifierData *mmd = (MeshDeformModifierData *)edit_modifier_property_get(op, ob, eModifierType_MeshDeform);
1338         
1339         if (!mmd)
1340                 return OPERATOR_CANCELLED;
1341
1342         if(mmd->bindcagecos) {
1343                 MEM_freeN(mmd->bindcagecos);
1344                 if(mmd->dyngrid) MEM_freeN(mmd->dyngrid);
1345                 if(mmd->dyninfluences) MEM_freeN(mmd->dyninfluences);
1346                 if(mmd->bindinfluences) MEM_freeN(mmd->bindinfluences);
1347                 if(mmd->bindoffsets) MEM_freeN(mmd->bindoffsets);
1348                 if(mmd->dynverts) MEM_freeN(mmd->dynverts);
1349                 if(mmd->bindweights) MEM_freeN(mmd->bindweights); /* deprecated */
1350                 if(mmd->bindcos) MEM_freeN(mmd->bindcos); /* deprecated */
1351
1352                 mmd->bindcagecos= NULL;
1353                 mmd->dyngrid= NULL;
1354                 mmd->dyninfluences= NULL;
1355                 mmd->bindoffsets= NULL;
1356                 mmd->dynverts= NULL;
1357                 mmd->bindweights= NULL; /* deprecated */
1358                 mmd->bindcos= NULL; /* deprecated */
1359                 mmd->totvert= 0;
1360                 mmd->totcagevert= 0;
1361                 mmd->totinfluence= 0;
1362                 
1363                 DAG_id_tag_update(&ob->id, OB_RECALC_DATA);
1364                 WM_event_add_notifier(C, NC_OBJECT|ND_MODIFIER, ob);
1365         }
1366         else {
1367                 DerivedMesh *dm;
1368                 int mode= mmd->modifier.mode;
1369
1370                 /* force modifier to run, it will call binding routine */
1371                 mmd->bindfunc= mesh_deform_bind;
1372                 mmd->modifier.mode |= eModifierMode_Realtime;
1373
1374                 if(ob->type == OB_MESH) {
1375                         dm= mesh_create_derived_view(scene, ob, 0);
1376                         dm->release(dm);
1377                 }
1378                 else if(ob->type == OB_LATTICE) {
1379                         lattice_calc_modifiers(scene, ob);
1380                 }
1381                 else if(ob->type==OB_MBALL) {
1382                         makeDispListMBall(scene, ob);
1383                 }
1384                 else if(ELEM3(ob->type, OB_CURVE, OB_SURF, OB_FONT)) {
1385                         makeDispListCurveTypes(scene, ob, 0);
1386                 }
1387
1388                 mmd->bindfunc= NULL;
1389                 mmd->modifier.mode= mode;
1390         }
1391         
1392         return OPERATOR_FINISHED;
1393 }
1394
1395 static int meshdeform_bind_invoke(bContext *C, wmOperator *op, wmEvent *UNUSED(event))
1396 {
1397         if (edit_modifier_invoke_properties(C, op))
1398                 return meshdeform_bind_exec(C, op);
1399         else 
1400                 return OPERATOR_CANCELLED;
1401 }
1402
1403 void OBJECT_OT_meshdeform_bind(wmOperatorType *ot)
1404 {
1405         /* identifiers */
1406         ot->name= "Mesh Deform Bind";
1407         ot->description = "Bind mesh to cage in mesh deform modifier";
1408         ot->idname= "OBJECT_OT_meshdeform_bind";
1409         
1410         /* api callbacks */
1411         ot->poll= meshdeform_poll;
1412         ot->invoke= meshdeform_bind_invoke;
1413         ot->exec= meshdeform_bind_exec;
1414         
1415         /* flags */
1416         ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
1417         edit_modifier_properties(ot);
1418 }
1419
1420 /****************** explode refresh operator *********************/
1421
1422 static int explode_poll(bContext *C)
1423 {
1424         return edit_modifier_poll_generic(C, &RNA_ExplodeModifier, 0);
1425 }
1426
1427 static int explode_refresh_exec(bContext *C, wmOperator *op)
1428 {
1429         Object *ob = ED_object_active_context(C);
1430         ExplodeModifierData *emd = (ExplodeModifierData *)edit_modifier_property_get(op, ob, eModifierType_Explode);
1431         
1432         if (!emd)
1433                 return OPERATOR_CANCELLED;
1434
1435         emd->flag |= eExplodeFlag_CalcFaces;
1436
1437         DAG_id_tag_update(&ob->id, OB_RECALC_DATA);
1438         WM_event_add_notifier(C, NC_OBJECT|ND_MODIFIER, ob);
1439         
1440         return OPERATOR_FINISHED;
1441 }
1442
1443 static int explode_refresh_invoke(bContext *C, wmOperator *op, wmEvent *UNUSED(event))
1444 {
1445         if (edit_modifier_invoke_properties(C, op))
1446                 return explode_refresh_exec(C, op);
1447         else
1448                 return OPERATOR_CANCELLED;
1449 }
1450
1451
1452 void OBJECT_OT_explode_refresh(wmOperatorType *ot)
1453 {
1454         ot->name= "Explode Refresh";
1455         ot->description= "Refresh data in the Explode modifier";
1456         ot->idname= "OBJECT_OT_explode_refresh";
1457
1458         ot->poll= explode_poll;
1459         ot->invoke= explode_refresh_invoke;
1460         ot->exec= explode_refresh_exec;
1461         
1462         /* flags */
1463         ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
1464         edit_modifier_properties(ot);
1465 }
1466
1467
1468 /****************** ocean bake operator *********************/
1469
1470 static int ocean_bake_poll(bContext *C)
1471 {
1472         return edit_modifier_poll_generic(C, &RNA_OceanModifier, 0);
1473 }
1474
1475 /* copied from init_ocean_modifier, MOD_ocean.c */
1476 static void init_ocean_modifier_bake(struct Ocean *oc, struct OceanModifierData *omd)
1477 {
1478         int do_heightfield, do_chop, do_normals, do_jacobian;
1479         
1480         if (!omd || !oc) return; 
1481         
1482         do_heightfield = TRUE;
1483         do_chop = (omd->chop_amount > 0);
1484         do_normals = (omd->flag & MOD_OCEAN_GENERATE_NORMALS);
1485         do_jacobian = (omd->flag & MOD_OCEAN_GENERATE_FOAM);
1486         
1487         BKE_init_ocean(oc, omd->resolution*omd->resolution, omd->resolution*omd->resolution, omd->spatial_size, omd->spatial_size, 
1488                                    omd->wind_velocity, omd->smallest_wave, 1.0, omd->wave_direction, omd->damp, omd->wave_alignment, 
1489                                    omd->depth, omd->time,
1490                                    do_heightfield, do_chop, do_normals, do_jacobian,
1491                                    omd->seed);
1492 }
1493
1494 typedef struct OceanBakeJob {
1495         /* from wmJob */
1496         void *owner;
1497         short *stop, *do_update;
1498         float *progress;
1499         int current_frame;
1500         struct OceanCache *och;
1501         struct Ocean *ocean;
1502         struct OceanModifierData *omd;
1503 } OceanBakeJob;
1504
1505 static void oceanbake_free(void *customdata)
1506 {
1507         OceanBakeJob *oj= customdata;
1508         MEM_freeN(oj);
1509 }
1510
1511 /* called by oceanbake, only to check job 'stop' value */
1512 static int oceanbake_breakjob(void *UNUSED(customdata))
1513 {
1514         //OceanBakeJob *ob= (OceanBakeJob *)customdata;
1515         //return *(ob->stop);
1516         
1517         /* this is not nice yet, need to make the jobs list template better 
1518          * for identifying/acting upon various different jobs */
1519         /* but for now we'll reuse the render break... */
1520         return (G.afbreek);
1521 }
1522
1523 /* called by oceanbake, wmJob sends notifier */
1524 static void oceanbake_update(void *customdata, float progress, int *cancel)
1525 {
1526         OceanBakeJob *oj= customdata;
1527         
1528         if (oceanbake_breakjob(oj))
1529                 *cancel = 1;
1530         
1531         *(oj->do_update)= 1;
1532         *(oj->progress)= progress;
1533 }
1534
1535 static void oceanbake_startjob(void *customdata, short *stop, short *do_update, float *progress)
1536 {
1537         OceanBakeJob *oj= customdata;
1538         
1539         oj->stop= stop;
1540         oj->do_update = do_update;
1541         oj->progress = progress;
1542         
1543         G.afbreek= 0;   /* XXX shared with render - replace with job 'stop' switch */
1544         
1545         BKE_bake_ocean(oj->ocean, oj->och, oceanbake_update, (void *)oj);
1546         
1547         *do_update= 1;
1548         *stop = 0;
1549 }
1550
1551 static void oceanbake_endjob(void *customdata)
1552 {
1553         OceanBakeJob *oj= customdata;
1554         
1555         if (oj->ocean) {
1556                 BKE_free_ocean(oj->ocean);
1557                 oj->ocean = NULL;
1558         }
1559         
1560         oj->omd->oceancache = oj->och;
1561         oj->omd->cached = TRUE;
1562 }
1563
1564 static int ocean_bake_exec(bContext *C, wmOperator *op)
1565 {
1566         Object *ob = ED_object_active_context(C);
1567         OceanModifierData *omd = (OceanModifierData *)edit_modifier_property_get(op, ob, eModifierType_Ocean);
1568         Scene *scene = CTX_data_scene(C);
1569         OceanCache *och;
1570         struct Ocean *ocean;
1571         int f, cfra, i=0;
1572         int free= RNA_boolean_get(op->ptr, "free");
1573         
1574         wmJob *steve;
1575         OceanBakeJob *oj;
1576         
1577         if (!omd)
1578                 return OPERATOR_CANCELLED;
1579         
1580         if (free) {
1581                 omd->refresh |= MOD_OCEAN_REFRESH_CLEAR_CACHE;
1582                 DAG_id_tag_update(&ob->id, OB_RECALC_DATA);
1583                 WM_event_add_notifier(C, NC_OBJECT|ND_MODIFIER, ob);
1584                 return OPERATOR_FINISHED;
1585         }
1586
1587         och = BKE_init_ocean_cache(omd->cachepath, modifier_path_relbase(ob),
1588                                    omd->bakestart, omd->bakeend, omd->wave_scale,
1589                                    omd->chop_amount, omd->foam_coverage, omd->foam_fade, omd->resolution);
1590         
1591         och->time = MEM_mallocN(och->duration*sizeof(float), "foam bake time");
1592         
1593         cfra = scene->r.cfra;
1594         
1595         /* precalculate time variable before baking */
1596         for (f=omd->bakestart; f<=omd->bakeend; f++) {
1597                 /* from physics_fluid.c:
1598                  *
1599                  * XXX: This can't be used due to an anim sys optimization that ignores recalc object animation,
1600                  * leaving it for the depgraph (this ignores object animation such as modifier properties though... :/ )
1601                  * --> BKE_animsys_evaluate_all_animation(G.main, eval_time);
1602                  * This doesn't work with drivers:
1603                  * --> BKE_animsys_evaluate_animdata(&fsDomain->id, fsDomain->adt, eval_time, ADT_RECALC_ALL);
1604                  */
1605                 
1606                 /* Modifying the global scene isn't nice, but we can do it in 
1607                  * this part of the process before a threaded job is created */
1608                 
1609                 //scene->r.cfra = f;
1610                 //ED_update_for_newframe(CTX_data_main(C), scene, CTX_wm_screen(C), 1);
1611                 
1612                 /* ok, this doesn't work with drivers, but is way faster. 
1613                  * let's use this for now and hope nobody wants to drive the time value... */
1614                 BKE_animsys_evaluate_animdata(scene, (ID *)ob, ob->adt, f, ADT_RECALC_ANIM);
1615                 
1616                 och->time[i] = omd->time;
1617                 i++;
1618         }
1619         
1620         /* make a copy of ocean to use for baking - threadsafety */
1621         ocean = BKE_add_ocean();
1622         init_ocean_modifier_bake(ocean, omd);
1623         
1624 #if 0
1625          BKE_bake_ocean(ocean, och);
1626         
1627         omd->oceancache = och;
1628         omd->cached = TRUE;
1629         
1630         scene->r.cfra = cfra;
1631         
1632         DAG_id_tag_update(&ob->id, OB_RECALC_DATA);
1633         WM_event_add_notifier(C, NC_OBJECT|ND_MODIFIER, ob);
1634 #endif
1635         
1636         /* job stuff */
1637         
1638         scene->r.cfra = cfra;
1639         
1640         /* setup job */
1641         steve= WM_jobs_get(CTX_wm_manager(C), CTX_wm_window(C), scene, "Ocean Simulation", WM_JOB_PROGRESS);
1642         oj= MEM_callocN(sizeof(OceanBakeJob), "ocean bake job");
1643         oj->ocean = ocean;
1644         oj->och = och;
1645         oj->omd = omd;
1646         
1647         WM_jobs_customdata(steve, oj, oceanbake_free);
1648         WM_jobs_timer(steve, 0.1, NC_OBJECT|ND_MODIFIER, NC_OBJECT|ND_MODIFIER);
1649         WM_jobs_callbacks(steve, oceanbake_startjob, NULL, NULL, oceanbake_endjob);
1650         
1651         WM_jobs_start(CTX_wm_manager(C), steve);
1652         
1653         
1654         
1655         return OPERATOR_FINISHED;
1656 }
1657
1658 static int ocean_bake_invoke(bContext *C, wmOperator *op, wmEvent *UNUSED(event))
1659 {
1660         if (edit_modifier_invoke_properties(C, op))
1661                 return ocean_bake_exec(C, op);
1662         else
1663                 return OPERATOR_CANCELLED;
1664 }
1665
1666
1667 void OBJECT_OT_ocean_bake(wmOperatorType *ot)
1668 {
1669         ot->name= "Bake Ocean";
1670         ot->description= "Bake an image sequence of ocean data";
1671         ot->idname= "OBJECT_OT_ocean_bake";
1672         
1673         ot->poll= ocean_bake_poll;
1674         ot->invoke= ocean_bake_invoke;
1675         ot->exec= ocean_bake_exec;
1676         
1677         /* flags */
1678         ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
1679         edit_modifier_properties(ot);
1680         
1681         RNA_def_boolean(ot->srna, "free", FALSE, "Free", "Free the bake, rather than generating it");
1682 }
1683