[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
///
////////////////////////////////////

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

#include "AliMUONVTrackReconstructor.h"
#include "AliMUONData.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 "AliMagF.h"
#include "AliLog.h"
#include "AliTracker.h"

ClassImp(AliMUONVTrackReconstructor) // Class implementation in ROOT context

//************* 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(AliMUONData* data)
  : TObject(),
    fMinBendingMomentum(fgkDefaultMinBendingMomentum),
    fMaxBendingMomentum(fgkDefaultMaxBendingMomentum),
    fBendingResolution(fgkDefaultBendingResolution),
    fNonBendingResolution(fgkDefaultNonBendingResolution),
    fBendingVertexDispersion(fgkDefaultBendingVertexDispersion),
    fNonBendingVertexDispersion(fgkDefaultNonBendingVertexDispersion),
    fMaxNormChi2MatchTrigger(fgkDefaultMaxNormChi2MatchTrigger),
    fSegmentMaxDistBending(0x0),
    fSegmentMaxDistNonBending(0x0),
    fHitsForRecPtr(0x0),
    fNHitsForRec(0),
    fNHitsForRecPerChamber(0x0),
    fIndexOfFirstHitForRecPerChamber(0x0),
    fRecTracksPtr(0x0),
    fNRecTracks(0),
    fMUONData(data),
    fTriggerTrack(new AliMUONTriggerTrack()),
    fTriggerCircuit(0x0)
{
  /// Constructor for class AliMUONVTrackReconstructor
  fSegmentMaxDistBending = new Double_t[AliMUONConstants::NTrackingSt()];
  fSegmentMaxDistNonBending = new Double_t[AliMUONConstants::NTrackingSt()];
  fNHitsForRecPerChamber = new Int_t[AliMUONConstants::NTrackingCh()];
  fIndexOfFirstHitForRecPerChamber = new Int_t[AliMUONConstants::NTrackingCh()];

  SetReconstructionParametersToDefaults();
  
  // 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);
}

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

  //__________________________________________________________________________
void AliMUONVTrackReconstructor::SetReconstructionParametersToDefaults(void)
{
  /// Set reconstruction parameters for making segments to default values
  // Would be much more convenient with a structure (or class) ????

  // ******** Parameters for making segments
  // should be parametrized ????
  // according to interval between chambers in a station ????
  // Maximum distance in non bending plane
  // 5 * 0.22 just to remember the way it was made in TRACKF_STAT
  // SIGCUT*DYMAX(IZ)
  for (Int_t st = 0; st < AliMUONConstants::NTrackingSt(); st++)
    fSegmentMaxDistNonBending[st] = 5. * 0.22;
  // Maximum distance in bending plane:
  // values from TRACKF_STAT, corresponding to (J psi 20cm),
  // scaled to the real distance between chambers in a station
  fSegmentMaxDistBending[0] = TMath::Abs( 1.5 *
					  (AliMUONConstants::DefaultChamberZ(1) - AliMUONConstants::DefaultChamberZ(0)) / 20.0);
  fSegmentMaxDistBending[1] = TMath::Abs( 1.5 *
					  (AliMUONConstants::DefaultChamberZ(3) - AliMUONConstants::DefaultChamberZ(2)) / 20.0);
  fSegmentMaxDistBending[2] = TMath::Abs( 3.0 *
					  (AliMUONConstants::DefaultChamberZ(5) - AliMUONConstants::DefaultChamberZ(4)) / 20.0);
  fSegmentMaxDistBending[3] = TMath::Abs( 6.0 *
					  (AliMUONConstants::DefaultChamberZ(7) - AliMUONConstants::DefaultChamberZ(6)) / 20.0);
  fSegmentMaxDistBending[4] = TMath::Abs( 6.0 *
					  (AliMUONConstants::DefaultChamberZ(9) - AliMUONConstants::DefaultChamberZ(8)) / 20.0);

  return;
}

  //__________________________________________________________________________
void AliMUONVTrackReconstructor::EventReconstruct(void)
{
  // To reconstruct one event
  AliDebug(1,"Enter EventReconstruct");
  ResetTracks(); //AZ
  ResetHitsForRec(); //AZ
  AddHitsForRecFromRawClusters();
  MakeTracks();
  if (fMUONData->IsTriggerTrackBranchesInTree()) 
    ValidateTracksWithTrigger(); 
  
  // 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, distBend, distNonBend, extBendCoor, extNonBendCoor, extrapFact;
  Double_t 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]);
    // extrapolation, on the straight line joining the HitForRec to the vertex (0,0,0),
    // to the Z of the HitForRec in the second chamber of the station
    // 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]);
      // absolute values of distances, in bending and non bending planes,
      // between the HitForRec in the second chamber
      // and the previous extrapolation
      extrapFact = hit2Ptr->GetZ()/ hit1Ptr->GetZ();
      extBendCoor = extrapFact * hit1Ptr->GetBendingCoor();
      extNonBendCoor = extrapFact * hit1Ptr->GetNonBendingCoor();
      distBend = TMath::Abs(hit2Ptr->GetBendingCoor() - extBendCoor);
      distNonBend = TMath::Abs(hit2Ptr->GetNonBendingCoor() - extNonBendCoor);
      // bending slope
      if ( hit1Ptr->GetZ() - hit2Ptr->GetZ() != 0. ) {
        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 distances not too large,
      // and impact parameter not too big if stations downstream of the dipole.
      // Conditions "distBend" and "impactParam" correlated for these stations ????
      if ((distBend < fSegmentMaxDistBending[station]) && (distNonBend < fSegmentMaxDistNonBending[station]) &&
	  (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()
	       << "  distBend: " << distBend
	       << "  distNonBend: " << distNonBend
	       << 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
  
  AliMUONTrack *track;
  AliMUONTrackParam trackParam; 
  AliMUONTriggerTrack *triggerTrack;
  
  fMUONData->SetTreeAddress("RL");
  fMUONData->GetRecTriggerTracks();
  TClonesArray *recTriggerTracks = fMUONData->RecTriggerTracks();
  
  Bool_t matchTrigger;
  Int_t loTrgNum;
  Double_t distTriggerTrack[3];
  Double_t xTrack, yTrack, ySlopeTrack, chi2MatchTrigger, minChi2MatchTrigger, chi2;
  
  track = (AliMUONTrack*) fRecTracksPtr->First();
  while (track) {
    matchTrigger = kFALSE;
    chi2MatchTrigger = 0.;
    loTrgNum = -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];
      loTrgNum=triggerTrack->GetLoTrgNum();
      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 < minChi2MatchTrigger && chi2 < fMaxNormChi2MatchTrigger) {
        minChi2MatchTrigger = chi2;
        matchTrigger = kTRUE;
        chi2MatchTrigger = chi2;
      }
      triggerTrack = (AliMUONTriggerTrack*) recTriggerTracks->After(triggerTrack);
    }
    
    track->SetMatchTrigger(matchTrigger);
    track->SetLoTrgNum(loTrgNum);
    track->SetChi2MatchTrigger(chi2MatchTrigger);
    
    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);      

    AliDebug(1, "AliMUONTrackReconstructor::MakeTriggerTrack using NEW trigger \n");
    AliMUONTriggerCircuit* circuit = 
      (AliMUONTriggerCircuit*)fTriggerCircuit->At(locTrg->LoCircuit()-1); // -1 !!!

    y11 = circuit->GetY11Pos(locTrg->LoStripX()); 
    stripX21 = locTrg->LoStripX()+locTrg->LoDev()+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);
  } // 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
06ca6d7b	18	/// \class AliMUONVTrackReconstructor
8d0843c6	19	////////////////////////////////////
06ca6d7b	20	///
	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	///
8d0843c6	32	////////////////////////////////////
	33
	34	#include <stdlib.h>
	35	#include <Riostream.h>
208f139e	36	#include <TClonesArray.h>
9edefa04	37	#include <TMath.h>
9edefa04	38	#include <TTree.h>
8d0843c6	39
	40	#include "AliMUONVTrackReconstructor.h"
	41	#include "AliMUONData.h"
	42	#include "AliMUONConstants.h"
	43	#include "AliMUONHitForRec.h"
208f139e	44	#include "AliMUONObjectPair.h"
8d0843c6	45	#include "AliMUONTriggerTrack.h"
e1a10d41	46	#include "AliMUONTriggerCircuit.h"
8d0843c6	47	#include "AliMUONLocalTrigger.h"
8d0843c6	48	#include "AliMUONGlobalTrigger.h"
8d0843c6	49	#include "AliMUONTrack.h"
37827b29	50	#include "AliMUONTrackParam.h"
37827b29	51	#include "AliMUONTrackExtrap.h"
8d0843c6	52	#include "AliMagF.h"
	53	#include "AliLog.h"
	54	#include "AliTracker.h"
	55
	56	ClassImp(AliMUONVTrackReconstructor) // Class implementation in ROOT context
	57
	58	//************* Defaults parameters for reconstruction
	59	const Double_t AliMUONVTrackReconstructor::fgkDefaultMinBendingMomentum = 3.0;
	60	const Double_t AliMUONVTrackReconstructor::fgkDefaultMaxBendingMomentum = 3000.0;
	61	const Double_t AliMUONVTrackReconstructor::fgkDefaultBendingResolution = 0.01;
	62	const Double_t AliMUONVTrackReconstructor::fgkDefaultNonBendingResolution = 0.144;
208f139e	63	const Double_t AliMUONVTrackReconstructor::fgkDefaultBendingVertexDispersion = 10.;
	64	const Double_t AliMUONVTrackReconstructor::fgkDefaultNonBendingVertexDispersion = 10.;
	65	const Double_t AliMUONVTrackReconstructor::fgkDefaultMaxNormChi2MatchTrigger = 16.0;
8d0843c6	66
	67	//__________________________________________________________________________
	68	AliMUONVTrackReconstructor::AliMUONVTrackReconstructor(AliMUONData* data)
	69	: TObject(),
	70	fMinBendingMomentum(fgkDefaultMinBendingMomentum),
	71	fMaxBendingMomentum(fgkDefaultMaxBendingMomentum),
	72	fBendingResolution(fgkDefaultBendingResolution),
	73	fNonBendingResolution(fgkDefaultNonBendingResolution),
208f139e	74	fBendingVertexDispersion(fgkDefaultBendingVertexDispersion),
	75	fNonBendingVertexDispersion(fgkDefaultNonBendingVertexDispersion),
	76	fMaxNormChi2MatchTrigger(fgkDefaultMaxNormChi2MatchTrigger),
8d0843c6	77	fSegmentMaxDistBending(0x0),
	78	fSegmentMaxDistNonBending(0x0),
	79	fHitsForRecPtr(0x0),
	80	fNHitsForRec(0),
	81	fNHitsForRecPerChamber(0x0),
	82	fIndexOfFirstHitForRecPerChamber(0x0),
8d0843c6	83	fRecTracksPtr(0x0),
	84	fNRecTracks(0),
	85	fMUONData(data),
	86	fTriggerTrack(new AliMUONTriggerTrack()),
	87	fTriggerCircuit(0x0)
	88	{
	89	/// Constructor for class AliMUONVTrackReconstructor
	90	fSegmentMaxDistBending = new Double_t[AliMUONConstants::NTrackingSt()];
	91	fSegmentMaxDistNonBending = new Double_t[AliMUONConstants::NTrackingSt()];
	92	fNHitsForRecPerChamber = new Int_t[AliMUONConstants::NTrackingCh()];
	93	fIndexOfFirstHitForRecPerChamber = new Int_t[AliMUONConstants::NTrackingCh()];
	94
	95	SetReconstructionParametersToDefaults();
	96
	97	// Memory allocation for the TClonesArray of hits for reconstruction
	98	// Is 10000 the right size ????
	99	fHitsForRecPtr = new TClonesArray("AliMUONHitForRec", 10000);
	100
208f139e	101	// set the magnetic field for track extrapolations
8d0843c6	102	const AliMagF* kField = AliTracker::GetFieldMap();
8d0843c6	103	if (!kField) AliFatal("No field available");
37827b29	104	AliMUONTrackExtrap::SetField(kField);
8d0843c6	105	}
	106
	107	//__________________________________________________________________________
	108	AliMUONVTrackReconstructor::~AliMUONVTrackReconstructor(void)
	109	{
	110	/// Destructor for class AliMUONVTrackReconstructor
	111	delete [] fSegmentMaxDistBending;
	112	delete [] fSegmentMaxDistNonBending;
	113	delete [] fNHitsForRecPerChamber;
	114	delete [] fIndexOfFirstHitForRecPerChamber;
	115	delete fTriggerTrack;
	116	delete fHitsForRecPtr;
8d0843c6	117	}
	118
	119	//__________________________________________________________________________
	120	void AliMUONVTrackReconstructor::SetReconstructionParametersToDefaults(void)
	121	{
	122	/// Set reconstruction parameters for making segments to default values
	123	// Would be much more convenient with a structure (or class) ????
	124
	125	// ******** Parameters for making segments
	126	// should be parametrized ????
	127	// according to interval between chambers in a station ????
	128	// Maximum distance in non bending plane
	129	// 5 * 0.22 just to remember the way it was made in TRACKF_STAT
	130	// SIGCUT*DYMAX(IZ)
208f139e	131	for (Int_t st = 0; st < AliMUONConstants::NTrackingSt(); st++)
8d0843c6	132	fSegmentMaxDistNonBending[st] = 5. * 0.22;
	133	// Maximum distance in bending plane:
	134	// values from TRACKF_STAT, corresponding to (J psi 20cm),
	135	// scaled to the real distance between chambers in a station
	136	fSegmentMaxDistBending[0] = TMath::Abs( 1.5 *
	137	(AliMUONConstants::DefaultChamberZ(1) - AliMUONConstants::DefaultChamberZ(0)) / 20.0);
	138	fSegmentMaxDistBending[1] = TMath::Abs( 1.5 *
	139	(AliMUONConstants::DefaultChamberZ(3) - AliMUONConstants::DefaultChamberZ(2)) / 20.0);
	140	fSegmentMaxDistBending[2] = TMath::Abs( 3.0 *
	141	(AliMUONConstants::DefaultChamberZ(5) - AliMUONConstants::DefaultChamberZ(4)) / 20.0);
	142	fSegmentMaxDistBending[3] = TMath::Abs( 6.0 *
	143	(AliMUONConstants::DefaultChamberZ(7) - AliMUONConstants::DefaultChamberZ(6)) / 20.0);
	144	fSegmentMaxDistBending[4] = TMath::Abs( 6.0 *
	145	(AliMUONConstants::DefaultChamberZ(9) - AliMUONConstants::DefaultChamberZ(8)) / 20.0);
	146
	147	return;
	148	}
	149
	150	//__________________________________________________________________________
	151	void AliMUONVTrackReconstructor::EventReconstruct(void)
	152	{
	153	// To reconstruct one event
	154	AliDebug(1,"Enter EventReconstruct");
	155	ResetTracks(); //AZ
8d0843c6	156	ResetHitsForRec(); //AZ
8d0843c6	157	AddHitsForRecFromRawClusters();
8d0843c6	158	MakeTracks();
	159	if (fMUONData->IsTriggerTrackBranchesInTree())
	160	ValidateTracksWithTrigger();
	161
	162	// Add tracks to MUON data container
	163	for(Int_t i=0; i<fNRecTracks; i++) {
	164	AliMUONTrack * track = (AliMUONTrack*) fRecTracksPtr->At(i);
	165	fMUONData->AddRecTrack(*track);
	166	}
	167	}
	168
	169	//__________________________________________________________________________
	170	void AliMUONVTrackReconstructor::ResetTracks(void)
	171	{
	172	/// To reset the TClonesArray of reconstructed tracks
	173	if (fRecTracksPtr) fRecTracksPtr->Delete();
	174	// Delete in order that the Track destructors are called,
	175	// hence the space for the TClonesArray of pointers to TrackHit's is freed
	176	fNRecTracks = 0;
	177	return;
	178	}
	179
8d0843c6	180	//__________________________________________________________________________
	181	void AliMUONVTrackReconstructor::ResetHitsForRec(void)
	182	{
	183	/// To reset the array and the number of HitsForRec,
	184	/// and also the number of HitsForRec
	185	/// and the index of the first HitForRec per chamber
	186	if (fHitsForRecPtr) fHitsForRecPtr->Delete();
	187	fNHitsForRec = 0;
	188	for (Int_t ch = 0; ch < AliMUONConstants::NTrackingCh(); ch++)
	189	fNHitsForRecPerChamber[ch] = fIndexOfFirstHitForRecPerChamber[ch] = 0;
	190	return;
	191	}
	192
	193	//__________________________________________________________________________
	194	void AliMUONVTrackReconstructor::SortHitsForRecWithIncreasingChamber()
	195	{
	196	/// Sort HitsForRec's in increasing order with respect to chamber number.
	197	/// Uses the function "Compare".
	198	/// Update the information for HitsForRec per chamber too.
	199	Int_t ch, nhits, prevch;
	200	fHitsForRecPtr->Sort();
	201	for (ch = 0; ch < AliMUONConstants::NTrackingCh(); ch++) {
	202	fNHitsForRecPerChamber[ch] = 0;
	203	fIndexOfFirstHitForRecPerChamber[ch] = 0;
	204	}
	205	prevch = 0; // previous chamber
	206	nhits = 0; // number of hits in current chamber
	207	// Loop over HitsForRec
	208	for (Int_t hit = 0; hit < fNHitsForRec; hit++) {
	209	// chamber number (0...)
	210	ch = ((AliMUONHitForRec) ((fHitsForRecPtr)[hit]))->GetChamberNumber();
	211	// increment number of hits in current chamber
	212	(fNHitsForRecPerChamber[ch])++;
	213	// update index of first HitForRec in current chamber
	214	// if chamber number different from previous one
	215	if (ch != prevch) {
	216	fIndexOfFirstHitForRecPerChamber[ch] = hit;
	217	prevch = ch;
	218	}
	219	}
	220	return;
	221	}
	222
	223	//__________________________________________________________________________
208f139e	224	TClonesArray* AliMUONVTrackReconstructor::MakeSegmentsInStation(Int_t station)
8d0843c6	225	{
208f139e	226	/// To make the list of segments in station(0..) "Station" from the list of hits to be reconstructed.
	227	/// Return a new TClonesArray of segments.
	228	/// It is the responsibility of the user to delete it afterward.
	229	AliDebug(1,Form("Enter MakeSegmentsPerStation (0...) %d",station));
	230
8d0843c6	231	AliMUONHitForRec hit1Ptr, hit2Ptr;
208f139e	232	AliMUONObjectPair *segment;
	233	Double_t bendingSlope, distBend, distNonBend, extBendCoor, extNonBendCoor, extrapFact;
	234	Double_t impactParam = 0., bendingMomentum = 0.; // to avoid compilation warning
8d0843c6	235	// first and second chambers (0...) in the station
208f139e	236	Int_t ch1 = 2 * station;
8d0843c6	237	Int_t ch2 = ch1 + 1;
208f139e	238	// list of segments
	239	TClonesArray *segments = new TClonesArray("AliMUONObjectPair", fNHitsForRecPerChamber[ch2]);
	240	// Loop over HitForRec's in the first chamber of the station
8d0843c6	241	for (Int_t hit1 = fIndexOfFirstHitForRecPerChamber[ch1];
	242	hit1 < fIndexOfFirstHitForRecPerChamber[ch1] + fNHitsForRecPerChamber[ch1];
	243	hit1++) {
	244	// pointer to the HitForRec
	245	hit1Ptr = (AliMUONHitForRec) ((fHitsForRecPtr)[hit1]);
208f139e	246	// extrapolation, on the straight line joining the HitForRec to the vertex (0,0,0),
	247	// to the Z of the HitForRec in the second chamber of the station
	248	// Loop over HitsForRec's in the second chamber of the station
8d0843c6	249	for (Int_t hit2 = fIndexOfFirstHitForRecPerChamber[ch2];
	250	hit2 < fIndexOfFirstHitForRecPerChamber[ch2] + fNHitsForRecPerChamber[ch2];
	251	hit2++) {
	252	// pointer to the HitForRec
	253	hit2Ptr = (AliMUONHitForRec) ((fHitsForRecPtr)[hit2]);
	254	// absolute values of distances, in bending and non bending planes,
	255	// between the HitForRec in the second chamber
	256	// and the previous extrapolation
	257	extrapFact = hit2Ptr->GetZ()/ hit1Ptr->GetZ();
	258	extBendCoor = extrapFact * hit1Ptr->GetBendingCoor();
	259	extNonBendCoor = extrapFact * hit1Ptr->GetNonBendingCoor();
	260	distBend = TMath::Abs(hit2Ptr->GetBendingCoor() - extBendCoor);
	261	distNonBend = TMath::Abs(hit2Ptr->GetNonBendingCoor() - extNonBendCoor);
208f139e	262	// bending slope
	263	if ( hit1Ptr->GetZ() - hit2Ptr->GetZ() != 0. ) {
	264	bendingSlope = (hit1Ptr->GetBendingCoor() - hit2Ptr->GetBendingCoor()) / (hit1Ptr->GetZ() - hit2Ptr->GetZ());
	265	// impact parameter
	266	impactParam = hit1Ptr->GetBendingCoor() - hit1Ptr->GetZ() * bendingSlope;
	267	// absolute value of bending momentum
	268	bendingMomentum = TMath::Abs(AliMUONTrackExtrap::GetBendingMomentumFromImpactParam(impactParam));
	269	} else {
	270	AliWarning("hit1Ptr->GetZ() = hit2Ptr->GetZ(): no segment created");
	271	continue;
	272	}
8d0843c6	273	// check for distances not too large,
	274	// and impact parameter not too big if stations downstream of the dipole.
	275	// Conditions "distBend" and "impactParam" correlated for these stations ????
208f139e	276	if ((distBend < fSegmentMaxDistBending[station]) && (distNonBend < fSegmentMaxDistNonBending[station]) &&
208f139e	277	(bendingMomentum < fMaxBendingMomentum) && (bendingMomentum > fMinBendingMomentum)) {
8d0843c6	278	// make new segment
208f139e	279	segment = new ((*segments)[segments->GetLast()+1]) AliMUONObjectPair(hit1Ptr, hit2Ptr, kFALSE, kFALSE);
8d0843c6	280	if (AliLog::GetGlobalDebugLevel() > 1) {
208f139e	281	cout << "segmentIndex(0...): " << segments->GetLast()
8d0843c6	282	<< " distBend: " << distBend
	283	<< " distNonBend: " << distNonBend
	284	<< endl;
	285	segment->Dump();
	286	cout << "HitForRec in first chamber" << endl;
	287	hit1Ptr->Dump();
	288	cout << "HitForRec in second chamber" << endl;
	289	hit2Ptr->Dump();
208f139e	290	}
8d0843c6	291	}
	292	} //for (Int_t hit2
	293	} // for (Int_t hit1...
208f139e	294	AliDebug(1,Form("Station: %d NSegments: %d ", station, segments->GetEntriesFast()));
208f139e	295	return segments;
8d0843c6	296	}
	297
	298	//__________________________________________________________________________
	299	void AliMUONVTrackReconstructor::ValidateTracksWithTrigger(void)
	300	{
	301	/// Try to match track from tracking system with trigger track
06ca6d7b	302	static const Double_t kDistSigma[3]={1,1,0.02}; // sigma of distributions (trigger-track) X,Y,slopeY
208f139e	303
8d0843c6	304	AliMUONTrack *track;
	305	AliMUONTrackParam trackParam;
	306	AliMUONTriggerTrack *triggerTrack;
	307
	308	fMUONData->SetTreeAddress("RL");
	309	fMUONData->GetRecTriggerTracks();
	310	TClonesArray *recTriggerTracks = fMUONData->RecTriggerTracks();
	311
	312	Bool_t matchTrigger;
c6ba19f7	313	Int_t loTrgNum;
8d0843c6	314	Double_t distTriggerTrack[3];
208f139e	315	Double_t xTrack, yTrack, ySlopeTrack, chi2MatchTrigger, minChi2MatchTrigger, chi2;
8d0843c6	316
	317	track = (AliMUONTrack*) fRecTracksPtr->First();
	318	while (track) {
	319	matchTrigger = kFALSE;
	320	chi2MatchTrigger = 0.;
c6ba19f7	321	loTrgNum = -1;
c6ba19f7	322
8d0843c6	323	trackParam = ((AliMUONTrackParam) (track->GetTrackParamAtHit()->Last()));
37827b29	324	AliMUONTrackExtrap::ExtrapToZ(&trackParam, AliMUONConstants::DefaultChamberZ(10)); // extrap to 1st trigger chamber
8d0843c6	325
	326	xTrack = trackParam.GetNonBendingCoor();
	327	yTrack = trackParam.GetBendingCoor();
	328	ySlopeTrack = trackParam.GetBendingSlope();
208f139e	329	minChi2MatchTrigger = 999.;
8d0843c6	330
	331	triggerTrack = (AliMUONTriggerTrack*) recTriggerTracks->First();
	332	while(triggerTrack){
06ca6d7b	333	distTriggerTrack[0] = (triggerTrack->GetX11()-xTrack)/kDistSigma[0];
	334	distTriggerTrack[1] = (triggerTrack->GetY11()-yTrack)/kDistSigma[1];
	335	distTriggerTrack[2] = (TMath::Tan(triggerTrack->GetThetay())-ySlopeTrack)/kDistSigma[2];
c6ba19f7	336	loTrgNum=triggerTrack->GetLoTrgNum();
208f139e	337	chi2 = 0.;
	338	for (Int_t iVar = 0; iVar < 3; iVar++) chi2 += distTriggerTrack[iVar]*distTriggerTrack[iVar];
	339	chi2 /= 3.; // Normalized Chi2: 3 degrees of freedom (X,Y,slopeY)
	340	if (chi2 < minChi2MatchTrigger && chi2 < fMaxNormChi2MatchTrigger) {
	341	minChi2MatchTrigger = chi2;
8d0843c6	342	matchTrigger = kTRUE;
208f139e	343	chi2MatchTrigger = chi2;
8d0843c6	344	}
	345	triggerTrack = (AliMUONTriggerTrack*) recTriggerTracks->After(triggerTrack);
	346	}
	347
	348	track->SetMatchTrigger(matchTrigger);
c6ba19f7	349	track->SetLoTrgNum(loTrgNum);
8d0843c6	350	track->SetChi2MatchTrigger(chi2MatchTrigger);
	351
	352	track = (AliMUONTrack*) fRecTracksPtr->After(track);
	353	}
	354
	355	return;
	356	}
	357
	358	//__________________________________________________________________________
	359	void AliMUONVTrackReconstructor::EventReconstructTrigger(void)
	360	{
	361	/// To reconstruct trigger for one event
	362	AliDebug(1,"Enter EventReconstructTrigger");
	363	MakeTriggerTracks();
	364	return;
	365	}
	366
	367	//__________________________________________________________________________
	368	Bool_t AliMUONVTrackReconstructor::MakeTriggerTracks(void)
	369	{
	370	// To make the trigger tracks from Local Trigger
	371	AliDebug(1, "Enter MakeTriggerTracks");
e6b25a6e	372
e6b25a6e	373	TTree* treeR;
e6b25a6e	374	UChar_t gloTrigPat;
	375	TClonesArray *localTrigger;
	376	TClonesArray *globalTrigger;
	377	AliMUONLocalTrigger *locTrg;
	378	AliMUONGlobalTrigger *gloTrg;
	379
	380	treeR = fMUONData->TreeR();
	381	if (!treeR) {
	382	AliWarning("TreeR is not loaded");
	383	return kFALSE;
	384	}
	385
e6b25a6e	386	fMUONData->SetTreeAddress("TC");
e6b25a6e	387	fMUONData->GetTrigger();
8d0843c6	388
e6b25a6e	389	// global trigger for trigger pattern
	390	gloTrigPat = 0;
	391	globalTrigger = fMUONData->GlobalTrigger();
	392	gloTrg = (AliMUONGlobalTrigger*)globalTrigger->UncheckedAt(0);
8d0843c6	393
e6b25a6e	394	if (gloTrg)
	395	gloTrigPat = gloTrg->GetGlobalResponse();
	396
	397
	398	// local trigger for tracking
	399	localTrigger = fMUONData->LocalTrigger();
	400	Int_t nlocals = (Int_t) (localTrigger->GetEntries());
8d0843c6	401
e6b25a6e	402	Float_t z11 = AliMUONConstants::DefaultChamberZ(10);
	403	Float_t z21 = AliMUONConstants::DefaultChamberZ(12);
	404
	405	Float_t y11 = 0.;
	406	Int_t stripX21 = 0;
	407	Float_t y21 = 0.;
	408	Float_t x11 = 0.;
	409
	410	for (Int_t i=0; i<nlocals; i++) { // loop on Local Trigger
	411	locTrg = (AliMUONLocalTrigger*)localTrigger->UncheckedAt(i);
	412
	413	AliDebug(1, "AliMUONTrackReconstructor::MakeTriggerTrack using NEW trigger \n");
	414	AliMUONTriggerCircuit* circuit =
	415	(AliMUONTriggerCircuit*)fTriggerCircuit->At(locTrg->LoCircuit()-1); // -1 !!!
	416
	417	y11 = circuit->GetY11Pos(locTrg->LoStripX());
	418	stripX21 = locTrg->LoStripX()+locTrg->LoDev()+1;
	419	y21 = circuit->GetY21Pos(stripX21);
	420	x11 = circuit->GetX11Pos(locTrg->LoStripY());
	421
	422	AliDebug(1, Form(" MakeTriggerTrack %d %d %d %d %d %f %f %f \n",i,locTrg->LoCircuit(),
	423	locTrg->LoStripX(),locTrg->LoStripX()+locTrg->LoDev()+1,locTrg->LoStripY(),y11, y21, x11));
	424
	425	Float_t thetax = TMath::ATan2( x11 , z11 );
	426	Float_t thetay = TMath::ATan2( (y21-y11) , (z21-z11) );
	427
	428	fTriggerTrack->SetX11(x11);
	429	fTriggerTrack->SetY11(y11);
	430	fTriggerTrack->SetThetax(thetax);
	431	fTriggerTrack->SetThetay(thetay);
	432	fTriggerTrack->SetGTPattern(gloTrigPat);
c6ba19f7	433	fTriggerTrack->SetLoTrgNum(i);
e6b25a6e	434
	435	fMUONData->AddRecTriggerTrack(*fTriggerTrack);
	436	} // end of loop on Local Trigger
	437
	438	return kTRUE;
8d0843c6	439	}
	440
	441	//__________________________________________________________________________
	442	void AliMUONVTrackReconstructor::EventDumpTrigger(void)
	443	{
	444	/// Dump reconstructed trigger event
	445	/// and the particle parameters
	446	AliMUONTriggerTrack *triggertrack ;
	447	Int_t nTriggerTracks = fMUONData->RecTriggerTracks()->GetEntriesFast();
	448
	449	AliDebug(1, "**** enter EventDumpTrigger ****");
	450	AliDebug(1, Form("Number of Reconstructed tracks : %d ", nTriggerTracks));
	451
	452	// Loop over reconstructed tracks
	453	for (Int_t trackIndex = 0; trackIndex < nTriggerTracks; trackIndex++) {
	454	triggertrack = (AliMUONTriggerTrack*)fMUONData->RecTriggerTracks()->At(trackIndex);
	455	printf(" trigger track number %i x11=%f y11=%f thetax=%f thetay=%f \n",
	456	trackIndex,
	457	triggertrack->GetX11(),triggertrack->GetY11(),
	458	triggertrack->GetThetax(),triggertrack->GetThetay());
	459	}
	460	}
	461