[u/mrichter/AliRoot.git] / MUON / AliMUONVTrackReconstructor.cxx

/**************************************************************************
 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 *                                                                        *
 * Author: The ALICE Off-line Project.                                    *
 * Contributors are mentioned in the code where appropriate.              *
 *                                                                        *
 * Permission to use, copy, modify and distribute this software and its   *
 * documentation strictly for non-commercial purposes is hereby granted   *
 * without fee, provided that the above copyright notice appears in all   *
 * copies and that both the copyright notice and this permission notice   *
 * appear in the supporting documentation. The authors make no claims     *
 * about the suitability of this software for any purpose. It is          *
 * provided "as is" without express or implied warranty.                  *
 **************************************************************************/

/* $Id$ */

////////////////////////////////////
///
/// \class AliMUONVTrackReconstructor
/// Virtual MUON track reconstructor in ALICE (class renamed from AliMUONEventReconstructor)
///
/// This class contains as data:
/// * a pointer to the array of hits to be reconstructed (the event)
/// * a pointer to the array of segments made with these hits inside each station
/// * a pointer to the array of reconstructed tracks
///
/// It contains as methods, among others:
/// * EventReconstruct to build the muon tracks
/// * EventReconstructTrigger to build the trigger tracks
///
///  \author Philippe Pillot
///
////////////////////////////////////

#include "AliMUONVTrackReconstructor.h"
#include "AliMUONRecData.h"
#include "AliMUONConstants.h"
#include "AliMUONHitForRec.h"
#include "AliMUONObjectPair.h"
#include "AliMUONTriggerTrack.h"
#include "AliMUONTriggerCircuit.h"
#include "AliMUONLocalTrigger.h"
#include "AliMUONGlobalTrigger.h"
#include "AliMUONTrack.h"
#include "AliMUONTrackParam.h"
#include "AliMUONTrackExtrap.h"
#include "AliMUONTrackHitPattern.h"

#include "AliLog.h"
#include "AliTracker.h"

#include <Riostream.h>
#include <TClonesArray.h>
#include <TMath.h>

/// \cond CLASSIMP
ClassImp(AliMUONVTrackReconstructor) // Class implementation in ROOT context
/// \endcond

//************* Defaults parameters for reconstruction
const Double_t AliMUONVTrackReconstructor::fgkDefaultMinBendingMomentum = 3.0;
const Double_t AliMUONVTrackReconstructor::fgkDefaultMaxBendingMomentum = 3000.0;
const Double_t AliMUONVTrackReconstructor::fgkDefaultBendingResolution = 0.01;
const Double_t AliMUONVTrackReconstructor::fgkDefaultNonBendingResolution = 0.144;
const Double_t AliMUONVTrackReconstructor::fgkDefaultBendingVertexDispersion = 10.;
const Double_t AliMUONVTrackReconstructor::fgkDefaultNonBendingVertexDispersion = 10.;
const Double_t AliMUONVTrackReconstructor::fgkDefaultMaxNormChi2MatchTrigger = 16.0;

//__________________________________________________________________________
AliMUONVTrackReconstructor::AliMUONVTrackReconstructor(AliMUONRecData* data)
  : TObject(),
    fMinBendingMomentum(fgkDefaultMinBendingMomentum),
    fMaxBendingMomentum(fgkDefaultMaxBendingMomentum),
    fBendingResolution(fgkDefaultBendingResolution),
    fNonBendingResolution(fgkDefaultNonBendingResolution),
    fBendingVertexDispersion(fgkDefaultBendingVertexDispersion),
    fNonBendingVertexDispersion(fgkDefaultNonBendingVertexDispersion),
    fMaxNormChi2MatchTrigger(fgkDefaultMaxNormChi2MatchTrigger),
    fHitsForRecPtr(0x0),
    fNHitsForRec(0),
    fNHitsForRecPerChamber(0x0),
    fIndexOfFirstHitForRecPerChamber(0x0),
    fRecTracksPtr(0x0),
    fNRecTracks(0),
    fMUONData(data),
    fTriggerTrack(new AliMUONTriggerTrack()),
    fTriggerCircuit(0x0)
{
  /// Constructor for class AliMUONVTrackReconstructor
  fNHitsForRecPerChamber = new Int_t[AliMUONConstants::NTrackingCh()];
  fIndexOfFirstHitForRecPerChamber = new Int_t[AliMUONConstants::NTrackingCh()];

  // Memory allocation for the TClonesArray of hits for reconstruction
  // Is 10000 the right size ????
  fHitsForRecPtr = new TClonesArray("AliMUONHitForRec", 10000);

  // set the magnetic field for track extrapolations
  const AliMagF* kField = AliTracker::GetFieldMap();
  if (!kField) AliFatal("No field available");
  AliMUONTrackExtrap::SetField(kField);

  fTrackHitPattern = new AliMUONTrackHitPattern(fMUONData);
}

  //__________________________________________________________________________
AliMUONVTrackReconstructor::~AliMUONVTrackReconstructor(void)
{
  /// Destructor for class AliMUONVTrackReconstructor
  delete [] fNHitsForRecPerChamber;
  delete [] fIndexOfFirstHitForRecPerChamber;
  delete fTriggerTrack;
  delete fHitsForRecPtr;
  delete fTrackHitPattern;
}

  //__________________________________________________________________________
void AliMUONVTrackReconstructor::EventReconstruct(void)
{
  /// To reconstruct one event
  AliDebug(1,"Enter EventReconstruct");
  
  ResetTracks(); //AZ
  ResetHitsForRec(); //AZ
  AddHitsForRecFromRawClusters();
  MakeTracks();
  if (fMUONData->IsTriggerTrackBranchesInTree()) 
    ValidateTracksWithTrigger(); 

  fTrackHitPattern->GetHitPattern(fRecTracksPtr);

  // Add tracks to MUON data container 
  for(Int_t i=0; i<fNRecTracks; i++) {
    AliMUONTrack * track = (AliMUONTrack*) fRecTracksPtr->At(i);
    fMUONData->AddRecTrack(*track);
  }
}

  //__________________________________________________________________________
void AliMUONVTrackReconstructor::ResetTracks(void)
{
  /// To reset the TClonesArray of reconstructed tracks
  if (fRecTracksPtr) fRecTracksPtr->Delete();
  // Delete in order that the Track destructors are called,
  // hence the space for the TClonesArray of pointers to TrackHit's is freed
  fNRecTracks = 0;
  return;
}

  //__________________________________________________________________________
void AliMUONVTrackReconstructor::ResetHitsForRec(void)
{
  /// To reset the array and the number of HitsForRec,
  /// and also the number of HitsForRec
  /// and the index of the first HitForRec per chamber
  if (fHitsForRecPtr) fHitsForRecPtr->Delete();
  fNHitsForRec = 0;
  for (Int_t ch = 0; ch < AliMUONConstants::NTrackingCh(); ch++)
    fNHitsForRecPerChamber[ch] = fIndexOfFirstHitForRecPerChamber[ch] = 0;
  return;
}

  //__________________________________________________________________________
void AliMUONVTrackReconstructor::SortHitsForRecWithIncreasingChamber()
{
  /// Sort HitsForRec's in increasing order with respect to chamber number.
  /// Uses the function "Compare".
  /// Update the information for HitsForRec per chamber too.
  Int_t ch, nhits, prevch;
  fHitsForRecPtr->Sort();
  for (ch = 0; ch < AliMUONConstants::NTrackingCh(); ch++) {
    fNHitsForRecPerChamber[ch] = 0;
    fIndexOfFirstHitForRecPerChamber[ch] = 0;
  }
  prevch = 0; // previous chamber
  nhits = 0; // number of hits in current chamber
  // Loop over HitsForRec
  for (Int_t hit = 0; hit < fNHitsForRec; hit++) {
    // chamber number (0...)
    ch = ((AliMUONHitForRec*)  ((*fHitsForRecPtr)[hit]))->GetChamberNumber();
    // increment number of hits in current chamber
    (fNHitsForRecPerChamber[ch])++;
    // update index of first HitForRec in current chamber
    // if chamber number different from previous one
    if (ch != prevch) {
      fIndexOfFirstHitForRecPerChamber[ch] = hit;
      prevch = ch;
    }
  }
  return;
}

  //__________________________________________________________________________
TClonesArray* AliMUONVTrackReconstructor::MakeSegmentsInStation(Int_t station)
{
  /// To make the list of segments in station(0..) "Station" from the list of hits to be reconstructed.
  /// Return a new TClonesArray of segments.
  /// It is the responsibility of the user to delete it afterward.
  AliDebug(1,Form("Enter MakeSegmentsPerStation (0...) %d",station));
  
  AliMUONHitForRec *hit1Ptr, *hit2Ptr;
  AliMUONObjectPair *segment;
  Double_t bendingSlope = 0, impactParam = 0., bendingMomentum = 0.; // to avoid compilation warning
  // first and second chambers (0...) in the station
  Int_t ch1 = 2 * station;
  Int_t ch2 = ch1 + 1;
  // list of segments
  TClonesArray *segments = new TClonesArray("AliMUONObjectPair", fNHitsForRecPerChamber[ch2]);
  // Loop over HitForRec's in the first chamber of the station
  for (Int_t hit1 = fIndexOfFirstHitForRecPerChamber[ch1];
       hit1 < fIndexOfFirstHitForRecPerChamber[ch1] + fNHitsForRecPerChamber[ch1];
       hit1++) {
    // pointer to the HitForRec
    hit1Ptr = (AliMUONHitForRec*) ((*fHitsForRecPtr)[hit1]);
    // Loop over HitsForRec's in the second chamber of the station
    for (Int_t hit2 = fIndexOfFirstHitForRecPerChamber[ch2];
	 hit2 < fIndexOfFirstHitForRecPerChamber[ch2] + fNHitsForRecPerChamber[ch2];
	 hit2++) {
      // pointer to the HitForRec
      hit2Ptr = (AliMUONHitForRec*) ((*fHitsForRecPtr)[hit2]);
      if ( hit1Ptr->GetZ() - hit2Ptr->GetZ() != 0. ) {
        // bending slope
        bendingSlope = (hit1Ptr->GetBendingCoor() - hit2Ptr->GetBendingCoor()) / (hit1Ptr->GetZ() - hit2Ptr->GetZ());
        // impact parameter
        impactParam = hit1Ptr->GetBendingCoor() - hit1Ptr->GetZ() * bendingSlope;
        // absolute value of bending momentum
	bendingMomentum = TMath::Abs(AliMUONTrackExtrap::GetBendingMomentumFromImpactParam(impactParam));
      } else {
        AliWarning("hit1Ptr->GetZ() = hit2Ptr->GetZ(): no segment created");
        continue;
      }   
      // check for bending momentum within tolerances
      if ((bendingMomentum < fMaxBendingMomentum) && (bendingMomentum > fMinBendingMomentum)) {
	// make new segment
	segment = new ((*segments)[segments->GetLast()+1]) AliMUONObjectPair(hit1Ptr, hit2Ptr, kFALSE, kFALSE);
	if (AliLog::GetGlobalDebugLevel() > 1) {
	  cout << "segmentIndex(0...): " << segments->GetLast() << endl;
	  segment->Dump();
	  cout << "HitForRec in first chamber" << endl;
	  hit1Ptr->Dump();
	  cout << "HitForRec in second chamber" << endl;
	  hit2Ptr->Dump();
	}
      }
    } //for (Int_t hit2
  } // for (Int_t hit1...
  AliDebug(1,Form("Station: %d  NSegments:  %d ", station, segments->GetEntriesFast()));
  return segments;
}

  //__________________________________________________________________________
void AliMUONVTrackReconstructor::ValidateTracksWithTrigger(void)
{
  /// Try to match track from tracking system with trigger track
  static const Double_t kDistSigma[3]={1,1,0.02}; // sigma of distributions (trigger-track) X,Y,slopeY
  
  Int_t loTrigger =  0;
  Int_t loCirc    = -1;
  Int_t loStripX  = -1;
  Int_t loStripY  = -1;
  Int_t loDev     = -1;
  Int_t loLpt     = -1;
  Int_t loHpt     = -1;

  AliMUONTrack *track;
  AliMUONTrackParam trackParam; 
  AliMUONTriggerTrack *triggerTrack;
  AliMUONLocalTrigger *locTrg;
  
  fMUONData->SetTreeAddress("RL");
  fMUONData->GetRecTriggerTracks();
  TClonesArray *recTriggerTracks = fMUONData->RecTriggerTracks();

  fMUONData->SetTreeAddress("TC");
  fMUONData->GetTrigger();
  TClonesArray *localTrigger = fMUONData->LocalTrigger();

  Int_t matchTrigger;
  Int_t loTrgNum;
  Double_t distTriggerTrack[3];
  Double_t xTrack, yTrack, ySlopeTrack, chi2MatchTrigger, minChi2MatchTrigger, chi2;
  
  track = (AliMUONTrack*) fRecTracksPtr->First();
  while (track) {
    matchTrigger = 0;
    chi2MatchTrigger = 0.;
    loTrgNum = -1;
    Int_t doubleMatch=-1; // Check if track matches 2 trigger tracks
    Double_t doubleChi2 = -1.;

    trackParam = *((AliMUONTrackParam*) (track->GetTrackParamAtHit()->Last()));
    AliMUONTrackExtrap::ExtrapToZ(&trackParam, AliMUONConstants::DefaultChamberZ(10)); // extrap to 1st trigger chamber
    
    xTrack = trackParam.GetNonBendingCoor();
    yTrack = trackParam.GetBendingCoor();
    ySlopeTrack = trackParam.GetBendingSlope();
    minChi2MatchTrigger = 999.;
  
    triggerTrack = (AliMUONTriggerTrack*) recTriggerTracks->First();
    while(triggerTrack){
      distTriggerTrack[0] = (triggerTrack->GetX11()-xTrack)/kDistSigma[0];
      distTriggerTrack[1] = (triggerTrack->GetY11()-yTrack)/kDistSigma[1];
      distTriggerTrack[2] = (TMath::Tan(triggerTrack->GetThetay())-ySlopeTrack)/kDistSigma[2];
      chi2 = 0.;
      for (Int_t iVar = 0; iVar < 3; iVar++) chi2 += distTriggerTrack[iVar]*distTriggerTrack[iVar];
      chi2 /= 3.; // Normalized Chi2: 3 degrees of freedom (X,Y,slopeY)
      if (chi2 < fMaxNormChi2MatchTrigger) {
	  Bool_t isDoubleTrack = (TMath::Abs(chi2 - minChi2MatchTrigger)<1.);
	  if (chi2 < minChi2MatchTrigger && chi2 < fMaxNormChi2MatchTrigger) {
	      if(isDoubleTrack){
		  doubleMatch = loTrgNum;
		  doubleChi2 = chi2MatchTrigger;
	      }
	      minChi2MatchTrigger = chi2;
	      chi2MatchTrigger = chi2;
	      loTrgNum=triggerTrack->GetLoTrgNum();
	      locTrg = (AliMUONLocalTrigger*)localTrigger->UncheckedAt(loTrgNum);
	      matchTrigger=1;
	      if(locTrg->LoLpt()>0)matchTrigger=2;
	      if(locTrg->LoHpt()>0)matchTrigger=3;
	  }
	  else if(isDoubleTrack) {
	      doubleMatch = triggerTrack->GetLoTrgNum();
	      doubleChi2 = chi2;
	  }
      }
      triggerTrack = (AliMUONTriggerTrack*) recTriggerTracks->After(triggerTrack);
    }
    if(doubleMatch>=0){ // If two trigger tracks match, select the one passing more trigger cuts
	AliDebug(1, Form("Two candidates found: %i and %i",loTrgNum,doubleMatch));
	AliMUONLocalTrigger *locTrg1 = (AliMUONLocalTrigger*)localTrigger->UncheckedAt(doubleMatch);
	if((locTrg1->LoLpt()>0 && matchTrigger<2) || (locTrg1->LoHpt() && matchTrigger<3)){
	    if(locTrg1->LoHpt()>0)matchTrigger=3;
	    else matchTrigger=2;
	    loTrgNum = doubleMatch;
	    chi2MatchTrigger=doubleChi2;
	}
    }
    
    track->SetMatchTrigger(matchTrigger);
    track->SetLoTrgNum(loTrgNum);
    track->SetChi2MatchTrigger(chi2MatchTrigger);

    if (loTrgNum >= 0 && loTrgNum < 234) {
      locTrg = (AliMUONLocalTrigger*)localTrigger->UncheckedAt(loTrgNum);
      
      loCirc   = locTrg->LoCircuit();
      loStripX = locTrg->LoStripX();
      loStripY = locTrg->LoStripY();
      loDev    = locTrg->LoDev();
      loLpt    = locTrg->LoLpt();
      loHpt    = locTrg->LoHpt();
      
      track->SetLocalTrigger(loCirc,loStripX,loStripY,loDev,loLpt,loHpt);
    }

    track = (AliMUONTrack*) fRecTracksPtr->After(track);
  }

  return;
}

//__________________________________________________________________________
void AliMUONVTrackReconstructor::EventReconstructTrigger(void)
{
  /// To reconstruct trigger for one event
  AliDebug(1,"Enter EventReconstructTrigger");
  MakeTriggerTracks();  
  return;
}

  //__________________________________________________________________________
Bool_t AliMUONVTrackReconstructor::MakeTriggerTracks(void)
{
  /// To make the trigger tracks from Local Trigger
  AliDebug(1, "Enter MakeTriggerTracks");
  
  TTree* treeR;
  UChar_t gloTrigPat;
  TClonesArray *localTrigger;
  TClonesArray *globalTrigger;
  AliMUONLocalTrigger *locTrg;
  AliMUONGlobalTrigger *gloTrg;

  treeR = fMUONData->TreeR();
  if (!treeR) {
    AliWarning("TreeR is not loaded");
    return kFALSE;
  }
  
  fMUONData->SetTreeAddress("TC");
  fMUONData->GetTrigger();

  // global trigger for trigger pattern
  gloTrigPat = 0;
  globalTrigger = fMUONData->GlobalTrigger(); 
  gloTrg = (AliMUONGlobalTrigger*)globalTrigger->UncheckedAt(0);
 
  if (gloTrg)
    gloTrigPat = gloTrg->GetGlobalResponse();
 

  // local trigger for tracking 
  localTrigger = fMUONData->LocalTrigger();    
  Int_t nlocals = (Int_t) (localTrigger->GetEntries());

  Float_t z11 = AliMUONConstants::DefaultChamberZ(10);
  Float_t z21 = AliMUONConstants::DefaultChamberZ(12);

  Float_t y11 = 0.;
  Int_t stripX21 = 0;
  Float_t y21 = 0.;
  Float_t x11 = 0.;

  for (Int_t i=0; i<nlocals; i++) { // loop on Local Trigger
      locTrg = (AliMUONLocalTrigger*)localTrigger->UncheckedAt(i);      
      
      Bool_t xTrig=kFALSE;
      Bool_t yTrig=kFALSE;
      
      if ( locTrg->LoSdev()==1 && locTrg->LoDev()==0 && 
	   locTrg->LoStripX()==0) xTrig=kFALSE; // no trigger in X
      else xTrig=kTRUE;                         // trigger in X
      if (locTrg->LoTrigY()==1 && 
	  locTrg->LoStripY()==15 ) yTrig = kFALSE; // no trigger in Y
      else yTrig = kTRUE;                          // trigger in Y

      if (xTrig && yTrig) { // make Trigger Track if trigger in X and Y
	  
	  AliDebug(1, "AliMUONTrackReconstructor::MakeTriggerTrack using NEW trigger \n");
	  AliMUONTriggerCircuit* circuit = 
	      (AliMUONTriggerCircuit*)fTriggerCircuit->At(locTrg->LoCircuit()-1); // -1 !!!
	  
	  y11 = circuit->GetY11Pos(locTrg->LoStripX()); 
// need first to convert deviation to [0-30] 
// (see AliMUONLocalTriggerBoard::LocalTrigger)
	  Int_t deviation = locTrg->LoDev(); 
	  Int_t sign = 0;
	  if ( !locTrg->LoSdev() &&  deviation ) sign=-1;
	  if ( !locTrg->LoSdev() && !deviation ) sign= 0;
	  if (  locTrg->LoSdev() == 1 )          sign=+1;
	  deviation *= sign;
	  deviation += 15;
	  stripX21 = locTrg->LoStripX()+deviation+1;
	  y21 = circuit->GetY21Pos(stripX21);       
	  x11 = circuit->GetX11Pos(locTrg->LoStripY());
	  
	  AliDebug(1, Form(" MakeTriggerTrack %d %d %d %d %d %f %f %f \n",i,locTrg->LoCircuit(),
			   locTrg->LoStripX(),locTrg->LoStripX()+locTrg->LoDev()+1,locTrg->LoStripY(),y11, y21, x11));
	  
	  Float_t thetax = TMath::ATan2( x11 , z11 );
	  Float_t thetay = TMath::ATan2( (y21-y11) , (z21-z11) );
	  
	  fTriggerTrack->SetX11(x11);
	  fTriggerTrack->SetY11(y11);
	  fTriggerTrack->SetThetax(thetax);
	  fTriggerTrack->SetThetay(thetay);
	  fTriggerTrack->SetGTPattern(gloTrigPat);
	  fTriggerTrack->SetLoTrgNum(i);
	      
	  fMUONData->AddRecTriggerTrack(*fTriggerTrack);
      } // board is fired 
  } // end of loop on Local Trigger
  
  return kTRUE;    
}

//__________________________________________________________________________
void AliMUONVTrackReconstructor::EventDumpTrigger(void)
{
  /// Dump reconstructed trigger event 
  /// and the particle parameters
  AliMUONTriggerTrack *triggertrack ;
  Int_t nTriggerTracks = fMUONData->RecTriggerTracks()->GetEntriesFast();
 
  AliDebug(1, "****** enter EventDumpTrigger ******");
  AliDebug(1, Form("Number of Reconstructed tracks : %d ",  nTriggerTracks));
  
  // Loop over reconstructed tracks
  for (Int_t trackIndex = 0; trackIndex < nTriggerTracks; trackIndex++) {
    triggertrack = (AliMUONTriggerTrack*)fMUONData->RecTriggerTracks()->At(trackIndex);
      printf(" trigger track number %i x11=%f y11=%f thetax=%f thetay=%f \n",
	     trackIndex,
	     triggertrack->GetX11(),triggertrack->GetY11(),
	     triggertrack->GetThetax(),triggertrack->GetThetay());      
  } 
}
Commit	Line	Data
8d0843c6	1	/**************************************************************************
	2	* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	3	* *
	4	* Author: The ALICE Off-line Project. *
	5	* Contributors are mentioned in the code where appropriate. *
	6	* *
	7	* Permission to use, copy, modify and distribute this software and its *
	8	* documentation strictly for non-commercial purposes is hereby granted *
	9	* without fee, provided that the above copyright notice appears in all *
	10	* copies and that both the copyright notice and this permission notice *
	11	* appear in the supporting documentation. The authors make no claims *
	12	* about the suitability of this software for any purpose. It is *
	13	* provided "as is" without express or implied warranty. *
	14	**************************************************************************/
	15
	16	/* $Id$ */
	17
	18	////////////////////////////////////
06ca6d7b	19	///
c4ee792d	20	/// \class AliMUONVTrackReconstructor
06ca6d7b	21	/// Virtual MUON track reconstructor in ALICE (class renamed from AliMUONEventReconstructor)
	22	///
	23	/// This class contains as data:
	24	/// * a pointer to the array of hits to be reconstructed (the event)
	25	/// * a pointer to the array of segments made with these hits inside each station
	26	/// * a pointer to the array of reconstructed tracks
	27	///
	28	/// It contains as methods, among others:
	29	/// * EventReconstruct to build the muon tracks
	30	/// * EventReconstructTrigger to build the trigger tracks
	31	///
c4ee792d	32	/// \author Philippe Pillot
c4ee792d	33	///
8d0843c6	34	////////////////////////////////////
8d0843c6	35
8d0843c6	36	#include "AliMUONVTrackReconstructor.h"
6b092dfc	37	#include "AliMUONRecData.h"
8d0843c6	38	#include "AliMUONConstants.h"
8d0843c6	39	#include "AliMUONHitForRec.h"
208f139e	40	#include "AliMUONObjectPair.h"
8d0843c6	41	#include "AliMUONTriggerTrack.h"
e1a10d41	42	#include "AliMUONTriggerCircuit.h"
8d0843c6	43	#include "AliMUONLocalTrigger.h"
8d0843c6	44	#include "AliMUONGlobalTrigger.h"
8d0843c6	45	#include "AliMUONTrack.h"
37827b29	46	#include "AliMUONTrackParam.h"
37827b29	47	#include "AliMUONTrackExtrap.h"
7771752e	48	#include "AliMUONTrackHitPattern.h"
4889d34c	49
8d0843c6	50	#include "AliLog.h"
	51	#include "AliTracker.h"
	52
4889d34c	53	#include <Riostream.h>
	54	#include <TClonesArray.h>
	55	#include <TMath.h>
	56
78649106	57	/// \cond CLASSIMP
8d0843c6	58	ClassImp(AliMUONVTrackReconstructor) // Class implementation in ROOT context
78649106	59	/// \endcond
8d0843c6	60
	61	//************* Defaults parameters for reconstruction
	62	const Double_t AliMUONVTrackReconstructor::fgkDefaultMinBendingMomentum = 3.0;
	63	const Double_t AliMUONVTrackReconstructor::fgkDefaultMaxBendingMomentum = 3000.0;
	64	const Double_t AliMUONVTrackReconstructor::fgkDefaultBendingResolution = 0.01;
	65	const Double_t AliMUONVTrackReconstructor::fgkDefaultNonBendingResolution = 0.144;
208f139e	66	const Double_t AliMUONVTrackReconstructor::fgkDefaultBendingVertexDispersion = 10.;
	67	const Double_t AliMUONVTrackReconstructor::fgkDefaultNonBendingVertexDispersion = 10.;
	68	const Double_t AliMUONVTrackReconstructor::fgkDefaultMaxNormChi2MatchTrigger = 16.0;
8d0843c6	69
8d0843c6	70	//__________________________________________________________________________
6b092dfc	71	AliMUONVTrackReconstructor::AliMUONVTrackReconstructor(AliMUONRecData* data)
8d0843c6	72	: TObject(),
	73	fMinBendingMomentum(fgkDefaultMinBendingMomentum),
	74	fMaxBendingMomentum(fgkDefaultMaxBendingMomentum),
	75	fBendingResolution(fgkDefaultBendingResolution),
	76	fNonBendingResolution(fgkDefaultNonBendingResolution),
208f139e	77	fBendingVertexDispersion(fgkDefaultBendingVertexDispersion),
	78	fNonBendingVertexDispersion(fgkDefaultNonBendingVertexDispersion),
	79	fMaxNormChi2MatchTrigger(fgkDefaultMaxNormChi2MatchTrigger),
8d0843c6	80	fHitsForRecPtr(0x0),
	81	fNHitsForRec(0),
	82	fNHitsForRecPerChamber(0x0),
	83	fIndexOfFirstHitForRecPerChamber(0x0),
8d0843c6	84	fRecTracksPtr(0x0),
	85	fNRecTracks(0),
	86	fMUONData(data),
	87	fTriggerTrack(new AliMUONTriggerTrack()),
	88	fTriggerCircuit(0x0)
	89	{
	90	/// Constructor for class AliMUONVTrackReconstructor
8d0843c6	91	fNHitsForRecPerChamber = new Int_t[AliMUONConstants::NTrackingCh()];
	92	fIndexOfFirstHitForRecPerChamber = new Int_t[AliMUONConstants::NTrackingCh()];
	93
8d0843c6	94	// Memory allocation for the TClonesArray of hits for reconstruction
	95	// Is 10000 the right size ????
	96	fHitsForRecPtr = new TClonesArray("AliMUONHitForRec", 10000);
	97
208f139e	98	// set the magnetic field for track extrapolations
8d0843c6	99	const AliMagF* kField = AliTracker::GetFieldMap();
8d0843c6	100	if (!kField) AliFatal("No field available");
37827b29	101	AliMUONTrackExtrap::SetField(kField);
7771752e	102
7771752e	103	fTrackHitPattern = new AliMUONTrackHitPattern(fMUONData);
8d0843c6	104	}
	105
	106	//__________________________________________________________________________
	107	AliMUONVTrackReconstructor::~AliMUONVTrackReconstructor(void)
	108	{
	109	/// Destructor for class AliMUONVTrackReconstructor
8d0843c6	110	delete [] fNHitsForRecPerChamber;
	111	delete [] fIndexOfFirstHitForRecPerChamber;
	112	delete fTriggerTrack;
	113	delete fHitsForRecPtr;
7771752e	114	delete fTrackHitPattern;
8d0843c6	115	}
8d0843c6	116
8d0843c6	117	//__________________________________________________________________________
	118	void AliMUONVTrackReconstructor::EventReconstruct(void)
	119	{
22ccc301	120	/// To reconstruct one event
8d0843c6	121	AliDebug(1,"Enter EventReconstruct");
22ccc301	122
8d0843c6	123	ResetTracks(); //AZ
8d0843c6	124	ResetHitsForRec(); //AZ
8d0843c6	125	AddHitsForRecFromRawClusters();
8d0843c6	126	MakeTracks();
	127	if (fMUONData->IsTriggerTrackBranchesInTree())
	128	ValidateTracksWithTrigger();
7771752e	129
	130	fTrackHitPattern->GetHitPattern(fRecTracksPtr);
	131
8d0843c6	132	// Add tracks to MUON data container
	133	for(Int_t i=0; i<fNRecTracks; i++) {
	134	AliMUONTrack * track = (AliMUONTrack*) fRecTracksPtr->At(i);
	135	fMUONData->AddRecTrack(*track);
	136	}
	137	}
	138
	139	//__________________________________________________________________________
	140	void AliMUONVTrackReconstructor::ResetTracks(void)
	141	{
	142	/// To reset the TClonesArray of reconstructed tracks
	143	if (fRecTracksPtr) fRecTracksPtr->Delete();
	144	// Delete in order that the Track destructors are called,
	145	// hence the space for the TClonesArray of pointers to TrackHit's is freed
	146	fNRecTracks = 0;
	147	return;
	148	}
	149
8d0843c6	150	//__________________________________________________________________________
	151	void AliMUONVTrackReconstructor::ResetHitsForRec(void)
	152	{
	153	/// To reset the array and the number of HitsForRec,
	154	/// and also the number of HitsForRec
	155	/// and the index of the first HitForRec per chamber
	156	if (fHitsForRecPtr) fHitsForRecPtr->Delete();
	157	fNHitsForRec = 0;
	158	for (Int_t ch = 0; ch < AliMUONConstants::NTrackingCh(); ch++)
	159	fNHitsForRecPerChamber[ch] = fIndexOfFirstHitForRecPerChamber[ch] = 0;
	160	return;
	161	}
	162
	163	//__________________________________________________________________________
	164	void AliMUONVTrackReconstructor::SortHitsForRecWithIncreasingChamber()
	165	{
	166	/// Sort HitsForRec's in increasing order with respect to chamber number.
	167	/// Uses the function "Compare".
	168	/// Update the information for HitsForRec per chamber too.
	169	Int_t ch, nhits, prevch;
	170	fHitsForRecPtr->Sort();
	171	for (ch = 0; ch < AliMUONConstants::NTrackingCh(); ch++) {
	172	fNHitsForRecPerChamber[ch] = 0;
	173	fIndexOfFirstHitForRecPerChamber[ch] = 0;
	174	}
	175	prevch = 0; // previous chamber
	176	nhits = 0; // number of hits in current chamber
	177	// Loop over HitsForRec
	178	for (Int_t hit = 0; hit < fNHitsForRec; hit++) {
	179	// chamber number (0...)
	180	ch = ((AliMUONHitForRec) ((fHitsForRecPtr)[hit]))->GetChamberNumber();
	181	// increment number of hits in current chamber
	182	(fNHitsForRecPerChamber[ch])++;
	183	// update index of first HitForRec in current chamber
	184	// if chamber number different from previous one
	185	if (ch != prevch) {
	186	fIndexOfFirstHitForRecPerChamber[ch] = hit;
	187	prevch = ch;
	188	}
	189	}
	190	return;
	191	}
	192
	193	//__________________________________________________________________________
208f139e	194	TClonesArray* AliMUONVTrackReconstructor::MakeSegmentsInStation(Int_t station)
8d0843c6	195	{
208f139e	196	/// To make the list of segments in station(0..) "Station" from the list of hits to be reconstructed.
	197	/// Return a new TClonesArray of segments.
	198	/// It is the responsibility of the user to delete it afterward.
	199	AliDebug(1,Form("Enter MakeSegmentsPerStation (0...) %d",station));
	200
8d0843c6	201	AliMUONHitForRec hit1Ptr, hit2Ptr;
208f139e	202	AliMUONObjectPair *segment;
22ccc301	203	Double_t bendingSlope = 0, impactParam = 0., bendingMomentum = 0.; // to avoid compilation warning
8d0843c6	204	// first and second chambers (0...) in the station
208f139e	205	Int_t ch1 = 2 * station;
8d0843c6	206	Int_t ch2 = ch1 + 1;
208f139e	207	// list of segments
	208	TClonesArray *segments = new TClonesArray("AliMUONObjectPair", fNHitsForRecPerChamber[ch2]);
	209	// Loop over HitForRec's in the first chamber of the station
8d0843c6	210	for (Int_t hit1 = fIndexOfFirstHitForRecPerChamber[ch1];
	211	hit1 < fIndexOfFirstHitForRecPerChamber[ch1] + fNHitsForRecPerChamber[ch1];
	212	hit1++) {
	213	// pointer to the HitForRec
	214	hit1Ptr = (AliMUONHitForRec) ((fHitsForRecPtr)[hit1]);
208f139e	215	// Loop over HitsForRec's in the second chamber of the station
8d0843c6	216	for (Int_t hit2 = fIndexOfFirstHitForRecPerChamber[ch2];
	217	hit2 < fIndexOfFirstHitForRecPerChamber[ch2] + fNHitsForRecPerChamber[ch2];
	218	hit2++) {
	219	// pointer to the HitForRec
	220	hit2Ptr = (AliMUONHitForRec) ((fHitsForRecPtr)[hit2]);
208f139e	221	if ( hit1Ptr->GetZ() - hit2Ptr->GetZ() != 0. ) {
22ccc301	222	// bending slope
208f139e	223	bendingSlope = (hit1Ptr->GetBendingCoor() - hit2Ptr->GetBendingCoor()) / (hit1Ptr->GetZ() - hit2Ptr->GetZ());
	224	// impact parameter
	225	impactParam = hit1Ptr->GetBendingCoor() - hit1Ptr->GetZ() * bendingSlope;
	226	// absolute value of bending momentum
	227	bendingMomentum = TMath::Abs(AliMUONTrackExtrap::GetBendingMomentumFromImpactParam(impactParam));
	228	} else {
	229	AliWarning("hit1Ptr->GetZ() = hit2Ptr->GetZ(): no segment created");
	230	continue;
	231	}
22ccc301	232	// check for bending momentum within tolerances
22ccc301	233	if ((bendingMomentum < fMaxBendingMomentum) && (bendingMomentum > fMinBendingMomentum)) {
8d0843c6	234	// make new segment
208f139e	235	segment = new ((*segments)[segments->GetLast()+1]) AliMUONObjectPair(hit1Ptr, hit2Ptr, kFALSE, kFALSE);
8d0843c6	236	if (AliLog::GetGlobalDebugLevel() > 1) {
22ccc301	237	cout << "segmentIndex(0...): " << segments->GetLast() << endl;
8d0843c6	238	segment->Dump();
	239	cout << "HitForRec in first chamber" << endl;
	240	hit1Ptr->Dump();
	241	cout << "HitForRec in second chamber" << endl;
	242	hit2Ptr->Dump();
208f139e	243	}
8d0843c6	244	}
	245	} //for (Int_t hit2
	246	} // for (Int_t hit1...
208f139e	247	AliDebug(1,Form("Station: %d NSegments: %d ", station, segments->GetEntriesFast()));
208f139e	248	return segments;
8d0843c6	249	}
	250
	251	//__________________________________________________________________________
	252	void AliMUONVTrackReconstructor::ValidateTracksWithTrigger(void)
	253	{
	254	/// Try to match track from tracking system with trigger track
06ca6d7b	255	static const Double_t kDistSigma[3]={1,1,0.02}; // sigma of distributions (trigger-track) X,Y,slopeY
208f139e	256
423b32ca	257	Int_t loTrigger = 0;
	258	Int_t loCirc = -1;
	259	Int_t loStripX = -1;
	260	Int_t loStripY = -1;
	261	Int_t loDev = -1;
	262	Int_t loLpt = -1;
	263	Int_t loHpt = -1;
	264
8d0843c6	265	AliMUONTrack *track;
	266	AliMUONTrackParam trackParam;
	267	AliMUONTriggerTrack *triggerTrack;
7771752e	268	AliMUONLocalTrigger *locTrg;
8d0843c6	269
	270	fMUONData->SetTreeAddress("RL");
	271	fMUONData->GetRecTriggerTracks();
	272	TClonesArray *recTriggerTracks = fMUONData->RecTriggerTracks();
7771752e	273
	274	fMUONData->SetTreeAddress("TC");
	275	fMUONData->GetTrigger();
	276	TClonesArray *localTrigger = fMUONData->LocalTrigger();
	277
	278	Int_t matchTrigger;
c6ba19f7	279	Int_t loTrgNum;
8d0843c6	280	Double_t distTriggerTrack[3];
208f139e	281	Double_t xTrack, yTrack, ySlopeTrack, chi2MatchTrigger, minChi2MatchTrigger, chi2;
8d0843c6	282
	283	track = (AliMUONTrack*) fRecTracksPtr->First();
	284	while (track) {
423b32ca	285	matchTrigger = 0;
8d0843c6	286	chi2MatchTrigger = 0.;
c6ba19f7	287	loTrgNum = -1;
7771752e	288	Int_t doubleMatch=-1; // Check if track matches 2 trigger tracks
7771752e	289	Double_t doubleChi2 = -1.;
c6ba19f7	290
8d0843c6	291	trackParam = ((AliMUONTrackParam) (track->GetTrackParamAtHit()->Last()));
37827b29	292	AliMUONTrackExtrap::ExtrapToZ(&trackParam, AliMUONConstants::DefaultChamberZ(10)); // extrap to 1st trigger chamber
8d0843c6	293
	294	xTrack = trackParam.GetNonBendingCoor();
	295	yTrack = trackParam.GetBendingCoor();
	296	ySlopeTrack = trackParam.GetBendingSlope();
208f139e	297	minChi2MatchTrigger = 999.;
8d0843c6	298
	299	triggerTrack = (AliMUONTriggerTrack*) recTriggerTracks->First();
	300	while(triggerTrack){
06ca6d7b	301	distTriggerTrack[0] = (triggerTrack->GetX11()-xTrack)/kDistSigma[0];
	302	distTriggerTrack[1] = (triggerTrack->GetY11()-yTrack)/kDistSigma[1];
	303	distTriggerTrack[2] = (TMath::Tan(triggerTrack->GetThetay())-ySlopeTrack)/kDistSigma[2];
208f139e	304	chi2 = 0.;
	305	for (Int_t iVar = 0; iVar < 3; iVar++) chi2 += distTriggerTrack[iVar]*distTriggerTrack[iVar];
	306	chi2 /= 3.; // Normalized Chi2: 3 degrees of freedom (X,Y,slopeY)
7771752e	307	if (chi2 < fMaxNormChi2MatchTrigger) {
	308	Bool_t isDoubleTrack = (TMath::Abs(chi2 - minChi2MatchTrigger)<1.);
	309	if (chi2 < minChi2MatchTrigger && chi2 < fMaxNormChi2MatchTrigger) {
	310	if(isDoubleTrack){
	311	doubleMatch = loTrgNum;
	312	doubleChi2 = chi2MatchTrigger;
	313	}
	314	minChi2MatchTrigger = chi2;
	315	chi2MatchTrigger = chi2;
	316	loTrgNum=triggerTrack->GetLoTrgNum();
	317	locTrg = (AliMUONLocalTrigger*)localTrigger->UncheckedAt(loTrgNum);
423b32ca	318	matchTrigger=1;
	319	if(locTrg->LoLpt()>0)matchTrigger=2;
	320	if(locTrg->LoHpt()>0)matchTrigger=3;
7771752e	321	}
	322	else if(isDoubleTrack) {
	323	doubleMatch = triggerTrack->GetLoTrgNum();
	324	doubleChi2 = chi2;
	325	}
8d0843c6	326	}
	327	triggerTrack = (AliMUONTriggerTrack*) recTriggerTracks->After(triggerTrack);
	328	}
7771752e	329	if(doubleMatch>=0){ // If two trigger tracks match, select the one passing more trigger cuts
	330	AliDebug(1, Form("Two candidates found: %i and %i",loTrgNum,doubleMatch));
	331	AliMUONLocalTrigger locTrg1 = (AliMUONLocalTrigger)localTrigger->UncheckedAt(doubleMatch);
423b32ca	332	if((locTrg1->LoLpt()>0 && matchTrigger<2) \|\| (locTrg1->LoHpt() && matchTrigger<3)){
	333	if(locTrg1->LoHpt()>0)matchTrigger=3;
	334	else matchTrigger=2;
7771752e	335	loTrgNum = doubleMatch;
	336	chi2MatchTrigger=doubleChi2;
	337	}
	338	}
8d0843c6	339
8d0843c6	340	track->SetMatchTrigger(matchTrigger);
c6ba19f7	341	track->SetLoTrgNum(loTrgNum);
8d0843c6	342	track->SetChi2MatchTrigger(chi2MatchTrigger);
423b32ca	343
	344	if (loTrgNum >= 0 && loTrgNum < 234) {
	345	locTrg = (AliMUONLocalTrigger*)localTrigger->UncheckedAt(loTrgNum);
	346
	347	loCirc = locTrg->LoCircuit();
	348	loStripX = locTrg->LoStripX();
	349	loStripY = locTrg->LoStripY();
	350	loDev = locTrg->LoDev();
	351	loLpt = locTrg->LoLpt();
	352	loHpt = locTrg->LoHpt();
	353
	354	track->SetLocalTrigger(loCirc,loStripX,loStripY,loDev,loLpt,loHpt);
	355	}
	356
8d0843c6	357	track = (AliMUONTrack*) fRecTracksPtr->After(track);
	358	}
	359
	360	return;
	361	}
	362
	363	//__________________________________________________________________________
	364	void AliMUONVTrackReconstructor::EventReconstructTrigger(void)
	365	{
	366	/// To reconstruct trigger for one event
	367	AliDebug(1,"Enter EventReconstructTrigger");
	368	MakeTriggerTracks();
	369	return;
	370	}
	371
	372	//__________________________________________________________________________
	373	Bool_t AliMUONVTrackReconstructor::MakeTriggerTracks(void)
	374	{
71a2d3aa	375	/// To make the trigger tracks from Local Trigger
8d0843c6	376	AliDebug(1, "Enter MakeTriggerTracks");
e6b25a6e	377
e6b25a6e	378	TTree* treeR;
e6b25a6e	379	UChar_t gloTrigPat;
	380	TClonesArray *localTrigger;
	381	TClonesArray *globalTrigger;
	382	AliMUONLocalTrigger *locTrg;
	383	AliMUONGlobalTrigger *gloTrg;
	384
	385	treeR = fMUONData->TreeR();
	386	if (!treeR) {
	387	AliWarning("TreeR is not loaded");
	388	return kFALSE;
	389	}
	390
e6b25a6e	391	fMUONData->SetTreeAddress("TC");
e6b25a6e	392	fMUONData->GetTrigger();
8d0843c6	393
e6b25a6e	394	// global trigger for trigger pattern
	395	gloTrigPat = 0;
	396	globalTrigger = fMUONData->GlobalTrigger();
	397	gloTrg = (AliMUONGlobalTrigger*)globalTrigger->UncheckedAt(0);
8d0843c6	398
e6b25a6e	399	if (gloTrg)
	400	gloTrigPat = gloTrg->GetGlobalResponse();
	401
	402
	403	// local trigger for tracking
	404	localTrigger = fMUONData->LocalTrigger();
	405	Int_t nlocals = (Int_t) (localTrigger->GetEntries());
8d0843c6	406
e6b25a6e	407	Float_t z11 = AliMUONConstants::DefaultChamberZ(10);
	408	Float_t z21 = AliMUONConstants::DefaultChamberZ(12);
	409
	410	Float_t y11 = 0.;
	411	Int_t stripX21 = 0;
	412	Float_t y21 = 0.;
	413	Float_t x11 = 0.;
	414
	415	for (Int_t i=0; i<nlocals; i++) { // loop on Local Trigger
7cf63da6	416	locTrg = (AliMUONLocalTrigger*)localTrigger->UncheckedAt(i);
7cf63da6	417
8b0baca4	418	Bool_t xTrig=kFALSE;
	419	Bool_t yTrig=kFALSE;
	420
	421	if ( locTrg->LoSdev()==1 && locTrg->LoDev()==0 &&
	422	locTrg->LoStripX()==0) xTrig=kFALSE; // no trigger in X
	423	else xTrig=kTRUE; // trigger in X
	424	if (locTrg->LoTrigY()==1 &&
	425	locTrg->LoStripY()==15 ) yTrig = kFALSE; // no trigger in Y
	426	else yTrig = kTRUE; // trigger in Y
	427
	428	if (xTrig && yTrig) { // make Trigger Track if trigger in X and Y
7cf63da6	429
	430	AliDebug(1, "AliMUONTrackReconstructor::MakeTriggerTrack using NEW trigger \n");
	431	AliMUONTriggerCircuit* circuit =
	432	(AliMUONTriggerCircuit*)fTriggerCircuit->At(locTrg->LoCircuit()-1); // -1 !!!
	433
	434	y11 = circuit->GetY11Pos(locTrg->LoStripX());
8602dabf	435	// need first to convert deviation to [0-30]
	436	// (see AliMUONLocalTriggerBoard::LocalTrigger)
	437	Int_t deviation = locTrg->LoDev();
	438	Int_t sign = 0;
	439	if ( !locTrg->LoSdev() && deviation ) sign=-1;
	440	if ( !locTrg->LoSdev() && !deviation ) sign= 0;
	441	if ( locTrg->LoSdev() == 1 ) sign=+1;
	442	deviation *= sign;
8b0baca4	443	deviation += 15;
8b0baca4	444	stripX21 = locTrg->LoStripX()+deviation+1;
7cf63da6	445	y21 = circuit->GetY21Pos(stripX21);
	446	x11 = circuit->GetX11Pos(locTrg->LoStripY());
	447
	448	AliDebug(1, Form(" MakeTriggerTrack %d %d %d %d %d %f %f %f \n",i,locTrg->LoCircuit(),
	449	locTrg->LoStripX(),locTrg->LoStripX()+locTrg->LoDev()+1,locTrg->LoStripY(),y11, y21, x11));
	450
	451	Float_t thetax = TMath::ATan2( x11 , z11 );
	452	Float_t thetay = TMath::ATan2( (y21-y11) , (z21-z11) );
	453
	454	fTriggerTrack->SetX11(x11);
	455	fTriggerTrack->SetY11(y11);
	456	fTriggerTrack->SetThetax(thetax);
	457	fTriggerTrack->SetThetay(thetay);
	458	fTriggerTrack->SetGTPattern(gloTrigPat);
	459	fTriggerTrack->SetLoTrgNum(i);
8b0baca4	460
7cf63da6	461	fMUONData->AddRecTriggerTrack(*fTriggerTrack);
7cf63da6	462	} // board is fired
e6b25a6e	463	} // end of loop on Local Trigger
	464
	465	return kTRUE;
8d0843c6	466	}
	467
	468	//__________________________________________________________________________
	469	void AliMUONVTrackReconstructor::EventDumpTrigger(void)
	470	{
	471	/// Dump reconstructed trigger event
	472	/// and the particle parameters
	473	AliMUONTriggerTrack *triggertrack ;
	474	Int_t nTriggerTracks = fMUONData->RecTriggerTracks()->GetEntriesFast();
	475
	476	AliDebug(1, "**** enter EventDumpTrigger ****");
	477	AliDebug(1, Form("Number of Reconstructed tracks : %d ", nTriggerTracks));
	478
	479	// Loop over reconstructed tracks
	480	for (Int_t trackIndex = 0; trackIndex < nTriggerTracks; trackIndex++) {
	481	triggertrack = (AliMUONTriggerTrack*)fMUONData->RecTriggerTracks()->At(trackIndex);
	482	printf(" trigger track number %i x11=%f y11=%f thetax=%f thetay=%f \n",
	483	trackIndex,
	484	triggertrack->GetX11(),triggertrack->GetY11(),
	485	triggertrack->GetThetax(),triggertrack->GetThetay());
	486	}
	487	}
	488