Implement color objects
This commit is contained in:
Daniel Rojas
KV
parent
03518abdcd
commit
1b31a6f044
199
src/wireviz/wv_colors_new.py
Normal file
199
src/wireviz/wv_colors_new.py
Normal file
@ -0,0 +1,199 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
from collections import namedtuple
|
||||
from dataclasses import dataclass, field
|
||||
from enum import Enum
|
||||
from typing import List
|
||||
|
||||
KnownColor = namedtuple("KnownColor", "html code_de full_en full_de")
|
||||
ColorOutputMode = Enum("ColorOutputMode", "EN_LOWER EN_UPPER DE_LOWER DE_UPPER HTML_LOWER HTML_UPPER")
|
||||
|
||||
known_colors = {
|
||||
"BK": KnownColor("#000000", "sw", "black", "schwarz"),
|
||||
"WH": KnownColor("#ffffff", "ws", "white", "weiß"),
|
||||
"GY": KnownColor("#999999", "gr", "grey", "grau"),
|
||||
"PK": KnownColor("#ff66cc", "rs", "pink", "rosa"),
|
||||
"RD": KnownColor("#ff0000", "rt", "red", "rot"),
|
||||
"OG": KnownColor("#ff8000", "or", "orange", "orange"),
|
||||
"YE": KnownColor("#ffff00", "ge", "yellow", "gelb"),
|
||||
"OL": KnownColor("#708000", "ol", "olive green", "olivgrün"),
|
||||
"GN": KnownColor("#00ff00", "gn", "green", "grün"),
|
||||
"TQ": KnownColor("#00ffff", "tk", "turquoise", "türkis"),
|
||||
"LB": KnownColor("#a0dfff", "hb", "light blue", "hellblau"),
|
||||
"BU": KnownColor("#0066ff", "bl", "blue", "blau"),
|
||||
"VT": KnownColor("#8000ff", "vi", "violet", "violett"),
|
||||
"BN": KnownColor("#895956", "br", "brown", "braun"),
|
||||
"BG": KnownColor("#ceb673", "bg", "beige", "beige"),
|
||||
"IV": KnownColor("#f5f0d0", "eb", "ivory", "elfenbein"),
|
||||
"SL": KnownColor("#708090", "si", "slate", "schiefer"),
|
||||
"CU": KnownColor("#d6775e", "ku", "copper", "Kupfer"),
|
||||
"SN": KnownColor("#aaaaaa", "vz", "tin", "verzinkt"),
|
||||
"SR": KnownColor("#84878c", "ag", "silver", "Silber"),
|
||||
"GD": KnownColor("#ffcf80", "au", "gold", "Gold"),
|
||||
}
|
||||
|
||||
color_output_mode = ColorOutputMode.EN_UPPER
|
||||
padding_amount = 1
|
||||
|
||||
def convert_case(inp):
|
||||
if "_LOWER" in color_output_mode.name:
|
||||
return inp.lower()
|
||||
elif "_UPPER" in color_output_mode.name:
|
||||
return inp.upper()
|
||||
else: # currently not used
|
||||
return inp
|
||||
|
||||
@dataclass
|
||||
class SingleColor():
|
||||
_code_en: str
|
||||
_code_de: str
|
||||
_html: str
|
||||
|
||||
@property
|
||||
def code_en(self):
|
||||
return convert_case(self._code_en)
|
||||
# repeat for _code_de, _html
|
||||
|
||||
@property
|
||||
def code_de(self):
|
||||
return convert_case(self._code_de)
|
||||
# repeat for _code_de, _html
|
||||
|
||||
@property
|
||||
def html(self):
|
||||
return convert_case(self._html)
|
||||
# repeat for _code_de, _html
|
||||
|
||||
@property
|
||||
def known(self):
|
||||
return self.code_en.upper() in known_colors.keys()
|
||||
|
||||
def __init__(self, inp):
|
||||
if isinstance(inp, int):
|
||||
hex_str = f"#{inp:06x}"
|
||||
self._html = hex_str
|
||||
self._code_en = hex_str # do not perform reverse lookup
|
||||
self._code_de = hex_str
|
||||
elif inp.upper() in known_colors.keys():
|
||||
inp_upper = inp.upper()
|
||||
self._code_en = inp_upper
|
||||
self._code_de = known_colors[inp_upper].code_de
|
||||
self._html = known_colors[inp_upper].html
|
||||
else: # assume valid HTML color
|
||||
self._html = inp
|
||||
self._code_en = inp
|
||||
self._code_de = inp
|
||||
|
||||
@property
|
||||
def html_padded(self):
|
||||
return ":".join([self.html] * padding_amount)
|
||||
|
||||
def __str__(self):
|
||||
if self.known and "EN_" in color_output_mode.name:
|
||||
return self.code_en
|
||||
elif self.known and "DE_" in color_output_mode.name:
|
||||
return self.code_de
|
||||
else:
|
||||
return self.html
|
||||
|
||||
@dataclass
|
||||
class MultiColor():
|
||||
colors: List[SingleColor] = field(default_factory=list)
|
||||
|
||||
def __init__(self, inp):
|
||||
self.colors = []
|
||||
if isinstance(inp, List): # input is already a list
|
||||
for item in inp:
|
||||
if isinstance(item, SingleColor):
|
||||
self.colors.append(item)
|
||||
else: # string
|
||||
self.colors.append(SingleColor(item))
|
||||
elif isinstance(inp, SingleColor): # single color
|
||||
self.colors = [inp]
|
||||
else: # split input into list
|
||||
if ":" in str(inp):
|
||||
self.colors = [SingleColor(item) for item in inp.split(":")]
|
||||
else:
|
||||
if isinstance(inp, int):
|
||||
self.colors = [SingleColor(inp)]
|
||||
elif len(inp) % 2 == 0:
|
||||
items = [inp[i:i+2] for i in range(0, len(inp), 2)]
|
||||
known = [item.upper() in known_colors.keys() for item in items]
|
||||
if all(known):
|
||||
self.colors = [SingleColor(item) for item in items]
|
||||
else: # assume it's a HTML color name
|
||||
self.colors = [SingleColor(inp)]
|
||||
else: # assume it's a HTML color name
|
||||
self.colors = [SingleColor(inp)]
|
||||
|
||||
|
||||
def __len__(self):
|
||||
return len(self.colors)
|
||||
|
||||
def __str__(self):
|
||||
if "EN_" in color_output_mode.name or "DE_" in color_output_mode.name:
|
||||
joiner = "" if self.all_known else ":"
|
||||
elif "HTML_" in color_output_mode.name:
|
||||
joiner = ":"
|
||||
else:
|
||||
joiner = "???"
|
||||
return joiner.join([str(color) for color in self.colors])
|
||||
|
||||
@property
|
||||
def all_known(self):
|
||||
return all([color.known for color in self.colors])
|
||||
|
||||
@property
|
||||
def html(self):
|
||||
return ":".join([color.html for color in self.colors])
|
||||
|
||||
@property
|
||||
def html_padded(self):
|
||||
if len(self) == 0:
|
||||
out = []
|
||||
elif len(self) == 1:
|
||||
out = [self.colors[0] for i in range(3)]
|
||||
elif len(self) == 2:
|
||||
out = [self.colors[0], self.colors[1], self.colors[0]]
|
||||
elif len(self) == 3:
|
||||
out = self.colors
|
||||
else:
|
||||
raise Exception(f"Padding not supported for len {len(selfq)}")
|
||||
return ":".join([str(color) for color in out])
|
||||
|
||||
COLOR_CODES = {
|
||||
# fmt: off
|
||||
"DIN": [
|
||||
"WH", "BN", "GN", "YE", "GY", "PK", "BU", "RD", "BK", "VT", "GYPK", "RDBU",
|
||||
"WHGN", "BNGN", "WHYE", "YEBN", "WHGY", "GYBN", "WHPK", "PKBN", "WHBU", "BNBU",
|
||||
"WHRD", "BNRD", "WHBK", "BNBK", "GYGN", "YEGY", "PKGN", "YEPK", "GNBU", "YEBU",
|
||||
"GNRD", "YERD", "GNBK", "YEBK", "GYBU", "PKBU", "GYRD", "PKRD", "GYBK", "PKBK",
|
||||
"BUBK", "RDBK", "WHBNBK", "YEGNBK", "GYPKBK", "RDBUBK", "WHGNBK", "BNGNBK",
|
||||
"WHYEBK", "YEBNBK", "WHGYBK", "GYBNBK", "WHPKBK", "PKBNBK", "WHBUBK",
|
||||
"BNBUBK", "WHRDBK", "BNRDBK",
|
||||
],
|
||||
# fmt: on
|
||||
"IEC": ["BN", "RD", "OG", "YE", "GN", "BU", "VT", "GY", "WH", "BK"],
|
||||
"BW": ["BK", "WH"],
|
||||
# 25-pair color code - see also https://en.wikipedia.org/wiki/25-pair_color_code
|
||||
# 5 major colors (WH,RD,BK,YE,VT) combined with 5 minor colors (BU,OG,GN,BN,SL).
|
||||
# Each POTS pair tip (+) had major/minor color, and ring (-) had minor/major color.
|
||||
# fmt: off
|
||||
"TEL": [ # 25x2: Ring and then tip of each pair
|
||||
"BUWH", "WHBU", "OGWH", "WHOG", "GNWH", "WHGN", "BNWH", "WHBN", "SLWH", "WHSL",
|
||||
"BURD", "RDBU", "OGRD", "RDOG", "GNRD", "RDGN", "BNRD", "RDBN", "SLRD", "RDSL",
|
||||
"BUBK", "BKBU", "OGBK", "BKOG", "GNBK", "BKGN", "BNBK", "BKBN", "SLBK", "BKSL",
|
||||
"BUYE", "YEBU", "OGYE", "YEOG", "GNYE", "YEGN", "BNYE", "YEBN", "SLYE", "YESL",
|
||||
"BUVT", "VTBU", "OGVT", "VTOG", "GNVT", "VTGN", "BNVT", "VTBN", "SLVT", "VTSL",
|
||||
],
|
||||
"TELALT": [ # 25x2: Tip and then ring of each pair
|
||||
"WHBU", "BU", "WHOG", "OG", "WHGN", "GN", "WHBN", "BN", "WHSL", "SL",
|
||||
"RDBU", "BURD", "RDOG", "OGRD", "RDGN", "GNRD", "RDBN", "BNRD", "RDSL", "SLRD",
|
||||
"BKBU", "BUBK", "BKOG", "OGBK", "BKGN", "GNBK", "BKBN", "BNBK", "BKSL", "SLBK",
|
||||
"YEBU", "BUYE", "YEOG", "OGYE", "YEGN", "GNYE", "YEBN", "BNYE", "YESL", "SLYE",
|
||||
"VTBU", "BUVT", "VTOG", "OGVT", "VTGN", "GNVT", "VTBN", "BNVT", "VTSL", "SLVT",
|
||||
],
|
||||
# fmt: on
|
||||
"T568A": ["WHGN", "GN", "WHOG", "BU", "WHBU", "OG", "WHBN", "BN"],
|
||||
"T568B": ["WHOG", "OG", "WHGN", "BU", "WHBU", "GN", "WHBN", "BN"],
|
||||
}
|
||||
Loading…
x
Reference in New Issue
Block a user