tikzplot42 0.2.8__py3-none-any.whl

tikzplot/colorbar.py

@@ -0,0 +1,261 @@
+import numpy as _np
+from .elements import Graph
+from .axes3d import Axes3
+from .config import TikzConfig
+
+class _Colorbar:
+    def __init__(self, im=None, **kwargs):
+        self._axis = None
+        self._cmap = None
+        self._lower = 0
+        self._upper = 1
+        self._ticks = None
+        self._tick_labels = None
+        self._label = None
+        self._width = 0.3
+        self._horizontal = False
+        self._rel_len = 1
+        self._divs = 0          # 0 for contiuous
+
+        if im is not None:
+            self._axis, self._cmap, self._lower, self._upper = im
+
+        if "axis" in kwargs:
+            self._axis = kwargs["axis"]
+        if "cmap" in kwargs:
+            self._cmap = kwargs["cmap"]
+        if "lower" in kwargs:
+            self._lower = kwargs["lower"]
+        if "upper" in kwargs:
+            self._upper = kwargs["upper"]
+        if "ticks" in kwargs:
+            self._ticks = kwargs["ticks"]
+        if "tick_labels" in kwargs:
+            self._tick_labels = kwargs["tick_labels"]
+        if "label" in kwargs:
+            self._label = kwargs["label"]
+        if "width" in kwargs:
+            self._width = kwargs["width"]
+        if "horizontal" in kwargs and kwargs["horizontal"]:
+            self._horizontal = True
+        if "rel_len" in kwargs:
+            self._rel_len = kwargs["rel_len"]
+        if "divisions" in kwargs:
+            self._divs = kwargs["divisions"]
+        if self._axis:
+            self._axis._show_colorbar(str(self))
+
+    _discrete = {'Pastel1':9, 'Pastel2':8, 'Paired':12, 'Accent':8, 'Dark2':8, 'Set1': 9, 'Set2':8, 'Set3':12, 'tab10':10, 'tab20':20, 'tab20b':20, 'tab20c':20}
+    _dictionary = {
+        'viridis':       [(0.267, 0.0049, 0.3294), (0.2297, 0.3224, 0.5457), (0.1276, 0.5669, 0.5506), (0.3692, 0.7889, 0.3829), (0.9932, 0.9062, 0.1439)],
+        'plasma':        [(0.0504, 0.0298, 0.528), (0.4949, 0.012, 0.6579), (0.7982, 0.2802, 0.4695), (0.9734, 0.5858, 0.2515), (0.94, 0.9752, 0.1313)],
+        'inferno':       [(0.0015, 0.0005, 0.0139), (0.3415, 0.0623, 0.4294), (0.7357, 0.2159, 0.3302), (0.9784, 0.5579, 0.0349), (0.9884, 0.9984, 0.6449)],
+        'magma':         [(0.0015, 0.0005, 0.0139), (0.3167, 0.0717, 0.4854), (0.7164, 0.215, 0.4753), (0.9867, 0.5356, 0.3822), (0.9871, 0.9914, 0.7495)],
+        'cividis':       [(0.0, 0.1351, 0.3048), (0.2637, 0.3078, 0.4228), (0.4887, 0.4853, 0.471), (0.7365, 0.6806, 0.424), (0.9957, 0.9093, 0.2178)],
+        
+        'Greys':         [(1.0, 1.0, 1.0), (0.0, 0.0, 0.0)],
+        'Purples':       [(0.9882, 0.9843, 0.9922), (0.2471, 0.0, 0.4902)],
+        'Blues':         [(0.9686, 0.9843, 1.0), (0.0314, 0.1882, 0.4196)],
+        'Greens':        [(0.9686, 0.9882, 0.9608), (0.0, 0.2667, 0.1059)],
+        'Oranges':       [(1.0, 0.9608, 0.9216), (0.498, 0.1529, 0.0157)],
+        'Reds':  [(1.0, 0.9608, 0.9412), (0.4039, 0.0, 0.051)],
+        'YlOrBr':        [(1.0, 1.0, 0.898), (0.4, 0.1451, 0.0235)],
+        'YlOrRd':        [(1.0, 1.0, 0.8), (0.502, 0.0, 0.149)],
+        'OrRd':  [(1.0, 0.9686, 0.9255), (0.498, 0.0, 0.0)],
+        'PuRd':  [(0.9686, 0.9569, 0.9765), (0.4039, 0.0, 0.1216)],
+        'RdPu':  [(1.0, 0.9686, 0.9529), (0.2863, 0.0, 0.4157)],
+        'BuPu':  [(0.9686, 0.9882, 0.9922), (0.302, 0.0, 0.2941)],
+        'GnBu':  [(0.9686, 0.9882, 0.9412), (0.0314, 0.251, 0.5059)],
+        'PuBu':  [(1.0, 0.9686, 0.9843), (0.0078, 0.2196, 0.3451)],
+        'YlGnBu':        [(1.0, 1.0, 0.851), (0.0314, 0.1137, 0.3451)],
+        'PuBuGn':        [(1.0, 0.9686, 0.9843), (0.0039, 0.2745, 0.2118)],
+        'BuGn':  [(0.9686, 0.9882, 0.9922), (0.0, 0.2667, 0.1059)],
+        'YlGn':  [(1.0, 1.0, 0.898), (0.0, 0.2706, 0.1608)],
+
+        'binary':        [(1.0, 1.0, 1.0), (0.498, 0.498, 0.498), (0.0, 0.0, 0.0)],
+        'gist_yarg':     [(1.0, 1.0, 1.0), (0.498, 0.498, 0.498), (0.0, 0.0, 0.0)],
+        'gist_gray':     [(0.0, 0.0, 0.0), (0.502, 0.502, 0.502), (1.0, 1.0, 1.0)],
+        'gray':  [(0.0, 0.0, 0.0), (0.502, 0.502, 0.502), (1.0, 1.0, 1.0)],
+        'bone':  [(0.0, 0.0, 0.0), (0.4392, 0.4841, 0.5642), (1.0, 1.0, 1.0)],
+        'pink':  [(0.1178, 0.0, 0.0), (0.8173, 0.674, 0.5785), (1.0, 1.0, 1.0)],
+        'spring':        [(1.0, 0.0, 1.0), (1.0, 0.502, 0.498), (1.0, 1.0, 0.0)],
+        'summer':        [(0.0, 0.5, 0.4), (0.502, 0.751, 0.4), (1.0, 1.0, 0.4)],
+        'autumn':        [(1.0, 0.0, 0.0), (1.0, 0.502, 0.0), (1.0, 1.0, 0.0)],
+        'winter':        [(0.0, 0.0, 1.0), (0.0, 0.502, 0.749), (0.0, 1.0, 0.5)],
+        'cool':  [(0.0, 1.0, 1.0), (0.502, 0.498, 1.0), (1.0, 0.0, 1.0)],
+        'Wistia':        [(0.8941, 1.0, 0.4784), (1.0, 0.7403, 0.0), (0.9882, 0.498, 0.0)],
+        'hot':   [(0.0416, 0.0, 0.0), (1.0, 0.3593, 0.0), (1.0, 1.0, 1.0)],
+        'afmhot':        [(0.0, 0.0, 0.0), (1.0, 0.5039, 0.0039), (1.0, 1.0, 1.0)],
+        'gist_heat':     [(0.0, 0.0, 0.0), (0.7529, 0.0039, 0.0), (1.0, 1.0, 1.0)],
+        'copper':        [(0.0, 0.0, 0.0), (0.6201, 0.3921, 0.2497), (1.0, 0.7812, 0.4975)],
+
+        'PiYG':  [(0.5569, 0.0039, 0.3216), (0.9673, 0.9685, 0.9656), (0.1529, 0.3922, 0.098)],
+        'PRGn':  [(0.251, 0.0, 0.2941), (0.9663, 0.9681, 0.9659), (0.0, 0.2667, 0.1059)],
+        'BrBG':  [(0.3294, 0.1882, 0.0196), (0.9572, 0.9599, 0.9596), (0.0, 0.2353, 0.1882)],
+        'PuOr':  [(0.498, 0.2314, 0.0314), (0.9662, 0.9664, 0.9677), (0.1765, 0.0, 0.2941)],
+        'RdGy':  [(0.4039, 0.0, 0.1216), (0.9976, 0.9976, 0.9976), (0.102, 0.102, 0.102)],
+        'RdBu':  [(0.4039, 0.0, 0.1216), (0.9657, 0.9672, 0.9681), (0.0196, 0.1882, 0.3804)],
+        'RdYlBu':        [(0.6471, 0.0, 0.149), (0.9976, 0.9991, 0.7534), (0.1922, 0.2118, 0.5843)],
+        'RdYlGn':        [(0.6471, 0.0, 0.149), (0.9971, 0.9988, 0.745), (0.0, 0.4078, 0.2157)],
+        'Spectral':      [(0.6196, 0.0039, 0.2588), (0.9981, 0.9992, 0.746), (0.3686, 0.3098, 0.6353)],
+        'coolwarm':      [(0.2298, 0.2987, 0.7537), (0.8674, 0.8644, 0.8626), (0.7057, 0.0156, 0.1502)],
+        'bwr':   [(0.0, 0.0, 1.0), (1.0, 0.9961, 0.9961), (1.0, 0.0, 0.0)],
+        'seismic':       [(0.0, 0.0, 0.3), (1.0, 0.9922, 0.9922), (0.5, 0.0, 0.0)],
+
+        'twilight':      [(0.8858, 0.85, 0.888), (0.5383, 0.678, 0.7711), (0.3732, 0.3814, 0.706), (0.2867, 0.0824, 0.3933), (0.3154, 0.0794, 0.2681), (0.6457, 0.2616, 0.3125), (0.7905, 0.5983, 0.4862), (0.8857, 0.85, 0.8857)],
+        'twilight_shifted':      [(0.1874, 0.0771, 0.2162), (0.36, 0.2069, 0.6029), (0.4114, 0.5368, 0.7467), (0.7387, 0.796, 0.8214), (0.8399, 0.7604, 0.7137), (0.7388, 0.4205, 0.3491), (0.4981, 0.1357, 0.3141), (0.1849, 0.0794, 0.2131)],
+        'hsv':   [(1.0, 0.0, 0.0), (1.0, 0.8338, 0.0), (0.3092, 1.0, 0.0), (0.0, 1.0, 0.5246), (0.0, 0.6184, 1.0), (0.2154, 0.0, 1.0), (1.0, 0.0, 0.9276), (1.0, 0.0, 0.0938)],
+
+        'flag':  [(1.0, 0.0, 0.0), (1.0, 0.9998, 1.0), (0.0, 0.037, 1.0), (0.0, 0.0, 0.0), (0.9897, 0.0, 0.0), (1.0, 0.9957, 0.9791), (0.0316, 0.1107, 1.0), (0.0, 0.0, 0.0425), (0.9464, 0.0, 0.0), (1.0, 0.9862, 0.9352), (0.0761, 0.1837, 1.0), (0.0, 0.0, 0.0877), (0.9007, 0.0, 0.0), (1.0, 0.9713, 0.8889), (0.123, 0.2558, 1.0), (0.0, 0.0, 0.135), (1.0, 0.0923, 0.0209), (0.9897, 0.9973, 1.0), (0.0, 0.0, 1.0), (0.0, 0.0, 0.0), (1.0, 0.0185, 0.0), (1.0, 1.0, 1.0), (0.0, 0.0185, 1.0), (0.0, 0.0, 0.0), (1.0, 0.0, 0.0), (1.0, 0.9973, 0.9897), (0.0209, 0.0923, 1.0), (0.0, 0.0, 0.0316), (0.9575, 0.0, 0.0), (1.0, 0.9891, 0.9464), (0.0648, 0.1656, 1.0), (0.0, 0.0, 0.0761), (1.0, 0.1837, 0.0761), (0.9352, 0.9862, 1.0), (0.0, 0.0, 0.9464), (0.0425, 0.0, 0.0), (1.0, 0.1107, 0.0316), (0.9791, 0.9957, 1.0), (0.0, 0.0, 0.9897), (0.0, 0.0, 0.0), (1.0, 0.037, 0.0), (1.0, 0.9998, 1.0), (0.0, 0.0, 1.0), (0.0, 0.0, 0.0), (1.0, 0.0, 0.0), (1.0, 0.9985, 1.0), (0.0103, 0.0739, 1.0), (0.0, 0.0, 0.0209), (1.0, 0.2737, 0.135), (0.877, 0.9667, 1.0), (0.0, 0.0, 0.8889), (0.0993, 0.0, 0.0), (1.0, 0.2019, 0.0877), (0.9239, 0.983, 1.0), (0.0, 0.0, 0.9352), (0.0536, 0.0, 0.0), (1.0, 0.129, 0.0425), (0.9684, 0.9939, 1.0), (0.0, 0.0, 0.9791), (0.0103, 0.0, 0.0), (1.0, 0.0554, 0.0), (1.0, 0.9993, 1.0), (0.0, 0.0, 1.0), (0.0, 0.0, 0.0)],
+        'prism':         [(1.0, 0.0, 0.0), (1.0, 0.5116, 0.0), (0.889, 1.0, 0.0), (0.1678, 0.887, 0.0), (0.0, 0.1863, 0.9124), (0.4142, 0.0, 0.9965), (1.0, 0.0, 0.1136), (1.0, 0.4354, 0.0), (0.9619, 1.0, 0.0), (0.228, 0.9359, 0.0), (0.0, 0.2631, 0.8441), (0.3427, 0.0, 1.0), (1.0, 0.0, 0.2261), (1.0, 0.3582, 0.0), (1.0, 0.987, 0.0), (0.293, 0.9784, 0.0), (0.0, 0.1493, 0.9423), (0.4501, 0.0, 0.9717), (1.0, 0.0, 0.0582), (1.0, 0.4728, 0.0), (0.9267, 1.0, 0.0), (0.198, 0.9129, 0.0), (0.0, 0.2255, 0.8786), (0.3772, 0.0, 1.0), (1.0, 0.0, 0.1714), (1.0, 0.396, 0.0), (0.9981, 1.0, 0.0), (0.2607, 0.9585, 0.0), (0.0, 0.3027, 0.8056), (0.3075, 0.0, 1.0), (1.0, 0.0, 0.2827), (1.0, 0.3185, 0.0), (0.8908, 1.0, 0.0), (0.1692, 0.8883, 0.0), (0.0, 0.1881, 0.9109), (0.4125, 0.0, 0.9976), (1.0, 0.0, 0.1163), (1.0, 0.4336, 0.0), (0.9636, 1.0, 0.0), (0.2295, 0.937, 0.0), (0.0, 0.2649, 0.8424), (0.3411, 0.0, 1.0), (1.0, 0.0, 0.2287), (1.0, 0.3563, 0.0), (1.0, 0.9861, 0.0), (0.2946, 0.9793, 0.0), (0.0, 0.3425, 0.7648), (0.2732, 0.0, 1.0), (1.0, 0.0, 0.061), (1.0, 0.471, 0.0), (0.9284, 1.0, 0.0), (0.1994, 0.914, 0.0), (0.0, 0.2273, 0.877), (0.3755, 0.0, 1.0), (1.0, 0.0, 0.1741), (1.0, 0.3941, 0.0), (0.9997, 1.0, 0.0), (0.2623, 0.9595, 0.0), (0.0, 0.3046, 0.8037), (0.3059, 0.0, 1.0), (1.0, 0.0, 0.2854), (1.0, 0.3166, 0.0), (1.0, 0.966, 0.0), (0.3295, 0.9983, 0.0)],
+        'gist_earth':    [(0.0, 0.0, 0.0), (0.0732, 0.1893, 0.4691), (0.1464, 0.4028, 0.489), (0.2122, 0.5313, 0.4381), (0.264, 0.5945, 0.3036), (0.4849, 0.6552, 0.3242), (0.6659, 0.7034, 0.3589), (0.752, 0.6459, 0.3949), (0.8694, 0.7323, 0.6555), (0.9922, 0.9843, 0.9843)],
+        'terrain':       [(0.2, 0.2, 0.6), (0.0536, 0.4928, 0.8928), (0.0, 0.7392, 0.5824), (0.3333, 0.8667, 0.4667), (0.7725, 0.9545, 0.5545), (0.8863, 0.8544, 0.5386), (0.6667, 0.5733, 0.42), (0.5608, 0.4378, 0.4115), (0.7804, 0.7189, 0.7057), (1.0, 1.0, 1.0)],
+        'gist_stern':    [(0.0, 0.0, 0.0), (0.7255, 0.1098, 0.2196), (0.1784, 0.2196, 0.4392), (0.3333, 0.3333, 0.6667), (0.4431, 0.4431, 0.8863), (0.5569, 0.5569, 0.758), (0.6667, 0.6667, 0.2908), (0.7804, 0.7804, 0.1713), (0.8902, 0.8902, 0.5856), (1.0, 1.0, 1.0)],
+        'gnuplot':       [(0.0, 0.0, 0.0), (0.3314, 0.0013, 0.6365), (0.4686, 0.0106, 0.9818), (0.5774, 0.037, 0.866), (0.6657, 0.087, 0.3497), (0.7462, 0.1727, 0.0), (0.8165, 0.2963, 0.0), (0.8834, 0.4753, 0.0), (0.9435, 0.7054, 0.0), (1.0, 1.0, 0.0)],
+        'gnuplot2':      [(0.0, 0.0, 0.0), (0.0, 0.0, 0.4392), (0.0, 0.0, 0.8784), (0.2604, 0.0, 1.0), (0.6036, 0.0463, 0.9537), (0.9589, 0.2737, 0.7263), (1.0, 0.4933, 0.5067), (1.0, 0.7208, 0.2792), (1.0, 0.9404, 0.0596), (1.0, 1.0, 1.0)],
+        'CMRmap':        [(0.0, 0.0, 0.0), (0.1318, 0.1318, 0.4392), (0.2635, 0.15, 0.6892), (0.5, 0.1833, 0.5833), (0.818, 0.2273, 0.3092), (0.9545, 0.3637, 0.0818), (0.9, 0.5833, 0.0333), (0.9, 0.7865, 0.1973), (0.9122, 0.9122, 0.5608), (1.0, 1.0, 1.0)],
+        'cubehelix':     [(0.0, 0.0, 0.0), (0.1036, 0.095, 0.2062), (0.0826, 0.2728, 0.3077), (0.17, 0.4368, 0.2237), (0.4588, 0.4806, 0.1997), (0.7577, 0.477, 0.4378), (0.83, 0.5632, 0.7763), (0.7639, 0.7573, 0.9494), (0.8107, 0.9218, 0.9374), (1.0, 1.0, 1.0)],
+        'brg':   [(0.0, 0.0, 1.0), (0.2196, 0.0, 0.7804), (0.4392, 0.0, 0.5608), (0.6667, 0.0, 0.3333), (0.8863, 0.0, 0.1137), (0.8863, 0.1137, 0.0), (0.6667, 0.3333, 0.0), (0.4392, 0.5608, 0.0), (0.2196, 0.7804, 0.0), (0.0, 1.0, 0.0)],
+        'gist_rainbow':  [(1.0, 0.0, 0.16), (1.0, 0.4314, 0.0), (0.9751, 1.0, 0.0), (0.3604, 1.0, 0.0), (0.0, 1.0, 0.2319), (0.0, 1.0, 0.8433), (0.0, 0.5616, 1.0), (0.0565, 0.0, 1.0), (0.6532, 0.0, 1.0), (1.0, 0.0, 0.75)],
+        'rainbow':       [(0.5, 0.0, 1.0), (0.2804, 0.3382, 0.9852), (0.0608, 0.6365, 0.9411), (0.1667, 0.866, 0.866), (0.3863, 0.9841, 0.7674), (0.6137, 0.9841, 0.6412), (0.8333, 0.866, 0.5), (1.0, 0.6365, 0.3382), (1.0, 0.3382, 0.1716), (1.0, 0.0, 0.0)],
+        'jet':   [(0.0, 0.0, 0.5), (0.0, 0.0, 0.9991), (0.0, 0.3784, 1.0), (0.0, 0.8333, 1.0), (0.3004, 1.0, 0.6673), (0.6673, 1.0, 0.3004), (1.0, 0.9012, 0.0), (1.0, 0.48, 0.0), (0.9991, 0.0733, 0.0), (0.5, 0.0, 0.0)],
+        'turbo':         [(0.19, 0.0718, 0.2322), (0.2733, 0.3801, 0.8404), (0.2138, 0.6589, 0.9796), (0.1034, 0.896, 0.715), (0.4432, 0.9955, 0.3734), (0.7856, 0.9358, 0.2034), (0.98, 0.73, 0.2216), (0.9619, 0.4109, 0.0931), (0.7912, 0.1637, 0.0139), (0.4796, 0.0158, 0.0106)],
+        'nipy_spectral':         [(0.0, 0.0, 0.0), (0.4287, 0.0, 0.6131), (0.0, 0.183, 0.8667), (0.0, 0.6445, 0.7334), (0.0, 0.6092, 0.0732), (0.0, 0.885, 0.0), (0.8, 0.9778, 0.0), (1.0, 0.6784, 0.0), (0.8928, 0.0, 0.0), (0.8, 0.8, 0.8)],
+        'gist_ncar':     [(0.0, 0.0, 0.502), (0.0, 0.0551, 1.0), (0.0, 0.9982, 0.897), (0.1499, 0.9408, 0.0), (0.5897, 1.0, 0.1093), (1.0, 0.913, 0.0), (1.0, 0.4404, 0.022), (1.0, 0.0, 0.7644), (0.8772, 0.4538, 0.9435), (0.9961, 0.9725, 0.9961)],
+
+        'Pastel1':       [(0.9843, 0.7059, 0.6824), (0.702, 0.8039, 0.8902), (0.8, 0.9216, 0.7725), (0.8706, 0.7961, 0.8941), (0.9961, 0.851, 0.651), (1.0, 1.0, 0.8), (0.898, 0.8471, 0.7412), (0.9922, 0.8549, 0.9255), (0.949, 0.949, 0.949)],
+        'Pastel2':       [(0.702, 0.8863, 0.8039), (0.9922, 0.8039, 0.6745), (0.7961, 0.8353, 0.9098), (0.9569, 0.7922, 0.8941), (0.902, 0.9608, 0.7882), (1.0, 0.949, 0.6824), (0.9451, 0.8863, 0.8), (0.8, 0.8, 0.8)],
+        'Paired':        [(0.651, 0.8078, 0.8902), (0.1216, 0.4706, 0.7059), (0.698, 0.8745, 0.5412), (0.2, 0.6275, 0.1725), (0.9843, 0.6039, 0.6), (0.8902, 0.102, 0.1098), (0.9922, 0.749, 0.4353), (1.0, 0.498, 0.0), (0.7922, 0.698, 0.8392), (0.4157, 0.2392, 0.6039), (1.0, 1.0, 0.6), (0.6941, 0.349, 0.1569)],
+        'Accent':        [(0.498, 0.7882, 0.498), (0.7451, 0.6824, 0.8314), (0.9922, 0.7529, 0.5255), (1.0, 1.0, 0.6), (0.2196, 0.4235, 0.6902), (0.9412, 0.0078, 0.498), (0.749, 0.3569, 0.0902), (0.4, 0.4, 0.4)],
+        'Dark2':         [(0.1059, 0.6196, 0.4667), (0.851, 0.3725, 0.0078), (0.4588, 0.4392, 0.702), (0.9059, 0.1608, 0.5412), (0.4, 0.651, 0.1176), (0.902, 0.6706, 0.0078), (0.651, 0.4627, 0.1137), (0.4, 0.4, 0.4)],
+        'Set1':  [(0.8941, 0.102, 0.1098), (0.2157, 0.4941, 0.7216), (0.302, 0.6863, 0.2902), (0.5961, 0.3059, 0.6392), (1.0, 0.498, 0.0), (1.0, 1.0, 0.2), (0.651, 0.3373, 0.1569), (0.9686, 0.5059, 0.749), (0.6, 0.6, 0.6)],
+        'Set2':  [(0.4, 0.7608, 0.6471), (0.9882, 0.5529, 0.3843), (0.5529, 0.6275, 0.7961), (0.9059, 0.5412, 0.7647), (0.651, 0.8471, 0.3294), (1.0, 0.851, 0.1843), (0.898, 0.7686, 0.5804), (0.702, 0.702, 0.702)],
+        'Set3':  [(0.5529, 0.8275, 0.7804), (1.0, 1.0, 0.702), (0.7451, 0.7294, 0.8549), (0.9843, 0.502, 0.4471), (0.502, 0.6941, 0.8275), (0.9922, 0.7059, 0.3843), (0.702, 0.8706, 0.4118), (0.9882, 0.8039, 0.898), (0.851, 0.851, 0.851), (0.7373, 0.502, 0.7412), (0.8, 0.9216, 0.7725), (1.0, 0.9294, 0.4353)],
+        'tab10':         [(0.1216, 0.4667, 0.7059), (1.0, 0.498, 0.0549), (0.1725, 0.6275, 0.1725), (0.8392, 0.1529, 0.1569), (0.5804, 0.4039, 0.7412), (0.549, 0.3373, 0.2941), (0.8902, 0.4667, 0.7608), (0.498, 0.498, 0.498), (0.7373, 0.7412, 0.1333), (0.0902, 0.7451, 0.8118)],
+        'tab20':         [(0.1216, 0.4667, 0.7059), (0.6824, 0.7804, 0.9098), (1.0, 0.498, 0.0549), (1.0, 0.7333, 0.4706), (0.1725, 0.6275, 0.1725), (0.5961, 0.8745, 0.5412), (0.8392, 0.1529, 0.1569), (1.0, 0.5961, 0.5882), (0.5804, 0.4039, 0.7412), (0.7725, 0.6902, 0.8353), (0.549, 0.3373, 0.2941), (0.7686, 0.6118, 0.5804), (0.8902, 0.4667, 0.7608), (0.9686, 0.7137, 0.8235), (0.498, 0.498, 0.498), (0.7804, 0.7804, 0.7804), (0.7373, 0.7412, 0.1333), (0.8588, 0.8588, 0.5529), (0.0902, 0.7451, 0.8118), (0.6196, 0.8549, 0.898)],
+        'tab20b':        [(0.2235, 0.2314, 0.4745), (0.3216, 0.3294, 0.6392), (0.4196, 0.4314, 0.8118), (0.6118, 0.6196, 0.8706), (0.3882, 0.4745, 0.2235), (0.549, 0.6353, 0.3216), (0.7098, 0.8118, 0.4196), (0.8078, 0.8588, 0.6118), (0.549, 0.4275, 0.1922), (0.7412, 0.6196, 0.2235), (0.9059, 0.7294, 0.3216), (0.9059, 0.7961, 0.5804), (0.5176, 0.2353, 0.2235), (0.6784, 0.2863, 0.2902), (0.8392, 0.3804, 0.4196), (0.9059, 0.5882, 0.6118), (0.4824, 0.2549, 0.451), (0.6471, 0.3176, 0.5804), (0.8078, 0.4275, 0.7412), (0.8706, 0.6196, 0.8392)],
+        'tab20c':        [(0.1922, 0.5098, 0.7412), (0.4196, 0.6824, 0.8392), (0.6196, 0.7922, 0.8824), (0.7765, 0.8588, 0.9373), (0.902, 0.3333, 0.051), (0.9922, 0.5529, 0.2353), (0.9922, 0.6824, 0.4196), (0.9922, 0.8157, 0.6353), (0.1922, 0.6392, 0.3294), (0.4549, 0.7686, 0.4627), (0.6314, 0.851, 0.6078), (0.7804, 0.9137, 0.7529), (0.4588, 0.4196, 0.6941), (0.6196, 0.6039, 0.7843), (0.7373, 0.7412, 0.8627), (0.8549, 0.8549, 0.9216), (0.3882, 0.3882, 0.3882), (0.5882, 0.5882, 0.5882), (0.7412, 0.7412, 0.7412), (0.851, 0.851, 0.851)]
+
+        
+}
+    
+    def _get_samples(self, colors, n):
+        if n == len(colors):
+            return colors
+        
+        colors_arr = _np.array(colors)
+        x_orig = _np.linspace(0, 1, len(colors))
+        x_new = _np.linspace(0, 1, n)
+        
+        r = _np.interp(x_new, x_orig, colors_arr[:, 0])
+        g = _np.interp(x_new, x_orig, colors_arr[:, 1])
+        b = _np.interp(x_new, x_orig, colors_arr[:, 2])
+        
+        return list(zip(r, g, b))
+    
+    def _generate_tex_colormap(self, cmap_name):
+        if isinstance(cmap_name, str):
+            rev = False
+            if cmap_name.endswith("_r"):
+                rev = True
+                cmap_name = cmap_name.removesuffix("_r")
+            if cmap_name not in self._dictionary:
+                raise Exception(f"% Error: {cmap_name} not found in _dictionary")
+            colors = self._dictionary[cmap_name]
+            if rev: colors=list(reversed(colors))
+        elif isinstance(cmap_name, list):
+            colors = cmap_name
+        else:
+            raise Exception(f"% Error: cmap_name must be a string or a list of colors")
+        output = [r"{urgb}{"]
+        if isinstance(cmap_name, str) and cmap_name in self._discrete:
+            n = self._discrete[cmap_name]
+            self._divs = n
+        elif self._divs > 0:
+            colors = self._get_samples(colors, self._divs)
+        if self._divs == 0:
+            for r, g, b in colors:
+                output.append(f"    rgb=({r:.4f}, {g:.4f}, {b:.4f})")
+        else:
+            lims = _np.linspace(0,1, self._divs+1, endpoint=True)
+            lims[-1] += 0.001
+            for i in range(self._divs):
+                r,g,b = colors[i]
+                output.append(f"    rgb({lims[i]})=({r:.4f}, {g:.4f}, {b:.4f}) rgb({lims[i+1]-0.001})=({r:.4f}, {g:.4f}, {b:.4f})")
+        output.append("}")        
+        return "\n".join(output)
+    
+    def __str__(self):
+        tridim = isinstance(self._axis, Axes3)
+        lines = []
+        lines.append(f"colormap={self._generate_tex_colormap(self._cmap)},")
+        lines.append(f"colorbar{' horizontal' if self._horizontal else ''},")
+        lines.append(r"colorbar style={")
+        if TikzConfig.USE_DECIMAL_COMMA:
+            lines.append("/pgf/number format/use comma,")
+        if self._horizontal:
+            lines.append(f"height={self._width}cm,")
+            lines.append(f"anchor=north,")
+            lines.append(f"at={{(p{self._axis._get_index()}.south)}},")
+            lines.append(f"yshift=-{TikzConfig.CBAR_Y_OFFSET * (not tridim) + TikzConfig.CBAR3_Z_OFFSET * tridim}cm,")
+        else:
+            lines.append(f"width={self._width}cm,")
+            lines.append(f"anchor=west,")
+            lines.append(f"at={{(p{self._axis._get_index()}.east)}},")
+            lines.append(f"xshift={TikzConfig.CBAR_X_OFFSET * (not tridim) + + TikzConfig.CBAR3_H_OFFSET * tridim}cm,")
+        if self._rel_len:
+            lines.append(f"{'width' if self._horizontal else 'height'}={self._rel_len}*\\pgfkeysvalueof{{/pgfplots/parent axis {'width' if self._horizontal else 'height'}}},")
+        if self._label:
+            lines.append(f"title={{{self._label}}},")
+            offset = 4 - 2 * (self._ticks == [] or self._tick_labels == [])
+            if self._horizontal:
+                lines.append(f"title style={{at={{(0.5, -{offset})}}, anchor=base}},")
+            else:
+                lines.append(f"title style={{at={{({offset},0.5)}}, anchor=base, yshift=-7pt, rotate=-90}},")
+        if self._ticks is not None:
+            lines.append(f"{'x' if self._horizontal else 'y'}tick={{{','.join(str(a) for a in self._ticks)}}},")
+            if self._tick_labels and len(self._tick_labels) == len(self._ticks):
+                lines.append(f"{'x' if self._horizontal else 'y'}ticklabels={{{','.join(str(a) for a in self._tick_labels)}}},")
+        if self._tick_labels == []:
+            lines.append(f"{'x' if self._horizontal else 'y'}ticklabels=\\empty,")
+            if self._divs > 0:
+                lines.append(f"{'x' if self._horizontal else 'y'}tick =\\empty,")
+        lines.append(r"},")
+        lines.append(f"point meta min={{{self._lower}}},")
+        lines.append(f"point meta max={{{self._upper}}}")
+
+        return "\n".join(lines)
+    
+    def color(self, value):
+        norm_val = (value - self._lower) / (self._upper - self._lower)
+        norm_val = _np.clip(norm_val, 0, 1)
+        is_reversed = False
+        if isinstance(self._cmap, str):
+            cmap_name = self._cmap
+            if cmap_name.endswith("_r"):
+                is_reversed = True
+                cmap_name = cmap_name.removesuffix("_r")
+            colors = self._dictionary[cmap_name]
+        elif isinstance(self._cmap, list):
+            colors = self._cmap
+        else:
+            raise Exception(f"% Error: cmap_name must be a string or a list of colors")
+        if is_reversed:
+            colors = list(reversed(colors))
+        n_colors = len(colors)
+        if n_colors == 1:
+            return colors[0]
+        elif self._divs == 0:
+            float_idx = norm_val * (n_colors - 1)
+            idx_low = int(_np.floor(float_idx))
+            idx_high = int(_np.ceil(float_idx))
+            weight = float_idx - idx_low
+            c_low = _np.array(colors[idx_low])
+            c_high = _np.array(colors[idx_high])
+            rgb = (1 - weight) * c_low + weight * c_high
+        else:
+            idx = int(norm_val * self._divs)
+            rgb = self._get_samples(colors, self._divs)[idx]
+        return tuple(rgb)
+    
+Colorbar = _Colorbar

tikzplot/colorbar.pyi

@@ -0,0 +1,128 @@
+from typing import Any, Iterable, Optional, Sequence, Tuple
+import numpy.typing as npt
+from .axes import Axes
+from .axes3d import Axes3
+
+ArrayLike = npt.ArrayLike
+
+
+class Colorbar:
+    """
+    PGFPlots-compatible colorbar generator.
+
+    This class converts a colormap definition into TikZ/PGFPlots
+    colorbar configuration and provides utilities for color mapping.
+
+    Typically constructed via:
+
+        Colorbar(im, **kwargs)
+
+    where `im` is a tuple describing an image:
+        (axis, cmap, lower, upper)
+    """
+
+    def __init__(
+        self,
+        im: Optional[Tuple[Axes, str, float, float]] = ...,
+        *,
+        axis: Optional[Axes | Axes3] = ...,
+        cmap: Optional[str] = ...,
+        lower: Optional[float] = ...,
+        upper: Optional[float] = ...,
+        ticks: Optional[Sequence[ float]] = ...,
+        tick_labels: Optional[Sequence[str | float]] = ...,
+        label: Optional[str] = ...,
+        width: float = ...,
+        horizontal: bool = ...,
+        rel_len: float = ...,
+        divisions: int = ...
+    ) -> None:
+        """
+        Parameters
+        ----------
+        im : imshow() return: tuple(axis, cmap, lower, upper), optional
+            Tuple describing the associated image:
+            - axis : parent axis object
+            - cmap : name of the colormap
+            - lower : minimum value of the color scale
+            - upper : maximum value of the color scale
+
+        axis: reference to axis, optional
+            Axis to which the color bar is plotted.
+
+        cmap : str, optional
+            Name of the colormap to use.
+            Supports reversed colormaps via suffix "_r".
+
+        lower : float, optional
+            Minimum value of the color scale.
+
+        upper : float, optional
+            Maximum value of the color scale.
+
+        ticks : sequence of str, optional
+            Positions of ticks on the colorbar.
+            Must be provided as strings for direct PGFPlots usage.
+
+        tick_labels : sequence of str, optional
+            Labels corresponding to `ticks`.
+            Must have the same length as `ticks`.
+
+            Special cases:
+            - empty list `[]` → disables tick labels
+
+        label : str, optional
+            Title of the colorbar.
+
+        width : float, default = 0.3
+            Thickness of the colorbar (in cm).
+
+            - vertical colorbar → width
+            - horizontal colorbar → height
+
+        horizontal : bool, default = False
+            If True, renders a horizontal colorbar.
+            Otherwise, a vertical colorbar is used.
+
+        rel_len : float, default = 1
+            Relative length of the colorbar with respect to the parent axis.
+
+            Internally mapped to:
+                width or height = rel_len * parent axis dimension
+        divisions: int, optional
+            May be used with continuous colormaps do discretize number of colors, 0 for continuous.
+        """
+        ...
+
+    def _generate_tex_colormap(self, cmap_name: str) -> str:
+        """
+        Generate a PGFPlots-compatible colormap definition.
+
+        Parameters
+        ----------
+        cmap_name : str
+            Name of the colormap.
+            Supports reversed maps using suffix "_r".
+
+        Returns
+        -------
+        str
+            TikZ colormap definition string.
+        """
+        ...
+
+    def color(self, value: float) -> Tuple[float, float, float]:
+        """
+        Map a scalar value to an RGB color using the current colormap.
+
+        Parameters
+        ----------
+        value : float
+            Value within the range [lower, upper].
+
+        Returns
+        -------
+        tuple of float
+            RGB color as (r, g, b), each in [0, 1].
+        """
+        ...

tikzplot/colors.py

@@ -0,0 +1,58 @@
+from .config import TikzConfig
+
+_COLOR_MAP = {'b':'blue', 'g':'teal', 'r':'red', 'c':'cyan', 'm':'magenta', 'y':'yellow', 'k':'black', 'w':'white', "orange":"orange", "green": "green", "cyan":"cyan", "peru": "brown", "lime": "lime", "gray": "gray", "magenta": "magetna", "purple": "violet"}
+_DEFAULT_CYCLE = ['#1f77b4','#ff7f0e','#2ca02c','#d62728','#9467bd','#8c564b','#e377c2','#7f7f7f','#bcbd22','#17becf']
+
+def _tex_color(input):
+    def color_string(r,g,b):
+        if TikzConfig.USE_XCOLOR:
+            return (r,g,b)
+        else:
+            return f"rgb:red,{r};green,{g};blue,{b}"
+    
+    def hex_to_rgb(hex):
+        if hex[0] == "#":
+            hex = hex[1:]
+        hex = hex.upper()
+        rgb = []
+        for i in (0, 2, 4):
+            decimal = int(hex[i:i+2], 16) / 255
+            rgb.append(decimal)  
+        return color_string(rgb[0],rgb[1],rgb[2])
+       
+    if isinstance(input, tuple):
+        opacity = False
+        if len(input) == 1:
+            input = input[0]
+        elif len(input) == 2:
+            opacity= input[1]
+            input = input[0]
+        else:
+            r,g,b = input[:3]
+            if len(input) == 4:
+                opacity = input[3]                    
+            return color_string(r,g,b), opacity
+    s = str(input)
+    if s[0] == "#":
+        opacity = False
+        if len(s) == 4:
+            hex = s[1] * 2 + s[2] * 2 + s[3] * 2
+        else:
+            hex = s[:8]
+            if len(s) > 8:
+                opacity = int(s[8:]) / 100
+        return hex_to_rgb(hex), opacity
+    if s.isdigit():
+        i = float(s)
+        return color_string(i,i,i), False
+    if s.startswith("C") and s[1:].isdigit():
+        index = int(s[1:]) % len(_DEFAULT_CYCLE)
+        return hex_to_rgb(_DEFAULT_CYCLE[index]), False
+    if s.lower() == "none":
+        return color_string(0,0,0), 0
+    if s in _COLOR_MAP.keys():
+        return _COLOR_MAP[s], False
+    if s in _COLOR_MAP.values():
+        return s, False
+    print(f"Unrecognized color {input}")
+    return None, False

tikzplot/colors.pyi

@@ -0,0 +1 @@
+from .config import TikzConfig as TikzConfig

tikzplot/config.py

@@ -0,0 +1,101 @@
+import json
+from pathlib import Path
+
+class _TikzConfig:
+
+    def __init__(self):
+
+        self.USE_DECIMAL_COMMA: bool = False
+
+        self.LEGEND_REL_X: float = 0.03
+        self.LEGEND_REL_Y: float = 0.03
+
+        self.LEFT_PADDING: float = 0.1
+        self.RIGHT_PADDING: float = 0.1
+        self.TOP_PADDING: float = 0.1
+        self.BOTTOM_PADDING: float = 0.1
+        self.Y_LABEL_PADDING: float = 0.6
+        self.SEC_Y_PADDING: float = 0.8
+        self.SEC_Y_LABEL_PADDING: float = 0.6
+        self.TITLE_PADDING: float = 0.6
+        self.X_LABEL_PADDING: float = 0.6
+        self.XTICK_PADDING: float = 0.7
+        self.YTICK_PADDING: float = 0.7
+
+        self.DEFAULT_WIDTH: float = 12
+        self.DEFAULT_HEIGHT: float = 10
+
+        self.SHARED_AXIS_REL_MARGIN: float = 0.03
+
+        self.SAVE_DATAPOINTS: bool = True
+        self.DATAPOINTS_DIR: str = "datapoints"  # ignored if SAVE_DATAPOINTS == False
+        self.UPDATE_DATA_ONLY: bool = False       # ignored if SAVE_DATAPOINTS == False
+        self.UPDATE_STYLE_ONLY: bool = False      # ignored if SAVE_DATAPOINTS == False
+        self.SHOW_SAVENAME: str = "showplot"
+        self.IMSHOW_SAVENAME: str = "imshow"
+
+        self.REDUCE_NUM_POINTS: bool = True
+        self.REDUCE_METHOD: int = 1              # 0: remove based on index; 1: remove based on plot distance; 2: remove based on curvature (good for line plots)
+        self.MAX_POINTS_PER_FIGURE: int = 10000
+        self.MAX_POINTS_PER_ELEMENT: int = 1000
+
+        self.STANDALONE: bool = False
+        self.TIKZ_COMPAT: float = 1.18 # only for standalone, used to change the setting in preambule, but does not guarantee the compatibility of the generated code wito older versions of pgfplots.
+        self.USE_XCOLOR: bool = True
+        self.USE_GROUPPLOTS: bool = True
+        self.SCHOOL_AXIS: bool = False
+        self.SCHOOL_AXIS_LABEL_MARGIN: float = 0.1
+
+        self.CBAR_X_OFFSET: float = 0.05
+        self.CBAR_Y_OFFSET: float = 0.2
+        self.CBAR_X_MARGIN: float = 0.5
+        self.CBAR_Y_MARGIN: float = 0.5
+
+        self.CBAR3_Z_OFFSET: float = 0.2   # for 3D vertical
+        self.CBAR3_H_OFFSET: float = 0.2    # for 3D horizontal
+
+        self.DEFAULT_3D_ROLL: float = 0
+        self.DEFAULT_3D_AZIM: float = 0
+        self.DEFAULT_3D_ELEV: float = 0
+
+        self._config_file: Path = Path.home() / ".tikz_userconf.json"
+
+        self._load_user_config()
+
+    def modifyParam(self, **kwargs):
+        for k, v in kwargs.items():
+            if not hasattr(self, k):
+                raise ValueError(f"Unknown TikzConfig parameter: {k}")
+            setattr(self, k, v)
+
+    def setPermanent(self, **kwargs):
+
+        self.modifyParam(**kwargs)
+
+        data = self._read_file()
+        data.update(kwargs)
+
+        with open(self._config_file, "w") as f:
+            json.dump(data, f, indent=2)
+
+    def _load_user_config(self):
+
+        if not self._config_file.exists():
+            return
+
+        with open(self._config_file) as f:
+            data = json.load(f)
+
+        for k, v in data.items():
+            if hasattr(self, k):
+                setattr(self, k, v)
+
+    def _read_file(self):
+
+        if not self._config_file.exists():
+            return {}
+
+        with open(self._config_file) as f:
+            return json.load(f)
+
+TikzConfig = _TikzConfig()

tikzplot/config.pyi

@@ -0,0 +1,95 @@
+# config.pyi
+
+from typing import Any, Dict
+from pathlib import Path
+
+from typing import TypedDict, Unpack, Any, ClassVar, Protocol
+
+class _ConfigParams(TypedDict, total=False):
+    USE_DECIMAL_COMMA: bool
+
+    LEGEND_REL_X: float
+    LEGEND_REL_Y: float
+
+    LEFT_PADDING: float
+    RIGHT_PADDING: float
+    TOP_PADDING: float
+    BOTTOM_PADDING: float
+    Y_LABEL_PADDING: float
+    SEC_Y_PADDING: float
+    SEC_Y_LABEL_PADDING: float
+    TITLE_PADDING: float
+    X_LABEL_PADDING: float
+
+    DEFAULT_WIDTH: float
+    DEFAULT_HEIGHT: float
+
+    SHARED_AXIS_REL_MARGIN: float
+
+    SAVE_DATAPOINTS: bool
+    DATAPOINTS_DIR: str
+    UPDATE_DATA_ONLY: bool
+    UPDATE_STYLE_ONLY: bool
+
+    SHOW_SAVENAME: str
+    IMSHOW_SAVENAME: str
+
+    REDUCE_NUM_POINTS: bool
+    REDUCE_METHOD: int
+    MAX_POINTS_PER_FIGURE: int
+    MAX_POINTS_PER_ELEMENT: int
+
+    STANDALONE: bool
+    TIKZ_COMPAT: float
+    USE_XCOLOR: bool
+    USE_GROUPPLOTS: bool
+    SCHOOL_AXIS: bool
+    SCHOOL_AXIS_LABEL_MARGIN: float
+
+    CBAR_X_OFFSET: float
+    CBAR_Y_OFFSET: float
+    CBAR_X_MARGIN: float
+    CBAR_Y_MARGIN: float
+
+    CBAR3_Z_OFFSET: float
+    CBAR3_H_OFFSET: float
+
+    DEFAULT_3D_ROLL: float
+    DEFAULT_3D_AZIM: float
+    DEFAULT_3D_ELEV: float
+
+    
+
+
+class TikzConfig:
+
+    def modifyParam(self, **kwargs: Unpack[_ConfigParams]) -> None:
+        """
+        Modify configuration parameters temporarily.
+
+        Parameters
+        ----------
+        kwargs:
+            Key-value pairs of configuration parameters to update.
+
+        Raises
+        ------
+        ValueError
+            If an unknown parameter is provided.
+        """
+        ...
+
+    def setPermanent(self, **kwargs: Unpack[_ConfigParams]) -> None:
+        """
+        Modify configuration parameters permanently.
+
+        Updates the in-memory configuration and writes changes
+        to the user configuration file (~/.tikz_userconf.json).
+
+        Parameters
+        ----------
+        kwargs:
+            Key-value pairs of configuration parameters to update.
+        """
+        ...
+