[u/mrichter/AliRoot.git] / PWGHF / vertexingHF / AliAnalysisTaskSELc2V0bachelor.h

#ifndef ALIANALYSISTASKSELC2V0BACHELOR_H
#define ALIANALYSISTASKSELC2V0BACHELOR_H
/**************************************************************************
 * Copyright(c) 1998-2009, 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.                  *
 **************************************************************************/

/* $Id$ */ 

#include <TH2F.h>
#include "TROOT.h"
#include "TSystem.h"

#include "AliAnalysisTaskSE.h"
#include "AliAODEvent.h"
#include "AliPID.h"
#include "AliAODTrack.h"
#include "AliPIDResponse.h"
#include "AliTPCPIDResponse.h"
#include "AliRDHFCutsLctoV0.h"
#include "AliNormalizationCounter.h"

class AliAnalysisTaskSELc2V0bachelor : public AliAnalysisTaskSE 
{
  
 public:
  
  AliAnalysisTaskSELc2V0bachelor();
  AliAnalysisTaskSELc2V0bachelor(const Char_t* name, AliRDHFCutsLctoV0* cutsA, AliRDHFCutsLctoV0* cutsB,
				 Bool_t useOnTheFly=kFALSE);
  virtual ~AliAnalysisTaskSELc2V0bachelor();

  // Implementation of interface methods  
  virtual void UserCreateOutputObjects();
  virtual void Init();
  virtual void LocalInit() {Init();}
  virtual void UserExec(Option_t *option);
  virtual void Terminate(Option_t *option);
 
  // histos
  void FillLc2pK0Sspectrum(AliAODRecoCascadeHF *part, Int_t isLc,
			   Int_t &nSelectedProd, AliRDHFCutsLctoV0 *cutsProd,
			   Int_t &nSelectedAnal, AliRDHFCutsLctoV0 *cutsAnal);

  void DefineHistograms();
  Int_t CheckOrigin(TClonesArray* arrayMC, AliAODMCParticle *mcPartCandidate) const;

  void MakeAnalysisForLc2prK0S(AliAODVertex * vtx1,
			       TClonesArray *arrayLctopK0s,
			       TClonesArray *mcArray,
			       Int_t &nSelectedProd, AliRDHFCutsLctoV0 *cutsProd,
			       Int_t &nSelectedAnal, AliRDHFCutsLctoV0 *cutsAnal);
 
  Int_t MatchToMC(AliAODRecoCascadeHF *lc2bacV0,
		  Int_t *pdgDgLc2bacV0, Int_t *pdgDgV0,
		  TClonesArray *mcArray);

  void SearchLcDaughter(TClonesArray *arrayMC);

  void DefineK0SHistos();

  // set MC usage
  void SetMC(Bool_t theMCon) {fUseMCInfo = theMCon;}
  Bool_t GetMC() const {return fUseMCInfo;}

  void FillArmPodDistribution(AliAODv0 *vZero, TString histoTitle, TList *histoList);

  void SetK0sAnalysis(Bool_t a) {fIsK0sAnalysis=a;}
  Bool_t GetK0sAnalysis() const {return fIsK0sAnalysis;}

  void SetUseOnTheFlyV0(Bool_t a) { fUseOnTheFlyV0=a; }
  Bool_t GetUseOnTheFlyV0() { return fUseOnTheFlyV0; }

 private:
  
  AliAnalysisTaskSELc2V0bachelor(const AliAnalysisTaskSELc2V0bachelor &source);
  AliAnalysisTaskSELc2V0bachelor& operator=(const AliAnalysisTaskSELc2V0bachelor& source); 
  
  Bool_t fUseMCInfo;          // Use MC info
  TList *fOutput;             // User output1 // general histos
  TList *fOutputAll;          // User output2 // histos without pid and cut on V0
  TList *fOutputPIDBach;      // User output3 // histos with PId on Bachelor

  // define the histograms
  TH1I *fCEvents;                    // Histogram to check selected events
  AliPIDResponse *fPIDResponse;      //! PID response object
  Bool_t fIsK0sAnalysis;             // switch between Lpi and K0sp
  AliNormalizationCounter *fCounter; // AliNormalizationCounter on output slot 4
  AliRDHFCutsLctoV0 *fProdCuts;      // Cuts - sent to output slot 5
  AliRDHFCutsLctoV0 *fAnalCuts;      // Cuts - sent to output slot 5
  TList *fListCuts;                  // list of cuts
  Bool_t fUseOnTheFlyV0;             // flag to analyze also on-the-fly V0 candidates
  Bool_t fIsEventSelected;           // flag for event selected

  ClassDef(AliAnalysisTaskSELc2V0bachelor,2); // class for Lc->p K0
};

#endif
Commit	Line	Data
992ba7d8	1	#ifndef ALIANALYSISTASKSELC2V0BACHELOR_H
	2	#define ALIANALYSISTASKSELC2V0BACHELOR_H
	3	/**************************************************************************
	4	* Copyright(c) 1998-2009, ALICE Experiment at CERN, All rights reserved. *
	5	* *
	6	* Author: The ALICE Off-line Project. *
	7	* Contributors are mentioned in the code where appropriate. *
	8	* *
	9	* Permission to use, copy, modify and distribute this software and its *
	10	* documentation strictly for non-commercial purposes is hereby granted *
	11	* without fee, provided that the above copyright notice appears in all *
	12	* copies and that both the copyright notice and this permission notice *
	13	* appear in the supporting documentation. The authors make no claims *
	14	* about the suitability of this software for any purpose. It is *
	15	* provided "as is" without express or implied warranty. *
	16	**************************************************************************/
	17
	18	/* $Id$ */
	19
	20	#include <TH2F.h>
	21	#include "TROOT.h"
	22	#include "TSystem.h"
	23
	24	#include "AliAnalysisTaskSE.h"
	25	#include "AliAODEvent.h"
	26	#include "AliPID.h"
	27	#include "AliAODTrack.h"
	28	#include "AliPIDResponse.h"
	29	#include "AliTPCPIDResponse.h"
	30	#include "AliRDHFCutsLctoV0.h"
	31	#include "AliNormalizationCounter.h"
	32
	33	class AliAnalysisTaskSELc2V0bachelor : public AliAnalysisTaskSE
	34	{
	35
	36	public:
	37
	38	AliAnalysisTaskSELc2V0bachelor();
0dbd1d4f	39	AliAnalysisTaskSELc2V0bachelor(const Char_t* name, AliRDHFCutsLctoV0* cutsA, AliRDHFCutsLctoV0* cutsB,
0dbd1d4f	40	Bool_t useOnTheFly=kFALSE);
992ba7d8	41	virtual ~AliAnalysisTaskSELc2V0bachelor();
	42
	43	// Implementation of interface methods
	44	virtual void UserCreateOutputObjects();
	45	virtual void Init();
	46	virtual void LocalInit() {Init();}
	47	virtual void UserExec(Option_t *option);
	48	virtual void Terminate(Option_t *option);
	49
	50	// histos
	51	void FillLc2pK0Sspectrum(AliAODRecoCascadeHF *part, Int_t isLc,
	52	Int_t &nSelectedProd, AliRDHFCutsLctoV0 *cutsProd,
	53	Int_t &nSelectedAnal, AliRDHFCutsLctoV0 *cutsAnal);
	54
	55	void DefineHistograms();
	56	Int_t CheckOrigin(TClonesArray* arrayMC, AliAODMCParticle *mcPartCandidate) const;
	57
	58	void MakeAnalysisForLc2prK0S(AliAODVertex * vtx1,
	59	TClonesArray *arrayLctopK0s,
	60	TClonesArray *mcArray,
	61	Int_t &nSelectedProd, AliRDHFCutsLctoV0 *cutsProd,
	62	Int_t &nSelectedAnal, AliRDHFCutsLctoV0 *cutsAnal);
	63
	64	Int_t MatchToMC(AliAODRecoCascadeHF *lc2bacV0,
	65	Int_t pdgDgLc2bacV0, Int_t pdgDgV0,
	66	TClonesArray *mcArray);
	67
	68	void SearchLcDaughter(TClonesArray *arrayMC);
	69
	70	void DefineK0SHistos();
	71
	72	// set MC usage
	73	void SetMC(Bool_t theMCon) {fUseMCInfo = theMCon;}
	74	Bool_t GetMC() const {return fUseMCInfo;}
	75
	76	void FillArmPodDistribution(AliAODv0 vZero, TString histoTitle, TList histoList);
	77
	78	void SetK0sAnalysis(Bool_t a) {fIsK0sAnalysis=a;}
	79	Bool_t GetK0sAnalysis() const {return fIsK0sAnalysis;}
	80
0dbd1d4f	81	void SetUseOnTheFlyV0(Bool_t a) { fUseOnTheFlyV0=a; }
0dbd1d4f	82	Bool_t GetUseOnTheFlyV0() { return fUseOnTheFlyV0; }
992ba7d8	83
	84	private:
	85
	86	AliAnalysisTaskSELc2V0bachelor(const AliAnalysisTaskSELc2V0bachelor &source);
	87	AliAnalysisTaskSELc2V0bachelor& operator=(const AliAnalysisTaskSELc2V0bachelor& source);
	88
	89	Bool_t fUseMCInfo; // Use MC info
	90	TList *fOutput; // User output1 // general histos
	91	TList *fOutputAll; // User output2 // histos without pid and cut on V0
	92	TList *fOutputPIDBach; // User output3 // histos with PId on Bachelor
	93
	94	// define the histograms
0dbd1d4f	95	TH1I *fCEvents; // Histogram to check selected events
	96	AliPIDResponse *fPIDResponse; //! PID response object
	97	Bool_t fIsK0sAnalysis; // switch between Lpi and K0sp
992ba7d8	98	AliNormalizationCounter *fCounter; // AliNormalizationCounter on output slot 4
	99	AliRDHFCutsLctoV0 *fProdCuts; // Cuts - sent to output slot 5
	100	AliRDHFCutsLctoV0 *fAnalCuts; // Cuts - sent to output slot 5
	101	TList *fListCuts; // list of cuts
0dbd1d4f	102	Bool_t fUseOnTheFlyV0; // flag to analyze also on-the-fly V0 candidates
0dbd1d4f	103	Bool_t fIsEventSelected; // flag for event selected
992ba7d8	104
0dbd1d4f	105	ClassDef(AliAnalysisTaskSELc2V0bachelor,2); // class for Lc->p K0
992ba7d8	106	};
	107
	108	#endif
	109