[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>
#include <AliTPCRecoParam.h>
#define kMAXCLUSTER 2500

class TFile;
class TClonesArray;
class AliTPCParam;
class AliTPCRecoParam;
class AliTPCclusterMI;
class AliTPCClustersRow;
class AliRawReader;
class AliSimDigits;
class TTree;
class TTreeSRedirector;
class  AliRawEventHeaderBase;
class AliTPCCalROC;

class AliTPCclustererMI : public TObject{
public:
  AliTPCclustererMI(const AliTPCParam* par, const AliTPCRecoParam * recoParam = 0);
  AliTPCclustererMI(const AliTPCclustererMI &param); // copy constructor
  AliTPCclustererMI &operator = (const AliTPCclustererMI & param); //assignment
  virtual ~AliTPCclustererMI();
  virtual void Digits2Clusters();
  virtual void Digits2ClustersOld(AliRawReader* rawReader);
  virtual void Digits2Clusters(AliRawReader* rawReader);
  virtual void SetInput(TTree * tree);  // set input tree with digits
  virtual void SetOutput(TTree * tree); // set output tree with 
  virtual void FillRow();               // fill the output container - Tree or TObjArray
  TObjArray * GetOutputArray(){return fOutputArray;}
  TClonesArray * GetOutputClonesArray(){return fOutputClonesArray;}

  void StoreInClonesArray(Bool_t bOutput = kTRUE) {fBClonesArray = bOutput;} // store output clusters in TClonesArray

  void SetUseHLTClusters(Int_t useHLTClusters) {fUseHLTClusters = useHLTClusters;} // set usage from HLT clusters from rec.C options

private:
  Bool_t IsMaximum(Float_t k, Int_t max, const Float_t *bins) const; 
  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, Float_t *matrix, Int_t pos);  // 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(AliTPCCalROC * noiseROC);
  Bool_t AcceptCluster(AliTPCclusterMI*c);
  Double_t  ProcesSignal(Float_t * signal, Int_t nchannels, Int_t id[3], Double_t &rms, Double_t &pedestalCalib);
  void ProcessSectorData();
  Int_t ReadHLTClusters();
  
  Float_t * fBins;       //!digits array
  Int_t   * fSigBins; //!digits array containg only timebins above threshold
  Int_t     fNSigBins;//!size of fSigBins
  Int_t fLoop;         //loop - cf in 2 loops
  Int_t fMaxBin;       //current ( for current sector)  maximal bin
  Int_t fMaxTime;      //current ( for current sector)  maximal time
  Int_t fMaxPad;       //current ( for current sector)  maximal pad
  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
  AliRawEventHeaderBase *fEventHeader; //! event header information
  UInt_t  fTimeStamp;   // Time Stamp
  UInt_t  fEventType;   // Event Type
  TTree * fInput;   //!input  tree with digits - object not owner
  TTree * fOutput;   //!output tree with digits - object not owner
  TObjArray *fOutputArray;     //! output TObjArray with pointers arrays of cluster
  TClonesArray *fOutputClonesArray; //! output TClonesArray with clusters
  AliTPCClustersRow * fRowCl;  //! current cluster row
  AliSimDigits * fRowDig;      //! current digits row
  const AliTPCParam * fParam;        //! tpc parameters
  Int_t fNcluster;             // number of clusters - for given row
  Int_t fNclusters;            // tot number of clusters
  TTreeSRedirector *fDebugStreamer;     //!debug streamer
  const AliTPCRecoParam  * fRecoParam;        //! reconstruction parameters
  Bool_t  fBDumpSignal; // dump signal flag
  Bool_t  fBClonesArray; // output clusters stored in TClonesArray 
  Int_t  fUseHLTClusters; // use HLT clusters instead of offline clusters

  // Non-persistent arrays

  Float_t** fAllBins; //! All sector bins
  Int_t** fAllSigBins;//! All signal bins in a sector
  Int_t*  fAllNSigBins;//! Number of signal bins in a sector


  ClassDef(AliTPCclustererMI,3)  // Time Projection Chamber digits
};

inline Bool_t AliTPCclustererMI::IsMaximum(Float_t q,Int_t max,const Float_t *bins) const {
  //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;
  //
  //
  if (fRecoParam->GetClusterMaxRange(1)>0){  //local maxima in z on more than 1 time bin 
    if (bins[-2] > q) return kFALSE;
    if (bins[ 2] > q) return kFALSE;
  }
  if (fRecoParam->GetClusterMaxRange(0)>0){  //local maxima in y on more than pad 
    if (bins[-2*max] > q) return kFALSE;
    if (bins[ 2*max] > q) return kFALSE;
  }

  return kTRUE; 
}


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

#endif
Commit	Line	Data
	1	#ifndef ALITPCCLUSTERERMI_H
	2	#define ALITPCCLUSTERERMI_H
	3
	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
	11	//-------------------------------------------------------
	12	// TPC clusterer
	13	//
	14	// Origin: Marian Ivanov
	15	//-------------------------------------------------------
	16	#include <Rtypes.h>
	17	#include <TObject.h>
	18	#include <AliTPCRecoParam.h>
	19	#define kMAXCLUSTER 2500
	20
	21	class TFile;
	22	class TClonesArray;
	23	class AliTPCParam;
	24	class AliTPCRecoParam;
	25	class AliTPCclusterMI;
	26	class AliTPCClustersRow;
	27	class AliRawReader;
	28	class AliSimDigits;
	29	class TTree;
	30	class TTreeSRedirector;
	31	class AliRawEventHeaderBase;
	32	class AliTPCCalROC;
	33
	34	class AliTPCclustererMI : public TObject{
	35	public:
	36	AliTPCclustererMI(const AliTPCParam* par, const AliTPCRecoParam * recoParam = 0);
	37	AliTPCclustererMI(const AliTPCclustererMI &param); // copy constructor
	38	AliTPCclustererMI &operator = (const AliTPCclustererMI & param); //assignment
	39	virtual ~AliTPCclustererMI();
	40	virtual void Digits2Clusters();
	41	virtual void Digits2ClustersOld(AliRawReader* rawReader);
	42	virtual void Digits2Clusters(AliRawReader* rawReader);
	43	virtual void SetInput(TTree * tree); // set input tree with digits
	44	virtual void SetOutput(TTree * tree); // set output tree with
	45	virtual void FillRow(); // fill the output container - Tree or TObjArray
	46	TObjArray * GetOutputArray(){return fOutputArray;}
	47	TClonesArray * GetOutputClonesArray(){return fOutputClonesArray;}
	48
	49	void StoreInClonesArray(Bool_t bOutput = kTRUE) {fBClonesArray = bOutput;} // store output clusters in TClonesArray
	50
	51	void SetUseHLTClusters(Int_t useHLTClusters) {fUseHLTClusters = useHLTClusters;} // set usage from HLT clusters from rec.C options
	52
	53	private:
	54	Bool_t IsMaximum(Float_t k, Int_t max, const Float_t *bins) const;
	55	void MakeCluster2(Int_t k,Int_t max,Float_t *bins,UInt_t m,
	56	AliTPCclusterMI &c);
	57	void MakeCluster(Int_t k,Int_t max,Float_t *bins,UInt_t m,
	58	AliTPCclusterMI &c);
	59	Float_t GetSigmaY2(Int_t iz);
	60	Float_t GetSigmaZ2(Int_t iz);
	61	Float_t FitMax(Float_t vmatrix[5][5], Float_t y, Float_t z, Float_t sigmay, Float_t sigmaz);
	62	void AddCluster(AliTPCclusterMI &c, Float_t *matrix, Int_t pos); // add the cluster to the array
	63	void UnfoldCluster(Float_t * matrix[7], Float_t recmatrix[5][5],
	64	Float_t & meani, Float_t & meanj, Float_t & sum, Float_t &overlap );
	65	void FindClusters(AliTPCCalROC * noiseROC);
	66	Bool_t AcceptCluster(AliTPCclusterMI*c);
	67	Double_t ProcesSignal(Float_t * signal, Int_t nchannels, Int_t id[3], Double_t &rms, Double_t &pedestalCalib);
	68	void ProcessSectorData();
	69	Int_t ReadHLTClusters();
	70
	71	Float_t * fBins; //!digits array
	72	Int_t * fSigBins; //!digits array containg only timebins above threshold
	73	Int_t fNSigBins;//!size of fSigBins
	74	Int_t fLoop; //loop - cf in 2 loops
	75	Int_t fMaxBin; //current ( for current sector) maximal bin
	76	Int_t fMaxTime; //current ( for current sector) maximal time
	77	Int_t fMaxPad; //current ( for current sector) maximal pad
	78	Int_t fSector; //!current sector
	79	Int_t fRow; //!current row
	80	Float_t fSign; //!current sign
	81	Float_t fRx; // current radius
	82	Float_t fPadWidth; // the width of the pad
	83	Float_t fPadLength; // the width of the pad
	84	Float_t fZWidth; //the z bin width
	85	Bool_t fPedSubtraction; // perform pedestal subtraction or not
	86	AliRawEventHeaderBase *fEventHeader; //! event header information
	87	UInt_t fTimeStamp; // Time Stamp
	88	UInt_t fEventType; // Event Type
	89	TTree * fInput; //!input tree with digits - object not owner
	90	TTree * fOutput; //!output tree with digits - object not owner
	91	TObjArray *fOutputArray; //! output TObjArray with pointers arrays of cluster
	92	TClonesArray *fOutputClonesArray; //! output TClonesArray with clusters
	93	AliTPCClustersRow * fRowCl; //! current cluster row
	94	AliSimDigits * fRowDig; //! current digits row
	95	const AliTPCParam * fParam; //! tpc parameters
	96	Int_t fNcluster; // number of clusters - for given row
	97	Int_t fNclusters; // tot number of clusters
	98	TTreeSRedirector *fDebugStreamer; //!debug streamer
	99	const AliTPCRecoParam * fRecoParam; //! reconstruction parameters
	100	Bool_t fBDumpSignal; // dump signal flag
	101	Bool_t fBClonesArray; // output clusters stored in TClonesArray
	102	Int_t fUseHLTClusters; // use HLT clusters instead of offline clusters
	103
	104	// Non-persistent arrays
	105
	106	Float_t** fAllBins; //! All sector bins
	107	Int_t** fAllSigBins;//! All signal bins in a sector
	108	Int_t* fAllNSigBins;//! Number of signal bins in a sector
	109
	110
	111	ClassDef(AliTPCclustererMI,3) // Time Projection Chamber digits
	112	};
	113
	114	inline Bool_t AliTPCclustererMI::IsMaximum(Float_t q,Int_t max,const Float_t *bins) const {
	115	//is this a local maximum ?
	116	if (bins[-max] >= q) return kFALSE;
	117	if (bins[-1 ] >= q) return kFALSE;
	118	if (bins[+max] > q) return kFALSE;
	119	if (bins[+1 ] > q) return kFALSE;
	120	if (bins[-max-1] >= q) return kFALSE;
	121	if (bins[+max-1] >= q) return kFALSE;
	122	if (bins[+max+1] > q) return kFALSE;
	123	if (bins[-max+1] >= q) return kFALSE;
	124	//
	125	//
	126	if (fRecoParam->GetClusterMaxRange(1)>0){ //local maxima in z on more than 1 time bin
	127	if (bins[-2] > q) return kFALSE;
	128	if (bins[ 2] > q) return kFALSE;
	129	}
	130	if (fRecoParam->GetClusterMaxRange(0)>0){ //local maxima in y on more than pad
	131	if (bins[-2*max] > q) return kFALSE;
	132	if (bins[ 2*max] > q) return kFALSE;
	133	}
	134
	135	return kTRUE;
	136	}
	137
	138
	139
	140	//-----------------------------------------------------------------
	141
	142	#endif
	143
	144