[u/mrichter/AliRoot.git] / TPC / AliTPCclustererMI.h

#ifndef ALITPCCLUSTERERMI_H
#define ALITPCCLUSTERERMI_H

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

/* $Id$ */


//-------------------------------------------------------
//                       TPC clusterer
//
//   Origin: Marian Ivanov  
//-------------------------------------------------------
#include <Rtypes.h>
#include <TObject.h>
#define kMAXCLUSTER 2500

class TFile;
class AliTPCParam;
class AliTPCclusterMI;
class AliTPCClustersRow;
class AliRawReader;
class AliSimDigits;
class TTree;

class AliTPCclustererMI : public TObject{
public:
  AliTPCclustererMI(const AliTPCParam* par);
  virtual void Digits2Clusters();
  virtual void Digits2Clusters(AliRawReader* rawReader);
  virtual void SetPedSubtraction(Bool_t pedestalSub = kFALSE)
    { fPedSubtraction = pedestalSub; };
  virtual void SetOldRCUFormat(Bool_t rcuFormat = kFALSE)
    { fIsOldRCUFormat = rcuFormat; };
  virtual void SetInput(TTree * tree);  // set input tree with digits    
  virtual void SetOutput(TTree * tree); //set output tree with 
private:
  Bool_t IsMaximum(Float_t k, Int_t max, const Float_t *bins); 
  void MakeCluster2(Int_t k,Int_t max,Float_t *bins,UInt_t m,
   AliTPCclusterMI &c);  
  void MakeCluster(Int_t k,Int_t max,Float_t *bins,UInt_t m,
   AliTPCclusterMI &c); 
  Float_t  GetSigmaY2(Int_t iz);
  Float_t  GetSigmaZ2(Int_t iz);
  Float_t  FitMax(Float_t vmatrix[5][5], Float_t y, Float_t z, Float_t sigmay, Float_t sigmaz);
  void AddCluster(AliTPCclusterMI &c);  // add the cluster to the array
  void UnfoldCluster(Float_t * matrix[7], Float_t recmatrix[5][5], 
		     Float_t & meani, Float_t & meanj, Float_t & sum, Float_t &overlap );
  void FindClusters();


  Float_t * fBins;       //!digits array
  Float_t * fResBins;    //!digits array with res. after 1 finder
  Int_t fLoop;         //loop - cf in 2 loops
  Int_t fMaxBin;
  Int_t fMaxTime;
  Int_t fMaxPad;
  Int_t fSector;      //!current sector
  Int_t fRow;         //!current row
  Float_t fSign;      //!current sign 
  Float_t fRx;        // current radius
  Float_t fPadWidth;  // the width of the pad
  Float_t fPadLength;  // the width of the pad
  Float_t fZWidth;     //the z bin width

  Bool_t  fPedSubtraction; // perform pedestal subtraction or not
  Bool_t  fIsOldRCUFormat; // assume old RCU raw data format

  TTree * fInput;   //!input  tree with digits - object not owner
  TTree * fOutput;   //!output tree with digits - object not owner
  AliTPCClustersRow * fRowCl;  //! current cluster row
  AliSimDigits * fRowDig;      //! current digits row
  const AliTPCParam * fParam;        //! tpc parameters
  Int_t fNcluster;             // number of clusters - for given row
  ClassDef(AliTPCclustererMI,1)  // Time Projection Chamber digits
};

inline Bool_t AliTPCclustererMI::IsMaximum(Float_t q,Int_t max,const Float_t *bins){
  //is this a local maximum ?
  if (bins[-max] >= q) return kFALSE;
  if (bins[-1  ] >= q) return kFALSE; 
  if (bins[+max] > q) return kFALSE; 
  if (bins[+1  ] > q) return kFALSE; 
  if (bins[-max-1] >= q) return kFALSE;
  if (bins[+max-1] >= q) return kFALSE; 
  if (bins[+max+1] > q) return kFALSE; 
  if (bins[-max+1] >= q) return kFALSE;
  return kTRUE; 
}


//-----------------------------------------------------------------

#endif
Commit	Line	Data
1c53abe2	1	#ifndef ALITPCCLUSTERERMI_H
	2	#define ALITPCCLUSTERERMI_H
	3
722aa38c	4	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	5	* See cxx source for full Copyright notice */
	6
	7	/* $Id$ */
	8
	9
	10
1c53abe2	11	//-------------------------------------------------------
	12	// TPC clusterer
	13	//
	14	// Origin: Marian Ivanov
	15	//-------------------------------------------------------
	16	#include <Rtypes.h>
	17	#include <TObject.h>
	18	#define kMAXCLUSTER 2500
	19
	20	class TFile;
	21	class AliTPCParam;
	22	class AliTPCclusterMI;
	23	class AliTPCClustersRow;
f8aae377	24	class AliRawReader;
1c53abe2	25	class AliSimDigits;
	26	class TTree;
	27
	28	class AliTPCclustererMI : public TObject{
	29	public:
f8aae377	30	AliTPCclustererMI(const AliTPCParam* par);
	31	virtual void Digits2Clusters();
	32	virtual void Digits2Clusters(AliRawReader* rawReader);
b6f060dc	33	virtual void SetPedSubtraction(Bool_t pedestalSub = kFALSE)
b6f060dc	34	{ fPedSubtraction = pedestalSub; };
c8c88e5d	35	virtual void SetOldRCUFormat(Bool_t rcuFormat = kFALSE)
c8c88e5d	36	{ fIsOldRCUFormat = rcuFormat; };
1c53abe2	37	virtual void SetInput(TTree * tree); // set input tree with digits
	38	virtual void SetOutput(TTree * tree); //set output tree with
	39	private:
13116aec	40	Bool_t IsMaximum(Float_t k, Int_t max, const Float_t *bins);
13116aec	41	void MakeCluster2(Int_t k,Int_t max,Float_t *bins,UInt_t m,
1c53abe2	42	AliTPCclusterMI &c);
13116aec	43	void MakeCluster(Int_t k,Int_t max,Float_t *bins,UInt_t m,
1c53abe2	44	AliTPCclusterMI &c);
	45	Float_t GetSigmaY2(Int_t iz);
	46	Float_t GetSigmaZ2(Int_t iz);
	47	Float_t FitMax(Float_t vmatrix[5][5], Float_t y, Float_t z, Float_t sigmay, Float_t sigmaz);
	48	void AddCluster(AliTPCclusterMI &c); // add the cluster to the array
13116aec	49	void UnfoldCluster(Float_t * matrix[7], Float_t recmatrix[5][5],
1c53abe2	50	Float_t & meani, Float_t & meanj, Float_t & sum, Float_t &overlap );
f8aae377	51	void FindClusters();
1c53abe2	52
	53
	54
13116aec	55	Float_t * fBins; //!digits array
13116aec	56	Float_t * fResBins; //!digits array with res. after 1 finder
1c53abe2	57	Int_t fLoop; //loop - cf in 2 loops
	58	Int_t fMaxBin;
	59	Int_t fMaxTime;
	60	Int_t fMaxPad;
	61	Int_t fSector; //!current sector
508541c7	62	Int_t fRow; //!current row
1c53abe2	63	Float_t fSign; //!current sign
	64	Float_t fRx; // current radius
	65	Float_t fPadWidth; // the width of the pad
	66	Float_t fPadLength; // the width of the pad
	67	Float_t fZWidth; //the z bin width
	68
b6f060dc	69	Bool_t fPedSubtraction; // perform pedestal subtraction or not
c8c88e5d	70	Bool_t fIsOldRCUFormat; // assume old RCU raw data format
b6f060dc	71
1c53abe2	72	TTree * fInput; //!input tree with digits - object not owner
	73	TTree * fOutput; //!output tree with digits - object not owner
	74	AliTPCClustersRow * fRowCl; //! current cluster row
	75	AliSimDigits * fRowDig; //! current digits row
	76	const AliTPCParam * fParam; //! tpc parameters
	77	Int_t fNcluster; // number of clusters - for given row
	78	ClassDef(AliTPCclustererMI,1) // Time Projection Chamber digits
	79	};
	80
13116aec	81	inline Bool_t AliTPCclustererMI::IsMaximum(Float_t q,Int_t max,const Float_t *bins){
1c53abe2	82	//is this a local maximum ?
	83	if (bins[-max] >= q) return kFALSE;
	84	if (bins[-1 ] >= q) return kFALSE;
	85	if (bins[+max] > q) return kFALSE;
	86	if (bins[+1 ] > q) return kFALSE;
	87	if (bins[-max-1] >= q) return kFALSE;
	88	if (bins[+max-1] >= q) return kFALSE;
	89	if (bins[+max+1] > q) return kFALSE;
	90	if (bins[-max+1] >= q) return kFALSE;
	91	return kTRUE;
	92	}
	93
	94
	95
	96	//-----------------------------------------------------------------
	97
	98	#endif
	99
	100