← Back to branch summary

~ubuntu-branches/ubuntu/wily/opencollada/wily-proposed

« back to all changes in this revision

Viewing changes to COLLADAMaya/src/COLLADAMayaSetHelper.cpp

  • Committer: Package Import Robot
  • Author(s): Matteo F. Vescovi
  • Date: 2015-05-14 17:23:27 UTC
  • Revision ID: package-import@ubuntu.com-20150514172327-f862u8envms01fra
Tags: upstream-0.1.0~20140703.ddf8f47+dfsg1
Import upstream version 0.1.0~20140703.ddf8f47+dfsg1

Show diffs side-by-side

added added

removed removed

Lines of Context:
COLLADAMaya/src/COLLADAMayaSetHelper.cpp
 
1
/*
 
2
    Copyright (c) 2008-2009 NetAllied Systems GmbH
 
3
 
 
4
        This file is part of COLLADAMaya.
 
5
 
 
6
    Portions of the code are:
 
7
    Copyright (c) 2005-2007 Feeling Software Inc.
 
8
    Copyright (c) 2005-2007 Sony Computer Entertainment America
 
9
    Copyright (c) 2004-2005 Alias Systems Corp.
 
10
        
 
11
    Licensed under the MIT Open Source License, 
 
12
    for details please see LICENSE file or the website
 
13
    http://www.opensource.org/licenses/mit-license.php
 
14
*/
 
15
 
 
16
#include "COLLADAMayaStableHeaders.h"
 
17
#include "COLLADAMayaSetHelper.h"
 
18
 
 
19
#include <maya/MStringArray.h>
 
20
#include <maya/MDagPath.h>
 
21
#include <maya/MItDependencyNodes.h>
 
22
#include <maya/MFnSet.h>
 
23
#include <maya/MFnDagNode.h>
 
24
#include <maya/MSelectionList.h>
 
25
 
 
26
namespace COLLADAMaya
 
27
{
 
28
    std::vector<MObjectHandle> SetHelper::setObjects;
 
29
    SetHelper::SetModes SetHelper::setMode;
 
30
 
 
31
    // ------------------------------------------------------
 
32
    // Unlike Maya's default behavior, we want to consider set membership to be inheritable
 
33
    bool SetHelper::isMemberOfSet ( const MDagPath& dagPath, MFnSet& Set )
 
34
    {
 
35
        if ( Set.isMember ( dagPath ) )
 
36
        {
 
37
            return true;
 
38
        }
 
39
 
 
40
        else
 
41
        {
 
42
            MFnDagNode dagNode ( dagPath );
 
43
            MSelectionList setMembers;
 
44
            Set.getMembers ( setMembers, true );
 
45
 
 
46
            for ( unsigned int i = 0; i < setMembers.length(); ++i )
 
47
            {
 
48
                MObject memberObject;
 
49
 
 
50
                if ( setMembers.getDependNode ( i, memberObject ) )
 
51
                {
 
52
                    if ( dagNode.isChildOf ( memberObject ) )
 
53
                    {
 
54
                        return true;
 
55
                    }
 
56
                }
 
57
            }
 
58
        }
 
59
 
 
60
        return false;
 
61
    }
 
62
 
 
63
    // ------------------------------------------------------
 
64
    bool SetHelper::isExcluded ( const MDagPath& dagPath )
 
65
    {
 
66
        MFnDagNode dagNode ( dagPath );
 
67
 
 
68
        if ( dagNode.name() == "world" )
 
69
            return false;
 
70
 
 
71
        bool bContainedInSet = false;
 
72
 
 
73
        for ( unsigned int i = 0; i < SetHelper::setObjects.size(); ++i )
 
74
        {
 
75
#  if MAYA_API_VERSION < 600
 
76
            MObject o = SetHelper::setObjects[i];
 
77
#  else
 
78
            MObject o = SetHelper::setObjects[i].object();
 
79
#  endif
 
80
            MFnSet currentSet ( o );
 
81
 
 
82
            if ( isMemberOfSet ( dagPath, currentSet ) )
 
83
            {
 
84
                bContainedInSet = true;
 
85
                break;
 
86
            }
 
87
        }
 
88
 
 
89
        if ( setMode == kExcluding )
 
90
        {
 
91
            return bContainedInSet;
 
92
        }
 
93
 
 
94
        else if ( setMode == kIncludeOnly )
 
95
        {
 
96
            return !bContainedInSet;
 
97
        }
 
98
 
 
99
        else
 
100
        {
 
101
            return false;
 
102
        }
 
103
    }
 
104
 
 
105
}
 
 
b'\\ No newline at end of file'