PndMCTrackInfoAnaTask.h

Go to the documentation of this file.

#pragma once

#include "PndPersistencyTask.h"
#include "PndMCTrackInfo.h"
#include "PndMCTrack.h"

#include "PndEventRequirements.h"
#include "PndEventCheck.h"

#include "TClonesArray.h"

#include <string>
#include <vector>
#include <map>

class TClonesArray;

class PndMCTrackInfoAnaTask : public PndPersistencyTask {
 public:
  PndMCTrackInfoAnaTask();

  virtual ~PndMCTrackInfoAnaTask();

  virtual void SetParContainers();
  virtual InitStatus Init();
  virtual InitStatus ReInit();

  virtual void Exec(Option_t *opt);

  void SetEventRequirements(PndEventRequirements eventRequirements) { fEventCheck.SetRequirements(eventRequirements); }

 protected:
  void Register();
  void Reset();
  void FinishTask();
  void SetupEventRequirements();

  void AssignTClonesArrays(std::map<TString, TClonesArray *> &map);
  void AnalyseResult(PndEventCheck &eventCheck);
  int GetPointOfFailure(std::vector<int> result);
  void AddPointOfFailure(int requirement, int position);
  void InitPointOfFailure();
  void InitInterestingEvents();
  std::vector<std::vector<std::vector<int>>> GetInterestingEvents() const { return fInterestingEvents; }  

 private:
  PndEventCheck fEventCheck;
  TClonesArray *fMCTracks = nullptr;
  TClonesArray *fMCTrackInfo = nullptr;
  TList *fBranchList = nullptr;
  std::vector<int> fReqResult;
  std::vector<std::vector<int>> fPointOfFailure;
  std::vector<std::vector<std::vector<int>>> fInterestingEvents;

  ClassDef(PndMCTrackInfoAnaTask, 1);
};