plotrobber 0.0.2 → 0.0.3

data/README.md

@@ -39,6 +39,15 @@ Do write this:
       }
     }
 
+Even more convenient:
+
+    plot('Fit some data') {
+      fn = fit SOME_DATA, [1,3] # Find linear fit using columns 1 and 3.
+
+      show fn                   # Plots the line of best fit
+      candlesticks SOME_DATA    # Plots candlesticks using 1:3:2:6:5 
+    }
+
 ## Installation
 
 The usual:

data/lib/plotrobber/version.rb

@@ -1,3 +1,3 @@
 module Plotrobber
-  VERSION = "0.0.2"
+  VERSION = "0.0.3"
 end

data/lib/plotrobber.rb

@@ -22,6 +22,32 @@ class Gnuplot::Plot
     arbitrary_lines << s
   end
 
+  def store(data)
+    (@store ||= {})[data] ||= begin
+      file = Tempfile.new(['plotrobber.','.dat'])
+      file.write data.map{|x|x.join("\t")}.join("\n")
+      file.write "\n"
+      file.rewind
+
+      file
+    end
+  end
+
+  def source_name(what)
+    name = case what
+      when String         then what
+      when Array          then store(what).path
+      when Tempfile, File then what.path
+      else                     what.to_s
+    end
+
+    unless name.start_with?("'") or name['(']
+      name = "'#{name}'"
+    end
+    
+    name
+  end
+
   def candlesticks(name, extra_opts = {})
     with_opt = case extra_opts[:with]
       when Hash, Array then 'candlesticks ' + extra_opts[:with].to_a.join(' ')
@@ -34,21 +60,24 @@ class Gnuplot::Plot
     show name, options
   end
 
-  def show(name, *args)
-    if Array === name
-      file = Tempfile.new(['plotrobber.','.dat'])
-      file.write name.map{|x|x.join("\t")}.join("\n")
-      file.close
+  def fit(what, using, use_b = true)
+    name = source_name what
 
-      @files ||= []
-      @files << file
+    @next_fn ||= 0
+    @next_fn += 1
 
-      name = file.path
-    end
+    f, m, b = %w[f m b].map {|x| "#{x}#{@next_fn}" }
+    using_str = using.join(':')
+    via_str = use_b ? "#{m}, #{b}" : m
 
-    unless name.start_with? "'"
-      name = "'#{name}'"
-    end
+    please '%s(x) = %s*x + %s' % [f, m, use_b ? b : 0]
+    please 'fit %s(x) %s using %s via %s' % [f, name, using_str, via_str]
+
+    "#{f}(x)"
+  end
+
+  def show(what, *args)
+    name = source_name what
 
     self.data << Gnuplot::DataSet.new(name) do |ds|
       for arg in args

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: plotrobber
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-03-13 00:00:00.000000000 Z
+date: 2013-03-14 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: gnuplot