[u/mrichter/AliRoot.git] / HLT / src / AliL3TrackArray.cxx

//$Id$

// Author: Uli Frankenfeld <mailto:franken@fi.uib.no>
//*-- Copyright &copy ULI

#include <math.h>
#include <string.h>
#include <iostream.h>
#include "AliL3Logging.h"
#include "AliL3TrackArray.h"
#include "AliL3HoughTrack.h"
#include "AliL3ModelTrack.h"
#include "AliL3ConfMapTrack.h"
#include "AliL3TrackSegmentData.h"
#include "AliL3Transform.h"
#include "AliL3ConfMapPoint.h"

//_____________________________________________________________
// AliL3TrackArray
//
// Track array class 
//

ClassImp(AliL3TrackArray)

AliL3TrackArray::AliL3TrackArray(){
  //Default constructor
  fSize = 0;
  fNTracks=0;
  fNAbsent=0;
  fTrackType='t';
  SetSize();
}


AliL3TrackArray::AliL3TrackArray(Int_t ntrack){
  //Constructor.
  fSize = 0;
  fNTracks=0;
  fNAbsent=0;
  fTrackType='t';
  SetSize(ntrack);
}

AliL3TrackArray::AliL3TrackArray(char* tracktype,Int_t ntrack){
  //Constructor.
  fSize = 0;
  fNTracks=0;
  fNAbsent=0;
  if(strcmp(tracktype,"AliL3Track")==0) fTrackType='t';
  if(strcmp(tracktype,"AliL3ConfMapTrack")==0) fTrackType='c';
  if(strcmp(tracktype,"AliL3HoughTrack")==0) fTrackType='h';
  if(strcmp(tracktype,"AliL3ModelTrack")==0) fTrackType='m';
  SetSize(ntrack);
}

AliL3TrackArray::AliL3TrackArray(char* tracktype){
  //Constructor.
  fSize = 0;
  fNTracks=0;
  fNAbsent=0;
  if(strcmp(tracktype,"AliL3Track")==0) fTrackType='t';
  if(strcmp(tracktype,"AliL3ConfMapTrack")==0) fTrackType='c';
  if(strcmp(tracktype,"AliL3HoughTrack")==0) fTrackType='h';
  if(strcmp(tracktype,"AliL3ModelTrack")==0) fTrackType='m';
  SetSize();
}


AliL3TrackArray::~AliL3TrackArray(){
  //Destructor
  DeleteArray();
}


AliL3Track *AliL3TrackArray::NextTrack(){
  if(fNTracks<fSize) return fTrack[fNTracks++];
  SetSize(fSize+100);
   return fTrack[fNTracks++]; 
}

void AliL3TrackArray::DeleteArray(){
  for(Int_t i=0; i<fSize;i++)
    delete fTrack[i];
  delete[] fIsPresent;
  delete[] fTrack;
}

Bool_t AliL3TrackArray::SetSize(Int_t newsize){
  if(newsize<=fSize) return kFALSE; //shrink comes later!! 
  if(!fSize){
    fSize = newsize;
    fTrack = new AliL3Track*[fSize];
    fIsPresent = new Bool_t[fSize];
    switch(fTrackType){
      case 't':
        for(Int_t i=0;i<fSize;i++){
          fTrack[i]   = new AliL3Track();
          fIsPresent[i] = kTRUE;
        }
        break;
      case 'c':  
        for(Int_t i=0;i<fSize;i++){
          fTrack[i]   = new AliL3ConfMapTrack();
          fIsPresent[i] = kTRUE;
        }
        break;
      case 'h':
        for(Int_t i=0;i<fSize;i++){
          fTrack[i]   = new AliL3HoughTrack();
          fIsPresent[i] = kTRUE;
        }
        break;
       case 'm':
        for(Int_t i=0;i<fSize;i++){
          fTrack[i]   = new AliL3ModelTrack();
          fIsPresent[i] = kTRUE;
        }
        break;
      default: 
        return kFALSE;
    }
    return kTRUE;
  }
  AliL3Track **tmp = new AliL3Track*[fSize];
  Bool_t *pre = new Bool_t[fSize];
  for(Int_t i=0; i<fSize;i++){
    tmp[i] = fTrack[i];
    pre[i] = fIsPresent[i];
  }
  delete[]  fTrack;
  delete[] fIsPresent;
  fTrack =  new AliL3Track*[newsize];
  fIsPresent = new Bool_t[newsize];
  for(Int_t i=0; i<fSize;i++){
    fTrack[i]   = tmp[i];
    fIsPresent[i] = pre[i];
  }
  delete[] tmp;
  delete[] pre;
  switch(fTrackType){
    case 't':
      for(Int_t i=fSize;i<newsize;i++){
        fTrack[i]   = new AliL3Track();
        fIsPresent[i] = kTRUE;
      }
      break;
    case 'c':  
      for(Int_t i=fSize;i<newsize;i++){
        fTrack[i]   = new AliL3ConfMapTrack();
        fIsPresent[i] = kTRUE;
      }
      break;
    case 'h':
      for(Int_t i=fSize;i<newsize;i++){
        fTrack[i]   = new AliL3HoughTrack();
        fIsPresent[i] = kTRUE;
      }
      break;
    case 'm':
      for(Int_t i=fSize;i<newsize;i++){
        fTrack[i]   = new AliL3ModelTrack();
        fIsPresent[i] = kTRUE;
      }
      break;
    default: 
      return kFALSE;
  }
  fSize = newsize;
  return kTRUE;
}

void AliL3TrackArray::Reset(){
  fNTracks=0;
  fNAbsent=0;
  for(Int_t i=0; i<fSize;i++)
    fIsPresent[i] = kTRUE; 
}

void AliL3TrackArray::Remove(Int_t track){
  if(fIsPresent[track]){
    fIsPresent[track]=kFALSE;
    fNAbsent++;
  }
}

void AliL3TrackArray::FillTracks(Int_t ntracks, AliL3TrackSegmentData* tr){
  //Read tracks from shared memory (or memory)
  AliL3TrackSegmentData *trs = tr;
   for(Int_t i=0; i<ntracks; i++){
    AliL3Track *track = NextTrack(); 
    track->SetPt(trs->fPt);
    track->SetPsi(trs->fPsi);
    track->SetTgl(trs->fTgl);
    track->SetNHits(trs->fNPoints);
    track->SetCharge(trs->fCharge);
    track->SetFirstPoint(trs->fX,trs->fY,trs->fZ);
    track->SetLastPoint(trs->fLastX,trs->fLastY,trs->fLastZ);
    track->SetHits( trs->fNPoints, trs->fPointIDs );
    UChar_t *tmpP = (UChar_t*)trs;
    tmpP += sizeof(AliL3TrackSegmentData)+trs->fNPoints*sizeof(UInt_t);
    trs = (AliL3TrackSegmentData*)tmpP;
  }
}

void AliL3TrackArray::FillTracks(Int_t ntracks, AliL3TrackSegmentData* tr,Int_t slice, AliL3Transform* trans){
  //Read tracks from shared memory (or memory)
  AliL3TrackSegmentData *trs = tr;
  for(Int_t i=0; i<ntracks; i++){
    AliL3Track *track = NextTrack(); 
    track->SetPt(trs->fPt);
    Float_t psi[1];
    psi[0]=trs->fPsi;
    trans->Local2GlobalAngle(psi,slice);
    track->SetPsi(psi[0]);
    track->SetTgl(trs->fTgl);
    track->SetNHits(trs->fNPoints);
    track->SetCharge(trs->fCharge);
    Float_t first[3];
    first[0]=trs->fX;first[1]=trs->fY;first[2]=trs->fZ;
    trans->Local2Global(first,slice);
    track->SetFirstPoint(first[0],first[1],first[2]);
    Float_t last[3];
    last[0]=trs->fLastX;last[1]=trs->fLastY;last[2]=trs->fLastZ;
    trans->Local2Global(last,slice);
    track->SetLastPoint(last[0],last[1],last[2]);
    track->SetHits( trs->fNPoints, trs->fPointIDs );
    UChar_t *tmpP = (UChar_t*)trs;
    tmpP += sizeof(AliL3TrackSegmentData)+trs->fNPoints*sizeof(UInt_t);
    trs = (AliL3TrackSegmentData*)tmpP;
  }
}

UInt_t AliL3TrackArray::GetOutSize(){
  UInt_t count = GetOutCount();   //use only present tracks
  UInt_t tHits = 0;
  for(Int_t i=0;i<fNTracks;i++){  //loop over all tracks
    AliL3Track *track = GetCheckedTrack(i);  //use only present tracks
    if(track)                                       //use only present tracks
      tHits += track->GetNHits();
  }

  //calculate size of track
  return count*sizeof(AliL3TrackSegmentData)+sizeof(UInt_t)*tHits;
}

UInt_t AliL3TrackArray::WriteTracks(UInt_t & ntracks,AliL3TrackSegmentData* tr){
  ntracks = GetOutCount();
  return WriteTracks(tr);
}

UInt_t AliL3TrackArray::WriteTracks(AliL3TrackSegmentData* tr){
  if(GetTrackType()=='c') return WriteConfMapTracks(tr);
  AliL3TrackSegmentData *tP = tr;
  UInt_t *pP;
  UInt_t size = 0;
  for(Int_t i=0; i<fNTracks; i++){  //loop over all tracks
    AliL3Track *track = GetCheckedTrack(i); //use only present tracks
    if(!track) continue;                           //use only present tracks
    tP->fX = track->GetFirstPointX();
    tP->fY = track->GetFirstPointY();
    tP->fZ = track->GetFirstPointZ();
    tP->fPt = track->GetPt();
    tP->fLastX = track->GetLastPointX();
    tP->fLastY = track->GetLastPointY();
    tP->fLastZ = track->GetLastPointZ();
    tP->fPsi = track->GetPsi();
    tP->fTgl = track->GetTgl();
    tP->fCharge = track->GetCharge();
    tP->fNPoints = track->GetNHits();
    pP = (UInt_t*)track->GetHitNumbers();
    for (UInt_t j=0;j<tP->fNPoints;j++){
      tP->fPointIDs[j] = pP[j];
    }
    Byte_t *tmpP = (Byte_t *)tP;
    tmpP += sizeof(AliL3TrackSegmentData)+tP->fNPoints*sizeof(UInt_t);
    size += sizeof(AliL3TrackSegmentData)+tP->fNPoints*sizeof(UInt_t);
    tP = (AliL3TrackSegmentData*)tmpP;
  }
  return size;
}

UInt_t AliL3TrackArray::WriteConfMapTracks(AliL3TrackSegmentData* tr){
  // use first and last point objects
  AliL3TrackSegmentData *tP = tr;
  UInt_t *pP;
  UInt_t size = 0;
  for(Int_t i=0; i<fNTracks; i++){  //loop over all tracks
    AliL3ConfMapTrack *track =(AliL3ConfMapTrack *) GetCheckedTrack(i); //use only present tracks
    if(!track) continue;                           //use only present tracks
    AliL3ConfMapPoint *hit = (AliL3ConfMapPoint*)track->lastHit;
    AliL3ConfMapPoint *lastHit = (AliL3ConfMapPoint*)track->firstHit;
    tP->fX = hit->GetX();
    tP->fY = hit->GetY();
    tP->fZ = hit->GetZ();
    tP->fLastX = lastHit->GetX();
    tP->fLastY = lastHit->GetY();
    tP->fLastZ = lastHit->GetZ();
   
//    tP->fX = track->GetFirstPointX();
//    tP->fY = track->GetFirstPointY();
//    tP->fZ = track->GetFirstPointZ();
    tP->fPt = track->GetPt();
//    tP->fLastX = track->GetLastPointX();
//    tP->fLastY = track->GetLastPointY();
//    tP->fLastZ = track->GetLastPointZ();
    tP->fPsi = track->GetPsi();
    tP->fTgl = track->GetTgl();
    tP->fCharge = track->GetCharge();
    tP->fNPoints = track->GetNHits();
    pP = (UInt_t*)track->GetHitNumbers();
    for (UInt_t j=0;j<tP->fNPoints;j++){
      tP->fPointIDs[j] = pP[j];
    }
    Byte_t *tmpP = (Byte_t *)tP;
    tmpP += sizeof(AliL3TrackSegmentData)+tP->fNPoints*sizeof(UInt_t);
    size +=sizeof(AliL3TrackSegmentData)+tP->fNPoints*sizeof(UInt_t);
    tP = (AliL3TrackSegmentData*)tmpP;
  }
  return size;
}

void AliL3TrackArray::AddLast(AliL3Track *track)
{
  AliL3Track *tpt = NextTrack();
  tpt->Set(track);
  
}


void AliL3TrackArray::AddTracks(AliL3TrackArray *newtrack,Bool_t remove_old,Int_t slice)
{
  if(GetTrackType() != newtrack->GetTrackType() && GetTrackType()!='t')
    {
      LOG(AliL3Log::kError,"AliL3TrackArray::AddTracks","Track types")
	<<"Bad idea to add tracks of different types"<<ENDLOG;
      return;
    }
  if(fSize < fNTracks+newtrack->GetNPresent())
    SetSize(fSize+newtrack->GetSize());
  for(Int_t i =0;i<newtrack->GetNTracks();i++){
    AliL3Track *tpt = newtrack->GetCheckedTrack(i);
    if(!tpt) continue;
    if(remove_old)
      newtrack->Remove(i);
    AliL3Track *track = NextTrack();
    track->Set(tpt);
    if(slice>=0)
      track->Rotate(slice); //Rotate track to global coordinates
    /*
      AliL3Track *track;
      if(GetTrackType()=='h')
      track = (AliL3HoughTrack*)NextTrack();
      else
      track = NextTrack();
      track->Set(tpt);
    */
  }
}


void AliL3TrackArray::Compress(){
  if(GetNPresent()==GetNTracks()) return;
  AliL3Track **tmp =  new AliL3Track *[fNTracks];
  Int_t present=0;
  Int_t absent=GetNPresent();
  for(Int_t i=0;i<GetNTracks();i++){
    if(fIsPresent[i]) tmp[present++] = fTrack[i];
    else tmp[absent++] = fTrack[i];
  }
  for(Int_t i=0;i<GetNTracks();i++)
    fIsPresent[i]=kTRUE;

  //Copy pointers back
  for(Int_t i=0; i<GetNTracks();i++){
    fTrack[i]=tmp[i];
  }

  delete[] tmp;

  fNTracks = GetNPresent();
  fNAbsent = 0;
}

void AliL3TrackArray::QSort(){
  // compress an sort
  Compress();
  QSort(fTrack,0,fNTracks);
}

void AliL3TrackArray::QSort( AliL3Track **a, Int_t first, Int_t last){

   // Sort array of AliL3Track pointers using a quicksort algorithm.
   // Uses TrackCompare() to compare objects.
   // Thanks to Root! 

   static AliL3Track *tmp;
   static int i;           // "static" to save stack space
   int j;

   while (last - first > 1) {
      i = first;
      j = last;
      for (;;) {
	while (++i < last && TrackCompare(a[i], a[first]) < 0)
	  ;
	while (--j > first && TrackCompare(a[j], a[first]) > 0)
	  ;
         if (i >= j)
            break;

         tmp  = a[i];
         a[i] = a[j];
         a[j] = tmp;
      }
      if (j == first) {
         ++first;
         continue;
      }
      tmp = a[first];
      a[first] = a[j];
      a[j] = tmp;
      if (j - first < last - (j + 1)) {
         QSort(a, first, j);
         first = j + 1;   // QSort(j + 1, last);
      } else {
         QSort(a, j + 1, last);
         last = j;        // QSort(first, j);
      }
   }
}

Int_t AliL3TrackArray::TrackCompare(AliL3Track *a, AliL3Track *b){
   // Compare the two tracks.
  
  return b->Compare(a);
  
  /*
    if(fTrackType=='h')
    {
    AliL3HoughTrack *tra = (AliL3HoughTrack*)a;
    AliL3HoughTrack *trb = (AliL3HoughTrack*)b;
    if(tra->GetWeight() < trb->GetWeight()) return 1;
    if(tra->GetWeight() > trb->GetWeight()) return -1;
    }
    else
    {
    if(a->GetNHits() < b->GetNHits()) return 1;
    if(a->GetNHits() > b->GetNHits()) return -1;
    }
    
    return 0;
  */
}
Commit	Line	Data
95a00d93	1	//$Id$
95a00d93	2
71df269a	3	// Author: Uli Frankenfeld <mailto:franken@fi.uib.no>
71df269a	4	//*-- Copyright &copy ULI
108615fc	5
	6	#include <math.h>
	7	#include <string.h>
	8	#include <iostream.h>
	9	#include "AliL3Logging.h"
	10	#include "AliL3TrackArray.h"
	11	#include "AliL3HoughTrack.h"
95a00d93	12	#include "AliL3ModelTrack.h"
108615fc	13	#include "AliL3ConfMapTrack.h"
	14	#include "AliL3TrackSegmentData.h"
	15	#include "AliL3Transform.h"
	16	#include "AliL3ConfMapPoint.h"
71df269a	17
108615fc	18	//_____________________________________________________________
71df269a	19	// AliL3TrackArray
108615fc	20	//
71df269a	21	// Track array class
108615fc	22	//
	23
	24	ClassImp(AliL3TrackArray)
	25
	26	AliL3TrackArray::AliL3TrackArray(){
	27	//Default constructor
	28	fSize = 0;
	29	fNTracks=0;
	30	fNAbsent=0;
	31	fTrackType='t';
	32	SetSize();
	33	}
	34
	35
	36	AliL3TrackArray::AliL3TrackArray(Int_t ntrack){
	37	//Constructor.
	38	fSize = 0;
	39	fNTracks=0;
	40	fNAbsent=0;
	41	fTrackType='t';
	42	SetSize(ntrack);
	43	}
	44
	45	AliL3TrackArray::AliL3TrackArray(char* tracktype,Int_t ntrack){
	46	//Constructor.
	47	fSize = 0;
	48	fNTracks=0;
	49	fNAbsent=0;
	50	if(strcmp(tracktype,"AliL3Track")==0) fTrackType='t';
	51	if(strcmp(tracktype,"AliL3ConfMapTrack")==0) fTrackType='c';
	52	if(strcmp(tracktype,"AliL3HoughTrack")==0) fTrackType='h';
95a00d93	53	if(strcmp(tracktype,"AliL3ModelTrack")==0) fTrackType='m';
108615fc	54	SetSize(ntrack);
	55	}
	56
	57	AliL3TrackArray::AliL3TrackArray(char* tracktype){
	58	//Constructor.
	59	fSize = 0;
	60	fNTracks=0;
	61	fNAbsent=0;
	62	if(strcmp(tracktype,"AliL3Track")==0) fTrackType='t';
	63	if(strcmp(tracktype,"AliL3ConfMapTrack")==0) fTrackType='c';
	64	if(strcmp(tracktype,"AliL3HoughTrack")==0) fTrackType='h';
95a00d93	65	if(strcmp(tracktype,"AliL3ModelTrack")==0) fTrackType='m';
108615fc	66	SetSize();
	67	}
	68
	69
	70	AliL3TrackArray::~AliL3TrackArray(){
	71	//Destructor
	72	DeleteArray();
	73	}
	74
	75
	76	AliL3Track *AliL3TrackArray::NextTrack(){
	77	if(fNTracks<fSize) return fTrack[fNTracks++];
	78	SetSize(fSize+100);
	79	return fTrack[fNTracks++];
	80	}
	81
	82	void AliL3TrackArray::DeleteArray(){
	83	for(Int_t i=0; i<fSize;i++)
	84	delete fTrack[i];
	85	delete[] fIsPresent;
	86	delete[] fTrack;
	87	}
	88
	89	Bool_t AliL3TrackArray::SetSize(Int_t newsize){
	90	if(newsize<=fSize) return kFALSE; //shrink comes later!!
	91	if(!fSize){
	92	fSize = newsize;
	93	fTrack = new AliL3Track*[fSize];
	94	fIsPresent = new Bool_t[fSize];
	95	switch(fTrackType){
	96	case 't':
	97	for(Int_t i=0;i<fSize;i++){
	98	fTrack[i] = new AliL3Track();
	99	fIsPresent[i] = kTRUE;
	100	}
	101	break;
	102	case 'c':
	103	for(Int_t i=0;i<fSize;i++){
	104	fTrack[i] = new AliL3ConfMapTrack();
	105	fIsPresent[i] = kTRUE;
	106	}
	107	break;
	108	case 'h':
	109	for(Int_t i=0;i<fSize;i++){
	110	fTrack[i] = new AliL3HoughTrack();
	111	fIsPresent[i] = kTRUE;
	112	}
	113	break;
95a00d93	114	case 'm':
	115	for(Int_t i=0;i<fSize;i++){
	116	fTrack[i] = new AliL3ModelTrack();
	117	fIsPresent[i] = kTRUE;
	118	}
	119	break;
108615fc	120	default:
	121	return kFALSE;
	122	}
	123	return kTRUE;
	124	}
	125	AliL3Track *tmp = new AliL3Track[fSize];
	126	Bool_t *pre = new Bool_t[fSize];
	127	for(Int_t i=0; i<fSize;i++){
	128	tmp[i] = fTrack[i];
	129	pre[i] = fIsPresent[i];
	130	}
	131	delete[] fTrack;
	132	delete[] fIsPresent;
12d59a2a	133	fTrack = new AliL3Track*[newsize];
108615fc	134	fIsPresent = new Bool_t[newsize];
	135	for(Int_t i=0; i<fSize;i++){
	136	fTrack[i] = tmp[i];
	137	fIsPresent[i] = pre[i];
	138	}
	139	delete[] tmp;
	140	delete[] pre;
	141	switch(fTrackType){
	142	case 't':
	143	for(Int_t i=fSize;i<newsize;i++){
	144	fTrack[i] = new AliL3Track();
	145	fIsPresent[i] = kTRUE;
	146	}
	147	break;
	148	case 'c':
	149	for(Int_t i=fSize;i<newsize;i++){
	150	fTrack[i] = new AliL3ConfMapTrack();
	151	fIsPresent[i] = kTRUE;
	152	}
	153	break;
	154	case 'h':
	155	for(Int_t i=fSize;i<newsize;i++){
	156	fTrack[i] = new AliL3HoughTrack();
	157	fIsPresent[i] = kTRUE;
	158	}
	159	break;
95a00d93	160	case 'm':
	161	for(Int_t i=fSize;i<newsize;i++){
	162	fTrack[i] = new AliL3ModelTrack();
	163	fIsPresent[i] = kTRUE;
	164	}
	165	break;
108615fc	166	default:
	167	return kFALSE;
	168	}
	169	fSize = newsize;
	170	return kTRUE;
	171	}
	172
	173	void AliL3TrackArray::Reset(){
	174	fNTracks=0;
	175	fNAbsent=0;
	176	for(Int_t i=0; i<fSize;i++)
	177	fIsPresent[i] = kTRUE;
	178	}
	179
	180	void AliL3TrackArray::Remove(Int_t track){
	181	if(fIsPresent[track]){
	182	fIsPresent[track]=kFALSE;
	183	fNAbsent++;
	184	}
	185	}
	186
	187	void AliL3TrackArray::FillTracks(Int_t ntracks, AliL3TrackSegmentData* tr){
	188	//Read tracks from shared memory (or memory)
	189	AliL3TrackSegmentData *trs = tr;
95a00d93	190	for(Int_t i=0; i<ntracks; i++){
108615fc	191	AliL3Track *track = NextTrack();
	192	track->SetPt(trs->fPt);
	193	track->SetPsi(trs->fPsi);
	194	track->SetTgl(trs->fTgl);
	195	track->SetNHits(trs->fNPoints);
	196	track->SetCharge(trs->fCharge);
	197	track->SetFirstPoint(trs->fX,trs->fY,trs->fZ);
	198	track->SetLastPoint(trs->fLastX,trs->fLastY,trs->fLastZ);
	199	track->SetHits( trs->fNPoints, trs->fPointIDs );
	200	UChar_t tmpP = (UChar_t)trs;
	201	tmpP += sizeof(AliL3TrackSegmentData)+trs->fNPoints*sizeof(UInt_t);
	202	trs = (AliL3TrackSegmentData*)tmpP;
	203	}
	204	}
	205
	206	void AliL3TrackArray::FillTracks(Int_t ntracks, AliL3TrackSegmentData* tr,Int_t slice, AliL3Transform* trans){
	207	//Read tracks from shared memory (or memory)
	208	AliL3TrackSegmentData *trs = tr;
	209	for(Int_t i=0; i<ntracks; i++){
	210	AliL3Track *track = NextTrack();
	211	track->SetPt(trs->fPt);
	212	Float_t psi[1];
	213	psi[0]=trs->fPsi;
	214	trans->Local2GlobalAngle(psi,slice);
	215	track->SetPsi(psi[0]);
	216	track->SetTgl(trs->fTgl);
	217	track->SetNHits(trs->fNPoints);
	218	track->SetCharge(trs->fCharge);
	219	Float_t first[3];
	220	first[0]=trs->fX;first[1]=trs->fY;first[2]=trs->fZ;
	221	trans->Local2Global(first,slice);
	222	track->SetFirstPoint(first[0],first[1],first[2]);
	223	Float_t last[3];
	224	last[0]=trs->fLastX;last[1]=trs->fLastY;last[2]=trs->fLastZ;
	225	trans->Local2Global(last,slice);
	226	track->SetLastPoint(last[0],last[1],last[2]);
	227	track->SetHits( trs->fNPoints, trs->fPointIDs );
	228	UChar_t tmpP = (UChar_t)trs;
	229	tmpP += sizeof(AliL3TrackSegmentData)+trs->fNPoints*sizeof(UInt_t);
	230	trs = (AliL3TrackSegmentData*)tmpP;
	231	}
	232	}
	233
	234	UInt_t AliL3TrackArray::GetOutSize(){
	235	UInt_t count = GetOutCount(); //use only present tracks
	236	UInt_t tHits = 0;
	237	for(Int_t i=0;i<fNTracks;i++){ //loop over all tracks
	238	AliL3Track *track = GetCheckedTrack(i); //use only present tracks
	239	if(track) //use only present tracks
	240	tHits += track->GetNHits();
	241	}
	242
	243	//calculate size of track
	244	return countsizeof(AliL3TrackSegmentData)+sizeof(UInt_t)tHits;
	245	}
	246
	247	UInt_t AliL3TrackArray::WriteTracks(UInt_t & ntracks,AliL3TrackSegmentData* tr){
	248	ntracks = GetOutCount();
	249	return WriteTracks(tr);
	250	}
	251
	252	UInt_t AliL3TrackArray::WriteTracks(AliL3TrackSegmentData* tr){
	253	if(GetTrackType()=='c') return WriteConfMapTracks(tr);
	254	AliL3TrackSegmentData *tP = tr;
255	UInt_t *pP;
256	UInt_t size = 0;
257	for(Int_t i=0; i<fNTracks; i++){ //loop over all tracks
258	AliL3Track *track = GetCheckedTrack(i); //use only present tracks
259	if(!track) continue; //use only present tracks
260	tP->fX = track->GetFirstPointX();
261	tP->fY = track->GetFirstPointY();
262	tP->fZ = track->GetFirstPointZ();
263	tP->fPt = track->GetPt();
264	tP->fLastX = track->GetLastPointX();
265	tP->fLastY = track->GetLastPointY();
266	tP->fLastZ = track->GetLastPointZ();
267	tP->fPsi = track->GetPsi();
268	tP->fTgl = track->GetTgl();
269	tP->fCharge = track->GetCharge();
270	tP->fNPoints = track->GetNHits();
271	pP = (UInt_t*)track->GetHitNumbers();
272	for (UInt_t j=0;j<tP->fNPoints;j++){
273	tP->fPointIDs[j] = pP[j];
274	}
275	Byte_t tmpP = (Byte_t )tP;
276	tmpP += sizeof(AliL3TrackSegmentData)+tP->fNPoints*sizeof(UInt_t);
277	size += sizeof(AliL3TrackSegmentData)+tP->fNPoints*sizeof(UInt_t);
278	tP = (AliL3TrackSegmentData*)tmpP;
279	}
280	return size;
281	}
282
283	UInt_t AliL3TrackArray::WriteConfMapTracks(AliL3TrackSegmentData* tr){
284	// use first and last point objects
285	AliL3TrackSegmentData *tP = tr;
286	UInt_t *pP;
287	UInt_t size = 0;
288	for(Int_t i=0; i<fNTracks; i++){ //loop over all tracks
289	AliL3ConfMapTrack track =(AliL3ConfMapTrack ) GetCheckedTrack(i); //use only present tracks
290	if(!track) continue; //use only present tracks
291	AliL3ConfMapPoint hit = (AliL3ConfMapPoint)track->lastHit;
292	AliL3ConfMapPoint lastHit = (AliL3ConfMapPoint)track->firstHit;
293	tP->fX = hit->GetX();
294	tP->fY = hit->GetY();
295	tP->fZ = hit->GetZ();
296	tP->fLastX = lastHit->GetX();
297	tP->fLastY = lastHit->GetY();
298	tP->fLastZ = lastHit->GetZ();
299
300	// tP->fX = track->GetFirstPointX();
301	// tP->fY = track->GetFirstPointY();
302	// tP->fZ = track->GetFirstPointZ();
303	tP->fPt = track->GetPt();
304	// tP->fLastX = track->GetLastPointX();
305	// tP->fLastY = track->GetLastPointY();
306	// tP->fLastZ = track->GetLastPointZ();
307	tP->fPsi = track->GetPsi();
308	tP->fTgl = track->GetTgl();
309	tP->fCharge = track->GetCharge();
310	tP->fNPoints = track->GetNHits();
311	pP = (UInt_t*)track->GetHitNumbers();
312	for (UInt_t j=0;j<tP->fNPoints;j++){
313	tP->fPointIDs[j] = pP[j];
314	}
315	Byte_t tmpP = (Byte_t )tP;
316	tmpP += sizeof(AliL3TrackSegmentData)+tP->fNPoints*sizeof(UInt_t);
317	size +=sizeof(AliL3TrackSegmentData)+tP->fNPoints*sizeof(UInt_t);
318	tP = (AliL3TrackSegmentData*)tmpP;
319	}
320	return size;
321	}
322
2e843b99	323	void AliL3TrackArray::AddLast(AliL3Track *track)
	324	{
	325	AliL3Track *tpt = NextTrack();
	326	tpt->Set(track);
44fd1eb4	327
2e843b99	328	}
	329
	330
71df269a	331	void AliL3TrackArray::AddTracks(AliL3TrackArray *newtrack,Bool_t remove_old,Int_t slice)
71df269a	332	{
d5fe1d26	333	if(GetTrackType() != newtrack->GetTrackType() && GetTrackType()!='t')
	334	{
	335	LOG(AliL3Log::kError,"AliL3TrackArray::AddTracks","Track types")
	336	<<"Bad idea to add tracks of different types"<<ENDLOG;
	337	return;
	338	}
108615fc	339	if(fSize < fNTracks+newtrack->GetNPresent())
	340	SetSize(fSize+newtrack->GetSize());
	341	for(Int_t i =0;i<newtrack->GetNTracks();i++){
	342	AliL3Track *tpt = newtrack->GetCheckedTrack(i);
	343	if(!tpt) continue;
15846008	344	if(remove_old)
15846008	345	newtrack->Remove(i);
108615fc	346	AliL3Track *track = NextTrack();
ce54fcc8	347	track->Set(tpt);
71df269a	348	if(slice>=0)
71df269a	349	track->Rotate(slice); //Rotate track to global coordinates
203925a9	350	/*
	351	AliL3Track *track;
	352	if(GetTrackType()=='h')
	353	track = (AliL3HoughTrack*)NextTrack();
	354	else
	355	track = NextTrack();
	356	track->Set(tpt);
	357	*/
108615fc	358	}
	359	}
	360
	361
	362	void AliL3TrackArray::Compress(){
	363	if(GetNPresent()==GetNTracks()) return;
	364	AliL3Track *tmp = new AliL3Track [fNTracks];
	365	Int_t present=0;
	366	Int_t absent=GetNPresent();
	367	for(Int_t i=0;i<GetNTracks();i++){
	368	if(fIsPresent[i]) tmp[present++] = fTrack[i];
	369	else tmp[absent++] = fTrack[i];
	370	}
	371	for(Int_t i=0;i<GetNTracks();i++)
	372	fIsPresent[i]=kTRUE;
	373
	374	//Copy pointers back
	375	for(Int_t i=0; i<GetNTracks();i++){
	376	fTrack[i]=tmp[i];
	377	}
	378
	379	delete[] tmp;
	380
	381	fNTracks = GetNPresent();
	382	fNAbsent = 0;
	383	}
	384
	385	void AliL3TrackArray::QSort(){
	386	// compress an sort
	387	Compress();
	388	QSort(fTrack,0,fNTracks);
	389	}
	390
203925a9	391	void AliL3TrackArray::QSort( AliL3Track **a, Int_t first, Int_t last){
108615fc	392
	393	// Sort array of AliL3Track pointers using a quicksort algorithm.
	394	// Uses TrackCompare() to compare objects.
	395	// Thanks to Root!
	396
	397	static AliL3Track *tmp;
	398	static int i; // "static" to save stack space
	399	int j;
	400
	401	while (last - first > 1) {
	402	i = first;
	403	j = last;
	404	for (;;) {
203925a9	405	while (++i < last && TrackCompare(a[i], a[first]) < 0)
	406	;
	407	while (--j > first && TrackCompare(a[j], a[first]) > 0)
	408	;
108615fc	409	if (i >= j)
	410	break;
	411
	412	tmp = a[i];
	413	a[i] = a[j];
	414	a[j] = tmp;
	415	}
	416	if (j == first) {
	417	++first;
	418	continue;
	419	}
	420	tmp = a[first];
	421	a[first] = a[j];
	422	a[j] = tmp;
	423	if (j - first < last - (j + 1)) {
	424	QSort(a, first, j);
	425	first = j + 1; // QSort(j + 1, last);
	426	} else {
	427	QSort(a, j + 1, last);
	428	last = j; // QSort(first, j);
	429	}
	430	}
	431	}
	432
	433	Int_t AliL3TrackArray::TrackCompare(AliL3Track a, AliL3Track b){
	434	// Compare the two tracks.
203925a9	435
	436	return b->Compare(a);
	437
	438	/*
	439	if(fTrackType=='h')
	440	{
	441	AliL3HoughTrack tra = (AliL3HoughTrack)a;
	442	AliL3HoughTrack trb = (AliL3HoughTrack)b;
	443	if(tra->GetWeight() < trb->GetWeight()) return 1;
	444	if(tra->GetWeight() > trb->GetWeight()) return -1;
	445	}
	446	else
	447	{
	448	if(a->GetNHits() < b->GetNHits()) return 1;
	449	if(a->GetNHits() > b->GetNHits()) return -1;
	450	}
	451
	452	return 0;
	453	*/
108615fc	454	}
	455
	456