This repository has been archived on 2023-03-25. You can view files and clone it, but cannot push or open issues or pull requests.
mightyscape-1.1-deprecated/extensions/fablabchemnitz/generate_palette/generate_palette.py

181 lines
5.3 KiB
Python
Raw Normal View History

2021-07-23 02:36:56 +02:00
#! /usr/bin/env python3
import os
import sys
import inkex
def log(text):
inkex.utils.debug(text)
def abort(text):
inkex.errormsg(_(text))
exit()
class GeneratePalette(inkex.EffectExtension):
def add_arguments(self, pars):
pars.add_argument('--name', help='Palette name')
pars.add_argument('--property', help='Color property')
pars.add_argument('--default', type=inkex.Boolean, help='Default grays')
pars.add_argument('--sort', help='Sort type')
pars.add_argument('--replace', type=inkex.Boolean, dest='replace', help='Replace existing')
def get_palettes_path(self):
if sys.platform.startswith('win'):
path = os.path.join(os.environ['APPDATA'], 'inkscape', 'palettes')
else:
path = os.environ.get('XDG_CONFIG_HOME', '~/.config')
path = os.path.join(path, 'inkscape', 'palettes')
return os.path.expanduser(path)
def get_file_path(self):
name = str(self.options.name).replace(' ', '-')
return "%s/%s.gpl" % (self.palettes_path, name)
def get_default_colors(self):
colors = [
" 0 0 0 Black",
" 26 26 26 90% Gray",
" 51 51 51 80% Gray",
" 77 77 77 70% Gray",
"102 102 102 60% Gray",
"128 128 128 50% Gray",
"153 153 153 40% Gray",
"179 179 179 30% Gray",
"204 204 204 20% Gray",
"230 230 230 10% Gray",
"236 236 236 7.5% Gray",
"242 242 242 5% Gray",
"249 249 249 2.5% Gray",
"255 255 255 White"
]
return colors if self.options.default else []
def get_node_prop(self, node, property):
attr = node.attrib.get('style')
style = dict(inkex.Style.parse_str(attr))
return style.get(property, 'none')
def get_node_index(self, item):
node = item[1]
id = node.attrib.get('id')
return self.options.ids.index(id)
def get_node_x(self, item):
node = item[1]
return node.bounding_box().center_x
def get_node_y(self, item):
node = item[1]
return node.bounding_box().center_y
def get_formatted_color(self, color):
rgb = inkex.Color(color).to_rgb()
if self.options.sort == 'hsl':
key = inkex.Color(color).to_hsl()
key = "{:03d}{:03d}{:03d}".format(*key)
else:
key = "{:03d}{:03d}{:03d}".format(*rgb)
rgb = "{:3d} {:3d} {:3d}".format(*rgb)
color = str(color).upper()
name = str(inkex.Color(color).to_named()).upper()
if name != color:
name = "%s (%s)" % (name.capitalize(), color)
return "%s %s %s" % (key, rgb, name)
def get_selected_colors(self):
colors = []
selected = list(self.svg.selected.items())
if self.options.sort == 'y_location':
selected.sort(key=self.get_node_x)
selected.sort(key=self.get_node_y)
elif self.options.sort == 'x_location':
selected.sort(key=self.get_node_y)
selected.sort(key=self.get_node_x)
else:
selected.sort(key=self.get_node_index)
for id, node in selected:
if self.options.property in ['fill', 'both']:
fill = self.get_node_prop(node, 'fill')
if inkex.colors.is_color(fill):
if fill != 'none' and fill not in colors:
colors.append(fill)
if self.options.property in ['stroke', 'both']:
stroke = self.get_node_prop(node, 'stroke')
if inkex.colors.is_color(stroke):
if stroke != 'none' and stroke not in colors:
colors.append(stroke)
colors = list(map(self.get_formatted_color, colors))
if self.options.sort == 'hsl' or self.options.sort == 'rgb':
colors.sort()
return list(map(lambda x : x[11:], colors))
def write_palette(self):
file = open(self.file_path, 'w')
try:
file.write("GIMP Palette\n")
file.write("Name: %s\n" % self.options.name)
file.write("#\n# Generated with Inkscape Generate Palette\n#\n")
for color in self.default_colors:
file.write("%s\n" % color)
for color in self.selected_colors:
if color[:11] not in list(map(lambda x : x[:11], self.default_colors)):
file.write("%s\n" % color)
finally:
file.close()
def effect(self):
self.palettes_path = self.get_palettes_path()
self.file_path = self.get_file_path()
self.default_colors = self.get_default_colors()
self.selected_colors = self.get_selected_colors()
if not self.options.replace and os.path.exists(self.file_path):
abort('Palette already exists!')
if not self.options.name:
abort('Please enter a palette name.')
if len(self.options.ids) < 2:
abort('Please select at least 2 objects.')
if not len(self.selected_colors):
abort('No colors found in selected objects!')
self.write_palette()
if __name__ == '__main__':
GeneratePalette().run()