* provided "as is" without express or implied warranty. *
**************************************************************************/
-/*
-$Log$
-Revision 1.22 2001/01/26 20:00:53 hristov
-Major upgrade of AliRoot code
-Revision 1.20 2001/01/08 11:01:02 gosset
-Modifications used for addendum to Dimuon TDR (JP Cussonneau):
-*. MaxBendingMomentum to make both a segment and a track (default 500)
-*. MaxChi2 per degree of freedom to make a track (default 100)
-*. MinBendingMomentum used also to make a track
- and not only a segment (default 3)
-*. wider roads for track search in stations 1 to 3
-*. extrapolation to actual Z instead of Z(chamber) in FollowTracks
-*. in track fit:
- - limits on parameters X and Y (+/-500)
- - covariance matrices in double precision
- - normalization of covariance matrices before inversion
- - suppression of Minuit printouts
-*. correction against memory leak (delete extrapHit) in FollowTracks
-*. RMax to 10 degrees with Z(chamber) instead of fixed values;
- RMin and Rmax cuts suppressed in NewHitForRecFromGEANT,
- because useless with realistic geometry
-
-Revision 1.19 2000/11/20 11:24:10 gosset
-New package for reconstructed tracks (A. Gheata):
-* tree output in the file "tree_reco.root"
-* display events and make histograms from this file
-
-Revision 1.18 2000/10/26 12:47:03 gosset
-Real distance between chambers of each station taken into account
-for the reconstruction parameters "fSegmentMaxDistBending[5]"
-
-Revision 1.17 2000/10/24 09:26:20 gosset
-Comments updated
-
-Revision 1.16 2000/10/24 09:22:35 gosset
-Method AddHitsForRecFromRawClusters: real Z of raw cluster and not Z of chamber
-
-Revision 1.15 2000/10/12 15:17:30 gosset
-Sign of fSimpleBValue corrected: sign ox Bx and not Bz (thanks to Galina)
-
-Revision 1.14 2000/10/04 18:21:26 morsch
-Include stdlib.h
-
-Revision 1.13 2000/10/02 21:28:09 fca
-Removal of useless dependecies via forward declarations
-
-Revision 1.12 2000/10/02 16:58:29 egangler
-Cleaning of the code :
--> coding conventions
--> void Streamers
--> some useless includes removed or replaced by "class" statement
-
-Revision 1.11 2000/09/22 09:16:33 hristov
-Casting needed on DEC
-
-Revision 1.10 2000/09/19 09:49:50 gosset
-AliMUONEventReconstructor package
-* track extrapolation independent from reco_muon.F, use of AliMagF...
-* possibility to use new magnetic field (automatic from generated root file)
-
-Revision 1.9 2000/07/20 12:45:27 gosset
-New "EventReconstructor..." structure,
- hopefully more adapted to tree/streamer.
-"AliMUONEventReconstructor::RemoveDoubleTracks"
- to keep only one track among similar ones.
-
-Revision 1.8 2000/07/18 16:04:06 gosset
-AliMUONEventReconstructor package:
-* a few minor modifications and more comments
-* a few corrections
- * right sign for Z of raw clusters
- * right loop over chambers inside station
- * symmetrized covariance matrix for measurements (TrackChi2MCS)
- * right sign of charge in extrapolation (ExtrapToZ)
- * right zEndAbsorber for Branson correction below 3 degrees
-* use of TVirtualFitter instead of TMinuit for AliMUONTrack::Fit
-* no parameter for AliMUONTrack::Fit() but more fit parameters in Track object
-
-Revision 1.7 2000/07/03 12:28:06 gosset
-Printout at the right place after extrapolation to vertex
-
-Revision 1.6 2000/06/30 12:01:06 gosset
-Correction for hit search in the right chamber (JPC)
-
-Revision 1.5 2000/06/30 10:15:48 gosset
-Changes to EventReconstructor...:
-precision fit with multiple Coulomb scattering;
-extrapolation to vertex with Branson correction in absorber (JPC)
-
-Revision 1.4 2000/06/27 14:11:36 gosset
-Corrections against violations of coding conventions
-
-Revision 1.3 2000/06/16 07:27:08 gosset
-To remove problem in running RuleChecker, like in MUON-dev
-
-Revision 1.1.2.5 2000/06/16 07:00:26 gosset
-To remove problem in running RuleChecker
-
-Revision 1.1.2.4 2000/06/12 08:00:07 morsch
-Dummy streamer to solve CINT compilation problem (to be investigated !)
-
-Revision 1.1.2.3 2000/06/09 20:59:57 morsch
-Make includes consistent with new file structure.
-
-Revision 1.1.2.2 2000/06/09 12:58:05 gosset
-Removed comment beginnings in Log sections of .cxx files
-Suppressed most violations of coding rules
-
-Revision 1.1.2.1 2000/06/07 14:44:53 gosset
-Addition of files for track reconstruction in C++
-*/
+/* $Id$ */
-//__________________________________________________________________________
+////////////////////////////////////
//
// MUON event reconstructor in ALICE
//
// * MakeEventToBeReconstructed to build the array of hits to be reconstructed
// * MakeSegments to build the segments
// * MakeTracks to build the tracks
-//__________________________________________________________________________
+//
+////////////////////////////////////
-#include <iostream.h>
#include <stdlib.h>
-
-#include <TRandom.h>
+#include <Riostream.h>
+#include <TDirectory.h>
#include <TFile.h>
-#include <TTree.h>
+#include <TMatrixD.h> //AZ
#include "AliMUONEventReconstructor.h"
#include "AliMUON.h"
-#include "AliMUONHitForRec.h"
-#include "AliMUONSegment.h"
+#include "AliMUONConstants.h"
#include "AliMUONHit.h"
+#include "AliMUONHitForRec.h"
+#include "AliMUONTriggerTrack.h"
+#include "AliMUONTriggerCircuit.h"
#include "AliMUONRawCluster.h"
+#include "AliMUONLocalTrigger.h"
+#include "AliMUONGlobalTrigger.h"
+#include "AliMUONRecoEvent.h"
+#include "AliMUONSegment.h"
#include "AliMUONTrack.h"
-#include "AliMUONChamber.h"
#include "AliMUONTrackHit.h"
#include "AliMagF.h"
-#include "AliRun.h"
-#include "TParticle.h"
-#include "AliMUONRecoEvent.h"
+#include "AliRun.h" // for gAlice
+#include "AliRunLoader.h"
+#include "AliLoader.h"
+#include "AliMUONTrackK.h" //AZ
+#include "AliMC.h"
+#include "AliLog.h"
+#include "AliTrackReference.h"
//************* Defaults parameters for reconstruction
-static const Double_t kDefaultMinBendingMomentum = 3.0;
-static const Double_t kDefaultMaxBendingMomentum = 500.0;
-static const Double_t kDefaultMaxChi2 = 100.0;
-static const Double_t kDefaultMaxSigma2Distance = 16.0;
-static const Double_t kDefaultBendingResolution = 0.01;
-static const Double_t kDefaultNonBendingResolution = 0.144;
-static const Double_t kDefaultChamberThicknessInX0 = 0.03;
+const Double_t AliMUONEventReconstructor::fgkDefaultMinBendingMomentum = 3.0;
+const Double_t AliMUONEventReconstructor::fgkDefaultMaxBendingMomentum = 500.0;
+const Double_t AliMUONEventReconstructor::fgkDefaultMaxChi2 = 100.0;
+const Double_t AliMUONEventReconstructor::fgkDefaultMaxSigma2Distance = 16.0;
+const Double_t AliMUONEventReconstructor::fgkDefaultBendingResolution = 0.01;
+const Double_t AliMUONEventReconstructor::fgkDefaultNonBendingResolution = 0.144;
+const Double_t AliMUONEventReconstructor::fgkDefaultChamberThicknessInX0 = 0.03;
// Simple magnetic field:
// Value taken from macro MUONtracking.C: 0.7 T, hence 7 kG
// Length and Position from reco_muon.F, with opposite sign:
// Length = ZMAGEND-ZCOIL
// Position = (ZMAGEND+ZCOIL)/2
// to be ajusted differently from real magnetic field ????
-static const Double_t kDefaultSimpleBValue = 7.0;
-static const Double_t kDefaultSimpleBLength = 428.0;
-static const Double_t kDefaultSimpleBPosition = 1019.0;
-static const Int_t kDefaultRecGeantHits = 0;
-static const Double_t kDefaultEfficiency = 0.95;
+const Double_t AliMUONEventReconstructor::fgkDefaultSimpleBValue = 7.0;
+const Double_t AliMUONEventReconstructor::fgkDefaultSimpleBLength = 428.0;
+const Double_t AliMUONEventReconstructor::fgkDefaultSimpleBPosition = 1019.0;
+const Int_t AliMUONEventReconstructor::fgkDefaultRecTrackRefHits = 0;
+const Double_t AliMUONEventReconstructor::fgkDefaultEfficiency = 0.95;
-static const Int_t kDefaultPrintLevel = 0;
+const Int_t AliMUONEventReconstructor::fgkDefaultPrintLevel = -1;
ClassImp(AliMUONEventReconstructor) // Class implementation in ROOT context
//__________________________________________________________________________
-AliMUONEventReconstructor::AliMUONEventReconstructor(void)
+AliMUONEventReconstructor::AliMUONEventReconstructor(AliLoader* loader)
+ : TObject()
{
// Constructor for class AliMUONEventReconstructor
SetReconstructionParametersToDefaults();
+ fTrackMethod = 1; //AZ - tracking method (1-default, 2-Kalman)
// Memory allocation for the TClonesArray of hits for reconstruction
// Is 10000 the right size ????
fHitsForRecPtr = new TClonesArray("AliMUONHitForRec", 10000);
fNHitsForRec = 0; // really needed or GetEntriesFast sufficient ????
// Memory allocation for the TClonesArray's of segments in stations
// Is 2000 the right size ????
- for (Int_t st = 0; st < kMaxMuonTrackingStations; st++) {
+ for (Int_t st = 0; st < fgkMaxMuonTrackingStations; st++) {
fSegmentsPtr[st] = new TClonesArray("AliMUONSegment", 2000);
fNSegments[st] = 0; // really needed or GetEntriesFast sufficient ????
}
// Is 10 the right size ????
fRecTracksPtr = new TClonesArray("AliMUONTrack", 10);
fNRecTracks = 0; // really needed or GetEntriesFast sufficient ????
+// trigger tracks
+ fRecTriggerTracksPtr = new TClonesArray("AliMUONTriggerTrack", 10);
+ fNRecTriggerTracks = 0; // really needed or GetEntriesFast sufficient ????
// Memory allocation for the TClonesArray of hits on reconstructed tracks
// Is 100 the right size ????
- fRecTrackHitsPtr = new TClonesArray("AliMUONTrack", 100);
+ //AZ fRecTrackHitsPtr = new TClonesArray("AliMUONTrack", 100);
+ fRecTrackHitsPtr = new TClonesArray("AliMUONTrackHit", 100);
fNRecTrackHits = 0; // really needed or GetEntriesFast sufficient ????
- // Sign of fSimpleBValue according to sign of Bx value at (50,50,950).
+ // Sign of fSimpleBValue according to sign of Bx value at (50,50,-950).
Float_t b[3], x[3];
- x[0] = 50.; x[1] = 50.; x[2] = 950.;
+ x[0] = 50.; x[1] = 50.; x[2] = -950.;
gAlice->Field()->Field(x, b);
fSimpleBValue = TMath::Sign(fSimpleBValue,(Double_t) b[0]);
+ fSimpleBPosition = TMath::Sign(fSimpleBPosition,(Double_t) x[2]);
// See how to get fSimple(BValue, BLength, BPosition)
// automatically calculated from the actual magnetic field ????
cout << endl;
}
+ // Initializions for track ref. background events
+ fBkgTrackRefFile = 0;
+ fBkgTrackRefTK = 0;
+ fBkgTrackRefParticles = 0;
+ fBkgTrackRefTTR = 0;
+ fBkgTrackRefEventNumber = -1;
+
// Initialize to 0 pointers to RecoEvent, tree and tree file
fRecoEvent = 0;
fEventTree = 0;
fTreeFile = 0;
-
+
+ // initialize loader's
+ fLoader = loader;
+
+ // initialize container
+ fMUONData = new AliMUONData(fLoader,"MUON","MUON");
+
+ // Loading AliRun master
+ AliRunLoader* runloader = fLoader->GetRunLoader();
+ if (runloader->GetAliRun() == 0x0) runloader->LoadgAlice();
+ gAlice = runloader->GetAliRun();
+
return;
}
-
-AliMUONEventReconstructor::AliMUONEventReconstructor (const AliMUONEventReconstructor& Reconstructor)
+ //__________________________________________________________________________
+AliMUONEventReconstructor::AliMUONEventReconstructor (const AliMUONEventReconstructor& rhs)
+ : TObject(rhs)
{
- // Dummy copy constructor
+// Protected copy constructor
+
+ AliFatal("Not implemented.");
}
-AliMUONEventReconstructor & AliMUONEventReconstructor::operator=(const AliMUONEventReconstructor& Reconstructor)
+AliMUONEventReconstructor &
+AliMUONEventReconstructor::operator=(const AliMUONEventReconstructor& rhs)
{
- // Dummy assignment operator
- return *this;
+// Protected assignement operator
+
+ if (this == &rhs) return *this;
+
+ AliFatal("Not implemented.");
+
+ return *this;
}
//__________________________________________________________________________
// if (fEventTree) delete fEventTree;
if (fRecoEvent) delete fRecoEvent;
delete fHitsForRecPtr; // Correct destruction of everything ???? or delete [] ????
- for (Int_t st = 0; st < kMaxMuonTrackingStations; st++)
+ for (Int_t st = 0; st < fgkMaxMuonTrackingStations; st++)
delete fSegmentsPtr[st]; // Correct destruction of everything ????
return;
}
{
// Set reconstruction parameters to default values
// Would be much more convenient with a structure (or class) ????
- fMinBendingMomentum = kDefaultMinBendingMomentum;
- fMaxBendingMomentum = kDefaultMaxBendingMomentum;
- fMaxChi2 = kDefaultMaxChi2;
- fMaxSigma2Distance = kDefaultMaxSigma2Distance;
+ fMinBendingMomentum = fgkDefaultMinBendingMomentum;
+ fMaxBendingMomentum = fgkDefaultMaxBendingMomentum;
+ fMaxChi2 = fgkDefaultMaxChi2;
+ fMaxSigma2Distance = fgkDefaultMaxSigma2Distance;
AliMUON *pMUON = (AliMUON*) gAlice->GetModule("MUON");
// ******** Parameters for making HitsForRec
// like in TRACKF_STAT:
// 2 degrees for stations 1 and 2, or ch(0...) from 0 to 3;
// 30 cm for stations 3 to 5, or ch(0...) from 4 to 9
- for (Int_t ch = 0; ch < kMaxMuonTrackingChambers; ch++) {
+ for (Int_t ch = 0; ch < fgkMaxMuonTrackingChambers; ch++) {
if (ch < 4) fRMin[ch] = TMath::Abs((&(pMUON->Chamber(ch)))->Z()) *
2.0 * TMath::Pi() / 180.0;
else fRMin[ch] = 30.0;
// Maximum distance in non bending plane
// 5 * 0.22 just to remember the way it was made in TRACKF_STAT
// SIGCUT*DYMAX(IZ)
- for (Int_t st = 0; st < kMaxMuonTrackingStations; st++)
+ for (Int_t st = 0; st < fgkMaxMuonTrackingStations; st++)
fSegmentMaxDistNonBending[st] = 5. * 0.22;
// Maximum distance in bending plane:
// values from TRACKF_STAT, corresponding to (J psi 20cm),
// scaled to the real distance between chambers in a station
- fSegmentMaxDistBending[0] = 1.5 *
- ((&(pMUON->Chamber(1)))->Z() - (&(pMUON->Chamber(0)))->Z()) / 20.0;
- fSegmentMaxDistBending[1] = 1.5 *
- ((&(pMUON->Chamber(3)))->Z() - (&(pMUON->Chamber(2)))->Z()) / 20.0;
- fSegmentMaxDistBending[2] = 3.0 *
- ((&(pMUON->Chamber(5)))->Z() - (&(pMUON->Chamber(4)))->Z()) / 20.0;
- fSegmentMaxDistBending[3] = 6.0 *
- ((&(pMUON->Chamber(7)))->Z() - (&(pMUON->Chamber(6)))->Z()) / 20.0;
- fSegmentMaxDistBending[4] = 6.0 *
- ((&(pMUON->Chamber(9)))->Z() - (&(pMUON->Chamber(8)))->Z()) / 20.0;
+ fSegmentMaxDistBending[0] = TMath::Abs( 1.5 *
+ ((&(pMUON->Chamber(1)))->Z() - (&(pMUON->Chamber(0)))->Z()) / 20.0);
+ fSegmentMaxDistBending[1] = TMath::Abs( 1.5 *
+ ((&(pMUON->Chamber(3)))->Z() - (&(pMUON->Chamber(2)))->Z()) / 20.0);
+ fSegmentMaxDistBending[2] = TMath::Abs( 3.0 *
+ ((&(pMUON->Chamber(5)))->Z() - (&(pMUON->Chamber(4)))->Z()) / 20.0);
+ fSegmentMaxDistBending[3] = TMath::Abs( 6.0 *
+ ((&(pMUON->Chamber(7)))->Z() - (&(pMUON->Chamber(6)))->Z()) / 20.0);
+ fSegmentMaxDistBending[4] = TMath::Abs( 6.0 *
+ ((&(pMUON->Chamber(9)))->Z() - (&(pMUON->Chamber(8)))->Z()) / 20.0);
- fBendingResolution = kDefaultBendingResolution;
- fNonBendingResolution = kDefaultNonBendingResolution;
- fChamberThicknessInX0 = kDefaultChamberThicknessInX0;
- fSimpleBValue = kDefaultSimpleBValue;
- fSimpleBLength = kDefaultSimpleBLength;
- fSimpleBPosition = kDefaultSimpleBPosition;
- fRecGeantHits = kDefaultRecGeantHits;
- fEfficiency = kDefaultEfficiency;
- fPrintLevel = kDefaultPrintLevel;
+ fBendingResolution = fgkDefaultBendingResolution;
+ fNonBendingResolution = fgkDefaultNonBendingResolution;
+ fChamberThicknessInX0 = fgkDefaultChamberThicknessInX0;
+ fSimpleBValue = fgkDefaultSimpleBValue;
+ fSimpleBLength = fgkDefaultSimpleBLength;
+ fSimpleBPosition = fgkDefaultSimpleBPosition;
+ fRecTrackRefHits = fgkDefaultRecTrackRefHits;
+ fEfficiency = fgkDefaultEfficiency;
+ fPrintLevel = fgkDefaultPrintLevel;
return;
}
//__________________________________________________________________________
-Double_t AliMUONEventReconstructor::GetImpactParamFromBendingMomentum(Double_t BendingMomentum)
+Double_t AliMUONEventReconstructor::GetImpactParamFromBendingMomentum(Double_t BendingMomentum) const
{
// Returns impact parameter at vertex in bending plane (cm),
// from the signed bending momentum "BendingMomentum" in bending plane (GeV/c),
}
//__________________________________________________________________________
-Double_t AliMUONEventReconstructor::GetBendingMomentumFromImpactParam(Double_t ImpactParam)
+Double_t AliMUONEventReconstructor::GetBendingMomentumFromImpactParam(Double_t ImpactParam) const
{
// Returns signed bending momentum in bending plane (GeV/c),
// the sign being the sign of the charge for particles moving forward in Z,
}
//__________________________________________________________________________
-void AliMUONEventReconstructor::SetBkgGeantFile(Text_t *BkgGeantFileName)
+void AliMUONEventReconstructor::SetBkgTrackRefFile(Text_t *BkgTrackRefFileName)
{
- // Set background file ... for GEANT hits
+ // Set background file ... for track ref. hits
// Must be called after having loaded the firts signal event
if (fPrintLevel >= 0) {
- cout << "Enter SetBkgGeantFile with BkgGeantFileName ``"
- << BkgGeantFileName << "''" << endl;}
- if (strlen(BkgGeantFileName)) {
- // BkgGeantFileName not empty: try to open the file
+ cout << "Enter SetBkgTrackRefFile with BkgTrackRefFileName ``"
+ << BkgTrackRefFileName << "''" << endl;}
+ if (strlen(BkgTrackRefFileName)) {
+ // BkgTrackRefFileName not empty: try to open the file
if (fPrintLevel >= 2) {cout << "Before File(Bkg)" << endl; gDirectory->Dump();}
- fBkgGeantFile = new TFile(BkgGeantFileName);
+ fBkgTrackRefFile = new TFile(BkgTrackRefFileName);
if (fPrintLevel >= 2) {cout << "After File(Bkg)" << endl; gDirectory->Dump();}
- if (fBkgGeantFile-> IsOpen()) {
+ if (fBkgTrackRefFile-> IsOpen()) {
if (fPrintLevel >= 0) {
- cout << "Background for GEANT hits in file: ``" << BkgGeantFileName
+ cout << "Background for Track ref. hits in file: ``" << BkgTrackRefFileName
<< "'' successfully opened" << endl;}
}
else {
- cout << "Background for GEANT hits in file: " << BkgGeantFileName << endl;
+ cout << "Background for Track Ref. hits in file: " << BkgTrackRefFileName << endl;
cout << "NOT FOUND: EXIT" << endl;
exit(0); // right instruction for exit ????
}
// Arrays for "particles" and "hits"
- fBkgGeantParticles = new TClonesArray("TParticle", 200);
- fBkgGeantHits = new TClonesArray("AliMUONHit", 2000);
+ fBkgTrackRefParticles = new TClonesArray("TParticle", 200);
// Event number to -1 for initialization
- fBkgGeantEventNumber = -1;
+ fBkgTrackRefEventNumber = -1;
// Back to the signal file:
// first signal event must have been loaded previously,
// otherwise, Segmentation violation at the next instruction
}
//__________________________________________________________________________
-void AliMUONEventReconstructor::NextBkgGeantEvent(void)
+void AliMUONEventReconstructor::NextBkgTrackRefEvent(void)
{
- // Get next event in background file for GEANT hits
+ // Get next event in background file for track ref. hits
// Goes back to event number 0 when end of file is reached
char treeName[20];
- TBranch *branch;
if (fPrintLevel >= 0) {
- cout << "Enter NextBkgGeantEvent" << endl;}
+ cout << "Enter NextBkgTrackRefEvent" << endl;}
// Clean previous event
- if(fBkgGeantTK) delete fBkgGeantTK;
- fBkgGeantTK = NULL;
- if(fBkgGeantParticles) fBkgGeantParticles->Clear();
- if(fBkgGeantTH) delete fBkgGeantTH;
- fBkgGeantTH = NULL;
- if(fBkgGeantHits) fBkgGeantHits->Clear();
+ if(fBkgTrackRefTK) delete fBkgTrackRefTK;
+ fBkgTrackRefTK = NULL;
+ if(fBkgTrackRefParticles) fBkgTrackRefParticles->Clear();
+ if(fBkgTrackRefTTR) delete fBkgTrackRefTTR;
+ fBkgTrackRefTTR = NULL;
// Increment event number
- fBkgGeantEventNumber++;
+ fBkgTrackRefEventNumber++;
// Get access to Particles and Hits for event from background file
if (fPrintLevel >= 2) {cout << "Before cd(Bkg)" << endl; gDirectory->Dump();}
- fBkgGeantFile->cd();
+ fBkgTrackRefFile->cd();
if (fPrintLevel >= 2) {cout << "After cd(Bkg)" << endl; gDirectory->Dump();}
// Particles: TreeK for event and branch "Particles"
- sprintf(treeName, "TreeK%d", fBkgGeantEventNumber);
- fBkgGeantTK = (TTree*)gDirectory->Get(treeName);
- if (!fBkgGeantTK) {
+ sprintf(treeName, "TreeK%d", fBkgTrackRefEventNumber);
+ fBkgTrackRefTK = (TTree*)gDirectory->Get(treeName);
+ if (!fBkgTrackRefTK) {
if (fPrintLevel >= 0) {
cout << "Cannot find Kine Tree for background event: " <<
- fBkgGeantEventNumber << endl;
+ fBkgTrackRefEventNumber << endl;
cout << "Goes back to event 0" << endl;
}
- fBkgGeantEventNumber = 0;
- sprintf(treeName, "TreeK%d", fBkgGeantEventNumber);
- fBkgGeantTK = (TTree*)gDirectory->Get(treeName);
- if (!fBkgGeantTK) {
- cout << "ERROR: cannot find Kine Tree for background event: " <<
- fBkgGeantEventNumber << endl;
- exit(0);
+ fBkgTrackRefEventNumber = 0;
+ sprintf(treeName, "TreeK%d", fBkgTrackRefEventNumber);
+ fBkgTrackRefTK = (TTree*)gDirectory->Get(treeName);
+ if (!fBkgTrackRefTK) {
+ AliError(Form("cannot find Kine Tree for background event: %d",fBkgTrackRefEventNumber));
+ exit(0);
}
}
- if (fBkgGeantTK)
- fBkgGeantTK->SetBranchAddress("Particles", &fBkgGeantParticles);
- fBkgGeantTK->GetEvent(0); // why event 0 ???? necessary ????
+ if (fBkgTrackRefTK)
+ fBkgTrackRefTK->SetBranchAddress("Particles", &fBkgTrackRefParticles);
+ fBkgTrackRefTK->GetEvent(0); // why event 0 ???? necessary ????
// Hits: TreeH for event and branch "MUON"
- sprintf(treeName, "TreeH%d", fBkgGeantEventNumber);
- fBkgGeantTH = (TTree*)gDirectory->Get(treeName);
- if (!fBkgGeantTH) {
- cout << "ERROR: cannot find Hits Tree for background event: " <<
- fBkgGeantEventNumber << endl;
+ sprintf(treeName, "TreeTR%d", fBkgTrackRefEventNumber);
+ fBkgTrackRefTTR = (TTree*)gDirectory->Get(treeName);
+ if (!fBkgTrackRefTTR) {
+ AliError(Form("cannot find Hits Tree for background event: %d",fBkgTrackRefEventNumber));
exit(0);
}
- if (fBkgGeantTH && fBkgGeantHits) {
- branch = fBkgGeantTH->GetBranch("MUON");
- if (branch) branch->SetAddress(&fBkgGeantHits);
- }
- fBkgGeantTH->GetEntries(); // necessary ????
+ fBkgTrackRefTTR->GetEntries(); // necessary ????
// Back to the signal file
((gAlice->TreeK())->GetCurrentFile())->cd();
if (fPrintLevel >= 2) {cout << "After cd(gAlice)" << endl; gDirectory->Dump();}
MakeEventToBeReconstructed();
MakeSegments();
MakeTracks();
+ if (fMUONData->IsTriggerTrackBranchesInTree())
+ ValidateTracksWithTrigger();
+
+ // Add tracks to MUON data container
+ for(Int_t i=0; i<GetNRecTracks(); i++) {
+ AliMUONTrack * track = (AliMUONTrack*) GetRecTracksPtr()->At(i);
+ fMUONData->AddRecTrack(*track);
+ }
+
+ return;
+}
+
+//__________________________________________________________________________
+void AliMUONEventReconstructor::EventReconstructTrigger(void)
+{
+ // To reconstruct one event
+ if (fPrintLevel >= 1) cout << "enter EventReconstructTrigger" << endl;
+ MakeTriggerTracks();
return;
}
// and the index of the first HitForRec per chamber
if (fHitsForRecPtr) fHitsForRecPtr->Clear();
fNHitsForRec = 0;
- for (Int_t ch = 0; ch < kMaxMuonTrackingChambers; ch++)
+ for (Int_t ch = 0; ch < fgkMaxMuonTrackingChambers; ch++)
fNHitsForRecPerChamber[ch] = fIndexOfFirstHitForRecPerChamber[ch] = 0;
return;
}
{
// To reset the TClonesArray of segments and the number of Segments
// for all stations
- for (Int_t st = 0; st < kMaxMuonTrackingStations; st++) {
+ for (Int_t st = 0; st < fgkMaxMuonTrackingStations; st++) {
if (fSegmentsPtr[st]) fSegmentsPtr[st]->Clear();
fNSegments[st] = 0;
}
return;
}
+ //__________________________________________________________________________
+void AliMUONEventReconstructor::ResetTriggerTracks(void)
+{
+ // To reset the TClonesArray of reconstructed trigger tracks
+ if (fRecTriggerTracksPtr) fRecTriggerTracksPtr->Delete();
+ // Delete in order that the Track destructors are called,
+ // hence the space for the TClonesArray of pointers to TrackHit's is freed
+ fNRecTriggerTracks = 0;
+ return;
+}
+
//__________________________________________________________________________
void AliMUONEventReconstructor::ResetTrackHits(void)
{
void AliMUONEventReconstructor::MakeEventToBeReconstructed(void)
{
// To make the list of hits to be reconstructed,
- // either from the GEANT hits or from the raw clusters
+ // either from the track ref. hits or from the raw clusters
// according to the parameter set for the reconstructor
+// TString evfoldname = AliConfig::GetDefaultEventFolderName();//to be interfaced properly
+
+// AliRunLoader* rl = AliRunLoader::GetRunLoader(evfoldname);
+// if (rl == 0x0)
+// {
+// Error("MakeEventToBeReconstructed",
+// "Can not find Run Loader in Event Folder named %s.",
+// evfoldname.Data());
+// return;
+// }
+// AliLoader* gime = rl->GetLoader("MUONLoader");
+// if (gime == 0x0)
+// {
+// Error("MakeEventToBeReconstructed","Can not get MUON Loader from Run Loader.");
+// return;
+// }
+ AliRunLoader *runLoader = fLoader->GetRunLoader();
+
if (fPrintLevel >= 1) cout << "enter MakeEventToBeReconstructed" << endl;
ResetHitsForRec();
- if (fRecGeantHits == 1) {
- // Reconstruction from GEANT hits
+ if (fRecTrackRefHits == 1) {
+ // Reconstruction from track ref. hits
// Back to the signal file
- ((gAlice->TreeK())->GetCurrentFile())->cd();
- // Signal hits
- // AliMUON *MUON = (AliMUON*) gAlice->GetModule("MUON"); // necessary ????
- // Security on MUON ????
- AddHitsForRecFromGEANT(gAlice->TreeH());
+ TTree* treeTR = runLoader->TreeTR();
+ if (treeTR == 0x0)
+ {
+ Int_t retval = runLoader->LoadTrackRefs("READ");
+ if ( retval)
+ {
+ AliError("Error occured while loading hits.");
+ return;
+ }
+ treeTR = runLoader->TreeTR();
+ if (treeTR == 0x0)
+ {
+ AliError("Can not get TreeTR");
+ return;
+ }
+ }
+
+ AddHitsForRecFromTrackRef(treeTR,1);
+
// Background hits
- AddHitsForRecFromBkgGEANT(fBkgGeantTH, fBkgGeantHits);
+ AddHitsForRecFromTrackRef(fBkgTrackRefTTR,0);
// Sort HitsForRec in increasing order with respect to chamber number
SortHitsForRecWithIncreasingChamber();
}
// Security on MUON ????
// TreeR assumed to be be "prepared" in calling function
// by "MUON->GetTreeR(nev)" ????
- TTree *treeR = gAlice->TreeR();
+ TTree *treeR = fLoader->TreeR();
+ fMUONData->SetTreeAddress("RC");
AddHitsForRecFromRawClusters(treeR);
// No sorting: it is done automatically in the previous function
}
if (fPrintLevel >= 10) {
cout << "end of MakeEventToBeReconstructed" << endl;
cout << "NHitsForRec: " << fNHitsForRec << endl;
- for (Int_t ch = 0; ch < kMaxMuonTrackingChambers; ch++) {
+ for (Int_t ch = 0; ch < fgkMaxMuonTrackingChambers; ch++) {
cout << "chamber(0...): " << ch
<< " NHitsForRec: " << fNHitsForRecPerChamber[ch]
<< " index(first HitForRec): " << fIndexOfFirstHitForRecPerChamber[ch]
}
//__________________________________________________________________________
-void AliMUONEventReconstructor::AddHitsForRecFromGEANT(TTree *TH)
+void AliMUONEventReconstructor::AddHitsForRecFromTrackRef(TTree *TTR, Int_t signal)
{
// To add to the list of hits for reconstruction
- // the GEANT signal hits from a hit tree TH.
- if (fPrintLevel >= 2)
- cout << "enter AddHitsForRecFromGEANT with TH: " << TH << endl;
- if (TH == NULL) return;
- AliMUON *pMUON = (AliMUON*) gAlice->GetModule("MUON"); // necessary ????
- // Security on MUON ????
- // See whether it could be the same for signal and background ????
- // Loop over tracks in tree
- Int_t ntracks = (Int_t) TH->GetEntries();
+ // the signal hits from a track reference tree TreeTR.
+ TClonesArray *listOfTrackRefs = NULL;
+ AliTrackReference *trackRef;
+
+ Int_t track = 0;
if (fPrintLevel >= 2)
- cout << "ntracks: " << ntracks << endl;
- for (Int_t track = 0; track < ntracks; track++) {
- gAlice->ResetHits();
- TH->GetEvent(track);
- // Loop over hits
- Int_t hit = 0;
- for (AliMUONHit* mHit = (AliMUONHit*) pMUON->FirstHit(-1);
- mHit;
- mHit = (AliMUONHit*) pMUON->NextHit(), hit++) {
- NewHitForRecFromGEANT(mHit,track, hit, 1);
- } // end of hit loop
- } // end of track loop
- return;
-}
+ cout << "enter AddHitsForRecFromTrackRef with TTR: " << TTR << endl;
+ if (TTR == NULL) return;
+
+ listOfTrackRefs = CleanTrackRefs(TTR);
+
+ Int_t ntracks = listOfTrackRefs->GetEntriesFast();
- //__________________________________________________________________________
-void AliMUONEventReconstructor::AddHitsForRecFromBkgGEANT(TTree *TH, TClonesArray *Hits)
-{
- // To add to the list of hits for reconstruction
- // the GEANT background hits from a hit tree TH and a pointer Hits to a hit list.
- // How to have only one function "AddHitsForRecFromGEANT" ????
- if (fPrintLevel >= 2)
- cout << "enter AddHitsForRecFromBkgGEANT with TH: " << TH << endl;
- if (TH == NULL) return;
- // Loop over tracks in tree
- Int_t ntracks = (Int_t) TH->GetEntries();
if (fPrintLevel >= 2)
cout << "ntracks: " << ntracks << endl;
- for (Int_t track = 0; track < ntracks; track++) {
- if (Hits) Hits->Clear();
- TH->GetEvent(track);
- // Loop over hits
- for (Int_t hit = 0; hit < Hits->GetEntriesFast(); hit++) {
- NewHitForRecFromGEANT((AliMUONHit*) (*Hits)[hit], track, hit, 0);
- } // end of hit loop
- } // end of track loop
+
+ for (Int_t index = 0; index < ntracks; index++) {
+ trackRef = (AliTrackReference*) listOfTrackRefs->At(index);
+ track = trackRef->GetTrack();
+
+ NewHitForRecFromTrackRef(trackRef,track,signal);
+ } // end of track ref.
+
+ listOfTrackRefs->Delete();
+ delete listOfTrackRefs;
return;
}
+
//__________________________________________________________________________
-AliMUONHitForRec* AliMUONEventReconstructor::NewHitForRecFromGEANT(AliMUONHit* Hit, Int_t TrackNumber, Int_t HitNumber, Int_t Signal)
+AliMUONHitForRec* AliMUONEventReconstructor::NewHitForRecFromTrackRef(AliTrackReference* Hit, Int_t TrackNumber, Int_t Signal)
{
- // To make a new hit for reconstruction from a GEANT hit pointed to by "Hit",
- // with hit number "HitNumber" in the track numbered "TrackNumber",
+ // To make a new hit for reconstruction from a track ref. hit pointed to by "Hit",
+ // with the track numbered "TrackNumber",
// either from signal ("Signal" = 1) or background ("Signal" = 0) event.
// Selects hits in tracking (not trigger) chambers.
// Takes into account the efficiency (fEfficiency)
// like in Fortran TRACKF_STAT.
AliMUONHitForRec* hitForRec;
Double_t bendCoor, nonBendCoor, radius;
- Int_t chamber = Hit->Chamber() - 1; // chamber(0...)
+ Int_t chamber = AliMUONConstants::ChamberNumber(Hit->Z()); // chamber(0...)
+ if (chamber < 0) return NULL;
// only in tracking chambers (fChamber starts at 1)
- if (chamber >= kMaxMuonTrackingChambers) return NULL;
+ if (chamber >= fgkMaxMuonTrackingChambers) return NULL;
// only if hit is efficient (keep track for checking ????)
if (gRandom->Rndm() > fEfficiency) return NULL;
// only if radius between RMin and RMax
radius = TMath::Sqrt((bendCoor * bendCoor) + (nonBendCoor * nonBendCoor));
// This cut is not needed with a realistic chamber geometry !!!!
// if ((radius < fRMin[chamber]) || (radius > fRMax[chamber])) return NULL;
- // new AliMUONHitForRec from GEANT hit and increment number of AliMUONHitForRec's
+ // new AliMUONHitForRec from track ref. hit and increment number of AliMUONHitForRec's
hitForRec = new ((*fHitsForRecPtr)[fNHitsForRec]) AliMUONHitForRec(Hit);
fNHitsForRec++;
// add smearing from resolution
// resolution: angular effect to be added here ????
hitForRec->SetBendingReso2(fBendingResolution * fBendingResolution);
hitForRec->SetNonBendingReso2(fNonBendingResolution * fNonBendingResolution);
- // GEANT track info
- hitForRec->SetHitNumber(HitNumber);
- hitForRec->SetTHTrack(TrackNumber);
- hitForRec->SetGeantSignal(Signal);
+ // track ref. info
+ hitForRec->SetTTRTrack(TrackNumber);
+ hitForRec->SetTrackRefSignal(Signal);
if (fPrintLevel >= 10) {
- cout << "track: " << TrackNumber << " hit: " << HitNumber << endl;
+ cout << "track: " << TrackNumber << endl;
Hit->Dump();
cout << "AliMUONHitForRec number (1...): " << fNHitsForRec << endl;
hitForRec->Dump();}
return hitForRec;
}
-
+ //__________________________________________________________________________
+TClonesArray* AliMUONEventReconstructor::CleanTrackRefs(TTree *treeTR)
+{
+ // Make hits from track ref..
+ // 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)
+
+ AliTrackReference *trackReference;
+ Float_t x1, y1, z1, pX1, pY1, pZ1;
+ Float_t x2, y2, z2, pX2, pY2, pZ2;
+ Int_t track1, track2;
+ Int_t nRec = 0;
+ Float_t maxGasGap = 1.; // cm
+ Int_t iHit1 = 0;
+ Int_t iHitMin = 0;
+
+ AliTrackReference *trackReferenceNew = new AliTrackReference();
+
+ TClonesArray* trackRefs = new TClonesArray("AliTrackReference", 10);
+ TClonesArray* cleanTrackRefs = new TClonesArray("AliTrackReference", 10);
+
+ if (treeTR == NULL) return NULL;
+ TBranch* branch = treeTR->GetBranch("MUON");
+ if (branch == NULL) return NULL;
+ branch->SetAddress(&trackRefs);
+
+ Int_t nTrack = (Int_t)treeTR->GetEntries();
+ for (Int_t iTrack = 0; iTrack < nTrack; iTrack++) {
+ treeTR->GetEntry(iTrack);
+ iHitMin = 0;
+ iHit1 = 0;
+ while (iHit1 < trackRefs->GetEntries()) {
+ trackReference = (AliTrackReference*)trackRefs->At(iHit1);
+ x1 = trackReference->X();
+ y1 = trackReference->Y();
+ z1 = trackReference->Z();
+ pX1 = trackReference->Px();
+ pY1 = trackReference->Py();
+ pZ1 = trackReference->Pz();
+ track1 = trackReference->GetTrack();
+ nRec = 1;
+ iHitMin = iHit1+1;
+ for (Int_t iHit2 = iHit1+1; iHit2 < trackRefs->GetEntries(); iHit2++) {
+ trackReference = (AliTrackReference*)trackRefs->At(iHit2);
+ x2 = trackReference->X();
+ y2 = trackReference->Y();
+ z2 = trackReference->Z();
+ pX2 = trackReference->Px();
+ pY2 = trackReference->Py();
+ pZ2 = trackReference->Pz();
+ track2 = trackReference->GetTrack();
+ if (track2 == track1 && TMath::Abs(z2-z1) < maxGasGap ) {
+ nRec++;
+ x1 += x2;
+ y1 += y2;
+ z1 += z2;
+ pX1 += pX2;
+ pY1 += pY2;
+ pZ1 += pZ2;
+ iHitMin = iHit2+1;
+ }
+
+ } // for iHit2
+ x1 /= (Float_t)nRec;
+ y1 /= (Float_t)nRec;
+ z1 /= (Float_t)nRec;
+ pX1 /= (Float_t)nRec;
+ pY1 /= (Float_t)nRec;
+ pZ1 /= (Float_t)nRec;
+ trackReferenceNew->SetPosition(x1,y1,z1);
+ trackReferenceNew->SetMomentum(pX1,pY1,pZ1);
+ trackReferenceNew->SetTrack(track1);
+ {new ((*cleanTrackRefs)[cleanTrackRefs->GetEntriesFast()]) AliTrackReference(*trackReferenceNew);}
+ iHit1 = iHitMin;
+ } // while iHit1
+ } // for track
+
+ trackRefs->Delete();
+ delete trackRefs;
+ delete trackReferenceNew;
+ return cleanTrackRefs;
+
+}
//__________________________________________________________________________
void AliMUONEventReconstructor::SortHitsForRecWithIncreasingChamber()
{
// Update the information for HitsForRec per chamber too.
Int_t ch, nhits, prevch;
fHitsForRecPtr->Sort();
- for (ch = 0; ch < kMaxMuonTrackingChambers; ch++) {
+ for (ch = 0; ch < fgkMaxMuonTrackingChambers; ch++) {
fNHitsForRecPerChamber[ch] = 0;
fIndexOfFirstHitForRecPerChamber[ch] = 0;
}
// AliMUON *MUON = (AliMUON*) gAlice->GetModule("MUON"); // necessary ????
// // Security on MUON ????
// // Loop over tracking chambers
-// for (Int_t ch = 0; ch < kMaxMuonTrackingChambers; ch++) {
+// for (Int_t ch = 0; ch < fgkMaxMuonTrackingChambers; ch++) {
// // number of HitsForRec to 0 for the chamber
// fNHitsForRecPerChamber[ch] = 0;
// // index of first HitForRec for the chamber
// // more information into HitForRec
// hitForRec->SetChamberNumber(ch);
// hitForRec->SetHitNumber(cor);
-// // Z coordinate of the chamber (cm) with sign opposite to GEANT sign
+// // Z coordinate of the chamber (cm) with sign opposite to TRACKREF sign
// // could (should) be more exact from chamber geometry ????
// hitForRec->SetZ(-(&(MUON->Chamber(ch)))->Z());
// if (fPrintLevel >= 10) {
// on the radius between RMin and RMax.
AliMUONHitForRec *hitForRec;
AliMUONRawCluster *clus;
- Int_t iclus, nclus;
+ Int_t iclus, nclus, nTRentries;
TClonesArray *rawclusters;
if (fPrintLevel >= 1) cout << "enter AddHitsForRecFromRawClusters" << endl;
- AliMUON *pMUON = (AliMUON*) gAlice->GetModule("MUON"); // necessary ????
- // Security on MUON ????
+
+// TString evfoldname = AliConfig::GetDefaultEventFolderName();//to be interfaced properly
+// AliRunLoader* rl = AliRunLoader::GetRunLoader(evfoldname);
+// if (rl == 0x0)
+// {
+// Error("MakeEventToBeReconstructed",
+// "Can not find Run Loader in Event Folder named %s.",
+// evfoldname.Data());
+// return;
+// }
+// AliLoader* gime = rl->GetLoader("MUONLoader");
+// if (gime == 0x0)
+// {
+// Error("MakeEventToBeReconstructed","Can not get MUON Loader from Run Loader.");
+// return;
+// }
+// // Loading AliRun master
+// rl->LoadgAlice();
+// gAlice = rl->GetAliRun();
+
+ // Loading MUON subsystem
+ // AliMUON * pMUON = (AliMUON *) gAlice->GetDetector("MUON");
+
+ nTRentries = Int_t(TR->GetEntries());
+ if (nTRentries != 1) {
+ AliError(Form("nTRentries = %d not equal to 1 ",nTRentries));
+ exit(0);
+ }
+ fLoader->TreeR()->GetEvent(0); // only one entry
+
// Loop over tracking chambers
- for (Int_t ch = 0; ch < kMaxMuonTrackingChambers; ch++) {
+ for (Int_t ch = 0; ch < fgkMaxMuonTrackingChambers; ch++) {
// number of HitsForRec to 0 for the chamber
fNHitsForRecPerChamber[ch] = 0;
// index of first HitForRec for the chamber
if (ch == 0) fIndexOfFirstHitForRecPerChamber[ch] = 0;
else fIndexOfFirstHitForRecPerChamber[ch] = fNHitsForRec;
- rawclusters = pMUON->RawClustAddress(ch);
- pMUON->ResetRawClusters();
- TR->GetEvent((Int_t) (TR->GetEntries()) - 1); // to be checked ????
+ rawclusters =fMUONData->RawClusters(ch);
+// pMUON->ResetRawClusters();
+// TR->GetEvent((Int_t) (TR->GetEntries()) - 1); // to be checked ????
nclus = (Int_t) (rawclusters->GetEntries());
// Loop over (cathode correlated) raw clusters
for (iclus = 0; iclus < nclus; iclus++) {
hitForRec->SetChamberNumber(ch);
hitForRec->SetHitNumber(iclus);
// Z coordinate of the raw cluster (cm)
- hitForRec->SetZ(clus->fZ[0]);
+ hitForRec->SetZ(clus->GetZ(0));
if (fPrintLevel >= 10) {
cout << "chamber (0...): " << ch <<
" raw cluster (0...): " << iclus << endl;
hitForRec->Dump();}
} // end of cluster loop
} // end of chamber loop
+ SortHitsForRecWithIncreasingChamber(); //AZ
return;
}
if (fPrintLevel >= 1) cout << "enter MakeSegments" << endl;
ResetSegments();
// Loop over stations
- for (Int_t st = 0; st < kMaxMuonTrackingStations; st++)
+ Int_t nb = (fTrackMethod == 2) ? 3 : 0; //AZ
+ //AZ for (Int_t st = 0; st < fgkMaxMuonTrackingStations; st++)
+ for (Int_t st = nb; st < fgkMaxMuonTrackingStations; st++) //AZ
MakeSegmentsPerStation(st);
if (fPrintLevel >= 10) {
cout << "end of MakeSegments" << endl;
- for (Int_t st = 0; st < kMaxMuonTrackingStations; st++) {
+ for (Int_t st = 0; st < fgkMaxMuonTrackingStations; st++) {
cout << "station(0...): " << st
<< " Segments: " << fNSegments[st]
<< endl;
Bool_t last2st;
Double_t bendingSlope, distBend, distNonBend, extBendCoor, extNonBendCoor,
impactParam = 0., maxImpactParam = 0., minImpactParam = 0.; // =0 to avoid compilation warnings.
- AliMUON *pMUON = (AliMUON*) gAlice->GetModule("MUON"); // necessary ????
if (fPrintLevel >= 1)
cout << "enter MakeSegmentsPerStation (0...) " << Station << endl;
// first and second chambers (0...) in the station
else last2st = kFALSE;
// extrapolation factor from Z of first chamber to Z of second chamber
// dZ to be changed to take into account fine structure of chambers ????
- Double_t extrapFact =
- (&(pMUON->Chamber(ch2)))->Z() / (&(pMUON->Chamber(ch1)))->Z();
+ Double_t extrapFact;
// index for current segment
Int_t segmentIndex = 0;
// Loop over HitsForRec in the first chamber of the station
// extrapolation,
// on the straight line joining the HitForRec to the vertex (0,0,0),
// to the Z of the second chamber of the station
- extBendCoor = extrapFact * hit1Ptr->GetBendingCoor();
- extNonBendCoor = extrapFact * hit1Ptr->GetNonBendingCoor();
// Loop over HitsForRec in the second chamber of the station
for (Int_t hit2 = fIndexOfFirstHitForRecPerChamber[ch2];
hit2 < fIndexOfFirstHitForRecPerChamber[ch2] + fNHitsForRecPerChamber[ch2];
// absolute values of distances, in bending and non bending planes,
// between the HitForRec in the second chamber
// and the previous extrapolation
+ extrapFact = hit2Ptr->GetZ()/ hit1Ptr->GetZ();
+ extBendCoor = extrapFact * hit1Ptr->GetBendingCoor();
+ extNonBendCoor = extrapFact * hit1Ptr->GetNonBendingCoor();
distBend = TMath::Abs(hit2Ptr->GetBendingCoor() - extBendCoor);
distNonBend = TMath::Abs(hit2Ptr->GetNonBendingCoor() - extNonBendCoor);
if (last2st) {
(hit1Ptr->GetZ() - hit2Ptr->GetZ());
// absolute value of impact parameter
impactParam =
- TMath::Abs(hit1Ptr->GetBendingCoor() - hit2Ptr->GetZ() * bendingSlope);
+ TMath::Abs(hit1Ptr->GetBendingCoor() - hit1Ptr->GetZ() * bendingSlope);
}
// check for distances not too large,
// and impact parameter not too big if stations downstream of the dipole.
// The order may be important for the following Reset's
ResetTracks();
ResetTrackHits();
- // Look for candidates from at least 3 aligned points in stations(1..) 4 and 5
- MakeTrackCandidates();
- // Follow tracks in stations(1..) 3, 2 and 1
- FollowTracks();
- // Remove double tracks
- RemoveDoubleTracks();
+ if (fTrackMethod == 2) { //AZ - Kalman filter
+ MakeTrackCandidatesK();
+ if (fRecTracksPtr->GetEntriesFast() == 0) return;
+ // Follow tracks in stations(1..) 3, 2 and 1
+ FollowTracksK();
+ // Remove double tracks
+ RemoveDoubleTracksK();
+ // Propagate tracks to the vertex thru absorber
+ GoToVertex();
+ // Fill AliMUONTrack data members
+ FillMUONTrack();
+ } else {
+ // Look for candidates from at least 3 aligned points in stations(1..) 4 and 5
+ MakeTrackCandidates();
+ // Follow tracks in stations(1..) 3, 2 and 1
+ FollowTracks();
+ // Remove double tracks
+ RemoveDoubleTracks();
+ UpdateTrackParamAtHit();
+ UpdateHitForRecAtHit();
+ }
return;
}
+ //__________________________________________________________________________
+void AliMUONEventReconstructor::ValidateTracksWithTrigger(void)
+{
+ // Try to match track from tracking system with trigger track
+ AliMUONTrack *track;
+ TClonesArray *recTriggerTracks;
+
+ fMUONData->ResetRecTriggerTracks();
+ fMUONData->SetTreeAddress("RL");
+ fMUONData->GetRecTriggerTracks();
+ recTriggerTracks = fMUONData->RecTriggerTracks();
+
+ track = (AliMUONTrack*) fRecTracksPtr->First();
+ while (track) {
+ track->MatchTriggerTrack(recTriggerTracks);
+ track = (AliMUONTrack*) fRecTracksPtr->After(track);
+ }
+
+}
+
+ //__________________________________________________________________________
+Bool_t AliMUONEventReconstructor::MakeTriggerTracks(void)
+{
+ // To make the trigger tracks from Local Trigger
+ if (fPrintLevel >= 1) cout << "enter MakeTriggerTracks" << endl;
+ // ResetTriggerTracks();
+
+ Int_t nTRentries;
+ Long_t gloTrigPat;
+ TClonesArray *localTrigger;
+ TClonesArray *globalTrigger;
+ AliMUONLocalTrigger *locTrg;
+ AliMUONGlobalTrigger *gloTrg;
+ AliMUONTriggerCircuit *circuit;
+ AliMUONTriggerTrack *recTriggerTrack = 0;
+
+ TTree* treeR = fLoader->TreeR();
+
+ // Loading MUON subsystem
+ AliMUON * pMUON = (AliMUON *) gAlice->GetDetector("MUON");
+
+ nTRentries = Int_t(treeR->GetEntries());
+
+ treeR->GetEvent(0); // only one entry
+
+ if (!(fMUONData->IsTriggerBranchesInTree())) {
+ AliWarning(Form("Trigger information is not avalaible, nTRentries = %d not equal to 1",nTRentries));
+ return kFALSE;
+ }
+
+ fMUONData->SetTreeAddress("TC");
+ fMUONData->GetTrigger();
+
+ // global trigger for trigger pattern
+ gloTrigPat = 0;
+ globalTrigger = fMUONData->GlobalTrigger();
+ gloTrg = (AliMUONGlobalTrigger*)globalTrigger->UncheckedAt(0);
+ if (gloTrg) {
+ if (gloTrg->SinglePlusLpt()) gloTrigPat|= 0x1;
+ if (gloTrg->SinglePlusHpt()) gloTrigPat|= 0x2;
+ if (gloTrg->SinglePlusApt()) gloTrigPat|= 0x4;
+
+ if (gloTrg->SingleMinusLpt()) gloTrigPat|= 0x8;
+ if (gloTrg->SingleMinusHpt()) gloTrigPat|= 0x10;
+ if (gloTrg->SingleMinusApt()) gloTrigPat|= 0x20;
+
+ if (gloTrg->SingleUndefLpt()) gloTrigPat|= 0x40;
+ if (gloTrg->SingleUndefHpt()) gloTrigPat|= 0x80;
+ if (gloTrg->SingleUndefApt()) gloTrigPat|= 0x100;
+
+ if (gloTrg->PairUnlikeLpt()) gloTrigPat|= 0x200;
+ if (gloTrg->PairUnlikeHpt()) gloTrigPat|= 0x400;
+ if (gloTrg->PairUnlikeApt()) gloTrigPat|= 0x800;
+
+ if (gloTrg->PairLikeLpt()) gloTrigPat|= 0x1000;
+ if (gloTrg->PairLikeHpt()) gloTrigPat|= 0x2000;
+ if (gloTrg->PairLikeApt()) gloTrigPat|= 0x4000;
+ }
+
+
+
+ // local trigger for tracking
+ localTrigger = fMUONData->LocalTrigger();
+ Int_t nlocals = (Int_t) (localTrigger->GetEntries());
+ Float_t z11 = ( &(pMUON->Chamber(10)) )->Z();
+ Float_t z21 = ( &(pMUON->Chamber(12)) )->Z();
+
+ for (Int_t i=0; i<nlocals; i++) { // loop on Local Trigger
+ locTrg = (AliMUONLocalTrigger*)localTrigger->UncheckedAt(i);
+ circuit = &(pMUON->TriggerCircuit(locTrg->LoCircuit()));
+ Float_t y11 = circuit->GetY11Pos(locTrg->LoStripX());
+ Int_t stripX21 = locTrg->LoStripX()+locTrg->LoDev()+1;
+ Float_t y21 = circuit->GetY21Pos(stripX21);
+ Float_t x11 = circuit->GetX11Pos(locTrg->LoStripY());
+ Float_t thetax = TMath::ATan2( x11 , z11 );
+ Float_t thetay = TMath::ATan2( (y21-y11) , (z21-z11) );
+
+ recTriggerTrack = new AliMUONTriggerTrack(x11,y11,thetax,thetay,gloTrigPat,this);
+ // since static statement does not work, set gloTrigPat for each track
+
+ // fNRecTriggerTracks++;
+ fMUONData->AddRecTriggerTrack(*recTriggerTrack);
+ } // end of loop on Local Trigger
+ return kTRUE;
+}
+
//__________________________________________________________________________
Int_t AliMUONEventReconstructor::MakeTrackCandidatesWithTwoSegments(AliMUONSegment *BegSegment)
{
// the first segment being pointed to by "BegSegment".
// Returns the number of such track candidates.
Int_t endStation, iEndSegment, nbCan2Seg;
- AliMUONSegment *endSegment, *extrapSegment;
+ AliMUONSegment *endSegment;
+ AliMUONSegment *extrapSegment = NULL;
AliMUONTrack *recTrack;
Double_t mcsFactor;
if (fPrintLevel >= 1) cout << "enter MakeTrackCandidatesWithTwoSegments" << endl;
GetBendingMomentumFromImpactParam(BegSegment->GetBendingImpact());
mcsFactor = fChamberThicknessInX0 * mcsFactor * mcsFactor;
// linear extrapolation to end station
- extrapSegment =
- BegSegment->CreateSegmentFromLinearExtrapToStation(endStation, mcsFactor);
// number of candidates with 2 segments to 0
nbCan2Seg = 0;
// Loop over segments in the end station
endSegment = (AliMUONSegment*) ((*fSegmentsPtr[endStation])[iEndSegment]);
// test compatibility between current segment and "extrapSegment"
// 4 because 4 quantities in chi2
+ extrapSegment =
+ BegSegment->CreateSegmentFromLinearExtrapToStation(endSegment->GetZ(), mcsFactor);
if ((endSegment->
NormalizedChi2WithSegment(extrapSegment,
fMaxSigma2Distance)) <= 4.0) {
// in stations(1..) 4 and 5,
// the segment being pointed to by "BegSegment".
Int_t ch, ch1, ch2, endStation, iHit, iHitMax, iHitMin, nbCan1Seg1Hit;
- AliMUONHitForRec *extrapHitForRec, *hit;
+ AliMUONHitForRec *extrapHitForRec= NULL;
+ AliMUONHitForRec *hit;
AliMUONTrack *recTrack;
Double_t mcsFactor;
if (fPrintLevel >= 1)
nbCan1Seg1Hit = 0;
// Loop over chambers of the end station
for (ch = ch2; ch >= ch1; ch--) {
- // linear extrapolation to chamber
- extrapHitForRec =
- BegSegment->CreateHitForRecFromLinearExtrapToChamber(ch, mcsFactor);
// limits for the hit index in the loop
iHitMin = fIndexOfFirstHitForRecPerChamber[ch];
iHitMax = iHitMin + fNHitsForRecPerChamber[ch];
hit = (AliMUONHitForRec*) ((*fHitsForRecPtr)[iHit]);
// test compatibility between current HitForRec and "extrapHitForRec"
// 2 because 2 quantities in chi2
+ // linear extrapolation to chamber
+ extrapHitForRec =
+ BegSegment->CreateHitForRecFromLinearExtrapToChamber( hit->GetZ(), mcsFactor);
if ((hit->
NormalizedChi2WithHitForRec(extrapHitForRec,
fMaxSigma2Distance)) <= 2.0) {
}
} // for (iHit = iHitMin;...
delete extrapHitForRec;
+ extrapHitForRec = NULL;
} // for (ch = ch2;...
return nbCan1Seg1Hit;
}
{
// Follow tracks in stations(1..) 3, 2 and 1
// too long: should be made more modular !!!!
- AliMUONHitForRec *bestHit, *extrapHit, *extrapCorrHit, *hit;
- AliMUONSegment *bestSegment, *extrapSegment, *extrapCorrSegment, *segment;
+ AliMUONHitForRec *bestHit, *extrapHit, *hit;
+ AliMUONSegment *bestSegment, *extrapSegment, *segment;
AliMUONTrack *track, *nextTrack;
AliMUONTrackParam *trackParam1, trackParam[2], trackParamVertex;
// -1 to avoid compilation warnings
// extrapolation to station
trackParam1->ExtrapToStation(station, trackParam);
extrapSegment = new AliMUONSegment(); // empty segment
- extrapCorrSegment = new AliMUONSegment(); // empty corrected segment
// multiple scattering factor corresponding to one chamber
// and momentum in bending plane (not total)
mcsFactor = 0.0136 * trackParam1->GetInverseBendingMomentum();
extrapSegment->UpdateFromStationTrackParam
(trackParam, mcsFactor, dZ1, dZ2, dZ3, station,
trackParam1->GetInverseBendingMomentum());
- // same thing for corrected segment
- // better to use copy constructor, after checking that it works properly !!!!
- extrapCorrSegment->SetBendingCoorReso2(fBendingResolution * fBendingResolution);
- extrapCorrSegment->
- SetNonBendingCoorReso2(fNonBendingResolution * fNonBendingResolution);
- extrapCorrSegment->UpdateFromStationTrackParam
- (trackParam, mcsFactor, dZ1, dZ2, dZ3, station,
- trackParam1->GetInverseBendingMomentum());
bestChi2 = 5.0;
bestSegment = NULL;
if (fPrintLevel >= 10) {
segment = (AliMUONSegment*) ((*fSegmentsPtr[station])[iSegment]);
// correction of corrected segment (fBendingCoor and fNonBendingCoor)
// according to real Z value of "segment" and slopes of "extrapSegment"
- extrapCorrSegment->
- SetBendingCoor(extrapSegment->GetBendingCoor() +
- extrapSegment->GetBendingSlope() *
- (segment->GetHitForRec1()->GetZ() -
- (&(pMUON->Chamber(2 * station)))->Z()));
- extrapCorrSegment->
- SetNonBendingCoor(extrapSegment->GetNonBendingCoor() +
- extrapSegment->GetNonBendingSlope() *
- (segment->GetHitForRec1()->GetZ() -
- (&(pMUON->Chamber(2 * station)))->Z()));
+ (&(trackParam[0]))->ExtrapToZ(segment->GetZ());
+ (&(trackParam[1]))->ExtrapToZ(segment->GetZ());
+ extrapSegment->SetBendingCoor((&(trackParam[0]))->GetBendingCoor());
+ extrapSegment->SetNonBendingCoor((&(trackParam[0]))->GetNonBendingCoor());
+ extrapSegment->SetBendingSlope((&(trackParam[0]))->GetBendingSlope());
+ extrapSegment->SetNonBendingSlope((&(trackParam[0]))->GetNonBendingSlope());
chi2 = segment->
- NormalizedChi2WithSegment(extrapCorrSegment, maxSigma2Distance);
+ NormalizedChi2WithSegment(extrapSegment, maxSigma2Distance);
if (chi2 < bestChi2) {
// update best Chi2 and Segment if better found
bestSegment = segment;
}
if (bestSegment) {
// best segment found: add it to track candidate
+ (&(trackParam[0]))->ExtrapToZ(bestSegment->GetZ());
+ (&(trackParam[1]))->ExtrapToZ(bestSegment->GetZ());
track->AddSegment(bestSegment);
// set track parameters at these two TrakHit's
track->SetTrackParamAtHit(track->GetNTrackHits() - 2, &(trackParam[0]));
// should consider all possibilities ????
// multiple scattering ???? do about like for extrapSegment !!!!
extrapHit = new AliMUONHitForRec(); // empty hit
- extrapCorrHit = new AliMUONHitForRec(); // empty corrected hit
bestChi2 = 3.0;
bestHit = NULL;
if (fPrintLevel >= 10) {
}
// Loop over chambers of the station
for (chInStation = 0; chInStation < 2; chInStation++) {
- // coordinates of extrapolated hit
- extrapHit->
- SetBendingCoor((&(trackParam[chInStation]))->GetBendingCoor());
- extrapHit->
- SetNonBendingCoor((&(trackParam[chInStation]))->GetNonBendingCoor());
- // resolutions from "extrapSegment"
- extrapHit->SetBendingReso2(extrapSegment->GetBendingCoorReso2());
- extrapHit->SetNonBendingReso2(extrapSegment->GetNonBendingCoorReso2());
- // same things for corrected hit
- // better to use copy constructor, after checking that it works properly !!!!
- extrapCorrHit->
- SetBendingCoor((&(trackParam[chInStation]))->GetBendingCoor());
- extrapCorrHit->
- SetNonBendingCoor((&(trackParam[chInStation]))->GetNonBendingCoor());
- extrapHit->SetBendingReso2(extrapSegment->GetBendingCoorReso2());
- extrapHit->SetNonBendingReso2(extrapSegment->GetNonBendingCoorReso2());
- // Loop over hits in the chamber
ch = 2 * station + chInStation;
for (iHit = fIndexOfFirstHitForRecPerChamber[ch];
iHit < fIndexOfFirstHitForRecPerChamber[ch] +
fNHitsForRecPerChamber[ch];
iHit++) {
hit = (AliMUONHitForRec*) ((*fHitsForRecPtr)[iHit]);
- // correction of corrected hit (fBendingCoor and fNonBendingCoor)
- // according to real Z value of "hit" and slopes of right "trackParam"
- extrapCorrHit->
- SetBendingCoor((&(trackParam[chInStation]))->GetBendingCoor() +
- (&(trackParam[chInStation]))->GetBendingSlope() *
- (hit->GetZ() -
- (&(trackParam[chInStation]))->GetZ()));
- extrapCorrHit->
- SetNonBendingCoor((&(trackParam[chInStation]))->GetNonBendingCoor() +
- (&(trackParam[chInStation]))->GetNonBendingSlope() *
- (hit->GetZ() -
- (&(trackParam[chInStation]))->GetZ()));
+ // coordinates of extrapolated hit
+ (&(trackParam[chInStation]))->ExtrapToZ(hit->GetZ());
+ extrapHit->
+ SetBendingCoor((&(trackParam[chInStation]))->GetBendingCoor());
+ extrapHit->
+ SetNonBendingCoor((&(trackParam[chInStation]))->GetNonBendingCoor());
+ // resolutions from "extrapSegment"
+ extrapHit->SetBendingReso2(extrapSegment->GetBendingCoorReso2());
+ extrapHit->SetNonBendingReso2(extrapSegment->GetNonBendingCoorReso2());
+ // Loop over hits in the chamber
// condition for hit not already in segment ????
chi2 = hit->NormalizedChi2WithHitForRec(extrapHit, maxSigma2Distance);
if (chi2 < bestChi2) {
}
if (bestHit) {
// best hit found: add it to track candidate
+ (&(trackParam[chBestHit]))->ExtrapToZ(bestHit->GetZ());
track->AddHitForRec(bestHit);
// set track parameters at this TrackHit
track->SetTrackParamAtHit(track->GetNTrackHits() - 1,
// and corresponding TrackHit's, ...
track->Remove();
delete extrapSegment;
- delete extrapCorrSegment;
delete extrapHit;
- delete extrapCorrHit;
break; // stop the search for this candidate:
// exit from the loop over station
}
delete extrapHit;
- delete extrapCorrHit;
}
delete extrapSegment;
- delete extrapCorrSegment;
// Sort track hits according to increasing Z
track->GetTrackHitsPtr()->Sort();
// Update track parameters at first track hit (smallest Z)
// after going through the first station
if (station == 0) {
trackParamVertex = *trackParam1;
- (&trackParamVertex)->ExtrapToVertex();
+ (&trackParamVertex)->ExtrapToVertex(0.,0.,0.);
track->SetTrackParamAtVertex(&trackParamVertex);
if (fPrintLevel >= 1) {
cout << "FollowTracks: track candidate(0..): " << trackIndex
return;
}
+ //__________________________________________________________________________
+void AliMUONEventReconstructor::UpdateTrackParamAtHit()
+{
+ // Set track parameters after track fitting. Fill fTrackParamAtHit of AliMUONTrack's
+ AliMUONTrack *track;
+ AliMUONTrackHit *trackHit;
+ AliMUONTrackParam *trackParam;
+ track = (AliMUONTrack*) fRecTracksPtr->First();
+ while (track) {
+ trackHit = (AliMUONTrackHit*) (track->GetTrackHitsPtr())->First();
+ while (trackHit) {
+ trackParam = trackHit->GetTrackParam();
+ track->AddTrackParamAtHit(trackParam);
+ trackHit = (AliMUONTrackHit*) (track->GetTrackHitsPtr())->After(trackHit);
+ } // trackHit
+ track = (AliMUONTrack*) fRecTracksPtr->After(track);
+ } // track
+ return;
+}
+
+ //__________________________________________________________________________
+void AliMUONEventReconstructor::UpdateHitForRecAtHit()
+{
+ // Set cluster parameterss after track fitting. Fill fHitForRecAtHit of AliMUONTrack's
+ AliMUONTrack *track;
+ AliMUONTrackHit *trackHit;
+ AliMUONHitForRec *hitForRec;
+ track = (AliMUONTrack*) fRecTracksPtr->First();
+ while (track) {
+ trackHit = (AliMUONTrackHit*) (track->GetTrackHitsPtr())->First();
+ while (trackHit) {
+ hitForRec = trackHit->GetHitForRecPtr();
+ track->AddHitForRecAtHit(hitForRec);
+ trackHit = (AliMUONTrackHit*) (track->GetTrackHitsPtr())->After(trackHit);
+ } // trackHit
+ track = (AliMUONTrack*) fRecTracksPtr->After(track);
+ } // track
+ return;
+}
+
+ //__________________________________________________________________________
+void AliMUONEventReconstructor::FillMUONTrack()
+{
+ // Set track parameters at hits for Kalman track. Fill fTrackParamAtHit of AliMUONTrack's
+ AliMUONTrackK *track;
+ track = (AliMUONTrackK*) fRecTracksPtr->First();
+ while (track) {
+ track->FillMUONTrack();
+ track = (AliMUONTrackK*) fRecTracksPtr->After(track);
+ }
+ return;
+}
+
//__________________________________________________________________________
void AliMUONEventReconstructor::EventDump(void)
{
AliMUONTrack *track;
AliMUONTrackParam *trackParam, *trackParam1;
- TParticle *p;
Double_t bendingSlope, nonBendingSlope, pYZ;
Double_t pX, pY, pZ, x, y, z, c;
Int_t np, trackIndex, nTrackHits;
fRecTracksPtr->Compress(); // for simple loop without "Next" since no hole
// Loop over reconstructed tracks
for (trackIndex = 0; trackIndex < fNRecTracks; trackIndex++) {
+ if (fTrackMethod != 1) continue; //AZ - skip the rest for now
if (fPrintLevel >= 1)
cout << " track number: " << trackIndex << endl;
// function for each track for modularity ????
z, x, y, pX, pY, pZ, c);
}
// informations about generated particles
- np = gAlice->GetNtrack();
+ np = gAlice->GetMCApp()->GetNtrack();
printf(" **** number of generated particles: %d \n", np);
- for (Int_t iPart = 0; iPart < np; iPart++) {
- p = gAlice->Particle(iPart);
- printf(" particle %d: type= %d px= %f py= %f pz= %f pdg= %d\n",
- iPart, p->GetPdgCode(), p->Px(), p->Py(), p->Pz(), p->GetPdgCode());
- }
+// for (Int_t iPart = 0; iPart < np; iPart++) {
+// p = gAlice->Particle(iPart);
+// printf(" particle %d: type= %d px= %f py= %f pz= %f pdg= %d\n",
+// iPart, p->GetPdgCode(), p->Px(), p->Py(), p->Pz(), p->GetPdgCode());
+// }
return;
}
+
+//__________________________________________________________________________
+void AliMUONEventReconstructor::EventDumpTrigger(void)
+{
+ // Dump reconstructed trigger event
+ // and the particle parameters
+
+ AliMUONTriggerTrack *triggertrack ;
+ Int_t nTriggerTracks = fMUONData->RecTriggerTracks()->GetEntriesFast();
+
+ if (fPrintLevel >= 1) {
+ cout << "****** enter EventDumpTrigger ******" << endl;
+ cout << " Number of Reconstructed tracks :" << nTriggerTracks << endl;
+ }
+ // Loop over reconstructed tracks
+ for (Int_t trackIndex = 0; trackIndex < nTriggerTracks; trackIndex++) {
+ triggertrack = (AliMUONTriggerTrack*)fMUONData->RecTriggerTracks()->At(trackIndex);
+ printf(" trigger track number %i x11=%f y11=%f thetax=%f thetay=%f \n",
+ trackIndex,
+ triggertrack->GetX11(),triggertrack->GetY11(),
+ triggertrack->GetThetax(),triggertrack->GetThetay());
+ }
+}
+
+//__________________________________________________________________________
void AliMUONEventReconstructor::FillEvent()
{
// Create a new AliMUONRecoEvent, fill its track list, then add it as a
TDirectory *current = gDirectory;
if (!fTreeFile) fTreeFile = new TFile("tree_reco.root", "RECREATE");
if (!fEventTree) fEventTree = new TTree("TreeRecoEvent", "MUON reconstructed events");
- if (fRecoEvent->MakeDumpTracks(fRecTracksPtr)) {
+ //AZif (fRecoEvent->MakeDumpTracks(fRecTracksPtr)) {
+ if (fRecoEvent->MakeDumpTracks(fMuons, fRecTracksPtr, this)) { //AZ
if (fPrintLevel > 1) fRecoEvent->EventInfo();
TBranch *branch = fEventTree->GetBranch("Event");
- if (!branch) branch = fEventTree->Branch("Event", "AliMUONRecoEvent", &fRecoEvent, 64000,1);
+ if (!branch) branch = fEventTree->Branch("Event", "AliMUONRecoEvent", &fRecoEvent, 64000);
branch->SetAutoDelete();
fTreeFile->cd();
fEventTree->Fill();
// restore directory
current->cd();
}
+
+//__________________________________________________________________________
+void AliMUONEventReconstructor::MakeTrackCandidatesK(void)
+{
+ // To make initial tracks for Kalman filter from the list of segments
+ Int_t istat, iseg;
+ AliMUONSegment *segment;
+ AliMUONTrackK *trackK;
+
+ if (fPrintLevel >= 1) cout << "enter MakeTrackCandidatesK" << endl;
+ // Reset the TClonesArray of reconstructed tracks
+ if (fRecTracksPtr) fRecTracksPtr->Delete();
+ // Delete in order that the Track destructors are called,
+ // hence the space for the TClonesArray of pointers to TrackHit's is freed
+ fNRecTracks = 0;
+
+ AliMUONTrackK a(this, fHitsForRecPtr); // bad idea ???
+ // Loop over stations(1...) 5 and 4
+ for (istat=4; istat>=3; istat--) {
+ // Loop over segments in the station
+ for (iseg=0; iseg<fNSegments[istat]; iseg++) {
+ // Transform segments to tracks and evaluate covariance matrix
+ segment = (AliMUONSegment*) ((*fSegmentsPtr[istat])[iseg]);
+ trackK = new ((*fRecTracksPtr)[fNRecTracks]) AliMUONTrackK(segment);
+ fNRecTracks++;
+ } // for (iseg=0;...)
+ } // for (istat=4;...)
+ return;
+}
+
+//__________________________________________________________________________
+void AliMUONEventReconstructor::FollowTracksK(void)
+{
+ // Follow tracks using Kalman filter
+ Bool_t ok;
+ Int_t icand, ichamBeg = 0, ichamEnd, chamBits;
+ Double_t zDipole1, zDipole2;
+ AliMUONTrackK *trackK;
+ AliMUONHitForRec *hit;
+ AliMUONRawCluster *clus;
+ TClonesArray *rawclusters;
+ clus = 0; rawclusters = 0;
+
+ //AZ(z->-z) zDipole1 = GetSimpleBPosition() - GetSimpleBLength()/2;
+ //AZ(z->-z) zDipole2 = zDipole1 + GetSimpleBLength();
+ zDipole1 = GetSimpleBPosition() + GetSimpleBLength()/2;
+ zDipole2 = zDipole1 - GetSimpleBLength();
+
+ // Print hits
+ trackK = (AliMUONTrackK*) ((*fRecTracksPtr)[0]);
+ if (trackK->DebugLevel() > 0) {
+ for (Int_t i1=0; i1<fNHitsForRec; i1++) {
+ hit = (AliMUONHitForRec*) ((*fHitsForRecPtr)[i1]);
+ //if (hit->GetTTRTrack() > 1 || hit->GetTrackRefSignal() == 0) continue;
+ printf(" Hit # %d %10.4f %10.4f %10.4f",
+ hit->GetChamberNumber(), hit->GetBendingCoor(),
+ hit->GetNonBendingCoor(), hit->GetZ());
+ if (fRecTrackRefHits) {
+ // from track ref hits
+ printf(" %3d %3d \n", hit->GetTrackRefSignal(), hit->GetTTRTrack());
+ } else {
+ // from raw clusters
+ rawclusters = fMUONData->RawClusters(hit->GetChamberNumber());
+ clus = (AliMUONRawCluster*) rawclusters->UncheckedAt(hit->
+ GetHitNumber());
+ printf("%3d", clus->GetTrack(1)-1);
+ if (clus->GetTrack(2) != 0) printf("%3d \n", clus->GetTrack(2)-1);
+ else printf("\n");
+ } // if (fRecTrackRefHits)
+ }
+ } // if (trackK->DebugLevel() > 0)
+
+ icand = -1;
+ Int_t nSeeds;
+ nSeeds = fNRecTracks; // starting number of seeds
+ // Loop over track candidates
+ while (icand < fNRecTracks-1) {
+ icand ++;
+ if (trackK->DebugLevel()>0) cout << " *** Kalman track candidate No. " << icand << endl;
+ trackK = (AliMUONTrackK*) ((*fRecTracksPtr)[icand]);
+ if (trackK->GetRecover() < 0) continue; // failed track
+
+ // Discard candidate which will produce the double track
+ /*
+ if (icand > 0) {
+ ok = CheckCandidateK(icand,nSeeds);
+ if (!ok) {
+ trackK->SetRecover(-1); // mark candidate to be removed
+ continue;
+ }
+ }
+ */
+
+ ok = kTRUE;
+ if (trackK->GetRecover() == 0) hit = (AliMUONHitForRec*)
+ trackK->GetHitOnTrack()->Last(); // last hit
+ //else hit = (AliMUONHitForRec*) (*trackK->GetHitOnTrack())[1]; // 2'nd hit
+ else hit = trackK->GetHitLastOk(); // hit where track stopped
+ if (hit) ichamBeg = hit->GetChamberNumber();
+ ichamEnd = 0;
+ // Check propagation direction
+ if (hit == NULL) ichamBeg = ichamEnd;
+ //AZ(z->-z) else if (trackK->GetTrackDir() > 0) {
+ else if (trackK->GetTrackDir() < 0) {
+ ichamEnd = 9; // forward propagation
+ ok = trackK->KalmanFilter(ichamBeg,ichamEnd,kFALSE,zDipole1,zDipole2);
+ if (ok) {
+ ichamBeg = ichamEnd;
+ ichamEnd = 6; // backward propagation
+ // Change weight matrix and zero fChi2 for backpropagation
+ trackK->StartBack();
+ //AZ trackK->SetTrackDir(-1);
+ trackK->SetTrackDir(1);
+ ok = trackK->KalmanFilter(ichamBeg,ichamEnd,kTRUE,zDipole1,zDipole2);
+ ichamBeg = ichamEnd;
+ ichamEnd = 0;
+ }
+ } else {
+ if (trackK->GetBPFlag()) {
+ // backpropagation
+ ichamEnd = 6; // backward propagation
+ // Change weight matrix and zero fChi2 for backpropagation
+ trackK->StartBack();
+ ok = trackK->KalmanFilter(ichamBeg,ichamEnd,kTRUE,zDipole1,zDipole2);
+ ichamBeg = ichamEnd;
+ ichamEnd = 0;
+ }
+ }
+
+ if (ok) {
+ //AZ trackK->SetTrackDir(-1);
+ trackK->SetTrackDir(1);
+ trackK->SetBPFlag(kFALSE);
+ ok = trackK->KalmanFilter(ichamBeg,ichamEnd,kFALSE,zDipole1,zDipole2);
+ }
+ if (!ok) { trackK->SetRecover(-1); continue; } // mark candidate to be removed
+
+ // Apply smoother
+ if (trackK->GetRecover() >= 0) {
+ ok = trackK->Smooth();
+ if (!ok) trackK->SetRecover(-1); // mark candidate to be removed
+ }
+
+ // Majority 3 of 4 in first 2 stations
+ if (!ok) continue;
+ chamBits = 0;
+ Double_t chi2_max = 0;
+ for (Int_t i=0; i<trackK->GetNTrackHits(); i++) {
+ hit = (AliMUONHitForRec*) (*trackK->GetHitOnTrack())[i];
+ chamBits |= BIT(hit->GetChamberNumber());
+ if (trackK->GetChi2PerPoint(i) > chi2_max) chi2_max = trackK->GetChi2PerPoint(i);
+ }
+ if (!((chamBits&3)==3 || (chamBits>>2&3)==3) && chi2_max > 25) {
+ //trackK->Recover();
+ trackK->SetRecover(-1); //mark candidate to be removed
+ continue;
+ }
+ if (ok) trackK->SetTrackQuality(0); // compute "track quality"
+ } // while
+
+ for (Int_t i=0; i<fNRecTracks; i++) {
+ trackK = (AliMUONTrackK*) ((*fRecTracksPtr)[i]);
+ if (trackK->GetRecover() < 0) fRecTracksPtr->RemoveAt(i);
+ }
+
+ // Compress TClonesArray
+ fRecTracksPtr->Compress();
+ fNRecTracks = fRecTracksPtr->GetEntriesFast();
+ return;
+}
+
+//__________________________________________________________________________
+Bool_t AliMUONEventReconstructor::CheckCandidateK(Int_t icand, Int_t nSeeds) const
+{
+ // Discards track candidate if it will produce the double track (having
+ // the same seed segment hits as hits of a good track found before)
+ AliMUONTrackK *track1, *track2;
+ AliMUONHitForRec *hit1, *hit2, *hit;
+
+ track1 = (AliMUONTrackK*) ((*fRecTracksPtr)[icand]);
+ hit1 = (AliMUONHitForRec*) (*track1->GetHitOnTrack())[0]; // 1'st hit
+ hit2 = (AliMUONHitForRec*) (*track1->GetHitOnTrack())[1]; // 2'nd hit
+
+ for (Int_t i=0; i<icand; i++) {
+ track2 = (AliMUONTrackK*) ((*fRecTracksPtr)[i]);
+ //if (track2->GetRecover() < 0) continue;
+ if (track2->GetRecover() < 0 && icand >= nSeeds) continue;
+
+ if (track1->GetStartSegment() == track2->GetStartSegment()) {
+ return kFALSE;
+ } else {
+ Int_t nSame = 0;
+ for (Int_t j=0; j<track2->GetNTrackHits(); j++) {
+ hit = (AliMUONHitForRec*) (*track2->GetHitOnTrack())[j];
+ if (hit == hit1 || hit == hit2) {
+ nSame++;
+ if (nSame == 2) return kFALSE;
+ }
+ } // for (Int_t j=0;
+ }
+ } // for (Int_t i=0;
+ return kTRUE;
+}
+
+//__________________________________________________________________________
+void AliMUONEventReconstructor::RemoveDoubleTracksK(void)
+{
+ // Removes double tracks (sharing more than half of their hits). Keeps
+ // the track with higher quality
+ AliMUONTrackK *track1, *track2, *trackToKill;
+
+ // Sort tracks according to their quality
+ fRecTracksPtr->Sort();
+
+ // Loop over first track of the pair
+ track1 = (AliMUONTrackK*) fRecTracksPtr->First();
+ Int_t debug = track1->DebugLevel();
+ while (track1) {
+ // Loop over second track of the pair
+ track2 = (AliMUONTrackK*) fRecTracksPtr->After(track1);
+ while (track2) {
+ // Check whether or not to keep track2
+ if (!track2->KeepTrack(track1)) {
+ if (debug >= 0) cout << " Killed track: " << 1/(*track2->GetTrackParameters())(4,0) <<
+ " " << track2->GetTrackQuality() << endl;
+ trackToKill = track2;
+ track2 = (AliMUONTrackK*) fRecTracksPtr->After(track2);
+ trackToKill->Kill();
+ fRecTracksPtr->Compress();
+ } else track2 = (AliMUONTrackK*) fRecTracksPtr->After(track2);
+ } // track2
+ track1 = (AliMUONTrackK*) fRecTracksPtr->After(track1);
+ } // track1
+
+ fNRecTracks = fRecTracksPtr->GetEntriesFast();
+ if (debug >= 0) cout << " Number of Kalman tracks: " << fNRecTracks << endl;
+}
+
+//__________________________________________________________________________
+void AliMUONEventReconstructor::GoToVertex(void)
+{
+ // Propagates track to the vertex thru absorber
+ // (using Branson correction for now)
+
+ Double_t zVertex;
+ zVertex = 0;
+ for (Int_t i=0; i<fNRecTracks; i++) {
+ //((AliMUONTrackK*)(*fRecTracksPtr)[i])->Branson();
+ ((AliMUONTrackK*)(*fRecTracksPtr)[i])->SetTrackQuality(1); // compute Chi2
+ //((AliMUONTrackK*)(*fRecTracksPtr)[i])->GoToZ(zVertex); // w/out absorber
+ ((AliMUONTrackK*)(*fRecTracksPtr)[i])->GoToVertex(1); // with absorber
+ }
+}
+
+//__________________________________________________________________________
+void AliMUONEventReconstructor::SetTrackMethod(Int_t iTrackMethod)
+{
+ // Set track method and recreate track container if necessary
+
+ fTrackMethod = iTrackMethod == 2 ? 2 : 1;
+ if (fTrackMethod == 2) {
+ if (fRecTracksPtr) delete fRecTracksPtr;
+ fRecTracksPtr = new TClonesArray("AliMUONTrackK", 10);
+ cout << " *** Tracking with the Kalman filter *** " << endl;
+ } else cout << " *** Traditional tracking *** " << endl;
+
+}