Weave.jl/src/JuliaReport.jl

module JuliaReport
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


global const report = Report("", false, "", "",  {}, "", "")

function listformats()
  pweave.listformats()
end

function weave(source ; doctype = "pandoc", 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

    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 = 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(string("Invalid format for chunk options line: ", chunk["start_line"]))
            end

            merge!(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
  return parsed
end

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

    chunkid = ((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

export weave

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