-
Notifications
You must be signed in to change notification settings - Fork 128
Expand file tree
/
Copy pathRemovePrimitiveVariables.py
More file actions
79 lines (67 loc) · 2.87 KB
/
Copy pathRemovePrimitiveVariables.py
File metadata and controls
79 lines (67 loc) · 2.87 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
##########################################################################
#
# Copyright (c) 2007-2010, Image Engine Design Inc. All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are
# met:
#
# * Redistributions of source code must retain the above copyright
# notice, this list of conditions and the following disclaimer.
#
# * Redistributions in binary form must reproduce the above copyright
# notice, this list of conditions and the following disclaimer in the
# documentation and/or other materials provided with the distribution.
#
# * Neither the name of Image Engine Design nor the names of any
# other contributors to this software may be used to endorse or
# promote products derived from this software without specific prior
# written permission.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS
# IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
# THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
# EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
# PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
# PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
# LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
# NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
# SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
#
##########################################################################
from fnmatch import fnmatchcase
import IECore
import IECoreScene
class RemovePrimitiveVariables( IECoreScene.PrimitiveOp ) :
def __init__( self ) :
IECoreScene.PrimitiveOp.__init__( self, "Removes variables from primitives" )
self.parameters().addParameters(
[
IECore.StringParameter(
name = "mode",
description = """This chooses whether or not the names parameter specifies the names of
variables to keep or the names of variables to remove.""",
defaultValue = "remove",
presets = (
( "keep", "keep" ),
( "remove", "remove" )
),
presetsOnly = True
),
IECore.StringVectorParameter(
name = "names",
description = "The names of variables. These can include * or ? characters to match many names.",
defaultValue = IECore.StringVectorData()
)
]
)
def modifyPrimitive( self, primitive, args ) :
keep = args["mode"].value == "keep"
for key in primitive.keys() :
for n in args["names"] :
m = fnmatchcase( key, n )
if (m and not keep) or (not m and keep) :
del primitive[key]
IECore.registerRunTimeTyped( RemovePrimitiveVariables )