AliCopyHeaderTask.h

Go to the documentation of this file.

#ifndef ALICOPYHEADERTASK_H
#define ALICOPYHEADERTASK_H

#include <AliAnalysisTaskSE.h>
class AliESDVertex;
class AliMultSelection;

class AliCopyHeaderTask : public AliAnalysisTaskSE
{
public:
  AliCopyHeaderTask(const char* name="header") 
    : AliAnalysisTaskSE(name),
      fCalculateRefMult(true),
      fCopyCentrality(true),
      fCopyTracklets(false),
      fCopyV0(false),
      fCopyAD(false),
      fCopyZDC(false),
      fMultSelection(0)
  {
    fBranchNames = "ESD:AliESDHeader.,AliESDRun.";
  }
  AliCopyHeaderTask(const AliCopyHeaderTask& other);
  virtual ~AliCopyHeaderTask() {}
  AliCopyHeaderTask& operator=(const AliCopyHeaderTask& other);
  virtual void  UserCreateOutputObjects();
  virtual void   Init() {}
  virtual void   LocalInit() {Init();}
  virtual void   UserExec(Option_t *option);
  virtual void   Terminate(Option_t *option);
  /* @} */
  virtual void SetCopyOptions(const TString& what);
  virtual void SetCalculateRefMult(Bool_t calc=true) { fCalculateRefMult=calc;}
  virtual void SetCopyCentrality(Bool_t copy=true) { fCopyCentrality = copy; }
  virtual void SetCopyTracklets(Bool_t copy=true) { fCopyTracklets = copy; }
  virtual void SetCopyV0(Bool_t copy) { fCopyV0 = copy; }
  virtual void SetCopyAD(Bool_t copy) { fCopyAD = copy; }
  virtual void SetCopyZDC(Bool_t copy) { fCopyZDC = copy; }
  virtual Bool_t Connect();
protected:
  void CopyVertex(AliAODEvent& aod, const AliESDVertex* vtx, Int_t type); 
  
  Bool_t fCalculateRefMult; // Whether to calculate reference multiplicity
  Bool_t fCopyCentrality;   // Whether to copy centrality information
  Bool_t fCopyTracklets;    // Whether to copy tracklets
  Bool_t fCopyV0;           // Whether to copy V0 data 
  Bool_t fCopyAD;           // Whether to copy AD data 
  Bool_t fCopyZDC;          // Whether to copy ZDC data 
  AliMultSelection* fMultSelection; 
  
  ClassDef(AliCopyHeaderTask,3); // Task to copy header from ESD to AOD
};

#endif
/* 
 * Local Variables:
 *  mode: C++ 
 * End:
 */