AliAnalysisTaskEmcalJetTree.h

Go to the documentation of this file.

#ifndef ALIANALYSISTASKEMCALJETTREE_H
#define ALIANALYSISTASKEMCALJETTREE_H

/* Copyright(c) 1998-2016, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

#include <map>
#include <vector>
#include <string>
#include "AliAnalysisTaskEmcalJetSpectraQA.h"

class AliAnalysisTaskEmcalJetTreeBase : public AliAnalysisTaskEmcalJetSpectraQA {
 public:
  enum EAnalisysType_t {
    kJetPP,
    kJetPbPb,
    kJetEmbedding
  };

  class AliEmcalJetInfoSummaryPP {
  public:
    AliEmcalJetInfoSummaryPP() : fPt(0), fEta(0), fPhi(0), fNEF(0), fZLeading(0), fNConstituents(0)  {;}
    AliEmcalJetInfoSummaryPP(const AliEmcalJetInfo& source);

    virtual ~AliEmcalJetInfoSummaryPP() {;}

    void Reset();
    void Set(const AliEmcalJetInfo& source);

    Double32_t  fPt            ; //[0,512,10]
    Double32_t  fEta           ; //[-1.28,1.28,8]
    Double32_t  fPhi           ; //[0,2*pi,9]
    // Fraction of neutral energy
    Double32_t  fNEF           ; //[0,1.28,7]
    // Momentum fraction of the leading particle
    Double32_t  fZLeading      ; //[0,1.28,7]
    Double32_t  fNConstituents ; //[0,256,8]

    ClassDef(AliEmcalJetInfoSummaryPP, 3);
  };

  class AliEmcalJetInfoSummaryPbPb {
  public:
    AliEmcalJetInfoSummaryPbPb() : fPt(0), fEta(0), fPhi(0), fNEF(0), fZLeading(0), fCent(0), fEP(0), fArea(0), fNConstituents(0), fCorrPt(0) {;}
    AliEmcalJetInfoSummaryPbPb(const AliEmcalJetInfo& source);

    virtual ~AliEmcalJetInfoSummaryPbPb() {;}

    void Reset();
    void Set(const AliEmcalJetInfo& source);

    Double32_t  fPt            ; //[0,512,10]
    Double32_t  fEta           ; //[-1.28,1.28,8]
    Double32_t  fPhi           ; //[0,2*pi,9]
    // Fraction of neutral energy
    Double32_t  fNEF           ; //[0,1.28,7]
    // Momentum fraction of the leading particle
    Double32_t  fZLeading      ; //[0,1.28,7]
    Double32_t  fCent          ; //[0,128,8]
    Double32_t  fEP            ; //[0,pi,8]
    Double32_t  fArea          ; //[0,5.12,9]
    Double32_t  fNConstituents ; //[0,1024,9]
    Double32_t  fCorrPt        ; //[-512,512,11]

    ClassDef(AliEmcalJetInfoSummaryPbPb, 3);
  };

  class AliEmcalJetInfoSummaryEmbedding {
  public:
    AliEmcalJetInfoSummaryEmbedding() : fPt(0), fEta(0), fPhi(0), fNEF(0), fZLeading(0), fCent(0), fEP(0), fArea(0), fNConstituents(0), fCorrPt(0), fMCPt(0) {;}
    AliEmcalJetInfoSummaryEmbedding(const AliEmcalJetInfo& source);

    virtual ~AliEmcalJetInfoSummaryEmbedding() {;}

    void Reset();
    void Set(const AliEmcalJetInfo& source);

    Double32_t  fPt            ; //[0,512,10]
    Double32_t  fEta           ; //[-1.28,1.28,8]
    Double32_t  fPhi           ; //[0,2*pi,9]
    // Fraction of neutral energy
    Double32_t  fNEF           ; //[0,1.28,7]
    // Momentum fraction of the leading particle
    Double32_t  fZLeading      ; //[0,1.28,7]
    Double32_t  fCent          ; //[0,128,8]
    Double32_t  fEP            ; //[0,pi,8]
    Double32_t  fArea          ; //[0,5.12,9]
    Double32_t  fNConstituents ; //[0,1024,9]
    Double32_t  fCorrPt        ; //[-512,512,11]
    Double32_t  fMCPt          ; //[0,512,10]

    ClassDef(AliEmcalJetInfoSummaryEmbedding, 3);
  };

  AliAnalysisTaskEmcalJetTreeBase() : AliAnalysisTaskEmcalJetSpectraQA(), fTree(0) {;}
  AliAnalysisTaskEmcalJetTreeBase(const char *name);
  virtual ~AliAnalysisTaskEmcalJetTreeBase() {;}

  static AliAnalysisTaskEmcalJetTreeBase* CreateInstance(const char* name, EAnalisysType_t type = kJetPP);

 protected:
  void AllocateTTree(const AliJetContainer* jets) = 0;
  void FillTTree(const AliEmcalJetInfo& jetInfo, const AliJetContainer* jets) = 0;

  TTree   *fTree;    

  ClassDef(AliAnalysisTaskEmcalJetTreeBase, 1)
};

template <class T>
class AliAnalysisTaskEmcalJetTree : public AliAnalysisTaskEmcalJetTreeBase {
 public:

  AliAnalysisTaskEmcalJetTree();
  AliAnalysisTaskEmcalJetTree(const char *name);

  virtual ~AliAnalysisTaskEmcalJetTree() {;}

  void                        UserCreateOutputObjects();
  Bool_t                      FillHistograms();

 protected:
  void                        AllocateTTree(const AliJetContainer* jets);
  void                        FillTTree(const AliEmcalJetInfo& jetInfo, const AliJetContainer* jets);

  std::map<std::string, std::vector<T> > *fCurrentOutput;  

 private:
  AliAnalysisTaskEmcalJetTree(const AliAnalysisTaskEmcalJetTree&);            // not implemented
  AliAnalysisTaskEmcalJetTree &operator=(const AliAnalysisTaskEmcalJetTree&); // not implemented

  ClassDef(AliAnalysisTaskEmcalJetTree, 1)
};

#endif