CoincidenceMatrix.hh

Go to the documentation of this file.

#ifndef GAMRSORT_COINCIDENCEMATRIX_HH
#define GAMRSORT_COINCIDENCEMATRIX_HH
 
#include "Sorter.hh"
 
#include <string>
#include <vector>
 
#include <TTreeReader.h>
#include <TTreeReaderValue.h>
 
namespace GamR {
namespace Sort {
namespace Type {
 
template <class DetX, class DetY, size_t X, size_t Y>
class CoincidenceMatrix {
private:
   Sorter *p;
   std::shared_ptr<TH2D> ptrMatrix;
   std::string Prefix;
   std::string GroupX;
   std::string GroupY;
 
public:
   CoincidenceMatrix(Sorter *parent, std::vector<std::string> opts) : p(parent)
   {
      if (opts.size() < 3)
         throw std::length_error("Invalid arguments, expect: Prefix GroupX GroupY");
      Prefix = opts[0];
      GroupX = opts[1];
      GroupY = opts[2];
      std::string name(Prefix + "_g" + GroupX + "m" + X + "g" + GroupY + "m" + Y);
      p->SetXsize(4096, 0, 4096);
      p->SetYsize(4096, 0, 4096);
      ptrMatrix = p->GetMatrix(name);
      // ptrMatrix -> SetBins(4096, 0, 4096, 4096, 0, 4096); // made knob.
   }
 
   void operator()(TTreeReader &R)
   {
      TTreeReaderValue<std::vector<DetX>> rDetsX(R, ("DetGroup._" + GroupX).c_str());
      TTreeReaderValue<std::vector<DetY>> rDetsY(R, ("DetGroup._" + GroupY).c_str());
      while (R.Next() && !p->Done()) {
         p->Tick();
         std::vector<DetX> vecDetsX = *rDetsX;
         std::vector<DetY> vecDetsY = *rDetsY;
         if (vecDetsX.size() + vecDetsY.size() < 2) {
            continue;
         }
         for (auto &Det1 : vecDetsX) {
            for (auto &Det2 : vecDetsY) {
               if (Det1.ID == Det2.ID) {
                  continue;
               }
               auto x = Det1.template GetCal<X>();
               auto y = Det2.template GetCal<Y>();
               if (!(x && y)) {
                  continue;
               }
               ptrMatrix->Fill(x, y);
            } // fold 2
         }    // fold 1
      }       // all events
   }          // Sorter::CoincidenceMatrix::operator()
};
 
} // namespace Type
} // namespace Sort
} // namespace GamR
 
#endif