[ViewVC] Diff of: cvsroot/UserCode/cbrown/Development/Plotting/Modules/EdgeLimit.C

Comparing UserCode/cbrown/Development/Plotting/Modules/EdgeLimit.C (file contents):
Revision 1.1 by buchmann, Tue Jun 12 06:12:29 2012 UTC vs.
Revision 1.5 by buchmann, Wed Jun 27 08:53:23 2012 UTC

+#include <iostream>
-+
+#include <RooRealVar.h>
-+
+#include <RooArgSet.h>
-+
+#include <RooDataSet.h>
-+
+#include <RooMCStudy.h>
-+
+#include <RooCategory.h>
-+
-+
+#include <RooPlot.h>
-+
+#include <RooSimultaneous.h>
-+
+#include <RooAddPdf.h>
-+
+#include <RooFitResult.h>
-+
+#include <RooVoigtian.h>
-+
+#include <RooMsgService.h>
-+
-+
+#include <RooStats/ModelConfig.h>
-+
+#include "RooStats/ProfileLikelihoodCalculator.h"
-+
+#include "RooStats/LikelihoodInterval.h"
-+
+#include "RooStats/HypoTestResult.h"
-+
+#include "RooStats/SimpleLikelihoodRatioTestStat.h"
-+
+#include "RooStats/ProfileLikelihoodTestStat.h"
-+
+#include "RooStats/HybridCalculatorOriginal.h"
-+
+#include "RooStats/HypoTestInverterOriginal.h"
-+
-+
+//#include "ParametrizedEdge.C"
-+
+#include "/shome/pablom/RooFit/Pdfs/RooSUSYTPdf.cxx"
-+
+#include "/shome/pablom/RooFit/Pdfs/RooSUSYBkgPdf.cxx"
-+
-+
+using namespace std;
+using namespace PlottingSetup;
-+
-+
+ShapeDroplet LimitsFromEdge(float low_fullCLs, float high_CLs, TTree *events, string addcut, string name, string mcjzb, string datajzb, vector<float> jzbbins, float jzbpeakerrormc, float jzbpeakerrordata) {
+  write_error(__FUNCTION__,"Not implemented edge limits yet (returning crap)");
+  ShapeDroplet beta;beta.observed=-12345;beta.SignalIntegral=1;return beta;
-+
+///------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------
-+
-+
-+
+namespace EdgeFitter {
-+
-+
+  void DoEdgeFit(string mcjzb, string datajzb, float DataPeakError, float MCPeakError, float jzb_cut, int icut, int is_data, TCut cut, TTree*);
-+
+  void DoEdgeFit(string mcjzb, string datajzb, float DataPeakError, float MCPeakError, vector<float> jzb_cut, int is_data, TCut cut, TTree*);
-+
+  void getMedianLimit(RooWorkspace *ws,vector<RooDataSet*> theToys,float &median,float &sigmaDown, float &sigmaUp, float &twoSigmaDown, float &twoSigmaUp);
-+
+  void InitializeVariables(float _mllmin, float _mllmax, float _jzbmax, TCut _cut);
-+
+  void PrepareDatasets(int);
-+
+  void DoFit(int is_data, float jzb_cut);
-+
+  string RandomStorageFile();
-+
+  Yield Get_Z_estimate(float,int);
-+
+  Yield Get_T_estimate(float,int);
-+
+  float calcExclusion(RooWorkspace *ws, RooDataSet *data = NULL);
-+
+  void prepareLimits(RooWorkspace *ws);
-+
+  vector<RooDataSet*> generateToys(RooWorkspace *ws, int nToys);
-+
+  void prepareLimits(RooWorkspace *ws, bool ComputeBands);
-+
+  TGraph* prepareLM(float mass, float nEv);
-+
-+
+  float jzbmax;
-+
+  float mllmin;
-+
+  float mllmax;
-+
+  TCut cut;
-+
-+
+  RooDataSet* AllData;
-+
+  RooDataSet* eeSample;
-+
+  RooDataSet* mmSample;
-+
+  RooDataSet* emSample;
-+
-+
+  bool MarcoDebug;
-+
+}
-+
-+
+TGraph* EdgeFitter::prepareLM(float mass, float nEv) {
-+
+  float massLM[1];
-+
+  massLM[0]=mass;
-+
+  float accEffLM[1];
-+
+  accEffLM[0]=nEv/PlottingSetup::luminosity;
-+
+  TGraph *lm = new TGraph(1, massLM, accEffLM);
-+
+  lm->GetXaxis()->SetNoExponent(true);
-+
+  lm->GetXaxis()->SetTitle("m_{cut} [GeV]");
-+
+  lm->GetYaxis()->SetTitle("#sigma #times A [pb] 95% CL UL");
-+
+  lm->GetXaxis()->SetLimits(0.,300.);
-+
+  lm->GetYaxis()->SetRangeUser(0.,0.08);
-+
+  lm->SetMarkerStyle(34);
-+
+  lm->SetMarkerColor(kRed);
-+
+  return lm;
-+
+}
-+
-+
+vector<RooDataSet*> EdgeFitter::generateToys(RooWorkspace *ws, int nToys) {
-+
+  ws->var("nSig")->setVal(0.);
-+
+  ws->var("nSig")->setConstant(true);
-+
+  RooFitResult* fit = ws->pdf("combModel")->fitTo(*ws->data("data_obs"),RooFit::Save());
-+
+  vector<RooDataSet*> theToys;
-+
-+
+  RooMCStudy mcEE(*ws->pdf("combModel"),RooArgSet(*ws->var("inv")),RooFit::Slice(*ws->cat("cat"),"EE"));
-+
+  mcEE.generate(nToys,44,true);
-+
+  RooMCStudy mcMM(*ws->pdf("combModel"),RooArgSet(*ws->var("inv")),RooFit::Slice(*ws->cat("cat"),"MM"));
-+
+  mcMM.generate(nToys,44,true);
-+
+  RooMCStudy mcOSOF(*ws->pdf("combModel"),RooArgSet(*ws->var("inv")),RooFit::Slice(*ws->cat("cat"),"OSOF"));
-+
+  mcOSOF.generate(nToys,44,true);
-+
-+
+  RooRealVar mll("mll","mll",mllmin,mllmax,"GeV/c^{2}");
-+
+  RooRealVar id1("id1","id1",0,1,"GeV/c^{2}");
-+
+  RooRealVar id2("id2","id2",0,1,"GeV/c^{2}");
-+
+  RooRealVar jzb("jzb","jzb",-jzbmax,jzbmax,"GeV/c");
-+
+  RooRealVar weight("weight","weight",0,1000,"");
-+
+  RooArgSet observables(mll,jzb,id1,id2,weight);
-+
-+
+  for(int i=0;i<nToys;i++) {
-+
+    RooDataSet* toyEE    = (RooDataSet*)mcEE.genData(i);
-+
+    RooDataSet* toyMM    = (RooDataSet*)mcMM.genData(i);
-+
+    RooDataSet* toyOSOF  = (RooDataSet*)mcOSOF.genData(i);
-+
+    stringstream toyname;
-+
+    toyname << "theToy_" << i;
-+
+    write_warning(__FUNCTION__,"Problem while adding toys");
-+
+//    RooDataSet *toyData = RooDataSet(toyname.str(),toyname.str(),observables,RooFit::Index(ws->cat("cat")),RooFit::Import("OSOF",*toyOSOF),RooFit::Import("EE",*toyEE),RooFit::Import("MM",*toyMM));
-+
+//    theToys.push_back(toyData);
-+
+  }
-+
+  ws->var("nSig")->setVal(17.0);
-+
+  ws->var("nSig")->setConstant(false);
-+
+  return theToys;
-+
+}
-+
-+
+void EdgeFitter::getMedianLimit(RooWorkspace *ws,vector<RooDataSet*> theToys,float &median,float &sigmaDown, float &sigmaUp, float &twoSigmaDown, float &twoSigmaUp) {
-+
+  TH1F *gauLimit = new TH1F("gausLimit","gausLimit",60,0.,80./PlottingSetup::luminosity);
-+
+  vector<float> theLimits;
-+
+  for(int itoy=0;itoy<theToys.size();itoy++) {
-+
+    float theLimit = calcExclusion(ws,theToys[itoy]);
-+
+    if(theLimit > 0 ) gauLimit->Fill(theLimit);
-+
+  }
-+
+  const Int_t nQ = 4;
-+
+  Double_t yQ[nQ] = {0.,0.,0.,0.};
-+
+  Double_t xQ[nQ] = {0.022750132,0.1586552539,0.84134474609999998,0.977249868};
-+
+  gauLimit->GetQuantiles(nQ,yQ,xQ);
-+
+  median = gauLimit->GetMean();
-+
+//  median = median1(gauLimit);
-+
+  twoSigmaDown = abs(yQ[0]-median);
-+
+  sigmaDown = abs(yQ[1]-median);
-+
+  sigmaUp = abs(yQ[2]-median);
-+
+  twoSigmaUp = abs(yQ[3]-median);
-+
+  cout << median * PlottingSetup::luminosity << " " << sigmaUp * PlottingSetup::luminosity << endl;
-+
+}
-+
-+
+void EdgeFitter::prepareLimits(RooWorkspace *ws, bool ComputeBands) {
-+
+  if(ComputeBands) {
-+
+    vector<RooDataSet*> theToys = EdgeFitter::generateToys(ws,50);
-+
+    vector<float> medVals;
-+
+    vector<float> medLimits;
-+
+    vector<float> sigmaLimitsDown;
-+
+    vector<float> sigmaLimitsUp;
-+
+    vector<float> twoSigmaLimitsDown;
-+
+    vector<float> twoSigmaLimitsUp;
-+
+    for(int i=20;i<=320;i+=40) {
-+
+      ws->var("nSig")->setVal(10.0);
-+
+      medVals.push_back((float)i);
-+
+      ws->var("m0")->setVal((float)i);
-+
+      ws->var("m0")->setConstant(true);
-+
+      float Smedian,SsigmaDown,SsigmaUp,StwoSigmaDown,StwoSigmaUp;
-+
+      EdgeFitter::getMedianLimit(ws,theToys,Smedian,SsigmaDown,SsigmaUp,StwoSigmaDown,StwoSigmaUp);
-+
+      medLimits.push_back(Smedian);
-+
+      sigmaLimitsDown.push_back(SsigmaDown);
-+
+      sigmaLimitsUp.push_back(SsigmaUp);
-+
+      twoSigmaLimitsDown.push_back(StwoSigmaDown);
-+
+      twoSigmaLimitsUp.push_back(StwoSigmaUp);
-+
+    }
-+
+    write_warning(__FUNCTION__,"Still need to store limits");
-+
+  } else {
-+
+    vector<float> theVals;
-+
+    vector<float> theLimits;
-+
+    for(int i=20;i<300;i+=5) {
-+
+      ws->var("nSig")->setVal(0.0);
-+
+      theVals.push_back((float)i);
-+
+      ws->var("m0")->setVal((float)i);
-+
+      ws->var("m0")->setConstant(true);
-+
+      theLimits.push_back(calcExclusion(ws));
-+
+    }
-+
-+
+    for(int i=0;i<theLimits.size();i++) {
-+
+      if((theLimits[i]<2.0/PlottingSetup::luminosity)||(theLimits[i]>40.0/PlottingSetup::luminosity)) {
-+
+        cout << i << " : " << theVals[i] << endl;
-+
+        theLimits[i] = (theLimits[i+2]+theLimits[i-2])/2.0;
-+
+        write_warning(__FUNCTION__,"Need to store limits");
-+
+      }
-+
+    write_warning(__FUNCTION__,"Need to store limits");
-+
+    }
-+
+}
-+
+}
-+
-+
-+
+float EdgeFitter::calcExclusion(RooWorkspace *ws, RooDataSet *data) {
-+
+  RooRealVar mu("mu","nSig",0,10000,"");
-+
+  RooArgSet poi = RooArgSet(mu);
-+
+  RooArgSet *nullParams = (RooArgSet*)poi.snapshot();
-+
+  nullParams->setRealValue("nSig",0);
-+
+  RooStats::ModelConfig *model = new RooStats::ModelConfig();
-+
+  model->SetWorkspace(*ws);
-+
+  model->SetPdf("combModel");
-+
+  model->SetParametersOfInterest(poi);
-+
+  if(!data) data = (RooDataSet*)ws->data("data_obs");
-+
-+
+  RooStats::ProfileLikelihoodCalculator plc(*data, *model);
-+
+  plc.SetNullParameters(*nullParams);
-+
+  plc.SetTestSize(0.05);
-+
+  RooStats::LikelihoodInterval* interval = plc.GetInterval();
-+
+  RooStats::HypoTestResult *htr = plc.GetHypoTest();
-+
+  double theLimit = interval->UpperLimit( mu );
-+
+  cout << "Significance " << htr->Significance() << endl;
-+
-+
+  ws->defineSet("obs","nB");
-+
+  ws->defineSet("poi","nSig");
-+
-+
+  RooStats::ModelConfig b_model = RooStats::ModelConfig("B_model", ws);
-+
+  b_model.SetPdf(*ws->pdf("combModel"));
-+
+  b_model.SetObservables(*ws->set("obs"));
-+
+  b_model.SetParametersOfInterest(*ws->set("poi"));
-+
+  ws->var("nSig")->setVal(0.0);  //# important!
-+
+  b_model.SetSnapshot(*ws->set("poi"));
-+
-+
+  RooStats::ModelConfig sb_model = RooStats::ModelConfig("S+B_model", ws);
-+
+  sb_model.SetPdf(*ws->pdf("combModel"));
-+
+  sb_model.SetObservables(*ws->set("obs"));
-+
+  sb_model.SetParametersOfInterest(*ws->set("poi"));
-+
+  ws->var("nSig")->setVal(64.0); //# important!
-+
+  sb_model.SetSnapshot(*ws->set("poi"));
-+
-+
+  RooStats::SimpleLikelihoodRatioTestStat slrts = RooStats::SimpleLikelihoodRatioTestStat(*b_model.GetPdf(),*sb_model.GetPdf());
-+
+  slrts.SetNullParameters(*b_model.GetSnapshot());
-+
+  slrts.SetAltParameters(*sb_model.GetSnapshot());
-+
+  RooStats::ProfileLikelihoodTestStat profll = RooStats::ProfileLikelihoodTestStat(*b_model.GetPdf());
-+
-+
+  RooStats::HybridCalculatorOriginal hc = RooStats::HybridCalculatorOriginal(*data, sb_model, b_model,0,0);
-+
+  hc.SetTestStatistic(2);
-+
+  hc.SetNumberOfToys(50);
-+
-+
+  RooStats::HypoTestInverterOriginal hcInv =  RooStats::HypoTestInverterOriginal(hc,*ws->var("nSig"));
-+
+  hcInv.SetTestSize(0.05);
-+
+  hcInv.UseCLs(true);
-+
+  hcInv.RunFixedScan(5,theLimit-0.5,theLimit+0.5);;
-+
+  RooStats::HypoTestInverterResult* hcInt = hcInv.GetInterval();
-+
+  float ulError = hcInt->UpperLimitEstimatedError();
-+
+  theLimit = hcInt->UpperLimit();
-+
+  cout << "Found upper limit : " << theLimit << "+/-" << ulError << endl;
-+
-+
+  return theLimit/PlottingSetup::luminosity;
-+
-+
+}
-+
-+
+TTree* SkimTree(int isample) {
-+
+  TTree* newTree = allsamples.collection[isample].events->CloneTree(0);
-+
+  float xsweight=1.0;
-+
+  if(allsamples.collection[isample].is_data==false) xsweight=luminosity*allsamples.collection[isample].weight;
-+
+  if(EdgeFitter::MarcoDebug) {
-+
+    cout << "   Original tree contains " << allsamples.collection[isample].events->GetEntries() << endl;
-+
+    cout << "   Going to reduce it with cut " << EdgeFitter::cut << endl;
-+
+  }
-+
+  TTreeFormula *select = new TTreeFormula("select", EdgeFitter::cut, allsamples.collection[isample].events);
-+
+  float wgt=1.0;
-+
+  allsamples.collection[isample].events->SetBranchAddress(cutWeight,&wgt);
-+
+  for (Int_t entry = 0 ; entry < allsamples.collection[isample].events->GetEntries() ; entry++) {
-+
+   allsamples.collection[isample].events->LoadTree(entry);
-+
+   if (select->EvalInstance()) {
-+
+     allsamples.collection[isample].events->GetEntry(entry);
-+
+     wgt=wgt*xsweight;
-+
+     newTree->Fill();
-+
+   }
-+
+  }
-+
-+
+  if(EdgeFitter::MarcoDebug) cout << "     Reduced tree contains " << newTree->GetEntries() << " entries " << endl;
-+
+  return newTree;
-+
+}
-+
-+
+void EdgeFitter::InitializeVariables(float _mllmin, float _mllmax, float _jzbmax, TCut _cut) {
-+
+  mllmin=_mllmin;
-+
+  mllmax=_mllmax;
-+
+  jzbmax=_jzbmax;
-+
+  cut=_cut;
-+
+}
-+
-+
+void EdgeFitter::PrepareDatasets(int is_data) {
-+
+  TTree *completetree;
-+
+  write_warning(__FUNCTION__,"Need to make this function ready for scans as well (use signal from scan samples)");
-+
+  bool hashit=0;
-+
+  for(int isample=0;isample<allsamples.collection.size();isample++) {
-+
+    if(!allsamples.collection[isample].is_active) continue;
-+
+    if(is_data==1&&allsamples.collection[isample].is_data==false) continue;//kick all samples that aren't data if we're looking for data.
-+
+    if(is_data==1&&allsamples.collection[isample].is_data==false) continue;//kick all samples that aren't data if we're looking for data.
-+
+    if(is_data!=1&&allsamples.collection[isample].is_data==true) continue;//kick all data samples when looking for MC
-+
+    if(is_data!=2&&allsamples.collection[isample].is_signal==true) continue;//remove signal sample if we don't want it.
-+
+    if(EdgeFitter::MarcoDebug) cout << "Considering : " << allsamples.collection[isample].samplename << endl;
-+
+    if(!hashit) {
-+
+      hashit=true;
-+
+      completetree = SkimTree(isample)->CloneTree();
-+
+    } else {
-+
+      completetree->CopyEntries(SkimTree(isample));
-+
+    }
-+
+    if(EdgeFitter::MarcoDebug) cout << "Complete tree now contains " << completetree->GetEntries() << " entries " << endl;
-+
+  }
-+
-+
+  RooRealVar mll("mll","mll",mllmin,mllmax,"GeV/c^{2}");
-+
+  RooRealVar id1("id1","id1",0,1,"GeV/c^{2}");
-+
+  RooRealVar id2("id2","id2",0,1,"GeV/c^{2}");
-+
+  RooRealVar jzb("jzb","jzb",-jzbmax,jzbmax,"GeV/c");
-+
+  RooRealVar weight("weight","weight",0,1000,"");
-+
+  RooArgSet observables(mll,jzb,id1,id2,weight);
-+
-+
+  string title="CMS Data";
-+
+  if(is_data!=1) title="CMS SIMULATION";
-+
+  RooDataSet LAllData("LAllData",title.c_str(),completetree,observables,"","weight");
-+
+  completetree->Write();
-+
+//  delete completetree;
-+
-+
+  EdgeFitter::eeSample = (RooDataSet*)LAllData.reduce("id1==id2");
-+
+  EdgeFitter::mmSample = (RooDataSet*)LAllData.reduce("id1==id2");
-+
+  EdgeFitter::emSample = (RooDataSet*)LAllData.reduce("id1!=id2");
-+
+  EdgeFitter::AllData  = (RooDataSet*)LAllData.reduce("id1!=id2||id1==id2");
-+
-+
+  eeSample->SetName("eeSample");
-+
+  mmSample->SetName("mmSample");
-+
+  emSample->SetName("emSample");
-+
+  AllData->SetName("AllData");
-+
-+
+  if(EdgeFitter::MarcoDebug) {
-+
+    cout << "Number of events in data sample = " << AllData->numEntries() << endl;
-+
+    cout << "Number of events in ee sample = " << eeSample->numEntries() << endl;
-+
+    cout << "Number of events in mm sample = " << mmSample->numEntries() << endl;
-+
+    cout << "Number of events in em sample = " << emSample->numEntries() << endl;
-+
+  }
-+
+}
-+
-+
+string EdgeFitter::RandomStorageFile() {
-+
+  TRandom3 *r = new TRandom3(0);
-+
+  int rho = (int)r->Uniform(1,10000000);
-+
+  stringstream RandomFile;
-+
+  RandomFile << PlottingSetup::cbafbasedir << "/exchange/TempEdgeFile_" << rho << ".root";
-+
+  delete r;
-+
+  return RandomFile.str();
-+
+}
-+
-+
+Yield EdgeFitter::Get_Z_estimate(float jzb_cut, int icut) {
-+
+  if(MarcoDebug) write_error(__FUNCTION__,"Returning random Z yield");
-+
+  Yield a(123,45,67); return a;
-+
+  return PlottingSetup::allresults.predictions[icut].Zbkg;
-+
+}
-+
-+
+Yield EdgeFitter::Get_T_estimate(float jzb_cut, int icut) {
-+
+  if(MarcoDebug) write_error(__FUNCTION__,"Returning random TTbar yield");
-+
+  Yield a(1234,56,78); return a;
-+
+  return PlottingSetup::allresults.predictions[icut].Flavorsym;
-+
+}
-+
-+
+void EdgeFitter::DoFit(int is_data, float jzb_cut) {
-+
+  RooRealVar mll("mll","mll",mllmin,mllmax,"GeV/c^{2}");
-+
+  RooCategory sample("sample","sample") ;
-+
+  sample.defineType("ee");
-+
+  //sample.defineType("mm");
-+
+  sample.defineType("em");
-+
+  //RooDataSet combData("combData","combined data",mll,Index(sample),Import("ee",*eeSample),Import("mm",*mmSample),Import("em",*emSample));
-+
+  RooDataSet combData("combData","combined data",mll,RooFit::Index(sample),RooFit::Import("ee",*eeSample),RooFit::Import("em",*emSample));
-+
-+
-+
-+
+  //First we make a fit to opposite flavor
-+
+  RooRealVar fttbarem("fttbarem", "fttbarem", 100, 0, 10000);
-+
+  RooRealVar par1ttbarem("par1ttbarem", "par1ttbarem", 1.6, 0.01, 4.0);
-+
+  RooRealVar par2ttbarem("par2ttbarem", "par2ttbarem", 1.0);
-+
+  RooRealVar par3ttbarem("par3ttbarem", "par3ttbarem", 0.028, 0.001, 1.0);
-+
+  RooRealVar par4ttbarem("par4ttbarem", "par4ttbarem", 2.0);
-+
+  RooSUSYBkgPdf ttbarem("ttbarem","ttbarem", mll , par1ttbarem, par2ttbarem, par3ttbarem, par4ttbarem);
-+
+  RooAddPdf model_em("model_em","model_em", ttbarem, fttbarem);
-+
+  RooSimultaneous simPdfOF("simPdfOF","simultaneous pdf", sample) ;
-+
+  simPdfOF.addPdf(model_em,"em");
-+
+  RooFitResult *resultOF = simPdfOF.fitTo(combData, RooFit::Save());
-+
+  resultOF->Print();
-+
-+
+  RooRealVar* resultOFpar1_ = (RooRealVar*) resultOF->floatParsFinal().find("par1ttbarem");
-+
+  float resultOFpar1 = resultOFpar1_->getVal();
-+
+  //RooRealVar* resultOFpar2_ = (RooRealVar*) resultOF->floatParsFinal().find("par2ttbarem");
-+
+  //float resultOFpar2 = resultOFpar2_->getVal();
-+
+  //cout << "caca2.txt" << endl;
-+
-+
+  RooRealVar* resultOFpar3_ = (RooRealVar*) resultOF->floatParsFinal().find("par3ttbarem");
-+
+  float resultOFpar3 = resultOFpar3_->getVal();
-+
-+
+  //RooRealVar* resultOFpar4_ = (RooRealVar*) resultOF->floatParsFinal().find("par4ttbarem");
-+
+  //float resultOFpar4 = resultOFpar4_->getVal();
-+
+  //cout << "caca4.txt" << endl;
-+
-+
-+
+  // Now same flavor
-+
+  RooRealVar fzee("fzee", "fzee", 5, 0, 100000);
-+
+  RooRealVar meanzee("meanzee", "meanzee", 91.1876, 89, 95);
-+
+  //RooRealVar sigmazee("sigmazee", "sigmazee", 0.5);
-+
+  RooRealVar sigmazee("sigmazee", "sigmazee", 5, 0, 100);
-+
+  RooRealVar widthzee("widthzee", "widthzee", 2.94);
-+
-+
+  RooRealVar fttbaree("fttbaree", "fttbaree", 100, 0, 100000);
-+
+  RooRealVar par1ttbaree("par1ttbaree", "par1ttbaree", resultOFpar1, 0, 100);
-+
+  RooRealVar par2ttbaree("par2ttbaree", "par2ttbaree", 1.0);
-+
+  RooRealVar par3ttbaree("par3ttbaree", "par3ttbaree", resultOFpar3, 0, 100);
-+
+  RooRealVar par4ttbaree("par4ttbaree", "par4ttbaree", 2.0);
-+
-+
+  RooRealVar fsignalee("fsignalee", "fsignalee", 10, 0, 400);
-+
+  RooRealVar par1signalee("par1signalee", "par1signalee", 45, 20, 100);
-+
+  RooRealVar par2signalee("par2signalee", "par2signalee", 2, 1, 10);
-+
+  RooRealVar par3signalee("par3signalee", "par3signalee", 45, 0, 200);
-+
-+
+  RooVoigtian zee("zee", "zee", mll, meanzee, widthzee, sigmazee);
-+
-+
-+
+  RooSUSYBkgPdf ttbaree("ttbaree","ttbaree", mll , par1ttbaree, par2ttbaree, par3ttbaree, par4ttbaree);
-+
+  //RooSUSYTPdf signalee("signalee","signalee", mll , par1signalee, par2signalee, par3signalee);
-+
+  RooSUSYTPdf signalee("signalee","signalee", mll , par1signalee, sigmazee, par3signalee);
-+
-+
+  //RooAddPdf model_ee("model_ee","model_ee", RooArgList(zee, ttbaree, signalee), RooArgList(fzee, fttbaree, fsignalee));
-+
+  RooAddPdf model_ee("model_ee","model_ee", RooArgList(zee, ttbaree, signalee), RooArgList(fzee, fttbaree, fsignalee));
-+
+  RooAddPdf model_emu("model_emu","model_emu", RooArgList(ttbaree), RooArgList(fttbaree));
-+
-+
-+
+  RooSimultaneous simPdf("simPdf","simultaneous pdf",sample) ;
-+
+  simPdf.addPdf(model_ee,"ee") ;
-+
+  simPdf.addPdf(model_emu,"em") ;
-+
-+
+  RooFitResult *result = simPdf.fitTo(combData, RooFit::Save());
-+
+  result->Print();
-+
-+
+  RooPlot* frame1 = mll.frame(RooFit::Bins(25),RooFit::Title("EE sample")) ;
-+
+  combData.plotOn(frame1,RooFit::Cut("sample==sample::ee")) ;
-+
+  simPdf.plotOn(frame1,RooFit::Slice(sample,"ee"),RooFit::ProjWData(sample,combData), RooFit::LineColor(kBlack)) ;
-+
+  simPdf.plotOn(frame1,RooFit::Slice(sample,"ee"),RooFit::Components("ttbaree"),RooFit::ProjWData(sample,combData),RooFit::LineStyle(kDashed)) ;
-+
+  simPdf.plotOn(frame1,RooFit::Slice(sample,"ee"),RooFit::Components("zee"), RooFit::ProjWData(sample, combData), RooFit::LineStyle(kDashed), RooFit::LineColor(kRed));
-+
+  simPdf.plotOn(frame1,RooFit::Slice(sample,"ee"),RooFit::Components("signalee"), RooFit::ProjWData(sample, combData), RooFit::LineStyle(kDashed), RooFit::LineColor(kGreen));
-+
-+
+  RooPlot* frame2 = mll.frame(RooFit::Bins(25),RooFit::Title("MM sample")) ;
-+
+  combData.plotOn(frame2,RooFit::Cut("sample==sample::mm")) ;
-+
+  simPdf.plotOn(frame2,RooFit::Slice(sample,"mm"),RooFit::ProjWData(sample,combData), RooFit::LineColor(kBlack)) ;
-+
+  simPdf.plotOn(frame2,RooFit::Slice(sample,"mm"),RooFit::Components("ttbarmm"),RooFit::ProjWData(sample,combData),RooFit::LineStyle(kDashed)) ;
-+
+  simPdf.plotOn(frame2,RooFit::Slice(sample,"mm"),RooFit::Components("zmm"), RooFit::ProjWData(sample, combData), RooFit::LineStyle(kDashed), RooFit::LineColor(kRed));
-+
+  simPdf.plotOn(frame2,RooFit::Slice(sample,"mm"),RooFit::Components("signalmm"), RooFit::ProjWData(sample, combData), RooFit::LineStyle(kDashed), RooFit::LineColor(kGreen));
-+
-+
-+
+  cout << "Result   : " << endl;
-+
+  cout << "f signal : " << fsignalee.getVal() << " +/- " << fsignalee.getError() << endl;
-+
+  cout << "f ttbar  : " << fttbaree.getVal() << " +/- " << fttbaree.getError() << endl;
-+
+  cout << "f tt em  : " << fttbarem.getVal() << " +/- " << fttbarem.getError() << endl;
-+
+  cout << "f z ee   : " << fzee.getVal() << " +/- " << fzee.getError() << endl;
-+
-+
+  // The same plot for the cointrol sample slice
-+
+  RooPlot* frame3 = mll.frame(RooFit::Bins(25),RooFit::Title("EM sample")) ;
-+
+  combData.plotOn(frame3,RooFit::Cut("sample==sample::em")) ;
-+
+  simPdfOF.plotOn(frame3,RooFit::Slice(sample,"em"),RooFit::ProjWData(sample,combData), RooFit::LineColor(kBlack)) ;
-+
+  simPdfOF.plotOn(frame3,RooFit::Slice(sample,"em"),RooFit::Components("ttbarem"),RooFit::ProjWData(sample,combData),RooFit::LineStyle(kDashed)) ;
-+
-+
+  stringstream prefix;
-+
+  if(is_data==data) prefix << "data_";
-+
+  if(is_data==mc) prefix << "mc_";
-+
+  if(is_data==mcwithsignal) prefix << "mcwithS_";
-+
-+
+  prefix << "JZB_" << jzb_cut;
-+
-+
+/*  cout << "fsignalee : " << fsignalee << endl;
-+
+  cout << "fttbaree : " << fttbaree << endl;
-+
+  cout << "fzee : " << fzee << endl;*/
-+
-+
-+
+  TCanvas* c = new TCanvas("rf501_simultaneouspdf","rf403_simultaneouspdf") ;
-+
+  c->cd() ;
-+
+  gPad->SetLeftMargin(0.15);
-+
+  frame1->GetYaxis()->SetTitleOffset(1.4);
-+
+  frame1->Draw();
-+
+  if(is_data==data) DrawPrelim();
-+
+  else DrawPrelim(PlottingSetup::luminosity,true);
-+
+  CompleteSave(c,"Edge/"+prefix.str()+"eemm");
-+
+  delete c;
-+
-+
+  TCanvas* d = new TCanvas("rf501_simultaneouspdf","rf403_simultaneouspdf") ;
-+
+  d->cd() ;
-+
+  gPad->SetLeftMargin(0.15);
-+
+  frame2->GetYaxis()->SetTitleOffset(1.4);
-+
+  frame2->Draw();
-+
+  if(is_data==data) DrawPrelim();
-+
+  else DrawPrelim(PlottingSetup::luminosity,true);
-+
+  CompleteSave(d,"Edge/"+prefix.str()+"mm");
-+
+  delete d;
-+
+  //c->cd(2) ; gPad->SetLeftMargin(0.15) ; frame2->GetYaxis()->SetTitleOffset(1.4) ; frame2->Draw();
-+
-+
+  TCanvas* e = new TCanvas("rf501_simultaneouspdfem","rf403_simultaneouspdfem") ;
-+
+  e->cd();
-+
+  gPad->SetLeftMargin(0.15);
-+
+  frame3->GetYaxis()->SetTitleOffset(1.4);
-+
+  frame3->Draw();
-+
+  if(is_data==data) DrawPrelim();
-+
+  else DrawPrelim(PlottingSetup::luminosity,true);
-+
+  CompleteSave(e,"Edge/"+prefix.str()+"emu");
-+
+  delete e;
-+
-+
+/*  TCanvas* f = new TCanvas("rf501_simultaneouspdfem","rf403_simultaneouspdfem") ;
-+
+  f->cd();
-+
+  gPad->SetLeftMargin(0.15);
-+
+  frame4->GetYaxis()->SetTitleOffset(1.4);
-+
+  frame4->Draw();
-+
+  if(is_data==data) DrawPrelim();
-+
+  else DrawPrelim(PlottingSetup::luminosity,true);
-+
+  CompleteSave(f,"Edge/"+prefix.str()+"eemm");
-+
+  delete f;*/
-+
-+
+}
-+
-+
+void EdgeFitter::DoEdgeFit(string mcjzb, string datajzb, float DataPeakError, float MCPeakError, float jzb_cut, int icut, int is_data, TCut cut, TTree *signalevents=0) {
-+
-+
+  TCut _cut(cut&&PlottingSetup::basiccut&&PlottingSetup::passtrig);
-+
-+
+  TFile *f = new TFile("workingfile.root","RECREATE");
-+
-+
+  EdgeFitter::InitializeVariables(PlottingSetup::iMllLow,PlottingSetup::iMllHigh,PlottingSetup::jzbHigh,_cut);
-+
-+
+  EdgeFitter::PrepareDatasets(is_data);
-+
-+
+  RooFit::MsgLevel msglevel = RooMsgService::instance().globalKillBelow();
-+
+  RooMsgService::instance().setGlobalKillBelow(RooFit::FATAL);
-+
+  EdgeFitter::DoFit(is_data, jzb_cut);
-+
+  RooMsgService::instance().setGlobalKillBelow(msglevel);
-+
-+
-+
+  f->Close();
-+
-+
+}
-+
-+
+void DoEdgeFit(string mcjzb, string datajzb, float DataPeakError, float MCPeakError, vector<float> jzb_cut, int is_data, TCut cut, TTree *signalevents=0) {
-+
+  for(int icut=0;icut<jzb_cut.size();icut++) {
-+
+    stringstream addcut;
-+
+    if(is_data==1) addcut << "(" << datajzb << ">" << jzb_cut[icut] << ")";
-+
+    if(is_data!=1) addcut << "(" << mcjzb << ">" << jzb_cut[icut] << ")";
-+
+    TCut jcut(addcut.str().c_str());
-+
-+
-+
+    EdgeFitter::DoEdgeFit(mcjzb, datajzb, DataPeakError, MCPeakError, jzb_cut[icut], icut, is_data, jcut&&cut, signalevents);
-+
-+
+  }
-+
+}

Diff Legend

-–
+Removed lines
-+
+Added lines
-<
+Changed lines
->
+Changed lines

Comparing UserCode/cbrown/Development/Plotting/Modules/EdgeLimit.C (file contents): Revision 1.1 by buchmann, Tue Jun 12 06:12:29 2012 UTC vs. Revision 1.5 by buchmann, Wed Jun 27 08:53:23 2012 UTC

Diff Legend

Comparing UserCode/cbrown/Development/Plotting/Modules/EdgeLimit.C (file contents):
Revision 1.1 by buchmann, Tue Jun 12 06:12:29 2012 UTC vs.
Revision 1.5 by buchmann, Wed Jun 27 08:53:23 2012 UTC