AliEmcalJetTaggerTaskFast.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 ALIEMCALJETTAGGERTASKFAST_H
#define ALIEMCALJETTAGGERTASKFAST_H

//#define JETTAGGERFAST_TEST

class TH1;
class TH2;
class TH3;
class AliJetContainer;

#include "AliAnalysisTaskEmcalJet.h"

namespace PWGJE {
namespace EMCALJetTasks {

class AliEmcalJetTaggerTaskFast : public AliAnalysisTaskEmcalJet {
 public:
  enum JetTaggingMethod {
    kGeo      = 0,
    kFraction = 1
  };

  enum JetTaggingType {
    kTag      = 0,
    kClosest  = 1
  };
  enum AcceptanceType {
    kNoLimit = 0,             
    kLimitTagEta = 1,         
    kLimitTagEtaPhi = 2,      
    kLimitBaseTagEtaPhi = 3   
  };

  AliEmcalJetTaggerTaskFast();

  AliEmcalJetTaggerTaskFast(const char *name);

  virtual ~AliEmcalJetTaggerTaskFast() {}

  void                                UserCreateOutputObjects();

  //Setters
  void SetJetContainerBase(Int_t c)                             { fContainerBase = c;}
  void SetJetContainerTag(Int_t c)                              { fContainerTag  = c;}

  void SetJetTaggingType(JetTaggingType t)                      { fJetTaggingType = t;}
  void SetJetTaggingMethod(JetTaggingMethod m)                  { fJetTaggingMethod = m;}

  void SetMinFractionShared(Double_t f)                         { fMinFractionShared = f; }

  void SetUseSumw2(Bool_t b)                                    { fUseSumw2 = b;}
  
  void SetTypeAcceptance(AcceptanceType type)                   { fTypeAcc = type; }
  void SetMaxDistance(Double_t dist)                            { fMaxDist = dist; }
  void SetSpecialParticleContainer(Int_t contnumb)              { fSpecPartContTag = contnumb; }


  static AliEmcalJetTaggerTaskFast *AddTaskJetTaggerFast(const char * njetsBase,
      const char * njetsTag,
      const Double_t R,
      const char * nrhoBase,
      const char * nrhoTag,
      const char * ntracks,
      const char * nclusters,
      const char * type,
      const char * CentEst,
      Int_t        pSel,
      const char * trigClass);

 protected:

  Bool_t                              Run();

  Bool_t                              FillHistograms();

  void                                Init();

  Double_t GetDeltaPhi(const AliEmcalJet* jet1, const AliEmcalJet* jet2);

  Double_t GetDeltaPhi(Double_t phi1,Double_t phi2);

  Bool_t     MatchJetsGeo(AliJetContainer &contBase, AliJetContainer &contTag, Float_t maxDist = 0.3) const;

  void     ResetTagging(const AliJetContainer &cont) const;
  
 private:
  JetTaggingType                      fJetTaggingType;             
  JetTaggingMethod                    fJetTaggingMethod;           
  Int_t                               fContainerBase;              
  Int_t                               fContainerTag;               
  Int_t                               fSpecPartContTag;            
  Double_t                            fMinFractionShared;          
  Bool_t                              fUseSumw2;                   
  Bool_t                              fMatchingDone;               
  AcceptanceType                      fTypeAcc;                    
  Double_t                            fMaxDist;                    
  Bool_t                              fInit;                       
  TH3            **fh3PtJet1VsDeltaEtaDeltaPhi;  
  TH2            **fh2PtJet1VsDeltaR;            
  TH2            **fh2PtJet2VsFraction;          

  TH2            **fh2PtJet1VsLeadPtAllSel;      
  TH2            **fh2PtJet1VsLeadPtTagged;      
  TH2            **fh2PtJet1VsPtJet2;            
  TH2            **fh2PtJet2VsRelPt;             
  
  TH3             *fh3PtJetDEtaDPhiConst;        
  TH3             *fh3PtJetAreaDRConst;          
  TH1             *fNAccJets;                    
#ifdef JETTAGGERFAST_TEST
  TH1             *fIndexErrorRateBase;          
  TH1             *fIndexErrorRateTag;           
  TH1             *fContainerErrorRateBase;      
  TH1             *fContainerErrorRateTag;       
#endif
  AliEmcalJetTaggerTaskFast(const AliEmcalJetTaggerTaskFast&);            // not implemented
  AliEmcalJetTaggerTaskFast &operator=(const AliEmcalJetTaggerTaskFast&); // not implemented

  ClassDef(AliEmcalJetTaggerTaskFast, 2);
};
}
}

#endif