NCExtractionPDF.cxx

Go to the documentation of this file.

// NCExtractionPDF.cxx
//
// A simple NC/CC separation class - PDFs based on TOm's variables
//
// C. Backhouse 06/2009

#include "NCUtils/Extraction/NCExtractionPDF.h"

#include "NCUtils/Cuts/NCAnalysisCutsNC.h"
#include "NCUtils/Extraction/MicroDSTMaker.h"

#include "MessageService/MsgService.h"

#include "TH2.h"

#include <cassert>

REGISTER_NCEXTRACTION(NCExtractionPDF, PDF)

ClassImp(NCExtractionPDF)

CVSID("$Id: NCExtractionPDF.cxx,v 1.2 2009/09/12 16:26:51 rodriges Exp $");

//----------------------------------------------------------------------
NCExtractionPDF::NCExtractionPDF(NCAnalysisCuts* cuts,
                                 const Registry& r)
  : NCExtraction(cuts, r)
{
  const char* tmps;
  TString trainingFilePath;
  if(r.Get("ExtractionPDFTrainingFilePath", tmps)) trainingFilePath = tmps;
  assert(trainingFilePath != "");

  TFile fin(trainingFilePath, "READ");
  fLikelihoodHist = (TH2*)fin.Get("hFrac");
  fLikelihoodHist->SetDirectory(0);
  fin.Close();

}

//----------------------------------------------------------------------
NCExtractionPDF::~NCExtractionPDF()
{
  if(fLikelihoodHist) delete fLikelihoodHist;
}

//----------------------------------------------------------------------
double NCExtractionPDF::GetIdProbability(NCEventInfo& evtInfo,
                                         int)
{
  assert(evtInfo.reco);
  assert(fLikelihoodHist);

  double idProb = 0;

  if(!evtInfo.reco->numberTracks){
    idProb = 1;
  }
  else{
    const double ext = evtInfo.reco->showerPlanes - evtInfo.reco->trackPlanes;

    const int bin = fLikelihoodHist->FindBin(evtInfo.reco->planes, ext);
    const double lhoodVal = fLikelihoodHist->GetBinContent(bin);

    if(lhoodVal == 0){
      idProb = (ext > 0) ? 1 : 0;
    }
    else{
      idProb = lhoodVal;
    }
  }

  return idProb;
}

---