BmnFillDstTask.h

Go to the documentation of this file.

/********************************************************************************
 *    BmnFillDstTask.h                                                          *
 *    BM@N Fill DST Task class declaration                                      *
 *    Add necessary data and event header to the DST file after reconstruction  *
 *******************************************************************************/
 
#ifndef BMNFILLDSTTASK_H
#define BMNFILLDSTTASK_H
 
#include "BmnEventHeader.h"
#include "BmnMCInfoDst.h"
#include "BmnTask.h"
#include "CbmVertex.h"
#include "DstEventHeader.h"
#include "DstRunHeader.h"
#include "FairMCEventHeader.h"
#include "TClonesArray.h"
 
#include <map>
 
class BmnFillDstTask : public BmnTask
{
  public:
    BmnFillDstTask();
 
    BmnFillDstTask(Long64_t nStartEvent, Long64_t nEvents);
 
    BmnFillDstTask(Long64_t nStartEvent,
                   Long64_t nEvents,
                   Bool_t isExp,
                   Int_t period_number = -1,
                   Int_t run_number = -1);
 
    ~BmnFillDstTask();
 
    virtual InitStatus Init();
 
    virtual InitStatus ReInit();
 
    virtual void Exec(Option_t* opt);
 
    virtual void SetParContainers();
 
    virtual void Finish();
 
    void SetRunNumber(Int_t period_number, Int_t run_number)
    {
        fPeriodNumber = period_number;
        fRunNumber = run_number;
    }
    void DoZCalibration(Bool_t cal) { fDoCalibration = cal; }
 
    void InitParticleInfo();
 
    struct stParticleInfo
    {
        Int_t A;
        Int_t Z;
    };
 
  private:
    int isExpData;
    Int_t fPeriodNumber;
    Int_t fRunNumber;
 
    FairMCEventHeader* fMCEventHeader;   
    TClonesArray* fMCTracks;   
    BmnMCInfoDst* fMCInfoDst;   
    BmnEventHeader* fEventHeader;   
    DstEventHeader* fDstHeader;   
    DstRunHeader* fRunHeader;   
    Long64_t fNStartEvent;
    Long64_t fNEvents;
    Long64_t fIEvent;
    CbmVertex* fPrimaryVertex;
    TClonesArray* fGlobalTracks;
 
    Double_t fZCalib1;
    Double_t fZCalib2;
    Double_t fBC1Calib;
    Double_t fBC2Calib;
    Bool_t fDoCalibration;
 
    /*For ADC, charge calculation*/
    TClonesArray *fT0, *fBC1, *fBC2, *fBC3, *fBC4;   
 
    map<TString, stParticleInfo> mapParticleInfo;
 
    BmnFillDstTask(const BmnFillDstTask&) = delete;
    BmnFillDstTask operator=(const BmnFillDstTask&) = delete;
 
    void Reset();
 
    void applyZCalibration();
    bool autoSetHeaders();
 
    ClassDef(BmnFillDstTask, 2);
};
 
#endif