1 files changed, 79 insertions, 0 deletions
diff --git a/examples/example-radon/Plotting.hs b/examples/example-radon/Plotting.hs
new file mode 100644
index 0000000..f4522da
--- /dev/null
+++ b/examples/example-radon/Plotting.hs
@@ -0,0 +1,79 @@
+{-# LANGUAGE OverloadedStrings #-}
+{-# LANGUAGE ExistentialQuantification #-}
+
+module Plotting where
+
+import Data.Aeson
+import Graphics.Vega.VegaLite
+import Data.Text (Text, pack)
+
+
+barPlot :: Text -> VLSpec
+barPlot xName = 
+    let enc = encoding
+            . position X [PName xName, PmType Nominal, PAxis [AxGrid True, AxTitle xName]]
+            . position Y [PName "binnedData", PAggregate Count, PmType Quantitative, PAxis [AxGrid False, AxTitle "count"]]
+    in asSpec $ [mark Bar [MOpacity 1.0, MColor "#a3c6de"], enc []]
+    
+barPlot2 :: Text -> Text -> VLSpec
+barPlot2 xName yName = 
+    let enc = encoding
+            . position X [PName xName, PmType Nominal, PAxis [AxGrid True, AxTitle xName]]
+            . position Y [PName yName, PmType Quantitative, PAxis [AxGrid False, AxTitle yName]]
+    in asSpec $ [mark Bar [MOpacity 1.0, MColor "#a3c6de"], enc []]
+    
+linePlot :: Text -> Text -> VLSpec
+linePlot xName yName = 
+  let enc = encoding
+            . position X [PName xName, PmType Quantitative, PAxis [AxGrid True, AxTitle xName]]
+            . position Y [PName yName, PmType Quantitative, PAxis [AxGrid False, AxTitle yName]]
+  in asSpec $ [mark Line [MColor "green"], enc []]
+
+density2DPlot :: Text -> Text -> (Double, Double) -> (Double, Double) -> VLSpec
+density2DPlot xName yName (xmin, xmax) (ymin, ymax) = 
+  let enc = encoding
+            . position X [PName xName
+                         ,PScale [SDomain (DNumbers [xmin, xmax])]
+                         ,PBin [Step 0.1], PmType Quantitative, PAxis [AxGrid True, AxTitle xName]]
+            . position Y [PName yName
+                         ,PScale [SDomain (DNumbers [ymin, ymax])]
+                         ,PBin [Step 0.1]
+                         ,PmType Quantitative
+                         ,PAxis [AxGrid True, AxTitle yName]]
+            . color [ MAggregate Count, MName "col", MmType Quantitative
+                    , MScale [{-SReverse False,-} SScheme "blues" [0.0, 1.0]]]
+  in asSpec $ [mark Rect [MClip True], enc []]
+  
+scatter2 :: Text -> Text -> (Double, Double) -> VLSpec
+scatter2 xName yName (ymin, ymax) = 
+  let enc = encoding
+            . position X [PName xName, PmType Nominal, PAxis [AxGrid True, AxTitle xName]]
+            . position Y [PName yName
+                         , PScale [SDomain (DNumbers [ymin, ymax])]
+                         , PmType Quantitative
+                         , PAxis [AxGrid True, AxTitle yName]]
+  in asSpec $ [mark Tick [MClip True], enc []]
+
+data SpecGrid = H [[VLSpec]] | V [[VLSpec]] | L [VLSpec] | S VLSpec
+
+data InputData = Cols [(Text, DataValues)]
+               | forall j. (ToJSON j) => J j
+               | File FilePath
+
+plot :: (Double, Double) -> SpecGrid -> InputData -> VegaLite
+plot (figw,figh) gridOfLayers dat =
+    let desc = description "Plot"
+        dat' = case dat of
+          Cols cols -> foldl (.) (dataFromColumns []) (map (uncurry dataColumn) cols) []
+          J o -> dataFromJson (toJSON o) []
+          File fp -> dataFromSource (pack fp) []
+        conf = configure
+            -- . configuration (Axis [ DomainWidth 1 ])
+            . configuration (SelectionStyle [ ( Single, [ On "dblclick" ] ) ])
+            . configuration (View [ViewStroke (Just "transparent")])
+        spec = case gridOfLayers of
+          S l -> layer [l]
+          L ls -> layer ls
+          H lss -> hConcat (map (asSpec . (:[]) . layer) lss)
+          V lss -> vConcat (map (asSpec . (:[]) . layer) lss)
+    in toVegaLite [width figw, height figh, conf [], desc, dat', spec]