initial implementation using StyleStrings

Author: t-bltg

Project.toml

@@ -19,6 +19,7 @@ Printf = "de0858da-6303-5e67-8744-51eddeeeb8d7"
 SparseArrays = "2f01184e-e22b-5df5-ae63-d93ebab69eaf"
 StaticArrays = "90137ffa-7385-5640-81b9-e52037218182"
 StatsBase = "2913bbd2-ae8a-5f71-8c99-4fb6c76f3a91"
+StyledStrings = "f489334b-da3d-4c2e-b8f0-e476e12c162b"
 
 [weakdeps]
 FileIO = "5789e2e9-d7fb-5bc7-8068-2c6fae9b9549"
@@ -53,6 +54,7 @@ Printf = "1"
 SparseArrays = "1"
 StaticArrays = "1"
 StatsBase = "0.33 - 0.34"
+StyledStrings = "1"
 Term = "2"
 Unitful = "1"
 julia = "1.10"

src/UnicodePlots.jl

@@ -1,6 +1,7 @@
 module UnicodePlots
 
 using PrecompileTools
+using StyledStrings
 using LinearAlgebra
 using StaticArrays
 using ColorTypes

src/common.jl

@@ -454,31 +454,44 @@ crayon_color(color::ColorType) =
         Crayons.ANSIColor(color - THRESHOLD, Crayons.COLORS_256)
     end
 
-function print_crayons(io, c, args...)
-    if CRAYONS_FAST[]
-        if Crayons.anyactive(c)  # bypass crayons checks (_have_color, _force_color)
-            print(io, Crayons.CSI)
-            Crayons._print(io, c)
-            print(io, Crayons.END_ANSI, args..., CRAYONS_RESET)
-        else
-            print(io, args...)
-        end
-    else
-        print(io, c, args..., CRAYONS_RESET)
+
+ss_color(::Union{Missing,Nothing}) = nothing
+ss_color(color::ColorType) =
+    if color ≡ INVALID_COLOR
+        nothing
+    elseif color < THRESHOLD  # 24bit
+        StyledStrings.SimpleColor(red(color), grn(color), blu(color))
+    else  # 8bit
+        StyledStrings.Legacy.legacy_color(Int(color - THRESHOLD))
     end
-end
 
-print_color(io::IO, color::Crayon, args...) = print_crayons(io, color, args...)
+# function print_crayons(io, c, args...)
+#     if CRAYONS_FAST[]
+#         if Crayons.anyactive(c)  # bypass crayons checks (_have_color, _force_color)
+#             print(io, Crayons.CSI)
+#             Crayons._print(io, c)
+#             print(io, Crayons.END_ANSI, args..., CRAYONS_RESET)
+#         else
+#             print(io, args...)
+#         end
+#     else
+#         print(io, c, args..., CRAYONS_RESET)
+#     end
+# end
+
+# print_color(io::IO, color::Crayon, args...) = print_crayons(io, color, args...)
 print_color(io::IO, color::UserColorType, args...) =
     print_color(io, ansi_color(color), args...)
 
 function print_color(io::IO, color::ColorType, args...; bgcol = missing)
     if get(io, :color, false)
-        print_crayons(
-            io,
-            Crayon(crayon_color(color), crayon_color(bgcol), CRAYONS_EMPTY_STYLES...),
-            args...,
-        )
+        face = StyledStrings.Face(; foreground = ss_color(color), background = ss_color(bgcol))
+        print(io, StyledStrings.face!(Base.annotatedstring(args...), face))
+        # print_crayons(
+        #     io,
+        #     Crayon(crayon_color(color), crayon_color(bgcol), CRAYONS_EMPTY_STYLES...),
+        #     args...,
+        # )
     else
         print(io, args...)
     end

src/show.jl

@@ -83,7 +83,7 @@ function print_title(
     post_pad = blank^(max(0, p_width - length(pre_pad) - length(title)))
     print_nocol(io, post_pad, right_pad)
     (
-        count(string('\n'), title) + 1,  # NOTE: string(...) for compat with 1.6
+        count('\n', title) + 1,
         length(strip(left_pad * pre_pad * title * post_pad * right_pad, '\n')),
     )
 end
@@ -145,7 +145,7 @@ end
 Base.show(io::IO, p::Plot) = _show(io, print, print_color, p)
 
 function _show(end_io::IO, print_nocol, print_color, p::Plot)
-    buf = PipeBuffer()  # buffering, for performance
+    buf = Base.AnnotatedIOBuffer()  # buffering, for performance
     io_color = get(end_io, :color, false)
     io = IOContext(buf, :color => io_color, :displaysize => displaysize(end_io))
 
@@ -385,7 +385,7 @@ function _show(end_io::IO, print_nocol, print_color, p::Plot)
     end
 
     # delayed print (buffering)
-    print_nocol(end_io, read(buf, String))
+    print(end_io, read(seekstart(buf), Base.AnnotatedString))
 
     # return the approximate image size
     (