Revision: 16715
          
http://projects.blender.org/plugins/scmsvn/viewcvs.php?view=rev&root=bf-blender&revision=16715
Author:   aligorith
Date:     2008-09-24 13:52:31 +0200 (Wed, 24 Sep 2008)

Log Message:
-----------
Made Object-level copy constraints work the same as for bones now too

Modified Paths:
--------------
    trunk/blender/source/blender/blenkernel/intern/modifier.c
    trunk/blender/source/blender/makesdna/DNA_modifier_types.h
    trunk/blender/source/blender/src/buttons_editing.c
    trunk/blender/source/blender/src/drawaction.c
    trunk/blender/source/blender/src/editobject.c

Modified: trunk/blender/source/blender/blenkernel/intern/modifier.c
===================================================================
--- trunk/blender/source/blender/blenkernel/intern/modifier.c   2008-09-24 
07:49:24 UTC (rev 16714)
+++ trunk/blender/source/blender/blenkernel/intern/modifier.c   2008-09-24 
11:52:31 UTC (rev 16715)
@@ -54,6 +54,7 @@
 
 #include "MEM_guardedalloc.h"
 
+#include "DNA_action_types.h"
 #include "DNA_armature_types.h"
 #include "DNA_camera_types.h"
 #include "DNA_cloth_types.h"
@@ -587,6 +588,361 @@
                   return result;
 }
 
+/* Mask */
+
+static void maskModifier_copyData(ModifierData *md, ModifierData *target)
+{
+       MaskModifierData *mmd = (MaskModifierData*) md;
+       MaskModifierData *tmmd = (MaskModifierData*) target;
+       
+       strcpy(tmmd->vgroup, mmd->vgroup);
+}
+
+static CustomDataMask maskModifier_requiredDataMask(ModifierData *md)
+{
+       return (1 << CD_MDEFORMVERT);
+}
+
+static void maskModifier_foreachObjectLink(
+                                             ModifierData *md, Object *ob,
+          void (*walk)(void *userData, Object *ob, Object **obpoin),
+                 void *userData)
+{
+       MaskModifierData *mmd = (MaskModifierData *)md;
+       walk(userData, ob, &mmd->ob_arm);
+}
+
+static void maskModifier_updateDepgraph(ModifierData *md, DagForest *forest,
+                                          Object *ob, DagNode *obNode)
+{
+       MaskModifierData *mmd = (MaskModifierData *)md;
+
+       if (mmd->ob_arm) 
+       {
+               DagNode *armNode = dag_get_node(forest, mmd->ob_arm);
+               
+               dag_add_relation(forest, armNode, obNode,
+                               DAG_RL_DATA_DATA | DAG_RL_OB_DATA, "Mask 
Modifier");
+       }
+}
+
+static DerivedMesh *maskModifier_applyModifier(ModifierData *md, Object *ob,
+               DerivedMesh *derivedData,
+  int useRenderParams, int isFinalCalc)
+{
+       MaskModifierData *mmd= (MaskModifierData *)md;
+       DerivedMesh *dm= derivedData, *result= NULL;
+       GHash *vertHash=NULL, *edgeHash, *faceHash;
+       GHashIterator *hashIter;
+       MDeformVert *dvert= NULL;
+       int numFaces=0, numEdges=0, numVerts=0;
+       int maxVerts, maxEdges, maxFaces;
+       int i;
+       
+       /* Overview of Method:
+        *      1. Get the vertices that are in the vertexgroup of interest 
+        *      2. Filter out unwanted geometry (i.e. not in vertexgroup), by 
populating mappings with new vs old indices
+        *      3. Make a new mesh containing only the mapping data
+        */
+       
+       /* get original number of verts, edges, and faces */
+       maxVerts= dm->getNumVerts(dm);
+       maxEdges= dm->getNumEdges(dm);
+       maxFaces= dm->getNumFaces(dm);
+       
+       /* check if we can just return the original mesh 
+        *      - must have verts and therefore verts assigned to vgroups to do 
anything useful
+        */
+       if ( !(ELEM(mmd->mode, MOD_MASK_MODE_ARM, MOD_MASK_MODE_VGROUP)) ||
+                (maxVerts == 0) || (ob->defbase.first == NULL) )
+       {
+               return derivedData;
+       }
+       
+       /* if mode is to use selected armature bones, aggregate the bone groups 
*/
+       if (mmd->mode == MOD_MASK_MODE_ARM) /* --- using selected bones --- */
+       {
+               GHash *vgroupHash, *boneHash;
+               Object *oba= mmd->ob_arm;
+               bPoseChannel *pchan;
+               bDeformGroup *def;
+               
+               /* check that there is armature object with bones to use, 
otherwise return original mesh */
+               if (ELEM(NULL, mmd->ob_arm, mmd->ob_arm->pose))
+                       return derivedData;             
+               
+               /* hashes for finding mapping of:
+                *      - vgroups to indicies -> vgroupHash  (string, int)
+                *      - bones to vgroup indices -> boneHash (index of vgroup, 
dummy)
+                */
+               vgroupHash= BLI_ghash_new(BLI_ghashutil_strhash, 
BLI_ghashutil_strcmp);
+               boneHash= BLI_ghash_new(BLI_ghashutil_inthash, 
BLI_ghashutil_intcmp);
+               
+               /* build mapping of names of vertex groups to indices */
+               for (i = 0, def = ob->defbase.first; def; def = def->next, i++) 
+                       BLI_ghash_insert(vgroupHash, def->name, 
SET_INT_IN_POINTER(i));
+               
+               /* get selected-posechannel <-> vertexgroup index mapping */
+               for (pchan= oba->pose->chanbase.first; pchan; pchan= 
pchan->next) 
+               {
+                       /* check if bone is selected */
+                       // TODO: include checks for visibility too?
+                       // FIXME: the depsgraph needs extensions to make this 
work in realtime...
+                       if ( (pchan->bone) && (pchan->bone->flag & 
BONE_SELECTED) ) 
+                       {
+                               /* check if hash has group for this bone */
+                               if (BLI_ghash_haskey(vgroupHash, pchan->name)) 
+                               {
+                                       int defgrp_index= 
GET_INT_FROM_POINTER(BLI_ghash_lookup(vgroupHash, pchan->name));
+                                       
+                                       /* add index to hash (store under key 
only) */
+                                       BLI_ghash_insert(boneHash, 
SET_INT_IN_POINTER(defgrp_index), pchan);
+                               }
+                       }
+               }
+               
+               /* if no bones selected, free hashes and return original mesh */
+               if (BLI_ghash_size(boneHash) == 0)
+               {
+                       BLI_ghash_free(vgroupHash, NULL, NULL);
+                       BLI_ghash_free(boneHash, NULL, NULL);
+                       
+                       return derivedData;
+               }
+               
+               /* repeat the previous check, but for dverts */
+               dvert= dm->getVertDataArray(dm, CD_MDEFORMVERT);
+               if (dvert == NULL)
+               {
+                       BLI_ghash_free(vgroupHash, NULL, NULL);
+                       BLI_ghash_free(boneHash, NULL, NULL);
+                       
+                       return derivedData;
+               }
+               
+               /* hashes for quickly providing a mapping from old to new - use 
key=oldindex, value=newindex */
+               vertHash= BLI_ghash_new(BLI_ghashutil_inthash, 
BLI_ghashutil_intcmp);
+               
+               /* add vertices which exist in vertexgroups into vertHash for 
filtering */
+               for (i = 0; i < maxVerts; i++) 
+               {
+                       MDeformWeight *def_weight = NULL;
+                       int j;
+                       
+                       for (j= 0; j < dvert[i].totweight; j++) 
+                       {
+                               if (BLI_ghash_haskey(boneHash, 
SET_INT_IN_POINTER(dvert[i].dw[j].def_nr))) 
+                               {
+                                       def_weight = &dvert[i].dw[j];
+                                       break;
+                               }
+                       }
+                       
+                       /* check if include vert in vertHash */
+                       if (mmd->flag & MOD_MASK_INV) {
+                               /* if this vert is in the vgroup, don't include 
it in vertHash */
+                               if (def_weight) continue;
+                       }
+                       else {
+                               /* if this vert isn't in the vgroup, don't 
include it in vertHash */
+                               if (!def_weight) continue;
+                       }
+                       
+                       /* add to ghash for verts (numVerts acts as counter for 
mapping) */
+                       BLI_ghash_insert(vertHash, SET_INT_IN_POINTER(i), 
SET_INT_IN_POINTER(numVerts));
+                       numVerts++;
+               }
+               
+               /* free temp hashes */
+               BLI_ghash_free(vgroupHash, NULL, NULL);
+               BLI_ghash_free(boneHash, NULL, NULL);
+       }
+       else            /* --- Using Nominated VertexGroup only --- */ 
+       {
+               int defgrp_index = -1;
+               
+               /* get index of vertex group */
+               if (mmd->vgroup[0]) 
+               {
+                       bDeformGroup *def;
+                       
+                       /* find index by comparing names - SLOW... */
+                       for (i = 0, def = ob->defbase.first; def; def = 
def->next, i++) 
+                       {
+                               if (!strcmp(def->name, mmd->vgroup)) 
+                               {
+                                       defgrp_index = i;
+                                       break;
+                               }
+                       }
+               }
+               
+               /* get dverts */
+               if (defgrp_index >= 0)
+                       dvert = dm->getVertDataArray(dm, CD_MDEFORMVERT);
+                       
+               /* if no vgroup (i.e. dverts) found, return the initial mesh */
+               if ((defgrp_index < 0) || (dvert == NULL))
+                       return dm;
+                       
+               /* hashes for quickly providing a mapping from old to new - use 
key=oldindex, value=newindex */
+               vertHash= BLI_ghash_new(BLI_ghashutil_inthash, 
BLI_ghashutil_intcmp);
+               
+               /* add vertices which exist in vertexgroup into ghash for 
filtering */
+               for (i = 0; i < maxVerts; i++) 
+               {
+                       MDeformWeight *def_weight = NULL;
+                       int j;
+                       
+                       for (j= 0; j < dvert[i].totweight; j++) 
+                       {
+                               if (dvert[i].dw[j].def_nr == defgrp_index) 
+                               {
+                                       def_weight = &dvert[i].dw[j];
+                                       break;
+                               }
+                       }
+                       
+                       /* check if include vert in vertHash */
+                       if (mmd->flag & MOD_MASK_INV) {
+                               /* if this vert is in the vgroup, don't include 
it in vertHash */
+                               if (def_weight) continue;
+                       }
+                       else {
+                               /* if this vert isn't in the vgroup, don't 
include it in vertHash */
+                               if (!def_weight) continue;
+                       }
+                       
+                       /* add to ghash for verts (numVerts acts as counter for 
mapping) */
+                       BLI_ghash_insert(vertHash, SET_INT_IN_POINTER(i), 
SET_INT_IN_POINTER(numVerts));
+                       numVerts++;
+               }
+       }
+       
+       /* hashes for quickly providing a mapping from old to new - use 
key=oldindex, value=newindex */
+       edgeHash= BLI_ghash_new(BLI_ghashutil_inthash, BLI_ghashutil_intcmp);
+       faceHash= BLI_ghash_new(BLI_ghashutil_inthash, BLI_ghashutil_intcmp);
+       
+       /* loop over edges and faces, and do the same thing to 
+        * ensure that they only reference existing verts 
+        */
+       for (i = 0; i < maxEdges; i++) 
+       {
+               MEdge me;
+               dm->getEdge(dm, i, &me);
+               
+               /* only add if both verts will be in new mesh */
+               if ( BLI_ghash_haskey(vertHash, SET_INT_IN_POINTER(me.v1)) &&
+                        BLI_ghash_haskey(vertHash, SET_INT_IN_POINTER(me.v2)) )
+               {
+                       BLI_ghash_insert(edgeHash, SET_INT_IN_POINTER(i), 
SET_INT_IN_POINTER(numEdges));
+                       numEdges++;
+               }
+       }
+       for (i = 0; i < maxFaces; i++) 
+       {
+               MFace mf;
+               dm->getFace(dm, i, &mf);
+               
+               /* all verts must be available */
+               if ( BLI_ghash_haskey(vertHash, SET_INT_IN_POINTER(mf.v1)) &&
+                        BLI_ghash_haskey(vertHash, SET_INT_IN_POINTER(mf.v2)) 
&&
+                        BLI_ghash_haskey(vertHash, SET_INT_IN_POINTER(mf.v3)) 
&&
+                       (mf.v4==0 || BLI_ghash_haskey(vertHash, 
SET_INT_IN_POINTER(mf.v4))) )
+               {
+                       BLI_ghash_insert(faceHash, SET_INT_IN_POINTER(i), 
SET_INT_IN_POINTER(numFaces));
+                       numFaces++;
+               }
+       }
+       
+       
+       /* now we know the number of verts, edges and faces, 
+        * we can create the new (reduced) mesh
+        */
+       result = CDDM_from_template(dm, numVerts, numEdges, numFaces);
+       
+       
+       /* using ghash-iterators, map data into new mesh */
+               /* vertices */
+       for ( hashIter = BLI_ghashIterator_new(vertHash);
+                 !BLI_ghashIterator_isDone(hashIter);
+                 BLI_ghashIterator_step(hashIter) ) 
+       {
+               MVert source;
+               MVert *dest;
+               int oldIndex = 
GET_INT_FROM_POINTER(BLI_ghashIterator_getKey(hashIter));
+               int newIndex = 
GET_INT_FROM_POINTER(BLI_ghashIterator_getValue(hashIter));
+               
+               dm->getVert(dm, oldIndex, &source);
+               dest = CDDM_get_vert(result, newIndex);
+               
+               DM_copy_vert_data(dm, result, oldIndex, newIndex, 1);
+               *dest = source;
+       }
+       BLI_ghashIterator_free(hashIter);
+               
+               /* edges */
+       for ( hashIter = BLI_ghashIterator_new(edgeHash);
+                 !BLI_ghashIterator_isDone(hashIter);
+                 BLI_ghashIterator_step(hashIter) ) 
+       {
+               MEdge source;
+               MEdge *dest;
+               int oldIndex = 
GET_INT_FROM_POINTER(BLI_ghashIterator_getKey(hashIter));
+               int newIndex = 
GET_INT_FROM_POINTER(BLI_ghashIterator_getValue(hashIter));
+               
+               dm->getEdge(dm, oldIndex, &source);
+               dest = CDDM_get_edge(result, newIndex);
+               
+               source.v1 = GET_INT_FROM_POINTER(BLI_ghash_lookup(vertHash, 
SET_INT_IN_POINTER(source.v1)));
+               source.v2 = GET_INT_FROM_POINTER(BLI_ghash_lookup(vertHash, 
SET_INT_IN_POINTER(source.v2)));
+               
+               DM_copy_edge_data(dm, result, oldIndex, newIndex, 1);
+               *dest = source;
+       }
+       BLI_ghashIterator_free(hashIter);
+       
+               /* faces */
+       for ( hashIter = BLI_ghashIterator_new(faceHash);

@@ Diff output truncated at 10240 characters. @@

_______________________________________________
Bf-blender-cvs mailing list
[email protected]
http://lists.blender.org/mailman/listinfo/bf-blender-cvs

Reply via email to