KVEventClassifier.h

Go to the documentation of this file.

#ifndef __KVEVENTCLASSIFIER_H
#define __KVEVENTCLASSIFIER_H
 
#include "KVVarGlob1.h"
 
class KVEventClassifier : public KVVarGlob1 {
    KVVarGlob* fVar;
    std::vector<double> fCuts;
    bool fBcuts;
    bool fBins;
 
   int calc_where() const;
 
public:
   KVEventClassifier() : KVVarGlob1("KVEventClassifier"), fVar(nullptr), fBcuts(false), fBins(false) {}
   KVEventClassifier(KVVarGlob* b) : KVVarGlob1(Form("%s_EC", b->GetName())), fVar(b), fBcuts(false), fBins(false) {}
   virtual ~KVEventClassifier() {}
   ROOT_COPY_CTOR(KVEventClassifier, KVVarGlob1)
   ROOT_COPY_ASSIGN_OP(KVEventClassifier)
 
   void Copy(TObject& other) const
   {
      KVVarGlob1::Copy(other);
      ((KVEventClassifier&)other).fVar = fVar;
      ((KVEventClassifier&)other).fCuts = fCuts;
      ((KVEventClassifier&)other).fBcuts = fBcuts;
      ((KVEventClassifier&)other).fBins = fBins;
   }
   Bool_t IsGlobalVariable() const
   {
      return kFALSE;
   }
   int GetEventClassification() const
   {
 
      int where = calc_where();
      if (fBins) {
         if (where % 2) {
            return where / 2;
         }
         return -1;
      }
      return where;
   }
   int GetMaxEventClassNumber() const
   {
 
      if (fBins) return fCuts.size() / 2;
      return fCuts.size();
   }
 
   void AddCut(double x)
   {
 
      if (fBins) {
         Error("AddCut", "You cannot mix cuts & bins (and you already defined some bins)");
         return;
      }
      fBcuts = true;
      fCuts.push_back(x);
   }
   void AddBin(double xmin, double xmax)
   {
 
      if (fBcuts) {
         Error("AddBin", "You cannot mix cuts & bins (and you already defined some cuts)");
         return;
      }
      fBins = true;
      fCuts.push_back(xmin);
      fCuts.push_back(xmax);
   }
 
   void Init()
   {
      std::sort(fCuts.begin(), fCuts.end());
   }
   void Reset() {}
   void Calculate()
   {
      SetValue(GetEventClassification());
   }
 
   ClassDef(KVEventClassifier, 1) //Simple class for sorting events according to global variables
};
 
#endif