palette_sortColors.py

Go to the documentation of this file.

# A script that sorts the colors in the group.
 
# By Wolthera(originally)
 
# SPDX-License-Identifier: CC0-1.0
 
# @package palette_docker
 
from krita import Palette
from builtins import Application
 
 
class sortColors(object):
 
    def __init__(self, name):
        # We want people to select a palette...
        allPalettes = Application.resources("palette")
        self.paletteName = name
        self.currentPalette = Palette(allPalettes[self.paletteName])
        self.sort_all_groups()
 
    def sort_all_groups(self):
        groupNames = self.currentPalette.groupNames()
        for groupName in groupNames:
            self.sort_color_by_name(groupName)
 
    def sort_color_by_name(self, groupName):
        d = {}
        slotCount = self.currentPalette.slotCountGroup(groupName)
        for i in range(slotCount - 1, -1, -1):
            entry = self.currentPalette.entryByIndexFromGroup((i), groupName)
            if not entry.isValid():
                continue
            d[entry.name() + str(i)] = entry
            self.currentPalette.removeEntryFromGroup((i), groupName)
 
        for s in sorted(d):
            self.currentPalette.addEntry(d[s], groupName)
 
    def sort_color_by_id(self, groupName):
        d = {}
        slotCount = self.currentPalette.slotCountGroup(groupName)
        for i in range(slotCount - 1, -1, -1):
            entry = self.currentPalette.entryByIndexFromGroup((i), groupName)
            if not entry.isValid():
                continue
            d[entry.id() + " " + str(i)] = entry
            self.currentPalette.removeEntryFromGroup((i), groupName)
 
        for s in sorted(d):
            self.currentPalette.addEntry(d[s], groupName)
 
    def sort_by_value(self, groupName):
        d = {}
        slotCount = self.currentPalette.slotCountGroup(groupName)
        for i in range(slotCount - 1, -1, -1):
            entry = self.currentPalette.entryByIndexFromGroup((i), groupName)
            if not entry.isValid():
                continue
            color = entry.color()
            color.setColorSpace("RGBA", "U8", "sRGB built-in")
            d[color.components()[0] +
              color.components()[1] +
              color.components()[2]] = entry
            self.currentPalette.removeEntryFromGroup((i), groupName)
 
        for s in sorted(d):
            self.currentPalette.addEntry(d[s], groupName)
 
    def sort_by_hue(self, stepsize, groupName):
        pass
 
    def palette(self):
        return self.currentPalette