[u/mrichter/AliRoot.git] / TRD / AliTRDtrackGTU.cxx

/**************************************************************************
 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 *                                                                        *
 * Author: The ALICE Off-line Project.                                    *
 * Contributors are mentioned in the code where appropriate.              *
 *                                                                        *
 * Permission to use, copy, modify and distribute this software and its   *
 * documentation strictly for non-commercial purposes is hereby granted   *
 * without fee, provided that the above copyright notice appears in all   *
 * copies and that both the copyright notice and this permission notice   *
 * appear in the supporting documentation. The authors make no claims     *
 * about the suitability of this software for any purpose. It is          *
 * provided "as is" without express or implied warranty.                  *
 **************************************************************************/

////////////////////////////////////////////////////////////////////////////
//                                                                        //
//  A GTU track                                                           //
//                                                                        //
//  Author: J. Klein (Jochen.Klein@cern.ch)                               //
//                                                                        //
////////////////////////////////////////////////////////////////////////////

/* $Id: AliTRDtrackGTU.cxx 27566 2008-07-24 15:31:08Z cblume $ */

#include "TObject.h"
#include "TObjArray.h"
#include "TClass.h"
#include "TMath.h"
#include "TH1F.h"

#include "AliESDTrdTrack.h"
#include "AliLog.h"
#include "AliTRDgtuParam.h"
#include "AliTRDtrackGTU.h"
#include "AliTRDtrackletGTU.h"
#include "AliTRDtrackletMCM.h"
#include "AliESDTrdTrack.h"

ClassImp(AliTRDtrackGTU)

AliTRDtrackGTU::AliTRDtrackGTU() :
  TObject(),
  fStack(-1),
  fSector(-1),
  fPID(0),
  fTracklets(0x0),
  fTrackletMask(0),
  fNTracklets(0),
  fRefLayerIdx(-1),
  fZChannel(-1),
  fZSubChannel(-1),
  fA(0),
  fB(0),
  fC(0),
  fLabel(-1)
{
// default ctor

  fTracklets = new TClonesArray("AliTRDtrackletGTU", 6);
  for (Int_t iTracklet = 0; iTracklet < 6; iTracklet++)
      new ((*fTracklets)[iTracklet]) AliTRDtrackletGTU();
//  fTracklets->BypassStreamer(kFALSE);
}

AliTRDtrackGTU::AliTRDtrackGTU(const AliTRDtrackGTU &rhs) :
  TObject(),
  fStack(rhs.fStack),
  fSector(rhs.fSector),
  fPID(rhs.fPID),
  fTracklets(0x0),
  fTrackletMask(rhs.fTrackletMask),
  fNTracklets(rhs.fNTracklets),
  fRefLayerIdx(rhs.fRefLayerIdx),
  fZChannel(rhs.fZChannel),
  fZSubChannel(rhs.fZSubChannel),
  fA(rhs.fA),
  fB(rhs.fB),
  fC(rhs.fC),
  fLabel(rhs.fLabel)
{
  fTracklets = new TClonesArray("AliTRDtrackletGTU", 6);
  for (Int_t iTracklet = 0; iTracklet < 6; iTracklet++)
    new ((*fTracklets)[iTracklet]) AliTRDtrackletGTU(*((AliTRDtrackletGTU*)(*(rhs.fTracklets))[iTracklet]));
}

AliTRDtrackGTU& AliTRDtrackGTU::operator=(const AliTRDtrackGTU &rhs)
{
  if (&rhs != this) {
    TObject::operator=(rhs);
    fStack         = rhs.fStack;
    fSector        = rhs.fSector;
    fPID           = rhs.fPID;
    fTrackletMask  = rhs.fTrackletMask;
    fNTracklets    = rhs.fNTracklets;
    fRefLayerIdx   = rhs.fRefLayerIdx;
    fZChannel      = rhs.fZChannel;
    fZSubChannel   = rhs.fZSubChannel;
    fA             = rhs.fA;
    fB             = rhs.fB;
    fC             = rhs.fC;
    fLabel         = rhs.fLabel;
    for (Int_t iTracklet = 0; iTracklet < 6; iTracklet++)
      new ((*fTracklets)[iTracklet]) AliTRDtrackletGTU(*((AliTRDtrackletGTU*)(*(rhs.fTracklets))[iTracklet])); 
  }

  return *this;
}

AliTRDtrackGTU& AliTRDtrackGTU::operator=(const AliESDTrdTrack &rhs)
{
  if ((void*) &rhs != (void*) this) {
    TObject::operator=(rhs);
    fStack         = rhs.GetStack();
    fSector        = rhs.GetSector();
    fPID           = rhs.GetPID();
    fTrackletMask  = rhs.GetLayerMask();
    fNTracklets    = 0;
    fRefLayerIdx   = -1;
    fZChannel      = -1;
    fZSubChannel   = -1;
    fA             = rhs.GetA();
    fB             = rhs.GetB();
    fC             = rhs.GetC();
    fLabel         = rhs.GetLabel();
    for (Int_t iTracklet = 0; iTracklet < 6; iTracklet++) {
      AliTRDtrackletGTU *trkl = new ((*fTracklets)[iTracklet]) AliTRDtrackletGTU();
      if (fTrackletMask & (1 << iTracklet)) {
	++fNTracklets;
	trkl->SetIndex(rhs.GetTrackletIndex(iTracklet));
      }
      else
	trkl->SetIndex(-1);
    }
  }

  return *this;
}

AliTRDtrackGTU::~AliTRDtrackGTU()
{
// dtor

  fTracklets->Delete();
  delete fTracklets;
}

void AliTRDtrackGTU::AddTracklet(const AliTRDtrackletGTU * const tracklet, Int_t layer)
{
// add a tracklet to this track

  if ( (fTrackletMask & (1 << layer)) != 0 ) {
    AliError(Form("Only one tracklet per layer (%i) possible! Mask: 0x%02x", layer, fTrackletMask));
    return;
  }

  new ((*fTracklets)[layer]) AliTRDtrackletGTU(*tracklet);
  fNTracklets++;
  fTrackletMask |= (1 << layer);
}

AliTRDtrackletGTU* AliTRDtrackGTU::GetTracklet(Int_t layer) const
{
// get a pointer to the tracklet in the layer specified

  if (IsTrackletInLayer(layer))
    return ((AliTRDtrackletGTU*) (*fTracklets)[layer]);
  else
    return 0x0;
}

Int_t AliTRDtrackGTU::GetNTracklets() const
{
// returns the number of tracklets in this track

  return fNTracklets;
}

Bool_t AliTRDtrackGTU::IsTrackletInLayer(Int_t layer) const
{
// checks for a tracklet in the given layer

  if ( (GetTrackletMask() & (1 << layer)) != 0)
    return kTRUE;
  else
    return kFALSE;
}

void AliTRDtrackGTU::SetFitParams(Float_t a, Float_t b, Float_t c)
{
// set the fit parameters

  fA = a;
  fB = b;
  fC = c;
}

Int_t AliTRDtrackGTU::GetZSubChannel()
{
// returns the z-subchannel

  if (fZSubChannel < 0) {
    for (Int_t layer = 0; layer < AliTRDgtuParam::GetNLayers(); layer++) {
      if (IsTrackletInLayer(layer)) {
	AliTRDtrackletGTU *trkl = (AliTRDtrackletGTU*) (*fTracklets)[layer];
	if (trkl) {
	  if ((fZSubChannel > -1) &&
	      (fZSubChannel != trkl->GetSubChannel(GetZChannel())))
	    AliError(Form("found inconsistent z-subchannels: track = %i/%i, trkl = %i",
			  GetZChannel(), fZSubChannel, trkl->GetSubChannel(GetZChannel())));
	  fZSubChannel = trkl->GetSubChannel(GetZChannel());
	}
	else {
	  AliError("no tracklet where one should be according to layer mask");
	}
      }
    }
  }
  return fZSubChannel;
}

Int_t AliTRDtrackGTU::GetYapprox()
{
  // returns an approximated y-position for the track
  // taken from the projected y-position of the tracklet in the reference layer
  // in which the track was found

  if ((fRefLayerIdx > -1) && (fRefLayerIdx < AliTRDgtuParam::GetNRefLayers()))
    return ((AliTRDtrackletGTU*) (*fTracklets)[AliTRDgtuParam::GetRefLayer(fRefLayerIdx)])->GetYProj();
  else
    return 0;
}

AliESDTrdTrack* AliTRDtrackGTU::CreateTrdTrack() const
{
  // creates an AliESDTrdTrack to be added to the ESD

  AliESDTrdTrack *trk = new AliESDTrdTrack();
  trk->SetA((Int_t) fA);
  trk->SetLayerMask(fTrackletMask);
  trk->SetPID(fPID);
  trk->SetB((Int_t) fB);
  trk->SetStack(fStack);
  trk->SetSector(fSector);
  trk->SetLabel(fLabel);

  for (Int_t iLayer = 0; iLayer < AliTRDgtuParam::GetNLayers(); iLayer++) {
    AliTRDtrackletGTU *trklGTU = GetTracklet(iLayer);
    if (trklGTU) {
      trk->SetTrackletIndex(trklGTU->GetIndex(), iLayer);
      AliESDTrdTracklet *trkl = trklGTU->GetTrackletESD();
      if (trkl)
	trk->AddTrackletReference(trkl, iLayer);
    }
  }

  return trk;
}

Bool_t AliTRDtrackGTU::CookLabel()
{
  // assign label from tracklets according to frequency

  Int_t nLabels = 0;
  Int_t label[6] = { 0 };
  Int_t count[6] = { 0 };

  for (Int_t iTracklet = 0; iTracklet < 6; iTracklet++) {
    if (!IsTrackletInLayer(iTracklet))
      continue;

    Int_t currLabel = GetTracklet(iTracklet)->GetLabel();

    Bool_t assigned = kFALSE;
    for (Int_t iLabel = 0; iLabel < nLabels; iLabel++) {
      if (currLabel == label[iLabel]) {
	count[iLabel]++;
	assigned = kTRUE;
	break;
      }
    }

    if (!assigned) {
      label[nLabels] = currLabel;
      count[nLabels] = 1;
      nLabels++;
    }
  }

  Int_t index[32];
  TMath::Sort(6, count, index);
  fLabel = label[index[0]];

  return kTRUE;
}
Commit	Line	Data
52c19022	1	/**************************************************************************
	2	* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	3	* *
	4	* Author: The ALICE Off-line Project. *
	5	* Contributors are mentioned in the code where appropriate. *
	6	* *
	7	* Permission to use, copy, modify and distribute this software and its *
	8	* documentation strictly for non-commercial purposes is hereby granted *
	9	* without fee, provided that the above copyright notice appears in all *
	10	* copies and that both the copyright notice and this permission notice *
	11	* appear in the supporting documentation. The authors make no claims *
	12	* about the suitability of this software for any purpose. It is *
	13	* provided "as is" without express or implied warranty. *
	14	**************************************************************************/
	15
	16	////////////////////////////////////////////////////////////////////////////
	17	// //
	18	// A GTU track //
	19	// //
	20	// Author: J. Klein (Jochen.Klein@cern.ch) //
	21	// //
	22	////////////////////////////////////////////////////////////////////////////
	23
	24	/* $Id: AliTRDtrackGTU.cxx 27566 2008-07-24 15:31:08Z cblume $ */
	25
	26	#include "TObject.h"
	27	#include "TObjArray.h"
4cc89512	28	#include "TClass.h"
2f1bde31	29	#include "TMath.h"
4cc89512	30	#include "TH1F.h"
52c19022	31
4d6aee34	32	#include "AliESDTrdTrack.h"
52c19022	33	#include "AliLog.h"
	34	#include "AliTRDgtuParam.h"
	35	#include "AliTRDtrackGTU.h"
	36	#include "AliTRDtrackletGTU.h"
4cc89512	37	#include "AliTRDtrackletMCM.h"
52c19022	38	#include "AliESDTrdTrack.h"
	39
	40	ClassImp(AliTRDtrackGTU)
5f006bd7	41
52c19022	42	AliTRDtrackGTU::AliTRDtrackGTU() :
	43	TObject(),
	44	fStack(-1),
	45	fSector(-1),
52c19022	46	fPID(0),
	47	fTracklets(0x0),
	48	fTrackletMask(0),
	49	fNTracklets(0),
	50	fRefLayerIdx(-1),
	51	fZChannel(-1),
	52	fZSubChannel(-1),
	53	fA(0),
	54	fB(0),
4cc89512	55	fC(0),
4cc89512	56	fLabel(-1)
52c19022	57	{
4cc89512	58	// default ctor
4cc89512	59
52c19022	60	fTracklets = new TClonesArray("AliTRDtrackletGTU", 6);
	61	for (Int_t iTracklet = 0; iTracklet < 6; iTracklet++)
	62	new ((*fTracklets)[iTracklet]) AliTRDtrackletGTU();
	63	// fTracklets->BypassStreamer(kFALSE);
	64	}
	65
dd09a2e2	66	AliTRDtrackGTU::AliTRDtrackGTU(const AliTRDtrackGTU &rhs) :
	67	TObject(),
	68	fStack(rhs.fStack),
	69	fSector(rhs.fSector),
	70	fPID(rhs.fPID),
	71	fTracklets(0x0),
	72	fTrackletMask(rhs.fTrackletMask),
	73	fNTracklets(rhs.fNTracklets),
	74	fRefLayerIdx(rhs.fRefLayerIdx),
	75	fZChannel(rhs.fZChannel),
	76	fZSubChannel(rhs.fZSubChannel),
	77	fA(rhs.fA),
	78	fB(rhs.fB),
	79	fC(rhs.fC),
	80	fLabel(rhs.fLabel)
	81	{
	82	fTracklets = new TClonesArray("AliTRDtrackletGTU", 6);
	83	for (Int_t iTracklet = 0; iTracklet < 6; iTracklet++)
	84	new ((fTracklets)[iTracklet]) AliTRDtrackletGTU(((AliTRDtrackletGTU)((rhs.fTracklets))[iTracklet]));
	85	}
	86
	87	AliTRDtrackGTU& AliTRDtrackGTU::operator=(const AliTRDtrackGTU &rhs)
	88	{
	89	if (&rhs != this) {
	90	TObject::operator=(rhs);
	91	fStack = rhs.fStack;
	92	fSector = rhs.fSector;
	93	fPID = rhs.fPID;
	94	fTrackletMask = rhs.fTrackletMask;
	95	fNTracklets = rhs.fNTracklets;
	96	fRefLayerIdx = rhs.fRefLayerIdx;
	97	fZChannel = rhs.fZChannel;
	98	fZSubChannel = rhs.fZSubChannel;
	99	fA = rhs.fA;
	100	fB = rhs.fB;
	101	fC = rhs.fC;
	102	fLabel = rhs.fLabel;
	103	for (Int_t iTracklet = 0; iTracklet < 6; iTracklet++)
	104	new ((fTracklets)[iTracklet]) AliTRDtrackletGTU(((AliTRDtrackletGTU)((rhs.fTracklets))[iTracklet]));
	105	}
	106
	107	return *this;
	108	}
	109
d9a2bdff	110	AliTRDtrackGTU& AliTRDtrackGTU::operator=(const AliESDTrdTrack &rhs)
	111	{
	112	if ((void) &rhs != (void) this) {
	113	TObject::operator=(rhs);
	114	fStack = rhs.GetStack();
	115	fSector = rhs.GetSector();
	116	fPID = rhs.GetPID();
	117	fTrackletMask = rhs.GetLayerMask();
	118	fNTracklets = 0;
	119	fRefLayerIdx = -1;
	120	fZChannel = -1;
	121	fZSubChannel = -1;
	122	fA = rhs.GetA();
	123	fB = rhs.GetB();
	124	fC = rhs.GetC();
	125	fLabel = rhs.GetLabel();
	126	for (Int_t iTracklet = 0; iTracklet < 6; iTracklet++) {
	127	AliTRDtrackletGTU trkl = new ((fTracklets)[iTracklet]) AliTRDtrackletGTU();
	128	if (fTrackletMask & (1 << iTracklet)) {
	129	++fNTracklets;
	130	trkl->SetIndex(rhs.GetTrackletIndex(iTracklet));
	131	}
	132	else
	133	trkl->SetIndex(-1);
	134	}
	135	}
	136
	137	return *this;
	138	}
	139
52c19022	140	AliTRDtrackGTU::~AliTRDtrackGTU()
52c19022	141	{
4cc89512	142	// dtor
4cc89512	143
52c19022	144	fTracklets->Delete();
	145	delete fTracklets;
	146	}
	147
5f006bd7	148	void AliTRDtrackGTU::AddTracklet(const AliTRDtrackletGTU * const tracklet, Int_t layer)
52c19022	149	{
4cc89512	150	// add a tracklet to this track
4cc89512	151
52c19022	152	if ( (fTrackletMask & (1 << layer)) != 0 ) {
	153	AliError(Form("Only one tracklet per layer (%i) possible! Mask: 0x%02x", layer, fTrackletMask));
	154	return;
	155	}
	156
	157	new ((fTracklets)[layer]) AliTRDtrackletGTU(tracklet);
	158	fNTracklets++;
	159	fTrackletMask \|= (1 << layer);
	160	}
	161
44eafcf2	162	AliTRDtrackletGTU* AliTRDtrackGTU::GetTracklet(Int_t layer) const
52c19022	163	{
4cc89512	164	// get a pointer to the tracklet in the layer specified
4cc89512	165
80f93426	166	if (IsTrackletInLayer(layer))
80f93426	167	return ((AliTRDtrackletGTU) (fTracklets)[layer]);
5f006bd7	168	else
80f93426	169	return 0x0;
52c19022	170	}
	171
	172	Int_t AliTRDtrackGTU::GetNTracklets() const
	173	{
4cc89512	174	// returns the number of tracklets in this track
4cc89512	175
52c19022	176	return fNTracklets;
	177	}
	178
5f006bd7	179	Bool_t AliTRDtrackGTU::IsTrackletInLayer(Int_t layer) const
52c19022	180	{
4cc89512	181	// checks for a tracklet in the given layer
4cc89512	182
52c19022	183	if ( (GetTrackletMask() & (1 << layer)) != 0)
52c19022	184	return kTRUE;
5f006bd7	185	else
52c19022	186	return kFALSE;
	187	}
	188
5f006bd7	189	void AliTRDtrackGTU::SetFitParams(Float_t a, Float_t b, Float_t c)
52c19022	190	{
4cc89512	191	// set the fit parameters
4cc89512	192
5f006bd7	193	fA = a;
52c19022	194	fB = b;
	195	fC = c;
	196	}
	197
5f006bd7	198	Int_t AliTRDtrackGTU::GetZSubChannel()
52c19022	199	{
4cc89512	200	// returns the z-subchannel
4cc89512	201
52c19022	202	if (fZSubChannel < 0) {
6dbd105b JK	203	for (Int_t layer = 0; layer < AliTRDgtuParam::GetNLayers(); layer++) {
	204	if (IsTrackletInLayer(layer)) {
	205	AliTRDtrackletGTU trkl = (AliTRDtrackletGTU) (*fTracklets)[layer];
	206	if (trkl) {
	207	if ((fZSubChannel > -1) &&
	208	(fZSubChannel != trkl->GetSubChannel(GetZChannel())))
	209	AliError(Form("found inconsistent z-subchannels: track = %i/%i, trkl = %i",
	210	GetZChannel(), fZSubChannel, trkl->GetSubChannel(GetZChannel())));
	211	fZSubChannel = trkl->GetSubChannel(GetZChannel());
	212	}
	213	else {
	214	AliError("no tracklet where one should be according to layer mask");
	215	}
	216	}
52c19022	217	}
	218	}
	219	return fZSubChannel;
	220	}
	221
5f006bd7	222	Int_t AliTRDtrackGTU::GetYapprox()
52c19022	223	{
d2c8b010	224	// returns an approximated y-position for the track
	225	// taken from the projected y-position of the tracklet in the reference layer
	226	// in which the track was found
4cc89512	227
d2c8b010	228	if ((fRefLayerIdx > -1) && (fRefLayerIdx < AliTRDgtuParam::GetNRefLayers()))
	229	return ((AliTRDtrackletGTU) (fTracklets)[AliTRDgtuParam::GetRefLayer(fRefLayerIdx)])->GetYProj();
	230	else
	231	return 0;
52c19022	232	}
	233
	234	AliESDTrdTrack* AliTRDtrackGTU::CreateTrdTrack() const
	235	{
44eafcf2	236	// creates an AliESDTrdTrack to be added to the ESD
	237
	238	AliESDTrdTrack *trk = new AliESDTrdTrack();
	239	trk->SetA((Int_t) fA);
	240	trk->SetLayerMask(fTrackletMask);
	241	trk->SetPID(fPID);
	242	trk->SetB((Int_t) fB);
	243	trk->SetStack(fStack);
	244	trk->SetSector(fSector);
2f1bde31	245	trk->SetLabel(fLabel);
44eafcf2	246
	247	for (Int_t iLayer = 0; iLayer < AliTRDgtuParam::GetNLayers(); iLayer++) {
	248	AliTRDtrackletGTU *trklGTU = GetTracklet(iLayer);
	249	if (trklGTU) {
a10c6981	250	trk->SetTrackletIndex(trklGTU->GetIndex(), iLayer);
44eafcf2	251	AliESDTrdTracklet *trkl = trklGTU->GetTrackletESD();
	252	if (trkl)
	253	trk->AddTrackletReference(trkl, iLayer);
	254	}
	255	}
52cd0cc0	256
44eafcf2	257	return trk;
52c19022	258	}
4cc89512	259
5f006bd7	260	Bool_t AliTRDtrackGTU::CookLabel()
4cc89512	261	{
2f1bde31	262	// assign label from tracklets according to frequency
	263
	264	Int_t nLabels = 0;
3eefbe0b	265	Int_t label[6] = { 0 };
3eefbe0b	266	Int_t count[6] = { 0 };
2f1bde31	267
	268	for (Int_t iTracklet = 0; iTracklet < 6; iTracklet++) {
	269	if (!IsTrackletInLayer(iTracklet))
	270	continue;
	271
	272	Int_t currLabel = GetTracklet(iTracklet)->GetLabel();
	273
	274	Bool_t assigned = kFALSE;
	275	for (Int_t iLabel = 0; iLabel < nLabels; iLabel++) {
	276	if (currLabel == label[iLabel]) {
	277	count[iLabel]++;
	278	assigned = kTRUE;
	279	break;
	280	}
4cc89512	281	}
2f1bde31	282
	283	if (!assigned) {
	284	label[nLabels] = currLabel;
	285	count[nLabels] = 1;
	286	nLabels++;
	287	}
	288	}
	289
6d96036a	290	Int_t index[32];
3eefbe0b	291	TMath::Sort(6, count, index);
3eefbe0b	292	fLabel = label[index[0]];
2f1bde31	293
2f1bde31	294	return kTRUE;
4cc89512	295	}