CbmEbEventEfficiencies.h

Go to the documentation of this file.

 
#include "CbmL1Counters.h"
 
#include <cstring>
#include <iostream>
 
using std::string;
 
class CbmEbEventEfficiencies {
public:
  CbmEbEventEfficiencies()
    : names()
    , indices()
    , ratio_reco()
    , ratio_ghosts(0)
    , mc()
    , reco()
    , ghosts(0)
    , nEvents(0)
    , ratio_clone()
    , clone() {
    AddCounter("reconstructable", "Reconstructable tracks   ");
    AddCounter("reconstructed", "Reconstructed tracks     ");
  }
 
  ~CbmEbEventEfficiencies() {};
 
  void AddCounter(string shortname, string name) {
    indices[shortname] = names.size();
    names.push_back(name);
 
    ratio_reco.AddCounter();
    mc.AddCounter();
    reco.AddCounter();
    ratio_clone.AddCounter();
    clone.AddCounter();
  }
 
  CbmEbEventEfficiencies& operator+=(CbmEbEventEfficiencies& a) {
    mc += a.mc;
    reco += a.reco;
    clone += a.clone;
    ghosts += a.ghosts;
    nEvents += a.nEvents;
    return *this;
  }
 
  void CalcEff() {
    ratio_reco         = reco / mc;
    const double total = reco.counters[indices["reconstructed"]] + ghosts
                         + clone.counters[indices["reconstructed"]];
    if (total > 0)
      ratio_ghosts = ghosts / total;
    else
      ratio_ghosts = -1;
    ratio_clone = clone / mc;
  }
 
 
  void Inc(bool isReco, int _nclones, string name) {
    const int index = indices[name];
    mc.counters[index]++;
    if (isReco) reco.counters[index]++;
    clone.counters[index] += _nclones;
  }
 
  void IncNEvents() { nEvents++; };
 
  void AddGhost(int i) { ghosts += i; }
 
  void PrintEff() {
 
    std::cout.setf(ios::fixed);
    std::cout.setf(ios::showpoint);
    std::cout.precision(3);
    std::cout << "Event category normalised to: "
              << " Eff  "
              << " / "
              << "Clones"
              << " / "
              << "All Reco"
              << " | "
              << "All MC" << std::endl;
 
    int NCounters = mc.NCounters;
    for (int iC = 0; iC < NCounters; iC++) {
      std::cout << names[iC] << "   : " << ratio_reco.counters[iC] << "  / "
                << ratio_clone.counters[iC]  // nCloneTracks/nMCTracks
                << "  / " << setw(8) << reco.counters[iC] / double(nEvents)
                << " | " << setw(8) << mc.counters[iC] / double(nEvents)
                << std::endl;
    }
    std::cout << "Ghost     probability       : " << ratio_ghosts << " | "
              << ghosts << std::endl;
  }
 
private:
  vector<string> names;  // names counters indexed by index of counter
  map<string, int>
    indices;  // indices of counters indexed by a counter shortname
 
  TL1TracksCatCounters<double> ratio_reco;
  double ratio_ghosts;
 
  TL1TracksCatCounters<int> mc;
  TL1TracksCatCounters<int> reco;
  int ghosts;
  int nEvents;
 
  TL1TracksCatCounters<double> ratio_clone;
  TL1TracksCatCounters<int> clone;
};