texlab 0.1.1 → 0.2.0

data/Gemfile

@@ -1,11 +1,12 @@
 source "http://rubygems.org"
 
-gem "rake4latex"
-gem "plusminus"
-gem "easystats"
-gem "to_latex"
-gem "gnuplot"
-gem "latex"
+gem "rake4latex", "0.1.3"
+gem "plusminus", "0.2.0"
+gem "easystats", "0.1.0"
+gem "to_latex", "0.4.0"
+gem "gnuplot", "~> 2.4.1"
+gem "latex", "0.1.3"
+gem "nio-percent_fmt", "0.0.0"
 
 group :development do
   gem "rspec", "~> 2.8.0"

data/Gemfile.lock

@@ -17,7 +17,9 @@ GEM
     log4r (1.1.10)
     nio (0.2.5)
       flt (>= 1.0.0)
-    plusminus (0.1.0)
+    nio-percent_fmt (0.0.0)
+      nio
+    plusminus (0.2.0)
       nio
     rake (0.9.2.2)
     rake4latex (0.1.3)
@@ -42,13 +44,14 @@ PLATFORMS
 
 DEPENDENCIES
   bundler (~> 1.0.0)
-  easystats
-  gnuplot
+  easystats (= 0.1.0)
+  gnuplot (~> 2.4.1)
   jeweler (~> 1.8.3)
-  latex
-  plusminus
-  rake4latex
+  latex (= 0.1.3)
+  nio-percent_fmt (= 0.0.0)
+  plusminus (= 0.2.0)
+  rake4latex (= 0.1.3)
   rcov
   rdoc (~> 3.12)
   rspec (~> 2.8.0)
-  to_latex
+  to_latex (= 0.4.0)

data/Rakefile

@@ -22,6 +22,7 @@ Jeweler::Tasks.new do |gem|
   gem.email = "kallo.bernat@gmail.com"
   gem.authors = ["Bernát Kalló"]
   gem.executables = ["texlab-compile"]
+  gem.files = FileList["lib/**/*.rb", "bin/*", "[A-Z]*", "spec/**/*"].to_a
   # dependencies defined in Gemfile
 end
 Jeweler::RubygemsDotOrgTasks.new

data/VERSION

@@ -1 +1 @@
-0.1.1
+0.2.0

data/bin/texlab-compile

@@ -18,11 +18,11 @@ begin
   infile = ARGV[0] || fail
   outfile = ARGV[1] || infile.sub(/(\.texlab)?$/, ".tex")
 
-  src = nil
+  content = nil
   File.open infile do |f|
-    src = f.read
+    content = f.read
   end
-  fail if !src
+  fail if !content
 
 rescue
         STDERR.puts File.basename($0) + " <inputfile> <outputfile>"
@@ -32,13 +32,25 @@ EOU
 
 else
 
-  # load boot code
-  File.open File.expand_path(__FILE__+"/../../lib/texlab/boot.rb") do |f|
-    src = "<%\n" << f.read << "\n%>\n" << src
+  src = "<% $_binding=binding %>"
+
+  # load modules
+
+  %w(hash latex debug figure macro math plot stats table tikz).each do |f|
+    f = File.expand_path(__FILE__+"/../../lib/texlab/#{f}.rb")
+    File.open f do |file|
+      s = file.read
+      s.gsub!(/([\\#"])/,"\\\\\\1"); s.gsub!(/\n/, "\\\\n")
+      f.gsub!(/([\\#"])/,"\\\\\\1"); f.gsub!(/\n/, "\\\\n")
+      src << "<% eval(\"#{s}\",binding,\"#{f}\") %>" 
+    end
   end
 
+  src << content
+
   # run erb
   erb = TEXERB.new(src.untaint, nil, ">", "$_erbout")
+  erb.filename = infile
 
   File.open outfile, "w" do |f|
     f.print erb.result(TOPLEVEL_BINDING.taint)

data/lib/texlab/debug.rb

@@ -0,0 +1,7 @@
+
+# debug
+def debug *args
+  args.each {|a| STDERR.puts a.inspect}
+  return *args
+end
+

data/lib/texlab/figure.rb

@@ -0,0 +1,32 @@
+# 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
+
+

data/lib/texlab/hash.rb

@@ -0,0 +1,52 @@
+
+# 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
+
+  def slice *args
+    args.mash{|k| self[k]} 
+  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
+

data/lib/texlab/latex.rb

@@ -0,0 +1,22 @@
+# latex generation helper
+require "plusminus"
+require "to_latex"
+require "nio/percent_fmt"
+
+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
+
+

data/lib/texlab/macro.rb

@@ -0,0 +1,101 @@
+# 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
+  hash.each do |key, value|
+    value = value.latex!.to_latex_math
+    define_latex_macro key, value
+    define_symbol_macro key, value
+    define_global_macro key, value
+  end
+end
+
+# a text macro is available in tex (\asdf) and in ruby (:asdf, $asdf). It will not be
+# forced to math mode 
+def text_macro hash
+  hash.each do |key, value|
+    value = "\\text{#{value}}".latex!
+    define_latex_macro key, value
+    define_symbol_macro key, value
+    define_global_macro key, value
+  end
+end
+
+def define_symbol_macro key, value
+  $_macros[key] = value
+end
+
+def define_latex_macro key, value
+  $_erbout << "\\def\\#{key}{#{value}}"
+end
+
+def define_global_macro key, value
+  eval "$#{key} = '#{value.gsub("\\","\\\\").gsub("'", "\\'")}'"
+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
+
+  def * unit
+    "\\unit[\\text{#{to_latex}}]{#{unit}}".latex!.to_latex_math
+  end
+end
+
+
+
+
+%w(
+  alpha beta gamma delta epsilon zeta eta theta iota kappa lambda mu nu xi omicron pi rho sigma tau upsilon phi chi psi omega
+).each do |a|
+  [a, a.capitalize].each do |b|
+    key = b.to_sym
+    value = "\\#{b}".latex!.to_latex_math
+    define_symbol_macro key, value
+    define_global_macro key, value
+  end
+end
+
+('a'..'z').each do |a|
+  [a, a.capitalize].each do |b|
+    key = b.to_sym
+    value = "#{b}".latex!.to_latex_math
+    define_symbol_macro key, value
+    define_global_macro key, value
+  end
+end
+
+
+# units
+$_units = {}
+
+def unit hash
+  # allow array keys to be DRY
+  hash.each do |ks,v|
+    [*ks].each do |k|
+      $_units[k] = v
+    end
+  end
+end
+
+class Symbol
+  def with_unit
+    raise "No unit defined for #{self}" unless $_units.key? self
+    if $_units[self]
+      self * "(#{$_units[self]})"
+    else
+      to_latex
+    end
+  end
+end
+
+

data/lib/texlab/math.rb

@@ -0,0 +1,13 @@
+# 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/plot.rb

@@ -0,0 +1,177 @@
+# plotting
+#
+
+require "gnuplot"
+require "open3"
+
+# 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) || (opts[:label] ? "htbp" : "H")
+  width   = opts.delete(:width) || "17cm"
+  height = opts.delete(:height) || "10cm"
+  cmd    = opts.delete(:cmd) || "plot"
+  label = opts.delete(:label)
+  debug = opts.delete(:debug)
+
+  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 = STDERR if debug
+        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
+            when false
+              gp << "unset #{key}\n"
+            else
+              plot.send key, value.to_s.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
+    $_erbout << "\\label{#{label}}" if label
+  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 do |t|
+        case value = t[i]
+        when String
+          "\"#{value.to_s.gsub(/([\\"])/, "\\\\\\1")}\""
+        else
+          value.to_s
+        end
+      end
+    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
+
+