Merge pull request #36 from IzaakWN/python3

Make `python3` compatible

.gitignore

@@ -1,4 +1,5 @@
 *.pyc
+*.bak
 *.d
 *.so
 *.pcm

Fitter/HighPT/DatacardsWToMuNu.py

@@ -123,7 +123,7 @@ def PlotWToMuNu(h_data_input,h_bkg_input,h_sig_input,era,var):
     canvas.cd()
     canvas.SetSelected(canvas)
     canvas.Update()
-    print
+    print()
     print('Creating control plot')
     canvas.Print(utils.figuresFolderWMuNu+"/wmunu_"+era+".png")
 
@@ -176,22 +176,22 @@ def CreateCardsWToMuNu(fileName,h_data,h_bkg,h_sig,uncs):
     basefolder = utils.picoFolder
     var = args.variable    
 
-    print
+    print()
     print('initializing SingleMuon samples >>>')
     singlemuSamples = {} # data samples dictionary
     singlemuNames = utils.singlemu[args.era]
     for singlemuName in singlemuNames:
         singlemuSamples[singlemuName] = utils.sampleHighPt(basefolder,args.era,
                                                            "munu",singlemuName,True)
 
-    print
+    print()
     print('initializing background samples >>>')
     bkgSamples = {} # MC bkg samples dictionary 
     for bkgSampleName in bkgSampleNames:
         bkgSamples[bkgSampleName] = utils.sampleHighPt(basefolder,args.era,
                                                        "munu",bkgSampleName,False)
 
-    print
+    print()
     print('initializing signal samples >>>')
     sigSamples = {} # MC signal samples dictionary 
     for sigSampleName in sigSampleNames:

Fitter/HighPT/DatacardsWToTauNu.py

@@ -25,7 +25,7 @@ def FitConst(x,par):
 def ComputeFake(h_wjets,h_dijets,h_fraction,name):
     nbins = h_wjets.GetNbinsX()
     hist = h_wjets.Clone(name)
-    print
+    print()
     print('Computing fake histogram ->',name)
     for i in range(1,nbins+1):
         x_wjets = h_wjets.GetBinContent(i)
@@ -53,7 +53,7 @@ def ComputeFake(h_wjets,h_dijets,h_fraction,name):
 ##################################
 def ComputeEWKFraction(h_data,h_mc):
 
-    print
+    print()
     print('Computing EWK fraction')
     nbins = h_data.GetNbinsX()
     h_fraction = h_data.Clone('fraction')
@@ -85,7 +85,7 @@ def ComputeEWKFraction(h_data,h_mc):
 ############################
 def PlotClosure(h_data,h_models,wp,era,var):
 
-    print
+    print()
     print("Plotting closure")
     h_model = h_models["bkg_fake_mc_wjets"]    
 
@@ -305,7 +305,7 @@ def PlotWToTauNu(h_data_input,h_fake_input,h_bkg_input,h_sig_input,wp,era,var):
     canvas.cd()
     canvas.SetSelected(canvas)
     canvas.Update()
-    print
+    print()
     print('Creating control plot')
     canvas.Print(utils.figuresFolderWTauNu+"/wtaunu_"+wp+"_"+era+".png")
 
@@ -382,7 +382,7 @@ def CreateCardsWToTauNu(fileName,h_data,h_fake,h_bkg,h_sig,uncs_fake,uncs_sig):
     uncert_names = ["JES","Unclustered","taues_1pr","taues_1pr1pi0","taues_3pr","taues_3pr1pi0"]
     #    uncert_names = ["JES","Unclustered","taues"]
 
-    print
+    print()
     print('initializing data samples >>>')
     metSamples = {} # data samples dictionary
     metNames = utils.met[args.era]
@@ -391,15 +391,15 @@ def CreateCardsWToTauNu(fileName,h_data,h_fake,h_bkg,h_sig,uncs_fake,uncs_sig):
                                                       "taunu",metName,True)
         metSamples[metName].SetTauNuConfig(fakeFactor,args.wp,wtaunuCuts)
 
-    print
+    print()
     print('initializing background samples >>>')
     bkgSamples = {} # MC bkg samples dictionary 
     for bkgSampleName in bkgSampleNames:
         bkgSamples[bkgSampleName] = utils.sampleHighPt(basefolder,args.era,
                                                        "taunu",bkgSampleName,False)
         bkgSamples[bkgSampleName].SetTauNuConfig(fakeFactor,args.wp,wtaunuCuts)
 
-    print
+    print()
     print('initializing signal samples >>>')
     sigSamples = {} # MC signal samples dictionary 
     for sigSampleName in sigSampleNames:
@@ -411,7 +411,7 @@ def CreateCardsWToTauNu(fileName,h_data,h_fake,h_bkg,h_sig,uncs_fake,uncs_sig):
     hists_sig_notFake = utils.RunSamplesTauNu(sigSamples,var,"",xbins,"_notFake","sig")
 
     # running on signal samples (central template and unceertainties)
-    print
+    print()
     print('Running on signal samples >>>')
     hists_sig_shape = {}
     commonCut = "metfilter>0.5&&mettrigger>0.5&&extraelec_veto<0.5&&extramuon_veto<0.5&&extratau_veto<0.5&&njets==0&&idDeepTau2017v2p1VSmu_1>=1&&idDeepTau2017v2p1VSe_1>=4&&genmatch_1==5&&idDeepTau2017v2p1VSjet_1>=" + utils.tauWPs[args.wp]
@@ -450,12 +450,12 @@ def CreateCardsWToTauNu(fileName,h_data,h_fake,h_bkg,h_sig,uncs_fake,uncs_sig):
 
 
     # running selection on data 
-    print
+    print()
     print('Running on data samples >>>')
     hists_data        = utils.RunSamplesTauNu(metSamples,var,"",xbins,"","data")
 
     # running selection on bkgd samples
-    print
+    print()
     print('Running on background samples >>>')
     hists_bkg_fake    = utils.RunSamplesTauNu(bkgSamples,var,"",xbins,"_fake","bkg")
     hists_bkg_notFake = utils.RunSamplesTauNu(bkgSamples,var,"",xbins,"_notFake","bkg")
@@ -524,7 +524,7 @@ def CreateCardsWToTauNu(fileName,h_data,h_fake,h_bkg,h_sig,uncs_fake,uncs_sig):
 
     # saving histograms to datacard file datacards
     outputFileName = utils.datacardsFolder + "/taunu_" + args.wp + "_" + args.era
-    print
+    print()
     print("Saving histograms to RooT file",outputFileName+".root")
     fileOutput = ROOT.TFile(outputFileName+".root","recreate")
     fileOutput.mkdir("taunu")

Fitter/HighPT/FakeFactorHighPt.py

@@ -50,7 +50,7 @@ def FitMass(x,par):
 ###########################
 def DrawFF(hist,era,channel,label,WP,**kwargs):
 
-    print
+    print()
     print('fitting FF histo >>>',era,channel,label)
 
     isdata = kwargs.get('isdata',True)
@@ -139,16 +139,16 @@ def main(outputfile,dataSamples,mcSamples,sigSamples,**kwargs):
     channel = kwargs.get("channel","wjets")
     mode = kwargs.get("mode",0)
 
-    print
+    print()
     print("+++++++++++++++++++++++++++++++++++++++++++")
-    print
+    print()
     if mode==0: 
         print('Computing FF as a function of tau pT in bins of pt_tau/pt_jet',era,wp,channel)
     elif mode==1: 
         print('Computing FF as a function of tau mass in bins of DM',era,wp,channel)
     else: 
         print('Computing FF as a function of tau pT in bins of (pt_tau/pt_jet,DM)',era,wp,channel)
-    print
+    print()
 
     cutTauDen = "idDeepTau2017v2p1VSjet_2<4"
     cutTauNum = "idDeepTau2017v2p1VSjet_2>=" + utils.tauWPs[wp]
@@ -179,10 +179,10 @@ def main(outputfile,dataSamples,mcSamples,sigSamples,**kwargs):
     histssig = {}
     for label in binCuts:
         xbins = []
-        print
+        print()
         print('***************************')
         print('Running over',label)
-        print
+        print()
 
         if mode==0: xbins=utils.xbinsPt
         elif mode==1: xbins=utils.xbinsMass[label]
@@ -256,15 +256,15 @@ def main(outputfile,dataSamples,mcSamples,sigSamples,**kwargs):
         singlemuSamples[singlemuName] = utils.sampleHighPt(basefolder,args.era,
                                                            "wjets",singlemuName,True)
 
-    print
+    print()
     print('initializing JetHT samples >>>')
     jethtSamples = {} # data samples disctionary
     jethtNames = utils.jetht[args.era]
     for jethtName in jethtNames:
         jethtSamples[jethtName] = utils.sampleHighPt(basefolder,args.era,
                                                      "dijets",jethtName,True)
 
-    print
+    print()
     print('initializing MC samples >>>')
     mcSamples = {} # mc samples dictionary
     for mcSampleName in mcSampleNames:
@@ -274,7 +274,7 @@ def main(outputfile,dataSamples,mcSamples,sigSamples,**kwargs):
         else:
             mcSamples[mcSampleName] = utils.sampleHighPt(basefolder,args.era,"wjets",mcSampleName,
                                                          False)
-    print
+    print()
     print('initializing W+Jets samples >>>') 
     sigSamples = {} # wjets samples dictionary
     for sigSampleName in sigSampleNames:

Fitter/HighPT/MetTrigger.py

@@ -47,7 +47,7 @@ def DrawEfficiency(histdata,histmc,era,legend):
 def main(args):
 
     era = args.era
-    print
+    print()
 
     channel = "munu"
     basefolder = utils.picoFolder
@@ -71,7 +71,7 @@ def main(args):
         dataSamples[dataSampleName] = utils.sampleHighPt(basefolder,era,channel,
                                                          dataSampleName,True)
 
-    print
+    print()
     mcSamples = {} # mc samples dictionary
     print('Initializing W+Jets MC samples>>>')
     for mcSampleName in mcSampleNames:
@@ -81,7 +81,7 @@ def main(args):
         else:
             mcSamples[mcSampleName] = utils.sampleHighPt(basefolder,era,channel,
                                                          mcSampleName,False)
-    print
+    print()
     histsdata = {} # data histo dictionary
     histsmc = {} # mc histo dictionary
 

Fitter/HighPT/PlotWToMuNu.py

@@ -98,7 +98,7 @@ def Plot(h_data_input,h_tot_input,h_bkg_input,h_sig_input,era,var,postFit):
     canvas.cd()
     canvas.SetSelected(canvas)
     canvas.Update()
-    print
+    print()
     print('Creating control plot')
     if postFit:
         canvas.Print(utils.figuresFolderWMuNu+"/wmunu_"+era+"_postFit.png")

Fitter/HighPT/PlotWToTauNu.py

@@ -112,7 +112,7 @@ def Plot(h_data_input,h_tot_input,h_fake_input,h_bkg_input,h_sig_input,wp,era,va
     canvas.cd()
     canvas.SetSelected(canvas)
     canvas.Update()
-    print
+    print()
     print('Creating control plot')
     if postFit:
         canvas.Print(utils.figuresFolderWTauNu+"/wtaunu_"+wp+"_"+era+"_postFit.png")

Fitter/TauES/correctionlib/tau_tes.py

@@ -366,5 +366,5 @@ def main():
 
 if __name__ == '__main__':
   main()
-  print()
+  print('')
 

Fitter/paper/writecards.py

@@ -240,7 +240,7 @@ def main():
 
 
 if __name__ == '__main__':
-  print
+  print('')
   main()
   print ">>>\n>>> Done harvesting\n"
 

Fitter/python/HighPT/utilsHighPT.py

@@ -276,7 +276,7 @@ def extractBinLabels(pt,ptratio):
 
 # Run over set of samples and create histogram
 def RunSamples(samples,var,weight,cut,xbins,name):
-    print
+    print('')
     print("Running",name,var,weight,cut)
     nbins = len(xbins)-1
     hist = ROOT.TH1D(name,"",nbins,array('d',list(xbins)))
@@ -289,7 +289,7 @@ def RunSamples(samples,var,weight,cut,xbins,name):
 # Run over set of samples and create histograms for W*->tau+v channel
 # for each sample loop over Tree entries is performed
 def RunSamplesTauNu(samples,var,unc,xbins,selection,name):
-    print
+    print('')
     print("Running",name,var,unc,selection)
     nbins = len(xbins)-1
     hists = {} # discionary of histograms
@@ -418,7 +418,7 @@ def __init__(self,**kwargs):
 class FakeFactorHighPt:
 
     def __init__(self,filename):
-        print
+        print('')
         print('Loading fake factors from file',filename," >>>>>")
         self.fileName = filename
         self.fileFF = ROOT.TFile(self.fileName,"READ")

Fitter/python/models/TagAndProbeModel.py

@@ -64,11 +64,11 @@ def doParametersOfInterest(self):
              elif self.passbin in bin:
                exp_pass += self.DC.exp[bin][param]
              if self.verbose>=3:
-               print ">>> Expected number of %s events in %s bin: %9.1f"%(param,bin,self.DC.exp[bin][param])
+               print(">>> Expected number of %s events in %s bin: %9.1f"%(param,bin,self.DC.exp[bin][param]))
         if self.verbose>=2:
-          print ">>> Expected number of signal events in pass region: %9.1f"%(exp_pass)
-          print ">>> Expected number of signal events in fail region: %9.1f"%(exp_fail)
-          print ">>> SF_fail = (%.2f+(1.-SF)*%.2f)/%.2f"%(exp_fail,exp_pass,exp_fail)
+          print(">>> Expected number of signal events in pass region: %9.1f"%(exp_pass))
+          print(">>> Expected number of signal events in fail region: %9.1f"%(exp_fail))
+          print(">>> SF_fail = (%.2f+(1.-SF)*%.2f)/%.2f"%(exp_fail,exp_pass,exp_fail))
         assert exp_fail, "Fail region has no signal events!"
 
         # DEFINE PARAMETERS
@@ -88,7 +88,7 @@ def getYieldScale(self,bin,process):
           elif self.passbin in bin:
             scale = 'SF'
           if self.verbose>=1:
-            print ">>> Scaling %3s in %s bin by '%s'"%(process,bin,scale)
+            print(">>> Scaling %3s in %s bin by '%s'"%(process,bin,scale))
           return scale
         return 1