py2ls 0.1.8.2__py3-none-any.whl → 0.1.8.3__py3-none-any.whl

py2ls/plot.py

@@ -983,9 +983,8 @@ def get_color(
     cmap: str = "auto",
     by: str = "start",
     alpha: float = 1.0,
-    output: str = "rgb",
+    output: str = "hue",
 ):
-    # Extract the colormap as a list
     def cmap2hex(cmap_name):
         cmap_ = matplotlib.pyplot.get_cmap(cmap_name)
         colors = [cmap_(i) for i in range(cmap_.N)]
@@ -1018,6 +1017,22 @@ def get_color(
             rgba_values = [hex_to_rgba(hex_color, alpha) for hex_color in hex_colors]
             return rgba_values
 
+    def rgba2hue(rgba_color):
+        if len(rgba_color) == 3:
+            r, g, b = rgba_color
+            a = 1
+        else:
+            r, g, b, a = rgba_color
+        # Convert each component to a scale of 0-255
+        r = int(r * 255)
+        g = int(g * 255)
+        b = int(b * 255)
+        a = int(a * 255)
+        if a < 255:
+            return "#{:02X}{:02X}{:02X}{:02X}".format(r, g, b, a)
+        else:
+            return "#{:02X}{:02X}{:02X}".format(r, g, b)
+
     # Determine color list based on cmap parameter
     if "aut" in cmap:
         colorlist = [
@@ -1042,15 +1057,17 @@ def get_color(
             for i in [int(i) for i in np.linspace(0, len(colorlist) - 1, n)]
         ]
 
-    if output == "rgb":
+    if "rgb" in output.lower():
         return hue2rgb(clist, alpha)
-    elif output == "hue":
-        return clist
+    elif "h" in output.lower():
+        hue_list = []
+        [hue_list.append(rgba2hue(i)) for i in hue2rgb(clist, alpha)]
+        return hue_list
     else:
         raise ValueError("Invalid output type. Choose 'rgb' or 'hue'.")
 
     # # Example usage
-    # colors = get_color(n=5, cmap="viridis", by="linear", alpha=0.5)
+    # colors = get_color(n=5, cmap="viridis", by="linear", alpha=0.5,output='rgb')
     # print(colors)
 
 

{py2ls-0.1.8.2.dist-info → py2ls-0.1.8.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: py2ls
-Version: 0.1.8.2
+Version: 0.1.8.3
 Summary: py(thon)2(too)ls
 Author: Jianfeng
 Author-email: Jianfeng.Liu0413@gmail.com

{py2ls-0.1.8.2.dist-info → py2ls-0.1.8.3.dist-info}/RECORD

@@ -136,12 +136,12 @@ py2ls/export_requirements.py,sha256=x2WgUF0jYKz9GfA1MVKN-MdsM-oQ8yUeC6Ua8oCymio,
 py2ls/freqanalysis.py,sha256=F4218VSPbgL5tnngh6xNCYuNnfR-F_QjECUUxrPYZss,32594
 py2ls/ips.py,sha256=nGGA9Xqa5MzU5X7aE7Mfkk0qrnOkn3DGYulKuh-Ew48,100571
 py2ls/netfinder.py,sha256=MY_0TQY_zaRBZ6wfR4RxNCGrU93HFmDVDRRy1EXl75o,47566
-py2ls/plot.py,sha256=2SVycCCPQG_N7Z6dYDCCMBXDyddogbpIu0SkjXJNI5M,49877
+py2ls/plot.py,sha256=WIAUDd4cRjFTX_Y_SWwsSw9_gIQF_Ye6R_-MvhB9G1k,50437
 py2ls/setuptools-70.1.0-py3-none-any.whl,sha256=2bi3cUVal8ip86s0SOvgspteEF8SKLukECi-EWmFomc,882588
 py2ls/sleep_events_detectors.py,sha256=bQA3HJqv5qnYKJJEIhCyhlDtkXQfIzqksnD0YRXso68,52145
 py2ls/stats.py,sha256=Wd9yCKQ_61QD29WMEgMuEcreFxF91NmlPW65iWT2B5w,39041
 py2ls/translator.py,sha256=bc5FB-wqC4TtQz9gyCP1mE38HqNRJ_pmuRIgKnAlMzM,30581
 py2ls/wb_detector.py,sha256=7y6TmBUj9exCZeIgBAJ_9hwuhkDh1x_-yg4dvNY1_GQ,6284
-py2ls-0.1.8.2.dist-info/METADATA,sha256=ZnmByp2nveooGWTwA9izvfzp7U-KgLT7vsaEtU9LteY,20017
-py2ls-0.1.8.2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-py2ls-0.1.8.2.dist-info/RECORD,,
+py2ls-0.1.8.3.dist-info/METADATA,sha256=-paqLVPq6PniDxsUdZ0UVwJQIry2DrS958ohish7h2o,20017
+py2ls-0.1.8.3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+py2ls-0.1.8.3.dist-info/RECORD,,