コード例 #1
0
    backgrounddiphotonmvafile=ROOT.TFile(options.diphotonmvahistfilename)
    signaldiphotonmvafile=ROOT.TFile(options.signalfilename)
  #toymaker = BdtToyMaker(options.tfileName,"data_pow_model_150.0")
  #toymaker.fitData()
  toymaker = CombinedToyMaker(options.inputmassfacws)

  if options.inputpdfworkspace:
    if not os.path.isfile(options.inputpdfworkspace): 
	sys.exit("No file named %s, generate it first (remove option)"%options.inputpdfworkspace)
    toymaker.loadKeysPdf(options.inputpdfworkspace)
    #if options.expSig>0: toymaker.loadKeysPdf(backgroundpdfws,1)
    #else: toymaker.loadKeysPdf(backgroundpdfws,0)
  else: 
    backgrounddiphotonmvahist=backgrounddiphotonmvafile.Get(options.diphotonmvahisttreename)
    print 'Creating keys pdf from ', backgrounddiphotonmvahist.GetName(), ' with ', backgrounddiphotonmvahist.GetEntries(), ' entries '
    toymaker.createKeysPdf(backgrounddiphotonmvahist)	
    #if options.expSig>0: 
    #  signaldiphotonmvahist=signaldiphotonmvafile.Get(options.signaltreename)
    #  print 'Creating keys pdf from ', signaldiphotonmvahist.GetName(), ' with ', signaldiphotonmvahist.GetEntries(), ' entries '
    #  toymaker.createSigHistPdf(signaldiphotonmvahist)
    toymaker.savePdfWorkspace(options.bdtworkspacename)

  toymaker.plotData(160,180)
  toymaker.genData(cardOutDir+"/"+options.outputmassfactoy)
  toymaker.plotToy(160,200)
  toymaker.saveToyWorkspace("testToyWS.root")
  #toymaker.genData(options.expSig)
  #if options.expSig>0: toymaker.plotSigData(160)
  ROOT.gROOT.ProcessLine(".L tmvaLoader.C+")
  from ROOT import tmvaLoader
  g_tmva = tmvaLoader(options.tmvaweightsfolder+"/TMVAClassification_BDT%sMIT.weights.xml"%options.bdtType,options.bdtType)