[u/mrichter/AliRoot.git] / CORRFW / test / AliCFRsnTask.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 : R. Vernet, Consorzio Cometa - Catania (it)
//-----------------------------------------------------------------------

#ifndef ALICFRSNTASK_H
#define ALICFRSNTASK_H

#include "AliAnalysisTask.h"

class TH1I;
class TParticle ;
class TFile ;
class AliMCEventHandler;
class AliESDEvent;
class AliStack ;
class AliCFManager;
class TChain;
class AliRsnDaughter;

class AliCFRsnTask : public AliAnalysisTask {
  public:

  enum {
    kStepGenerated       = 0,
    kStepReconstructible = 1,
    kStepReconstructed   = 2,
    kStepSelected        = 3
  };

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

  // ANALYSIS FRAMEWORK STUFF to loop on data and fill output objects
  void     ConnectInputData(Option_t *option="");
  void     CreateOutputObjects();
  void     Exec(Option_t *option);
  void     Init(); //loads the CF manager
  void     LocalInit() {Init();} //needed for the slaves 
  void     Terminate(Option_t *);
  
  // CORRECTION FRAMEWORK RELATED FUNCTIONS
  void           SetCFManager(AliCFManager* io) {fCFManager = io;}   // global correction manager
  AliCFManager * GetCFManager()                 {return fCFManager;} // get corr manager

  void     SetRsnPDG(Int_t code)             {fRsnPDG = code; }            // defines the PDG code of searched resonance
  static   Double_t GetRapidity(Double_t, Double_t) ;  // returns the rapidity of the Resonance (assuming PDG code)


 protected:

  Int_t           fRsnPDG;       //  PDG code of searched resonance
  TChain         *fChain      ;  // chained files
  AliESDEvent    *fESD        ;  // pointer to the ESD event read
  AliCFManager   *fCFManager  ;  // pointer to the CF manager

  // Histograms
  //Number of events
  TH1I *fHistEventsProcessed; //! simple histo for monitoring the number of events processed
  
  ClassDef(AliCFRsnTask,1);
};

#endif
Commit	Line	Data
2fbc0b17	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 : R. Vernet, Consorzio Cometa - Catania (it)
	18	//-----------------------------------------------------------------------
	19
	20	#ifndef ALICFRSNTASK_H
	21	#define ALICFRSNTASK_H
	22
	23	#include "AliAnalysisTask.h"
	24
	25	class TH1I;
	26	class TParticle ;
	27	class TFile ;
	28	class AliMCEventHandler;
	29	class AliESDEvent;
	30	class AliStack ;
	31	class AliCFManager;
	32	class TChain;
	33	class AliRsnDaughter;
	34
	35	class AliCFRsnTask : public AliAnalysisTask {
	36	public:
	37
	38	enum {
	39	kStepGenerated = 0,
	40	kStepReconstructible = 1,
	41	kStepReconstructed = 2,
	42	kStepSelected = 3
	43	};
	44
	45	AliCFRsnTask();
	46	AliCFRsnTask(const Char_t* name);
	47	AliCFRsnTask& operator= (const AliCFRsnTask& c);
	48	AliCFRsnTask(const AliCFRsnTask& c);
	49	virtual ~AliCFRsnTask();
	50
	51	// ANALYSIS FRAMEWORK STUFF to loop on data and fill output objects
	52	void ConnectInputData(Option_t *option="");
	53	void CreateOutputObjects();
	54	void Exec(Option_t *option);
	55	void Init(); //loads the CF manager
	56	void LocalInit() {Init();} //needed for the slaves
	57	void Terminate(Option_t *);
	58
	59	// CORRECTION FRAMEWORK RELATED FUNCTIONS
	60	void SetCFManager(AliCFManager* io) {fCFManager = io;} // global correction manager
	61	AliCFManager * GetCFManager() {return fCFManager;} // get corr manager
	62
	63	void SetRsnPDG(Int_t code) {fRsnPDG = code; } // defines the PDG code of searched resonance
	64	static Double_t GetRapidity(Double_t, Double_t) ; // returns the rapidity of the Resonance (assuming PDG code)
65
66
67	protected:
68
69	Int_t fRsnPDG; // PDG code of searched resonance
70	TChain *fChain ; // chained files
71	AliESDEvent *fESD ; // pointer to the ESD event read
72	AliCFManager *fCFManager ; // pointer to the CF manager
73
74	// Histograms
75	//Number of events
76	TH1I *fHistEventsProcessed; //! simple histo for monitoring the number of events processed
77
78	ClassDef(AliCFRsnTask,1);
79	};
80
81	#endif