Full color support

Cargo.toml

@@ -32,6 +32,7 @@ pretty_toa = "1.0.0"
 str_stack = "0.1"
 addr2line = "0.8"
 indexmap = "1.0"
+rand = "0.6.5"
 
 [dev-dependencies]
 pretty_assertions = "0.5"

src/bin/flamegraph.rs

@@ -3,7 +3,7 @@ use std::io::{self, BufReader, Read};
 use std::path::PathBuf;
 use structopt::StructOpt;
 
-use inferno::flamegraph::{self, Direction, FuncFrameAttrsMap, Options};
+use inferno::flamegraph::{self, BackgroundColor, Direction, FuncFrameAttrsMap, Options, Palette};
 
 #[derive(Debug, StructOpt)]
 #[structopt(name = "inferno-flamegraph", author = "")]
@@ -21,6 +21,25 @@ struct Opt {
     /// Collapsed perf output files. With no INFILE, or INFILE is -, read STDIN.
     #[structopt(name = "INFILE", parse(from_os_str))]
     infiles: Vec<PathBuf>,
+    /// set color palette
+    #[structopt(
+        short = "c",
+        long = "colors",
+        default_value = "hot",
+        raw(
+            possible_values = r#"&["hot","mem","io","wakeup","java","js","perl","red","green","blue","aqua","yellow","purple","orange"]"#
+        )
+    )]
+    colors: Palette,
+    /// set background colors. Gradient choices are yellow (default), blue, green, grey; flat colors use "#rrggbb"
+    #[structopt(long = "bgcolors")]
+    bgcolors: Option<BackgroundColor>,
+    /// colors are keyed by function name hash
+    #[structopt(long = "hash")]
+    hash: bool,
+    /// use consistent palette (palette.map)
+    #[structopt(long = "cp")]
+    cp: bool,
 }
 
 impl Into<Options> for Opt {
@@ -43,6 +62,10 @@ impl Into<Options> for Opt {
             "Flame Graph".to_string()
         };
         Options {
+            colors: self.colors,
+            bgcolors: self.bgcolors,
+            hash: self.hash,
+            consistent_palette: self.cp,
             func_frameattrs,
             direction,
             title,