fixed bugs while streaming histos for weighting, added new trainconfig for pPb

[u/mrichter/AliRoot.git] / PWGGA / GammaConv / AliAnalysisTaskGCPartToPWG4Part.h

#ifndef AliAnalysisTaskGCPartToPWG4Part_cxx\r
#define AliAnalysisTaskGCPartToPWG4Part_cxx\r
\r
// example of an analysis task creating a p_t spectrum\r
// Authors: Svein Lindal\r
\r
class TH1F;\r
class AliESDEvent;\r
class AliAODConversionPhoton;\r
class AliAODPWG4ParticleCorrelation;\r
class AliAODPWG4Particle;\r
class TClonesArray;\r
class TString;\r
class AliMCAnalysisUtils;\r
class AliAODMCHeader;\r
\r
#include "AliAnalysisTaskSE.h"\r
\r
class AliAnalysisTaskGCPartToPWG4Part : public AliAnalysisTaskSE {\r
\r
public:\r
\r
  AliAnalysisTaskGCPartToPWG4Part(); \r
  AliAnalysisTaskGCPartToPWG4Part(const char *name);\r
  virtual ~AliAnalysisTaskGCPartToPWG4Part();\r
  \r
  virtual void   UserCreateOutputObjects();\r
  virtual void   UserExec(Option_t *option);\r
  virtual void   Terminate(Option_t *);\r
\r
  void SetDeltaAODFileName(TString string) { fDeltaAODFileName = string;}\r
  void SetGammaBranchName(TString string) { fAODBranchName = string; }\r
 \r
\r
  void SetDebugLevel(Int_t debugLevel) { fDebugLevel = debugLevel; }\r
  Int_t GetDebugLevel() const { return fDebugLevel; }\r
\r
  void SetGammaCutId(TString cut) { fGammaCutString = Form("GammaConv_%s", cut.Data());}\r
  void SetPionCutId(TString cut) { fPionCutString = Form("GammaConv_%s", cut.Data());}\r
\r
  \r
 private:\r
\r
  //Clean up\r
  void CleanUp();\r
\r
  //Get the AOD event from whereever it might be accessible\r
  AliAODEvent * GetAODEvent();\r
\r
  Bool_t BothTracksPresent(const AliAODConversionPhoton * const photon, const TClonesArray * const tracks) const;\r
  Bool_t BothGammaPresent(const AliAODConversionPhoton * const pion, const TClonesArray * const photons, const TClonesArray * const tracks) const;\r
\r
  //Get Conversion gammas branch\r
  TClonesArray * GetConversionGammas(const AliAODEvent * aodEvent) const;\r
  TClonesArray * GetPions(const AliAODEvent * aodEvent) const;\r
  TClonesArray * GetAODBranch(const AliAODEvent * aodEvent, TString branchName) const;\r
\r
  //Fill AOD tree with PWG4 particles\r
  AliAODPWG4ParticleCorrelation * AddToAOD(AliAODConversionPhoton * aodO, TClonesArray * branch, TString detector);\r
  AliAODPWG4ParticleCorrelation * AddPionToAOD(AliAODConversionPhoton * pion, TClonesArray * branch, TString detector);  \r
  //Process conv gamma\r
  void ProcessConvGamma( const AliAODEvent * const aodEvent );\r
\r
  TString     fDeltaAODFileName;//! File where Gamma Conv AOD is located, if not in default AOD\r
  TString     fGammaCutString;   //! The cut string of the conversion analysis used to produce input AOD\r
  TString     fPionCutString;   //! The cut string of the conversion analysis used to produce input AOD\r
  TString     fAODBranchName;\r
  TClonesArray * fAODPWG4Photons;\r
  TClonesArray * fAODPWG4Pi0;\r
\r
\r
  Int_t fDebugLevel;\r
\r
\r
  AliAnalysisTaskGCPartToPWG4Part(const AliAnalysisTaskGCPartToPWG4Part&); // not implemented\r
  AliAnalysisTaskGCPartToPWG4Part& operator=(const AliAnalysisTaskGCPartToPWG4Part&); // not implemented\r
\r
  //Int_t CheckTag(AliAODPWG4ParticleCorrelation * particle, TClonesArray * tracks, TClonesArray * arrayMC, AliAODMCHeader * mcHeader);\r
  \r
  ClassDef(AliAnalysisTaskGCPartToPWG4Part, 1); // example of analysis\r
};\r
\r
#endif\r