Skip to content

Commit

Permalink
Add efficiency plots vs. eta and phi for cut on discriminators
Browse files Browse the repository at this point in the history
  • Loading branch information
Sebastien Wertz committed Apr 28, 2017
1 parent 40ac737 commit eb9325d
Show file tree
Hide file tree
Showing 7 changed files with 296 additions and 222 deletions.
2 changes: 1 addition & 1 deletion DQMOffline/RecoB/interface/EtaPtBin.h
View file
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@

/** \class EtaPtBin
*
* Decide if jet/parton lie within desired rapidity/pt range.
* Decide if jet/parton lie within desired abs(pseudo-rapidity)/pt range.
*
*/

Expand Down
24 changes: 17 additions & 7 deletions DQMOffline/RecoB/interface/JetTagPlotter.h
View file
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@ class JetTagPlotter : public BaseBTagPlotter {

JetTagPlotter (const std::string & tagName, const EtaPtBin & etaPtBin,
const edm::ParameterSet& pSet, const unsigned int& mc ,
const bool& willFinalize, DQMStore::IBooker & ibook, const bool & doCTagPlots = false);
const bool& willFinalize, DQMStore::IBooker & ibook, const bool & doCTagPlots = false, bool doDifferentialPlots=false, double discrCut=-999.);

virtual ~JetTagPlotter () ;

Expand Down Expand Up @@ -45,17 +45,21 @@ class JetTagPlotter : public BaseBTagPlotter {

// binning and bounds
// 1) for 'efficiency' versus discriminator cut histos
int discrBins ;
double discrStart_ ;
double discrEnd_ ;
int nBinEffPur_ ;
double startEffPur_ ;
double endEffPur_ ;
int discrBins;
double discrStart_;
double discrEnd_;
int nBinEffPur_;
double startEffPur_;
double endEffPur_;

unsigned int mcPlots_;
bool willFinalize_;

bool doCTagPlots_;

// Differential plots: efficiency vs. variable for cut on discrimator > cutValue_
bool doDifferentialPlots_;
double cutValue_;

int *nJets;
// jet multiplicity
Expand All @@ -80,6 +84,12 @@ class JetTagPlotter : public BaseBTagPlotter {

// reconstructed jet phi
FlavourHistograms<double> * dJetRecPhi;

// jet Phi larger than requested discrimnator cut
FlavourHistograms<double> * dJetPhiDiscrCut;

// jet Eta larger than requested discrimnator cut
FlavourHistograms<double> * dJetPseudoRapidityDiscrCut;
};

#endif
121 changes: 65 additions & 56 deletions DQMOffline/RecoB/plugins/BTagPerformanceAnalyzerOnData.cc
View file
Original file line number Diff line number Diff line change
Expand Up @@ -28,6 +28,12 @@ BTagPerformanceAnalyzerOnData::BTagPerformanceAnalyzerOnData(const edm::Paramete
slInfoToken = consumes<SoftLeptonTagInfoCollection>(pSet.getParameter<InputTag>("softLeptonInfo"));
jecMCToken = mayConsume<JetCorrector>(pSet.getParameter<edm::InputTag>( "JECsourceMC" ));
jecDataToken = consumes<JetCorrector>(pSet.getParameter<edm::InputTag>( "JECsourceData" ));

if (etaRanges.size() == 0)
etaRanges = { pSet.getParameter<double>("etaMin"), pSet.getParameter<double>("etaMax") };
if (ptRanges.size() == 0)
ptRanges = { pSet.getParameter<double>("ptRecJetMin"), pSet.getParameter<double>("ptRecJetMax") };

for (vector<edm::ParameterSet>::const_iterator iModule = moduleConfig.begin();
iModule != moduleConfig.end(); ++iModule) {

Expand Down Expand Up @@ -56,13 +62,13 @@ BTagPerformanceAnalyzerOnData::BTagPerformanceAnalyzerOnData(const edm::Paramete
const std::vector<InputTag> listInfo = iModule->getParameter<vector<InputTag>>("listTagInfos");
for(unsigned int ITi=0; ITi<listInfo.size(); ITi++){
tokens.push_back(consumes< View<BaseTagInfo> >(listInfo[ITi]));
vIP.push_back(listInfo[ITi]);
}
vIP.push_back(listInfo[ITi]);
}
}
else {
const InputTag& moduleLabel = iModule->getParameter<InputTag>("label");
tokens.push_back(consumes< View<BaseTagInfo> >(moduleLabel));
vIP.push_back(moduleLabel);
vIP.push_back(moduleLabel);
}
tagInfoToken.push_back(tokens);
tagInfoInputTags.push_back(vIP);
Expand Down Expand Up @@ -95,17 +101,22 @@ void BTagPerformanceAnalyzerOnData::bookHistograms(DQMStore::IBooker & ibook, ed

// eta loop
for ( int iEta = iEtaStart ; iEta < iEtaEnd ; ++iEta ) {
// pt loop
for ( int iPt = iPtStart ; iPt < iPtEnd ; ++iPt ) {

const EtaPtBin& etaPtBin = getEtaPtBin(iEta, iPt);

// Instantiate the genertic b tag plotter
JetTagPlotter *jetTagPlotter = new JetTagPlotter(folderName, etaPtBin,
iModule->getParameter<edm::ParameterSet>("parameters"), 0, false, ibook);
binJetTagPlotters.at(iTag).push_back ( jetTagPlotter ) ;

}
// pt loop
for ( int iPt = iPtStart ; iPt < iPtEnd ; ++iPt ) {

const EtaPtBin& etaPtBin = getEtaPtBin(iEta, iPt);

// Instantiate the genertic b tag plotter
bool doDifferentialPlots = false;
double discrCut = -999.;
if (iModule->exists("differentialPlots") && iModule->getParameter<bool>("differentialPlots") == true) {
doDifferentialPlots = true;
discrCut = iModule->getParameter<double>("discrCut");
}
JetTagPlotter *jetTagPlotter = new JetTagPlotter(folderName, etaPtBin, iModule->getParameter<edm::ParameterSet>("parameters"), 0, false, ibook, false, doDifferentialPlots, discrCut);
binJetTagPlotters.at(iTag).push_back ( jetTagPlotter ) ;

}
}

} else if(dataFormatType == "TagCorrelation") {
Expand Down Expand Up @@ -133,18 +144,18 @@ void BTagPerformanceAnalyzerOnData::bookHistograms(DQMStore::IBooker & ibook, ed

// eta loop
for ( int iEta = iEtaStart ; iEta < iEtaEnd ; ++iEta ) {
// pt loop
for ( int iPt = iPtStart ; iPt < iPtEnd ; ++iPt ) {
const EtaPtBin& etaPtBin = getEtaPtBin(iEta, iPt);
// Instantiate the tagInfo plotter
BaseTagInfoPlotter *jetTagPlotter = theFactory.buildPlotter(dataFormatType, moduleLabel.label(),
etaPtBin, iModule->getParameter<edm::ParameterSet>("parameters"), folderName,
0, false, ibook);
binTagInfoPlotters.at(iInfoTag).push_back ( jetTagPlotter ) ;
// pt loop
for ( int iPt = iPtStart ; iPt < iPtEnd ; ++iPt ) {
const EtaPtBin& etaPtBin = getEtaPtBin(iEta, iPt);
// Instantiate the tagInfo plotter
BaseTagInfoPlotter *jetTagPlotter = theFactory.buildPlotter(dataFormatType, moduleLabel.label(),
etaPtBin, iModule->getParameter<edm::ParameterSet>("parameters"), folderName,
0, false, ibook);
binTagInfoPlotters.at(iInfoTag).push_back ( jetTagPlotter ) ;
binTagInfoPlottersToModuleConfig.insert(make_pair(jetTagPlotter, iModule - moduleConfig.begin()));
}
}
}
}
}
Expand All @@ -154,31 +165,30 @@ EtaPtBin BTagPerformanceAnalyzerOnData::getEtaPtBin(const int& iEta, const int&
{
// DEFINE BTagBin:
bool etaActive_ , ptActive_;
double etaMin_, etaMax_, ptMin_, ptMax_ ;
double etaMin_, etaMax_, ptMin_, ptMax_;

if ( iEta != -1 ) {
etaActive_ = true ;
etaMin_ = etaRanges[iEta] ;
etaMax_ = etaRanges[iEta+1] ;
etaActive_ = true;
etaMin_ = etaRanges[iEta];
etaMax_ = etaRanges[iEta+1];
}
else {
etaActive_ = false ;
etaMin_ = etaRanges[0] ;
etaMax_ = etaRanges[etaRanges.size() - 1] ;
etaActive_ = false;
etaMin_ = etaRanges[0];
etaMax_ = etaRanges[etaRanges.size() - 1];
}

if ( iPt != -1 ) {
ptActive_ = true ;
ptMin_ = ptRanges[iPt] ;
ptMax_ = ptRanges[iPt+1] ;
ptActive_ = true;
ptMin_ = ptRanges[iPt];
ptMax_ = ptRanges[iPt+1];
}
else {
ptActive_ = false ;
ptMin_ = ptRanges[0] ;
ptMax_ = ptRanges[ptRanges.size() - 1] ;
ptActive_ = false;
ptMin_ = ptRanges[0];
ptMax_ = ptRanges[ptRanges.size() - 1];
}
return EtaPtBin(etaActive_ , etaMin_ , etaMax_ ,
ptActive_ , ptMin_ , ptMax_ );
return EtaPtBin(etaActive_, etaMin_, etaMax_, ptActive_, ptMin_ , ptMax_);
}

BTagPerformanceAnalyzerOnData::~BTagPerformanceAnalyzerOnData()
Expand Down Expand Up @@ -238,24 +248,23 @@ void BTagPerformanceAnalyzerOnData::analyze(const edm::Event& iEvent, const edm:
LogDebug("Info") << "Found " << tagColl.size() << " B candidates in collection " << jetTagInputTags[iJetLabel];

int plotterSize = binJetTagPlotters[iJetLabel].size();
for (JetTagCollection::const_iterator tagI = tagColl.begin();
tagI != tagColl.end(); ++tagI) {
for (JetTagCollection::const_iterator tagI = tagColl.begin(); tagI != tagColl.end(); ++tagI) {

//JEC
reco::Jet correctedJet = *(tagI->first);
double jec = 1.0;
if(doJEC && corrector) {
jec = corrector->correction(*(tagI->first));
jec = corrector->correction(*(tagI->first));
}

if (!jetSelector(*(tagI->first), -1, infoHandle, jec))
continue;
continue;

for (int iPlotter = 0; iPlotter != plotterSize; ++iPlotter) {
bool inBin = binJetTagPlotters[iJetLabel][iPlotter]->etaPtBin().inBin(*tagI->first, jec);
// Fill histograms if in desired pt/rapidity bin.
if (inBin)
binJetTagPlotters[iJetLabel][iPlotter]->analyzeTag(*tagI, jec, -1);
bool inBin = binJetTagPlotters[iJetLabel][iPlotter]->etaPtBin().inBin(*tagI->first, jec);
// Fill histograms if in desired pt/rapidity bin.
if (inBin)
binJetTagPlotters[iJetLabel][iPlotter]->analyzeTag(*tagI, jec, -1);
}
}
for (int iPlotter = 0; iPlotter != plotterSize; ++iPlotter) {
Expand All @@ -280,7 +289,7 @@ void BTagPerformanceAnalyzerOnData::analyze(const edm::Event& iEvent, const edm:
reco::Jet correctedJet = *(tagI->first);
double jec = 1.0;
if(doJEC && corrector) {
jec = corrector->correction(*(tagI->first));
jec = corrector->correction(*(tagI->first));
}

if (!jetSelector(*(tagI->first), -1, infoHandle, jec))
Expand Down Expand Up @@ -323,8 +332,8 @@ void BTagPerformanceAnalyzerOnData::analyze(const edm::Event& iEvent, const edm:
// protect against missing products
//
if (tagInfoHandle.isValid() == false){
edm::LogWarning("BTagPerformanceAnalyzerOnData")<<" Collection "<<tagInfoInputTags[iJetLabel][iInputTags]<<" not present. Skipping it for this event.";
continue;
edm::LogWarning("BTagPerformanceAnalyzerOnData")<<" Collection "<<tagInfoInputTags[iJetLabel][iInputTags]<<" not present. Skipping it for this event.";
continue;
}

unsigned int size = tagInfoHandle->size();
Expand Down Expand Up @@ -355,17 +364,17 @@ void BTagPerformanceAnalyzerOnData::analyze(const edm::Event& iEvent, const edm:
reco::Jet correctedJet = *(jetRef);
double jec = 1.0;
if(doJEC && corrector) {
jec = corrector->correction(*(jetRef));
jec = corrector->correction(*(jetRef));
}

if (!jetSelector(*jetRef, -1, infoHandle, jec))
continue;

for (int iPlotter = 0; iPlotter != plotterSize; ++iPlotter) {
bool inBin = binTagInfoPlotters[iJetLabel][iPlotter]->etaPtBin().inBin(*jetRef, jec);
// Fill histograms if in desired pt/rapidity bin.
if (inBin)
binTagInfoPlotters[iJetLabel][iPlotter]->analyzeTag(baseTagInfos, jec, -1);
bool inBin = binTagInfoPlotters[iJetLabel][iPlotter]->etaPtBin().inBin(*jetRef, jec);
// Fill histograms if in desired pt/rapidity bin.
if (inBin)
binTagInfoPlotters[iJetLabel][iPlotter]->analyzeTag(baseTagInfos, jec, -1);
}
}
}
Expand Down
Loading

0 comments on commit eb9325d

Please sign in to comment.