* about the suitability of this software for any purpose. It is *
* provided "as is" without express or implied warranty. *
**************************************************************************/
-/*
-$Log$
-Revision 1.3 2000/10/02 21:28:09 fca
-Removal of useless dependecies via forward declarations
-Revision 1.2 2000/06/15 07:58:49 morsch
-Code from MUON-dev joined
-
-Revision 1.1.2.7 2000/06/09 22:06:29 morsch
-Some coding rule violations corrected. Will soon be obsolete.
-
-Revision 1.1.2.6 2000/05/02 07:15:29 morsch
-Put back TH1.h and TH2.h includes.
-
-Revision 1.1.2.5 2000/02/17 18:12:43 morsch
-Corrections in trackf_read_spoint causing segmentation violations in previous version (I. Chevrot)
-New histos (I. Chevrot)
-
-Revision 1.1.2.4 2000/02/15 18:01:08 morsch
-Log messages
-
-Revision 1.1.2.3 2000/02/15 17:59:01 morsch
-Log message added
-
-Revision 1.1.2.2 2000/02/15 18:54:56 morsch
-Reference between track contributing to reconstructed hit and particle corrected
-*/
-
-#include "AliCallf77.h"
-#include "AliMUONTrackReconstructor.h"
-#include "AliRun.h"
-#include "AliMUON.h"
-#include "AliMC.h"
-
-#include "AliMUONHit.h"
-#include "AliMUONPadHit.h"
-#include "AliMUONDigit.h"
+/* $Id$ */
+
+/// \class AliMUONTrackReconstructor
+/// MUON track reconstructor using the original method
+///
+/// This class contains as data:
+/// - the parameters for the track reconstruction
+///
+/// It contains as methods, among others:
+/// - MakeTracks to build the tracks
+///
+
+#include <stdlib.h>
+#include <Riostream.h>
+#include <TMatrixD.h>
+
+#include "AliMUONVTrackReconstructor.h"
+#include "AliMUONTrackReconstructor.h"
+#include "AliMUONData.h"
+#include "AliMUONConstants.h"
#include "AliMUONRawCluster.h"
-#include "AliMUONReconstHit.h"
-
-#include "AliPDG.h"
-
-#include <TRandom.h>
-#include <TFile.h>
-#include <TH1.h>
-#include <TH2.h>
-#include <TTree.h>
-#include <TParticle.h>
+#include "AliMUONHitForRec.h"
+#include "AliMUONObjectPair.h"
+#include "AliMUONTrack.h"
+#include "AliMUONTrackParam.h"
+#include "AliMUONTrackExtrap.h"
+#include "AliLog.h"
#include <TMinuit.h>
-#include <iostream.h>
-#ifndef WIN32
-# define reco_init reco_init_
-# define cutpxz cutpxz_
-# define sigmacut sigmacut_
-# define xpreci xpreci_
-# define ypreci ypreci_
-# define reconstmuon reconstmuon_
-# define reconstmuon2 reconstmuon2_
-# define trackf_read_geant trackf_read_geant_
-# define trackf_read_fit trackf_read_fit_
-# define trackf_read_spoint trackf_read_spoint_
-# define chfill chfill_
-# define chfill2 chfill2_
-# define chf1 chf1_
-# define chfnt chfnt_
-# define hist_create hist_create_
-# define hist_closed hist_closed_
-# define rndm rndm_
-# define fcn fcn_
-# define trackf_fit trackf_fit_
-# define prec_fit prec_fit_
-# define fcnfit fcnfit_
-# define reco_term reco_term_
-#else
-# define reco_init RECO_INIT
-# define cutpxz CUTPXZ
-# define sigmacut SIGMACUT
-# define xpreci XPRECI
-# define ypreci YPRECI
-# define reconstmuon RECONSTMUON
-# define reconstmuon2 RECONSTMUON2
-# define trackf_read_geant TRACKF_READ_GEANT
-# define trackf_read_fit TRACKF_READ_FIT
-# define trackf_read_spoint TRACKF_READ_SPOINT
-# define chfill CHFILL
-# define chfill2 CHFILL2
-# define chf1 CHF1
-# define chfnt CHFNT
-# define hist_create HIST_CREATE
-# define hist_closed HIST_CLOSED
-# define rndm RNDM
-# define fcn FCN
-# define trackf_fit TRACKF_FIT
-# define prec_fit PREC_FIT
-# define fcnfit FCNFIT
-# define reco_term RECO_TERM
-#endif
+// Functions to be minimized with Minuit
+void TrackChi2(Int_t &NParam, Double_t *Gradient, Double_t &Chi2, Double_t *Param, Int_t Flag);
+void TrackChi2MCS(Int_t &NParam, Double_t *Gradient, Double_t &Chi2, Double_t *Param, Int_t Flag);
-extern "C"
-{
-void type_of_call reco_init(Double_t &, Double_t &, Double_t &);
-void type_of_call reco_term();
-void type_of_call cutpxz(Double_t &);
-void type_of_call sigmacut(Double_t &);
-void type_of_call xpreci(Double_t &);
-void type_of_call ypreci(Double_t &);
-void type_of_call reconstmuon(Int_t &, Int_t &, Int_t &, Int_t &, Int_t &);
-void type_of_call reconstmuon2(Int_t &, Int_t &, Int_t &);
-void type_of_call trackf_read_fit(Int_t &, Int_t &, Int_t &, Int_t *, Double_t *, Double_t *);
-void type_of_call trackf_read_geant(Int_t *, Double_t *, Double_t *, Double_t *, Int_t *, Int_t *, Double_t *, Double_t *, Double_t *, Double_t *,Int_t &, Double_t *, Double_t *, Double_t *, Int_t &, Int_t &, Double_t *, Double_t *, Double_t *, Double_t *);
-void type_of_call trackf_read_spoint(Int_t *, Double_t *, Double_t *, Double_t *, Int_t *, Int_t *, Double_t *, Double_t *, Double_t *, Double_t *,Int_t &, Double_t *, Double_t *, Double_t *, Int_t &, Int_t &, Double_t *, Double_t *, Double_t *, Double_t *);
-void type_of_call chfill(Int_t &, Float_t &, Float_t &, Float_t &);
-void type_of_call chfill2(Int_t &, Float_t &, Float_t &, Float_t &);
-void type_of_call chf1(Int_t &, Float_t &, Float_t &);
-void type_of_call chfnt(Int_t &, Int_t &, Int_t *, Int_t *, Float_t *, Float_t *, Float_t *, Float_t *, Float_t *, Float_t *, Float_t *, Float_t *);
-void type_of_call hist_create();
-void type_of_call hist_closed();
-void type_of_call fcnf(Int_t &, Double_t *, Double_t &, Double_t *, Int_t);
-void type_of_call fcn(Int_t &, Double_t *, Double_t &, Double_t *, Int_t &, Int_t &);
-void type_of_call trackf_fit(Int_t &, Double_t *, Double_t *, Double_t &, Double_t &, Double_t &, Double_t &, Double_t &);
-void type_of_call prec_fit(Double_t &, Double_t &, Double_t &, Double_t &, Double_t&, Double_t &, Double_t &, Double_t &, Double_t &, Double_t &, Double_t &, Double_t &, Double_t &, Double_t &, Double_t &);
-void type_of_call fcnfitf(Int_t &, Double_t *, Double_t &, Double_t *, Int_t);
-void type_of_call fcnfit(Int_t &, Double_t *, Double_t &, Double_t *, Int_t &, Int_t &);
-Float_t type_of_call rndm() {return gRandom->Rndm();}
-void type_of_call fit_trace(Float_t &, Float_t &, Float_t &, Float_t &, Float_t &, Float_t &, Int_t &, Int_t &, Int_t &, Int_t &);
-}
+Double_t MultipleScatteringAngle2(AliMUONTrackParam *param);
-static TTree *gAliNtupleGlobal;
-static TFile *gAliFileGlobal;
-static TTree *gAliTreeK1;
-static TTree *gAliTrH1;
-static TClonesArray *gAliHits2; //List of hits for one track only
-static TClonesArray *gAliParticles2; //List of particles in the Kine tree
+/// \cond CLASSIMP
+ClassImp(AliMUONTrackReconstructor) // Class implementation in ROOT context
+/// \endcond
+//************* Defaults parameters for reconstruction
+const Double_t AliMUONTrackReconstructor::fgkMaxNormChi2 = 100.0;
+const Bool_t AliMUONTrackReconstructor::fgkTrackAllTracks = kFALSE;
-// variables of the tracking ntuple
-struct {
- Int_t ievr; // number of event
- Int_t ntrackr; // number of tracks per event
- Int_t istatr[500]; // 1 = good muon, 2 = ghost, 0 = something else
- Int_t isignr[500]; // sign of the track
- Float_t pxr[500]; // x momentum of the reconstructed track
- Float_t pyr[500]; // y momentum of the reconstructed track
- Float_t pzr[500]; // z momentum of the reconstructed track
- Float_t zvr[500]; // z vertex
- Float_t chi2r[500]; // chi2 of the fit of the track with the field map
- Float_t pxv[500]; // x momentum at vertex
- Float_t pyv[500]; // y momentum at vertex
- Float_t pzv[500]; // z momentum at vertex
-} NtupleSt;
-
-ClassImp(AliMUONTrackReconstructor)
-
-//___________________________________________________
-AliMUONTrackReconstructor::AliMUONTrackReconstructor()
+//__________________________________________________________________________
+AliMUONTrackReconstructor::AliMUONTrackReconstructor(AliMUONData* data)
+ : AliMUONVTrackReconstructor(data)
{
-// Constructor
-//
- fSPxzCut = 3.0;
- fSSigmaCut = 4.0;
- fSXPrec = 0.01;
- fSYPrec = 0.144;
+ /// Constructor for class AliMUONTrackReconstructor
+
+ // Memory allocation for the TClonesArray of reconstructed tracks
+ fRecTracksPtr = new TClonesArray("AliMUONTrack", 10);
}
-//_____________________________________________________________________________
-void AliMUONTrackReconstructor::Reconst2(Int_t &ifit, Int_t &idebug, Int_t &nev)
+ //__________________________________________________________________________
+AliMUONTrackReconstructor::~AliMUONTrackReconstructor(void)
{
-//
-//
- reconstmuon2(ifit,idebug,nev);
+ /// Destructor for class AliMUONTrackReconstructor
+ delete fRecTracksPtr;
}
-//_____________________________________________________________________________
-void AliMUONTrackReconstructor::Reconst(Int_t &ifit, Int_t &idebug, Int_t bgdEvent, Int_t &nev, Int_t &idres, Int_t &ireadgeant, Option_t *option,Text_t *filename)
+ //__________________________________________________________________________
+void AliMUONTrackReconstructor::AddHitsForRecFromRawClusters()
{
- //
- // open kine and hits tree of background file for reconstruction of geant hits
- // call tracking fortran program
- static Bool_t first=kTRUE;
- static TFile *pFile;
- char *addBackground = strstr(option,"Add");
+ /// To add to the list of hits for reconstruction all the raw clusters
+ TTree *treeR;
+ AliMUONHitForRec *hitForRec;
+ AliMUONRawCluster *clus;
+ Int_t iclus, nclus;
+ TClonesArray *rawclusters;
+ AliDebug(1,"Enter AddHitsForRecFromRawClusters");
- if (addBackground ) { // only in case of background with geant hits
- if(first) {
- fFileName=filename;
- cout<<"filename "<<fFileName<<endl;
- pFile=new TFile(fFileName);
- cout<<"I have opened "<<fFileName<<" file "<<endl;
- gAliHits2 = new TClonesArray("AliMUONHit",1000);
- gAliParticles2 = new TClonesArray("TParticle",1000);
- first=kFALSE;
- }
- pFile->cd();
- if(gAliHits2) gAliHits2->Clear();
- if(gAliParticles2) gAliParticles2->Clear();
- if(gAliTrH1) delete gAliTrH1;
- gAliTrH1=0;
- if(gAliTreeK1) delete gAliTreeK1;
- gAliTreeK1=0;
- // Get Hits Tree header from file
- char treeName[20];
- sprintf(treeName,"TreeH%d",bgdEvent);
- gAliTrH1 = (TTree*)gDirectory->Get(treeName);
- // printf("gAliTrH1 %p of treename %s for event %d \n",gAliTrH1,treeName,bgdEvent);
- if (!gAliTrH1) {
- printf("ERROR: cannot find Hits Tree for event:%d\n",bgdEvent);
- }
- // set branch addresses
- TBranch *branch;
- char branchname[30];
- AliMUON *pMUON = (AliMUON*) gAlice->GetModule("MUON");
- sprintf(branchname,"%s",pMUON->GetName());
- if (gAliTrH1 && gAliHits2) {
- branch = gAliTrH1->GetBranch(branchname);
- if (branch) branch->SetAddress(&gAliHits2);
- }
- gAliTrH1->GetEntries();
- // get the Kine tree
- sprintf(treeName,"TreeK%d",bgdEvent);
- gAliTreeK1 = (TTree*)gDirectory->Get(treeName);
- if (!gAliTreeK1) {
- printf("ERROR: cannot find Kine Tree for event:%d\n",bgdEvent);
+ treeR = fMUONData->TreeR();
+ if (!treeR) {
+ AliError("TreeR must be loaded");
+ exit(0);
+ }
+
+ fMUONData->SetTreeAddress("RC");
+ fMUONData->GetRawClusters(); // only one entry
+
+ // Loop over tracking chambers
+ for (Int_t ch = 0; ch < AliMUONConstants::NTrackingCh(); ch++) {
+ rawclusters =fMUONData->RawClusters(ch);
+ nclus = (Int_t) (rawclusters->GetEntries());
+ // Loop over (cathode correlated) raw clusters
+ for (iclus = 0; iclus < nclus; iclus++) {
+ clus = (AliMUONRawCluster*) rawclusters->UncheckedAt(iclus);
+ // new AliMUONHitForRec from raw cluster
+ // and increment number of AliMUONHitForRec's (total and in chamber)
+ hitForRec = new ((*fHitsForRecPtr)[fNHitsForRec]) AliMUONHitForRec(clus);
+ fNHitsForRec++;
+ // more information into HitForRec
+ hitForRec->SetBendingReso2(clus->GetErrY() * clus->GetErrY());
+ hitForRec->SetNonBendingReso2(clus->GetErrX() * clus->GetErrX());
+ // original raw cluster
+ hitForRec->SetChamberNumber(ch);
+ hitForRec->SetHitNumber(iclus);
+ // Z coordinate of the raw cluster (cm)
+ hitForRec->SetZ(clus->GetZ(0));
+ StdoutToAliDebug(3,
+ cout << "Chamber " << ch <<
+ " raw cluster " << iclus << " : " << endl;
+ clus->Print("full");
+ cout << "AliMUONHitForRec number (1...): " << fNHitsForRec << endl;
+ hitForRec->Print("full");
+ );
+ } // end of cluster loop
+ } // end of chamber loop
+ SortHitsForRecWithIncreasingChamber();
+
+ AliDebug(1,"End of AddHitsForRecFromRawClusters");
+ if (AliLog::GetGlobalDebugLevel() > 0) {
+ AliDebug(1, Form("NHitsForRec: %d",fNHitsForRec));
+ for (Int_t ch = 0; ch < AliMUONConstants::NTrackingCh(); ch++) {
+ AliDebug(1, Form("Chamber(0...): %d",ch));
+ AliDebug(1, Form("NHitsForRec: %d", fNHitsForRecPerChamber[ch]));
+ AliDebug(1, Form("Index(first HitForRec): %d", fIndexOfFirstHitForRecPerChamber[ch]));
+ for (Int_t hit = fIndexOfFirstHitForRecPerChamber[ch];
+ hit < fIndexOfFirstHitForRecPerChamber[ch] + fNHitsForRecPerChamber[ch];
+ hit++) {
+ AliDebug(1, Form("HitForRec index(0...): %d",hit));
+ ((*fHitsForRecPtr)[hit])->Dump();
+ }
}
- // set branch addresses
- if (gAliTreeK1)
- gAliTreeK1->SetBranchAddress("Particles", &gAliParticles2);
- gAliTreeK1->GetEvent(0);
-
- // get back to the first file
- TTree *treeK = gAlice->TreeK();
- TFile *file1 = 0;
- if (treeK) file1 = treeK->GetCurrentFile();
- file1->cd();
-
- } // end if addBackground
+ }
- // call tracking fortran program
- reconstmuon(ifit,idebug,nev,idres,ireadgeant);
+ return;
}
-//________________________________________________________________________________
-void AliMUONTrackReconstructor::Init(Double_t &seff, Double_t &sb0, Double_t &sbl3)
+ //__________________________________________________________________________
+void AliMUONTrackReconstructor::MakeTracks(void)
{
- //
- // introduce in fortran program somme parameters and cuts for tracking
- // create output file "reconst.root" (histos + ntuple)
- cutpxz(fSPxzCut); // Pxz cut (GeV/c) to begin the track finding
- sigmacut(fSSigmaCut); // Number of sigmas delimiting the searching areas
- xpreci(fSXPrec); // Chamber precision in X (cm)
- ypreci(fSYPrec); // Chamber precision in Y (cm)
- reco_init(seff,sb0,sbl3);
+ /// To make the tracks from the list of segments and points in all stations
+ AliDebug(1,"Enter MakeTracks");
+ // 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();
+ // Propagate tracks to the vertex through absorber
+ ExtrapTracksToVertex();
+ // Fill out the AliMUONTrack's
+ FillMUONTrack();
}
-//__________________________________________
-void AliMUONTrackReconstructor::FinishEvent()
+ //__________________________________________________________________________
+void AliMUONTrackReconstructor::MakeTrackCandidates(void)
{
-// Finish
- TTree *treeK = gAlice->TreeK();
- TFile *file1 = 0;
- if (treeK) file1 = treeK->GetCurrentFile();
- file1->cd();
+ /// To make track candidates:
+ /// Start with segments station(1..) 4 or 5 then follow track in station 5 or 4.
+ /// Good candidates are made of at least three hitForRec's.
+ /// Keep only best candidates or all of them according to the flag fgkTrackAllTracks.
+ TClonesArray *segments;
+ AliMUONObjectPair *segment;
+ AliMUONHitForRec *hitForRec1, *hitForRec2;
+ AliMUONTrack *track;
+ AliMUONTrackParam *trackParamAtFirstHit;
+ Int_t iCandidate = 0;
+
+ AliDebug(1,"Enter MakeTrackCandidates");
+
+ // Loop over stations(1..) 5 and 4 and make track candidates
+ for (Int_t istat=4; istat>=3; istat--) {
+ // Make segments in the station
+ segments = MakeSegmentsInStation(istat);
+ // Loop over segments
+ for (Int_t iseg=0; iseg<segments->GetEntriesFast(); iseg++) {
+ AliDebug(1,Form("Making primary candidate(1..) %d",++iCandidate));
+ // Transform segments to tracks and put them at the end of fRecTracksPtr
+ segment = (AliMUONObjectPair*) ((*segments)[iseg]);
+ hitForRec1 = (AliMUONHitForRec*) segment->First();
+ hitForRec2 = (AliMUONHitForRec*) segment->Second();
+ track = new ((*fRecTracksPtr)[fRecTracksPtr->GetLast()+1]) AliMUONTrack(hitForRec1, hitForRec2);
+ fNRecTracks++;
+ // Add MCS effects in parameter covariances
+ trackParamAtFirstHit = (AliMUONTrackParam*) (track->GetTrackParamAtHit()->First());
+ AliMUONTrackExtrap::AddMCSEffectInTrackParamCov(trackParamAtFirstHit,AliMUONConstants::ChamberThicknessInX0(),1.);
+ // Printout for debuging
+ if ((AliLog::GetDebugLevel("MUON","AliMUONTrackReconstructor") >= 2) || (AliLog::GetGlobalDebugLevel() >= 2)) {
+ cout<<endl<<"Track parameter covariances at first hit with multiple Coulomb scattering effects:"<<endl;
+ trackParamAtFirstHit->GetCovariances()->Print();
+ }
+ // Look for compatible hitForRec(s) in the other station
+ FollowTrackInStation(track,7-istat);
+ }
+ // delete the array of segments
+ delete segments;
+ }
+ fRecTracksPtr->Compress(); // this is essential before checking tracks
+
+ // Keep all different tracks or only the best ones as required
+ if (fgkTrackAllTracks) RemoveIdenticalTracks();
+ else RemoveDoubleTracks();
+
+ AliDebug(1,Form("Number of good candidates = %d",fNRecTracks));
+
}
-//_____________________________________
-void AliMUONTrackReconstructor::Close()
+ //__________________________________________________________________________
+void AliMUONTrackReconstructor::RemoveIdenticalTracks(void)
{
- //
- // write histos and ntuple to "reconst.root" file
- reco_term();
+ /// To remove identical tracks:
+ /// Tracks are considered identical if they have all their hits in common.
+ /// One keeps the track with the larger number of hits if need be
+ AliMUONTrack *track1, *track2, *trackToRemove;
+ Int_t hitsInCommon, nHits1, nHits2;
+ Bool_t removedTrack1;
+ // Loop over first track of the pair
+ track1 = (AliMUONTrack*) fRecTracksPtr->First();
+ while (track1) {
+ removedTrack1 = kFALSE;
+ nHits1 = track1->GetNTrackHits();
+ // Loop over second track of the pair
+ track2 = (AliMUONTrack*) fRecTracksPtr->After(track1);
+ while (track2) {
+ nHits2 = track2->GetNTrackHits();
+ // number of hits in common between two tracks
+ hitsInCommon = track1->HitsInCommon(track2);
+ // check for identical tracks
+ if ((hitsInCommon == nHits1) || (hitsInCommon == nHits2)) {
+ // decide which track to remove
+ if (nHits2 > nHits1) {
+ // remove track1 and continue the first loop with the track next to track1
+ trackToRemove = track1;
+ track1 = (AliMUONTrack*) fRecTracksPtr->After(track1);
+ fRecTracksPtr->Remove(trackToRemove);
+ fRecTracksPtr->Compress(); // this is essential to retrieve the TClonesArray afterwards
+ fNRecTracks--;
+ removedTrack1 = kTRUE;
+ break;
+ } else {
+ // remove track2 and continue the second loop with the track next to track2
+ trackToRemove = track2;
+ track2 = (AliMUONTrack*) fRecTracksPtr->After(track2);
+ fRecTracksPtr->Remove(trackToRemove);
+ fRecTracksPtr->Compress(); // this is essential to retrieve the TClonesArray afterwards
+ fNRecTracks--;
+ }
+ } else track2 = (AliMUONTrack*) fRecTracksPtr->After(track2);
+ } // track2
+ if (removedTrack1) continue;
+ track1 = (AliMUONTrack*) fRecTracksPtr->After(track1);
+ } // track1
+ return;
}
-//________________________________________________________
-void chfill(Int_t &id, Float_t &x, Float_t &y, Float_t &w)
+ //__________________________________________________________________________
+void AliMUONTrackReconstructor::RemoveDoubleTracks(void)
{
- //
- // fill histo like hfill in fortran
- char name[5];
- sprintf(name,"h%d",id);
- TH1F *h1 = (TH1F*) gDirectory->Get(name);
- h1->Fill(x);
+ /// To remove double tracks:
+ /// Tracks are considered identical if more than half of the hits of the track
+ /// which has the smaller number of hits are in common with the other track.
+ /// Among two identical tracks, one keeps the track with the larger number of hits
+ /// or, if these numbers are equal, the track with the minimum chi2.
+ AliMUONTrack *track1, *track2, *trackToRemove;
+ Int_t hitsInCommon, nHits1, nHits2;
+ Bool_t removedTrack1;
+ // Loop over first track of the pair
+ track1 = (AliMUONTrack*) fRecTracksPtr->First();
+ while (track1) {
+ removedTrack1 = kFALSE;
+ nHits1 = track1->GetNTrackHits();
+ // Loop over second track of the pair
+ track2 = (AliMUONTrack*) fRecTracksPtr->After(track1);
+ while (track2) {
+ nHits2 = track2->GetNTrackHits();
+ // number of hits in common between two tracks
+ hitsInCommon = track1->HitsInCommon(track2);
+ // check for identical tracks
+ if (((nHits1 < nHits2) && (2 * hitsInCommon > nHits1)) || (2 * hitsInCommon > nHits2)) {
+ // decide which track to remove
+ if ((nHits1 > nHits2) || ((nHits1 == nHits2) && (track1->GetFitFMin() <= track2->GetFitFMin()))) {
+ // remove track2 and continue the second loop with the track next to track2
+ trackToRemove = track2;
+ track2 = (AliMUONTrack*) fRecTracksPtr->After(track2);
+ fRecTracksPtr->Remove(trackToRemove);
+ fRecTracksPtr->Compress(); // this is essential to retrieve the TClonesArray afterwards
+ fNRecTracks--;
+ } else {
+ // else remove track1 and continue the first loop with the track next to track1
+ trackToRemove = track1;
+ track1 = (AliMUONTrack*) fRecTracksPtr->After(track1);
+ fRecTracksPtr->Remove(trackToRemove);
+ fRecTracksPtr->Compress(); // this is essential to retrieve the TClonesArray afterwards
+ fNRecTracks--;
+ removedTrack1 = kTRUE;
+ break;
+ }
+ } else track2 = (AliMUONTrack*) fRecTracksPtr->After(track2);
+ } // track2
+ if (removedTrack1) continue;
+ track1 = (AliMUONTrack*) fRecTracksPtr->After(track1);
+ } // track1
+ return;
}
-//_________________________________________________________
-void chfill2(Int_t &id, Float_t &x, Float_t &y, Float_t &w)
+ //__________________________________________________________________________
+void AliMUONTrackReconstructor::FollowTracks(void)
{
- //
- // fill histo like hfill2 in fortran
- char name[5];
- sprintf(name,"h%d",id);
- TH2F *h2 = (TH2F*) gDirectory->Get(name);
- h2->Fill(x,y,w);
+ /// Follow tracks in stations(1..) 3, 2 and 1
+ AliDebug(1,"Enter FollowTracks");
+
+ AliMUONTrack *track, *nextTrack;
+ AliMUONTrackParam *trackParamAtFirstHit;
+ Double_t numberOfDegFree, chi2Norm;
+ Int_t CurrentNRecTracks;
+
+ for (Int_t station = 2; station >= 0; station--) {
+ // Save the actual number of reconstructed track in case of
+ // tracks are added or suppressed during the tracking procedure
+ // !! Do not compress fRecTracksPtr until the end of the loop over tracks !!
+ CurrentNRecTracks = fNRecTracks;
+ for (Int_t iRecTrack = 0; iRecTrack <CurrentNRecTracks; iRecTrack++) {
+ AliDebug(1,Form("FollowTracks: track candidate(1..) %d", iRecTrack+1));
+ track = (AliMUONTrack*) fRecTracksPtr->UncheckedAt(iRecTrack);
+ // Fit the track:
+ // Do not take into account the multiple scattering to speed up the fit
+ // Calculate the track parameter covariance matrix
+ // If "station" is station(1..) 3 then use the vertex to better constrain the fit
+ if (station==2) {
+ SetVertexForFit(track);
+ track->SetFitWithVertex(kTRUE);
+ } else track->SetFitWithVertex(kFALSE);
+ Fit(track,kFALSE, kTRUE);
+ // Remove the track if the normalized chi2 is too high
+ numberOfDegFree = (2. * track->GetNTrackHits() - 5.);
+ if (numberOfDegFree > 0) chi2Norm = track->GetFitFMin() / numberOfDegFree;
+ else chi2Norm = 1.e10;
+ if (chi2Norm > fgkMaxNormChi2) {
+ fRecTracksPtr->Remove(track);
+ fNRecTracks--;
+ continue;
+ }
+ // Add MCS effects in parameter covariances
+ trackParamAtFirstHit = (AliMUONTrackParam*) (track->GetTrackParamAtHit()->First());
+ AliMUONTrackExtrap::AddMCSEffectInTrackParamCov(trackParamAtFirstHit,AliMUONConstants::ChamberThicknessInX0(),1.);
+ // Printout for debuging
+ if ((AliLog::GetDebugLevel("MUON","AliMUONTrackReconstructor") >= 2) || (AliLog::GetGlobalDebugLevel() >= 2)) {
+ cout<<endl<<"Track parameter covariances at first hit with multiple Coulomb scattering effects:"<<endl;
+ trackParamAtFirstHit->GetCovariances()->Print();
+ }
+ // Look for compatible hitForRec in station(0..) "station"
+ FollowTrackInStation(track,station);
+ }
+ // Compress fRecTracksPtr for the next step
+ fRecTracksPtr->Compress();
+ // Keep only the best tracks if required
+ if (!fgkTrackAllTracks) RemoveDoubleTracks();
+ }
+
+ // Last fit of track candidates with all station
+ // Take into account the multiple scattering and remove bad tracks
+ Int_t trackIndex = -1;
+ track = (AliMUONTrack*) fRecTracksPtr->First();
+ while (track) {
+ trackIndex++;
+ nextTrack = (AliMUONTrack*) fRecTracksPtr->After(track); // prepare next track
+ track->SetFitWithVertex(kFALSE); // just to be sure
+ Fit(track,kTRUE, kFALSE);
+ // Printout for debuging
+ if (AliLog::GetGlobalDebugLevel() >= 3) {
+ cout << "FollowTracks: track candidate(0..) " << trackIndex << " after final fit" << endl;
+ track->RecursiveDump();
+ }
+ // Remove the track if the normalized chi2 is too high
+ numberOfDegFree = (2.0 * track->GetNTrackHits() - 5);
+ if (numberOfDegFree > 0) chi2Norm = track->GetFitFMin() / numberOfDegFree;
+ else chi2Norm = 1.e10;
+ if (chi2Norm > fgkMaxNormChi2) {
+ fRecTracksPtr->Remove(track);
+ fNRecTracks--;
+ }
+ track = nextTrack;
+ }
+ fRecTracksPtr->Compress();
+
}
-//__________________________________________
-void chf1(Int_t &id, Float_t &x, Float_t &w)
+ //__________________________________________________________________________
+void AliMUONTrackReconstructor::FollowTrackInStation(AliMUONTrack* trackCandidate, Int_t nextStation)
{
+ /// Follow trackCandidate in station(0..) nextStation and search for compatible HitForRec(s)
+ /// Keep all possibilities or only the best one(s) according to the flag fgkTrackAllTracks:
+ /// kTRUE: duplicate "trackCandidate" if there are several possibilities and add the new tracks at the end of
+ /// fRecTracksPtr to avoid conficts with other track candidates at this current stage of the tracking procedure.
+ /// Remove the obsolete "trackCandidate" at the end.
+ /// kFALSE: add only the best hit(s) to the "trackCandidate". Try to add a couple of hits in priority.
+ AliDebug(1,Form("Enter FollowTrackInStation(1..) %d", nextStation+1));
+
+ Int_t ch1 = 2*nextStation;
+ Int_t ch2 = 2*nextStation+1;
+ Double_t zCh2 = AliMUONConstants::DefaultChamberZ(ch2);
+ Double_t chi2WithOneHitForRec = 1.e10;
+ Double_t chi2WithTwoHitForRec = 1.e10;
+ Double_t maxChi2WithOneHitForRec = 2.*fgkMaxNormChi2; // 2 because 2 quantities in chi2
+ Double_t maxChi2WithTwoHitForRec = 4.*fgkMaxNormChi2; // 4 because 4 quantities in chi2
+ Double_t bestChi2WithOneHitForRec = maxChi2WithOneHitForRec;
+ Double_t bestChi2WithTwoHitForRec = maxChi2WithTwoHitForRec;
+ AliMUONTrack *newTrack = 0x0;
+ AliMUONHitForRec *hitForRecCh1, *hitForRecCh2;
+ AliMUONHitForRec *bestHitForRec1 = 0x0, *bestHitForRec2 = 0x0;
+ Bool_t *hitForRecCh1Used = new Bool_t[fNHitsForRecPerChamber[ch1]];
+ for (Int_t hit1 = 0; hit1 < fNHitsForRecPerChamber[ch1]; hit1++) hitForRecCh1Used[hit1] = kFALSE;
//
- // fill histo like hf1 in fortran
- char name[5];
- sprintf(name,"h%d",id);
- TH1F *h1 = (TH1F*) gDirectory->Get(name);
- h1->Fill(x,w);
-}
-
-//_________________
-void hist_create()
-{
+ //Extrapolate trackCandidate to chamber "ch2" to save computing time in the next steps
+ AliMUONTrackParam *extrapTrackParamPtr = trackCandidate->GetExtrapTrackParam();
+ *extrapTrackParamPtr = *((AliMUONTrackParam*)(trackCandidate->GetTrackParamAtHit()->First()));
+ AliMUONTrackExtrap::ExtrapToZCov(extrapTrackParamPtr, zCh2);
+ AliMUONTrackParam extrapTrackParamSave(*extrapTrackParamPtr);
//
- // Create an output file ("reconst.root")
- // Create some histograms and an ntuple
-
- gAliFileGlobal = new TFile("reconst.root","RECREATE","Ntuple - reconstruction");
-
- gAliNtupleGlobal = new TTree("ntuple","Reconst ntuple");
- gAliNtupleGlobal->Branch("ievr",&NtupleSt.ievr,"ievr/I");
- gAliNtupleGlobal->Branch("ntrackr",&NtupleSt.ntrackr,"ntrackr/I");
- gAliNtupleGlobal->Branch("istatr",&NtupleSt.istatr[0],"istatr[500]/I");
- gAliNtupleGlobal->Branch("isignr",&NtupleSt.isignr[0],"isignr[500]/I");
- gAliNtupleGlobal->Branch("pxr",&NtupleSt.pxr[0],"pxr[500]/F");
- gAliNtupleGlobal->Branch("pyr",&NtupleSt.pyr[0],"pyr[500]/F");
- gAliNtupleGlobal->Branch("pzr",&NtupleSt.pzr[0],"pzr[500]/F");
- gAliNtupleGlobal->Branch("zvr",&NtupleSt.zvr[0],"zvr[500]/F");
- gAliNtupleGlobal->Branch("chi2r",&NtupleSt.chi2r[0],"chi2r[500]/F");
- gAliNtupleGlobal->Branch("pxv",&NtupleSt.pxv[0],"pxv[500]/F");
- gAliNtupleGlobal->Branch("pyv",&NtupleSt.pyv[0],"pyv[500]/F");
- gAliNtupleGlobal->Branch("pzv",&NtupleSt.pzv[0],"pzv[500]/F");
-
- // test aliroot
-
- new TH1F("h100","particule id du hit geant",20,0.,20.);
- new TH1F("h101","position en x du hit geant",100,-200.,200.);
- new TH1F("h102","position en y du hit geant",100,-200.,200.);
- new TH1F("h103","chambre de tracking concernee",15,0.,14.);
- new TH1F("h104","moment ptot du hit geant",50,0.,100.);
- new TH1F("h105","px au vertex",50,0.,20.);
- new TH1F("h106","py au vertex",50,0.,20.);
- new TH1F("h107","pz au vertex",50,0.,20.);
- new TH1F("h108","position zv",50,-15.,15.);
- new TH1F("h109","position en x du hit reconstruit",100,-300.,300.);
- new TH1F("h110","position en y du hit reconstruit",100,-300.,300.);
- new TH1F("h111","delta x station 1",100,-0.3,0.3);
- new TH1F("h112","delta x station 2",100,-0.3,0.3);
- new TH1F("h113","delta x station 3",100,-0.3,0.3);
- new TH1F("h114","delta x station 4",100,-0.5,0.5);
- new TH1F("h115","delta x station 5",100,-0.5,0.5);
- new TH1F("h116","delta x station 1",100,-2,2);
- new TH1F("h117","delta x station 2",100,-2,2);
- new TH1F("h121","delta y station 1",100,-0.04,0.04);
- new TH1F("h122","delta y station 2",100,-0.04,0.04);
- new TH1F("h123","delta y station 3",100,-0.04,0.04);
- new TH1F("h124","delta y station 4",100,-0.04,0.04);
- new TH1F("h125","delta y station 5",100,-0.04,0.04);
-
- /* char hname[30];
- char hname1[30];
- for (int i=0;i<10;i++) {
- sprintf(hname,"deltax%d",i);
- sprintf(hname1,"h12%d",i);
- new TH1F(hname1,hname ,100,-0.4,0.4);
- sprintf(hname,"deltay%d",i);
- sprintf(hname1,"h13%d",i);
- new TH1F(hname1,hname ,100,-0.4,0.4);
+ // Printout for debuging
+ if ((AliLog::GetDebugLevel("MUON","AliMUONTrackReconstructor") >= 2) || (AliLog::GetGlobalDebugLevel() >= 2)) {
+ TMatrixD* paramCovForDebug = extrapTrackParamPtr->GetCovariances();
+ cout<<endl<<"Track parameter covariances at first hit extrapolated to z = "<<zCh2<<":"<<endl;
+ paramCovForDebug->Print();
+ }
+ //
+ // look for candidates in chamber 2
+ // Printout for debuging
+ if ((AliLog::GetDebugLevel("MUON","AliMUONTrackReconstructor") >= 1) || (AliLog::GetGlobalDebugLevel() >= 1)) {
+ cout << "FollowTrackInStation: look for hits in chamber(1..): " << ch2+1 << endl;
+ }
+ for (Int_t hit2 = 0; hit2 < fNHitsForRecPerChamber[ch2]; hit2++) {
+ hitForRecCh2 = (AliMUONHitForRec*) fHitsForRecPtr->UncheckedAt(fIndexOfFirstHitForRecPerChamber[ch2]+hit2);
+ // extrapolate track parameters and covariances only once for this hit
+ AliMUONTrackExtrap::ExtrapToZCov(extrapTrackParamPtr, hitForRecCh2->GetZ());
+ chi2WithOneHitForRec = trackCandidate->TryOneHitForRec(hitForRecCh2);
+ // if good chi2 then try to attach a hitForRec in the other chamber too
+ if (chi2WithOneHitForRec < maxChi2WithOneHitForRec) {
+ // Printout for debuging
+ if ((AliLog::GetDebugLevel("MUON","AliMUONTrackReconstructor") >= 1) || (AliLog::GetGlobalDebugLevel() >= 1)) {
+ cout << "FollowTrackInStation: look for second hits in chamber(1..): " << ch1+1 << endl;
}
- */
- new TH2F("h2000","VAR X st. 5",30,3.0,183.0,100,0.,25.);
- new TH2F("h2001","VAR Y st. 5",30,3.0,183.0,100,0.,25.);
-
- new TH2F("h2500","P vs X HHIT",30,3.0,183.0,200,0.,200.);
- new TH2F("h2501","P vs X HHIT**2",30,3.0,183.0,200,0.,5000.);
- new TH2F("h2502","P vs X EPH2 st. 5",30,3.0,183.0,100,0.,0.000005);
- new TH2F("h2503","P vs X EAL2 st. 5",30,3.0,183.0,100,0.,0.01);
- //new TH2F("h2504","P vs X EXM2 st. 5",30,3.0,183.0,100,0.,1.5);
- new TH2F("h2504","P vs X EXM2 st. 5",30,3.0,183.0,100,0.,0.1);
- new TH2F("h2505","P vs X EYM2 st. 5",30,3.0,183.0,100,0.,30.);
-
- new TH2F("h2507","P vs X EPH st. 5",30,3.0,183.0,100,0.,0.003);
- new TH2F("h2508","P vs X EAL st. 5",30,3.0,183.0,100,0.,0.3);
- //new TH2F("h2509","P vs X EXM st. 5",30,3.0,183.0,100,0.,1.5);
- new TH2F("h2509","P vs X EXM st. 5",30,3.0,183.0,100,0.,0.4);
- new TH2F("h2510","P vs X EYM st. 5",30,3.0,183.0,100,0.,30.);
-
- new TH2F("h2511","P vs X EPH cut st. 5",30,3.0,183.0,100,0.,0.01);
- new TH2F("h2512","P vs X EAL cut st. 5",30,3.0,183.0,100,0.,0.3);
- //new TH2F("h2513","P vs X EXM cut st. 5",30,3.0,183.0,100,0.,1.5);
- new TH2F("h2513","P vs X EXM cut st. 5",30,3.0,183.0,100,0.,0.4);
- new TH2F("h2514","P vs X EYM cut st. 5",30,3.0,183.0,100,0.,30.);
- // 4
- new TH2F("h2400","P vs X HHIT",30,3.0,183.0,200,0.,200.);
- new TH2F("h2401","P vs X HHIT**2",30,3.0,183.0,200,0.,5000.);
- new TH2F("h2402","P vs X EPH2 st. 4",30,3.0,183.0,100,0.,0.000005);
- new TH2F("h2403","P vs X EAL2 st. 4",30,3.0,183.0,100,0.,0.05);
- //new TH2F("h2404","P vs X EXM2 st. 4",30,3.0,183.0,100,0.,1.5);
- new TH2F("h2404","P vs X EXM2 st. 4",30,3.0,183.0,100,0.,0.1);
- new TH2F("h2405","P vs X EYM2 st. 4",30,3.0,183.0,100,0.,30.);
-
- new TH2F("h2407","P vs X EPH st. 4",30,3.0,183.0,100,0.,0.003);
- new TH2F("h2408","P vs X EAL st. 4",30,3.0,183.0,100,0.,0.3);
- //new TH2F("h2409","P vs X EXM st. 4",30,3.0,183.0,100,0.,1.5);
- new TH2F("h2409","P vs X EXM st. 4",30,3.0,183.0,100,0.,0.1);
- new TH2F("h2410","P vs X EYM st. 4",30,3.0,183.0,100,0.,30.);
-
- new TH2F("h2411","P vs X EPH cut st. 4",30,3.0,183.0,100,0.,0.01);
- new TH2F("h2412","P vs X EAL cut st. 4",30,3.0,183.0,100,0.,0.3);
- //new TH2F("h2413","P vs X EXM cut st. 4",30,3.0,183.0,100,0.,1.5);
- new TH2F("h2413","P vs X EXM cut st. 4",30,3.0,183.0,100,0.,0.1);
- new TH2F("h2414","P vs X EYM cut st. 4",30,3.0,183.0,100,0.,30.);
- // 3
- new TH1F("h2301","P2",30,3.0,183.0);
- new TH2F("h2302","P2 vs X EPH2 st. 3",30,3.0,183.0,100,0.,0.0006);
- new TH2F("h2303","P2 vs X EAL2 st. 3",30,3.0,183.0,100,0.,0.0005);
- //new TH2F("h2304","P2 vs X EXM2 st. 3",30,3.0,183.0,100,0.,1.5);
- new TH2F("h2304","P2 vs X EXM2 st. 3",30,3.0,183.0,100,0.,2.);
- new TH2F("h2305","P2 vs X EYM2 st. 3",30,3.0,183.0,100,0.,3.);
-
- new TH2F("h2307","P vs X EPH2 st. 3",30,3.0,183.0,100,0.,0.0006);
- new TH2F("h2308","P vs X EAL2 st. 3",30,3.0,183.0,100,0.,0.005);
- //new TH2F("h2309","P vs X EXM2 st. 3",30,3.0,183.0,100,0.,1.5);
- new TH2F("h2309","P vs X EXM2 st. 3",30,3.0,183.0,100,0.,2.);
- new TH2F("h2310","P vs X EYM2 st. 3",30,3.0,183.0,100,0.,3.);
-
- new TH2F("h2311","P vs X EPH cut st. 3",30,3.0,183.0,100,0.,0.06);
- new TH2F("h2312","P vs X EAL cut st. 3",30,3.0,183.0,100,0.,0.05);
- //new TH2F("h2313","P vs X EXM cut st. 3",30,3.0,183.0,100,0.,1.5);
- new TH2F("h2313","P vs X EXM cut st. 3",30,3.0,183.0,100,0.,6.);
- new TH2F("h2314","P vs X EYM cut st. 3",30,3.0,183.0,100,0.,7.);
-
- new TH2F("h2315","P2 vs X EPH cut st. 3",30,3.0,183.0,100,0.,0.06);
- new TH2F("h2316","P2 vs X EAL cut st. 3",30,3.0,183.0,100,0.,0.05);
- //new TH2F("h2317","P2 vs X EXM cut st. 3",30,3.0,183.0,100,0.,1.5);
- new TH2F("h2317","P2 vs X EXM cut st. 3",30,3.0,183.0,100,0.,6.);
- new TH2F("h2318","P2 vs X EYM cut st. 3",30,3.0,183.0,100,0.,7.);
-
- // 2
- new TH1F("h2201","P2",30,3.0,183.0);
- new TH2F("h2202","P2 vs X EPH2 st. 2",30,3.0,183.0,100,0.,0.0006);
- new TH2F("h2203","P2 vs X EAL2 st. 2",30,3.0,183.0,100,0.,0.005);
- //new TH2F("h2204","P2 vs X EXM2 st. 2",30,3.0,183.0,100,0.,1.5);
- new TH2F("h2204","P2 vs X EXM2 st. 2",30,3.0,183.0,100,0.,7.);
- new TH2F("h2205","P2 vs X EYM2 st. 2",30,3.0,183.0,100,0.,5.);
-
- new TH2F("h2207","P vs X EPH2 st. 2",30,3.0,183.0,100,0.,0.0006);
- new TH2F("h2208","P vs X EAL2 st. 2",30,3.0,183.0,100,0.,0.005);
- //new TH2F("h2209","P vs X EXM2 st. 2",30,3.0,183.0,100,0.,1.5);
- new TH2F("h2209","P vs X EXM2 st. 2",30,3.0,183.0,100,0.,7.);
- new TH2F("h2210","P vs X EYM2 st. 2",30,3.0,183.0,100,0.,5.);
-
- new TH2F("h2211","P vs X EPH cut st. 2",30,3.0,183.0,100,0.,0.05);
- new TH2F("h2212","P vs X EAL cut st. 2",30,3.0,183.0,100,0.,0.2);
- //new TH2F("h2213","P vs X EXM cut st. 2",30,3.0,183.0,100,0.,1.5);
- new TH2F("h2213","P vs X EXM cut st. 2",30,3.0,183.0,100,0.,11.);
- new TH2F("h2214","P vs X EYM cut st. 2",30,3.0,183.0,100,0.,10.);
-
- new TH2F("h2215","P2 vs X EPH cut st. 2",30,3.0,183.0,100,0.,0.05);
- new TH2F("h2216","P2 vs X EAL cut st. 2",30,3.0,183.0,100,0.,0.2);
- //new TH2F("h2217","P2 vs X EXM cut st. 2",30,3.0,183.0,100,0.,1.5);
- new TH2F("h2217","P2 vs X EXM cut st. 2",30,3.0,183.0,100,0.,11.);
- new TH2F("h2218","P2 vs X EYM cut st. 2",30,3.0,183.0,100,0.,10.);
-
- // 1
- new TH2F("h2102","P2 vs X EPH2 st. 2",30,3.0,183.0,100,0.,0.0006);
- new TH2F("h2103","P2 vs X EAL2 st. 2",30,3.0,183.0,100,0.,0.005);
- //new TH2F("h2104","P2 vs X EXM2 st. 2",30,3.0,183.0,100,0.,1.5);
- new TH2F("h2104","P2 vs X EXM2 st. 2",30,3.0,183.0,100,0.,7.);
- new TH2F("h2105","P2 vs X EYM2 st. 2",30,3.0,183.0,100,0.,7.);
-
- new TH2F("h2107","P vs X EPH2 st. 2",30,3.0,183.0,100,0.,0.0006);
- new TH2F("h2108","P vs X EAL2 st. 2",30,3.0,183.0,100,0.,0.005);
- //new TH2F("h2109","P vs X EXM2 st. 2",30,3.0,183.0,100,0.,1.5);
- new TH2F("h2109","P vs X EXM2 st. 2",30,3.0,183.0,100,0.,7.);
- new TH2F("h2110","P vs X EYM2 st. 2",30,3.0,183.0,100,0.,7.);
-
- new TH2F("h2111","P vs X EPH cut st. 2",30,3.0,183.0,100,0.,0.1);
- new TH2F("h2112","P vs X EAL cut st. 2",30,3.0,183.0,100,0.,0.2);
- //new TH2F("h2113","P vs X EXM cut st. 2",30,3.0,183.0,100,0.,1.5);
- new TH2F("h2113","P vs X EXM cut st. 2",30,3.0,183.0,100,0.,11.);
- new TH2F("h2114","P vs X EYM cut st. 2",30,3.0,183.0,100,0.,11.);
-
- new TH2F("h2115","P2 vs X EPH cut st. 2",30,3.0,183.0,100,0.,0.1);
- new TH2F("h2116","P2 vs X EAL cut st. 2",30,3.0,183.0,100,0.,0.2);
- //new TH2F("h2117","P2 vs X EXM cut st. 2",30,3.0,183.0,100,0.,1.5);
- new TH2F("h2117","P2 vs X EXM cut st. 2",30,3.0,183.0,100,0.,11.);
- new TH2F("h2118","P2 vs X EYM cut st. 2",30,3.0,183.0,100,0.,11.);
-
- // 2,3,4,5
- new TH1F("h2701","P2 fit 2",30,3.0,183.0);
- new TH2F("h2702","P2 vs X EPH2 st. 1 fit 2",30,3.0,183.0,100,0.,0.0006);
- new TH2F("h2703","P2 vs X EAL2 st. 1 fit 2",30,3.0,183.0,100,0.,0.005);
- // new TH2F("h2704","P2 vs X EXM2 st. 1 fit 2",30,3.0,183.0,100,0.,1.5);
- new TH2F("h2704","P2 vs X EXM2 st. 1 fit 2",30,3.0,183.0,100,0.,2.);
- new TH2F("h2705","P2 vs X EYM2 st. 1 fit 2",30,3.0,183.0,100,0.,3.);
-
- new TH2F("h2707","P vs X EPH2 st. 1 fit 2",30,3.0,183.0,100,0.,0.0006);
- new TH2F("h2708","P vs X EAL2 st. 1 fit 2",30,3.0,183.0,100,0.,0.005);
- //new TH2F("h2709","P vs X EXM2 st. 1 fit 2",30,3.0,183.0,100,0.,1.5);
- new TH2F("h2709","P vs X EXM2 st. 1 fit 2",30,3.0,183.0,100,0.,2.);
- new TH2F("h2710","P vs X EYM2 st. 1 fit 2",30,3.0,183.0,100,0.,3.);
-
- new TH2F("h2711","P vs X EPH cut st. 1 fit 2",30,3.0,183.0,100,0.,0.07);
- new TH2F("h2712","P vs X EAL cut st. 1 fit 2",30,3.0,183.0,100,0.,0.2);
- //new TH2F("h2713","P vs X EXM cut st. 1 fit 2",30,3.0,183.0,100,0.,1.5);
- new TH2F("h2713","P vs X EXM cut st. 1 fit 2",30,3.0,183.0,100,0.,6.);
- new TH2F("h2714","P vs X EYM cut st. 1 fit 2",30,3.0,183.0,100,0.,7.);
-
- new TH2F("h2715","P2 vs X EPH cut st. 1 fit 2",30,3.0,183.0,100,0.,0.07);
- new TH2F("h2716","P2 vs X EAL cut st. 1 fit 2",30,3.0,183.0,100,0.,0.2);
- //new TH2F("h2717","P2 vs X EXM cut st. 1 fit 2",30,3.0,183.0,100,0.,1.5);
- new TH2F("h2717","P2 vs X EXM cut st. 1 fit 2",30,3.0,183.0,100,0.,6.);
- new TH2F("h2718","P2 vs X EYM cut st. 1 fit 2",30,3.0,183.0,100,0.,7.);
-
- // 1,3,4,5
- new TH1F("h2801","P2 fit 1",30,3.0,183.0);
- new TH2F("h2802","P2 vs X EPH2 st. 2 fit 1",30,3.0,183.0,100,0.,0.0006);
- new TH2F("h2803","P2 vs X EAL2 st. 2 fit 1",30,3.0,183.0,100,0.,0.005);
- //new TH2F("h2804","P2 vs X EXM2 st. 2 fit 1",30,3.0,183.0,100,0.,1.5);
- new TH2F("h2804","P2 vs X EXM2 st. 2 fit 1",30,3.0,183.0,100,0.,2.);
- new TH2F("h2805","P2 vs X EYM2 st. 2 fit 1",30,3.0,183.0,100,0.,3.);
-
- new TH2F("h2807","P vs X EPH2 st. 2 fit 1",30,3.0,183.0,100,0.,0.0006);
- new TH2F("h2808","P vs X EAL2 st. 2 fit 1",30,3.0,183.0,100,0.,0.005);
- //new TH2F("h2809","P vs X EXM2 st. 2 fit 1",30,3.0,183.0,100,0.,1.5);
- new TH2F("h2809","P vs X EXM2 st. 2 fit 1",30,3.0,183.0,100,0.,2.);
- new TH2F("h2810","P vs X EYM2 st. 2 fit 1",30,3.0,183.0,100,0.,3.);
-
- new TH2F("h2811","P vs X EPH cut st. 2 fit 1",30,3.0,183.0,100,0.,0.05);
- new TH2F("h2812","P vs X EAL cut st. 2 fit 1",30,3.0,183.0,100,0.,0.2);
- //new TH2F("h2813","P vs X EXM cut st. 2 fit 1",30,3.0,183.0,100,0.,1.5);
- new TH2F("h2813","P vs X EXM cut st. 2 fit 1",30,3.0,183.0,100,0.,5.);
- new TH2F("h2814","P vs X EYM cut st. 2 fit 1",30,3.0,183.0,100,0.,7.);
-
- new TH2F("h2815","P2 vs X EPH cut st. 2 fit 1",30,3.0,183.0,100,0.,0.05);
- new TH2F("h2816","P2 vs X EAL cut st. 2 fit 1",30,3.0,183.0,100,0.,0.2);
- //new TH2F("h2817","P2 vs X EXM cut st. 2 fit 1",30,3.0,183.0,100,0.,1.5);
- new TH2F("h2817","P2 vs X EXM cut st. 2 fit 1",30,3.0,183.0,100,0.,5.);
- new TH2F("h2818","P2 vs X EYM cut st. 2 fit 1",30,3.0,183.0,100,0.,7.);
-
-
- new TH2F("h1111","dx vs x station 1",30,-250.,250.,30,-0.5,0.5);
- new TH2F("h1112","dx vs x station 2",30,-250.,250.,30,-0.5,0.5);
- new TH2F("h1113","dx vs x station 3",30,-250.,250.,30,-0.5,0.5);
- new TH2F("h1114","dx vs x station 4",30,-250.,250.,30,-0.5,0.5);
- new TH2F("h1115","dx vs x station 5",30,-250.,250.,30,-0.5,0.5);
- new TH2F("h1121","dy vs y station 1",30,-250.,250.,30,-0.04,0.04);
- new TH2F("h1122","dy vs y station 2",30,-250.,250.,30,-0.04,0.04);
- new TH2F("h1123","dy vs y station 3",30,-250.,250.,30,-0.04,0.04);
- new TH2F("h1124","dy vs y station 4",30,-250.,250.,30,-0.04,0.04);
- new TH2F("h1125","dy vs y station 5",30,-250.,250.,30,-0.04,0.04);
-
- // fin de test
-
- new TH1F("h500","Acceptance en H st. 4",500,0.,500.);
- new TH1F("h600","Acceptance en H st. 5",500,0.,500.);
- new TH1F("h700","X vertex track found",200,-10.,10.);
- new TH1F("h701","Y vertex track found",200,-10.,10.);
- new TH1F("h800","Rap. muon gen.",100,0.,5.);
- new TH1F("h801","Rap. muon gen. recons.",100,0.,5.);
- new TH1F("h802","Rap. muon gen. ghost ",100,0.,5.);
- new TH1F("h900","Pt muon gen.",100,0.,20.);
- new TH1F("h901","Pt muon gen. recons.",100,0.,20.);
- new TH1F("h902","Pt muon gen. ghost",100,0.,20.);
- new TH1F("h910","phi muon gen.",100,-10.,10.);
- new TH1F("h911","phi muon gen. recons.",100,-10.,10.);
- new TH1F("h912","phi muon gen. ghost",100,-10.,10.);
- new TH2F("h1001","Y VS X hit st. 1",300,-300.,300.,300,-300.,300.);
- new TH2F("h1002","Y VS X hit st. 2",300,-300.,300.,300,-300.,300.);
- new TH2F("h1003","Y VS X hit st. 3",300,-300.,300.,300,-300.,300.);
- new TH2F("h1004","Y VS X hit st. 4",300,-300.,300.,300,-300.,300.);
- new TH2F("h1005","Y VS X hit st. 5",300,-300.,300.,300,-300.,300.);
- // Histos variance dans 4
- new TH2F("h11","VAR X st. 4",30,3.0,183.0,100,0.,2.);
- new TH2F("h12","VAR Y st. 4",30,3.0,183.0,100,0.,600.);
- new TH2F("h13","VAR PHI st. 4",30,3.0,183.0,100,0.,0.0001);
- new TH2F("h14","VAR ALM st. 4",30,3.0,183.0,100,0.,0.05);
- new TH1F("h15","P",30,3.0,183.0);
- new TH1F("h411","VAR X st. 4",100,-1.42,1.42);
- new TH1F("h412","VAR Y st. 4",100,-25.,25.);
- new TH1F("h413","VAR PHI st. 4",100,-0.01,0.01);
- new TH1F("h414","VAR ALM st. 4",100,-0.23,0.23);
- // histo2
- new TH2F("h211","histo2-VAR X st. 4",30,3.0,183.0,100,0.,2.);
- new TH2F("h212","histo2-VAR Y st. 4",30,3.0,183.0,100,0.,600.);
- new TH1F("h213","histo2-VAR X st. 4",100,-1.42,1.42);
- new TH1F("h214","histo2-VAR Y st. 4",100,-25.,25.);
- new TH1F("h215","histo2-P",30,3.0,183.0);
-
- // Histos variance dans 2
- new TH2F("h21","VAR X st. 2",30,3.0,183.0,100,0.,3.);
- new TH2F("h22","VAR Y st. 2",30,3.0,183.0,100,0.,7.);
- new TH2F("h23","VAR PHI st. 2",30,3.0,183.0,100,0.,0.006);
- new TH2F("h24","VAR ALM st. 2",30,3.0,183.0,100,0.,0.005);
- new TH1F("h25","P",30,3.0,183.0);
- new TH1F("h421","VAR X st. 2",100,-1.72,1.72);
- new TH1F("h422","VAR Y st. 2",100,-2.7,2.7);
- new TH1F("h423","VAR PHI st. 2",100,-0.08,0.08);
- new TH1F("h424","VAR ALM st. 2",100,-0.072,0.072);
- // histo2
- new TH2F("h221","histo2-VAR X st. 2",30,3.0,183.0,100,0.,3.);
- new TH2F("h222","histo2-VAR Y st. 2",30,3.0,183.0,100,0.,7.);
- new TH1F("h223","histo2-VAR X st. 2",100,-1.72,1.72);
- new TH1F("h224","histo2-VAR Y st. 2",100,-2.7,2.7);
- new TH1F("h225","histo2-P",30,3.0,183.0);
-
- // Histos variance dans 1
- new TH2F("h31","VAR X st. 1",30,3.0,183.0,100,0.,2.);
- new TH2F("h32","VAR Y st. 1",30,3.0,183.0,100,0.,0.5);
- new TH2F("h33","VAR PHI st. 1",30,3.0,183.0,100,0.,0.006);
- new TH2F("h34","VAR ALM st. 1",30,3.0,183.0,100,0.,0.005);
- new TH1F("h35","P",30,3.0,183.0);
- new TH1F("h431","VAR X st. 1",100,-1.42,1.42);
- new TH1F("h432","VAR Y st. 1",100,-0.72,0.72);
- new TH1F("h433","VAR PHI st. 1",100,-0.08,0.08);
- new TH1F("h434","VAR ALM st. 1",100,-0.072,0.072);
- // Histos variance dans 1
- new TH2F("h41","VAR X st. 1 fit 5,4,3,2,V",30,3.0,183.0,100,0.,4.);
- new TH2F("h42","VAR Y st. 1 fit 5,4,3,2,V",30,3.0,183.0,100,0.,20.);
- new TH2F("h43","VAR PHI st. 1 fit 5,4,3,2,V",30,3.0,183.0,100,0.,0.005);
- new TH2F("h44","VAR ALM st. 1 fit 5,4,3,2,V",30,3.0,183.0,100,0.,0.005);
- new TH1F("h45","P",30,3.0,183.0);
- new TH1F("h441","VAR X st. 1 fit 5,4,3,2,V",100,-2.,2.);
- new TH1F("h442","VAR Y st. 1 fit 5,4,3,2,V",100,-4.5,4.5);
- new TH1F("h443","VAR PHI st. 1 fit 5,4,3,2,V",100,-0.072,0.072);
- new TH1F("h444","VAR ALM st. 1 fit 5,4,3,2,V",100,-0.072,0.072);
- // histo2
- new TH2F("h241","histo2-VAR X st. 1 fit 5,4,3,2,V",30,3.0,183.0,100,0.,4.);
- new TH2F("h242","histo2-VAR Y st. 1 fit 5,4,3,2,V",30,3.0,183.0,100,0.,20.);
- new TH1F("h243","histo2-VAR X st. 1 fit 5,4,3,2,V",100,-2.,2.);
- new TH1F("h244","histo2-VAR Y st. 1 fit 5,4,3,2,V",100,-4.5,4.5);
- new TH1F("h245","histo2-P",30,3.0,183.0);
-
- // Histos variance dans 2
- new TH2F("h51","VAR X st. 2 fit 5,4,3,1,V",30,3.0,183.0,100,0.,0.5);
- new TH2F("h52","VAR Y st. 2 fit 5,4,3,1,V",30,3.0,183.0,100,0.,2.);
- new TH2F("h53","VAR PHI st. 2 fit 5,4,3,1,V",30,3.0,183.0,100,0.,0.005);
- new TH2F("h54","VAR ALM st. 2 fit 5,4,3,1,V",30,3.0,183.0,100,0.,0.01);
- new TH1F("h55","P",30,3.0,183.0);
- new TH1F("h451","VAR X st. 2 fit 5,4,3,1,V",100,-0.72,0.72);
- new TH1F("h452","VAR Y st. 2 fit 5,4,3,1,V",100,-1.42,1.42);
- new TH1F("h453","VAR PHI st. 2 fit 5,4,3,1,V",100,-0.072,0.072);
- new TH1F("h454","VAR ALM st. 2 fit 5,4,3,1,V",100,-0.1,0.1);
- new TH1F("h999","PTOT",30,3.0,183.0);
- // histo2
- new TH2F("h251","histo2-VAR X st. 2 fit 5,4,3,1,V",30,3.0,183.0,100,0.,0.5);
- new TH2F("h252","histo2-VAR Y st. 2 fit 5,4,3,1,V",30,3.0,183.0,100,0.,2.);
- new TH1F("h253","histo2-VAR X st. 2 fit 5,4,3,1,V",100,-0.72,0.72);
- new TH1F("h254","histo2-VAR Y st. 2 fit 5,4,3,1,V",100,-1.42,1.42);
- new TH1F("h255","histo2-P",30,3.0,183.0);
- // Histos variance dans 3
- new TH2F("h61","VAR X st. 3 fit 4,5,V",30,3.0,183.0,100,0.,5.);
- new TH2F("h62","VAR Y st. 3 fit 4,5,V",30,3.0,183.0,100,0.,2.);
- new TH2F("h63","VAR PHI st. 3 fit 4,5,V",30,3.0,183.0,100,0.,0.0006);
- new TH2F("h64","VAR ALM st. 3 fit 4,5,V",30,3.0,183.0,100,0.,0.0006);
- new TH1F("h65","P",30,3.0,183.0);
- new TH1F("h461","VAR X st. 3 fit 4,5,V",100,-2.25,2.25);
- new TH1F("h462","VAR Y st. 3 fit 4,5,V",100,-1.42,1.42);
- new TH1F("h463","VAR PHI st. 3 fit 4,5,V",100,-0.024,0.024);
- new TH1F("h464","VAR ALM st. 3 fit 4,5,V",100,-0.024,0.024);
- // histo2
- new TH2F("h261","histo2-VAR X st. 3 fit 4,5,V",30,3.0,183.0,100,0.,5.);
- new TH2F("h262","histo2-VAR Y st. 3 fit 4,5,V",30,3.0,183.0,100,0.,2.);
- new TH1F("h263","histo2-VAR X st. 3 fit 4,5,V",100,-2.25,2.25);
- new TH1F("h264","histo2-VAR Y st. 3 fit 4,5,V",100,-1.42,1.42);
- new TH1F("h265","Phisto2-",30,3.0,183.0);
- // Histos dx,dy distribution between chambers inside stations
- new TH1F("h71","DX in st. ID-70",100,-5.,5.);
- new TH1F("h81","DY in st. ID-80",100,-5.,5.);
- new TH1F("h72","DX in st. ID-70",100,-5.,5.);
- new TH1F("h82","DY in st. ID-80",100,-5.,5.);
- new TH1F("h73","DX in st. ID-70",100,-5.,5.);
- new TH1F("h83","DY in st. ID-80",100,-5.,5.);
- new TH1F("h74","DX in st. ID-70",100,-5.,5.);
- new TH1F("h84","DY in st. ID-80",100,-5.,5.);
- new TH1F("h75","DX in st. ID-70",100,-5.,5.);
- new TH1F("h85","DY in st. ID-80",100,-5.,5.);
-}
-
-//_____________________________________________________________________________
-void chfnt(Int_t &ievr, Int_t &ntrackr, Int_t *istatr, Int_t *isignr, Float_t *pxr, Float_t *pyr, Float_t *pzr, Float_t *zvr, Float_t *chi2r, Float_t *pxv, Float_t *pyv, Float_t *pzv)
-{
+ Bool_t foundSecondHit = kFALSE;
+ for (Int_t hit1 = 0; hit1 < fNHitsForRecPerChamber[ch1]; hit1++) {
+ hitForRecCh1 = (AliMUONHitForRec*) fHitsForRecPtr->UncheckedAt(fIndexOfFirstHitForRecPerChamber[ch1]+hit1);
+ chi2WithTwoHitForRec = trackCandidate->TryTwoHitForRec(hitForRecCh2, hitForRecCh1); // order hits like that to save computing time
+ // if good chi2 then create a new track by adding the 2 hitForRec to the "trackCandidate"
+ if (chi2WithTwoHitForRec < maxChi2WithTwoHitForRec) {
+ foundSecondHit = kTRUE;
+ if (fgkTrackAllTracks) {
+ // copy trackCandidate into a new track put at the end of fRecTracksPtr and add the new hitForRec's
+ newTrack = new ((*fRecTracksPtr)[fRecTracksPtr->GetLast()+1]) AliMUONTrack(*trackCandidate);
+ fNRecTracks++;
+ AliMUONTrackParam TrackParam1(extrapTrackParamSave);
+ AliMUONTrackExtrap::ExtrapToZ(&TrackParam1, hitForRecCh1->GetZ());
+ newTrack->AddTrackParamAtHit(&TrackParam1,hitForRecCh1);
+ AliMUONTrackParam TrackParam2(extrapTrackParamSave);
+ AliMUONTrackExtrap::ExtrapToZ(&TrackParam2, hitForRecCh2->GetZ());
+ newTrack->AddTrackParamAtHit(&TrackParam2,hitForRecCh2);
+ // Sort TrackParamAtHit according to increasing Z
+ newTrack->GetTrackParamAtHit()->Sort();
+ // Update the chi2 of the new track
+ if (newTrack->GetFitFMin()<0) newTrack->SetFitFMin(chi2WithTwoHitForRec);
+ else newTrack->SetFitFMin(newTrack->GetFitFMin() + chi2WithTwoHitForRec);
+ // Tag hitForRecCh1 as used
+ hitForRecCh1Used[hit1] = kTRUE;
+ // Printout for debuging
+ if ((AliLog::GetDebugLevel("MUON","AliMUONTrackReconstructor") >= 1) || (AliLog::GetGlobalDebugLevel() >= 1)) {
+ cout << "FollowTrackInStation: added two hits in station(1..): " << nextStation+1
+ << " (Chi2 = " << chi2WithTwoHitForRec << ")" << endl;
+ if (AliLog::GetGlobalDebugLevel() >= 3) newTrack->RecursiveDump();
+ }
+ } else if (chi2WithTwoHitForRec < bestChi2WithTwoHitForRec) {
+ // keep track of the best couple of hits
+ bestChi2WithTwoHitForRec = chi2WithTwoHitForRec;
+ bestHitForRec1 = hitForRecCh1;
+ bestHitForRec2 = hitForRecCh2;
+ }
+ }
+ }
+ // if no hitForRecCh1 found then consider to add hitForRecCh2 only
+ if (!foundSecondHit) {
+ if (fgkTrackAllTracks) {
+ // copy trackCandidate into a new track put at the end of fRecTracksPtr and add the new hitForRec's
+ newTrack = new ((*fRecTracksPtr)[fRecTracksPtr->GetLast()+1]) AliMUONTrack(*trackCandidate);
+ fNRecTracks++;
+ AliMUONTrackParam TrackParam1(extrapTrackParamSave);
+ AliMUONTrackExtrap::ExtrapToZ(&TrackParam1, hitForRecCh2->GetZ());
+ newTrack->AddTrackParamAtHit(&TrackParam1,hitForRecCh2);
+ // Sort TrackParamAtHit according to increasing Z
+ newTrack->GetTrackParamAtHit()->Sort();
+ // Update the chi2 of the new track
+ if (newTrack->GetFitFMin()<0) newTrack->SetFitFMin(chi2WithOneHitForRec);
+ else newTrack->SetFitFMin(newTrack->GetFitFMin() + chi2WithOneHitForRec);
+ // Printout for debuging
+ if ((AliLog::GetDebugLevel("MUON","AliMUONTrackReconstructor") >= 1) || (AliLog::GetGlobalDebugLevel() >= 1)) {
+ cout << "FollowTrackInStation: added one hit in chamber(1..): " << ch2+1
+ << " (Chi2 = " << chi2WithOneHitForRec << ")" << endl;
+ if (AliLog::GetGlobalDebugLevel() >= 3) newTrack->RecursiveDump();
+ }
+ } if (!bestHitForRec2 && chi2WithOneHitForRec < bestChi2WithOneHitForRec) {
+ // keep track of the best single hitForRec except if a couple
+ // of hits has already been found (i.e. bestHitForRec2!=0x0)
+ bestChi2WithOneHitForRec = chi2WithOneHitForRec;
+ bestHitForRec1 = hitForRecCh2;
+ }
+ }
+ }
+ // reset the extrapolated track parameter for next step
+ trackCandidate->SetExtrapTrackParam(&extrapTrackParamSave);
+ }
//
- // fill the ntuple
- NtupleSt.ievr = ievr;
- NtupleSt.ntrackr = ntrackr;
- for (Int_t i=0; i<500; i++) {
- NtupleSt.istatr[i] = istatr[i];
- NtupleSt.isignr[i] = isignr[i];
- NtupleSt.pxr[i] = pxr[i];
- NtupleSt.pyr[i] = pyr[i];
- NtupleSt.pzr[i] = pzr[i];
- NtupleSt.zvr[i] = zvr[i];
- NtupleSt.chi2r[i] = chi2r[i];
- NtupleSt.pxv[i] = pxv[i];
- NtupleSt.pyv[i] = pyv[i];
- NtupleSt.pzv[i] = pzv[i];
+ // look for candidates in chamber 1 not already attached to a track
+ // if we want to keep all possible tracks or if no good couple of hitForRec has been found
+ if (fgkTrackAllTracks || !bestHitForRec2) {
+ if ((AliLog::GetDebugLevel("MUON","AliMUONTrackReconstructor") >= 1) || (AliLog::GetGlobalDebugLevel() >= 1)) {
+ cout << "FollowTrackInStation: look for single hits in chamber(1..): " << ch1+1 << endl;
}
- gAliNtupleGlobal->Fill();
-}
-
-//________________
-void hist_closed()
-{
+ for (Int_t hit1 = 0; hit1 < fNHitsForRecPerChamber[ch1]; hit1++) {
+ hitForRecCh1 = (AliMUONHitForRec*) fHitsForRecPtr->UncheckedAt(fIndexOfFirstHitForRecPerChamber[ch1]+hit1);
+ if (hitForRecCh1Used[hit1]) continue; // Skip hitForRec already used
+ chi2WithOneHitForRec = trackCandidate->TryOneHitForRec(hitForRecCh1);
+ // if good chi2 then create a new track by adding the good hitForRec in "ch1" to the "trackCandidate"
+ // We do not try to attach a hitForRec in the other chamber too since it has already been done above
+ if (chi2WithOneHitForRec < maxChi2WithOneHitForRec) {
+ if (fgkTrackAllTracks) {
+ // copy trackCandidate into a new track put at the end of fRecTracksPtr and add the new hitForRec's
+ newTrack = new ((*fRecTracksPtr)[fRecTracksPtr->GetLast()+1]) AliMUONTrack(*trackCandidate);
+ fNRecTracks++;
+ AliMUONTrackParam TrackParam1(extrapTrackParamSave);
+ AliMUONTrackExtrap::ExtrapToZ(&TrackParam1, hitForRecCh1->GetZ());
+ newTrack->AddTrackParamAtHit(&TrackParam1,hitForRecCh1);
+ // Sort TrackParamAtHit according to increasing Z
+ newTrack->GetTrackParamAtHit()->Sort();
+ // Update the chi2 of the new track
+ if (newTrack->GetFitFMin()<0) newTrack->SetFitFMin(chi2WithOneHitForRec);
+ else newTrack->SetFitFMin(newTrack->GetFitFMin() + chi2WithOneHitForRec);
+ // Printout for debuging
+ if ((AliLog::GetDebugLevel("MUON","AliMUONTrackReconstructor") >= 1) || (AliLog::GetGlobalDebugLevel() >= 1)) {
+ cout << "FollowTrackInStation: added one hit in chamber(1..): " << ch1+1
+ << " (Chi2 = " << chi2WithOneHitForRec << ")" << endl;
+ if (AliLog::GetGlobalDebugLevel() >= 3) newTrack->RecursiveDump();
+ }
+ } if (!bestHitForRec2 && chi2WithOneHitForRec < bestChi2WithOneHitForRec) {
+ // keep track of the best single hitForRec except if a couple
+ // of hits has already been found (i.e. bestHitForRec1!=0x0)
+ bestChi2WithOneHitForRec = chi2WithOneHitForRec;
+ bestHitForRec1 = hitForRecCh1;
+ }
+ }
+ }
+ }
//
- // write histos and ntuple to "reconst.root" file
- gAliFileGlobal->Write();
+ // fill out the best track if required else clean up the fRecTracksPtr array
+ if (!fgkTrackAllTracks && bestHitForRec1) {
+ AliMUONTrackParam TrackParam1(extrapTrackParamSave);
+ AliMUONTrackExtrap::ExtrapToZ(&TrackParam1, bestHitForRec1->GetZ());
+ trackCandidate->AddTrackParamAtHit(&TrackParam1,bestHitForRec1);
+ if (bestHitForRec2) {
+ AliMUONTrackParam TrackParam2(extrapTrackParamSave);
+ AliMUONTrackExtrap::ExtrapToZ(&TrackParam2, bestHitForRec2->GetZ());
+ trackCandidate->AddTrackParamAtHit(&TrackParam2,bestHitForRec2);
+ // Sort TrackParamAtHit according to increasing Z
+ trackCandidate->GetTrackParamAtHit()->Sort();
+ // Update the chi2 of the new track
+ if (trackCandidate->GetFitFMin()<0) trackCandidate->SetFitFMin(bestChi2WithTwoHitForRec);
+ else trackCandidate->SetFitFMin(trackCandidate->GetFitFMin() + bestChi2WithTwoHitForRec);
+ // Printout for debuging
+ if ((AliLog::GetDebugLevel("MUON","AliMUONTrackReconstructor") >= 1) || (AliLog::GetGlobalDebugLevel() >= 1)) {
+ cout << "FollowTrackInStation: added the two best hits in station(1..): " << nextStation+1
+ << " (Chi2 = " << bestChi2WithTwoHitForRec << ")" << endl;
+ if (AliLog::GetGlobalDebugLevel() >= 3) newTrack->RecursiveDump();
+ }
+ } else {
+ // Sort TrackParamAtHit according to increasing Z
+ trackCandidate->GetTrackParamAtHit()->Sort();
+ // Update the chi2 of the new track
+ if (trackCandidate->GetFitFMin()<0) trackCandidate->SetFitFMin(bestChi2WithOneHitForRec);
+ else trackCandidate->SetFitFMin(trackCandidate->GetFitFMin() + bestChi2WithOneHitForRec);
+ // Printout for debuging
+ if ((AliLog::GetDebugLevel("MUON","AliMUONTrackReconstructor") >= 1) || (AliLog::GetGlobalDebugLevel() >= 1)) {
+ cout << "FollowTrackInStation: added the best hit in station(1..): " << nextStation+1
+ << " (Chi2 = " << bestChi2WithOneHitForRec << ")" << endl;
+ if (AliLog::GetGlobalDebugLevel() >= 3) newTrack->RecursiveDump();
+ }
+ }
+ } else {
+ fRecTracksPtr->Remove(trackCandidate); // obsolete track
+ fNRecTracks--;
+ }
+
}
-//________________________________________________________________________
-void trackf_read_fit(Int_t &ievr, Int_t &nev, Int_t &nhittot1, Int_t *izch, Double_t *xgeant, Double_t *ygeant)
+ //__________________________________________________________________________
+void AliMUONTrackReconstructor::SetVertexForFit(AliMUONTrack* trackCandidate)
{
-
- // introduce aliroot variables in fortran common
- // tracking study from geant hits
- //
-
- AliMUON *pMUON = (AliMUON*) gAlice->GetModule("MUON");
- TTree *treeH = gAlice->TreeH();
- Int_t ntracks = (Int_t)treeH->GetEntries();
- cout<<"ntrack="<<ntracks<<endl;
-
- nhittot1 = 0;
-
-// Loop over tracks
- for (Int_t track=0; track<ntracks;track++) {
- gAlice->ResetHits();
- treeH->GetEvent(track);
-
- if (pMUON) {
-
-// Loop over hits
- for(AliMUONHit* mHit=(AliMUONHit*)pMUON->FirstHit(-1);
- mHit;
- mHit=(AliMUONHit*)pMUON->NextHit())
- {
- if (mHit->fChamber > 10) continue;
- TClonesArray *fPartArray = gAlice->Particles();
- Int_t ftrack = mHit->Track();
- Int_t id = ((TParticle*) fPartArray->UncheckedAt(ftrack))->GetPdgCode();
-
- if (id==kMuonPlus||id==kMuonMinus) {
- xgeant[nhittot1] = mHit->Y();
- ygeant[nhittot1] = mHit->X();
- izch[nhittot1] = mHit->fChamber;
-// printf("id %d ch %d x %f y %f\n",id,izch[nhittot1],xgeant[nhittot1],ygeant[nhittot1]);
- nhittot1++;
- }
- } // hit loop
- } // if pMUON
- } // track loop
-
- ievr=nev;
- gAliFileGlobal->cd();
+ /// Add the vertex as a measured hit to constrain the fit of the "trackCandidate"
+ /// Compute the vertex resolution from natural vertex dispersion and
+ /// multiple scattering effets according to trackCandidate path in absorber
+ /// It is necessary to account for multiple scattering effects here instead of during the fit of
+ /// the "trackCandidate" to do not influence the result by changing track resolution at vertex
+ AliDebug(1,"Enter SetVertexForFit");
+
+ Double_t nonBendingReso2 = fNonBendingVertexDispersion * fNonBendingVertexDispersion;
+ Double_t bendingReso2 = fBendingVertexDispersion * fBendingVertexDispersion;
+ // add multiple scattering effets
+ AliMUONTrackParam paramAtVertex(*((AliMUONTrackParam*)(trackCandidate->GetTrackParamAtHit()->First())));
+ paramAtVertex.DeleteCovariances(); // to be sure to account only for multiple scattering
+ AliMUONTrackExtrap::ExtrapToVertexUncorrected(¶mAtVertex,0.);
+ TMatrixD* paramCov = paramAtVertex.GetCovariances();
+ nonBendingReso2 += (*paramCov)(0,0);
+ bendingReso2 += (*paramCov)(2,2);
+ // Set the vertex
+ AliMUONHitForRec vertex; // Coordinates set to (0.,0.,0.) by default
+ vertex.SetNonBendingReso2(nonBendingReso2);
+ vertex.SetBendingReso2(bendingReso2);
+ trackCandidate->SetVertex(&vertex);
}
-//______________________________________________________________________________
-void trackf_read_geant(Int_t *itypg, Double_t *xtrg, Double_t *ytrg, Double_t *ptotg, Int_t *idg, Int_t *izch, Double_t *pvert1g, Double_t *pvert2g, Double_t *pvert3g, Double_t *zvertg, Int_t &nhittot1, Double_t *cx, Double_t *cy, Double_t *cz, Int_t &ievr,Int_t &nev,Double_t *xgeant, Double_t *ygeant, Double_t *clsize1, Double_t *clsize2)
+ //__________________________________________________________________________
+void AliMUONTrackReconstructor::Fit(AliMUONTrack *track, Bool_t includeMCS, Bool_t calcCov)
{
- //
- // introduce aliroot variables in fortran common
- // tracking study from geant hits
- //
-
- AliMUON *pMUON = (AliMUON*) gAlice->GetModule("MUON");
+ /// Fit the track "track" with or without multiple Coulomb scattering according to "includeMCS".
- // TTree *treeK = gAlice->TreeK();
- TTree *treeH = gAlice->TreeH();
- Int_t ntracks = (Int_t)treeH->GetEntries();
- cout<<"ntrack="<<ntracks<<endl;
-
- Int_t maxidg = 0;
- Int_t nres=0;
+ Double_t benC, errorParam, invBenP, nonBenC, x, y;
+ AliMUONTrackParam *trackParam;
+ Double_t arg[1], fedm, errdef, fitFMin;
+ Int_t npari, nparx;
+ Int_t status, covStatus;
+
+ // Clear MINUIT parameters
+ gMinuit->mncler();
+ // Give the fitted track to MINUIT
+ gMinuit->SetObjectFit(track);
+ if ((AliLog::GetDebugLevel("MUON","AliMUONTrackReconstructor") >= 2) || (AliLog::GetGlobalDebugLevel() >= 2)) {
+ // Define print level
+ arg[0] = 1;
+ gMinuit->mnexcm("SET PRI", arg, 1, status);
+ // Print covariance matrix
+ gMinuit->mnexcm("SHO COV", arg, 0, status);
+ } else {
+ arg[0] = -1;
+ gMinuit->mnexcm("SET PRI", arg, 1, status);
+ }
+ // No warnings
+ gMinuit->mnexcm("SET NOW", arg, 0, status);
+ // Define strategy
+ //arg[0] = 2;
+ //gMinuit->mnexcm("SET STR", arg, 1, status);
+
+ // Switch between available FCN according to "includeMCS"
+ if (includeMCS) gMinuit->SetFCN(TrackChi2MCS);
+ else gMinuit->SetFCN(TrackChi2);
+
+ // Set fitted parameters (!! The order is very important for the covariance matrix !!)
+ trackParam = (AliMUONTrackParam*) (track->GetTrackParamAtHit()->First());
+ // could be tried with no limits for the search (min=max=0) ????
+ // mandatory limits in non Bending to avoid NaN values of parameters
+ gMinuit->mnparm(0, "X", trackParam->GetNonBendingCoor(), 0.03, -500.0, 500.0, status);
+ gMinuit->mnparm(1, "NonBenS", trackParam->GetNonBendingSlope(), 0.001, -0.5, 0.5, status);
+ // mandatory limits in Bending to avoid NaN values of parameters
+ gMinuit->mnparm(2, "Y", trackParam->GetBendingCoor(), 0.10, -500.0, 500.0, status);
+ gMinuit->mnparm(3, "BenS", trackParam->GetBendingSlope(), 0.001, -0.5, 0.5, status);
+ gMinuit->mnparm(4, "InvBenP", trackParam->GetInverseBendingMomentum(), 0.003, -0.4, 0.4, status);
+
+ // minimization
+ gMinuit->mnexcm("MIGRAD", arg, 0, status);
+
+ // Calculate the covariance matrix more accurately if required
+ if (calcCov) gMinuit->mnexcm("HESSE", arg, 0, status);
+
+ // get results into "invBenP", "benC", "nonBenC" ("x", "y")
+ gMinuit->GetParameter(0, x, errorParam);
+ trackParam->SetNonBendingCoor(x);
+ gMinuit->GetParameter(1, nonBenC, errorParam);
+ trackParam->SetNonBendingSlope(nonBenC);
+ gMinuit->GetParameter(2, y, errorParam);
+ trackParam->SetBendingCoor(y);
+ gMinuit->GetParameter(3, benC, errorParam);
+ trackParam->SetBendingSlope(benC);
+ gMinuit->GetParameter(4, invBenP, errorParam);
+ trackParam->SetInverseBendingMomentum(invBenP);
+
+ // global result of the fit
+ gMinuit->mnstat(fitFMin, fedm, errdef, npari, nparx, covStatus);
+ track->SetFitFMin(fitFMin);
+
+ // Get the covariance matrix if required
+ if (calcCov) {
+ // Covariance matrix according to HESSE status
+ // If problem then keep only the diagonal terms (variances)
+ Double_t matrix[5][5];
+ gMinuit->mnemat(&matrix[0][0],5);
+ if (covStatus == 3) trackParam->SetCovariances(matrix);
+ else trackParam->SetVariances(matrix);
+ } else *(trackParam->GetCovariances()) = 0;
-//
-// Loop over tracks
-//
-
- for (Int_t track=0; track<ntracks;track++) {
- gAlice->ResetHits();
- treeH->GetEvent(track);
-
- if (pMUON) {
-//
-// Loop over hits
-//
- for(AliMUONHit* mHit=(AliMUONHit*)pMUON->FirstHit(-1);
- mHit;
- mHit=(AliMUONHit*)pMUON->NextHit())
- {
- if (maxidg<=20000) {
-
- if (mHit->fChamber > 10) continue;
- TClonesArray *fPartArray = gAlice->Particles();
- TParticle *particle;
- Int_t ftrack = mHit->Track();
- Int_t id = ((TParticle*) fPartArray->UncheckedAt(ftrack))->GetPdgCode();
-
-// if (id==kMuonPlus||id==kMuonMinus) {
-
- // inversion de x et y car le champ est inverse dans le programme tracking
- xtrg[maxidg] = 0;
- ytrg[maxidg] = 0;
- xgeant[maxidg] = mHit->Y(); // x-pos of hit
- ygeant[maxidg] = mHit->X(); // y-pos of hit
- clsize1[maxidg] = 0; // cluster size on 1-st cathode
- clsize2[maxidg] = 0; // cluster size on 2-nd cathode
- cx[maxidg] = mHit->fCyHit; // Px/P of hit
- cy[maxidg] = mHit->fCxHit; // Py/P of hit
- cz[maxidg] = mHit->fCzHit; // Pz/P of hit
- izch[maxidg] = mHit->fChamber;
- /*
- Int_t pdgtype = Int_t(mHit->fParticle); // particle number
- itypg[maxidg] = gMC->IdFromPDG(pdgtype);
-
- */
- itypg[maxidg] = 0;
- if (id==kMuonPlus) itypg[maxidg] = 5;
- if (id==kMuonMinus) itypg[maxidg] = 6;
-
- //printf("ich, itypg[maxidg] %d %d\n",izch[maxidg],itypg[maxidg]);
-
- ptotg[maxidg] = mHit->fPTot; // P of hit
-
- particle = (TParticle*) fPartArray->UncheckedAt(ftrack);
- Float_t thet = particle->Theta();
- thet = thet*180./3.1416;
-
- //cout<<"chambre "<<izch[maxidg]<<" ptot="<<ptotg[maxidg]<<" theta="<<thet<<" phi="<<mHit->fPhi<<" z="<<zz<<endl;
-
- Int_t iparent = particle->GetFirstMother();
- if (iparent >= 0) {
- Int_t ip;
- while(1) {
- ip=((TParticle*) fPartArray->UncheckedAt(iparent))->GetFirstMother();
- if (ip < 0) {
- break;
- } else {
- iparent = ip;
- }
- }
- }
- //printf("iparent - %d\n",iparent);
- Int_t id1 = ftrack; // numero de la particule generee au vertex
- Int_t idum = track+1;
- Int_t id2 = ((TParticle*) fPartArray->UncheckedAt(iparent))->GetPdgCode();
-
- if (id2==443) id2=114;
- else id2=116;
-
- if (id2==116) {
- nres++;
- }
- //printf("id2 %d\n",id2);
- idg[maxidg] = 30000*id1+10000*idum+id2;
-
- pvert1g[maxidg] = particle->Py(); // Px vertex
- pvert2g[maxidg] = particle->Px(); // Py vertex
- pvert3g[maxidg] = particle->Pz(); // Pz vertex
- zvertg[maxidg] = particle->Vz(); // z vertex
-
- // cout<<"x="<<xgeant[maxidg]<<endl;
- //cout<<"y="<<ygeant[maxidg]<<endl;
- //cout<<"typ="<<itypg[maxidg]<<endl;
-
- maxidg ++;
-
- }
- }
- } // hit loop
-// } // if pMUON
- } // track loop first file
-
- if (gAliTrH1 && gAliHits2 ) { // if background file
- ntracks =(Int_t)gAliTrH1->GetEntries();
- printf("Trackf_read - 2-nd file - ntracks %d\n",ntracks);
-
- // Loop over tracks
- for (Int_t track=0; track<ntracks; track++) {
-
- if (gAliHits2) gAliHits2->Clear();
- gAliTrH1->GetEvent(track);
-
- // Loop over hits
- AliMUONHit *mHit;
- for (int i=0;i<gAliHits2->GetEntriesFast();i++)
- {
- mHit=(AliMUONHit*) (*gAliHits2)[i];
- if (mHit->fChamber > 10) continue;
- if (maxidg<=20000) {
-
- // inversion de x et y car le champ est inverse dans le programme tracking !!!!
- xtrg[maxidg] = 0; // only for reconstructed point
- ytrg[maxidg] = 0; // only for reconstructed point
- xgeant[maxidg] = mHit->Y(); // x-pos of hit
- ygeant[maxidg] = mHit->X(); // y-pos of hit
- clsize1[maxidg] = 0; // cluster size on 1-st cathode
- clsize2[maxidg] = 0; // cluster size on 2-nd cathode
- cx[maxidg] = mHit->fCyHit; // Px/P of hit
- cy[maxidg] = mHit->fCxHit; // Py/P of hit
- cz[maxidg] = mHit->fCzHit; // Pz/P of hit
- izch[maxidg] = mHit->fChamber; // chamber number
- ptotg[maxidg] = mHit->fPTot; // P of hit
-
- Int_t ftrack = mHit->Track();
- Int_t id1 = ftrack; // track number
- Int_t idum = track+1;
-
- TClonesArray *fPartArray = gAliParticles2;
-// TParticle *Part=NULL;
- Int_t id = ((TParticle*) fPartArray->UncheckedAt(ftrack))->GetPdgCode();
- if (id==kMuonPlus||id==kMuonMinus) {
- if (id==kMuonPlus) itypg[maxidg] = 5;
- else itypg[maxidg] = 6;
- } else itypg[maxidg]=0;
-
- Int_t id2=0; // set parent to 0 for background !!
- idg[maxidg] = 30000*id1+10000*idum+id2;
-
-
- pvert1g[maxidg] = 0; // Px vertex
- pvert2g[maxidg] = 0; // Py vertex
- pvert3g[maxidg] = 0; // Pz vertex
- zvertg[maxidg] = 0; // z vertex
- maxidg ++;
-
- } // check limits (maxidg)
- } // hit loop
- } // track loop
- } // if gAliTrH1
-
- ievr = nev;
- nhittot1 = maxidg ;
- cout<<"nhittot1="<<nhittot1<<endl;
-
- static Int_t nbres=0;
- if (nres>=19) nbres++;
- printf("nres, nbres %d %d \n",nres,nbres);
-
- gAliFileGlobal->cd();
-
}
-//________________________________________________________________________
-void trackf_read_spoint(Int_t *itypg, Double_t *xtrg, Double_t *ytrg, Double_t *ptotg, Int_t *idg, Int_t *izch, Double_t *pvert1g, Double_t *pvert2g, Double_t *pvert3g, Double_t *zvertg, Int_t &nhittot1, Double_t *cx, Double_t *cy, Double_t *cz, Int_t &ievr,Int_t &nev,Double_t *xgeant, Double_t *ygeant,Double_t *clsize1, Double_t *clsize2)
-
+ //__________________________________________________________________________
+void TrackChi2(Int_t & /*NParam*/, Double_t * /*Gradient*/, Double_t &Chi2, Double_t *Param, Int_t /*Flag*/)
{
- //
- // introduce aliroot variables in fortran common
- // tracking study from reconstructed points
- //
- AliMUON *pMUON = (AliMUON*) gAlice->GetModule("MUON");
-
- cout<<"numero de l'evenement "<<nev<<endl;
-
- TTree *treeR = gAlice->TreeR();
- Int_t nent=(Int_t)treeR->GetEntries();
- if (nev < 10)
- printf("Found %d entries in the tree (must be one per cathode per event! + 1empty)\n",
- nent);
-//
-
- Int_t mult1, mult2;
-
- if (pMUON) {
- Int_t mpoi=0;
- for (Int_t ich=0;ich<10;ich++) {
- printf("chambre %d\n",ich+1);
- TClonesArray *reconstPoints = pMUON->RawClustAddress(ich);
-
- pMUON->ResetRawClusters();
- treeR->GetEvent(nent-1);
- Int_t npoints = (Int_t) reconstPoints->GetEntries();
- if (!npoints) continue;
- printf("\n ch %d npoints = %d\n",ich+1,npoints);
- // Loop over reconstruted points
- for (Int_t ipoi=0; ipoi<npoints; ipoi++) {
- printf(" point %d\n",ipoi);
- AliMUONRawCluster* point =
- (AliMUONRawCluster*) reconstPoints->UncheckedAt(ipoi);
-
- mult1=point->fMultiplicity[0];
- mult2=point->fMultiplicity[1];
- xtrg[mpoi]=(Double_t) point->fY[0];
- ytrg[mpoi]=(Double_t) point->fX[0];
- izch[mpoi]=ich+1;
- Int_t itrack = point->fTracks[1];
- Int_t ihit = point->fTracks[0];
- xgeant[mpoi] = 0;
- ygeant[mpoi] = 0;
- clsize1[mpoi] = mult1;
- clsize2[mpoi] = mult2;
- Int_t id1, id2, idum;
- id1=id2=idum=-1;
- itypg[mpoi]=0;
- ihit = ihit-1;
- if (ihit >=0 && itrack >=0) {
- TClonesArray *fPartArray = gAlice->Particles();
- gAlice->ResetHits();
- gAlice->TreeH()->GetEvent(itrack);
- TClonesArray *pMUONhits = pMUON->Hits();
- AliMUONHit* mHit;
- mHit=(AliMUONHit*) (pMUONhits->UncheckedAt(ihit));
- Int_t id = (Int_t) mHit->fParticle;
- xgeant[mpoi] = mHit->Y();
- ygeant[mpoi] = mHit->X();
- if (id == kMuonPlus) itypg[mpoi]=5;
- if (id == kMuonMinus) itypg[mpoi]=6;
- TParticle *particle;
- particle = (TParticle*)
- (fPartArray->UncheckedAt(mHit->Track()));
- TParticle* particleM=(TParticle*)
- (fPartArray->UncheckedAt(particle->GetFirstMother()));
- Int_t iparent=particleM->GetPdgCode();
- printf("\n Particle Id:%d %d \n", id, iparent);
- if (iparent == 443) id2=114;
- if (iparent == 553) id2=116;
- }
- id1=itrack;
- idum=itrack+1;
- idg[mpoi] = 30000*id1+10000*idum+id2;
- mpoi++;
- } // loop over points
- } // loop over chamber
- ievr = nev;
- cout<<"evenement "<<ievr<<endl;
- nhittot1 = mpoi;
- cout<<"nhittot1="<<nhittot1<<endl;
-
- treeR->Reset();
-
- gAliFileGlobal->cd();
-
- } // if pMUON
+ /// Return the "Chi2" to be minimized with Minuit for track fitting,
+ /// with "NParam" parameters
+ /// and their current values in array pointed to by "Param".
+ /// Assumes that the track hits are sorted according to increasing Z.
+ /// Track parameters at each TrackHit are updated accordingly.
+ /// Multiple Coulomb scattering is not taken into account
+
+ AliMUONTrack *trackBeingFitted = (AliMUONTrack*) gMinuit->GetObjectFit();
+// AliMUONTrack *trackBeingFitted = (AliMUONTrack*) AliMUONTrackReconstructor::Fitter()->GetObjectFit();
+ AliMUONTrackParam param1;
+ AliMUONTrackParam* trackParamAtHit;
+ AliMUONHitForRec* hitForRec;
+ Chi2 = 0.0; // initialize Chi2
+ Double_t dX, dY;
+
+ // copy of track parameters to be fitted
+ param1 = *((AliMUONTrackParam*) (trackBeingFitted->GetTrackParamAtHit()->First()));
+ param1.SetNonBendingCoor(Param[0]);
+ param1.SetNonBendingSlope(Param[1]);
+ param1.SetBendingCoor(Param[2]);
+ param1.SetBendingSlope(Param[3]);
+ param1.SetInverseBendingMomentum(Param[4]);
+
+ // Take the vertex into account in the fit if required
+ if (trackBeingFitted->GetFitWithVertex()) {
+ AliMUONTrackParam paramAtVertex(param1);
+ AliMUONTrackExtrap::ExtrapToZ(¶mAtVertex, 0.);
+ AliMUONHitForRec *vertex = trackBeingFitted->GetVertex();
+ if (!vertex) {
+ cout<<"Error in TrackChi2: Want to use the vertex in tracking but it has not been created!!"<<endl;
+ exit(-1);
+ }
+ dX = vertex->GetNonBendingCoor() - paramAtVertex.GetNonBendingCoor();
+ dY = vertex->GetBendingCoor() - paramAtVertex.GetBendingCoor();
+ Chi2 += dX * dX / vertex->GetNonBendingReso2() + dY * dY / vertex->GetBendingReso2();
+ }
+
+ // Follow track through all planes of track hits
+ trackParamAtHit = (AliMUONTrackParam*) (trackBeingFitted->GetTrackParamAtHit()->First());
+ while (trackParamAtHit) {
+ hitForRec = trackParamAtHit->GetHitForRecPtr();
+ // extrapolation to the plane of the hitForRec attached to the current trackParamAtHit
+ AliMUONTrackExtrap::ExtrapToZ(¶m1, hitForRec->GetZ());
+ // update track parameters of the current hit
+ trackParamAtHit->SetTrackParam(param1);
+ // Increment Chi2
+ // done hit per hit, with hit resolution,
+ // and not with point and angle like in "reco_muon.F" !!!!
+ dX = hitForRec->GetNonBendingCoor() - param1.GetNonBendingCoor();
+ dY = hitForRec->GetBendingCoor() - param1.GetBendingCoor();
+ Chi2 = Chi2 + dX * dX / hitForRec->GetNonBendingReso2() + dY * dY / hitForRec->GetBendingReso2();
+ trackParamAtHit = (AliMUONTrackParam*) (trackBeingFitted->GetTrackParamAtHit()->After(trackParamAtHit));
+ }
}
-//____________________________________________________________________________
-void trackf_fit(Int_t &ivertex, Double_t *pest, Double_t *pstep, Double_t &pxzinv, Double_t &tphi, Double_t &talam, Double_t &xvert, Double_t &yvert)
+ //__________________________________________________________________________
+void TrackChi2MCS(Int_t & /*NParam*/, Double_t * /*Gradient*/, Double_t &Chi2, Double_t *Param, Int_t /*Flag*/)
{
- //
- // Fit a track candidate with the following input parameters:
- // INPUT : IVERTEX : vertex flag, if IVERTEX=1 (XVERT,YVERT) are free paramaters
- // if IVERTEX=1 (XVERT,YVERT)=(0.,0.)
- // PEST(5) : starting value of parameters (minuit)
- // PSTEP(5) : step size for parameters (minuit)
- // OUTPUT : PXZINV,TPHI,TALAM,XVERT,YVERT : fitted value of the parameters
-
- static Double_t arglist[10];
- static Double_t c[5] = {0.4, 0.45, 0.45, 90., 90.};
- static Double_t b1, b2, epxz, efi, exs, exvert, eyvert;
- TString chname;
- Int_t ierflg = 0;
-
- TMinuit *gMinuit = new TMinuit(5);
- gMinuit->mninit(5,10,7);
- gMinuit->SetFCN(fcnf); // constant m.f.
-
- arglist[0] = -1;
+ /// Return the "Chi2" to be minimized with Minuit for track fitting,
+ /// with "NParam" parameters
+ /// and their current values in array pointed to by "Param".
+ /// Assumes that the track hits are sorted according to increasing Z.
+ /// Track parameters at each TrackHit are updated accordingly.
+ /// Multiple Coulomb scattering is taken into account with covariance matrix.
- gMinuit->mnexcm("SET PRINT", arglist, 1, ierflg);
- // gMinuit->mnseti('track fitting');
+ AliMUONTrack *trackBeingFitted = (AliMUONTrack*) gMinuit->GetObjectFit();
+// AliMUONTrack *trackBeingFitted = (AliMUONTrack*) AliMUONTrackReconstructor::Fitter()->GetObjectFit();
+ AliMUONTrackParam param1;
+ AliMUONTrackParam* trackParamAtHit;
+ AliMUONHitForRec* hitForRec;
+ Chi2 = 0.0; // initialize Chi2
+ Int_t chCurrent, chPrev = 0, hitNumber, hitNumber1, hitNumber2, hitNumber3;
+ Double_t z1, z2, z3;
+ AliMUONTrackParam *trackParamAtHit1, *trackParamAtHit2, *trackParamAtHit3;
+ AliMUONHitForRec *hitForRec1, *hitForRec2;
+ Double_t hbc1, hbc2, pbc1, pbc2;
+ Double_t hnbc1, hnbc2, pnbc1, pnbc2;
+ Int_t numberOfHit = trackBeingFitted->GetNTrackHits();
+ TMatrixD *covBending = new TMatrixD(numberOfHit, numberOfHit);
+ TMatrixD *covNonBending = new TMatrixD(numberOfHit, numberOfHit);
+ Double_t *msa2 = new Double_t[numberOfHit];
- gMinuit->mnparm(0, "invmom", pest[0], pstep[0], -c[0], c[0], ierflg);
- gMinuit->mnparm(1, "azimuth", pest[1], pstep[1], -c[1], c[1], ierflg);
- gMinuit->mnparm(2, "deep", pest[2], pstep[2], -c[2], c[2], ierflg);
- if (ivertex==1) {
- gMinuit->mnparm(3, "x ", pest[3], pstep[3], -c[3], c[3], ierflg);
- gMinuit->mnparm(4, "y ", pest[4], pstep[4], -c[4], c[4], ierflg);
- }
+ // copy of track parameters to be fitted
+ param1 = *((AliMUONTrackParam*) (trackBeingFitted->GetTrackParamAtHit()->First()));
+ param1.SetNonBendingCoor(Param[0]);
+ param1.SetNonBendingSlope(Param[1]);
+ param1.SetBendingCoor(Param[2]);
+ param1.SetBendingSlope(Param[3]);
+ param1.SetInverseBendingMomentum(Param[4]);
+
+ // Take the vertex into account in the fit if required
+ if (trackBeingFitted->GetFitWithVertex()) {
+ AliMUONTrackParam paramAtVertex(param1);
+ AliMUONTrackExtrap::ExtrapToZ(¶mAtVertex, 0.);
+ AliMUONHitForRec *vertex = trackBeingFitted->GetVertex();
+ if (!vertex) {
+ cout<<"Error in TrackChi2MCS: Want to use the vertex in tracking but it has not been created!!"<<endl;
+ exit(-1);
+ }
+ Double_t dX = vertex->GetNonBendingCoor() - paramAtVertex.GetNonBendingCoor();
+ Double_t dY = vertex->GetBendingCoor() - paramAtVertex.GetBendingCoor();
+ Chi2 += dX * dX / vertex->GetNonBendingReso2() + dY * dY / vertex->GetBendingReso2();
+ }
- gMinuit->mnexcm("SET NOGR", arglist, 0, ierflg);
- gMinuit->mnexcm("MINIMIZE", arglist, 0, ierflg);
- gMinuit->mnexcm("EXIT" , arglist, 0, ierflg);
+ // Predicted coordinates and multiple scattering angles are first calculated
+ for (hitNumber = 0; hitNumber < numberOfHit; hitNumber++) {
+ trackParamAtHit = (AliMUONTrackParam*) (trackBeingFitted->GetTrackParamAtHit()->UncheckedAt(hitNumber));
+ hitForRec = trackParamAtHit->GetHitForRecPtr();
+ // extrapolation to the plane of the hitForRec attached to the current trackParamAtHit
+ AliMUONTrackExtrap::ExtrapToZ(¶m1, hitForRec->GetZ());
+ // update track parameters of the current hit
+ trackParamAtHit->SetTrackParam(param1);
+ // square of multiple scattering angle at current hit, with one chamber
+ msa2[hitNumber] = MultipleScatteringAngle2(¶m1);
+ // correction for eventual missing hits or multiple hits in a chamber,
+ // according to the number of chambers
+ // between the current hit and the previous one
+ chCurrent = hitForRec->GetChamberNumber();
+ if (hitNumber > 0) msa2[hitNumber] = msa2[hitNumber] * (chCurrent - chPrev);
+ chPrev = chCurrent;
+ }
+
+ // Calculates the covariance matrix
+ for (hitNumber1 = 0; hitNumber1 < numberOfHit; hitNumber1++) {
+ trackParamAtHit1 = (AliMUONTrackParam*) (trackBeingFitted->GetTrackParamAtHit()->UncheckedAt(hitNumber1));
+ hitForRec1 = trackParamAtHit1->GetHitForRecPtr();
+ z1 = hitForRec1->GetZ();
+ for (hitNumber2 = hitNumber1; hitNumber2 < numberOfHit; hitNumber2++) {
+ trackParamAtHit2 = (AliMUONTrackParam*) (trackBeingFitted->GetTrackParamAtHit()->UncheckedAt(hitNumber2));
+ z2 = trackParamAtHit2->GetHitForRecPtr()->GetZ();
+ // initialization to 0 (diagonal plus upper triangular part)
+ (*covBending)(hitNumber2, hitNumber1) = 0.0;
+ // contribution from multiple scattering in bending plane:
+ // loop over upstream hits
+ for (hitNumber3 = 0; hitNumber3 < hitNumber1; hitNumber3++) {
+ trackParamAtHit3 = (AliMUONTrackParam*) (trackBeingFitted->GetTrackParamAtHit()->UncheckedAt(hitNumber3));
+ z3 = trackParamAtHit3->GetHitForRecPtr()->GetZ();
+ (*covBending)(hitNumber2, hitNumber1) = (*covBending)(hitNumber2, hitNumber1) + ((z1 - z3) * (z2 - z3) * msa2[hitNumber3]);
+ }
+ // equal contribution from multiple scattering in non bending plane
+ (*covNonBending)(hitNumber2, hitNumber1) = (*covBending)(hitNumber2, hitNumber1);
+ if (hitNumber1 == hitNumber2) {
+ // Diagonal elements: add contribution from position measurements
+ // in bending plane
+ (*covBending)(hitNumber2, hitNumber1) = (*covBending)(hitNumber2, hitNumber1) + hitForRec1->GetBendingReso2();
+ // and in non bending plane
+ (*covNonBending)(hitNumber2, hitNumber1) = (*covNonBending)(hitNumber2, hitNumber1) + hitForRec1->GetNonBendingReso2();
+ } else {
+ // Non diagonal elements: symmetrization
+ // for bending plane
+ (*covBending)(hitNumber1, hitNumber2) = (*covBending)(hitNumber2, hitNumber1);
+ // and non bending plane
+ (*covNonBending)(hitNumber1, hitNumber2) = (*covNonBending)(hitNumber2, hitNumber1);
+ }
+ } // for (hitNumber2 = hitNumber1;...
+ } // for (hitNumber1 = 0;...
+
+ // Inversion of covariance matrices
+ Int_t ifailBending;
+ gMinuit->mnvert(&((*covBending)(0,0)), numberOfHit, numberOfHit, numberOfHit, ifailBending);
+ Int_t ifailNonBending;
+ gMinuit->mnvert(&((*covNonBending)(0,0)), numberOfHit, numberOfHit, numberOfHit, ifailNonBending);
+
+ // It would be worth trying to calculate the inverse of the covariance matrix
+ // only once per fit, since it cannot change much in principle,
+ // and it would save a lot of computing time !!!!
- gMinuit->mnpout(0, chname, pxzinv, epxz, b1, b2, ierflg);
- gMinuit->mnpout(1, chname, tphi, efi, b1, b2, ierflg);
- gMinuit->mnpout(2, chname, talam, exs, b1, b2, ierflg);
- if (ivertex==1) {
- gMinuit->mnpout(3, chname, xvert, exvert, b1, b2, ierflg);
- gMinuit->mnpout(4, chname, yvert, eyvert, b1, b2, ierflg);
- }
+ // Calculates Chi2
+ if ((ifailBending == 0) && (ifailNonBending == 0)) {
+ // with Multiple Scattering if inversion correct
+ for (hitNumber1 = 0; hitNumber1 < numberOfHit ; hitNumber1++) {
+ trackParamAtHit1 = (AliMUONTrackParam*) (trackBeingFitted->GetTrackParamAtHit()->UncheckedAt(hitNumber1));
+ hitForRec1 = trackParamAtHit1->GetHitForRecPtr();
+ hbc1 = hitForRec1->GetBendingCoor();
+ pbc1 = trackParamAtHit1->GetBendingCoor();
+ hnbc1 = hitForRec1->GetNonBendingCoor();
+ pnbc1 = trackParamAtHit1->GetNonBendingCoor();
+ for (hitNumber2 = 0; hitNumber2 < numberOfHit; hitNumber2++) {
+ trackParamAtHit2 = (AliMUONTrackParam*) (trackBeingFitted->GetTrackParamAtHit()->UncheckedAt(hitNumber2));
+ hitForRec2 = trackParamAtHit2->GetHitForRecPtr();
+ hbc2 = hitForRec2->GetBendingCoor();
+ pbc2 = trackParamAtHit2->GetBendingCoor();
+ hnbc2 = hitForRec2->GetNonBendingCoor();
+ pnbc2 = trackParamAtHit2->GetNonBendingCoor();
+ Chi2 += ((*covBending)(hitNumber2, hitNumber1) * (hbc1 - pbc1) * (hbc2 - pbc2)) +
+ ((*covNonBending)(hitNumber2, hitNumber1) * (hnbc1 - pnbc1) * (hnbc2 - pnbc2));
+ }
+ }
+ } else {
+ // without Multiple Scattering if inversion impossible
+ for (hitNumber1 = 0; hitNumber1 < numberOfHit ; hitNumber1++) {
+ trackParamAtHit1 = (AliMUONTrackParam*) (trackBeingFitted->GetTrackParamAtHit()->UncheckedAt(hitNumber1));
+ hitForRec1 = trackParamAtHit1->GetHitForRecPtr();
+ hbc1 = hitForRec1->GetBendingCoor();
+ pbc1 = trackParamAtHit1->GetBendingCoor();
+ hnbc1 = hitForRec1->GetNonBendingCoor();
+ pnbc1 = trackParamAtHit1->GetNonBendingCoor();
+ Chi2 += ((hbc1 - pbc1) * (hbc1 - pbc1) / hitForRec1->GetBendingReso2()) +
+ ((hnbc1 - pnbc1) * (hnbc1 - pnbc1) / hitForRec1->GetNonBendingReso2());
+ }
+ }
- delete gMinuit;
+ delete covBending;
+ delete covNonBending;
+ delete [] msa2;
}
-
-//________________________________________________________________________________
-void fcnf(Int_t &npar, Double_t *grad, Double_t &fval, Double_t *pest, Int_t iflag)
+
+ //__________________________________________________________________________
+Double_t MultipleScatteringAngle2(AliMUONTrackParam *param)
{
- //
- // function called by trackf_fit
- Int_t futil = 0;
- fcn(npar,grad,fval,pest,iflag,futil);
+ /// Returns square of multiple Coulomb scattering angle
+ /// from TrackParamAtHit pointed to by "param"
+ Double_t slopeBending, slopeNonBending, radiationLength, inverseBendingMomentum2, inverseTotalMomentum2;
+ Double_t varMultipleScatteringAngle;
+ slopeBending = param->GetBendingSlope();
+ slopeNonBending = param->GetNonBendingSlope();
+ // thickness in radiation length for the current track,
+ // taking local angle into account
+ radiationLength = AliMUONConstants::ChamberThicknessInX0() *
+ TMath::Sqrt(1.0 + slopeBending*slopeBending + slopeNonBending*slopeNonBending);
+ inverseBendingMomentum2 = param->GetInverseBendingMomentum() * param->GetInverseBendingMomentum();
+ inverseTotalMomentum2 = inverseBendingMomentum2 * (1.0 + slopeBending * slopeBending) /
+ (1.0 + slopeBending *slopeBending + slopeNonBending * slopeNonBending);
+ varMultipleScatteringAngle = 0.0136 * (1.0 + 0.038 * TMath::Log(radiationLength));
+ // The velocity is assumed to be 1 !!!!
+ varMultipleScatteringAngle = inverseTotalMomentum2 * radiationLength * varMultipleScatteringAngle * varMultipleScatteringAngle;
+ return varMultipleScatteringAngle;
}
-//____________________________________________________________________________
-void prec_fit(Double_t &pxzinv, Double_t &fis, Double_t &alams, Double_t &xvert, Double_t &yvert, Double_t &pxzinvf, Double_t &fif, Double_t &alf, Double_t &xvertf, Double_t &yvertf, Double_t &epxzinv, Double_t &efi, Double_t &exs, Double_t &exvert, Double_t &eyvert)
+ //__________________________________________________________________________
+void AliMUONTrackReconstructor::ExtrapTracksToVertex()
{
- //
- // minuit fits for tracking finding
-
- static Double_t arglist[10];
- static Double_t c1[5] = {0.001, 0.001, 0.001, 1., 1.};
- static Double_t c2[5] = {0.5, 0.5, 0.5, 120., 120.};
- static Double_t emat[9];
- static Double_t b1, b2;
- Double_t fmin, fedm, errdef;
- Int_t npari, nparx, istat;
-
- TString chname;
- Int_t ierflg = 0;
-
- TMinuit *gMinuit = new TMinuit(5);
- gMinuit->mninit(5,10,7);
- gMinuit->SetFCN(fcnfitf);
-
- arglist[0] = -1.;
- gMinuit->mnexcm("SET PRINT", arglist, 1, ierflg);
-
- // gMinuit->mnseti('track fitting');
-
- gMinuit->mnparm(0,"invmom", pxzinv, c1[0], -c2[0], c2[0], ierflg); // 0.003, 0.5
- gMinuit->mnparm(1,"azimuth ", fis, c1[1], -c2[1], c2[1], ierflg);
- gMinuit->mnparm(2,"deep ", alams, c1[2], -c2[2], c2[2], ierflg);
- gMinuit->mnparm(3,"xvert", xvert, c1[3], -c2[3], c2[3], ierflg);
- gMinuit->mnparm(4,"yvert", yvert, c1[4], -c2[4], c2[4], ierflg);
-
- gMinuit->mnexcm("SET NOGR", arglist, 0, ierflg);
- arglist[0] = 2.;
- gMinuit->mnexcm("MINIMIZE", arglist, 0, ierflg);
- gMinuit->mnexcm("EXIT", arglist, 0, ierflg);
-
- gMinuit->mnpout(0, chname, pxzinvf, epxzinv, b1, b2, ierflg);
- gMinuit->mnpout(1, chname, fif, efi, b1, b2, ierflg);
- gMinuit->mnpout(2, chname, alf, exs, b1, b2, ierflg);
- gMinuit->mnpout(3, chname, xvertf, exvert, b1, b2, ierflg);
- gMinuit->mnpout(4, chname, yvertf, eyvert, b1, b2, ierflg);
+ /// propagate tracks to the vertex through the absorber (Branson)
+ AliMUONTrack *track;
+ AliMUONTrackParam trackParamVertex;
+ AliMUONHitForRec *vertex;
+ Double_t vX, vY, vZ;
+
+ for (Int_t iRecTrack = 0; iRecTrack <fNRecTracks; iRecTrack++) {
+ track = (AliMUONTrack*) fRecTracksPtr->UncheckedAt(iRecTrack);
+ trackParamVertex = *((AliMUONTrackParam*)(track->GetTrackParamAtHit()->First()));
+ vertex = track->GetVertex();
+ if (vertex) { // if track as a vertex defined, use it
+ vX = vertex->GetNonBendingCoor();
+ vY = vertex->GetBendingCoor();
+ vZ = vertex->GetZ();
+ } else { // else vertex = (0.,0.,0.)
+ vX = 0.;
+ vY = 0.;
+ vZ = 0.;
+ }
+ AliMUONTrackExtrap::ExtrapToVertex(&trackParamVertex, vX, vY, vZ);
+ track->SetTrackParamAtVertex(&trackParamVertex);
+
+ if (AliLog::GetGlobalDebugLevel() > 0) {
+ cout << "FollowTracks: track candidate(0..): " << iRecTrack
+ << " after extrapolation to vertex" << endl;
+ track->RecursiveDump();
+ }
+ }
- gMinuit->mnemat(emat, 3);
- gMinuit->mnstat(fmin, fedm, errdef, npari, nparx, istat);
+}
- delete gMinuit;
+ //__________________________________________________________________________
+void AliMUONTrackReconstructor::FillMUONTrack()
+{
+ /// Fill fHitForRecAtHit of AliMUONTrack's
+ AliMUONTrack *track;
+ AliMUONTrackParam *trackParamAtHit;
+ track = (AliMUONTrack*) fRecTracksPtr->First();
+ while (track) {
+ trackParamAtHit = (AliMUONTrackParam*) (track->GetTrackParamAtHit()->First());
+ while (trackParamAtHit) {
+ track->AddHitForRecAtHit(trackParamAtHit->GetHitForRecPtr());
+ trackParamAtHit = (AliMUONTrackParam*) (track->GetTrackParamAtHit()->After(trackParamAtHit));
+ }
+ track = (AliMUONTrack*) fRecTracksPtr->After(track);
+ }
+ return;
}
-//____________________________________________________________________
-void fcnfitf(Int_t &npar, Double_t *grad, Double_t &fval, Double_t *xval, Int_t iflag)
+ //__________________________________________________________________________
+void AliMUONTrackReconstructor::EventDump(void)
{
- //
- // function called by prec_fit
- Int_t futil = 0;
- fcnfit(npar,grad,fval,xval,iflag,futil);
+ /// Dump reconstructed event (track parameters at vertex and at first hit),
+ /// and the particle parameters
+ AliMUONTrack *track;
+ AliMUONTrackParam *trackParam, *trackParam1;
+ Double_t bendingSlope, nonBendingSlope, pYZ;
+ Double_t pX, pY, pZ, x, y, z, c;
+ Int_t trackIndex, nTrackHits;
+
+ AliDebug(1,"****** enter EventDump ******");
+ AliDebug(1, Form("Number of Reconstructed tracks : %d", fNRecTracks));
+
+ fRecTracksPtr->Compress(); // for simple loop without "Next" since no hole
+ // Loop over reconstructed tracks
+ for (trackIndex = 0; trackIndex < fNRecTracks; trackIndex++) {
+ AliDebug(1, Form("track number: %d", trackIndex));
+ // function for each track for modularity ????
+ track = (AliMUONTrack*) ((*fRecTracksPtr)[trackIndex]);
+ nTrackHits = track->GetNTrackHits();
+ AliDebug(1, Form("Number of track hits: %d ", nTrackHits));
+ // track parameters at Vertex
+ trackParam = track->GetTrackParamAtVertex();
+ x = trackParam->GetNonBendingCoor();
+ y = trackParam->GetBendingCoor();
+ z = trackParam->GetZ();
+ bendingSlope = trackParam->GetBendingSlope();
+ nonBendingSlope = trackParam->GetNonBendingSlope();
+ pYZ = 1/TMath::Abs(trackParam->GetInverseBendingMomentum());
+ pZ = pYZ/TMath::Sqrt(1+bendingSlope*bendingSlope);
+ pX = pZ * nonBendingSlope;
+ pY = pZ * bendingSlope;
+ c = TMath::Sign(1.0, trackParam->GetInverseBendingMomentum());
+ AliDebug(1, Form("Track parameters at Vertex z= %f: X= %f Y= %f pX= %f pY= %f pZ= %f c= %f\n",
+ z, x, y, pX, pY, pZ, c));
+
+ // track parameters at first hit
+ trackParam1 = (AliMUONTrackParam*) track->GetTrackParamAtHit()->First();
+ x = trackParam1->GetNonBendingCoor();
+ y = trackParam1->GetBendingCoor();
+ z = trackParam1->GetZ();
+ bendingSlope = trackParam1->GetBendingSlope();
+ nonBendingSlope = trackParam1->GetNonBendingSlope();
+ pYZ = 1/TMath::Abs(trackParam1->GetInverseBendingMomentum());
+ pZ = pYZ/TMath::Sqrt(1.0 + bendingSlope * bendingSlope);
+ pX = pZ * nonBendingSlope;
+ pY = pZ * bendingSlope;
+ c = TMath::Sign(1.0, trackParam1->GetInverseBendingMomentum());
+ AliDebug(1, Form("track parameters at z= %f: X= %f Y= %f pX= %f pY= %f pZ= %f c= %f\n",
+ z, x, y, pX, pY, pZ, c));
+ }
+ // informations about generated particles NO !!!!!!!!
+
+// 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;
}
+
+
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff