Weave.jl/src/JuliaReport.jl

module JuliaReport
using Compat
using PyCall
#using PyPlot
@pyimport pweave #Output formatting uses Pweave


#Contains report global properties
#Similar to pweave.PwebProcessor
type Report
  source::String
  documentationmode::Bool
  cwd::String
  basename::String
  formatdict
  pending_code::String
  figdir::String
end

const report = Report("", false, "", "",  Any[], "", "")

function listformats()
  pweave.listformats()
end

function weave(source ; doctype = "pandoc", plotlib="PyPlot", informat="noweb", figdir = "figures", figformat = nothing)
    pweave.rcParams["chunk"]["defaultoptions"]["engine"] = "julia"

    doc = pweave.Pweb(source, doctype, shell="julia")
    #doc[:setreader](informat)
    #doc[:parse]()

    cwd, fname = splitdir(abspath(source))
    basename = splitext(fname)[1]
    formatdict = doc[:formatter][:getformatdict]()
    figformat == nothing || (formatdict["figfmt"] = figformat)
    #println(formatdict["figfmt"])
    #report = Report(source, false, cwd, basename, formatdict, "", figdir)

    report.source = source
    report.cwd = cwd
    report.basename = basename
    report.figdir = figdir
    report.formatdict = formatdict

    l_plotlib = lowercase(plotlib)
    if l_plotlib == "winston"
      eval(Expr(:using, :Winston))
      rcParams[:plotlib] = "Winston"
    elseif l_plotlib == "pyplot"
      #eval(parse("import PyPlot.plt"))
      eval(Expr(:using, :PyPlot))
      rcParams[:plotlib] = "PyPlot"
    end


    parsed = read_noweb(source)
    doc[:executed] = run(parsed)

    #Formatting with pweave
    #doc[:executed] = run(PyVector(doc["parsed"]))
    doc[:isexecuted] = true
    doc[:format]()
    doc[:write]()
end


function run_block(code_str)
    oldSTDOUT = STDOUT
    #If there is nothing to read code will hang
    println()
    rw, wr = redirect_stdout()
    include_string(code_str)
    redirect_stdout(oldSTDOUT)
    close(wr)
    result = readall(rw)
    close(rw)
    return string("\n", result)
end

function run_term(code_str)
    oldSTDOUT = STDOUT
    #If there is nothing to read code will hang
    println()
    rw, wr = redirect_stdout()


    #Emulate terminal
    n = length(code_str)
    pos = 2 #The first character is extra line end
    while pos < n
        oldpos = pos
        code, pos = parse(code_str, pos)
        println(string("\njulia> ", rstrip(code_str[oldpos:(pos-1)])))
        s = eval(code)
        s == nothing || (smime = reprmime(MIME("text/plain"), s))  #display(s)
        println(smime)
    end

    redirect_stdout(oldSTDOUT)
    close(wr)
    result = readall(rw)
    close(rw)
    return string(result)
end


function run(parsed)
  i = 1
  for chunk in copy(parsed)
    if chunk[:type] == "code"
      #print(chunk["content"])
      info("Weaving chunk $(chunk[:number]) from line $(chunk[:start_line])")
      defaults = copy(rcParams[:chunk][:defaultoptions])
      options = copy(chunk[:options])
      try
        options = merge(rcParams[:chunk][:defaultoptions], options)
      catch
        options = rcParams[:chunk][:defaultoptions]
        warn("Invalid format for chunk options line: $(chunk[:start_line])")
      end

      merge!(chunk, options)
      delete!(chunk, :options)

      chunk[:evaluate] || (chunk[:result] = ""; continue) #Do nothing if eval is false
      if chunk[:term]
        chunk[:result] = run_term(chunk[:content])
      else
        chunk[:result] = run_block(chunk[:content])
      end

      chunk[:fig] && (chunk[:figure] = savefigs(chunk))
    end
    parsed[i] = copy(chunk)
    i += 1
  end
  parsed
end

function savefigs(chunk)
    l_plotlib = lowercase(rcParams[:plotlib])
    if l_plotlib == "pyplot"
      return savefigs_pyplot(chunk)
    elseif l_plotlib == "winston"
      return savefigs_winston(chunk)
    end
end

function savefigs_pyplot(chunk)
    fignames = String[]
    ext = report.formatdict["figfmt"]
    figpath = joinpath(report.cwd, report.figdir)
    isdir(figpath) || mkdir(figpath)

    chunkid = get(chunk,:name,chunk[:number]) #((chunk[:name] == nothing) ? chunk[:number] : chunk[:name])
    #Iterate over all open figures, save them and store names
    for fig = plt.get_fignums()
        full_name = joinpath(report.cwd, report.figdir, "$(report.basename)_$(chunkid)_$fig$ext")
        rel_name = "$(report.figdir)/$(report.basename)_$(chunkid)_$fig$ext" #Relative path is used in output
        savefig(full_name)
        push!(fignames, rel_name)
        plt.draw()
        plt.close()
    end
    return fignames
end

function savefigs_winston(chunk)
    fignames = String[]
    ext = report.formatdict["figfmt"]
    figpath = joinpath(report.cwd, report.figdir)
    isdir(figpath) || mkdir(figpath)

    chunkid = get(chunk,:name,chunk[:number])#((chunk[:name] == nothing) ? chunk[:number] : chunk[:name])
    #Iterate over all open figures, save them and store names
    #println(Winston._display.figs)
    #println(Winston._display.fig_order)

    for fig = copy(Winston._display.fig_order)
        full_name = joinpath(report.cwd, report.figdir, "$(report.basename)_$(chunkid)_$fig$ext")
        rel_name = "$(report.figdir)/$(report.basename)_$(chunkid)_$fig$ext" #Relative path is used in output
        println(rel_name)
        #figure(fig) #Calling figure clears the canvas!
        #savefig(Winston._display.figs[gcf()].plot, full_name)
        savefig(full_name)
        closefig()
        #savefig(full_name)
        push!(fignames, rel_name)
    end
    return fignames
end

#Saving Winston figures
#savefig(Winston._display.figs[1].plot, "test.png")

export weave

include("config.jl")
include("readers.jl")
end