Merge branch 'heat-maps'

Author: bakpakin

examples/heat-map.janet

@@ -0,0 +1,48 @@
+###
+### Heat maps in 2 ways - closures vs. dataframes
+###
+
+(import spork/charts)
+(import spork/gfx2d)
+
+###
+### The closure driven way
+###
+(def cmap (get charts/color-maps :turbo))
+(defn distfrom [px py] (fn [x y] (let [dx (- px x) dy (- py y)] (math/sqrt (+ (* dx dx) (* dy dy))))))
+(def d1 (distfrom 10 10))
+(def d2 (distfrom 20 18))
+(def d3 (distfrom 37 8))
+(charts/heat-map-chart
+  :width (* 2 1920) :height (* 2 1080)
+  :num-columns (* 1 48) :num-rows (* 1 27)
+  :cell-text-fn (fn [x y] (string/format "%d,%d" x y))
+  :color-fn (fn [x y]
+              (def t (min (* 0.1 (d1 x y)) (* 0.2 (d2 x y)) (* 0.03 (d3 x y))))
+              (cmap (+ (* 0.01 (math/random)) t)))
+  :box-gap 2
+  :font (gfx2d/load-font "examples/fonts/Roboto-Regular.ttf" 24)
+  :title "Heat Map Example"
+  :save-as "tmp/heat-map.png")
+(print "Made heat map at tmp/heat-map.png")
+
+###
+### The data-driven way
+###
+(def dfdata
+  {:a (range 0 1 0.04)
+   :b (range 0 1 0.03)
+   :c (range 0 1 0.035)
+   :d (range 0 2 0.029)
+   :e (range 0.2 1.4 0.0391)})
+(charts/dark-mode)
+(charts/heat-map-chart
+  :width 700 :height 700
+  :data dfdata
+  :legend :right
+  :legend-labels (seq [x :range [0 5]] (case x 0 "" 4 "" (string/format "%.2f" (/ x 4))))
+  :font (gfx2d/load-font "examples/fonts/Roboto-Regular.ttf" 24)
+  :title "Heat Map"
+  :color-map :viridis
+  :save-as "tmp/heat-map-dataframe.png")
+(print "Made heat map at tmp/heat-map-dataframe.png")

examples/multi-charts.janet

@@ -0,0 +1,39 @@
+###
+### See available color maps with heat charts
+###
+
+(import spork/charts)
+(import spork/gfx2d)
+
+(def dfdata
+  {:a (range 0 1 0.04)
+   :b (range 0 1 0.03)
+   :c (range 0 1 0.035)
+   :d (range 0 2 0.029)
+   :e (range 0.2 1.4 0.0391)})
+(def font (gfx2d/load-font "examples/fonts/Roboto-Regular.ttf" 16))
+(defn make-a-chart
+  [color-map]
+  (charts/heat-map-chart
+    :width 500 :height 500
+    :data dfdata
+    :legend :right
+    :legend-labels (seq [x :range [0 5]] (case x 0 "" 4 "" (string/format "%.2f" (/ x 4))))
+    :font font
+    :legend-width 20
+    :title (string ":" color-map)
+    :color-map color-map))
+
+# Make a grid of charts
+(def the-charts (map make-a-chart [:grayscale :turbo :viridis :magma]))
+(def outer-canvas (gfx2d/blank 1000 1100 4))
+(gfx2d/fill-rect outer-canvas 0 0 2000 2000 gfx2d/white)
+(def title "spork/charts color maps")
+(def [w h] (gfx2d/measure-text font title 3.0))
+(gfx2d/draw-text outer-canvas font (- 500 (div w 2)) 20 title gfx2d/black 3.0)
+(eachp [i chart] the-charts
+  (def x (if (even? i) 0 500))
+  (def y (if (> i 1) 100 600))
+  (gfx2d/stamp outer-canvas chart x y))
+(gfx2d/save "tmp/chart-grid.png" outer-canvas)
+(print "Wrote to tmp/chart-grid.png")