[u/mrichter/AliRoot.git] / STEER / AliESDfriendTrack.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.                  *
 **************************************************************************/

//-------------------------------------------------------------------------
//               Implementation of the AliESDfriendTrack class
//  This class keeps complementary to the AliESDtrack information 
//      Origin: Iouri Belikov, CERN, Jouri.Belikov@cern.ch
//-------------------------------------------------------------------------
#include "AliTrackPointArray.h"
#include "AliESDfriendTrack.h"
#include "TObjArray.h"
#include "TClonesArray.h"
#include "AliKalmanTrack.h"

ClassImp(AliESDfriendTrack)

AliESDfriendTrack::AliESDfriendTrack(): 
TObject(), 
f1P(0), 
fHmpPhotClus(new TClonesArray("AliHMPIDCluster",1000)),
fnMaxITScluster(0),
fnMaxTPCcluster(0),
fnMaxTRDcluster(0),
fITSindex(0x0),
fTPCindex(0x0),
fTRDindex(0x0),
fPoints(0),
fCalibContainer(0),
fITStrack(0),
fTRDtrack(0),
fTPCOut(0),
fITSOut(0),
fTRDIn(0)
{
  //
  // Default constructor
  //
	//  Int_t i;
  //  fITSindex = new Int_t[fnMaxITScluster];
  //fTPCindex = new Int_t[fnMaxTPCcluster];
  //fTRDindex = new Int_t[fnMaxTRDcluster];
  //for (i=0; i<kMaxITScluster; i++) fITSindex[i]=-2;
  //for (i=0; i<kMaxTPCcluster; i++) fTPCindex[i]=-2;
  //for (i=0; i<kMaxTRDcluster; i++) fTRDindex[i]=-2;
  
  //fHmpPhotClus->SetOwner(kTRUE); 
  
}

AliESDfriendTrack::AliESDfriendTrack(const AliESDfriendTrack &t): 
TObject(t),
f1P(t.f1P),
fHmpPhotClus(t.fHmpPhotClus),        
fnMaxITScluster(t.fnMaxITScluster),
fnMaxTPCcluster(t.fnMaxTPCcluster),
fnMaxTRDcluster(t.fnMaxTRDcluster),
fITSindex(0x0),
fTPCindex(0x0),
fTRDindex(0x0),
fPoints(0),
fCalibContainer(0),
fITStrack(0),
fTRDtrack(0),
fTPCOut(0),
fITSOut(0),
fTRDIn(0)
{
  //
  // Copy constructor
  //
  AliDebug(2,"Calling copy constructor");

  Int_t i;
  if (fnMaxITScluster != 0){
	  fITSindex = new Int_t[fnMaxITScluster];
	  for (i=0; i<fnMaxITScluster; i++) fITSindex[i]=t.fITSindex[i];
  }
  if (fnMaxTPCcluster != 0){
	  fTPCindex = new Int_t[fnMaxTPCcluster];
	  for (i=0; i<fnMaxTPCcluster; i++) fTPCindex[i]=t.fTPCindex[i];
  }
  if (fnMaxTRDcluster != 0){
	  fTRDindex = new Int_t[fnMaxTRDcluster];
	  for (i=0; i<fnMaxTRDcluster; i++) fTRDindex[i]=t.fTRDindex[i]; 
  }
  AliDebug(2,Form("fnMaxITScluster = %d",fnMaxITScluster));
  AliDebug(2,Form("fnMaxTPCcluster = %d",fnMaxTPCcluster));
  AliDebug(2,Form("fnMaxTRDcluster = %d",fnMaxTRDcluster));
  if (t.fPoints) fPoints=new AliTrackPointArray(*t.fPoints);
  if (t.fCalibContainer) {
     fCalibContainer = new TObjArray(5);
     Int_t no=t.fCalibContainer->GetEntriesFast();
     for (i=0; i<no; i++) {
       TObject *o=t.fCalibContainer->At(i);
       fCalibContainer->AddLast(o->Clone());
     }  
  }

  if (t.fTPCOut) fTPCOut = new AliExternalTrackParam(*(t.fTPCOut));
  if (t.fITSOut) fITSOut = new AliExternalTrackParam(*(t.fITSOut));
  if (t.fTRDIn)  fTRDIn = new AliExternalTrackParam(*(t.fTRDIn));

  fHmpPhotClus = new TClonesArray(*t.fHmpPhotClus);  
  
}

AliESDfriendTrack::~AliESDfriendTrack() {
  //
  // Simple destructor
  //
   delete fPoints;
   if (fCalibContainer) fCalibContainer->Delete();
   delete fCalibContainer;
   delete fHmpPhotClus;   
   delete fITStrack;
   delete fTRDtrack;
   delete fTPCOut;
   delete fITSOut;
   delete fTRDIn;
   delete[] fITSindex;
   delete[] fTPCindex;
   delete[] fTRDindex;
}


void AliESDfriendTrack::AddCalibObject(TObject * calibObject){
  //
  // add calibration object to array -
  // track is owner of the objects in the container 
  //
  if (!fCalibContainer) fCalibContainer = new TObjArray(5);
  fCalibContainer->AddLast(calibObject);
}

TObject * AliESDfriendTrack::GetCalibObject(Int_t index){
  //
  //
  //
  if (!fCalibContainer) return 0;
  if (index>=fCalibContainer->GetEntriesFast()) return 0;
  return fCalibContainer->At(index);
}


void AliESDfriendTrack::SetTPCOut(const AliExternalTrackParam &param) {
  // 
  // backup TPC out track
  //
  delete fTPCOut;
  fTPCOut=new AliExternalTrackParam(param);
} 
void AliESDfriendTrack::SetITSOut(const AliExternalTrackParam &param) {
  //
  // backup ITS out track
  //
  delete fITSOut;
  fITSOut=new AliExternalTrackParam(param);
} 
void AliESDfriendTrack::SetTRDIn(const AliExternalTrackParam  &param)  {
  //
  // backup TRD in track
  //
  delete fTRDIn;
  fTRDIn=new AliExternalTrackParam(param);
} 


void AliESDfriendTrack::SetHmpPhotClus(TClonesArray  *array)  {
  //
  // 
  //
  fHmpPhotClus->Clear("C");
  for(Int_t iClu=0; iClu<array->GetEntriesFast(); iClu++){
     TObject *pClu = (TObject*)array->UncheckedAt(iClu);
     new ((*fHmpPhotClus)[iClu]) TObject(*pClu);
  }
}

void AliESDfriendTrack::SetITSIndices(Int_t* indices, Int_t n){

	//
	// setting fITSindex
	// instantiating the pointer if still NULL
	//

	fnMaxITScluster = n;
	AliDebug(2,Form("fnMaxITScluster = %d",fnMaxITScluster));
	if (fITSindex == 0x0){
		fITSindex = new Int_t[fnMaxITScluster];
	}
	for (Int_t i = 0; i < fnMaxITScluster; i++){
		fITSindex[i] = indices[i];
	}
}

void AliESDfriendTrack::SetTPCIndices(Int_t* indices, Int_t n){

	//
	// setting fTPCindex
	// instantiating the pointer if still NULL
	//

	fnMaxTPCcluster = n;
	AliDebug(2,Form("fnMaxTPCcluster = %d",fnMaxTPCcluster));
	if (fTPCindex == 0x0){
		fTPCindex = new Int_t[fnMaxTPCcluster];
	}
	for (Int_t i = 0; i < fnMaxTPCcluster; i++){
		fTPCindex[i] = indices[i];
	}
}

void AliESDfriendTrack::SetTRDIndices(Int_t* indices, Int_t n){

	//
	// setting fTRDindex
	// instantiating the pointer if still NULL
	//

	fnMaxTRDcluster = n;
	AliDebug(2,Form("fnMaxTRDcluster = %d",fnMaxTRDcluster));
	if (fTRDindex == 0x0){
		fTRDindex = new Int_t[fnMaxTRDcluster];
	}
	for (Int_t i = 0; i < fnMaxTRDcluster; i++){
		fTRDindex[i] = indices[i];
	}
}
Commit	Line	Data
1d99986f	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	// Implementation of the AliESDfriendTrack class
	18	// This class keeps complementary to the AliESDtrack information
	19	// Origin: Iouri Belikov, CERN, Jouri.Belikov@cern.ch
	20	//-------------------------------------------------------------------------
	21	#include "AliTrackPointArray.h"
	22	#include "AliESDfriendTrack.h"
00dce61a	23	#include "TObjArray.h"
5637918c	24	#include "TClonesArray.h"
be713d4d	25	#include "AliKalmanTrack.h"
1d99986f	26
	27	ClassImp(AliESDfriendTrack)
	28
15e85efa	29	AliESDfriendTrack::AliESDfriendTrack():
	30	TObject(),
	31	f1P(0),
5637918c	32	fHmpPhotClus(new TClonesArray("AliHMPIDCluster",1000)),
6d3a7bbf	33	fnMaxITScluster(0),
	34	fnMaxTPCcluster(0),
	35	fnMaxTRDcluster(0),
	36	fITSindex(0x0),
	37	fTPCindex(0x0),
	38	fTRDindex(0x0),
00dce61a	39	fPoints(0),
00dce61a	40	fCalibContainer(0),
15e85efa	41	fITStrack(0),
ded25cc6	42	fTRDtrack(0),
	43	fTPCOut(0),
	44	fITSOut(0),
	45	fTRDIn(0)
15e85efa	46	{
1d99986f	47	//
	48	// Default constructor
	49	//
6d3a7bbf	50	// Int_t i;
	51	// fITSindex = new Int_t[fnMaxITScluster];
	52	//fTPCindex = new Int_t[fnMaxTPCcluster];
	53	//fTRDindex = new Int_t[fnMaxTRDcluster];
	54	//for (i=0; i<kMaxITScluster; i++) fITSindex[i]=-2;
	55	//for (i=0; i<kMaxTPCcluster; i++) fTPCindex[i]=-2;
	56	//for (i=0; i<kMaxTRDcluster; i++) fTRDindex[i]=-2;
5637918c	57
	58	//fHmpPhotClus->SetOwner(kTRUE);
	59
1d99986f	60	}
	61
	62	AliESDfriendTrack::AliESDfriendTrack(const AliESDfriendTrack &t):
	63	TObject(t),
	64	f1P(t.f1P),
5637918c	65	fHmpPhotClus(t.fHmpPhotClus),
6d3a7bbf	66	fnMaxITScluster(t.fnMaxITScluster),
	67	fnMaxTPCcluster(t.fnMaxTPCcluster),
	68	fnMaxTRDcluster(t.fnMaxTRDcluster),
	69	fITSindex(0x0),
	70	fTPCindex(0x0),
	71	fTRDindex(0x0),
15e85efa	72	fPoints(0),
00dce61a	73	fCalibContainer(0),
15e85efa	74	fITStrack(0),
ded25cc6	75	fTRDtrack(0),
	76	fTPCOut(0),
	77	fITSOut(0),
	78	fTRDIn(0)
1d99986f	79	{
	80	//
	81	// Copy constructor
	82	//
6d3a7bbf	83	AliDebug(2,"Calling copy constructor");
6d3a7bbf	84
1d99986f	85	Int_t i;
6d3a7bbf	86	if (fnMaxITScluster != 0){
	87	fITSindex = new Int_t[fnMaxITScluster];
	88	for (i=0; i<fnMaxITScluster; i++) fITSindex[i]=t.fITSindex[i];
	89	}
	90	if (fnMaxTPCcluster != 0){
	91	fTPCindex = new Int_t[fnMaxTPCcluster];
	92	for (i=0; i<fnMaxTPCcluster; i++) fTPCindex[i]=t.fTPCindex[i];
	93	}
	94	if (fnMaxTRDcluster != 0){
	95	fTRDindex = new Int_t[fnMaxTRDcluster];
	96	for (i=0; i<fnMaxTRDcluster; i++) fTRDindex[i]=t.fTRDindex[i];
	97	}
	98	AliDebug(2,Form("fnMaxITScluster = %d",fnMaxITScluster));
	99	AliDebug(2,Form("fnMaxTPCcluster = %d",fnMaxTPCcluster));
	100	AliDebug(2,Form("fnMaxTRDcluster = %d",fnMaxTRDcluster));
1d99986f	101	if (t.fPoints) fPoints=new AliTrackPointArray(*t.fPoints);
f333ed51	102	if (t.fCalibContainer) {
	103	fCalibContainer = new TObjArray(5);
	104	Int_t no=t.fCalibContainer->GetEntriesFast();
6c27b212	105	for (i=0; i<no; i++) {
f333ed51	106	TObject *o=t.fCalibContainer->At(i);
	107	fCalibContainer->AddLast(o->Clone());
	108	}
	109	}
ded25cc6	110
	111	if (t.fTPCOut) fTPCOut = new AliExternalTrackParam(*(t.fTPCOut));
	112	if (t.fITSOut) fITSOut = new AliExternalTrackParam(*(t.fITSOut));
	113	if (t.fTRDIn) fTRDIn = new AliExternalTrackParam(*(t.fTRDIn));
6d3a7bbf	114
5637918c	115	fHmpPhotClus = new TClonesArray(*t.fHmpPhotClus);
5637918c	116
1d99986f	117	}
1d99986f	118
15e85efa	119	AliESDfriendTrack::~AliESDfriendTrack() {
1d99986f	120	//
15e85efa	121	// Simple destructor
1d99986f	122	//
15e85efa	123	delete fPoints;
f333ed51	124	if (fCalibContainer) fCalibContainer->Delete();
2c56c9f6	125	delete fCalibContainer;
5637918c	126	delete fHmpPhotClus;
15e85efa	127	delete fITStrack;
15e85efa	128	delete fTRDtrack;
ded25cc6	129	delete fTPCOut;
	130	delete fITSOut;
	131	delete fTRDIn;
82c82b62	132	delete[] fITSindex;
	133	delete[] fTPCindex;
	134	delete[] fTRDindex;
1d99986f	135	}
00dce61a	136
	137
	138	void AliESDfriendTrack::AddCalibObject(TObject * calibObject){
	139	//
	140	// add calibration object to array -
	141	// track is owner of the objects in the container
	142	//
	143	if (!fCalibContainer) fCalibContainer = new TObjArray(5);
	144	fCalibContainer->AddLast(calibObject);
	145	}
	146
	147	TObject * AliESDfriendTrack::GetCalibObject(Int_t index){
	148	//
	149	//
	150	//
	151	if (!fCalibContainer) return 0;
	152	if (index>=fCalibContainer->GetEntriesFast()) return 0;
	153	return fCalibContainer->At(index);
	154	}
ded25cc6	155
	156
	157	void AliESDfriendTrack::SetTPCOut(const AliExternalTrackParam &param) {
	158	//
	159	// backup TPC out track
	160	//
	161	delete fTPCOut;
	162	fTPCOut=new AliExternalTrackParam(param);
	163	}
	164	void AliESDfriendTrack::SetITSOut(const AliExternalTrackParam &param) {
	165	//
	166	// backup ITS out track
	167	//
	168	delete fITSOut;
	169	fITSOut=new AliExternalTrackParam(param);
	170	}
	171	void AliESDfriendTrack::SetTRDIn(const AliExternalTrackParam &param) {
	172	//
	173	// backup TRD in track
	174	//
	175	delete fTRDIn;
	176	fTRDIn=new AliExternalTrackParam(param);
	177	}
	178
5637918c	179
	180	void AliESDfriendTrack::SetHmpPhotClus(TClonesArray *array) {
	181	//
	182	//
	183	//
	184	fHmpPhotClus->Clear("C");
	185	for(Int_t iClu=0; iClu<array->GetEntriesFast(); iClu++){
	186	TObject pClu = (TObject)array->UncheckedAt(iClu);
	187	new ((fHmpPhotClus)[iClu]) TObject(pClu);
	188	}
	189	}
	190
6d3a7bbf	191	void AliESDfriendTrack::SetITSIndices(Int_t* indices, Int_t n){
	192
	193	//
	194	// setting fITSindex
	195	// instantiating the pointer if still NULL
	196	//
	197
	198	fnMaxITScluster = n;
	199	AliDebug(2,Form("fnMaxITScluster = %d",fnMaxITScluster));
	200	if (fITSindex == 0x0){
	201	fITSindex = new Int_t[fnMaxITScluster];
	202	}
	203	for (Int_t i = 0; i < fnMaxITScluster; i++){
	204	fITSindex[i] = indices[i];
	205	}
	206	}
	207
	208	void AliESDfriendTrack::SetTPCIndices(Int_t* indices, Int_t n){
	209
	210	//
	211	// setting fTPCindex
	212	// instantiating the pointer if still NULL
	213	//
	214
	215	fnMaxTPCcluster = n;
	216	AliDebug(2,Form("fnMaxTPCcluster = %d",fnMaxTPCcluster));
	217	if (fTPCindex == 0x0){
	218	fTPCindex = new Int_t[fnMaxTPCcluster];
	219	}
	220	for (Int_t i = 0; i < fnMaxTPCcluster; i++){
	221	fTPCindex[i] = indices[i];
	222	}
	223	}
	224
	225	void AliESDfriendTrack::SetTRDIndices(Int_t* indices, Int_t n){
	226
	227	//
	228	// setting fTRDindex
	229	// instantiating the pointer if still NULL
	230	//
	231
	232	fnMaxTRDcluster = n;
	233	AliDebug(2,Form("fnMaxTRDcluster = %d",fnMaxTRDcluster));
	234	if (fTRDindex == 0x0){
	235	fTRDindex = new Int_t[fnMaxTRDcluster];
	236	}
	237	for (Int_t i = 0; i < fnMaxTRDcluster; i++){
	238	fTRDindex[i] = indices[i];
	239	}
	240	}
	241