CbmMCEventFilter.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 CBMMCEVENTFILTER_H
#define CBMMCEVENTFILTER_H 1
 
#include "CbmDefs.h"
 
#include "FairMCApplication.h"
#include "FairTask.h"
 
#include "TClonesArray.h"
 
#include <map>
#include <string>
#include <vector>
 
class TClonesArray;
 
 
class CbmMCEventFilter : public FairTask {
 
public:
  CbmMCEventFilter();
 
 
  virtual ~CbmMCEventFilter() {};
 
 
  Int_t GetNofInputEvents() const { return fNofEventsIn; }
 
 
  void SetMinNofData(ECbmDataType type, Int_t value) { fMinNofData[type] = value; }
 
 
protected:
  TObject* GetData(ECbmDataType type, Int_t index) const;
 
 
  Int_t GetNofData(ECbmDataType type) const
  {
    return (fData.at(type) == nullptr ? 0 : fData.at(type)->GetEntriesFast());
  }
 
 
  Bool_t SelectEvent() const;
 
 
private:
  std::map<ECbmDataType, TClonesArray*> fData;  
  //            std::vector<TClonesArray*> fData;          //!  Data arrays
  std::map<ECbmDataType, Int_t> fMinNofData;  
  Int_t fNofEventsIn;                         
  Int_t fNofEventsOut;                        
 
  virtual void Exec(Option_t*);
 
 
  virtual void Finish();
 
 
  virtual InitStatus Init();
 
 
  void GetBranch(ECbmDataType type);
 
 
  std::string Statistics() const;
 
 
  TString GetBranchName(ECbmDataType type) const;
 
 
  ClassDef(CbmMCEventFilter, 2);
};
 
#endif /* CBMMCEVENTFILTER_H */