rootlogon.py

Go to the documentation of this file.

#  cout << "Loading WCSim Library" << endl
#  gSystem->Load("libWCSimRoot.so")

from ROOT import *
from array import *




alexStyle = TStyle("alexStyle", "Alex's Style")


#set the background color to white
alexStyle.SetFillColor(10)
alexStyle.SetFrameFillColor(10)
alexStyle.SetCanvasColor(10)
alexStyle.SetPadColor(10)
alexStyle.SetTitleFillColor(0)
alexStyle.SetStatColor(10)

#dont put a colored frame around the plots
alexStyle.SetFrameBorderMode(0)
alexStyle.SetCanvasBorderMode(0)
alexStyle.SetPadBorderMode(0)

#use the primary color palette in heatmap style
#alexStyle.SetPalette(1,0)

# Set a custom color palette
#Int_t  colors[50]

#Double_t Red[3]    = { 0.00, 0.00, 0.00}
#Double_t Green[3]  = { 0.00, 0.00, 0.00}
#Double_t Blue[3]   = { 0.70, 1.00, 1.00}
#Double_t Length[3] = { 0.00, 1.00, 1.00 }
#Int_t FI = TColor::CreateGradientColorTable(2,Length,Red,Green,Blue,50)
#for (int i=0 i<50 i++) colors[i] = FI+i
#gStyle.SetPalette(50,colors)


#navy    = (0, 0, 0.5)
#cyan    = (0, 1, 1)
#yellow  = (1, 1, 0)
#scarlet = (1, 0, 0)

#Red    = array("d", [navy[0], cyan[0], yellow[0], scarlet[0]])
#Green  = array("d", [navy[1], cyan[1], yellow[1], scarlet[1]])
#Blue   = array("d", [navy[2], cyan[2], yellow[2], scarlet[2]])
#Length = array("d", [   0.00,    0.25,      0.55,       1.00])

#TColor.CreateGradientColorTable(4,Length,Red,Green,Blue,100)


def SetupColors(colors, lens = []):
    Red    = array("d", [c[0] for c in colors] )
    Green  = array("d", [c[1] for c in colors] )
    Blue   = array("d", [c[2] for c in colors] )
    if not lens: lens = [ float(x)/(len(colors)-1.) for x in range(len(colors)) ]
    Length = array("d",  lens)
    TColor.CreateGradientColorTable(len(colors),Length,Red,Green,Blue,101)


navy    = (0, 0, 0.5)
blue    = (0, 0.2, 1)
cyan    = (0, 1, 1)
yellow  = (1, 1, 0)
orange  = (1, 0.5, 0)
magenta = (1, 0, 1)
red     = (1, 0, 0)
white   = (1, 1, 1)
black   = (0, 0, 0)

alexStyle.SetNumberContours(101)

#colors = [navy, blue, yellow, orange, red]
#SetupColors(colors, [0.0, 0.15, 0.5, 0.85, 1.0])
SetupColors( [black, red, orange, yellow, white] )


#set the default line color for a histogram to be black
alexStyle.SetHistLineColor(kBlue)

#set the default line color for a fit function to be red
alexStyle.SetFuncColor(kRed)

#make the axis labels black
alexStyle.SetLabelColor(kBlack,"xyz")

#set the default title color to be black
alexStyle.SetTitleColor(kBlack)


# Sizes

#set the margins
alexStyle.SetPadBottomMargin(0.17)
alexStyle.SetPadLeftMargin(0.17)
#alexStyle.SetPadTopMargin(0.075)
#alexStyle.SetPadLeftMargin(0.15)

#set axis label and title text sizes
alexStyle.SetLabelSize(0.06,"xyz")
alexStyle.SetTitleSize(0.07,"xyz")
alexStyle.SetTitleOffset(1.00,"x")
alexStyle.SetTitleOffset(1.05,"y")
alexStyle.SetStatFontSize(0.04)
alexStyle.SetTextSize(0.05)

# Cut off number of digits for switch to exponent
TGaxis.SetMaxDigits(4)

#set the plot title style
alexStyle.SetTitleAlign(23)
alexStyle.SetTitleX(.5)
alexStyle.SetTitleY(0.99)
alexStyle.SetTitleBorderSize(0)
alexStyle.SetTitleFillColor(0)
alexStyle.SetTitleFontSize(0.07)

#set the legend style
alexStyle.SetLegendBorderSize(0)
#alexStyle.SetLegendFillColor(0)

#set stat box style
alexStyle.SetOptFit(1111)
alexStyle.SetOptStat(0000000)
alexStyle.SetStatBorderSize(0)

#set line widths
alexStyle.SetHistLineWidth(2)
alexStyle.SetFrameLineWidth(2)
alexStyle.SetFuncWidth(2)

# Misc

#set the number of divisions to show
alexStyle.SetNdivisions(506, "xy")

#turn off xy grids
alexStyle.SetPadGridX(0)
alexStyle.SetPadGridY(0)

#set the tick mark style
alexStyle.SetPadTickX(1)
alexStyle.SetPadTickY(1)


#marker settings
#alexStyle.SetMarkerStyle(8)
alexStyle.SetMarkerSize(0.9)
alexStyle.SetMarkerStyle(1)
alexStyle.SetMarkerSize(1.0)

# Fonts

kMinosFont = 42
alexStyle.SetStatFont(kMinosFont)
alexStyle.SetLabelFont(kMinosFont,"xyz")
alexStyle.SetTitleFont(kMinosFont,"xyz")
alexStyle.SetTextFont(kMinosFont)
#alexStyle.SetLegendFont(kMinosFont)

#done
alexStyle.cd()

gROOT.ForceStyle()
gStyle.ls()










#import os
#from ctypes import *
#def startWC():
#       cdll.LoadLibrary(os.path.join(os.environ['HOME'],'WCSim','libWCSimRoot.so'))

# History and tab completion

#import os
#try: 
#    import readline, rlcompleter, atexit
#
#    history_file = os.path.join(os.environ['HOME'], '.python_history')
#    try:
#        readline.read_history_file(history_file)
#    except IOError:
#        pass
#    readline.parse_and_bind("tab: complete")
#    readline.set_history_length(1000)
#    atexit.register(readline.write_history_file, history_file)
#
#    del readline, rlcompleter, atexit, history_file, __file__
#except ImportError: 
#    print "No tab completion"

#started = True