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 THistManager;
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;            
  Double_t fMug;            
  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() {}

    virtual const char *what() const throw() { return "Error in reclusterizing in fastjet"; }
  };
  class SubstructureException : public std::exception {
  public:
    SubstructureException() : std::exception() {}
    virtual ~SubstructureException() throw() {}

    virtual const char *what() const throw() { return "Error in builing substructure observable"; }
  };
  class SoftDropException : public std::exception {
  public:
    SoftDropException() : std::exception() {}
    virtual ~SoftDropException() throw() {}

    virtual const char *what() const throw() { return "No associated softdrop structure found for jet - softdrop algorithm failing"; }
  };
  enum Reclusterizer_t {
    kCAAlgo = 0,
    kKTAlgo = 1,
    kAKTAlgo = 2
  };
  enum JetTreeEntry {
    kTRadius = 0,
    kTWeight = 1,
    kTPtJetRec = 2,
    kTPtJetSim = 3,
    kTEJetRec = 4,
    kTEJetSim = 5,
    kTEtaRec = 6,
    kTEtaSim = 7,
    kTPhiRec = 8,
    kTPhiSim = 9,
    kTRhoPtRec = 10,
    kTRhoPtSim = 11,
    kTRhoMassRec = 12,
    kTRhoMassSim = 13,
    kTAreaRec = 14,
    kTAreaSim = 15,
    kTNEFRec = 16,
    kTNEFSim = 17,
    kTMassRec = 18,
    kTMassSim = 19,
    kTZgMeasured = 20,
    kTZgTrue = 21,
    kTRgMeasured = 22,
    kTRgTrue = 23,
    kTMgMeasured = 24,
    kTMgTrue = 25,
    kTPtgMeasured = 26,
    kTPtgTrue = 27,
    kTMugMeasured = 28,
    kTMugTrue = 29,
    kTOneNSubjettinessMeasured = 30,
    kTOneNSubjettinessTrue = 31,
    kTTwoNSubjettinessMeasured = 32,
    kTTwoNSubjettinessTrue = 33,
    kTAngularityMeasured = 34,
    kTAngularityTrue = 35,
    kTPtDMeasured = 36,
    kTPtDTrue = 37,
    kTNCharged = 38,
    kTNNeutral = 39,
    kTNConstTrue = 40,
    kTNDroppedMeasured = 41,
    kTNDroppedTrue = 42,
    kTNVar = 43
  };

  AliAnalysisTaskEmcalJetSubstructureTree();
  AliAnalysisTaskEmcalJetSubstructureTree(const char *name);
  virtual ~AliAnalysisTaskEmcalJetSubstructureTree();

  void SetTriggerBits(UInt_t triggersel) { fTriggerSelectionBits = triggersel; }
  void SetTriggerString(TString triggerstring) { fTriggerSelectionString = triggerstring; }
  void SetUseDownscaleWeight(Bool_t usedownscale) { fUseDownscaleWeight = usedownscale; }

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

  void SetFillPartLevelBranches(Bool_t doFill) { fFillPart = doFill; }
  void SetFillAcceptance(Bool_t doFill) { fFillAcceptance = doFill; }
  void SetFillRhoBranches(Bool_t doFill) { fFillRho = doFill; }
  void SetFillMassBranches(Bool_t doFill) { fFillMass = doFill; }
  void SetFillSoftdropBranches(Bool_t doFill) { fFillSoftDrop = doFill; }
  void SetFillNSubjettinessBranches(Bool_t doFill) { fFillNSub = doFill; }
  void SetFillSubstructureBranches(Bool_t doFill) { fFillStructGlob = doFill; }

  static AliAnalysisTaskEmcalJetSubstructureTree *AddEmcalJetSubstructureTreeMaker(Bool_t isMC, Bool_t isData, Double_t jetradius, AliJetContainer::EJetType_t jettype, AliJetContainer::ERecoScheme_t recombinationScheme, const char *name);

protected:
  virtual void UserCreateOutputObjects();
  virtual bool Run();
  virtual void RunChanged(Int_t newrun);

  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);

  void DoConstituentQA(const AliEmcalJet *jet, const AliParticleContainer *tracks, const AliClusterContainer *clusters);

  void LinkOutputBranch(const TString &branchname, Double_t *datalocation);

  bool IsPartBranch(const TString &branchname) const;
  bool IsAcceptanceBranch(const TString &branchname) const;
  bool IsRhoBranch(const TString &branchname) const;
  bool IsMassBranch(const TString &branchname) const;
  bool IsSoftdropBranch(const TString &branchname) const;
  bool IsNSubjettinessBranch(const TString &branchname) const;
  bool IsStructbranch(const TString &branchname) const;

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

  Double_t                     fSDZCut;                     
  Double_t                     fSDBetaCut;                  
  Reclusterizer_t              fReclusterizer;              

  UInt_t                       fTriggerSelectionBits;       
  TString                      fTriggerSelectionString;     
  Bool_t                       fUseDownscaleWeight;         

  // Fill levels for tree (save disk space when information is not needed)
  Bool_t                       fFillPart;                   
  Bool_t                       fFillAcceptance;             
  Bool_t                       fFillRho;                    
  Bool_t                       fFillMass;                   
  Bool_t                       fFillSoftDrop;               
  Bool_t                       fFillNSub;                   
  Bool_t                       fFillStructGlob;             

  ClassDef(AliAnalysisTaskEmcalJetSubstructureTree, 1);
};

} /* namespace EmcalTriggerJets */

#endif /* ALIANALYSISTASKEMCALJETSUBSTRUCTURETREE_H */