[u/mrichter/AliRoot.git] / ITS / AliITSClusterFinder.h

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

////////////////////////////////////////////////
//  ITS Cluster Finder Class                  //
////////////////////////////////////////////////

#include <TObject.h>
#include <TClonesArray.h>

class AliITSMap;
class AliITSresponse;
class AliITSsegmentation;
class AliITSRawCluster;
class AliITS;
class AliITSdigit;
class AliITSRecPoint;

//----------------------------------------------------------------------
class AliITSClusterFinder :public TObject{
 public:
    AliITSClusterFinder(); // Default constructor
    AliITSClusterFinder(AliITSsegmentation *seg, AliITSresponse *resp,
			TClonesArray *digits);// Standard Constructor
    virtual ~AliITSClusterFinder(); // Destructor
    AliITSClusterFinder(const AliITSClusterFinder &source); // copy constructor
    // assignment operator
    AliITSClusterFinder& operator=(const AliITSClusterFinder &source);
    virtual void SetResponse(AliITSresponse *response) {// set response
	fResponse=response;}
    virtual void SetSegmentation(AliITSsegmentation *segmentation) {
	// set segmentation
	fSegmentation=segmentation;}
    virtual void SetDigits(TClonesArray *ITSdigits) {// set digits
	fDigits=ITSdigits;fNdigits = fDigits->GetEntriesFast();}
    virtual AliITSdigit* GetDigit(Int_t i){ // Returns ith digit
	return (AliITSdigit*) fDigits->UncheckedAt(i);}
    virtual TClonesArray* Digits(){ // Gets fDigits
	return fDigits;}
    virtual Int_t   NDigits() const {// Get Number of Digits
	return fNdigits;}
    // Standard Getters.
    virtual AliITSresponse * GetResp(){// Returns fResponse
	return fResponse;}
    virtual AliITSsegmentation * GetSeg(){// Returns fSegmentation
	return fSegmentation;}
    virtual Int_t GetNRawClusters() const { // returns fNRawClusters
	return fNRawClusters;}
    AliITSMap   *Map() {// map
	return fMap;}
    virtual Int_t GetNperMax() const { // returns fNperMax
	return fNperMax;}
    virtual Int_t GetDeclusterFlag() const { // returns fDeclusterFlag
	return fDeclusterFlag;}
    virtual Int_t GetClusterSize() const { // returns fClusterSize
	return fClusterSize;}
    virtual Int_t GetNPeaks() const { // returns fNPeaks
	return fNPeaks;}
    //
    virtual void AddCluster(Int_t branch, AliITSRawCluster *c);
    virtual void AddCluster(Int_t branch, AliITSRawCluster *c,
			    AliITSRecPoint &rp);
    virtual void FindRawClusters(Int_t mod=0); // Finds cluster of digits.
    // Determins if digit i has a neighbor and if so that neighor index is j.
    virtual Bool_t IsNeighbor(TObjArray *digs,Int_t i,Int_t j[]) const;
    // Given a cluster of digits, creates the nessesary RecPoint. May also
    // do some peak separation.
    virtual void CreateRecPoints(TObjArray *,Int_t){};
    virtual void FindCluster(Int_t,Int_t,AliITSRawCluster *) {}// find cluster
    virtual void Decluster(AliITSRawCluster *) {}// Decluster
    virtual void SetNperMax(Int_t npermax=3) {
	// Set max. Number of cells per local cluster
	fNperMax = npermax;
    }
    virtual void SetDeclusterFlag(Int_t flag=1) {
	// Decluster ?
	fDeclusterFlag =flag;
    }
    virtual void SetClusterSize(Int_t clsize=3) {
	// Set max. cluster size ; bigger clusters will be rejected
	fClusterSize = clsize;
    }
    virtual void CalibrateCOG() {
	// Self Calibration of COG 
    }
    virtual void CorrectCOG(){
	// correct COG
    }
    virtual Bool_t Centered(AliITSRawCluster *) const {// cluster
	return kTRUE;
    }
    virtual void SplitByLocalMaxima(AliITSRawCluster *){}//split by local maxima
    virtual void   FillCluster(AliITSRawCluster *,Int_t) {}// fiil cluster
    virtual void   FillCluster(AliITSRawCluster *cluster) {// fill cluster
	FillCluster(cluster,1);
    }
    // set the fitting methods in the derived classes
    // data members

protected:
    TClonesArray       *fDigits;       //! digits
    Int_t              fNdigits;       //! num of digits

    AliITSresponse     *fResponse;     //! response
    AliITSsegmentation *fSegmentation; //!segmentation
    Int_t              fNRawClusters;  //! in case we split the cluster
                                       // and want to keep track of 
                                       // the cluster which was splitted
    AliITSMap          *fMap;          //! map
    Int_t              fNperMax;       //! NperMax
    Int_t              fDeclusterFlag; //! DeclusterFlag
    Int_t              fClusterSize;   //! ClusterSize
    Int_t              fNPeaks;        //! NPeaks  
  
    ClassDef(AliITSClusterFinder,2) //Class for clustering and reconstruction of space points
};
#endif
Commit	Line	Data
b0f5e3fc	1	#ifndef ALIITSCLUSTERFINDER_H
b0f5e3fc	2	#define ALIITSCLUSTERFINDER_H
9de0700b	3
	4	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	5	* See cxx source for full Copyright notice */
	6
	7	/* $Id$ */
b0f5e3fc	8
b0f5e3fc	9	////////////////////////////////////////////////
e8189707	10	// ITS Cluster Finder Class //
b0f5e3fc	11	////////////////////////////////////////////////
b0f5e3fc	12
e8189707	13	#include <TObject.h>
	14	#include <TClonesArray.h>
	15
	16	class AliITSMap;
	17	class AliITSresponse;
	18	class AliITSsegmentation;
	19	class AliITSRawCluster;
	20	class AliITS;
8ca6a9a5	21	class AliITSdigit;
9355b256	22	class AliITSRecPoint;
b0f5e3fc	23
9de0700b	24	//----------------------------------------------------------------------
	25	class AliITSClusterFinder :public TObject{
	26	public:
28a570fd	27	AliITSClusterFinder(); // Default constructor
9de0700b	28	AliITSClusterFinder(AliITSsegmentation seg, AliITSresponse resp,
28a570fd	29	TClonesArray *digits);// Standard Constructor
28a570fd	30	virtual ~AliITSClusterFinder(); // Destructor
9de0700b	31	AliITSClusterFinder(const AliITSClusterFinder &source); // copy constructor
	32	// assignment operator
	33	AliITSClusterFinder& operator=(const AliITSClusterFinder &source);
28a570fd	34	virtual void SetResponse(AliITSresponse *response) {// set response
28a570fd	35	fResponse=response;}
9de0700b	36	virtual void SetSegmentation(AliITSsegmentation *segmentation) {
9de0700b	37	// set segmentation
28a570fd	38	fSegmentation=segmentation;}
	39	virtual void SetDigits(TClonesArray *ITSdigits) {// set digits
	40	fDigits=ITSdigits;fNdigits = fDigits->GetEntriesFast();}
	41	virtual AliITSdigit* GetDigit(Int_t i){ // Returns ith digit
	42	return (AliITSdigit*) fDigits->UncheckedAt(i);}
	43	virtual TClonesArray* Digits(){ // Gets fDigits
	44	return fDigits;}
	45	virtual Int_t NDigits() const {// Get Number of Digits
	46	return fNdigits;}
	47	// Standard Getters.
	48	virtual AliITSresponse * GetResp(){// Returns fResponse
	49	return fResponse;}
	50	virtual AliITSsegmentation * GetSeg(){// Returns fSegmentation
	51	return fSegmentation;}
93f82b23	52	virtual Int_t GetNRawClusters() const { // returns fNRawClusters
28a570fd	53	return fNRawClusters;}
	54	AliITSMap *Map() {// map
	55	return fMap;}
93f82b23	56	virtual Int_t GetNperMax() const { // returns fNperMax
28a570fd	57	return fNperMax;}
93f82b23	58	virtual Int_t GetDeclusterFlag() const { // returns fDeclusterFlag
28a570fd	59	return fDeclusterFlag;}
93f82b23	60	virtual Int_t GetClusterSize() const { // returns fClusterSize
28a570fd	61	return fClusterSize;}
93f82b23	62	virtual Int_t GetNPeaks() const { // returns fNPeaks
28a570fd	63	return fNPeaks;}
9de0700b	64	//
	65	virtual void AddCluster(Int_t branch, AliITSRawCluster *c);
	66	virtual void AddCluster(Int_t branch, AliITSRawCluster *c,
	67	AliITSRecPoint &rp);
f8d9a5b8	68	virtual void FindRawClusters(Int_t mod=0); // Finds cluster of digits.
f8d9a5b8	69	// Determins if digit i has a neighbor and if so that neighor index is j.
bf3f2830	70	virtual Bool_t IsNeighbor(TObjArray *digs,Int_t i,Int_t j[]) const;
f8d9a5b8	71	// Given a cluster of digits, creates the nessesary RecPoint. May also
f8d9a5b8	72	// do some peak separation.
ac74f489	73	virtual void CreateRecPoints(TObjArray *,Int_t){};
	74	virtual void FindCluster(Int_t,Int_t,AliITSRawCluster *) {}// find cluster
	75	virtual void Decluster(AliITSRawCluster *) {}// Decluster
9de0700b	76	virtual void SetNperMax(Int_t npermax=3) {
	77	// Set max. Number of cells per local cluster
	78	fNperMax = npermax;
	79	}
	80	virtual void SetDeclusterFlag(Int_t flag=1) {
	81	// Decluster ?
	82	fDeclusterFlag =flag;
	83	}
	84	virtual void SetClusterSize(Int_t clsize=3) {
	85	// Set max. cluster size ; bigger clusters will be rejected
	86	fClusterSize = clsize;
	87	}
	88	virtual void CalibrateCOG() {
	89	// Self Calibration of COG
	90	}
	91	virtual void CorrectCOG(){
	92	// correct COG
	93	}
ac74f489	94	virtual Bool_t Centered(AliITSRawCluster *) const {// cluster
9de0700b	95	return kTRUE;
9de0700b	96	}
ac74f489	97	virtual void SplitByLocalMaxima(AliITSRawCluster *){}//split by local maxima
	98	virtual void FillCluster(AliITSRawCluster *,Int_t) {}// fiil cluster
	99	virtual void FillCluster(AliITSRawCluster *cluster) {// fill cluster
9de0700b	100	FillCluster(cluster,1);
	101	}
	102	// set the fitting methods in the derived classes
	103	// data members
e8189707	104
93f82b23	105	protected:
bf3f2830	106	TClonesArray *fDigits; //! digits
88cb7938	107	Int_t fNdigits; //! num of digits
88cb7938	108
9de0700b	109	AliITSresponse *fResponse; //! response
	110	AliITSsegmentation *fSegmentation; //!segmentation
	111	Int_t fNRawClusters; //! in case we split the cluster
	112	// and want to keep track of
	113	// the cluster which was splitted
	114	AliITSMap *fMap; //! map
	115	Int_t fNperMax; //! NperMax
	116	Int_t fDeclusterFlag; //! DeclusterFlag
	117	Int_t fClusterSize; //! ClusterSize
	118	Int_t fNPeaks; //! NPeaks
b0f5e3fc	119
2b00434a	120	ClassDef(AliITSClusterFinder,2) //Class for clustering and reconstruction of space points
9de0700b	121	};
b0f5e3fc	122	#endif