convert COLOR_ARG macro to a function

Signed-off-by: Ildar Muslukhov <ildarm@google.com>