AtDataReductionTask.h

Go to the documentation of this file.

#ifndef ATDATAREDUCTIONTSAK_H
#define ATDATAREDUCTIONTSAK_H
 
#include <FairTask.h>
 
#include <Rtypes.h>
#include <TString.h>
 
#include <algorithm>  // for max
#include <functional> // for function
#include <utility>    // for move
#include <vector>     // for vector
 
class AtBaseEvent;
class TClonesArray;
class TBuffer;
class TClass;
class TMemberInspector;
 
class AtDataReductionTask : public FairTask {
private:
   using ReductionFunction = std::function<bool()>;
 
   ReductionFunction fReductionFunc{nullptr};
 
   TClonesArray *fInputEventArray{nullptr}; //< AtRawEvent branch object
   TString fInputBranchName{"AtRawEvent"};  //< Name of AtRawEvent branch to mark as good/not good
   AtBaseEvent *fEvent;
   std::vector<TString> fOutputBranchs;
 
public:
   AtDataReductionTask() = default;
   ~AtDataReductionTask() = default;
 
   void SetReductionFunction(ReductionFunction func) { fReductionFunc = std::move(func); }
   void SetReductionFunction(std::function<bool(AtBaseEvent *)> func);
 
   template <class T>
   void SetReductionFunction(std::function<bool(T *)> func)
   {
      fReductionFunc = [this, func]() { return func(dynamic_cast<T *>(fEvent)); };
   }
   void SetInputBranch(TString inputBranch)
   {
      fInputBranchName = inputBranch;
      fOutputBranchs.push_back(inputBranch);
   }
   void SetOutputBranch(TString outputBranch) { fOutputBranchs.push_back(outputBranch); }
 
   virtual InitStatus Init() override;
   virtual void Exec(Option_t *opt) override;
 
   ClassDefOverride(AtDataReductionTask, 2)
};
 
#endif //#ifndef ATDATAREDUCTIONTSAK_H