def conclude(self) :
for configId, tag in enumerate(self.sideBySideAnalysisTags()) :
org=organizer.organizer( self.sampleSpecs(tag) )
org.mergeSamples(targetSpec = {"name":"g_jets_mg", "color":r.kGreen},
sources = ["g_jets_mg_pt%s"%bin for bin in ["40_100","100_200","200"] ])
org.scale(10)
pl = plotter.plotter(org, psFileName = self.psFileName(tag))
pl.plotAll()
def drawCut(hist,cut) :
line = r.TLine(cut,0,cut,hist.GetMaximum())
line.SetLineColor(r.kBlue)
line.SetLineStyle(r.kDashed)
r.SetOwnership( line, False )
line.Draw("same")
def drawOne() :
one = r.TF1("one","1",0,5)
one.SetLineColor(r.kRed)
one.SetLineStyle(r.kDashed)
one.SetLineWidth(1)
r.SetOwnership(one,False)
one.Draw("same")
import utils
r.gStyle.SetOptStat(0)
c = r.TCanvas("c","",800,1000)
outDir = self.outputDirectory(configId)+"/"
for s in org.samples : c.Print(outDir+s["name"]+".ps[")
iSelection = -4
for key in sorted(org.selections[iSelection]):
if key == "counts" : continue
for hist,sample in zip(org.selections[iSelection][key],org.samples) :
if not hist : continue
ctr = hist.ProjectionX(hist.GetName()+"_ctr",0,2)
fwd = hist.ProjectionX(hist.GetName()+"_fwd",3)
if min(ctr.GetEntries(),fwd.GetEntries()) < 10: continue
leg = r.TLegend(0.52,0.7,0.9,0.9);
ctr.Scale(1/ctr.Integral()); ctr.SetLineColor(r.kBlack); ctr.SetMarkerColor(r.kBlack); leg.AddEntry(ctr,"Central","lp")
fwd.Scale(1/fwd.Integral()); fwd.SetLineColor(r.kRed); fwd.SetMarkerColor(r.kRed); leg.AddEntry(fwd,"Forward","lp")
ratio = utils.ratioHistogram(fwd,ctr)
m = 2*max(fwd.GetMaximum(),ctr.GetMaximum())
fwd.SetMaximum(m)
ctr.SetMaximum(m)
ratio.SetMinimum(0); ratio.SetMaximum(2)
ratio.SetTitle("fwd/ctr")
c.Clear(); c.Divide(1,2);
c.cd(1).SetLogy(); ctr.Draw(); fwd.Draw("same"); drawCut(ctr,0.55); leg.Draw()
c.cd(2); ratio.Draw(); drawOne(); drawCut(ratio,0.55)
c.Print(outDir+sample["name"]+".ps")
for s in org.samples :
fileName = outDir+s["name"]+".ps]"
c.Print(fileName)
fileNamePDF = fileName.replace(".ps]",".pdf")
os.system("ps2pdf "+fileName[:-1] + " " + fileNamePDF)
print fileNamePDF, "has been printed."
def conclude(self) :
for tag in self.sideBySideAnalysisTags() :
org=organizer.organizer( self.sampleSpecs(tag) )
pl = plotter.plotter(org,
psFileName = self.psFileName(tag),
blackList = ["lumiHisto","xsHisto","nJobsHisto"]
)
pl.plotAll()
def conclude(self) :
for tag in self.sideBySideAnalysisTags() :
#organize
org=organizer.organizer( self.sampleSpecs(tag) )
org.mergeSamples(targetSpec = {"name":"g_jets_mg", "color":r.kGreen}, sources = ["g_jets_mg_pt%s"%bin for bin in ["40_100","100_200","200"] ])
#org.mergeSamples(targetSpec = {"name":"qcd_py" , "color":r.kBlue}, sources = ["qcd_py_pt%d"%i for i in [30,80,170,300,470,800,1400] ])
#org.mergeSamples(targetSpec = {"name":"standard_model","color":r.kGreen+3},
# sources = ["g_jets_mg","qcd_py","tt_tauola_mg","z_inv_mg_skim","z_jets_mg_skim","w_jets_mg_skim"], keepSources = True
# )
org.scale(100)
#plot
pl = plotter.plotter(org,
psFileName = self.psFileName(tag),
samplesForRatios=("JetMET_skim","standard_model"),
sampleLabelsForRatios=("ctr","fwd"),
)
pl.plotAll()
import utils
c = r.TCanvas("c","",800,1000)
for key in org.selections[-1]:
if key == "counts" : continue
for hist,sample in zip(org.selections[-1][key],org.samples) :
if not hist : continue
c.Clear()
c.Divide(1,2)
c.cd(1).SetLogy()
div = 1 if "z" in key else 2
ctr = hist.ProjectionX(hist.GetName()+"_ctr",0,div)
fwd = hist.ProjectionX(hist.GetName()+"_fwd",div+1)
if max(ctr.GetEntries(),fwd.GetEntries()) < 10: continue
ctr.Scale(1/ctr.Integral())
fwd.Scale(1/fwd.Integral())
ctr.SetLineColor(r.kBlack); ctr.SetMarkerColor(r.kBlack)
fwd.SetLineColor(r.kRed); fwd.SetMarkerColor(r.kRed)
leg = r.TLegend(0.52,0.7,0.9,0.9);
leg.AddEntry(ctr,"Central","lp")
leg.AddEntry(fwd,"Forward","lp")
m = 2*max(fwd.GetMaximum(),ctr.GetMaximum())
fwd.SetMaximum(m)
ctr.SetMaximum(m)
ctr.Draw("")
fwd.Draw("same")
leg.Draw()
c.cd(2)
ratio = utils.ratioHistogram(fwd,ctr)
ratio.SetMaximum(2)
one = r.TF1("one","1",0,5)
one.SetLineColor(r.kRed)
one.SetLineStyle(r.kDashed)
one.SetLineWidth(1)
ratio.Draw()
one.Draw("same")
c.Print(sample["name"]+'_'+key+"normSlice.eps")
def conclude(self) :
org=organizer.organizer( self.sampleSpecs() )
org.mergeSamples(targetSpec = {"name":"qcd_py6","color":r.kBlue}, sources = ["qcd_py6_pt%d"%i for i in [15,30,80] ])
org.scale(100.0)
pl = supy.plotter(org,
pdfFileName = self.pdfFileName(""),
)
pl.plotAll()
def conclude(self) :
for tag in self.sideBySideAnalysisTags() :
#organize
org=organizer.organizer( self.sampleSpecs(tag) )
org.mergeSamples(targetSpec = {"name":"g_jets_mg", "color":r.kGreen}, sources = ["g_jets_mg_pt%s"%bin for bin in ["40_100","100_200","200"] ])
smSources = ["g_jets_mg","tt_tauola_mg","z_inv_mg_skim","z_jets_mg_skim","w_jets_mg_skim"]
if "pythia6" in tag :
org.mergeSamples(targetSpec = {"name":"qcd_py6", "color":r.kBlue}, sources = ["qcd_py6_pt%d"%i for i in [80,170,300] ])
smSources.append("qcd_py6")
if "pythia8" in tag :
lowerPtList = [0,15,20,30,50,80,
120,170,230,300,380,470,600,800,
1000,1400,1800,2200,2600,3000,3500]
org.mergeSamples(targetSpec = {"name":"qcd_py8","color":r.kBlue},
sources = ["qcd_py8_pt%dto%d"%(lowerPtList[i],lowerPtList[i+1]) for i in range(len(lowerPtList)-1)] )
smSources.append("qcd_py8")
if "madgraph" in tag :
org.mergeSamples(targetSpec = {"name":"qcd_mg", "color":r.kBlue},
sources = ["qcd_mg_ht_%s"%bin for bin in ["50_100","100_250","250_500","500_1000","1000_inf"] ])
smSources.append("qcd_mg")
org.mergeSamples(targetSpec = {"name":"standard_model","color":r.kGreen+3}, sources = smSources, keepSources = True)
org.scale()
def makeRate(hist,total,name) :
if (not hist) or (not total): return 0
hist.Scale(1./total)
hist.SetMaximum(0.91)
hist.GetXaxis().SetTitle("")
if hist.GetYaxis().GetNbins()>1:
hist.GetYaxis().SetTitle("")
hist.GetZaxis().SetTitle("rate")
else:
hist.GetYaxis().SetTitle("rate")
hist.SetTitle(name)
return 1
for iSel,selection in enumerate(org.selections) :
if len(selection)>1 :
totals = tuple(map(lambda h: h.GetBinContent(2) if h else 0, selection["counts"]))
totals2 = tuple(map(lambda h: h.GetBinContent(1)+h.GetBinContent(2) if h else 0, org.selections[iSel+1]["counts"])) if \
iSel+1<len(org.selections) else [0]*len(totals)
totals = map(max,totals,totals2)
for name in selection:
if "VetoCounts" in name :
result = map(makeRate, selection[name],totals, [name]*len(totals))
pl = plotter.plotter(org,
psFileName = self.psFileName(tag),
samplesForRatios=("JetMET_skim","standard_model"),
sampleLabelsForRatios=("data","s.m."),
#compactOutput=True
)
pl.plotAll()
def main():
media = ['.jpg', '.mp4', '.mp3', '.png', '.jpeg']
programs = ['.deb', '.tar.gz', '.tar.bz2', '.exe', '.msi', '.AppImage', '.tar.xz', '.tgz']
code = ['.py', '.js', '.html', '.css', '.jar', '.cpp', '.c', '.rs', '.cs', '.php', '.sh']
compress = ['.zip', '.rar', '.7zip']
text = ['.doc', '.docx', '.pdf', '.txt', '.log', '.odt', '.ppt', '.pptx']
user_system_type = platform.system()
user_name = getpass.getuser()
folderSystem.detect_create_folders(user_name, user_system_type, 'MEDIA')
folderSystem.detect_create_folders(user_name, user_system_type, 'PROGRAMS')
folderSystem.detect_create_folders(user_name, user_system_type, 'TEXT')
folderSystem.detect_create_folders(user_name, user_system_type, 'CODE')
folderSystem.detect_create_folders(user_name, user_system_type, 'COMPRESS')
folderSystem.detect_create_folders(user_name, user_system_type, 'OTHER')
organizer.organizer(user_name, programs, 'PROGRAMS', user_system_type)
organizer.organizer(user_name, text, 'TEXT', user_system_type)
organizer.organizer(user_name, media, 'MEDIA', user_system_type)
organizer.organizer(user_name, code, 'CODE', user_system_type)
organizer.organizer(user_name, compress, 'COMPRESS', user_system_type)
organizer.organizer_other(user_name, user_system_type)
def conclude(self) :
for tag in self.sideBySideAnalysisTags() :
#organize
org=organizer.organizer( self.sampleSpecs(tag) )
self.mergeSamples(org, tag)
org.scale()
pl = plotter.plotter(org,
psFileName = self.psFileName(tag),
samplesForRatios=("2010 Data","standard_model"),
sampleLabelsForRatios=("data","s.m."),
)
pl.plotAll()
def conclude(self) :
for tag in self.sideBySideAnalysisTags() :
org=organizer.organizer( self.sampleSpecs(tag) )
org.mergeSamples(targetSpec = {"name":"qcd_py8", "color":r.kBlue}, allWithPrefix="qcd_py8")
org.mergeSamples(targetSpec = {"name":"2010 Data", "color":r.kBlack, "markerStyle":20}, allWithPrefix="Run2010")
org.scale()
pl = supy.plotter(org,
pdfFileName = self.pdfFileName(tag),
samplesForRatios=("2010 Data","qcd_py8"),
sampleLabelsForRatios=("data","qcd"),
)
pl.plotAll()
def conclude(self) :
#organize
org = organizer.organizer( self.sampleSpecs() )
#org.scale()
##plot
#pl = plotter.plotter(org,
# psFileName = self.psFileName(),
# doLog = False,
# #compactOutput = True,
# #noSci = True,
# #pegMinimum = 0.1,
# blackList = ["lumiHisto","xsHisto","nJobsHisto"],
# )
#pl.plotAll()
self.makeXsPlot(org, "", "scan_tanbeta10_tanja1")
def conclude(self) :
for tag in self.sideBySideAnalysisTags() :
#organize
org=organizer.organizer( self.sampleSpecs(tag) )
org.mergeSamples(targetSpec = {"name":"g_jets_mg", "color":r.kGreen}, sources = ["g_jets_mg_pt%s"%bin for bin in ["40_100","100_200","200"] ])
smSources = ["g_jets_mg","tt_tauola_mg","z_inv_mg_skim","z_jets_mg_skim","w_jets_mg_skim"]
if "pythia6" in tag :
org.mergeSamples(targetSpec = {"name":"qcd_py6", "color":r.kBlue}, sources = ["qcd_py6_pt%d"%i for i in [80,170,300] ])
smSources.append("qcd_py6")
if "pythia8" in tag :
lowerPtList = [0,15,20,30,50,80,
120,170,230,300,380,470,600,800,
1000,1400,1800,2200,2600,3000,3500]
org.mergeSamples(targetSpec = {"name":"qcd_py8","color":r.kBlue},
sources = ["qcd_py8_pt%dto%d"%(lowerPtList[i],lowerPtList[i+1]) for i in range(len(lowerPtList)-1)] )
smSources.append("qcd_py8")
if "madgraph" in tag :
org.mergeSamples(targetSpec = {"name":"qcd_mg", "color":r.kBlue},
sources = ["qcd_mg_ht_%s"%bin for bin in ["50_100","100_250","250_500","500_1000","1000_inf"] ])
smSources.append("qcd_mg")
org.mergeSamples(targetSpec = {"name":"standard_model","color":r.kGreen+3}, sources = smSources, keepSources = True)
org.scale()
##other
#import deltaPhiLook
#numers = ["JetMET_skim","standard_model"]
#if "pythia" in tag : qcdLabel = "qcd_py"
#if "madgraph" in tag : qcdLabel = "qcd_mg"
#
#for numer,denom in zip(numers,[qcdLabel]*len(numers)) :
# d = deltaPhiLook.deltaPhiLooker(org,tag,(numer,denom))
# d.makeSlicePlots()
# if numer==numers[0] :
# d.makeSliceComparisonPlots(qcdLabel)
#plot
pl = plotter.plotter(org,
psFileName = self.psFileName(tag),
samplesForRatios=("JetMET_skim","standard_model"),
sampleLabelsForRatios=("data","s.m."),
)
pl.plotAll()
def conclude(self) :
##for skimming only
#org = organizer.organizer( self.sampleSpecs() )
#utils.printSkimResults(org)
#organize
org=organizer.organizer( self.sampleSpecs() )
#org.scale()
##plot
#pl = supy.plotter(org,
# pdfFileName = self.pdfFileName(),
# doLog = False,
# #compactOutput = True,
# #noSci = True,
# #pegMinimum = 0.1,
# blackList = ["lumiHisto","xsHisto","nJobsHisto"],
# )
#pl.plotAll()
self.makeScanPlots(org)
def conclude(self) :
smSources = ["g_jets_mg","tt_tauola_mg","z_inv_mg_skim","z_jets_mg_skim","w_jets_mg_skim"]
for tag in self.sideBySideAnalysisTags() :
org = organizer.organizer( self.sampleSpecs(tag) )
org.mergeSamples(targetSpec = {"name":"g_jets_mg", "color":r.kGreen}, allWithPrefix = "g_jets_mg")
for color,qcd in zip([r.kRed,r.kBlue,r.kGreen],["qcd_py6", "qcd_py8", "qcd_mg"]) :
org.mergeSamples(targetSpec = {"name":qcd, "color":r.kBlue}, allWithPrefix = qcd)
if qcd in [sample["name"] for sample in org.samples]:
org.mergeSamples(targetSpec = {"name":"sm_w_"+qcd,"color":color}, sources = smSources+[qcd], keepSources = True)
org.drop(qcd)
for sample in smSources: org.drop(sample)
org.scale()
pl = plotter.plotter(org,
psFileName = self.psFileName(tag),
samplesForRatios=("JetMET_skim", filter(lambda name: "sm_w_qcd" in name, [sample["name"] for sample in org.samples])),
sampleLabelsForRatios=("data","s.m.")
)
pl.plotAll()
def conclude(self) :
for tag in self.sideBySideAnalysisTags() :
##for skimming only
#org = organizer.organizer( self.sampleSpecs(tag) )
#utils.printSkimResults(org)
#organize
org=organizer.organizer( self.sampleSpecs(tag) )
org.mergeSamples( targetSpec = {"name":"2010 Data", "color":r.kBlack, "markerStyle":20}, allWithPrefix="Run2010" )
#self.mergeSamples(org, tag)
org.scale()
#plot
pl = plotter.plotter(org,
psFileName = self.psFileName(tag),
#samplesForRatios = ("2010 Data","standard_model"),
#sampleLabelsForRatios = ("data","s.m."),
#whiteList = ["xcak5JetAlphaTPat","xcak5JetAlphaTZoomPat"],
blackList = ["nJobsHisto","lumiHisto"]
#compactOutput = True
)
pl.plotAll()
def conclude(self):
org = organizer.organizer(self.sampleSpecs())
pl = plotter.plotter(
org, psFileName=self.psFileName(), doLog=False, blackList=["lumiHisto", "xsHisto", "nJobsHisto"]
)
pl.plotAll()
def conclude(self) :
for tag in self.sideBySideAnalysisTags() :
org = organizer.organizer( self.sampleSpecs(tag) )
utils.printSkimResults(org)
import case
import organizer
from zencad import *
w = 27
h = 20
l = 64
t = 1.5
r = 27 / 2 - 4
z = 1
s = 0.965
d = 5
d2 = 5
m = 3
n = 5
st = organizer.organizer(m, n, w, h, l, t, d, d2)
cs = case.case(w, h, l, t, r, z, s)
ucs = union([
cs.translate(t * 1.5 + (w + t) * i, 0, t + (h + t) * j)
for i in range(0, m) for j in range(0, n)
])
disp(ucs)
disp(st, color(1, 1, 1))
show()
def organizer(self, config) : return organizer.organizer(config['tag'], self.sampleSpecs(config['tag']))
############
def __init__(self, options) :
def conclude(self) :
org = organizer.organizer( self.sampleSpecs() )
utils.printSkimResults(org)
Default is false.
If it is desired to be made true,
it should be included in command without any value.
''')
parser.add_argument('--ignore',
type=str,
metavar='IGNORE',
nargs='+',
help='''
Used to ignore files/directories from the source directory.
Add space-separated names.
''')
args = parser.parse_args()
if args.src[0] == '.':
args.src = args.src[1:]
args.src = os.getcwd() + args.src
if args.dist[0] == '.':
args.dist = args.dist[1:]
args.dist = os.getcwd() + args.dist
if __name__ == "__main__":
print('\norganize_by:', args.organize_by, '\nsrc:', args.src, '\ndist:',
args.dist, '\nignore:', args.ignore, '\ninclude_sub_directories:',
args.include_sub_directories, '\n')
organizer(args.organize_by, args.src, args.dist, args.ignore,
args.include_sub_directories)
def conclude(self) :
for tag in self.sideBySideAnalysisTags() :
org=organizer.organizer( self.sampleSpecs(tag) )
org.mergeSamples(targetSpec = {"name":"2010 Data", "color":r.kBlack, "markerStyle":20}, allWithPrefix = "Run2010" )
pl = supy.plotter(org, pdfFileName = self.pdfFileName(tag))
pl.plotAll()
