[u/mrichter/AliRoot.git] / PWG3 / vertexingHF / AliCFHeavyFlavourTask.h

/**************************************************************************
 * Copyright(c) 1998-1999, 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.                  *
 **************************************************************************/

//-----------------------------------------------------------------------
// Author : C. Zampolli, on an example from R. Vernet
//-----------------------------------------------------------------------

#ifndef ALICFHEAVYFLAVOURTASK_H
#define ALICFHEAVYFLAVOURTASK_H

#include "AliAnalysisTaskSE.h"

class TH1I;
class AliCFManager;
class AliAODRecoDecayHF2Prong;

class AliCFHeavyFlavourTask : public AliAnalysisTaskSE {
  public:

  enum {
    kStepGenerated       = 0,
    kStepReconstructed   = 1
  };

  AliCFHeavyFlavourTask();
  AliCFHeavyFlavourTask(const Char_t* name);
  AliCFHeavyFlavourTask& operator= (const AliCFHeavyFlavourTask& c);
  AliCFHeavyFlavourTask(const AliCFHeavyFlavourTask& c);
  virtual ~AliCFHeavyFlavourTask();

  // ANALYSIS FRAMEWORK STUFF to loop on data and fill output objects
  void UserCreateOutputObjects();
  void UserExec(Option_t *option);
  void Terminate(Option_t *);
  
  // CORRECTION FRAMEWORK RELATED FUNCTIONS
  void SetCFManager(AliCFManager* const io) {fCFManager = io;}   // global correction manager
  AliCFManager* GetCFManager() const {return fCFManager;} // get corr manager

  void SetPDG(Int_t code) {fPDG = code; } // defines the PDG code of searched HF
  
 protected:
  Int_t fPDG;         //  PDG code of searched V0's
  AliCFManager* fCFManager  ; //  pointer to the CF manager
  TH1I* fHistEventsProcessed;   //! simple histo for monitoring the number of events processed
  Int_t fCountMC;               // MC particle found
  Int_t fEvents;                // n. of events
  
  ClassDef(AliCFHeavyFlavourTask,0);
};

#endif
Commit	Line	Data
8eb4dc33	1	/**************************************************************************
	2	* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	3	* *
	4	* Author: The ALICE Off-line Project. *
	5	* Contributors are mentioned in the code where appropriate. *
	6	* *
	7	* Permission to use, copy, modify and distribute this software and its *
	8	* documentation strictly for non-commercial purposes is hereby granted *
	9	* without fee, provided that the above copyright notice appears in all *
	10	* copies and that both the copyright notice and this permission notice *
	11	* appear in the supporting documentation. The authors make no claims *
	12	* about the suitability of this software for any purpose. It is *
	13	* provided "as is" without express or implied warranty. *
	14	**************************************************************************/
	15
	16	//-----------------------------------------------------------------------
	17	// Author : C. Zampolli, on an example from R. Vernet
	18	//-----------------------------------------------------------------------
	19
	20	#ifndef ALICFHEAVYFLAVOURTASK_H
	21	#define ALICFHEAVYFLAVOURTASK_H
	22
	23	#include "AliAnalysisTaskSE.h"
	24
	25	class TH1I;
	26	class AliCFManager;
	27	class AliAODRecoDecayHF2Prong;
	28
	29	class AliCFHeavyFlavourTask : public AliAnalysisTaskSE {
	30	public:
	31
	32	enum {
	33	kStepGenerated = 0,
079cb93b	34	kStepReconstructed = 1
8eb4dc33	35	};
	36
	37	AliCFHeavyFlavourTask();
	38	AliCFHeavyFlavourTask(const Char_t* name);
	39	AliCFHeavyFlavourTask& operator= (const AliCFHeavyFlavourTask& c);
	40	AliCFHeavyFlavourTask(const AliCFHeavyFlavourTask& c);
	41	virtual ~AliCFHeavyFlavourTask();
	42
	43	// ANALYSIS FRAMEWORK STUFF to loop on data and fill output objects
	44	void UserCreateOutputObjects();
	45	void UserExec(Option_t *option);
	46	void Terminate(Option_t *);
	47
	48	// CORRECTION FRAMEWORK RELATED FUNCTIONS
	49	void SetCFManager(AliCFManager* const io) {fCFManager = io;} // global correction manager
	50	AliCFManager* GetCFManager() const {return fCFManager;} // get corr manager
	51
	52	void SetPDG(Int_t code) {fPDG = code; } // defines the PDG code of searched HF
8eb4dc33	53
	54	protected:
	55	Int_t fPDG; // PDG code of searched V0's
	56	AliCFManager* fCFManager ; // pointer to the CF manager
	57	TH1I* fHistEventsProcessed; //! simple histo for monitoring the number of events processed
	58	Int_t fCountMC; // MC particle found
	59	Int_t fEvents; // n. of events
	60
	61	ClassDef(AliCFHeavyFlavourTask,0);
	62	};
	63
	64	#endif