AliAnalysisTaskEmcalJetSubstructureTree.h

Go to the documentation of this file.

/************************************************************************************
 * Copyright (C) 2017, Copyright Holders of the ALICE Collaboration                 *
 * All rights reserved.                                                             *
 *                                                                                  *
 * Redistribution and use in source and binary forms, with or without               *
 * modification, are permitted provided that the following conditions are met:      *
 *     * Redistributions of source code must retain the above copyright             *
 *       notice, this list of conditions and the following disclaimer.              *
 *     * Redistributions in binary form must reproduce the above copyright          *
 *       notice, this list of conditions and the following disclaimer in the        *
 *       documentation and/or other materials provided with the distribution.       *
 *     * Neither the name of the <organization> nor the                             *
 *       names of its contributors may be used to endorse or promote products       *
 *       derived from this software without specific prior written permission.      *
 *                                                                                  *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND  *
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED    *
 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE           *
 * DISCLAIMED. IN NO EVENT SHALL ALICE COLLABORATION BE LIABLE FOR ANY              *
 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES       *
 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;     *
 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND      *
 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT       *
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS    *
 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.                     *
 ************************************************************************************/
#ifndef ALIANALYSISTASKEMCALJETSUBSTRUCTURETREE_H
#define ALIANALYSISTASKEMCALJETSUBSTRUCTURETREE_H

#include "AliAnalysisTaskEmcalJet.h"
#include <exception>
#include <TString.h>
#include <fastjet/PseudoJet.hh>
#include <fastjet/JetDefinition.hh>

class TTree;
class AliClusterContainer;
class AliEmcalJet;
class AliParticleContainer;
class AliTrackContainer;

namespace EmcalTriggerJets {

struct AliNSubjettinessParameters {
  Double_t fOneSubjettiness;      
  Double_t fTwoSubjettiness;      
};

struct AliNSubjettinessDefinition{
  Double_t fBeta;                 
  Double_t fRadius;               
};

struct AliSoftDropParameters {
  Double_t fZg;             
  Double_t fMg;             
  Double_t fRg;             
  Double_t fPtg;            
  Int_t fNDropped;          
};

struct AliSoftdropDefinition {
  Double_t fZ;                              
  Double_t fBeta;                           
  fastjet::JetAlgorithm fRecluserAlgo;      
};

struct AliJetSubstructureSettings {
  AliSoftdropDefinition fSoftdropSettings;
  AliNSubjettinessDefinition fSubjettinessSettings;
};

struct AliJetSubstructureData {
  AliSoftDropParameters fSoftDrop;
  AliNSubjettinessParameters fNsubjettiness;
};

class AliAnalysisTaskEmcalJetSubstructureTree : public AliAnalysisTaskEmcalJet {
public:
  class ReclusterizerException : public std::exception {
  public:
    ReclusterizerException() : std::exception() {}
    virtual ~ReclusterizerException() throw() {}

    const char *what() const throw() { return "Error in reclusterizing in fastjet"; }
  };
  enum Reclusterizer_t {
    kCAAlgo = 0,
    kKTAlgo = 1,
    kAKTAlgo = 2
  };
  enum JetTreeEntry {
    kTRadius = 0,
    kTWeight = 1,
    kTPtJetRec = 2,
    kTPtJetSim = 3,
    kTRhoPtRec = 4,
    kTRhoPtSim = 5,
    kTRhoMassRec = 6,
    kTRhoMassSim = 7,
    kTAreaRec = 8,
    kTAreaSim = 9,
    kTNEFRec = 10,
    kTNEFSim = 11,
    kTMassRec = 12,
    kTMassSim = 13,
    kTZgMeasured = 14,
    kTZgTrue = 15,
    kTRgMeasured = 16,
    kTRgTrue = 17,
    kTMgMeasured = 18,
    kTMgTrue = 19,
    kTPtgMeasured = 20,
    kTPtgTrue = 21,
    kTOneNSubjettinessMeasured = 22,
    kTOneNSubjettinessTrue = 23,
    kTTwoNSubjettinessMeasured = 24,
    kTTwoNSubjettinessTrue = 25,
    kTAngularityMeasured = 26,
    kTAngularityTrue = 27,
    kTPtDMeasured = 28,
    kTPtDTrue = 29,
    kTNCharged = 30,
    kTNNeutral = 31,
    kTNConstTrue = 32,
    kTNDroppedMeasured = 33,
    kTNDroppedTrue = 34,
    kTNVar = 35
  };
  AliAnalysisTaskEmcalJetSubstructureTree();
  AliAnalysisTaskEmcalJetSubstructureTree(const char *name);
  virtual ~AliAnalysisTaskEmcalJetSubstructureTree();

  void SetTriggerBits(UInt_t triggersel) { fTriggerSelectionBits = triggersel; }
  void SetTriggerString(TString triggerstring) { fTriggerSelectionString = triggerstring; }

  void SetNameRhoPtRec(const char *rhoname) { fRhoNamePtRec = rhoname; }
  void SetNameRhoMassRec(const char *rhoname) { fRhoNameMassRec =  rhoname; }
  void SetNameRhoPtSim(const char *rhoname) { fRhoNamePtSim = rhoname; }
  void SetNameRhoMassSim(const char *rhoname) { fRhoNameMassSim = rhoname; }

  void SetSoftdropDefiniion(Double_t zcut, Double_t betacut, Reclusterizer_t reclusterizer) {
    fSDZCut = zcut;
    fSDBetaCut = betacut;
    fReclusterizer = reclusterizer;
  }

  static AliAnalysisTaskEmcalJetSubstructureTree *AddEmcalJetSubstructureTreeMaker(Bool_t isMC, Bool_t isData, Double_t jetradius, const char *name);

protected:
  virtual void UserCreateOutputObjects();
  virtual bool Run();

  AliJetSubstructureData MakeJetSubstructure(const AliEmcalJet &jet, double jetradius, const AliParticleContainer *tracks, const AliClusterContainer *clusters, const AliJetSubstructureSettings &settings) const;

  AliSoftDropParameters MakeSoftDropParameters(const fastjet::PseudoJet &jet, const AliSoftdropDefinition &cut) const;

  AliNSubjettinessParameters MakeNsubjettinessParameters(const fastjet::PseudoJet &jet, const AliNSubjettinessDefinition &cut) const;

  Double_t MakeAngularity(const AliEmcalJet &jet, const AliParticleContainer *tracks, const AliClusterContainer *clusters) const;

  Double_t MakePtD(const AliEmcalJet &jet, const AliParticleContainer *const particles, const AliClusterContainer *const clusters) const;

  void FillTree(double r, double weight, const AliEmcalJet *datajet, const AliEmcalJet *mcjet, AliSoftDropParameters *dataSoftdrop, AliSoftDropParameters *mcsoftdrop, AliNSubjettinessParameters *dataSubjettiness, AliNSubjettinessParameters *mcSubjettiness, Double_t *angularity, Double_t *ptd, Double_t *rhoparameters);

private:
  TTree                       *fJetSubstructureTree;        
  Double_t                     fJetTreeData[kTNVar];        

  Double_t                     fSDZCut;                     
  Double_t                     fSDBetaCut;                  
  Reclusterizer_t              fReclusterizer;              

  UInt_t                       fTriggerSelectionBits;       
  TString                      fTriggerSelectionString;     

  TString                      fRhoNamePtRec;               
  TString                      fRhoNameMassRec;             
  TString                      fRhoNamePtSim;               
  TString                      fRhoNameMassSim;             

  ClassDef(AliAnalysisTaskEmcalJetSubstructureTree, 1);
};

} /* namespace EmcalTriggerJets */

#endif /* ALIANALYSISTASKEMCALJETSUBSTRUCTURETREE_H */