mantao/Plots.jl
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

speed up no-color

Browse Source
This commit is contained in:
t-bltg 2021-12-30 16:29:36 +01:00
parent 523af85af5
commit e0beaaac28
1 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
src/backends/unicodeplots.jl
View File

@ -205,11 +205,12 @@ function Base.show(io::IO, plt::Plot{UnicodePlotsBackend})
i < n && println(io) i < n && println(io)
end end
else else
have_color = Base.get_have_color()
lines_colored = Array{Union{Nothing,Vector{String}}}(undef, nr, nc) lines_colored = Array{Union{Nothing,Vector{String}}}(undef, nr, nc)
lines_uncolored = similar(lines_colored) lines_uncolored = have_color ? similar(lines_colored) : lines_colored
l_max = zeros(Int, nr) l_max = zeros(Int, nr)
w_max = zeros(Int, nc) w_max = zeros(Int, nc)
buf = IOContext(PipeBuffer(), :color => Base.get_have_color()) buf = IOContext(PipeBuffer(), :color => have_color)
re_col = r"\e\[[0-9;]*[a-zA-Z]" re_col = r"\e\[[0-9;]*[a-zA-Z]"
sps = 0 sps = 0
for r in 1:nr for r in 1:nr
@ -225,9 +226,11 @@ function Base.show(io::IO, plt::Plot{UnicodePlotsBackend})
sp = plt.o[sps += 1] sp = plt.o[sps += 1]
show(buf, sp) show(buf, sp)
colored = read(buf, String) colored = read(buf, String)
uncolored = replace(colored, re_col => "") lines_colored[r, c] = lu = lc = split(colored, '\n')
lines_colored[r, c] = lc = split(colored, "\n") if have_color
lines_uncolored[r, c] = lu = split(uncolored, "\n") uncolored = replace(colored, re_col => '\0')
lines_uncolored[r, c] = lu = split(uncolored, '\n')
end
lmax = max(length(lc), lmax) lmax = max(length(lc), lmax)
w_max[c] = max(maximum(length.(lu)), w_max[c]) w_max[c] = max(maximum(length.(lu)), w_max[c])
end end