scales methods v0

dreamRs · Aug 28, 2018 · 6d17c67 · 6d17c67
1 parent 1c89c1c
commit 6d17c67
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/R/chroma-class.R b/R/chroma-class.R
@@ -205,6 +205,30 @@ chroma <- R6::R6Class(
       return(res)
     },
 
+    # Scales
+    scale = function(colors) {
+      not_initialized(private$initialized)
+      private$chroma$scale <- glue::glue(
+        "scale([{colors}])",
+        colors = stri_c(glue::single_quote(colors), collapse = ", ")
+      )
+    },
+    mode = function(mode = c("rgb", "lab", "lrgb", "hsl", "lch")) {
+      mode <- match.arg(mode)
+      private$chroma$mode <- glue::glue("mode('{mode}')")
+    },
+    correctLightness = function() {
+      private$chroma$correctLightness <- "correctLightness()"
+    },
+    colors = function(n) {
+      if (length(n) == 1) {
+        private$chroma$colors <- glue::glue("colors({n})")
+      } else {
+        private$chroma$colors <- glue::glue("([{n}])", n = stri_c(n, collapse = ", "))
+      }
+    },
+
+
     # R methods
     print = function() {
       code <- private$chroma