set_palette.py

Go to the documentation of this file.

#!/usr/bin/env python
import os, sys, re
import glob
import string
from array import array 
from pprint import pprint
import ROOT


def set_palette(name="palette", ncontours=999):
    """Set a color palette from a given RGB list
    stops, red, green and blue should all be lists of the same length
    see set_decent_colors for an example"""

    if name == "gray" or name == "grayscale":
        stops = [0.00, 0.34, 0.61, 0.84, 1.00]
        red   = [1.00, 0.84, 0.61, 0.34, 0.00]
        green = [1.00, 0.84, 0.61, 0.34, 0.00]
        blue  = [1.00, 0.84, 0.61, 0.34, 0.00]
    elif name == "susy":
    #NRGBs = 5
     #   NCont = 255
        stops = [0.00, 0.34, 0.61, 0.84, 1.00]
        red = [0.50, 0.50, 1.00, 1.00, 1.00]
        green = [ 0.50, 1.00, 1.00, 0.60, 0.50]
        blue = [1.00, 1.00, 0.50, 0.40, 0.50]
       
    else:
        # default palette, looks cool
        stops = [0.00, 0.34, 0.61, 0.84, 1.00]
        red   = [0.00, 0.00, 0.87, 1.00, 0.51]
        green = [0.00, 0.81, 1.00, 0.20, 0.00]
        blue  = [0.51, 1.00, 0.12, 0.00, 0.00]

    s = array('d', stops)
    r = array('d', red)
    g = array('d', green)
    b = array('d', blue)

    npoints = len(s)
    ROOT.TColor.CreateGradientColorTable(npoints, s, r, g, b, ncontours)
    ROOT.gStyle.SetNumberContours(ncontours)