CbmCutMap.h

Go to the documentation of this file.

/* Copyright (C) 2023 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Dario Ramirez [committer] */
 
#ifndef CBMCUTMAP_H
#define CBMCUTMAP_H
 
#include "CbmCut.h"
#include "CbmEvent.h"
#include "CbmStsHit.h"
#include "CbmStsUtils.h"
 
#include <TClonesArray.h>

class CbmCutMap {
 public:
  CbmCutMap()  = default;
  ~CbmCutMap() = default;

  std::unordered_map<CbmCutId, CbmCut<float>> GetMap() const { return fCbmCuts; }

  CbmCut<float>* AddCbmCut(CbmCutId id);

  bool Check(CbmCutId id, double value);

  bool CheckStsHit(CbmStsHit*, TClonesArray*);

  bool CheckEvent(CbmEvent* evt);

  friend std::ostream& operator<<(std::ostream& out, const CbmCutMap& obj)
  {
    for (auto& [id, cut] : obj.GetMap()) {
      out << ToString(id) << "\t" << cut << std::endl;
    }
    return out;
  }

  void Print()
  {
    for (auto& [id, cut] : fCbmCuts) {
      std::cout << int(id) << ": " << cut << "\tFailed: " << fFailedCounter[id] << std::endl;
    }
  }
 
 private:
  std::unordered_map<CbmCutId, CbmCut<float>> fCbmCuts;
  std::unordered_map<CbmCutId, unsigned long int> fFailedCounter;
};
 
#endif