[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 "AliAnalysisTaskSE.h"

class TH1I;
class TParticle ;
class AliCFManager;

class AliCFRsnTask : public AliAnalysisTaskSE {
  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     UserCreateOutputObjects();
  void     UserExec(Option_t *option);
  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
  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
318a0e1f	23	#include "AliAnalysisTaskSE.h"
2fbc0b17	24
	25	class TH1I;
	26	class TParticle ;
2fbc0b17	27	class AliCFManager;
2fbc0b17	28
318a0e1f	29	class AliCFRsnTask : public AliAnalysisTaskSE {
2fbc0b17	30	public:
	31
	32	enum {
	33	kStepGenerated = 0,
	34	kStepReconstructible = 1,
	35	kStepReconstructed = 2,
	36	kStepSelected = 3
	37	};
	38
	39	AliCFRsnTask();
	40	AliCFRsnTask(const Char_t* name);
	41	AliCFRsnTask& operator= (const AliCFRsnTask& c);
	42	AliCFRsnTask(const AliCFRsnTask& c);
	43	virtual ~AliCFRsnTask();
	44
	45	// ANALYSIS FRAMEWORK STUFF to loop on data and fill output objects
318a0e1f	46	void UserCreateOutputObjects();
318a0e1f	47	void UserExec(Option_t *option);
2fbc0b17	48	void Terminate(Option_t *);
	49
	50	// CORRECTION FRAMEWORK RELATED FUNCTIONS
	51	void SetCFManager(AliCFManager* io) {fCFManager = io;} // global correction manager
	52	AliCFManager * GetCFManager() {return fCFManager;} // get corr manager
	53
	54	void SetRsnPDG(Int_t code) {fRsnPDG = code; } // defines the PDG code of searched resonance
	55	static Double_t GetRapidity(Double_t, Double_t) ; // returns the rapidity of the Resonance (assuming PDG code)
	56
	57
	58	protected:
	59
	60	Int_t fRsnPDG; // PDG code of searched resonance
2fbc0b17	61	AliCFManager *fCFManager ; // pointer to the CF manager
	62
	63	// Histograms
	64	//Number of events
	65	TH1I *fHistEventsProcessed; //! simple histo for monitoring the number of events processed
	66
	67	ClassDef(AliCFRsnTask,1);
	68	};
	69
	70	#endif