OnlineDBuilderTask.h

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----                 PNDMCMATCHSELECTORTASK header file             -----
// -----                  Created 18/01/10  by T.Stockmanns             -----
// -------------------------------------------------------------------------

#ifndef OnlineDBuilderTask_H
#define OnlineDBuilderTask_H

// framework includes
#include "FairTask.h"
#include "PndDetectorList.h"
#include "PndTrackCand.h"
#include "TH2.h"
#include "THStack.h"

#include "TLorentzVector.h"

#include <vector>
#include <map>

class TClonesArray;
class PndTrack;
class FairTrackParP;

class OnlineDBuilderTask : public FairTask {
 public:
  OnlineDBuilderTask(TString trackBranchName);

  virtual ~OnlineDBuilderTask();

  virtual void SetParContainers();
  virtual InitStatus Init();
  virtual void AddTrackBranch(TString trackBranchName) { fTrack[trackBranchName] = 0; }

  virtual void Exec(Option_t *opt);

  virtual void Finish();

  // void SetTrackBranchName(TString name) {fTrackBranchName = name;} //alternatively PndTrackCandData can be analysed. Therefore pndTrackData has to be set to kFALSE

 private:
  //
  //  virtual void FillMapTrackQualifikation();

  std::vector<std::pair<TLorentzVector, int>> CombineFirstWithTwoSecond(std::vector<PndTrack *> first, std::vector<PndTrack *> second);
  TVector3 CorrectTrackParForPrimaryVertex(FairTrackParP input);
  std::map<TString, TClonesArray *> fTrack; 
  TClonesArray *fMCTrack;

  TClonesArray *fDPlusArray;
  TClonesArray *fDMinusArray;

  TClonesArray *fDPlusArrayTrue;
  TClonesArray *fDMinusArrayTrue;

  FairRootManager *ioman;

  void Register();

  void Reset();

  ClassDef(OnlineDBuilderTask, 1);
};

#endif