[u/mrichter/AliRoot.git] / RALICE / AliInvmass.h

#ifndef ALIINVMASS_H
#define ALIINVMASS_H
/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

// $Id$

#include <math.h>
 
#include "TObject.h"
#include "TObjArray.h"

#include "AliRandom.h"
#include "AliTrack.h"

class AliInvmass : public TObject
{
 public:
  AliInvmass();                                    // Default constructor
  ~AliInvmass();                                   // Destructor
  void SetStorageMode(Int_t m);                    // Set storage mode (1=single, 2=multiple)
  void SetThetaSwitch(Int_t i=1);                  // Enable (1/0) new theta for comb. bkg. reco.
  void SetPhiSwitch(Int_t i=1);                    // Enable (1/0) new phi for comb. bkg. reco.
  Int_t GetStorageMode() const;                    // Provide storage mode
  Int_t GetThetaSwitch() const;                    // Provide theta switch flag
  Int_t GetPhiSwitch() const;                      // Provide phi switch flag
  TObjArray* Invmass(TObjArray* a1,TObjArray* a2); // Two-particle inv. mass reco.
  TObjArray* CombBkg(TObjArray* a1,TObjArray* a2); // Two-particle comb. background reco.

 protected:
  Double_t fPi;     // Value of pi
  Int_t fMode;      // Storage mode for signal and bkg. results (2=separate arrays)
  Int_t fBkg;       // Flag to denote comb. background processing
  AliRandom fRndm;  // The random number generator for the comb. bkg. reconstruction
  Int_t fNewtheta;  // Flag to denote enabling of switching theta for comb. bkg. reco.
  Int_t fNewphi;    // Flag to denote enabling of switching phi for comb. bkg. reco.
  TObjArray* fMinv; // Array with reconstructed invariant mass 'tracks'
  TObjArray* fMbkg; // Array with reconstructed comb. background 'tracks'

 private:
  void Combine(TObjArray* a1,TObjArray* a2); // Make two-particle combinations

 ClassDef(AliInvmass,2) // Construction of invariant mass and combinatorial background.
};
#endif
Commit	Line	Data
d88f97cc	1	#ifndef ALIINVMASS_H
d88f97cc	2	#define ALIINVMASS_H
3da30618	3	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	4	* See cxx source for full Copyright notice */
	5
f531a546	6	// $Id$
3da30618	7
d88f97cc	8	#include <math.h>
	9
	10	#include "TObject.h"
	11	#include "TObjArray.h"
	12
	13	#include "AliRandom.h"
	14	#include "AliTrack.h"
	15
	16	class AliInvmass : public TObject
	17	{
	18	public:
	19	AliInvmass(); // Default constructor
	20	~AliInvmass(); // Destructor
	21	void SetStorageMode(Int_t m); // Set storage mode (1=single, 2=multiple)
	22	void SetThetaSwitch(Int_t i=1); // Enable (1/0) new theta for comb. bkg. reco.
	23	void SetPhiSwitch(Int_t i=1); // Enable (1/0) new phi for comb. bkg. reco.
261c0caf	24	Int_t GetStorageMode() const; // Provide storage mode
	25	Int_t GetThetaSwitch() const; // Provide theta switch flag
	26	Int_t GetPhiSwitch() const; // Provide phi switch flag
d88f97cc	27	TObjArray* Invmass(TObjArray* a1,TObjArray* a2); // Two-particle inv. mass reco.
	28	TObjArray* CombBkg(TObjArray* a1,TObjArray* a2); // Two-particle comb. background reco.
	29
	30	protected:
	31	Double_t fPi; // Value of pi
	32	Int_t fMode; // Storage mode for signal and bkg. results (2=separate arrays)
	33	Int_t fBkg; // Flag to denote comb. background processing
	34	AliRandom fRndm; // The random number generator for the comb. bkg. reconstruction
	35	Int_t fNewtheta; // Flag to denote enabling of switching theta for comb. bkg. reco.
	36	Int_t fNewphi; // Flag to denote enabling of switching phi for comb. bkg. reco.
	37	TObjArray* fMinv; // Array with reconstructed invariant mass 'tracks'
	38	TObjArray* fMbkg; // Array with reconstructed comb. background 'tracks'
	39
	40	private:
	41	void Combine(TObjArray* a1,TObjArray* a2); // Make two-particle combinations
	42
261c0caf	43	ClassDef(AliInvmass,2) // Construction of invariant mass and combinatorial background.
d88f97cc	44	};
d88f97cc	45	#endif