Allow strings in datasets (#11)

* Allow spaces in strings in datasets

Useful for labels
Closes #10

Author: devstopfix

README.md

@@ -100,7 +100,7 @@ by adding `gnuplot` to your list of dependencies in `mix.exs`:
 ```elixir
 def deps do
   [
-    {:gnuplot, "~> 0.19.87"}
+    {:gnuplot, "~> 1.19.88"}
   ]
 end
 ```

examples/so327576.exs

@@ -4,7 +4,6 @@ defmodule BarChart do
   @moduledoc "Chart from https://stackoverflow.com/a/11551808/3366"
 
   def run do
-
     chart = [
       [:set, :term, :png, :size, '512,512'],
       [:set, :output, Path.join("/tmp", "barchart.PNG")],
@@ -13,11 +12,11 @@ defmodule BarChart do
       [:plot, "-", :using, '1:3:xtic(2)', :with, :boxes]
     ]
 
-    dataset = [[0, "label", 100], [1, "label2", 450], [2, "bar_label", 75]]
+    dataset = [[0, "label", 100], [1, "label2", 450], [2, "bar label", 75]]
 
     plot(chart, [dataset])
   end
 end
 
-# mix run examples/so327576.exs
+# mix run examples/so327576.exs && open /tmp/barchart.PNG
 BarChart.run()

lib/gnuplot/dataset.ex

@@ -31,7 +31,19 @@ defmodule Gnuplot.Dataset do
 
   defp format_point(point) do
     point
-    |> Enum.map(&Kernel.to_string/1)
+    |> Enum.map(&to_str/1)
     |> Enum.join(" ")
   end
+
+  defp to_str(f) when is_float(f), do: Float.to_string(f)
+  defp to_str(i) when is_integer(i), do: Integer.to_string(i)
+  defp to_str(s) when is_binary(s) do
+    if contains_space?(s) do
+      "\"" <> s <> "\""
+    else
+      s
+    end
+  end
+
+  def contains_space?(s), do: String.contains?(s, " ")
 end

mix.exs

@@ -4,7 +4,7 @@ defmodule Gnuplot.MixProject do
   def project do
     [
       app: :gnuplot,
-      version: "0.19.87",
+      version: "1.19.88",
       elixir: "~> 1.6",
       start_permanent: Mix.env() == :prod,
       description: "Interface between Elixir and Gnuplot graphing library",
@@ -41,7 +41,7 @@ defmodule Gnuplot.MixProject do
       licenses: ["EPL-2.0"],
       links: %{
         "GitHub" => "https://github.com/devstopfix/gnuplot-elixir",
-        "Travis" => "https://travis-ci.org/devstopfix/gnuplot-elixir"
+        "Travis CI" => "https://travis-ci.org/devstopfix/gnuplot-elixir"
       }
     ]
   end

test/gnuplot_test.exs

@@ -83,4 +83,11 @@ defmodule GnuplotTest do
     expected = "set xrange [-3:3];\nset yrange [-3:3];\nsplot sin(x) * cos(y)"
     assert {:ok, expected} == G.plot(plot)
   end
+
+  test "Strings with spaces in datasets" do
+    input = [[0, "label", 100], [1, "label2", 450], [2, "bar label", 75]]
+    expected = ["0 label 100", "\n", "1 label2 450", "\n", "2 \"bar label\" 75", "\ne\n"]
+    assert expected ==
+             [input] |> D.format_datasets() |> Enum.to_list()
+  end
 end