guide is set to colourbar

for continuous/gradient scales Fixes #26
Mikata-Project · Apr 16, 2018 · f0f2f7b · f0f2f7b
1 parent 0d9d350
commit f0f2f7b
Showing 1 changed file with 34 additions and 9 deletions.
diff --git a/R/theme_scales.R b/R/theme_scales.R
@@ -3,16 +3,41 @@ theme_scales <- function(palette) {
 
   colours <- palette$swatch[-1]
 
-  new_scale_functions <- list(
-    scale_colour_discrete   = function(...) discrete_scale(  'colour', 'ggthemr', discrete_colours(colours), ...),
-    scale_fill_discrete     = function(...) discrete_scale(  'fill',   'ggthemr', discrete_colours(colours), ...),
-    scale_colour_continuous = function(...) continuous_scale('colour', 'ggthemr', seq_gradient_pal(palette$gradient[['low']], palette$gradient[['high']]), ...),
-    scale_fill_continuous   = function(...) continuous_scale('fill',   'ggthemr', seq_gradient_pal(palette$gradient[['low']], palette$gradient[['high']]), ...),
-    scale_colour_gradient   = function(...) continuous_scale('colour', 'ggthemr', seq_gradient_pal(palette$gradient[['low']], palette$gradient[['high']]), ...),
-    scale_fill_gradient     = function(...) continuous_scale('fill',   'ggthemr', seq_gradient_pal(palette$gradient[['low']], palette$gradient[['high']]), ...)
-  )
+  scale_colour_discrete = function(...) discrete_scale('colour', 'ggthemr', discrete_colours(colours), ...)
+
+  scale_fill_discrete = function(...) discrete_scale('fill', 'ggthemr', discrete_colours(colours), ...)
+
+  scale_colour_continuous = function(...,
+                                     low = palette$gradient[['low']],
+                                     high = palette$gradient[['high']],
+                                     space = "Lab",
+                                     guide = "colourbar") {
+    continuous_scale('colour', 'ggthemr',
+                     seq_gradient_pal(low, high, space),
+                     guide = guide, ...)
+  }
 
-  new_scale_functions
+  scale_fill_continuous   = function(...,
+                                     low = palette$gradient[['low']],
+                                     high = palette$gradient[['high']],
+                                     space = "Lab",
+                                     guide = "colourbar") {
+    continuous_scale('fill', 'ggthemr',
+                     seq_gradient_pal(low, high, space),
+                     guide = guide, ...)
+  }
+
+  list(
+    scale_colour_discrete   = scale_colour_discrete,
+    scale_color_discrete    = scale_colour_discrete,
+    scale_fill_discrete     = scale_fill_discrete,
+    scale_colour_continuous = scale_colour_continuous,
+    scale_color_continuous  = scale_colour_continuous,
+    scale_colour_gradient   = scale_colour_continuous,
+    scale_color_gradient    = scale_colour_continuous,
+    scale_fill_continuous   = scale_fill_continuous,
+    scale_fill_gradient     = scale_fill_continuous
+  )
 }