CbmStackFilter.h

Go to the documentation of this file.

/* Copyright (C) 2019-2020 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Volker Friese [committer] */
 
#
#ifndef CBMSTACKFILTER_H
#define CBMSTACKFILTER_H 1
 
#include "CbmDefs.h"
 
#include "Rtypes.h"
 
#include <map>
#include <vector>
 
class TClonesArray;
class TParticle;
 
 
class CbmStackFilter {
 
public:
  typedef std::map<std::pair<Int_t, ECbmModuleId>, Int_t> PointMap;
 
 
  CbmStackFilter();
 
 
  virtual ~CbmStackFilter();
 
 
  virtual const std::vector<Bool_t>& Select(const TClonesArray& particles, const PointMap& points);
 
 
  void SetMinEkin(Double_t minimum) { fMinEkin = minimum; }
 
 
  void SetMinNofPoints(ECbmModuleId detector, UInt_t minimum)
  {
    if (detector >= ECbmModuleId::kNofSystems) return;
    fMinNofPoints[detector] = minimum;
  }
 
 
  void SetStoreAllMothers(Bool_t choice) { fStoreAllMothers = choice; }
 
 
  void SetStoreAllPrimaries(Bool_t choice) { fStoreAllPrimaries = choice; }
 
 
  void SetStoreAllPrimaryDecays(Bool_t choice = kTRUE) { fStoreAllDecays = choice; }
 
 
private:
  Bool_t fStoreAllPrimaries;                     
  Bool_t fStoreAllMothers;                       
  Bool_t fStoreAllDecays;                        
  std::map<ECbmModuleId, UInt_t> fMinNofPoints;  
  Double_t fMinEkin;                             
  std::vector<Bool_t> fStore;                    
 
  ClassDef(CbmStackFilter, 2);
};
 
#endif /* CBMSTACKFILTER_H */