AliAnalysisTaskDmesonJetsDetectorResponse.h

Go to the documentation of this file.

#ifndef ALIANALYSISTASKDMESONJETSDETECTORRESPONSE_H
#define ALIANALYSISTASKDMESONJETSDETECTORRESPONSE_H

/**************************************************************************
* Copyright(c) 1998-2016, ALICE Experiment at CERN, All rights reserved. *
*                                                                        *
* Author: The ALICE Off-line Project.                                    *
* Contributors are mentioned in the code where appropriate.              *
*                                                                        *
* Permission to use, copy, modify and distribute this software and its   *
* documentation strictly for non-commercial purposes is hereby granted   *
* without fee, provided that the above copyright notice appears in all   *
* copies and that both the copyright notice and this permission notice   *
* appear in the supporting documentation. The authors make no claims     *
* about the suitability of this software for any purpose. It is          *
* provided "as is" without express or implied warranty.                  *
**************************************************************************/

#include "AliAnalysisTaskDmesonJets.h"

class AliAnalysisTaskDmesonJetsDetectorResponse : public AliAnalysisTaskDmesonJets
{
 public:

  class AliDmesonMatchInfoSummary {
  public:
    AliDmesonMatchInfoSummary() {;}
    virtual ~AliDmesonMatchInfoSummary() {}

    virtual void Reset() = 0;
    virtual void SetReconstructed(const AliDmesonJetInfo& reco) = 0;
    virtual void SetGenerated(const AliDmesonJetInfo& truth) = 0;
    virtual AliDmesonInfoSummary* GetReconstructed() = 0;
    virtual AliDmesonInfoSummary* GetGenerated() = 0;

    ClassDef(AliDmesonMatchInfoSummary, 1);
  };

  class AliD0MatchInfoSummary : public AliDmesonMatchInfoSummary {
  public:
    AliD0MatchInfoSummary() : AliDmesonMatchInfoSummary(), fReconstructed(), fGenerated() {}

    virtual void Reset();
    virtual void SetReconstructed(const AliDmesonJetInfo& reco);
    virtual void SetGenerated(const AliDmesonJetInfo& truth);

    virtual AliDmesonInfoSummary* GetReconstructed() { return &fReconstructed; }
    virtual AliDmesonInfoSummary* GetGenerated()     { return &fGenerated    ; }

    AliD0InfoSummary       fReconstructed ; 
    AliDmesonMCInfoSummary fGenerated     ; 

    ClassDef(AliD0MatchInfoSummary, 2);
  };

  class AliDStarMatchInfoSummary : public AliDmesonMatchInfoSummary {
  public:
    AliDStarMatchInfoSummary() : AliDmesonMatchInfoSummary(), fReconstructed(), fGenerated() {}

    virtual void Reset();
    virtual void SetReconstructed(const AliDmesonJetInfo& reco);
    virtual void SetGenerated(const AliDmesonJetInfo& truth);

    virtual AliDmesonInfoSummary* GetReconstructed() { return &fReconstructed; }
    virtual AliDmesonInfoSummary* GetGenerated()     { return &fGenerated    ; }

    AliDStarInfoSummary     fReconstructed ; 
    AliDmesonInfoSummary    fGenerated     ; 

    ClassDef(AliDStarMatchInfoSummary, 1);
  };

  class ResponseEngine : public TObject {
  public:
    ResponseEngine();
    ResponseEngine(ECandidateType_t type);
    ResponseEngine(const ResponseEngine &source);
    ResponseEngine& operator=(const ResponseEngine& source);
    virtual ~ResponseEngine() {;}

    void SetMaxJetDmesonDistance(Double_t d) { fMaxJetDmesonDistance = d; }

    void SetReconstructedAnalysisEngine(AnalysisEngine* reco) { fRecontructed = reco; }
    void SetGeneratedAnalysisEngine(AnalysisEngine* truth) { fGenerated = truth; }

    Bool_t CheckInit();
    Bool_t IsInhibit() const { return fInhibit; }

    const char* GetName() const { return fName.Data(); }

    void RunAnalysis();

    TTree* BuildTree(const char* taskName);
    TTree* GetTree() const { return fTree; }
    Bool_t FillTree(Bool_t applyKinCuts, Bool_t findNoDMesonRecoJets);

    void AssignDataSlot(Int_t n) { fDataSlotNumber = n; }
    Int_t GetDataSlotNumber() const { return fDataSlotNumber; }

    friend bool        operator==(const ResponseEngine& lhs, const ResponseEngine& rhs) { return (lhs.fCandidateType == rhs.fCandidateType);}
    friend inline bool operator!=(const ResponseEngine& lhs, const ResponseEngine& rhs) { return !(lhs == rhs); }

  protected:

    ECandidateType_t              fCandidateType       ; 
    Bool_t                        fInhibit             ; 
    Double_t                      fMaxJetDmesonDistance; 

    TString                       fName                ; 
    TTree                        *fTree                ; 
    AliDmesonMatchInfoSummary    *fCurrentDmeson       ; 
    AliJetInfoSummary           **fCurrentJetInfoReco  ; 
    AliJetInfoSummary           **fCurrentJetInfoTruth ; 
    Int_t                         fDataSlotNumber      ; 

    AnalysisEngine               *fRecontructed        ; 
    AnalysisEngine               *fGenerated           ; 

    friend class AliAnalysisTaskDmesonJetsDetectorResponse;

  private:

    ClassDef(ResponseEngine, 2);
  };

  AliAnalysisTaskDmesonJetsDetectorResponse();
  AliAnalysisTaskDmesonJetsDetectorResponse(const char* name, Int_t nOutputTrees=2);
  virtual ~AliAnalysisTaskDmesonJetsDetectorResponse() {;}

  void SetFindRecoJetsForLostDMesons(Bool_t b) { fFindRecoJetsForLostDMesons = b; }

  virtual void         UserCreateOutputObjects();
  virtual void         ExecOnce();
  virtual Bool_t       Run();
  virtual Bool_t       FillHistograms();

  static AliAnalysisTaskDmesonJetsDetectorResponse* AddTaskDmesonJetsDetectorResponse(TString trackName, TString clusName, TString mcPartName, Int_t nMaxTrees, TString suffix);

 protected:

  virtual void SetOutputTypeInternal(EOutputType_t b);

  Int_t PostDataFromResponseEngine(const ResponseEngine& eng);

  Bool_t                                      fFindRecoJetsForLostDMesons; 
  std::map<ECandidateType_t, ResponseEngine>  fResponseEngines           ; 

 private:

  AliAnalysisTaskDmesonJetsDetectorResponse(const AliAnalysisTaskDmesonJetsDetectorResponse &source);
  AliAnalysisTaskDmesonJetsDetectorResponse& operator=(const AliAnalysisTaskDmesonJetsDetectorResponse& source);

  ClassDef(AliAnalysisTaskDmesonJetsDetectorResponse, 2);
};

#endif