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

Code re-factored. Breaking changes: thw macros are now named @gp and @gsp

Browse Source
This commit is contained in:
Giorgio Calderone 2018-04-19 18:21:48 +02:00
parent 106f8470ba
commit a536004b0d
4 changed files with 231 additions and 160 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
README.md
View File

@ -42,7 +42,6 @@ using Gnuplot
A slightly more complicated one showing a parabola with a solid line and a title: A slightly more complicated one showing a parabola with a solid line and a title:
``` Julia ``` Julia
using Gnuplot
x = 1:10 x = 1:10
@gp x x.^2 "w l tit 'Parabola'" @gp x x.^2 "w l tit 'Parabola'"
``` ```
@ -50,8 +49,6 @@ x = 1:10
A real life example showing some random noise generated data: A real life example showing some random noise generated data:
``` Julia ``` Julia
using Gnuplot
# Create some noisy data... # Create some noisy data...
x = linspace(-2pi, 2pi, 100); x = linspace(-2pi, 2pi, 100);
y = 1.5 * sin.(0.3 + 0.7x) ; y = 1.5 * sin.(0.3 + 0.7x) ;
@ -68,7 +65,7 @@ e = 0.5 * ones(x);
That's it for the first plots. The syntax should be familiar to most gnuplot users, with this code we: That's it for the first plots. The syntax should be familiar to most gnuplot users, with this code we:
- set a few gnuplot properties (`key` and `grid`); - set a few gnuplot properties (`key` and `grid`);
- set the X axis range and Y axis label; - set the X axis range and Y axis label;
- passed the data to gnuplot; - send the data to gnuplot;
- plot two data sets specifying a few details (style, line width, color, legend, etc...). - plot two data sets specifying a few details (style, line width, color, legend, etc...).
Note that this simple example already covers the vast majority of use cases, since the remaining details of the plot can be easily tweaked by adding the appropriate gnuplot command. Also note that you would barely recognize the Julia language by just looking at the `@gp` call since **Gnuplot.jl** aims to be mostly transparent: the user is supposed to focus only on the data and on the gnuplot commands, rather than the package details. Note that this simple example already covers the vast majority of use cases, since the remaining details of the plot can be easily tweaked by adding the appropriate gnuplot command. Also note that you would barely recognize the Julia language by just looking at the `@gp` call since **Gnuplot.jl** aims to be mostly transparent: the user is supposed to focus only on the data and on the gnuplot commands, rather than the package details.
@ -80,8 +77,10 @@ GNUPLOT (1) -> reset session
GNUPLOT (1) -> GNUPLOT (1) ->
GNUPLOT (1) -> set key horizontal GNUPLOT (1) -> set key horizontal
GNUPLOT (1) -> set grid GNUPLOT (1) -> set grid
GNUPLOT (1) -> set title 'My title'
GNUPLOT (1) -> set xrange [-7:7] GNUPLOT (1) -> set xrange [-7:7]
GNUPLOT (1) -> set ylabel 'Y label' GNUPLOT (1) -> set ylabel 'Y label'
GNUPLOT (1) -> set xlabel 'X label'
GNUPLOT (1) -> $data0 << EOD GNUPLOT (1) -> $data0 << EOD
GNUPLOT (1) -> -6.283185307179586 1.2258873407968363 GNUPLOT (1) -> -6.283185307179586 1.2258873407968363
GNUPLOT (1) -> -6.156252270670907 1.1443471266509504 GNUPLOT (1) -> -6.156252270670907 1.1443471266509504
@ -89,9 +88,9 @@ GNUPLOT (1) -> -6.029319234162229 1.05377837392046
GNUPLOT (1) -> ... GNUPLOT (1) -> ...
GNUPLOT (1) -> EOD GNUPLOT (1) -> EOD
GNUPLOT (1) -> $data1 << EOD GNUPLOT (1) -> $data1 << EOD
GNUPLOT (1) -> -6.283185307179586 1.770587856071291 0.5 GNUPLOT (1) -> -6.283185307179586 2.25743603855675 0.5
GNUPLOT (1) -> -6.156252270670907 0.9350095514668977 0.5 GNUPLOT (1) -> -6.156252270670907 0.8313068798234011 0.5
GNUPLOT (1) -> -6.029319234162229 0.8960704540397358 0.5 GNUPLOT (1) -> -6.029319234162229 0.6077957618755075 0.5
GNUPLOT (1) -> ... GNUPLOT (1) -> ...
GNUPLOT (1) -> EOD GNUPLOT (1) -> EOD
GNUPLOT (1) -> plot \ GNUPLOT (1) -> plot \
@ -111,28 +110,35 @@ Note the lack of ` -> ` and the different color in the reply (if your terminal i
The default verbosity level is 4. The default verbosity level is 4.
So far we have shown how to produce plots with a single command, however such task can also be break into multiple statements by using `@gpi` in place of `@gp`. The syntax is exactly the same, but we should explicitly take care of resetting the gnuplot session (by using the `0` number) and send the final plot commands (using the `:.` symbol), e.g.: So far we have shown how to produce plots with a single command, however such task can also be performed using multiple statements. The syntax is exactly the same, but we should use the `:-` symbol at the beginning of each statement (ecept the first) and at the end of each statement (except the last), e.g.:
``` Julia ``` Julia
# Reset the gnuplot session and give the dataset the name :aa # Reset the gnuplot session and give the dataset the name :aa
@gpi 0 x y+noise e :aa @gp x y+noise e :aa :-
# Define a model function to be fitted # Define a model function to be fitted
@gpi "f(x) = a * sin(b + c*x); a = 1; b = 1; c = 1;" @gp :- "f(x) = a * sin(b + c*x); a = 1; b = 1; c = 1;" :-
# Fit the function to the :aa dataset # Fit the function to the :aa dataset
@gpi "fit f(x) \$aa u 1:2:3 via a, b, c;" @gp :- "fit f(x) \$aa u 1:2:3 via a, b, c;" :-
# Prepare a multiplot showing the data, the model... # Prepare a multiplot showing the data, the model...
@gpi "set multiplot layout 2,1" @gp :- "set multiplot layout 2,1" :-
@gpi "plot \$aa w points tit 'Data'" ylab="Data and model" @gp :- "plot \$aa w points tit 'Data'" ylab="Data and model" :-
@gpi "plot \$aa u 1:(f(\$1)) w lines tit 'Best fit'" @gp :- "plot \$aa u 1:(f(\$1)) w lines tit 'Best fit'" :-
# ... and the residuals (the `2` here refer to the second plot in the multiplot. Also note the `:.` symbol has last argument which triggers the actual plot generation. # ... and the residuals (the `2` here refer to the second plot in the multiplot.
@gpi 2 xlab="X label" ylab="Residuals" @gp :- 2 xlab="X label" ylab="Residuals" :-
@gpi "plot \$aa u 1:((f(\$1)-\$2) / \$3):(1) w errorbars notit" :. @gp :- "plot \$aa u 1:((f(\$1)-\$2) / \$3):(1) w errorbars notit"
``` ```
Further documentation for the `@gp` and `@gpi` macros is available in the REPL by means of the `@doc` macro or by typing `?` in the REPL followed by the macro name. The **Gnuplot.jl** package also provide support for 3D plots using the `@gsp` macro in place of `@gp`, e.g.:
``` Julia
@gsp randn(Float64, 30, 50)
```
Further documentation for the `@gp` and `@gsp` macros is available in the REPL by means of the `@doc` macro or by typing `?` in the REPL followed by the macro name.
@ -189,6 +195,3 @@ GNUPLOT (2) Process exited with status 0
0 0
``` ```
Note that `GnuplotQuit` returns the exit code of the underlying gnuplot process. Alternatively you can use `GnuplotQuitAll()` to terminate all active istances. Note that `GnuplotQuit` returns the exit code of the underlying gnuplot process. Alternatively you can use `GnuplotQuitAll()` to terminate all active istances.

1
REQUIRE
View File

@ -1,2 +1,3 @@
julia 0.6 julia 0.6
AbbrvKW 0.3.1 AbbrvKW 0.3.1
ColorTypes 0.6.7

318
src/Gnuplot.jl
View File

@ -3,6 +3,7 @@ __precompile__(true)
module Gnuplot module Gnuplot
using AbbrvKW using AbbrvKW
using ColorTypes
import Base.send import Base.send
import Base.reset import Base.reset
@ -14,7 +15,7 @@ import Base.reset
export CheckGnuplotVersion, GnuplotSession, GnuplotProc, export CheckGnuplotVersion, GnuplotSession, GnuplotProc,
GnuplotQuit, GnuplotQuitAll, GnuplotGet, setCurrent, getCurrent, GnuplotQuit, GnuplotQuitAll, GnuplotGet, setCurrent, getCurrent,
@gp, @gpi, @gp_str, @gp_cmd @gp, @gsp, @gp_str, @gp_cmd
###################################################################### ######################################################################
@ -274,7 +275,7 @@ end
#--------------------------------------------------------------------- #---------------------------------------------------------------------
function addData(gp::GnuplotSession, data::Vararg{AbstractArray{T},M}; name="") where {T<:Number,M} function addData(gp::GnuplotSession, args...; name="")
if name == "" if name == ""
name = string("data", gp.blockCnt) name = string("data", gp.blockCnt)
gp.blockCnt += 1 gp.blockCnt += 1
@ -282,86 +283,131 @@ function addData(gp::GnuplotSession, data::Vararg{AbstractArray{T},M}; name="")
name = "\$$name" name = "\$$name"
# Check dimensions # Check dimensions
dimX = (size(data[1]))[1] maxDim = 0
dimY = 0 for iarg in 1:length(args)
is2D = false d = args[iarg]
first1D = 0 ok = false
coordX = Vector{Float64}() if typeof(d) <: AbstractArray
coordY = Vector{Float64}() if typeof(d[1]) <: Number
for i in length(data):-1:1 ok = true
d = data[i] end
@assert ndims(d) <=2 "Array dimensions must be <= 2" if typeof(d[1]) <: ColorTypes.RGB
ok = true
end
end
if ndims(d) > maxDim
maxDim = ndims(d)
end
if ndims(d) == 2 @assert ok "Invalid argument at position $iarg"
dimY == 0 && (dimY = (size(d))[2]) @assert maxDim <= 3 "Array dimensions must be <= 3"
end
dimX = 0
dimY = 0
dimZ = 0
count1D = 0
for iarg in 1:length(args)
d = args[iarg]
if ndims(d) == 1
count1D += 1
if maxDim == 1
(iarg == 1) && (dimX = length(d))
@assert dimX == length(d) "Array size are incompatible"
else
(iarg == 1) && (dimX = length(d))
(maxDim == 2) && (iarg == 2) && (dimY = length(d))
(maxDim == 3) && (iarg == 3) && (dimZ = length(d))
@assert iarg <= maxDim "2D and 3D data must be given at the end of argument list"
end
elseif ndims(d) == 2
if iarg == 1
dimX = (size(d))[1]
dimY = (size(d))[2]
end
@assert dimX == (size(d))[1] "Array size are incompatible" @assert dimX == (size(d))[1] "Array size are incompatible"
@assert dimY == (size(d))[2] "Array size are incompatible" @assert dimY == (size(d))[2] "Array size are incompatible"
@assert first1D == 0 "2D data must be given at the end of argument list" @assert dimZ == 0 "Mixing 2D and 3D data is not allowed"
is2D = true elseif ndims(d) == 3
end if iarg == 1
dimX = (size(d))[1]
if ndims(d) == 1 dimY = (size(d))[2]
if !is2D dimZ = (size(d))[3]
@assert dimX == (size(d))[1] "Array size are incompatible"
else
@assert i <= 2 "When 2D data are given only the first two arrays must be 1D"
if i == 1
@assert dimX == (size(d))[1] "Array size are incompatible"
end
if i == 2
@assert dimY == (size(d))[1] "Array size are incompatible"
end
end end
@assert dimX == (size(d))[1] "Array size are incompatible"
first1D = i @assert dimY == (size(d))[2] "Array size are incompatible"
@assert dimZ == (size(d))[3] "Array size are incompatible"
end end
end end
if is2D if (dimZ > 0) && (count1D != 0) && (count1D != 3)
if ndims(data[1]) == 1 error("Either zero or three 1D arrays must be given before 3D data")
@assert ndims(data[2]) == 1 "Only one coordinate of a 2D dataset has been given" elseif (dimY > 0) && (count1D != 0) && (count1D != 2)
coordX = deepcopy(data[1]) error("Either zero or two 1D arrays must be given before 2D data")
coordY = deepcopy(data[2])
else
coordX = collect(1.:1.:dimX)
coordY = collect(1.:1.:dimY)
end
end end
# Prepare data
v = "$name << EOD" v = "$name << EOD"
push!(gp.data, inputData(v)) push!(gp.data, inputData(v))
if !is2D if dimZ > 0 # 3D
for i in 1:dimX for ix in 1:dimX
v = "" for iy in 1:dimY
for j in 1:length(data) for iz in 1:dimZ
v *= " " * string(data[j][i]) if count1D == 0
v = string(ix) * " " * string(iy) * " " * string(iz)
else
v = string(args[1][ix]) * " " * string(args[2][iy]) * " " * string(args[3][iz])
end
for iarg in count1D+1:length(args)
d = args[iarg]
v *= " " * string(d[ix,iy,iz])
end
push!(gp.data, inputData(v))
end
end end
push!(gp.data, inputData(v)) push!(gp.data, inputData(""))
end end
else elseif dimY > 0 # 2D
for i in 1:dimX for ix in 1:dimX
for j in 1:dimY for iy in 1:dimY
v = string(coordX[i]) * " " * string(coordY[j]) if count1D == 0
for d in data v = string(ix) * " " * string(iy)
ndims(d) == 1 && (continue) else
v *= " " * string(d[i,j]) v = string(args[1][ix]) * " " * string(args[2][iy])
end
for iarg in count1D+1:length(args)
d = args[iarg]
if typeof(d[ix,iy]) <: ColorTypes.RGB
tmp = d[ix,iy]
v *= " " * string(float(tmp.r)*255) * " " * string(float(tmp.g)*255) * " " * string(float(tmp.b)*255)
else
v *= " " * string(d[ix,iy])
end
end end
push!(gp.data, inputData(v)) push!(gp.data, inputData(v))
end end
push!(gp.data, inputData("")) push!(gp.data, inputData(""))
end end
else # 1D
for ix in 1:dimX
v = ""
for iarg in 1:length(args)
d = args[iarg]
v *= " " * string(d[ix])
end
push!(gp.data, inputData(v))
end
end end
v = "EOD" push!(gp.data, inputData("EOD"))
push!(gp.data, inputData(v))
return (name, is2D) return name
end end
function addData(gp::GnuplotProc, data::Vararg{AbstractArray{T},M}; name="") where {T<:Number,M} function addData(gp::GnuplotProc, args...; name="")
name = addData(gp.session, data..., name=name) name = addData(gp.session, args..., name=name)
first = true first = true
count = 0 count = 0
@ -397,9 +443,7 @@ setMultiID(gp::GnuplotProc, id::Int) = setMultiID(gp.session, id)
#--------------------------------------------------------------------- #---------------------------------------------------------------------
function setSplot(gp::GnuplotSession, splot::Bool) function setSplot(gp::GnuplotSession, splot::Bool)
if splot gp.plot[gp.multiID].splot = splot
gp.plot[gp.multiID].splot = splot
end
end end
setSplot(gp::GnuplotProc, splot::Bool) = setSplot(gp.session, splot) setSplot(gp::GnuplotProc, splot::Bool) = setSplot(gp.session, splot)
@ -533,49 +577,30 @@ end
#--------------------------------------------------------------------- #---------------------------------------------------------------------
function gpDriver(args...) function gpDriver(splot, args...)
gp = nothing
for arg in args
if typeof(arg) == GnuplotProc ||
typeof(arg) == GnuplotSession
gp = arg
end
end
if gp == nothing
gp = getCurrent()
end
if length(args) == 0 if length(args) == 0
#gpDump(gp) gpDump(getCurrent())
return nothing return nothing
end end
gp = nothing
eData = Vector{Any}() eData = Vector{Any}()
dataName = "" dataName = ""
addDump = false addDump = true
term = ("", "") term = ("", "")
file="" file=""
stream=nothing stream=nothing
function endOfData(associatedPlot=nothing) function endOfData(associatedPlot=nothing)
if length(eData) > 0 if length(eData) > 0
(last, splot) = addData(gp, eData...; name=dataName) last = addData(gp, eData...; name=dataName)
if associatedPlot != nothing if associatedPlot != nothing
setSplot(gp, splot)
addPlot(gp, last, associatedPlot) addPlot(gp, last, associatedPlot)
end end
end end
eData = Vector{Any}() eData = Vector{Any}()
dataName = "" dataName = ""
end end
function endOfStream(forceDump::Bool)
endOfData("")
(forceDump || addDump) && (gpDump(gp; term=term, file=file, stream=stream))
addDump = false
term = ("", "")
file=""
stream=nothing
end
function isPlotCmd(s::String) function isPlotCmd(s::String)
(length(s) >= 2) && (s[1:2] == "p " ) && (return (true, false, strip(s[2:end]))) (length(s) >= 2) && (s[1:2] == "p " ) && (return (true, false, strip(s[2:end])))
(length(s) >= 3) && (s[1:3] == "pl " ) && (return (true, false, strip(s[3:end]))) (length(s) >= 3) && (s[1:3] == "pl " ) && (return (true, false, strip(s[3:end])))
@ -593,25 +618,35 @@ function gpDriver(args...)
for iarg in 1:length(args) for iarg in 1:length(args)
arg = args[iarg] arg = args[iarg]
if typeof(arg) == GnuplotProc ||
typeof(arg) == GnuplotSession
gp = arg
end
if gp == nothing
gp = getCurrent()
end
if iarg == 1
if (typeof(arg) != Symbol) || (arg != :-)
reset(gp)
end
setSplot(gp, splot)
end
if typeof(arg) == GnuplotProc || if typeof(arg) == GnuplotProc ||
typeof(arg) == GnuplotSession typeof(arg) == GnuplotSession
continue continue
elseif typeof(arg) == Symbol end
if arg == :.
addDump = true if typeof(arg) == Symbol
elseif arg == :splot if arg == :-
setSplot(gp, true) (iarg == length(args)) && (addDump = false)
else else
dataName = string(arg) dataName = string(arg)
endOfData() endOfData()
end end
elseif isa(arg, Int) elseif isa(arg, Int)
if arg == 0 @assert arg > 0
reset(gp) endOfData("")
else setMultiID(gp, arg)
endOfData("")
setMultiID(gp, arg)
end
elseif isa(arg, String) elseif isa(arg, String)
# Either a plot or cmd string # Either a plot or cmd string
if length(eData) > 0 if length(eData) > 0
@ -650,7 +685,9 @@ function gpDriver(args...)
push!(eData, arg) push!(eData, arg)
end end
end end
endOfStream(false)
endOfData("")
(addDump) && (gpDump(gp; term=term, file=file, stream=stream))
return nothing return nothing
end end
@ -862,20 +899,26 @@ end
""" """
# @gp # @gp
The `@gp` (and its companion `@gpi`) allows to exploit all of the The `@gp`, and its companion `@gsp`(to be used for the `splot`
**Gnuplot** package functionalities using an extremely efficient and operations) allows to exploit all of the **Gnuplot** package
concise syntax. functionalities using an extremely efficient and concise syntax. Both
macros accept the same syntax, described below:
The `@gp` accepts any number of arguments, with the following meaning: The `@gp` macro accepts any number of arguments, with the following
meaning:
- a string: a command (e.g. "set key left") or plot specification; - a string: a command (e.g. "set key left") or plot specification;
- a `GnuplotProc` or `GnuplotSession` object: set the current sink; - a `GnuplotProc` or `GnuplotSession` object: set the current destination;
- a symbol: specifies the data set name; - a symbol: specifies the data set name;
- an `Int`: if >0 set the current plot destination (if multiplot is - an `Int`: if >0 set the current plot destination (if multiplot is
enabled). If 0 reset the whole session. enabled);
- a keyword: set the keyword value (see below); - a keyword: set the keyword value (see below);
- any other data type: data to be passed to Gnuplot. Each dataset - any other data type: data to be passed to Gnuplot. Each dataset
must be terminated by either: a symbol (i.e. the data set name) or a must be terminated by either: a symbol (i.e. the data set name) or a
string with the plot specifications (e.g. "with lines"). string with the plot specifications (e.g. "with lines");
- the `:-` symbol, used as first argument, avoids resetting the
Gnuplot session. Used as last argument avoids immediate execution of
the plot/splot command. This symbol can be used to split a single
`@gp` call in multiple ones.
All entries are optional, and there is no mandatory order. The plot All entries are optional, and there is no mandatory order. The plot
specification can either be: a complete plot/splot command (e.g., specification can either be: a complete plot/splot command (e.g.,
@ -941,8 +984,8 @@ should use the `@gpi` macro instead, with exaclty the same syntax as
@gp "plo sin(x)" "s cos(x)" @gp "plo sin(x)" "s cos(x)"
# Split a `@gp` call in two # Split a `@gp` call in two
@gpi 0 "plot sin(x)" @gp "plot sin(x)" :-
@gpi "plot cos(x)" :. @gp :- "plot cos(x)"
# Insert a 3 second pause between one plot and the next # Insert a 3 second pause between one plot and the next
@gp "plot sin(x)" 2 xr=(-2pi,2pi) "pause 3" "plot cos(4*x)" @gp "plot sin(x)" 2 xr=(-2pi,2pi) "pause 3" "plot cos(4*x)"
@ -975,11 +1018,11 @@ lw = 3
4, "p sin(4*x)") 4, "p sin(4*x)")
# or equivalently # or equivalently
@gpi 0 xr=(-2pi,2pi) "unset key" "set multi layout 2,2 title 'Multiplot title'" @gp xr=(-2pi,2pi) "unset key" "set multi layout 2,2 title 'Multiplot title'" :-
for i in 1:4 for i in 1:4
@gpi i "p sin(\$i*x)" @gp :- i "p sin(\$i*x)" :-
end end
@gpi :. @gp
# Multiple gnuplot instances # Multiple gnuplot instances
@ -999,7 +1042,9 @@ noise = randn(length(x))./2;
e = 0.5 * ones(x); e = 0.5 * ones(x);
@gp verb=2 x y :aa "plot \\\$aa w l" "pl \\\$aa u 1:(2*\\\$2) w l" @gp verb=2 x y :aa "plot \\\$aa w l" "pl \\\$aa u 1:(2*\\\$2) w l"
@gp randn(Float64, 30, 50)
@gsp randn(Float64, 30, 50)
@gp randn(Float64, 30, 50) "w image"
@gp("set key horizontal", "set grid", @gp("set key horizontal", "set grid",
xrange=(-7,7), ylabel="Y label", xrange=(-7,7), ylabel="Y label",
@ -1007,36 +1052,57 @@ e = 0.5 * ones(x);
x, y+noise, e, "w errorbars t 'Data'"); x, y+noise, e, "w errorbars t 'Data'");
@gpi 0 "f(x) = a * sin(b + c*x); a = 1; b = 1; c = 1;" @gp "f(x) = a * sin(b + c*x); a = 1; b = 1; c = 1;" :-
@gpi x y+noise e :aa @gp :- x y+noise e :aa :-
@gpi "fit f(x) \\\$aa u 1:2:3 via a, b, c;" @gp :- "fit f(x) \\\$aa u 1:2:3 via a, b, c;" :-
@gpi "set multiplot layout 2,1" @gp :- "set multiplot layout 2,1" :-
@gpi "plot \\\$aa w points tit 'Data'" ylab="Data and model" @gp :- "plot \\\$aa w points tit 'Data'" ylab="Data and model" :-
@gpi "plot \\\$aa u 1:(f(\\\$1)) w lines tit 'Best fit'" @gp :- "plot \\\$aa u 1:(f(\\\$1)) w lines tit 'Best fit'" :-
@gpi 2 xlab="X label" ylab="Residuals" @gp :- 2 xlab="X label" ylab="Residuals" :-
@gpi "plot \\\$aa u 1:((f(\\\$1)-\\\$2) / \\\$3):(1) w errorbars notit" :. @gp :- "plot \\\$aa u 1:((f(\\\$1)-\\\$2) / \\\$3):(1) w errorbars notit"
# Display an image
using TestImages
img = testimage("lena");
@gp img "w image"
@gp "set size square" img "w rgbimage" # Color image with correct proportions
@gp "set size square" img "u 2:(-\\\$1):3:4:5 with rgbimage" # Correct orientation
``` ```
""" """
macro gp(args...) macro gp(args...)
esc_args = Vector{Any}() # esc_args = Vector{Any}()
for arg in args # for arg in args
push!(esc_args, esc(arg)) # push!(esc_args, esc(arg))
end # end
e = :(@gpi(0, $(esc_args...), :.)) # e = :(@gp(splot=true, $(esc_args...)))
# return e
return e out = Expr(:call)
push!(out.args, :(Gnuplot.gpDriver))
push!(out.args, false)
for iarg in 1:length(args)
arg = args[iarg ]
if (isa(arg, Expr) && (arg.head == :(=)))
sym = string(arg.args[1])
val = arg.args[2]
push!(out.args, :((Symbol($sym),$val)))
else
push!(out.args, arg)
end
end
return esc(out)
end end
""" """
# @gpi # @gsp
See documentation for `@gp`. See documentation for `@gp`.
""" """
macro gpi(args...) macro gsp(args...)
out = Expr(:call) out = Expr(:call)
push!(out.args, :(Gnuplot.gpDriver)) push!(out.args, :(Gnuplot.gpDriver))
push!(out.args, true)
for iarg in 1:length(args) for iarg in 1:length(args)
arg = args[iarg ] arg = args[iarg ]
if (isa(arg, Expr) && (arg.head == :(=))) if (isa(arg, Expr) && (arg.head == :(=)))

27
test/runtests.jl
View File

@ -22,11 +22,11 @@ function gp_test()
@gp "plot sin(x)" "pl cos(x)" @gp "plot sin(x)" "pl cos(x)"
@gp "plo sin(x)" "s cos(x)" @gp "plo sin(x)" "s cos(x)"
@gpi 0 "plot sin(x)" @gp "plot sin(x)" :-
@gpi "plot cos(x)" :. @gp :- "plot cos(x)"
@gp "plot sin(x)" 2 xr=(-2pi,2pi) "pause 3" "plot cos(4*x)" @gp "plot sin(x)" 2 xr=(-2pi,2pi) "pause 2" "plot cos(4*x)"
x = linspace(-2pi, 2pi, 100); x = linspace(-2pi, 2pi, 100);
y = 1.5 * sin.(0.3 + 0.7x) ; y = 1.5 * sin.(0.3 + 0.7x) ;
@ -37,8 +37,9 @@ function gp_test()
@gp x y "w l" @gp x y "w l"
@gp x y :aa "plot \$aa w l" "pl \$aa u 1:(2*\$2) w l" @gp x y :aa "plot \$aa w l" "pl \$aa u 1:(2*\$2) w l"
@gp randn(Float64, 30, 50) @gsp randn(Float64, 30, 50)
@gp :splot x y y @gp randn(Float64, 30, 50) "w image"
@gsp x y y
@gp("set key horizontal", "set grid", @gp("set key horizontal", "set grid",
xrange=(-7,7), ylabel="Y label", xrange=(-7,7), ylabel="Y label",
@ -46,14 +47,14 @@ function gp_test()
x, y+noise, e, "w errorbars t 'Data'"); x, y+noise, e, "w errorbars t 'Data'");
@gpi 0 "f(x) = a * sin(b + c*x); a = 1; b = 1; c = 1;" @gp "f(x) = a * sin(b + c*x); a = 1; b = 1; c = 1;" :-
@gpi x y+noise e :aa @gp :- x y+noise e :aa :-
@gpi "fit f(x) \$aa u 1:2:3 via a, b, c;" @gp :- "fit f(x) \$aa u 1:2:3 via a, b, c;" :-
@gpi "set multiplot layout 2,1" @gp :- "set multiplot layout 2,1" :-
@gpi "plot \$aa w points" ylab="Data and model" @gp :- "plot \$aa w points" ylab="Data and model" :-
@gpi "plot \$aa u 1:(f(\$1)) w lines" @gp :- "plot \$aa u 1:(f(\$1)) w lines" :-
@gpi 2 xlab="X label" ylab="Residuals" @gp :- 2 xlab="X label" ylab="Residuals" :-
@gpi "plot \$aa u 1:((f(\$1)-\$2) / \$3):(1) w errorbars notit" :. @gp :- "plot \$aa u 1:((f(\$1)-\$2) / \$3):(1) w errorbars notit"
#----------------------------------------------------------------- #-----------------------------------------------------------------