ntree colors

src/tree_repr.py

@@ -1,6 +1,6 @@
 import os
 
-# from colors import RED, YELLOW, GREEN, CYAN , BLUE, PURPLE
+from src.colors import RED, YELLOW, GREEN, CYAN , BLUE, PURPLE
 
 EMPTY_TREE_LINE_LENGTH = 160
 EMPTY_TREE_LINE = ' ' * EMPTY_TREE_LINE_LENGTH
@@ -71,3 +71,15 @@ def format(num, unit=None):
   # human_number = num
   
   return human_number, unit
+
+
+def get_color_formatter(t):
+  '''
+    @param        t      string can be 'd', 'f', 'l', 'x'
+  '''
+  if t == 'd':
+    return BLUE
+  if t == 'x':
+    return GREEN
+  return None
+