X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=MUON%2FAliMUONTrackReconstructor.cxx;h=4d1ae1125fd6ab0172dac2dcd6661bc66be47e7e;hb=d5c9a22c384e2ab9fb589650a71cea4ffa9bb8ac;hp=adfe7c36e7158f66646b2ef1a40b7fdad495294d;hpb=6a9bc54137706d73455d459d5c6f96d7e516e478;p=u%2Fmrichter%2FAliRoot.git diff --git a/MUON/AliMUONTrackReconstructor.cxx b/MUON/AliMUONTrackReconstructor.cxx index adfe7c36e71..4d1ae1125fd 100644 --- a/MUON/AliMUONTrackReconstructor.cxx +++ b/MUON/AliMUONTrackReconstructor.cxx @@ -12,1150 +12,1042 @@ * 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 +#include +#include + +#include "AliMUONVTrackReconstructor.h" +#include "AliMUONTrackReconstructor.h" +#include "AliMUONData.h" +#include "AliMUONConstants.h" #include "AliMUONRawCluster.h" -#include "AliMUONReconstHit.h" - -#include "AliPDG.h" - -#include -#include -#include -#include -#include -#include +#include "AliMUONHitForRec.h" +#include "AliMUONObjectPair.h" +#include "AliMUONTrack.h" +#include "AliMUONTrackParam.h" +#include "AliMUONTrackExtrap.h" +#include "AliLog.h" #include -#include -#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 "<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; isegGetEntriesFast(); 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<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 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<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<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="<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="<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; trackResetHits(); - 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 "<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="<Clear(); - gAliTrH1->GetEvent(track); - - // Loop over hits - AliMUONHit *mHit; - for (int i=0;iGetEntriesFast();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="<=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 "<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; ipoiUncheckedAt(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 "<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!!"<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!!"<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 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; } + +