Add Jochens GTU simulation code

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

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

/* $Id: AliTRDtrackletGTU.h 27496 2008-07-22 08:35:45Z cblume $ */

// --------------------------------------------------------
// 
// GTU tracklet
// 
//
// --------------------------------------------------------

#include "AliTRDtrackletBase.h"
#include "AliLog.h"

class AliTRDmcmTracklet;
class AliTRDgtuParam;

class AliTRDtrackletGTU : public AliTRDtrackletBase {
 public:
  AliTRDtrackletGTU();
//  AliTRDtrackletGTU(UInt_t tracklet_word = 0);
  AliTRDtrackletGTU(AliTRDtrackletBase *tracklet); 
  AliTRDtrackletGTU(const AliTRDtrackletGTU& trk);

  ~AliTRDtrackletGTU();

  AliTRDtrackletGTU& operator=(const AliTRDtrackletGTU &rhs); 

  Bool_t IsSortable() const { return kTRUE; }
  Int_t Compare(const TObject *o) const;

  // ----- Getters for information from the tracklet word -----
  Int_t GetYbin() const { return fTracklet->GetYbin(); }
  Int_t GetdY() const { return fTracklet->GetdY(); }
  Int_t GetZbin() const { return fTracklet->GetZbin(); }
  Double_t GetPID() const { return fTracklet->GetPID(); }
  Double_t GetPID(Int_t is) const { return fTracklet->GetPID(is); }

  // ----- Getters for calculated properties -----
  Int_t GetYProj() const { return fYProj; }
  Int_t GetAlpha() const { return fAlpha; }
  Int_t GetYPrime() const { return fYPrime; }
  Int_t GetSubChannel(Int_t zch);

  // ----- Getters for offline corresponding values -----
  Bool_t CookPID() { return kFALSE; }
  Int_t GetDetector() const { return fTracklet->GetDetector(); }
  Int_t GetIndex() const { return fIndex; }

  Float_t GetdYdX() const { return (GetdY() * 140e-4 / 3.); }
  Float_t GetX() const { return 0; }
  Float_t GetY() const { return (GetYbin() * 160e-4); }
  Float_t GetZ() const { return 0; }

//  AliTRDtrackletBase* GetTracklet() const { return fTracklet; }
  UInt_t GetTrackletWord() const { return fTracklet->GetTrackletWord(); }

  Int_t GetSide() const { return GetYbin() < 0 ? 0 : 1; } 

  // ----- Setters -----
  void SetAlpha(Int_t alpha) { fAlpha = alpha; }
  void SetYProj(Int_t yproj) { fYProj = yproj; }
  void SetYPrime(Int_t yprime) { fYPrime = yprime; }

  void SetSubChannel(Int_t zch, Int_t subch);
  void SetDetector(Int_t /* id */ ) { AliError("Cannot change base tracklet"); }
  void SetIndex(Int_t idx) { fIndex = idx; }

  void RemoveTracklet() { fTracklet = fgkDummyTracklet; }

 protected:
  AliTRDgtuParam *fGtuParam;	 //!
  const AliTRDtrackletBase *fTracklet; // always points to a valid tracklet

  Int_t *fSubChannel;		//! [AliTRDgtuParam::GetNZChannels()]
  Bool_t fAssignedZ;		// tracklet assigned to a Z-channel

  Int_t fAlpha;			// calculated value for alpha
  Int_t fYProj;			// calculated value for y_proj
  Int_t fYPrime;		// calculated value for y'
  Int_t fIndex;

  static AliTRDtrackletBase* fgkDummyTracklet;

 private:

  ClassDef(AliTRDtrackletGTU, 1);
};

#endif