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,
    kJetPPCharged,
    kJetPbPbCharged,
    kJetPPSimulation,
    kJetPPChargedSimulation
  };

  class EventInfo {
  public:
    EventInfo() : fCent(-1), fEP(-1), fWeight(1), fPtHard(0) {}
    EventInfo(double cent, double ep, double w, double pt) : fCent(cent), fEP(ep), fWeight(w), fPtHard(pt) {}

    double fCent;
    double fEP;
    double fWeight;
    double fPtHard;
  };

  class AliEmcalJetEventInfoSummaryPP {
  public:
    AliEmcalJetEventInfoSummaryPP() {;}
    AliEmcalJetEventInfoSummaryPP(EventInfo) {;}

    virtual ~AliEmcalJetEventInfoSummaryPP() {;}

    void Reset() {;}
    void Set(EventInfo) {;}

    ClassDef(AliEmcalJetEventInfoSummaryPP, 1);
  };

  class AliEmcalJetEventInfoSummaryPPSimulation {
  public:
    AliEmcalJetEventInfoSummaryPPSimulation() : fWeight(1), fPtHard(0) {;}
    AliEmcalJetEventInfoSummaryPPSimulation(EventInfo event);

    virtual ~AliEmcalJetEventInfoSummaryPPSimulation() {;}

    void Reset();
    void Set(EventInfo event);

    Double32_t  fWeight          ; //[0,0,12]
    Double32_t  fPtHard          ; //[0,512,10]

    ClassDef(AliEmcalJetEventInfoSummaryPPSimulation, 1);
  };

  class AliEmcalJetEventInfoSummaryPbPb {
  public:
    AliEmcalJetEventInfoSummaryPbPb() : fCent(0), fEP(0) {;}
    AliEmcalJetEventInfoSummaryPbPb(EventInfo event);

    virtual ~AliEmcalJetEventInfoSummaryPbPb() {;}

    void Reset();
    void Set(EventInfo event);

    Double32_t  fCent          ; //[-10,118,7]
    Double32_t  fEP            ; //[0,2*pi,9]

    ClassDef(AliEmcalJetEventInfoSummaryPbPb, 2);
  };

  class AliEmcalJetInfoSummaryPP {
  public:
    AliEmcalJetInfoSummaryPP() : fPt(0), fEta(0), fPhi(0), fNEF(0), fLeadingPt(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 of leading particle
    Double32_t  fLeadingPt     ; //[0,512,10]
    Double32_t  fNConstituents ; //[0,256,8]

    ClassDef(AliEmcalJetInfoSummaryPP, 4);
  };

  class AliEmcalJetInfoSummaryPPCharged {
  public:
    AliEmcalJetInfoSummaryPPCharged() : fPt(0), fEta(0), fPhi(0), fLeadingPt(0), fNConstituents(0)  {;}
    AliEmcalJetInfoSummaryPPCharged(const AliEmcalJetInfo& source);

    virtual ~AliEmcalJetInfoSummaryPPCharged() {;}

    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]
    // Momentum of leading particle
    Double32_t  fLeadingPt     ; //[0,512,10]
    Double32_t  fNConstituents ; //[0,256,8]

    ClassDef(AliEmcalJetInfoSummaryPPCharged, 1);
  };

  class AliEmcalJetInfoSummaryPbPb {
  public:
    AliEmcalJetInfoSummaryPbPb() : fPt(0), fEta(0), fPhi(0), fNEF(0), fLeadingPt(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,9]
    Double32_t  fEta           ; //[-1.28,1.28,5]
    Double32_t  fPhi           ; //[0,2*pi,7]
    // Fraction of neutral energy
    Double32_t  fNEF           ; //[0,1.28,5]
    // Momentum of leading particle
    Double32_t  fLeadingPt     ; //[0,512,9]
    Double32_t  fArea          ; //[0,1.28,6]
    Double32_t  fNConstituents ; //[0,1024,9]
    Double32_t  fCorrPt        ; //[-256,256,9]

    ClassDef(AliEmcalJetInfoSummaryPbPb, 5);
  };

  class AliEmcalJetInfoSummaryPbPbCharged {
  public:
    AliEmcalJetInfoSummaryPbPbCharged() : fPt(0), fEta(0), fPhi(0), fLeadingPt(0), fArea(0), fNConstituents(0), fCorrPt(0) {;}
    AliEmcalJetInfoSummaryPbPbCharged(const AliEmcalJetInfo& source);

    virtual ~AliEmcalJetInfoSummaryPbPbCharged() {;}

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

    Double32_t  fPt            ; //[0,512,9]
    Double32_t  fEta           ; //[-1.28,1.28,5]
    Double32_t  fPhi           ; //[0,2*pi,7]
    // Momentum of leading particle
    Double32_t  fLeadingPt     ; //[0,512,9]
    Double32_t  fArea          ; //[0,1.28,6]
    Double32_t  fNConstituents ; //[0,1024,9]
    Double32_t  fCorrPt        ; //[-256,256,9]

    ClassDef(AliEmcalJetInfoSummaryPbPbCharged, 2);
  };

  class AliEmcalJetInfoSummaryEmbedding {
  public:
    AliEmcalJetInfoSummaryEmbedding() : fPt(0), fEta(0), fPhi(0), fNEF(0), fLeadingPt(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 of leading particle
    Double32_t  fLeadingPt     ; //[0,512,9]
    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, 4);
  };

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

  static AliAnalysisTaskEmcalJetTreeBase* CreateInstance(const char* name, EAnalisysType_t type = kJetPP);
  static AliAnalysisTaskEmcalJetTreeBase* AddTaskEmcalJetTree(TString ntracks = "usedefault", TString nclusters = "usedefault", Double_t trackPtCut = 0.15, Double_t clusECut = 0.30, EAnalisysType_t type = kJetPP, TString suffix = "");

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

  TTree   *fTree;    

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

  ClassDef(AliAnalysisTaskEmcalJetTreeBase, 1)
};

template <class T, class U>
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;  
  U                                      *fCurrentEvent ;  

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

  ClassDef(AliAnalysisTaskEmcalJetTree, 1)
};

#endif