Source code for warg.colors.label_colors

#!/usr/bin/env python3

__author__ = "Christian Heider Lindbjerg"
__doc__ = r"""

           Created on 20/03/2020
           """

__all__ = ["compute_color_for_labels"]

import logging
from typing import Tuple

from warg import TripleNumber

_logger = logging.getLogger(__name__)




def compute_color_for_labels(label: int, palette: TripleNumber = (2**11 - 1, 2**15 - 1, 2**20 - 1)) -> Tuple:
    """
    Simple function that adds fixed color depending on the class"""
    return (*[int(((label > 0) * p * (label**2 - label + 1)) % 255) for p in palette],)



if __name__ == "__main__":
    for i in range(9):
        _logger.info(compute_color_for_labels(i))