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 AliEmcalJetInfoSummaryBase {
  public:
    AliEmcalJetInfoSummaryBase() : fPt(0), fEta(0), fPhi(0), fNEF(0), fZLeading(0) {;}
    AliEmcalJetInfoSummaryBase(const AliAnalysisTaskEmcalJetSpectraQA::AliEmcalJetInfo& source);
    virtual ~AliEmcalJetInfoSummaryBase() {;}

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

    Double32_t  fPt        ; //[0,200,12]
    Double32_t  fEta       ; //[-2,2,10]
    Double32_t  fPhi       ; //[0,2*pi,10]
    // Fraction of neutral energy
    Double32_t  fNEF       ; //[0,1,6]
    // Momentum fraction of the leading particle
    Double32_t  fZLeading  ; //[0,1,8]

    ClassDef(AliEmcalJetInfoSummaryBase, 1);
  };

  class AliEmcalJetInfoSummaryPP : public AliEmcalJetInfoSummaryBase {
  public:
    AliEmcalJetInfoSummaryPP() : AliEmcalJetInfoSummaryBase(), fNConstituents(0)  {;}
    AliEmcalJetInfoSummaryPP(const AliEmcalJetInfo& source);

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

    Char_t fNConstituents; 

    ClassDef(AliEmcalJetInfoSummaryPP, 1);
  };

  class AliEmcalJetInfoSummaryPbPb : public AliEmcalJetInfoSummaryBase {
  public:
    AliEmcalJetInfoSummaryPbPb() : AliEmcalJetInfoSummaryBase(), fCent(0), fEP(0), fArea(0), fNConstituents(0), fCorrPt(0) {;}
    AliEmcalJetInfoSummaryPbPb(const AliEmcalJetInfo& source);

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

    Char_t      fCent          ; 
    Double32_t  fEP            ; //[0,pi,10]
    Double32_t  fArea          ; //[0,10,11]
    Double32_t  fNConstituents ; //[0,1000,9]
    Double32_t  fCorrPt        ; //[0,200,12]

    ClassDef(AliEmcalJetInfoSummaryPbPb, 1);
  };

  class AliEmcalJetInfoSummaryEmbedding : public AliEmcalJetInfoSummaryPbPb {
  public:
    AliEmcalJetInfoSummaryEmbedding() : AliEmcalJetInfoSummaryPbPb(), fMCPt(0) {;}
    AliEmcalJetInfoSummaryEmbedding(const AliEmcalJetInfo& source);

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

    Double32_t  fMCPt      ; //[0,200,12]

    ClassDef(AliEmcalJetInfoSummaryEmbedding, 1);
  };

  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