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

gadfly color groups for line segments

Browse Source
This commit is contained in:
Thomas Breloff 2015-09-25 16:47:17 -04:00
parent d4da910f19
commit 6f1ef82e89
2 changed files with 119 additions and 32 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

125
examples/playground.ipynb
View File

File diff suppressed because one or more lines are too long

26
src/backends/gadfly.jl
View File

@ -102,7 +102,8 @@ end
# error("unsupported linestyle: ", linestyle) # error("unsupported linestyle: ", linestyle)
# end # end
createSegments(z) = collect(repmat(z',2,1))[2:end]
Base.first(c::Colorant) = c
function addGadflySeries!(gplt, d::Dict, initargs::Dict) function addGadflySeries!(gplt, d::Dict, initargs::Dict)
@ -121,8 +122,9 @@ function addGadflySeries!(gplt, d::Dict, initargs::Dict)
# set theme: color, line width, and point size # set theme: color, line width, and point size
line_width = d[:width] * (d[:linetype] in (:none, :ohlc, :scatter) ? 0 : 1) * Gadfly.px # 0 width when we don't show a line line_width = d[:width] * (d[:linetype] in (:none, :ohlc, :scatter) ? 0 : 1) * Gadfly.px # 0 width when we don't show a line
line_color = isa(d[:color], AbstractVector) ? colorant"black" : d[:color]
# fg = initargs[:foreground_color] # fg = initargs[:foreground_color]
theme = Gadfly.Theme(; default_color = d[:color], theme = Gadfly.Theme(; default_color = line_color,
line_width = line_width, line_width = line_width,
default_point_size = 0.5 * d[:markersize] * Gadfly.px, default_point_size = 0.5 * d[:markersize] * Gadfly.px,
# grid_color = fg, # grid_color = fg,
@ -159,6 +161,22 @@ function addGadflySeries!(gplt, d::Dict, initargs::Dict)
# add the Geoms # add the Geoms
append!(gfargs, getLineGeoms(d)) append!(gfargs, getLineGeoms(d))
# colorgroup
if isa(d[:color], AbstractVector)
# create a color scale, and set the color group to the index of the color
push!(gplt.scales, Gadfly.Scale.color_discrete_manual(d[:color]...))
# this is super weird, but... oh well... for some reason this creates n separate line segments...
# create a list of vertices that go: [x1,x2,x2,x3,x3, ... ,xi,xi, ... xn,xn] (same for y)
# then the vector passed to the "color" keyword should be a vector: [1,1,2,2,3,3,4,4, ..., i,i, ... , n,n]
csindices = Int[mod1(i,length(d[:color])) for i in 1:length(d[:y])]
cs = collect(repmat(csindices', 2, 1))[1:end-1]
d[:x], d[:y] = map(createSegments, (d[:x], d[:y]))
colorgroup = [(:color, cs)]
else
colorgroup = []
end
# fillto # fillto
if d[:fillto] == nothing if d[:fillto] == nothing
yminmax = [] yminmax = []
@ -190,7 +208,7 @@ function addGadflySeries!(gplt, d::Dict, initargs::Dict)
# Should ensure from this side that colors which are the same are merged together # Should ensure from this side that colors which are the same are merged together
push!(gplt.guides[1].labels, d[:label]) push!(gplt.guides[1].labels, d[:label])
push!(gplt.guides[1].colors, d[:marker] == :none ? d[:color] : d[:markercolor]) push!(gplt.guides[1].colors, d[:marker] == :none ? first(d[:color]) : d[:markercolor])
# println("updated legend: ", gplt.guides) # println("updated legend: ", gplt.guides)
end end
@ -204,7 +222,7 @@ function addGadflySeries!(gplt, d::Dict, initargs::Dict)
# add the layer to the Gadfly.Plot # add the layer to the Gadfly.Plot
# prepend!(gplt.layers, Gadfly.layer(unique(gfargs)..., d[:args]...; x = x, y = d[:y], d[:kwargs]...)) # prepend!(gplt.layers, Gadfly.layer(unique(gfargs)..., d[:args]...; x = x, y = d[:y], d[:kwargs]...))
prepend!(gplt.layers, Gadfly.layer(unique(gfargs)...; x = x, y = d[:y], yminmax...)) prepend!(gplt.layers, Gadfly.layer(unique(gfargs)...; x = x, y = d[:y], colorgroup..., yminmax...))
nothing nothing
end end