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

/* $Id$ */

// -----------------------
// Class AliMUONRecoCheck
// -----------------------
// Utility class to check the muon reconstruction. Reconstructed tracks are compared
// to reference tracks. The reference tracks are built from AliTrackReference for the
// hit in chamber (0..9) and from kinematics for the vertex parameters.     

#include "AliMUON.h"
#include "AliMUONRecoCheck.h"
#include "AliMUONTrack.h"
#include "AliMUONData.h"
#include "AliMUONConstants.h"

#include "AliRun.h" // for gAlice
#include "AliLoader.h" 
#include "AliRunLoader.h" 
#include "AliHeader.h"
#include "AliStack.h"
#include "AliTrackReference.h"
#include "AliLog.h" 

#include <TParticle.h>

ClassImp(AliMUONRecoCheck)

//_____________________________________________________________________________
AliMUONRecoCheck::AliMUONRecoCheck(Char_t *chLoader)
{
/// Constructor
  fMuonTrackRef = new TClonesArray("AliMUONTrack", 10);

  // open the run loader
  fRunLoader = AliRunLoader::Open(chLoader);
  if (!fRunLoader) {
    AliError(Form("no run loader found in file %s","galice.root" ));
    return;
  }
  // initialize loader's
  AliLoader *loader = fRunLoader->GetLoader("MUONLoader");

  // initialize container
  fMUONData  = new AliMUONData(loader,"MUON","MUON");

   // Loading AliRun master
  if (fRunLoader->GetAliRun() == 0x0) fRunLoader->LoadgAlice();

  fRunLoader->LoadKinematics("READ");
  fRunLoader->LoadTrackRefs("READ");
  loader->LoadTracks("READ");

  fReconstructibleTracks = 0; 
  fRecoTracks = 0;
}

//____________________________________________________________________
AliMUONRecoCheck::AliMUONRecoCheck(const AliMUONRecoCheck& rhs)
 : TObject(rhs)
{
/// Protected copy constructor

  AliFatal("Not implemented.");
}

//_____________________________________________________________________________
AliMUONRecoCheck::~AliMUONRecoCheck()
{
/// Destructor

  fRunLoader->UnloadKinematics();
  fRunLoader->UnloadTrackRefs();
  fRunLoader->UnloadTracks();
  fMuonTrackRef->Delete();
  delete fMuonTrackRef;
  delete fMUONData;
}

//________________________________________________________________________
AliMUONRecoCheck& AliMUONRecoCheck::operator = (const AliMUONRecoCheck& rhs)
{
/// Protected assignement operator

  if (this == &rhs) return *this;

  AliFatal("Not implemented.");
    
  return *this;  
}

//_____________________________________________________________________________
void AliMUONRecoCheck::MakeTrackRef()
{
/// Make reconstructible tracks

  AliTrackReference *trackReference;
  AliMUONTrack *muonTrack;
  AliMUONTrackParam *trackParam;
  AliMUONHitForRec *hitForRec;
  Float_t x, y, z, pX, pY, pZ, pYZ;
  Int_t track, trackSave;
  TParticle *particle;   
  Float_t bendingSlope = 0;
  Float_t  nonBendingSlope = 0;
  Float_t inverseBendingMomentum = 0;
  
  TTree* treeTR = fRunLoader->TreeTR();
  if (treeTR == NULL) return;

  TBranch* branch = treeTR->GetBranch("MUON");
  if (branch == NULL) return;

  TClonesArray* trackRefs = 0;
  branch->SetAddress(&trackRefs);
  branch->SetAutoDelete(kTRUE);  

  Int_t nTrackRef = (Int_t)branch->GetEntries();
 
  track = trackSave = -999;
  Bool_t isNewTrack;
  Int_t iHitMin, iChamber;

  trackParam = new AliMUONTrackParam();
  hitForRec = new AliMUONHitForRec();
  muonTrack = new AliMUONTrack();

  Int_t max = fRunLoader->GetHeader()->Stack()->GetNtrack();
  for (Int_t iTrackRef  = 0; iTrackRef < nTrackRef; iTrackRef++) {

    branch->GetEntry(iTrackRef);
    
    iHitMin = 0;
    isNewTrack = kTRUE;
    
    if (!trackRefs->GetEntries()) continue; 

    while (isNewTrack) {
      for (Int_t iHit = iHitMin; iHit < trackRefs->GetEntries(); iHit++) {
      
	trackReference = (AliTrackReference*)trackRefs->At(iHit);
	x = trackReference->X();
	y = trackReference->Y();
	z = trackReference->Z();
	pX = trackReference->Px();
	pY = trackReference->Py();
	pZ = trackReference->Pz();

	track = trackReference->GetTrack();

	if(track >= max){
	  AliWarningStream()
	    << "Track ID " << track 
	    << " larger than max number of particles " << max << endl;
	  isNewTrack = kFALSE;
	  break;
	}
	if (track != trackSave && iHit != 0) {
	  iHitMin = iHit;
	  trackSave = track;
	  break;
	}

	// track parameters at hit
	trackParam->SetBendingCoor(y);
	trackParam->SetNonBendingCoor(x);
	trackParam->SetZ(z);

	if (TMath::Abs(pZ) > 0) {
	  bendingSlope = pY/pZ;
	  nonBendingSlope = pX/pZ;
	}
	pYZ = TMath::Sqrt(pY*pY+pZ*pZ);
	if (pYZ >0) inverseBendingMomentum = 1/pYZ; 

	trackParam->SetBendingSlope(bendingSlope);
	trackParam->SetNonBendingSlope(nonBendingSlope);
	trackParam->SetInverseBendingMomentum(inverseBendingMomentum);

	hitForRec->SetBendingCoor(y);
	hitForRec->SetNonBendingCoor(x);
	hitForRec->SetZ(z);
	hitForRec->SetBendingReso2(0.0); 
	hitForRec->SetNonBendingReso2(0.0);  
	iChamber = AliMUONConstants::ChamberNumber(z);
	hitForRec->SetChamberNumber(iChamber);

	muonTrack->AddTrackParamAtHit(trackParam);
	muonTrack->AddHitForRecAtHit(hitForRec);
	muonTrack->SetTrackID(track);

	trackSave = track;
	if (iHit == trackRefs->GetEntries()-1) isNewTrack = kFALSE;
      }

      // track parameters at vertex 
      particle = fRunLoader->GetHeader()->Stack()->Particle(muonTrack->GetTrackID());

      if (particle) {

	x = particle->Vx();
	y = particle->Vy();
	z = particle->Vz();
	pX = particle->Px();
	pY = particle->Py();
	pZ = particle->Pz();

	trackParam->SetBendingCoor(y);
	trackParam->SetNonBendingCoor(x);
	trackParam->SetZ(z);
	if (TMath::Abs(pZ) > 0) {
	  bendingSlope = pY/pZ;
	  nonBendingSlope = pX/pZ;
	}
	pYZ = TMath::Sqrt(pY*pY+pZ*pZ);
	if (pYZ >0) inverseBendingMomentum = 1/pYZ;       
	trackParam->SetBendingSlope(bendingSlope);
	trackParam->SetNonBendingSlope(nonBendingSlope);
	trackParam->SetInverseBendingMomentum(inverseBendingMomentum);
      
	muonTrack->SetTrackParamAtVertex(trackParam);
      }

      AddMuonTrackReference(muonTrack);
      muonTrack->ResetTrackParamAtHit();
      muonTrack->ResetHitForRecAtHit();
      
    } // end while isNewTrack
  }
  
  CleanMuonTrackRef();
  
  ReconstructibleTracks();

  delete muonTrack;
  delete trackParam;
  delete hitForRec;

}

//____________________________________________________________________________
TClonesArray* AliMUONRecoCheck::GetTrackReco()
{
/// Return TClonesArray of reconstructed tracks

  GetMUONData()->ResetRecTracks();
  GetMUONData()->SetTreeAddress("RT");
  fTrackReco = GetMUONData()->RecTracks(); 
  GetMUONData()->GetRecTracks();
  fRecoTracks = fTrackReco->GetEntriesFast();
  return fTrackReco;
}
//_____________________________________________________________________________
void AliMUONRecoCheck::PrintEvent() const
{
/// Debug facility

  AliMUONTrack *track;
  AliMUONHitForRec *hitForRec;
  TClonesArray *  hitForRecAtHit = 0;
  Float_t xRec,yRec,zRec;

  Int_t nTrackRef = fMuonTrackRef->GetEntriesFast();
  
  printf(" ******************************* \n");
  printf(" nb of tracks %d \n",nTrackRef);

  for (Int_t index = 0; index < nTrackRef; index++) {
    track = (AliMUONTrack*)fMuonTrackRef->At(index);
    hitForRecAtHit = track->GetHitForRecAtHit();
    Int_t nTrackHits = hitForRecAtHit->GetEntriesFast();
    printf(" track number %d \n",index);
    for (Int_t iHit = 0; iHit < nTrackHits; iHit++){
      hitForRec = (AliMUONHitForRec*) hitForRecAtHit->At(iHit); 
      xRec  = hitForRec->GetNonBendingCoor();
      yRec  = hitForRec->GetBendingCoor();
      zRec  = hitForRec->GetZ();
      printf("  x,y,z: %f , %f , %f \n",xRec,yRec,zRec);
    }
  }
}

//_____________________________________________________________________________
void AliMUONRecoCheck::ResetTracks() const
{
/// Reset tracks

  if (fMuonTrackRef) fMuonTrackRef->Clear();
}
//_____________________________________________________________________________
void AliMUONRecoCheck::CleanMuonTrackRef()
{
/// Re-calculate hits parameters because two AliTrackReferences are recorded for
/// each chamber (one when particle is entering + one when particle is leaving 
/// the sensitive volume) 

  Float_t maxGasGap = 1.; // cm 
  AliMUONTrack *track, *trackNew;
  AliMUONHitForRec *hitForRec, *hitForRec1, *hitForRec2;
  AliMUONTrackParam *trackParam, *trackParam1, *trackParam2, *trackParamAtVertex;
  TClonesArray *  hitForRecAtHit = 0;
  TClonesArray *  trackParamAtHit = 0;
  Float_t xRec,yRec,zRec;
  Float_t xRec1,yRec1,zRec1;
  Float_t xRec2,yRec2,zRec2;
  Float_t bendingSlope,nonBendingSlope,bendingMomentum;
  Float_t bendingSlope1,nonBendingSlope1,bendingMomentum1;
  Float_t bendingSlope2,nonBendingSlope2,bendingMomentum2;
  TClonesArray *newMuonTrackRef = new TClonesArray("AliMUONTrack", 10);
  Int_t iHit1;
  Int_t iChamber = 0;
  Int_t nRec = 0;
  Int_t nTrackHits = 0;

  hitForRec = new AliMUONHitForRec();
  trackParam = new AliMUONTrackParam();
  trackNew = new AliMUONTrack();

  Int_t nTrackRef = fMuonTrackRef->GetEntriesFast();
  for (Int_t index = 0; index < nTrackRef; index++) {
    track = (AliMUONTrack*)fMuonTrackRef->At(index);
    hitForRecAtHit = track->GetHitForRecAtHit();
    trackParamAtHit = track->GetTrackParamAtHit();
    trackParamAtVertex = track->GetTrackParamAtVertex();
    nTrackHits = hitForRecAtHit->GetEntriesFast();
    iHit1 = 0;
    while (iHit1 < nTrackHits) {
      hitForRec1 = (AliMUONHitForRec*) hitForRecAtHit->At(iHit1); 
      trackParam1 = (AliMUONTrackParam*) trackParamAtHit->At(iHit1); 
      xRec1  = hitForRec1->GetNonBendingCoor();
      yRec1  = hitForRec1->GetBendingCoor();
      zRec1  = hitForRec1->GetZ();	
      xRec   = xRec1;
      yRec   = yRec1;
      zRec   = zRec1;
      bendingSlope1 = trackParam1->GetBendingSlope();
      nonBendingSlope1 = trackParam1->GetNonBendingSlope();
      bendingMomentum1 = 0;
      if (TMath::Abs(trackParam1->GetInverseBendingMomentum()) > 0)
	bendingMomentum1 = 1./trackParam1->GetInverseBendingMomentum();
      bendingSlope = bendingSlope1;
      nonBendingSlope = nonBendingSlope1;
      bendingMomentum = bendingMomentum1;
      nRec = 1;  
      for (Int_t iHit2 = iHit1+1; iHit2 < nTrackHits; iHit2++) {
	hitForRec2 = (AliMUONHitForRec*) hitForRecAtHit->At(iHit2); 
	trackParam2 = (AliMUONTrackParam*) trackParamAtHit->At(iHit2); 
	xRec2  = hitForRec2->GetNonBendingCoor();
	yRec2  = hitForRec2->GetBendingCoor();
	zRec2  = hitForRec2->GetZ();	  
	bendingSlope2 = trackParam2->GetBendingSlope();
	nonBendingSlope2 = trackParam2->GetNonBendingSlope();
	bendingMomentum2 = 0;
	if (TMath::Abs(trackParam2->GetInverseBendingMomentum()) > 0)
	  bendingMomentum2 = 1./trackParam2->GetInverseBendingMomentum();
	
	if ( TMath::Abs(zRec2-zRec1) < maxGasGap ) {

	  nRec++;
	  xRec += xRec2;
	  yRec += yRec2;
	  zRec += zRec2;
	  bendingSlope += bendingSlope2;
	  nonBendingSlope += nonBendingSlope2;
	  bendingMomentum += bendingMomentum2;
	  iHit1 = iHit2;
	}
	
      } // end iHit2
      xRec /= (Float_t)nRec;
      yRec /= (Float_t)nRec;
      zRec /= (Float_t)nRec;
      bendingSlope /= (Float_t)nRec;
      nonBendingSlope /= (Float_t)nRec;
      bendingMomentum /= (Float_t)nRec;
      
      hitForRec->SetNonBendingCoor(xRec);
      hitForRec->SetBendingCoor(yRec);
      hitForRec->SetZ(zRec);
      iChamber = AliMUONConstants::ChamberNumber(zRec);
      hitForRec->SetChamberNumber(iChamber);
      hitForRec->SetBendingReso2(0.0); 
      hitForRec->SetNonBendingReso2(0.0); 
      trackParam->SetNonBendingCoor(xRec);
      trackParam->SetBendingCoor(yRec);
      trackParam->SetZ(zRec);
      trackParam->SetNonBendingSlope(nonBendingSlope);
      trackParam->SetBendingSlope(bendingSlope);
      if (TMath::Abs(bendingMomentum) > 0)
	trackParam->SetInverseBendingMomentum(1./bendingMomentum);

      trackNew->AddHitForRecAtHit(hitForRec);
      trackNew->AddTrackParamAtHit(trackParam);
      
      iHit1++;
    } // end iHit1

    trackNew->SetTrackID(track->GetTrackID());
    trackNew->SetTrackParamAtVertex(trackParamAtVertex);
    {new ((*newMuonTrackRef)[newMuonTrackRef->GetEntriesFast()]) AliMUONTrack(*trackNew);}    
    trackNew->ResetHitForRecAtHit();
    trackNew->ResetTrackParamAtHit();
    
  } // end trackRef

  fMuonTrackRef->Clear();
  nTrackRef = newMuonTrackRef->GetEntriesFast();
  for (Int_t index = 0; index < nTrackRef; index++) {
    track = (AliMUONTrack*)newMuonTrackRef->At(index);
    AddMuonTrackReference(track);
  }
  

  delete trackNew;
  delete hitForRec;
  delete trackParam;
  newMuonTrackRef->Delete();
  delete newMuonTrackRef;
  
}

//_____________________________________________________________________________
void AliMUONRecoCheck::ReconstructibleTracks()
{
/// Calculate the number of reconstructible tracks
  
  AliMUONTrack* track;
  TClonesArray* hitForRecAtHit = NULL;
  AliMUONHitForRec* hitForRec;
  Float_t zRec;
  Int_t nTrackRef, nTrackHits;
  Int_t isChamberInTrack[10];
  Int_t iChamber = 0;
  Bool_t isTrackOK = kTRUE;

  fReconstructibleTracks = 0;

  nTrackRef = fMuonTrackRef->GetEntriesFast();
  for (Int_t index = 0; index < nTrackRef; index++) {
    track = (AliMUONTrack*)fMuonTrackRef->At(index);
    hitForRecAtHit = track->GetHitForRecAtHit();
    nTrackHits = hitForRecAtHit->GetEntriesFast();
    for (Int_t ch = 0; ch < 10; ch++) isChamberInTrack[ch] = 0;
 
    for ( Int_t iHit = 0; iHit < nTrackHits; iHit++) {
      hitForRec = (AliMUONHitForRec*) hitForRecAtHit->At(iHit); 
      zRec  = hitForRec->GetZ();
      iChamber = hitForRec->GetChamberNumber();
      if (iChamber < 0 || iChamber > 10) continue;
      isChamberInTrack[iChamber] = 1;
    } 
    // track is reconstructible if the particle is depositing a hit
    // in the following chamber combinations:

    isTrackOK = kTRUE;
    if (!isChamberInTrack[0] && !isChamberInTrack[1]) isTrackOK = kFALSE;
    if (!isChamberInTrack[2] && !isChamberInTrack[3]) isTrackOK = kFALSE;
    if (!isChamberInTrack[4] && !isChamberInTrack[5]) isTrackOK = kFALSE;
    Int_t nHitsInLastStations=0;
    for (Int_t ch = 6; ch < AliMUONConstants::NTrackingCh(); ch++)
      if (isChamberInTrack[ch]) nHitsInLastStations++; 
    if(nHitsInLastStations < 3) isTrackOK = kFALSE;

    if (isTrackOK) fReconstructibleTracks++;
    if (!isTrackOK) fMuonTrackRef->Remove(track); // remove non reconstructible tracks
  }
  fMuonTrackRef->Compress();
}
Commit	Line	Data
b8dc484b	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
d19b6003	16	/* $Id$ */
	17
	18	// -----------------------
	19	// Class AliMUONRecoCheck
	20	// -----------------------
b8dc484b	21	// Utility class to check the muon reconstruction. Reconstructed tracks are compared
	22	// to reference tracks. The reference tracks are built from AliTrackReference for the
	23	// hit in chamber (0..9) and from kinematics for the vertex parameters.
b8dc484b	24
cc056de1	25	#include "AliMUON.h"
	26	#include "AliMUONRecoCheck.h"
	27	#include "AliMUONTrack.h"
	28	#include "AliMUONData.h"
	29	#include "AliMUONConstants.h"
b8dc484b	30
b8dc484b	31	#include "AliRun.h" // for gAlice
b8dc484b	32	#include "AliLoader.h"
b8dc484b	33	#include "AliRunLoader.h"
b8dc484b	34	#include "AliHeader.h"
b8dc484b	35	#include "AliStack.h"
cc056de1	36	#include "AliTrackReference.h"
	37	#include "AliLog.h"
	38
	39	#include <TParticle.h>
b8dc484b	40
	41	ClassImp(AliMUONRecoCheck)
	42
	43	//_____________________________________________________________________________
	44	AliMUONRecoCheck::AliMUONRecoCheck(Char_t *chLoader)
	45	{
d19b6003	46	/// Constructor
b8dc484b	47	fMuonTrackRef = new TClonesArray("AliMUONTrack", 10);
	48
	49	// open the run loader
	50	fRunLoader = AliRunLoader::Open(chLoader);
	51	if (!fRunLoader) {
	52	AliError(Form("no run loader found in file %s","galice.root" ));
	53	return;
	54	}
	55	// initialize loader's
	56	AliLoader *loader = fRunLoader->GetLoader("MUONLoader");
	57
	58	// initialize container
	59	fMUONData = new AliMUONData(loader,"MUON","MUON");
	60
	61	// Loading AliRun master
	62	if (fRunLoader->GetAliRun() == 0x0) fRunLoader->LoadgAlice();
	63
	64	fRunLoader->LoadKinematics("READ");
	65	fRunLoader->LoadTrackRefs("READ");
	66	loader->LoadTracks("READ");
	67
	68	fReconstructibleTracks = 0;
	69	fRecoTracks = 0;
	70	}
	71
f29ba3e1	72	//____________________________________________________________________
	73	AliMUONRecoCheck::AliMUONRecoCheck(const AliMUONRecoCheck& rhs)
	74	: TObject(rhs)
	75	{
d19b6003	76	/// Protected copy constructor
f29ba3e1	77
	78	AliFatal("Not implemented.");
	79	}
	80
b8dc484b	81	//_____________________________________________________________________________
	82	AliMUONRecoCheck::~AliMUONRecoCheck()
	83	{
d19b6003	84	/// Destructor
d19b6003	85
b8dc484b	86	fRunLoader->UnloadKinematics();
	87	fRunLoader->UnloadTrackRefs();
	88	fRunLoader->UnloadTracks();
29fc2c86	89	fMuonTrackRef->Delete();
b8dc484b	90	delete fMuonTrackRef;
	91	delete fMUONData;
	92	}
	93
f29ba3e1	94	//________________________________________________________________________
	95	AliMUONRecoCheck& AliMUONRecoCheck::operator = (const AliMUONRecoCheck& rhs)
	96	{
d19b6003	97	/// Protected assignement operator
f29ba3e1	98
	99	if (this == &rhs) return *this;
	100
	101	AliFatal("Not implemented.");
	102
	103	return *this;
	104	}
	105
b8dc484b	106	//_____________________________________________________________________________
	107	void AliMUONRecoCheck::MakeTrackRef()
	108	{
d19b6003	109	/// Make reconstructible tracks
b8dc484b	110
	111	AliTrackReference *trackReference;
	112	AliMUONTrack *muonTrack;
	113	AliMUONTrackParam *trackParam;
	114	AliMUONHitForRec *hitForRec;
	115	Float_t x, y, z, pX, pY, pZ, pYZ;
	116	Int_t track, trackSave;
	117	TParticle *particle;
	118	Float_t bendingSlope = 0;
	119	Float_t nonBendingSlope = 0;
	120	Float_t inverseBendingMomentum = 0;
	121
	122	TTree* treeTR = fRunLoader->TreeTR();
	123	if (treeTR == NULL) return;
	124
	125	TBranch* branch = treeTR->GetBranch("MUON");
	126	if (branch == NULL) return;
	127
de876dd1	128	TClonesArray* trackRefs = 0;
b8dc484b	129	branch->SetAddress(&trackRefs);
de876dd1	130	branch->SetAutoDelete(kTRUE);
b8dc484b	131
de876dd1	132	Int_t nTrackRef = (Int_t)branch->GetEntries();
b8dc484b	133
	134	track = trackSave = -999;
	135	Bool_t isNewTrack;
	136	Int_t iHitMin, iChamber;
	137
	138	trackParam = new AliMUONTrackParam();
	139	hitForRec = new AliMUONHitForRec();
	140	muonTrack = new AliMUONTrack();
	141
cc056de1	142	Int_t max = fRunLoader->GetHeader()->Stack()->GetNtrack();
b8dc484b	143	for (Int_t iTrackRef = 0; iTrackRef < nTrackRef; iTrackRef++) {
cc056de1	144
de876dd1	145	branch->GetEntry(iTrackRef);
b8dc484b	146
	147	iHitMin = 0;
	148	isNewTrack = kTRUE;
dba3c809	149
29fc2c86	150	if (!trackRefs->GetEntries()) continue;
b8dc484b	151
29fc2c86	152	while (isNewTrack) {
b8dc484b	153	for (Int_t iHit = iHitMin; iHit < trackRefs->GetEntries(); iHit++) {
	154
	155	trackReference = (AliTrackReference*)trackRefs->At(iHit);
	156	x = trackReference->X();
	157	y = trackReference->Y();
	158	z = trackReference->Z();
	159	pX = trackReference->Px();
	160	pY = trackReference->Py();
	161	pZ = trackReference->Pz();
	162
	163	track = trackReference->GetTrack();
	164
cc056de1	165	if(track >= max){
	166	AliWarningStream()
	167	<< "Track ID " << track
	168	<< " larger than max number of particles " << max << endl;
	169	isNewTrack = kFALSE;
	170	break;
	171	}
b8dc484b	172	if (track != trackSave && iHit != 0) {
	173	iHitMin = iHit;
	174	trackSave = track;
	175	break;
	176	}
	177
	178	// track parameters at hit
	179	trackParam->SetBendingCoor(y);
	180	trackParam->SetNonBendingCoor(x);
	181	trackParam->SetZ(z);
cc056de1	182
b8dc484b	183	if (TMath::Abs(pZ) > 0) {
	184	bendingSlope = pY/pZ;
	185	nonBendingSlope = pX/pZ;
	186	}
	187	pYZ = TMath::Sqrt(pYpY+pZpZ);
	188	if (pYZ >0) inverseBendingMomentum = 1/pYZ;
	189
	190	trackParam->SetBendingSlope(bendingSlope);
	191	trackParam->SetNonBendingSlope(nonBendingSlope);
	192	trackParam->SetInverseBendingMomentum(inverseBendingMomentum);
	193
	194	hitForRec->SetBendingCoor(y);
	195	hitForRec->SetNonBendingCoor(x);
	196	hitForRec->SetZ(z);
	197	hitForRec->SetBendingReso2(0.0);
	198	hitForRec->SetNonBendingReso2(0.0);
29fc2c86	199	iChamber = AliMUONConstants::ChamberNumber(z);
b8dc484b	200	hitForRec->SetChamberNumber(iChamber);
	201
	202	muonTrack->AddTrackParamAtHit(trackParam);
	203	muonTrack->AddHitForRecAtHit(hitForRec);
	204	muonTrack->SetTrackID(track);
cc056de1	205
b8dc484b	206	trackSave = track;
	207	if (iHit == trackRefs->GetEntries()-1) isNewTrack = kFALSE;
	208	}
cc056de1	209
b8dc484b	210	// track parameters at vertex
b8dc484b	211	particle = fRunLoader->GetHeader()->Stack()->Particle(muonTrack->GetTrackID());
cc056de1	212
b8dc484b	213	if (particle) {
cc056de1	214
b8dc484b	215	x = particle->Vx();
	216	y = particle->Vy();
	217	z = particle->Vz();
	218	pX = particle->Px();
	219	pY = particle->Py();
	220	pZ = particle->Pz();
	221
	222	trackParam->SetBendingCoor(y);
	223	trackParam->SetNonBendingCoor(x);
	224	trackParam->SetZ(z);
	225	if (TMath::Abs(pZ) > 0) {
	226	bendingSlope = pY/pZ;
	227	nonBendingSlope = pX/pZ;
	228	}
	229	pYZ = TMath::Sqrt(pYpY+pZpZ);
	230	if (pYZ >0) inverseBendingMomentum = 1/pYZ;
	231	trackParam->SetBendingSlope(bendingSlope);
	232	trackParam->SetNonBendingSlope(nonBendingSlope);
	233	trackParam->SetInverseBendingMomentum(inverseBendingMomentum);
	234
	235	muonTrack->SetTrackParamAtVertex(trackParam);
	236	}
cc056de1	237
b8dc484b	238	AddMuonTrackReference(muonTrack);
	239	muonTrack->ResetTrackParamAtHit();
	240	muonTrack->ResetHitForRecAtHit();
	241
	242	} // end while isNewTrack
b8dc484b	243	}
	244
	245	CleanMuonTrackRef();
	246
	247	ReconstructibleTracks();
	248
	249	delete muonTrack;
	250	delete trackParam;
	251	delete hitForRec;
b8dc484b	252
	253	}
	254
	255	//____________________________________________________________________________
	256	TClonesArray* AliMUONRecoCheck::GetTrackReco()
	257	{
d19b6003	258	/// Return TClonesArray of reconstructed tracks
b8dc484b	259
29fc2c86	260	GetMUONData()->ResetRecTracks();
b8dc484b	261	GetMUONData()->SetTreeAddress("RT");
	262	fTrackReco = GetMUONData()->RecTracks();
	263	GetMUONData()->GetRecTracks();
	264	fRecoTracks = fTrackReco->GetEntriesFast();
	265	return fTrackReco;
	266	}
	267	//_____________________________________________________________________________
	268	void AliMUONRecoCheck::PrintEvent() const
	269	{
d19b6003	270	/// Debug facility
b8dc484b	271
	272	AliMUONTrack *track;
	273	AliMUONHitForRec *hitForRec;
	274	TClonesArray * hitForRecAtHit = 0;
	275	Float_t xRec,yRec,zRec;
	276
	277	Int_t nTrackRef = fMuonTrackRef->GetEntriesFast();
	278
	279	printf(" ******************************* \n");
	280	printf(" nb of tracks %d \n",nTrackRef);
	281
	282	for (Int_t index = 0; index < nTrackRef; index++) {
	283	track = (AliMUONTrack*)fMuonTrackRef->At(index);
	284	hitForRecAtHit = track->GetHitForRecAtHit();
	285	Int_t nTrackHits = hitForRecAtHit->GetEntriesFast();
	286	printf(" track number %d \n",index);
	287	for (Int_t iHit = 0; iHit < nTrackHits; iHit++){
	288	hitForRec = (AliMUONHitForRec*) hitForRecAtHit->At(iHit);
	289	xRec = hitForRec->GetNonBendingCoor();
	290	yRec = hitForRec->GetBendingCoor();
	291	zRec = hitForRec->GetZ();
	292	printf(" x,y,z: %f , %f , %f \n",xRec,yRec,zRec);
	293	}
	294	}
	295	}
	296
	297	//_____________________________________________________________________________
	298	void AliMUONRecoCheck::ResetTracks() const
	299	{
d19b6003	300	/// Reset tracks
d19b6003	301
b8dc484b	302	if (fMuonTrackRef) fMuonTrackRef->Clear();
	303	}
	304	//_____________________________________________________________________________
	305	void AliMUONRecoCheck::CleanMuonTrackRef()
	306	{
d19b6003	307	/// Re-calculate hits parameters because two AliTrackReferences are recorded for
	308	/// each chamber (one when particle is entering + one when particle is leaving
	309	/// the sensitive volume)
b8dc484b	310
	311	Float_t maxGasGap = 1.; // cm
	312	AliMUONTrack track, trackNew;
	313	AliMUONHitForRec hitForRec, hitForRec1, *hitForRec2;
	314	AliMUONTrackParam trackParam, trackParam1, trackParam2, trackParamAtVertex;
	315	TClonesArray * hitForRecAtHit = 0;
	316	TClonesArray * trackParamAtHit = 0;
	317	Float_t xRec,yRec,zRec;
	318	Float_t xRec1,yRec1,zRec1;
	319	Float_t xRec2,yRec2,zRec2;
	320	Float_t bendingSlope,nonBendingSlope,bendingMomentum;
	321	Float_t bendingSlope1,nonBendingSlope1,bendingMomentum1;
	322	Float_t bendingSlope2,nonBendingSlope2,bendingMomentum2;
	323	TClonesArray *newMuonTrackRef = new TClonesArray("AliMUONTrack", 10);
	324	Int_t iHit1;
	325	Int_t iChamber = 0;
	326	Int_t nRec = 0;
	327	Int_t nTrackHits = 0;
	328
	329	hitForRec = new AliMUONHitForRec();
	330	trackParam = new AliMUONTrackParam();
	331	trackNew = new AliMUONTrack();
	332
	333	Int_t nTrackRef = fMuonTrackRef->GetEntriesFast();
b8dc484b	334	for (Int_t index = 0; index < nTrackRef; index++) {
	335	track = (AliMUONTrack*)fMuonTrackRef->At(index);
	336	hitForRecAtHit = track->GetHitForRecAtHit();
	337	trackParamAtHit = track->GetTrackParamAtHit();
	338	trackParamAtVertex = track->GetTrackParamAtVertex();
	339	nTrackHits = hitForRecAtHit->GetEntriesFast();
	340	iHit1 = 0;
	341	while (iHit1 < nTrackHits) {
	342	hitForRec1 = (AliMUONHitForRec*) hitForRecAtHit->At(iHit1);
	343	trackParam1 = (AliMUONTrackParam*) trackParamAtHit->At(iHit1);
	344	xRec1 = hitForRec1->GetNonBendingCoor();
	345	yRec1 = hitForRec1->GetBendingCoor();
	346	zRec1 = hitForRec1->GetZ();
	347	xRec = xRec1;
	348	yRec = yRec1;
	349	zRec = zRec1;
	350	bendingSlope1 = trackParam1->GetBendingSlope();
	351	nonBendingSlope1 = trackParam1->GetNonBendingSlope();
de876dd1	352	bendingMomentum1 = 0;
	353	if (TMath::Abs(trackParam1->GetInverseBendingMomentum()) > 0)
	354	bendingMomentum1 = 1./trackParam1->GetInverseBendingMomentum();
b8dc484b	355	bendingSlope = bendingSlope1;
	356	nonBendingSlope = nonBendingSlope1;
	357	bendingMomentum = bendingMomentum1;
	358	nRec = 1;
	359	for (Int_t iHit2 = iHit1+1; iHit2 < nTrackHits; iHit2++) {
	360	hitForRec2 = (AliMUONHitForRec*) hitForRecAtHit->At(iHit2);
	361	trackParam2 = (AliMUONTrackParam*) trackParamAtHit->At(iHit2);
	362	xRec2 = hitForRec2->GetNonBendingCoor();
	363	yRec2 = hitForRec2->GetBendingCoor();
	364	zRec2 = hitForRec2->GetZ();
	365	bendingSlope2 = trackParam2->GetBendingSlope();
	366	nonBendingSlope2 = trackParam2->GetNonBendingSlope();
de876dd1	367	bendingMomentum2 = 0;
	368	if (TMath::Abs(trackParam2->GetInverseBendingMomentum()) > 0)
	369	bendingMomentum2 = 1./trackParam2->GetInverseBendingMomentum();
b8dc484b	370
b8dc484b	371	if ( TMath::Abs(zRec2-zRec1) < maxGasGap ) {
cc056de1	372
b8dc484b	373	nRec++;
	374	xRec += xRec2;
	375	yRec += yRec2;
	376	zRec += zRec2;
	377	bendingSlope += bendingSlope2;
	378	nonBendingSlope += nonBendingSlope2;
	379	bendingMomentum += bendingMomentum2;
	380	iHit1 = iHit2;
	381	}
	382
	383	} // end iHit2
	384	xRec /= (Float_t)nRec;
	385	yRec /= (Float_t)nRec;
	386	zRec /= (Float_t)nRec;
	387	bendingSlope /= (Float_t)nRec;
	388	nonBendingSlope /= (Float_t)nRec;
	389	bendingMomentum /= (Float_t)nRec;
	390
	391	hitForRec->SetNonBendingCoor(xRec);
	392	hitForRec->SetBendingCoor(yRec);
	393	hitForRec->SetZ(zRec);
29fc2c86	394	iChamber = AliMUONConstants::ChamberNumber(zRec);
b8dc484b	395	hitForRec->SetChamberNumber(iChamber);
	396	hitForRec->SetBendingReso2(0.0);
	397	hitForRec->SetNonBendingReso2(0.0);
	398	trackParam->SetNonBendingCoor(xRec);
	399	trackParam->SetBendingCoor(yRec);
	400	trackParam->SetZ(zRec);
	401	trackParam->SetNonBendingSlope(nonBendingSlope);
	402	trackParam->SetBendingSlope(bendingSlope);
de876dd1	403	if (TMath::Abs(bendingMomentum) > 0)
de876dd1	404	trackParam->SetInverseBendingMomentum(1./bendingMomentum);
b8dc484b	405
	406	trackNew->AddHitForRecAtHit(hitForRec);
	407	trackNew->AddTrackParamAtHit(trackParam);
	408
	409	iHit1++;
	410	} // end iHit1
	411
	412	trackNew->SetTrackID(track->GetTrackID());
	413	trackNew->SetTrackParamAtVertex(trackParamAtVertex);
	414	{new ((newMuonTrackRef)[newMuonTrackRef->GetEntriesFast()]) AliMUONTrack(trackNew);}
	415	trackNew->ResetHitForRecAtHit();
	416	trackNew->ResetTrackParamAtHit();
	417
	418	} // end trackRef
	419
	420	fMuonTrackRef->Clear();
	421	nTrackRef = newMuonTrackRef->GetEntriesFast();
	422	for (Int_t index = 0; index < nTrackRef; index++) {
	423	track = (AliMUONTrack*)newMuonTrackRef->At(index);
	424	AddMuonTrackReference(track);
	425	}
	426
	427
	428	delete trackNew;
	429	delete hitForRec;
	430	delete trackParam;
29fc2c86	431	newMuonTrackRef->Delete();
b8dc484b	432	delete newMuonTrackRef;
	433
	434	}
	435
	436	//_____________________________________________________________________________
	437	void AliMUONRecoCheck::ReconstructibleTracks()
	438	{
d19b6003	439	/// Calculate the number of reconstructible tracks
b8dc484b	440
	441	AliMUONTrack* track;
	442	TClonesArray* hitForRecAtHit = NULL;
	443	AliMUONHitForRec* hitForRec;
	444	Float_t zRec;
	445	Int_t nTrackRef, nTrackHits;
	446	Int_t isChamberInTrack[10];
	447	Int_t iChamber = 0;
	448	Bool_t isTrackOK = kTRUE;
	449
	450	fReconstructibleTracks = 0;
	451
	452	nTrackRef = fMuonTrackRef->GetEntriesFast();
	453	for (Int_t index = 0; index < nTrackRef; index++) {
	454	track = (AliMUONTrack*)fMuonTrackRef->At(index);
	455	hitForRecAtHit = track->GetHitForRecAtHit();
	456	nTrackHits = hitForRecAtHit->GetEntriesFast();
	457	for (Int_t ch = 0; ch < 10; ch++) isChamberInTrack[ch] = 0;
	458
	459	for ( Int_t iHit = 0; iHit < nTrackHits; iHit++) {
	460	hitForRec = (AliMUONHitForRec*) hitForRecAtHit->At(iHit);
	461	zRec = hitForRec->GetZ();
	462	iChamber = hitForRec->GetChamberNumber();
	463	if (iChamber < 0 \|\| iChamber > 10) continue;
	464	isChamberInTrack[iChamber] = 1;
cc056de1	465	}
	466	// track is reconstructible if the particle is depositing a hit
	467	// in the following chamber combinations:
	468
b8dc484b	469	isTrackOK = kTRUE;
cc056de1	470	if (!isChamberInTrack[0] && !isChamberInTrack[1]) isTrackOK = kFALSE;
	471	if (!isChamberInTrack[2] && !isChamberInTrack[3]) isTrackOK = kFALSE;
	472	if (!isChamberInTrack[4] && !isChamberInTrack[5]) isTrackOK = kFALSE;
	473	Int_t nHitsInLastStations=0;
	474	for (Int_t ch = 6; ch < AliMUONConstants::NTrackingCh(); ch++)
	475	if (isChamberInTrack[ch]) nHitsInLastStations++;
	476	if(nHitsInLastStations < 3) isTrackOK = kFALSE;
	477
	478	if (isTrackOK) fReconstructibleTracks++;
b8dc484b	479	if (!isTrackOK) fMuonTrackRef->Remove(track); // remove non reconstructible tracks
	480	}
	481	fMuonTrackRef->Compress();
	482	}
	483