texlab 0.0.0

data/lib/texlab.rake

@@ -0,0 +1,7 @@
+require "rubygems"
+require "rake4latex"
+
+rule ".tex" => ".texlab" do |t|
+  sh %("#{File.expand_path(__FILE__+"/../../bin/texlab-compile")}" "#{t.source}" > "#{t.name}")
+end
+

data/lib/texlab.rb

@@ -0,0 +1,20 @@
+# To be sourced in VIM at startup
+
+# Set up path
+$: << (ENV["TEXLAB"]+"/lib") << (ENV["TEXLAB"]+"/lib/ruby") 
+Dir.glob(ENV["TEXLAB"]+"/gems/**/lib").each do |dir|
+  $: << dir
+end
+
+
+require "rubygems"
+#require "irb"
+require "yaml"
+require "rake4latex"
+
+
+task :compile do
+  Rake::Task[VIM::Buffer.current.name.sub(/\.[^.]+\z/, ".pdf")].invoke
+end
+
+

data/lib/texlab/boot.rb

@@ -0,0 +1,373 @@
+# This is the boot file to be loaded before parsing a .texlab file
+# encoding: ASCII-8bit
+require "easystats"
+require "plusminus"
+require "to_latex"
+require "gnuplot"
+require "open3"
+
+# tweak Array
+class Array
+  def extract_options!
+    if last.is_a? Hash
+      pop
+    else
+      {}
+    end
+  end
+end
+
+# tweak Hash
+class Hash
+  # Object-like behavior
+  def method_missing name, *args
+    name_string = name.to_s
+    case name_string[-1]
+    when "="
+      self[name_string[0..-2].to_sym] = args[0]
+    when "!"
+      self[name_string[0..-2].to_sym] = {}
+    when "?"
+      !! self[name_string[0..-2]]
+    else
+      self[name]
+    end
+  end
+end
+
+module Enumerable
+  # map-hash
+  def mash
+    res = {}
+    each do |*x|
+      k,v = yield(*x)
+      res[k] = v
+    end
+    res
+  end
+end
+
+class Hash
+  def mash! &block
+    replace mash(&block)
+  end
+end
+
+
+# Average method for array with uncertainty calculation
+class Array
+  def average
+    m = mean
+    m.pm(Math.sqrt(m.delta**2 + (standard_deviation*3)**2))
+  end
+end
+
+
+# latex generation helper
+require "texlab/texlabfile"
+
+$_latexfile = TexlabFile.new $_erbout
+
+def documentHeader *args
+  $_latexfile.documentHeader *args
+end
+
+def env *args
+  $_latexfile.env(*args){yield}
+end
+
+def puts string
+  $_erbout << string.to_latex << "\n"
+end
+
+# table generation
+#
+
+def table title, settings={}
+  @_table = {}
+  yield
+  entries = @_table
+
+  unless @_tables
+    $_latexfile.table(entries, title, settings)
+  else
+    @_tables << [entries, title, settings]
+  end
+end
+
+def column title
+  @_table_columns << (@_table_columns.last[title] || {})
+
+  result = yield
+  subcolumns = @_table_columns.pop
+
+  unless subcolumns.empty?
+    result = subcolumns
+  end
+
+  @_table_columns.last[title] = result
+end
+
+def row title
+  @_table_columns = [@_table[title] || {}]
+  yield
+  @_table[title] = @_table_columns.first
+end
+
+def tables *args
+  opts = args.extract_options!
+  caption = args.shift
+
+  @_tables = []
+  yield
+
+  $_latexfile.tables(@_tables, caption, opts)
+
+  @_tables = nil
+end
+
+
+# Figure generation
+#
+
+def figure *args
+  opts = args.extract_options!
+  caption = args.shift
+  format = opts[:width] ? "width=#{opts[:width]}" : opts[:format]
+  label = opts[:label]
+  filename = opts[:filename]
+  if @_figures
+    @_figures << Figure.new(format, filename, caption, label)
+  else
+    raise "not implemented"
+  end
+end
+
+def figures *args
+  opts = args.extract_options!
+  placement = opts.delete(:placement) || "htbp"
+  label = opts.delete(:label)
+  newPageThreshold = opts.delete(:newPageThreshold) || 29
+  caption = args.shift
+
+  @_figures = []
+  yield
+
+  $_latexfile.figures(caption, @_figures, newPageThreshold, placement, label)
+
+  @_figures = nil
+end
+
+
+# plotting
+#
+
+require "gnuplot"
+
+# override to have "w+" mode
+def Gnuplot.open( persist=true )
+  cmd = Gnuplot.gnuplot( persist ) or raise 'gnuplot not found'
+  IO::popen( cmd, "w+") { |io| yield io }
+end 
+
+def plot *args
+  opts = args.extract_options!
+  title = args[0]
+
+  placement = opts.delete(:placement) || "htbp"
+  width   = opts.delete(:width) || "17cm"
+  height = opts.delete(:height) || "10cm"
+  cmd    = opts.delete(:cmd) || "plot"
+
+  env :figure, "[#{placement}]" do
+    env :center do
+      @_datasets = []
+      yield
+      @_datasets
+
+      gnuplot = Gnuplot.gnuplot(true) or raise "gnuplot not found"
+      Open3.popen3(gnuplot) do |gp, out, err, external|
+        gp <<<<-GP
+          set terminal latex size #{width}, #{height}
+        GP
+
+        Gnuplot::Plot.new(gp, cmd) do |plot|
+          opts.each do |key, value|
+            case value
+            when true
+              plot.send key
+            else
+              plot.send key, value.gsub(/([\\ ])/, "\\\\\\1")
+            end
+          end
+
+          @_datasets.each do |ds|
+            plot.data << ds
+          end
+        end
+        gp.close
+        $_erbout << out.readlines.join("\n")
+
+        if not external.value.success?
+          errlines = err.readlines
+          raise "could not plot:\n" + errlines.join("\n")
+        end
+      end
+    end
+    $_erbout << "\\caption{#{title}}" if title
+  end
+end
+
+def splot *args, &block
+  opts = args.extract_options!
+  opts.cmd = "splot"
+  plot *args, opts, &block
+end
+
+def dataset *args
+  opts = args.extract_options!
+
+  data = args.shift
+  data_is_expr = !!data
+  if data
+    # simple function
+    ds = Gnuplot::DataSet.new data
+  else
+    # data
+    @_datastrings = []
+    yield
+
+    c = @_datastrings.first.count
+    data = (0...c).map do |i|
+      @_datastrings.map{|t|t[i]}
+    end
+
+    ds = Gnuplot::DataSet.new data
+  end
+  
+  # check args
+  raise ArgumentError, "Unnecessary parameters: #{args}" unless args.empty?
+
+  # tweak args (syntax candy)
+  opts[:title] = nil unless data_is_expr or opts.has_key? :title
+  if opts.key? :title and not opts[:title]
+    opts.delete(:title)
+    opts[:notitle] = true
+  end
+
+  # send options
+  opts.each do |key, value|
+    case value
+    when true
+      ds.send key
+    else
+      ds.send :"#{key}=", value
+    end
+  end
+
+  @_datasets << ds
+
+end
+
+def datastring arg
+  @_datastrings << arg
+end
+
+def data *args
+  datastring args
+end
+
+
+# fitting
+def fit expr, opts={}
+  raise "via: is a necessary parameter" if not opts[:via]
+  via = opts[:via]
+  vars = via.split(",")
+
+  using = opts[:using]
+
+  @_datasets = []
+
+  dataset do 
+    yield
+  end
+
+  ds = @_datasets.first
+
+  errlines = []
+  result = {}
+
+  gnuplot = Gnuplot.gnuplot(true) or raise "gnuplot not found"
+  Open3.popen3(gnuplot) do |gp, out, err, external|
+    vars.each do |var|
+      gp << "#{var} = #{rand}\n"
+    end
+    gp <<<<-GP
+      fit #{expr} '-' #{"using " + using if using} via #{via}
+    GP
+    gp << ds.to_gplot
+    gp.close
+    
+    errlines = err.readlines
+    errlines.each do |l|
+      if l =~ /\A([a-zA-Z_][a-zA-Z0-9_]*)\s*=\s*([0-9eE.+-]+)\s*\+\/-\s*([0-9eE.+-]+)/
+        result[$1.to_sym] = $2.to_f.pm($3.to_f)
+      end
+    end
+  end
+
+  vars.map do |v|
+    result[v.to_sym] or raise "could not fit:\n" + errlines.join("\n")
+  end
+end
+
+
+# macros
+
+$_macros={}
+
+# a macro is available in tex (\asdf) and in ruby (:asdf.to_latex or $asdf). It will go to math
+# mode
+def macro hash
+  text_macro hash.mash{|key, value| [key, value.latex!.to_latex_math] }
+end
+
+# a text macro is available in tex (\asdf) and in ruby (:asdf). It will not be
+# forced to math mode 
+def text_macro hash
+  hash.each do |key, value|
+    $_macros[key] = value
+    eval "$#{key} = '#{value.gsub("\\","\\\\").gsub("'", "\\'")}'"
+    $_erbout << "\\def\\#{key}{#{value}}"
+  end
+end
+
+class Symbol
+  # symbol's to_latex is the macro with the same name
+  def to_latex
+    $_macros[self] or raise "Undefined macro: #{self}"
+  end
+
+  def + str
+    "#{to_latex}#{str}".latex!.to_latex_math
+  end
+end
+
+
+# debug
+def debug *args
+  args.each {|a| STDERR.puts a.inspect}
+  return *args
+end
+
+# include Math
+include Math
+
+# degree radian
+def degrees x
+  x * 180.0 / Math::PI
+end
+
+def radians x
+  x / 180.0 * Math::PI
+end

data/lib/texlab/texlabfile.rb

@@ -0,0 +1,164 @@
+# encoding: ASCII-8bit
+require "latex"
+
+class TexlabFile < LatexFile
+  # create new instance.
+  # @param _erbout the string or file to write to
+  def initialize _erbout
+    # we don't call super
+
+    @_erbout = _erbout
+
+    # replay actions in super
+
+    @extras = {
+      :documentclass => :article,
+      :fontsize => "10pt"
+    }
+
+    #use defaults for unused entries:
+    @indent = 0 # indent lines in blocks
+
+    @usePackages = {
+      :inputenc=>:latin1,
+      :graphics=>:final,
+      :graphicx=>:pdftex,
+      :color=>:dvips,
+      :amsfonts=>true,
+      :subfigure=>true,
+      :lscape=>true,
+      :hyperref=>true,
+      :amsmath=>true,
+      :units=>true,
+      :float=>true,
+      :xcolor=>"table"
+      #:subfig=>["lofdepth", "lotdepth"]
+    }
+      
+    @default_table_align = 'c'
+
+    @lastWasPrint = false
+  end
+
+  def puts string
+    lines = string.split("\n")
+    for line in lines
+      @indent -= 2 if line[0,5]=='\\end{' && @indent >= 2
+      if @lastWasPrint
+        @_erbout << line.latex! << "\n"
+        @lastWasPrint = false
+      else
+        @_erbout << (" " * @indent).latex! << line.latex! << "\n"
+      end
+      @indent += 2 if line[0,7]=='\\begin{'
+    end
+  end
+
+  def print string
+    @lastWasPrint = true
+    @indent -= 2 if string[/\\end/]
+    @indent += 2 if string[/\\begin/]
+    @_erbout << string.latex! << "\n"
+  end
+
+  def prettyPrintCell x
+    x.to_latex
+  end
+
+  def documentHeader extras={}
+    @extras.merge! extras
+    @usePackages.merge! extras[:usePackages] || {}
+      
+    dcsettings = [@extras[:fontsize]]
+    dcsettings << "twocolumn" if @extras[:twocolumn]
+
+    puts "\\documentclass[#{dcsettings.join(",")}]{#{@extras[:documentclass]}}"
+
+    @usePackages.each do |package, settings|
+      case settings
+      when true then puts "\\usepackage{#{package}}"
+      when Array  then puts "\\usepackage[#{settings.join(",")}]{#{package}}"
+      else puts "\\usepackage[#{settings}]{#{package}}"
+      end
+    end
+    puts "\\addtolength{\\oddsidemargin}{-3.5cm}"
+    puts "\\addtolength{\\textwidth}{7cm}"
+    puts "\\addtolength{\\topmargin}{-3cm}"
+    puts "\\addtolength{\\textheight}{5cm}"
+    puts "\\newcommand{\\hide}[1]{}"
+
+    puts "\\special{landscape}" if @extras[:landscape]
+
+    #puts "\\begin{document}"
+    puts "\\DeclareGraphicsExtensions{.jpg,.pdf,.mps,.png}"
+  end
+  
+
+  def env(name, *args)
+    # overridden for compatibility with new Array#to_s
+    puts "\\begin{#{name}}#{args.join("")}"
+    yield if block_given?
+    puts "\\end{#{name}}"
+  end
+
+  def table entries, caption, args={}
+
+    # override defaults
+    args = {
+      :sort=> proc{0},
+      :header_hlines=>true
+    }.merge(args)
+
+
+    args[:rowTitle] = args[:rowTitle].to_latex if args[:rowTitle]
+
+    entries = convertKeysToLatex(entries)
+
+    super(entries, caption, args)
+  end
+
+  def plain_table entries, caption, opts={}
+    # capture output of table()
+    temp_erbout = @_erbout; @_erbout = ""
+    table(entries, caption, opts)
+    src = @_erbout; @_erbout = temp_erbout
+
+    src.sub!(/\A.*?\\begin{center}(.*)\\end{center}.*\z/m, '\1') or raise "internal error"
+    src.sub!(/\\caption{#{caption}}/, "") or raise "internal error"
+
+  end
+
+  def tables data, caption, opts={}
+    placement        = opts.delete(:placement) || "htbp"
+    tableLandscape   = opts.delete(:landscape)
+    tableFontSize    = opts.delete(:fontSize)  || @defaultTableFontSize
+    tableLabel       = opts.delete(:label)     || ""
+    tableCaption     = caption
+
+    puts "\\begin{landscape}" if tableLandscape
+    puts "\n\\begin{table}[#{placement}]\n\\begin{center}"
+    puts "\\#{tableFontSize}" if tableFontSize
+    
+    puts data.map { |args|
+      entries, caption, opts = *args
+      "\\subtable[#{caption}]{#{plain_table(*args)}}"
+    }.join("\\qquad")
+
+    puts "\\caption{#{tableCaption}}\n" if tableCaption
+    puts "\\label{tab:#{tableLabel}}" if tableLabel
+    puts "\\end{center}\n\\end{table}\n"
+    puts "\\end{landscape}" if @tableLandscape
+  end
+
+  private
+
+  def convertKeysToLatex value
+    case value
+    when Hash
+      value.mash{|k,v| [k.to_latex, convertKeysToLatex(v)] }
+    else
+      value
+    end
+  end
+
+end