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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
|
#!/usr/bin/env python
# coding=utf-8
#
# Copyright (C) 2009 Aurelio A. Heckert, aurium (a) gmail dot com
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
#
"""
Interpolation of attributes in selected objects or group's children.
"""
import inkex
class InterpAttG(inkex.EffectExtension):
"""
This effect applies a value for any interpolatable attribute for all
elements inside the selected group or for all elements in a multiple selection.
"""
def __init__(self):
super(InterpAttG, self).__init__()
self.arg_parser.add_argument(
"-a", "--att", type=str, dest="att", default="fill",
help="Attribute to be interpolated.")
self.arg_parser.add_argument(
"-o", "--att-other", type=str, dest="att_other",
help="Other attribute (for a limited UI).")
self.arg_parser.add_argument(
"-t", "--att-other-type", type=str, dest="att_other_type",
help="The other attribute type.")
self.arg_parser.add_argument(
"-w", "--att-other-where", type=str, dest="att_other_where",
help="That is a tag attribute or a style attribute?")
self.arg_parser.add_argument(
"-s", "--start-val", type=str, dest="start_val", default="#F00",
help="Initial interpolation value.")
self.arg_parser.add_argument(
"-e", "--end-val", type=str, dest="end_val", default="#00F",
help="End interpolation value.")
self.arg_parser.add_argument(
"-u", "--unit", type=str, dest="unit", default="color",
help="Values unit.")
self.arg_parser.add_argument(
"--zsort", type=inkex.Boolean, dest="zsort", default=True,
help="use z-order instead of selection order")
self.arg_parser.add_argument(
"--tab", type=str, dest="tab",
help="The selected UI-tab when OK was pressed")
def get_color_steps(self, total):
"""Get the color value, returning the start color and a single increment step"""
start_value = inkex.Color(self.options.start_val)
end_value = inkex.Color(self.options.end_val)
color_inc = [
(end_value[v] - start_value[v]) / float(total - 1)
for v in range(3)]
return start_value, color_inc
def get_number_steps(self, total):
"""Get the number value, returning the start float and a single increment step"""
start_value = self.options.start_val.replace(",", ".")
end_value = self.options.end_val.replace(",", ".")
unit = self.options.unit
if unit != 'none':
start_value = self.svg.unittouu(start_value + unit)
end_value = self.svg.unittouu(end_value + unit)
try:
start_value = float(start_value)
end_value = float(end_value)
except ValueError:
inkex.errormsg(
_("Bad values for a number field: {}, {}.".format(start_value, end_value)))
return 0, 0
val_inc = (end_value - start_value) / float(total - 1)
return start_value, val_inc
def get_elements(self):
"""Returns a list of elements to work on"""
if not self.svg.selection:
return []
if len(self.svg.selection) > 1:
# multiple selection
if self.options.zsort:
return self.svg.selection.paint_order()
return self.svg.selected
# must be a group
node = self.svg.selection.filter(inkex.Group).first()
return list(node) or []
def effect(self):
if self.options.att == 'other':
if self.options.att_other is not None:
inte_att = self.options.att_other
else:
inkex.errormsg(_("You selected 'Other'. Please enter an attribute to interpolate."))
return
inte_att_type = self.options.att_other_type
where = self.options.att_other_where
else:
inte_att = self.options.att
inte_att_type = 'float'
if inte_att in ('width', 'height'):
where = 'tag'
elif inte_att in ('scale', 'trans-x', 'trans-y'):
where = 'transform'
elif inte_att == 'opacity':
where = 'style'
elif inte_att in ('fill', 'stroke'):
inte_att_type = 'color'
where = 'style'
collection = self.get_elements()
if not collection:
inkex.errormsg(_('There is no selection to interpolate'))
return False
if inte_att_type == 'color':
cur, inc = self.get_color_steps(len(collection))
else:
cur, inc = self.get_number_steps(len(collection))
for node in collection:
if inte_att_type == 'color':
val = inkex.Color([int(cur[i]) for i in range(3)])
elif inte_att_type == 'float':
val = cur
elif inte_att_type == 'int':
val = int(round(cur))
else:
raise KeyError("Unknown attr type: {}".format(inte_att_type))
if where == 'style':
node.style[inte_att] = str(val)
elif where == 'transform':
if inte_att == 'trans-x':
node.transform.add_translate(val, 0)
elif inte_att == 'trans-y':
node.transform.add_translate(0, val)
elif inte_att == 'scale':
node.transform.add_scale(val)
elif where == 'tag':
node.set(inte_att, str(val))
else:
raise KeyError("Unknown update {}".format(where))
if inte_att_type == 'color':
cur = [cur[i] + inc[i] for i in range(3)]
else:
cur += inc
return True
if __name__ == '__main__':
InterpAttG().run()
|
