Update to track display by Chuncheng

[u/mrichter/AliRoot.git] / TRD / AliTRDcluster.h

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

/* $Id$ */


#include "AliCluster.h"  

class AliTRDrecPoint;

class AliTRDcluster : public AliCluster {

 public:

  AliTRDcluster() : AliCluster() { fQ=0; fTimeBin=0; fDetector=0; }
  AliTRDcluster(const AliTRDcluster &c);
  AliTRDcluster(const AliTRDrecPoint &p);

  virtual void    AddTrackIndex(Int_t *i); 


  Int_t   IsUsed() const      { return (fQ < 0) ? 1 : 0; }
  void    Use()               { fQ = -fQ; }
  
  Bool_t  From2pad() const    { return TestBit(k2pad); }
  Bool_t  From3pad() const    { return TestBit(k3pad); }
  Bool_t  From4pad() const    { return TestBit(k4pad); }
  Bool_t  From5pad() const    { return TestBit(k5pad); }
  Bool_t  FromLarge() const   { return TestBit(kLarge);}
  Bool_t  Isolated() const    { return (TestBit(k2pad) || TestBit(k3pad)); }
 
  void    SetDetector(Int_t d)         { fDetector  = d; }
  void    SetLocalTimeBin(Int_t t)     { fTimeBin   = t; }
  void    SetQ(Float_t q)              { fQ         = q; }
  
  Int_t   GetDetector() const          { return fDetector; }
  Int_t   GetLocalTimeBin() const      { return fTimeBin;  }
  Float_t GetQ() const                 { return fQ; }

  void    Set2pad()                    { SetBit(k2pad);  }
  void    Set3pad()                    { SetBit(k3pad);  }
  void    Set4pad()                    { SetBit(k4pad);  }
  void    Set5pad()                    { SetBit(k5pad);  }
  void    SetLarge()                   { SetBit(kLarge); }
  
 protected:

  enum {
    k2pad  = 0x00000001,   // 2 pad cluster
    k3pad  = 0x00000002,   // 3 pad cluster
    k4pad  = 0x00000004,   // 4 pad cluster
    k5pad  = 0x00000008,   // 5 pad cluster
    kLarge = 0x00000016    // Large cluster
  };
  
  Int_t    fDetector;       // TRD detector number
  Int_t    fTimeBin;        // Time bin number within the detector
  Float_t  fQ;              // amplitude 
				 
  ClassDef(AliTRDcluster,1) // Cluster for the TRD
 
};

#endif