uniform length 4 indentation in utils.jl
This commit is contained in:
parent
2702dc6654
commit
55bb6fda1b
126
src/utils.jl
126
src/utils.jl
@ -1,119 +1,4 @@
|
|||||||
|
|
||||||
calcMidpoints(edges::AbstractVector) = Float64[0.5 * (edges[i] + edges[i+1]) for i in 1:length(edges)-1]
|
|
||||||
|
|
||||||
"Make histogram-like bins of data"
|
|
||||||
function binData(data, nbins)
|
|
||||||
lo, hi = ignorenan_extrema(data)
|
|
||||||
edges = collect(range(lo, stop=hi, length=nbins+1))
|
|
||||||
midpoints = calcMidpoints(edges)
|
|
||||||
buckets = Int[max(2, min(searchsortedfirst(edges, x), length(edges)))-1 for x in data]
|
|
||||||
counts = zeros(Int, length(midpoints))
|
|
||||||
for b in buckets
|
|
||||||
counts[b] += 1
|
|
||||||
end
|
|
||||||
edges, midpoints, buckets, counts
|
|
||||||
end
|
|
||||||
|
|
||||||
"""
|
|
||||||
A hacky replacement for a histogram when the backend doesn't support histograms directly.
|
|
||||||
Convert it into a bar chart with the appropriate x/y values.
|
|
||||||
"""
|
|
||||||
function histogramHack(; kw...)
|
|
||||||
plotattributes = KW(kw)
|
|
||||||
|
|
||||||
# we assume that the y kwarg is set with the data to be binned, and nbins is also defined
|
|
||||||
edges, midpoints, buckets, counts = binData(plotattributes[:y], plotattributes[:bins])
|
|
||||||
plotattributes[:x] = midpoints
|
|
||||||
plotattributes[:y] = float(counts)
|
|
||||||
plotattributes[:seriestype] = :bar
|
|
||||||
plotattributes[:fillrange] = plotattributes[:fillrange] === nothing ? 0.0 : plotattributes[:fillrange]
|
|
||||||
plotattributes
|
|
||||||
end
|
|
||||||
|
|
||||||
"""
|
|
||||||
A hacky replacement for a bar graph when the backend doesn't support bars directly.
|
|
||||||
Convert it into a line chart with fillrange set.
|
|
||||||
"""
|
|
||||||
function barHack(; kw...)
|
|
||||||
plotattributes = KW(kw)
|
|
||||||
midpoints = plotattributes[:x]
|
|
||||||
heights = plotattributes[:y]
|
|
||||||
fillrange = plotattributes[:fillrange] === nothing ? 0.0 : plotattributes[:fillrange]
|
|
||||||
|
|
||||||
# estimate the edges
|
|
||||||
dists = diff(midpoints) * 0.5
|
|
||||||
edges = zeros(length(midpoints)+1)
|
|
||||||
for i in eachindex(edges)
|
|
||||||
if i == 1
|
|
||||||
edge = midpoints[1] - dists[1]
|
|
||||||
elseif i == length(edges)
|
|
||||||
edge = midpoints[i-1] + dists[i-2]
|
|
||||||
else
|
|
||||||
edge = midpoints[i-1] + dists[i-1]
|
|
||||||
end
|
|
||||||
edges[i] = edge
|
|
||||||
end
|
|
||||||
|
|
||||||
x = Float64[]
|
|
||||||
y = Float64[]
|
|
||||||
for i in eachindex(heights)
|
|
||||||
e1, e2 = edges[i:i+1]
|
|
||||||
append!(x, [e1, e1, e2, e2])
|
|
||||||
append!(y, [fillrange, heights[i], heights[i], fillrange])
|
|
||||||
end
|
|
||||||
|
|
||||||
plotattributes[:x] = x
|
|
||||||
plotattributes[:y] = y
|
|
||||||
plotattributes[:seriestype] = :path
|
|
||||||
plotattributes[:fillrange] = fillrange
|
|
||||||
plotattributes
|
|
||||||
end
|
|
||||||
|
|
||||||
|
|
||||||
"""
|
|
||||||
A hacky replacement for a sticks graph when the backend doesn't support sticks directly.
|
|
||||||
Convert it into a line chart that traces the sticks, and a scatter that sets markers at the points.
|
|
||||||
"""
|
|
||||||
function sticksHack(; kw...)
|
|
||||||
plotattributesLine = KW(kw)
|
|
||||||
plotattributesScatter = copy(plotattributesLine)
|
|
||||||
|
|
||||||
# these are the line vertices
|
|
||||||
x = Float64[]
|
|
||||||
y = Float64[]
|
|
||||||
fillrange = plotattributesLine[:fillrange] === nothing ? 0.0 : plotattributesLine[:fillrange]
|
|
||||||
|
|
||||||
# calculate the vertices
|
|
||||||
yScatter = plotattributesScatter[:y]
|
|
||||||
for (i,xi) in enumerate(plotattributesScatter[:x])
|
|
||||||
yi = yScatter[i]
|
|
||||||
for j in 1:3 push!(x, xi) end
|
|
||||||
append!(y, [fillrange, yScatter[i], fillrange])
|
|
||||||
end
|
|
||||||
|
|
||||||
# change the line args
|
|
||||||
plotattributesLine[:x] = x
|
|
||||||
plotattributesLine[:y] = y
|
|
||||||
plotattributesLine[:seriestype] = :path
|
|
||||||
plotattributesLine[:markershape] = :none
|
|
||||||
plotattributesLine[:fillrange] = nothing
|
|
||||||
|
|
||||||
# change the scatter args
|
|
||||||
plotattributesScatter[:seriestype] = :none
|
|
||||||
|
|
||||||
plotattributesLine, plotattributesScatter
|
|
||||||
end
|
|
||||||
|
|
||||||
function regressionXY(x, y)
|
|
||||||
# regress
|
|
||||||
β, α = convert(Matrix{Float64}, [x ones(length(x))]) \ convert(Vector{Float64}, y)
|
|
||||||
|
|
||||||
# make a line segment
|
|
||||||
regx = [ignorenan_minimum(x), ignorenan_maximum(x)]
|
|
||||||
regy = β * regx + α
|
|
||||||
regx, regy
|
|
||||||
end
|
|
||||||
|
|
||||||
function replace_image_with_heatmap(z::Array{T}) where T<:Colorant
|
function replace_image_with_heatmap(z::Array{T}) where T<:Colorant
|
||||||
n, m = size(z)
|
n, m = size(z)
|
||||||
colors = ColorGradient(vec(z))
|
colors = ColorGradient(vec(z))
|
||||||
@ -279,8 +164,6 @@ function _expand_limits(lims, x)
|
|||||||
e1, e2 = ignorenan_extrema(x)
|
e1, e2 = ignorenan_extrema(x)
|
||||||
lims[1] = NaNMath.min(lims[1], e1)
|
lims[1] = NaNMath.min(lims[1], e1)
|
||||||
lims[2] = NaNMath.max(lims[2], e2)
|
lims[2] = NaNMath.max(lims[2], e2)
|
||||||
# catch err
|
|
||||||
# @warn(err)
|
|
||||||
catch
|
catch
|
||||||
end
|
end
|
||||||
nothing
|
nothing
|
||||||
@ -771,9 +654,9 @@ function with(f::Function, args...; kw...)
|
|||||||
backend(arg)
|
backend(arg)
|
||||||
end
|
end
|
||||||
|
|
||||||
# # TODO: generalize this strategy to allow args as much as possible
|
# TODO: generalize this strategy to allow args as much as possible
|
||||||
# # as in: with(:gr, :scatter, :legend, :grid) do; ...; end
|
# as in: with(:gr, :scatter, :legend, :grid) do; ...; end
|
||||||
# # TODO: can we generalize this enough to also do something similar in the plot commands??
|
# TODO: can we generalize this enough to also do something similar in the plot commands??
|
||||||
|
|
||||||
# k = :seriestype
|
# k = :seriestype
|
||||||
# if arg in _allTypes
|
# if arg in _allTypes
|
||||||
@ -858,13 +741,12 @@ end
|
|||||||
# used in updating an existing series
|
# used in updating an existing series
|
||||||
|
|
||||||
extendSeriesByOne(v::UnitRange{Int}, n::Int = 1) = isempty(v) ? (1:n) : (minimum(v):maximum(v)+n)
|
extendSeriesByOne(v::UnitRange{Int}, n::Int = 1) = isempty(v) ? (1:n) : (minimum(v):maximum(v)+n)
|
||||||
extendSeriesByOne(v::AVec, n::Integer = 1) = isempty(v) ? (1:n) : vcat(v, (1:n) + ignorenan_maximum(v))
|
extendSeriesByOne(v::AVec, n::Integer = 1) = isempty(v) ? (1:n) : vcat(v, (1:n) .+ ignorenan_maximum(v))
|
||||||
extendSeriesData(v::AbstractRange{T}, z::Real) where {T} = extendSeriesData(float(collect(v)), z)
|
extendSeriesData(v::AbstractRange{T}, z::Real) where {T} = extendSeriesData(float(collect(v)), z)
|
||||||
extendSeriesData(v::AbstractRange{T}, z::AVec) where {T} = extendSeriesData(float(collect(v)), z)
|
extendSeriesData(v::AbstractRange{T}, z::AVec) where {T} = extendSeriesData(float(collect(v)), z)
|
||||||
extendSeriesData(v::AVec{T}, z::Real) where {T} = (push!(v, convert(T, z)); v)
|
extendSeriesData(v::AVec{T}, z::Real) where {T} = (push!(v, convert(T, z)); v)
|
||||||
extendSeriesData(v::AVec{T}, z::AVec) where {T} = (append!(v, convert(Vector{T}, z)); v)
|
extendSeriesData(v::AVec{T}, z::AVec) where {T} = (append!(v, convert(Vector{T}, z)); v)
|
||||||
|
|
||||||
|
|
||||||
# -------------------------------------------------------
|
# -------------------------------------------------------
|
||||||
# NOTE: backends should implement the following methods to get/set the x/y/z data objects
|
# NOTE: backends should implement the following methods to get/set the x/y/z data objects
|
||||||
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user