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