def add_path(lmon_top, geant_inst):
sys.path.append(lmon_top + "/macro")
sys.path.append(lmon_top + "/macro/Q2rec")
gSystem.AddIncludePath(" -I" + lmon_top + "/include")
gSystem.AddIncludePath(" -I" + geant_inst)
gROOT.ProcessLine(".L " + lmon_top + "/src/GeoParser.cxx+")
def compileAll(
branches = [],
treename = 'tau',
inputpath = '',
outputdump = False,
removeunselected = False,
cutflows = [],
toolset = 'LepHad2011',
):
global compiledir
compiledir = 'libs/' + toolset
from ROOT import gSystem
gSystem.SetAclicMode(gSystem.kOpt)
gSystem.AddIncludePath('-I' + filedir + '/../External/TauSpinner/extern/include')
gSystem.AddIncludePath('-I' + filedir + '/../External/TauSpinner/extern/include/Tauola')
compileC(filedir+"/Loader.C")
compileC(filedir+"/CutFlowHist.C")
print "@@@@@ Setting up the branches, kernel, whiteboard and external tools..."
if type(branches) is str:
branches = [branches]
b = []
for branch in branches:
if not hasattr(__import__('Branches.'+branch), branch):
raise Exception("Can't find Branches: " + branch)
b.append(getattr(__import__('Branches.'+branch), branch))
import CodeGenerator, NTupleTools
if not CodeGenerator.setupBranches(b, treename, inputpath, outputdump):
raise Exception("Setup branches failed!")
compileC(filedir+"/WhiteBoard.C")
compileC(filedir+"/CutFlow_Base.C")
CodeGenerator.setupForkItem(removeunselected, outputdump)
import External
External.initialiseAllTools(toolset)
import CutFlows
CutFlows.load(cutflows)
compileC(filedir+"/PostLoader.C")
def compileAClick(dictionary, g4=True):
from ROOT import gInterpreter, gSystem
import os.path
dd4hep = os.environ['DD4hepINSTALL']
inc = ' -I' + os.environ['ROOTSYS'] + '/include -I' + dd4hep + '/include '
lib = ' -L' + dd4hep + '/lib -lDDCore -lDDG4 -lDDSegmentation '
if g4:
geant4 = os.environ['G4INSTALL']
inc = inc + ' -I' + geant4 + '/include/Geant4 -Wno-shadow -g -O0 '
lib = lib + ' -L' + geant4 + '/lib -L' + geant4 + '/lib64 -lG4event -lG4tracking -lG4particles '
gSystem.AddIncludePath(inc)
gSystem.AddLinkedLibs(lib)
logger.info('Loading AClick %s', dictionary)
package = imp.find_module('DDG4')
dic = os.path.dirname(package[1]) + os.sep + dictionary
gInterpreter.ProcessLine('.L ' + dic + '+')
from ROOT import dd4hep as module
return module
def RooFitInclude():
print "adding RooFit ...",
scramCmd = ['scram','tool','info','roofitcore']
grepCmd = ['grep', 'INCLUDE']
pscram = subprocess.Popen(scramCmd, stdout = subprocess.PIPE)
pgrep = subprocess.Popen(grepCmd, stdin=pscram.stdout,
stdout=subprocess.PIPE)
pscram.stdout.close()
output = pgrep.communicate()[0]
if (pgrep.returncode == 0):
print "done"
roofitinc = output.split("=")[1].rstrip()
# print roofitinc
gROOT.GetInterpreter().AddIncludePath(roofitinc)
roofitinc = '-I"' + roofitinc + '"'
# print roofitinc
gSystem.AddIncludePath(roofitinc)
return True
else:
print "failed"
print 'scram returned:',pscram.returncode,'grep:',pgrep.returncode
return False
def compileAClick(dictionary,g4=True):
from ROOT import gInterpreter, gSystem
import sys, imp, exceptions
import os.path
dd4hep = os.environ['DD4hepINSTALL']
inc = ' -I'+os.environ['ROOTSYS']+'/include -I'+dd4hep+'/include '
lib = ' -L'+dd4hep+'/lib -lDDCore -lDDG4 -lDDSegmentation '
if g4:
geant4 = os.environ['G4INSTALL']
inc = inc + ' -I'+geant4+'/include/Geant4 -Wno-shadow -g -O0 '
lib = lib + ' -L'+geant4+'/lib -L'+geant4+'/lib64 -lG4event -lG4tracking -lG4particles '
gSystem.AddIncludePath(inc)
gSystem.AddLinkedLibs(lib)
#####print "Includes: ",gSystem.GetIncludePath(),"\n","Linked libs:",gSystem.GetLinkedLibs()
print 'Loading AClick ',dictionary
package = imp.find_module('DDG4')
dic = os.path.dirname(package[1])+os.sep+dictionary
###print dic
gInterpreter.ProcessLine('.L '+dic+'+')
#####gInterpreter.Load('DDG4Dict_C.so')
from ROOT import DD4hep as module
return module
if arguments['--noRoot']:
print 'Not loading any ROOT modules. usefull for generating files on oslic'
else:
from ROOT import TRandom3
from ROOT import TChain, TGraph, TGraphErrors, gSystem, gROOT, TH1D, TH2D, TFile, TCanvas, TF1
from ROOT import THStack, Double
from ROOT import kRed, kBlue, kGreen, kCyan, kOrange
from ROOT import kOrange as kO, kBlue as kB, kGreen as kG
from ROOT import kMagenta as kM, kAzure as kA, kRed as kR
from ROOT import TCanvas, TLine, TLatex
from ROOT import gStyle, gPad, TPaletteAxis
gSystem.Load("$RATROOT/lib/libRATEvent")
gSystem.AddIncludePath(" -I$RATROOT/include")
# gROOT.LoadMacro("$WATCHENV/watchmakers/goldenFileExtractor.C")
# from ROOT import goldenFileExtractor
gStyle.SetOptStat(1112211)
def loadSimulationParametersNew():
#Chain and subsequent isotopes
d = {}
d['CHAIN_238U_NA'] = ['234Pa', '214Pb', '214Bi', '210Bi', '210Tl']
d['CHAIN_232Th_NA'] = ['228Ac', '212Pb', '212Bi', '208Tl']
d['CHAIN_235U_NA'] = ['231Th', '223Fr', '211Pb', '211Bi', '207Tl']
d['40K_NA'] = ['40K']
ut.put_yx_tit(hZ, "Counts", "Hit #it{z} (mm)", 1.5, 1.4)
ut.set_margin_lbtr(gPad, 0.11, 0.11, 0.02, 0.03)
gPad.SetGrid()
gPad.SetLogy()
ut.invert_col(rt.gPad)
can.SaveAs("01fig.pdf")
#hit_z
#_____________________________________________________________________________
if __name__ == "__main__":
gROOT.SetBatch()
gStyle.SetPadTickX(1)
gStyle.SetFrameLineWidth(2)
gROOT.ProcessLine(".L ../acc_Q2_kine.h+")
gSystem.AddIncludePath(" -I/home/jaroslav/sim/lmon/include")
gSystem.AddIncludePath(
" -I/home/jaroslav/sim/geant/geant4.10.07.p01/install/include/Geant4")
gROOT.ProcessLine(".L ../../src/GeoParser.cxx+")
main()
gStyle.SetPadRightMargin(0.06) ## was 0.02
gStyle.SetTitleColor(1, "XYZ")
gStyle.SetTitleFont(42, "XYZ")
gStyle.SetTitleSize(0.07, "XYZ")
gStyle.SetTitleXOffset(0.9)
gStyle.SetTitleYOffset(1.4) ## was 1.25
## For the axis labels:
gStyle.SetLabelColor(1, "XYZ")
gStyle.SetLabelFont(42, "XYZ")
gStyle.SetLabelOffset(0.007, "XYZ")
gStyle.SetLabelSize(0.06, "XYZ")
gStyle.SetNdivisions(505, "XYZ")
if (gSystem.DynamicPathName("libFWCoreFWLite.so", True)):
gROOT.GetInterpreter().AddIncludePath(cmssw_base + '/src')
gSystem.AddIncludePath('-I"' + cmssw_base + '/src"')
if not RooFitInclude():
workingdir = os.getcwd()
print 'changing to', cmssw_base, 'directory'
os.chdir(cmssw_base)
RooFitInclude()
print 'returning to working directory', workingdir
os.chdir(workingdir)
print 'end of pyroot_logon'
if __name__ == '__main__':
from ROOT import *
