PndHypGeIdealTrackFinderTask.h

Go to the documentation of this file.

#ifndef PNDHYPGEIDEALTRACKFINDERTASK_H
#define PNDHYPGEIDEALTRACKFINDERTASK_H

#include "FairTask.h"
#include "PndMCTrack.h"

#include "TH1.h"
#include "PndHypGePoint.h"

//#include "GFTrackCand.h"
//#include "PndHypCluster.h"
#include <string>
#include <vector>
#include <map>

class TClonesArray;

class PndHypGeIdealTrackFinderTask : public FairTask {
 public:
  PndHypGeIdealTrackFinderTask();
  virtual ~PndHypGeIdealTrackFinderTask();

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

  virtual void Exec(Option_t *opt);

  void WriteHistograms();

 private:
  TString fMcBranch;
  TString fTrackBranch;

  TClonesArray *fMcArray;
  TClonesArray *fTrackArray;

  TH1F *hismom;
  int fEventNr;
  std::map<int, std::vector<int>> mcHitMap;
  Int_t fVerbose;

  // void Register();
  // void Reset();
  // void ProduceHits();
  // std::map<int, std::vector<int> > AssignHitsToTracks();

  ClassDef(PndHypGeIdealTrackFinderTask, 1);
};

#endif /* PndHypGeIdealTrackFinderTASK_H */