Bogdan: new version of MUON visualization.

[u/mrichter/AliRoot.git] / HLT / comp / AliHLTModeller.h

// @(#) $Id$

#ifndef AliHLT_Modeller
#define AliHLT_Modeller


#include "AliHLTRootTypes.h"

class AliHLTTrackArray;
class AliHLTMemHandler;
class AliHLTDigitRowData;
class AliHLTModelTrack;

struct Cluster {
  UInt_t fCharge; // Charge
  UInt_t fPad; // Pad
  UInt_t fTime; // Time
  UInt_t fSigmaY2; // SigmaY2
  UInt_t fSigmaZ2; // SigmaZ2
};

struct Digit {
  Short_t fCharge; // Charge
  Bool_t fUsed; // Flag if used
};

struct ClusterRegion {
  Int_t fMintime; // Min time
  Int_t fMaxtime; // Max time
};

class AliHLTModeller {
  
 public:
  
  AliHLTModeller();
  virtual ~AliHLTModeller();
  
  virtual void FindClusters();
  void Init(Int_t slice,Int_t patch,Char_t *trackdata,Char_t *path,Bool_t houghtracks,Bool_t binary=kTRUE);
  void CheckForOverlaps(Float_t dangle=-1,Int_t *rowrange=0);
  void CalculateCrossingPoints();
  void RemoveBadTracks();
  void WriteRemaining();
  
  void SetInputData(AliHLTDigitRowData *digits) {fRowData = digits;}
  void SetTrackThreshold(Int_t i=0) {fTrackThreshold=i;}
  void SetOverlap(Int_t p=6,Int_t t=8) {fPadOverlap=p;fTimeOverlap=t;}
  void SetSearchRange(Int_t p=1,Int_t t=2) {fPadSearch=p;fTimeSearch=t;}
  void SetInnerSearchRange(Int_t p,Int_t t) {fInnerPadSearch=p; fInnerTimeSearch=t;}
  void SetOuterSearchRange(Int_t p,Int_t t) {fOuterPadSearch=p; fOuterTimeSearch=t;}
  void SetMaxClusterRange(Int_t p,Int_t t) {fMaxPads=p; fMaxTimebins=t;}
  void Debug() {fDebug=kTRUE;}
  virtual Bool_t SetFitRange(AliHLTModelTrack */*track*/,Int_t */*p*/,Int_t */*t*/) {return kFALSE;}
  virtual void SetNmaxOverlaps(Int_t /*i*/) {return;}
  virtual void SetChiSqMax(Float_t /*f*/,Int_t /*p*/) {return;}
  
  AliHLTTrackArray *GetTracks() {return fTracks;}
    
 protected:
  
  AliHLTTrackArray *fTracks; //! Array of tracks
  AliHLTDigitRowData *fRowData;//! Row data
  Digit *fRow; //! Current row
  Char_t fPath[1024]; // Path to the files

  Bool_t fDebug; // Flag to switch on/off the debugging
  Int_t fNClusters; // Number of clusters
  Int_t fMaxClusters; // Max clusters (?)
  Int_t fCurrentPadRow; // Current pad row
  Int_t fMaxPads; // Max pads (?)
  Int_t fMaxTimebins; // Max time bins (?)
  Int_t fPadSearch; // Pad search (?)
  Int_t fTimeSearch; // Time search (?)
  Int_t fInnerPadSearch; // Inner pad search (?)
  Int_t fInnerTimeSearch; // Inner time search (?)
  Int_t fOuterPadSearch; // Outer Pad search (?)
  Int_t fOuterTimeSearch; // Outer time search (?)
  
  Int_t fSlice; // Slice
  Int_t fPatch; // Patch
  
  void FillCluster(AliHLTModelTrack *track,Cluster *cluster,Int_t row,Int_t npads);
  void FillZeros(AliHLTDigitRowData *digPt,Bool_t reversesign=kFALSE);
  void LocateCluster(AliHLTModelTrack *track,ClusterRegion *region,Int_t &padmin,Int_t &padmax);
  void GetTrackID(Int_t pad,Int_t time,Int_t *trackID);
  
 private:
  Bool_t fHoughTracks; // Flag to switch on/off Hough tracks
  Bool_t CheckCluster(Int_t hitpad,Int_t hittime);
  Float_t fPadOverlap; // Pad overlap (?)
  Float_t fTimeOverlap; // Time overlap (?)
  Int_t fTrackThreshold; //minimum weigth track need in order to be included.(=Nhits/weight)
  AliHLTMemHandler *fMemHandler; //! Pointer to the memory handler

  void CalcClusterWidth(Cluster *cl,Float_t &sigmaY2,Float_t &sigmaZ2);
  
  ClassDef(AliHLTModeller,1) //Modeller class
    
};

typedef AliHLTModeller AliL3Modeller; // for backward compatibility

#endif