visualisation-utils 0.4 → 0.5

data/README.md

@@ -14,7 +14,7 @@ scatter-plot
 ============
 
 Prints scatter plots of one or more columns of whitespace separated data using
-gnuplot as a backend
+gnuplot as a backend.
 
 
 ~~~ .bash
@@ -31,16 +31,41 @@ This incantation will bring up the gnuplot gui with the following graph:
 
 ![gui](https://raw.github.com/programmiersportgruppe/visualisation-utils/master/doc/gnuplot-gui.png)
 
+heat-map
+=====-==
+
+Prints a heat map visualising the distribution of
+geo coordinate samples.
+
+~~~ .bash
+cat | < | heat-map <<'END'
+38.6,-90.5
+38.6,-90.5
+40.5,-74.3
+34.4,-92.2
+42.5,-83.0
+34.1,-92.0
+33.3,-111.9
+34.1,-91.8
+34.2,-86.8
+END
+~~~
+
+This will render to the following visualisation:
+
+![map](https://raw.github.com/programmiersportgruppe/visualisation-utils/master/doc/heat-map.png)
+
+
 TODO
 ====
 
-* Overriding of autoscale
-* Create nice PNGs
-* Line plots
-* Changing changing the size of the dots
-* Secondary y-axis
-* Bar chart single series
-* Bar chart multiple series
-* Discrete y-values, e.g. to visualise events over time
-* Histogram plotting
+[X] Overriding of autoscale
+[X] Create nice PNGs
+[X] Line plots
+[ ] Changing changing the size of the dots
+[X] Secondary y-axis
+[ ] Bar chart single series
+[ ] Bar chart multiple series
+[ ] Discrete y-values, e.g. to visualise events over time
+[ ] Histogram plotting
 

data/bin/heat-map

@@ -0,0 +1,125 @@
+#!/usr/bin/env ruby
+require 'trollop'
+
+DATA_FILE_NAME="/tmp/plot-"+`date  "+%s"`.strip+".dat"
+PLOT_FILE_NAME="/tmp/plot-"+`date  "+%s"`.strip+".plt"
+
+
+@opts = Trollop::options do
+    banner "heat-map takes comma and newline separated table of coordinate from stdin and renders them using gnuplot"
+    opt :outfile, "Output file name", :type => String, :short => 'o'
+    opt :debug, "debug"
+    opt :title, "Title", :type=> String, :short => 't'
+    opt :font, "Font", :type => String, :short => 'f'
+    opt :extra_header, "Extra commands to be put in the header of the gnuplot file", :type => String
+end
+
+title = @opts[:title] || ""
+
+# TODO I do want to be streamed!
+input = `cat`
+
+freqs = input.split("\n").map{|line|
+    line.split(",").map{|number | number.to_f.round(0)}.join(",")
+}
+.group_by{|coords| coords}
+.map{|pair| [pair[0], pair[1].length]}
+.inject({}) { |memo, obj|
+   memo[obj.first] = obj.last
+   memo
+}
+
+freq_values = freqs.values
+max_val = freq_values.max
+min_val = freq_values.min
+mid_val = (min_val + (max_val - min_val)/2).round(1)
+
+puts "Min. value: #{min_val}, max value: #{max_val}"
+
+full_map = (-179..179).map{|lon|
+   (-89..89).map{|lat|
+       key = "#{lat},#{lon}"
+       "#{lat} #{lon} #{freqs[key]||0}"
+   }
+}.join("\n")
+
+
+File.open(DATA_FILE_NAME, "w"){|f| f.write(full_map)}
+
+font = @opts[:font] || "Futura"
+font_size = "12"
+
+extra_header = @opts[:extra_header] || ""
+
+if (@opts[:outfile])
+    filename = @opts[:outfile]
+    extension = filename.gsub(/[^.]*\./,"")
+    if (extension == "png")
+        terminal=<<EOF
+            set term pngcairo font '#{font},#{font_size}' transparent size 1200,800
+            set output '#{filename}'
+
+EOF
+    elsif (extension == "eps")
+        terminal=<<EOF
+            set term epscairo size 1200,800 font '#{font},#{font_size}'
+            set output '#{filename}'
+EOF
+    elsif (extension == "pdf")
+        terminal=<<EOF
+            set term pdfcairo size 22cm,14cm font '#{font},#{font_size}'
+            set output '#{filename}'
+EOF
+    elsif (extension == "svg")
+        terminal=<<EOF
+            set term svg size 1200,800 font '#{font},#{font_size}'
+            set output '#{filename}'
+EOF
+    else
+        throw "Unknown output format '.#{extension}'."
+    end
+
+
+end
+
+
+world_dat=File.join(File.dirname(File.expand_path(__FILE__)), '/../world.dat')
+
+
+script = <<EOF
+
+
+
+#{terminal}
+
+
+set title "#{title}"
+
+# color definitions
+set border lw 1.5
+set style line 1 lc rgb 'black' lt 1 lw 2
+
+set rmargin screen 0.85
+
+unset key
+set tics scale 0.5
+unset xtics
+unset ytics
+set xrange[-179:179]
+set yrange[-89:89]
+#set format '%g'
+set palette defined (0 "white",#{min_val} "#00ffff",#{mid_val} "yellow",#{max_val} "red")
+plot '#{DATA_FILE_NAME}' u 2:1:3 w image, \
+     '#{world_dat}' with lines linestyle 1
+EOF
+
+
+puts "data file " + DATA_FILE_NAME
+puts "plot file " + PLOT_FILE_NAME
+
+
+
+File.open(PLOT_FILE_NAME, "w"){|f| f.write(script)}
+
+`gnuplot #{PLOT_FILE_NAME}`
+

data/visualisation-utils.gemspec

@@ -3,7 +3,7 @@ Gem::Specification.new do |s|
   s.summary      = 'utilities for quick visualisation'
   s.description  = 'visualisation-utils provides a number of utilities
                     for visualising data from the command line.'
-  s.version      = '0.4'
+  s.version      = '0.5'
   s.platform     = Gem::Platform::RUBY
 
   s.files        = ['bin/scatter-plot']