[u/mrichter/AliRoot.git] / MUON / AliMUONReconstructor.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 for MUON reconstruction                                             //
//                                                                           //
///////////////////////////////////////////////////////////////////////////////
#include <TParticle.h>
#include <TArrayF.h>

#include "AliRunLoader.h"
#include "AliHeader.h"
#include "AliGenEventHeader.h"
#include "AliESD.h"
#include "AliMUONReconstructor.h"
 
#include "AliMUONData.h"
#include "AliMUONTrackReconstructor.h"
#include "AliMUONClusterReconstructor.h"
#include "AliMUONClusterFinderVS.h"
#include "AliMUONTrack.h"
#include "AliMUONTrackParam.h"
#include "AliMUONTriggerTrack.h"
#include "AliESDMuonTrack.h"
#include "AliMUONRawData.h"

#include "AliRawReader.h"


ClassImp(AliMUONReconstructor)
//_____________________________________________________________________________
AliMUONReconstructor::AliMUONReconstructor()
  : AliReconstructor()
{
}
//_____________________________________________________________________________
AliMUONReconstructor::~AliMUONReconstructor()
{
}
//_____________________________________________________________________________
void AliMUONReconstructor::Reconstruct(AliRunLoader* runLoader) const
{
//  AliLoader
  AliLoader* loader = runLoader->GetLoader("MUONLoader");
  Int_t nEvents = runLoader->GetNumberOfEvents();

// used local container for each method
// passing fLoader as argument, could be avoided ???
  AliMUONTrackReconstructor* recoEvent = new AliMUONTrackReconstructor(loader);
  AliMUONData* dataEvent = recoEvent->GetMUONData();

  AliMUONClusterReconstructor* recoCluster = new AliMUONClusterReconstructor(loader);
  AliMUONData* dataCluster = recoCluster->GetMUONData();
  AliMUONClusterFinderVS *recModel = recoCluster->GetRecoModel();
  recModel->SetGhostChi2Cut(10);

  loader->LoadDigits("READ");
  loader->LoadRecPoints("RECREATE");
  loader->LoadTracks("RECREATE");
  
  //   Loop over events              
  for(Int_t ievent = 0; ievent < nEvents; ievent++) {
    printf("Event %d\n",ievent);
    runLoader->GetEvent(ievent);

    //----------------------- digit2cluster & Trigger2Trigger -------------------
    if (!loader->TreeR()) loader->MakeRecPointsContainer();
     
    // tracking branch
    dataCluster->MakeBranch("RC");
    dataCluster->SetTreeAddress("D,RC");
    recoCluster->Digits2Clusters(); 
    dataCluster->Fill("RC"); 

    // trigger branch
    dataCluster->MakeBranch("TC");
    dataCluster->SetTreeAddress("TC");
    recoCluster->Trigger2Trigger(); 
    dataCluster->Fill("TC");

    loader->WriteRecPoints("OVERWRITE");

    //---------------------------- Track & TriggerTrack ---------------------
    if (!loader->TreeT()) loader->MakeTracksContainer();

    // trigger branch
    dataEvent->MakeBranch("RL"); //trigger track
    dataEvent->SetTreeAddress("RL");
    recoEvent->EventReconstructTrigger();
    dataEvent->Fill("RL");

    // tracking branch
    dataEvent->MakeBranch("RT"); //track
    dataEvent->SetTreeAddress("RT");
    recoEvent->EventReconstruct();
    dataEvent->Fill("RT");

    loader->WriteTracks("OVERWRITE"); 
  
    //--------------------------- Resetting branches -----------------------
    dataCluster->ResetDigits();
    dataCluster->ResetRawClusters();
    dataCluster->ResetTrigger();

    dataEvent->ResetRawClusters();
    dataEvent->ResetTrigger();
    dataEvent->ResetRecTracks();  
    dataEvent->ResetRecTriggerTracks();

  }
  loader->UnloadDigits();
  loader->UnloadRecPoints();
  loader->UnloadTracks();

  delete recoCluster;
  delete recoEvent;
}

//_____________________________________________________________________________
void AliMUONReconstructor::Reconstruct(AliRunLoader* runLoader, AliRawReader* rawReader) const
{
//  AliLoader
  AliLoader* loader = runLoader->GetLoader("MUONLoader");

// used local container for each method
// passing fLoader as argument, could be avoided ???
  AliMUONTrackReconstructor* recoEvent = new AliMUONTrackReconstructor(loader);
  AliMUONData* dataEvent = recoEvent->GetMUONData();

  AliMUONRawData* rawData = new AliMUONRawData(loader);
  AliMUONData* dataCluster = rawData->GetMUONData();

  AliMUONClusterReconstructor* recoCluster = new AliMUONClusterReconstructor(loader, dataCluster);
  AliMUONClusterFinderVS *recModel = recoCluster->GetRecoModel();
  recModel->SetGhostChi2Cut(10);

  loader->LoadRecPoints("RECREATE");
  loader->LoadTracks("RECREATE");
  loader->LoadDigits("RECREATE");


  //   Loop over events  
  Int_t iEvent = 0;
            
  while (rawReader->NextEvent()) {
    printf("Event %d\n",iEvent);
    runLoader->GetEvent(iEvent++);

    //----------------------- raw2digits & raw2trigger-------------------
    if (!loader->TreeD()) loader->MakeDigitsContainer();

    dataCluster->MakeBranch("D,GLT");
    dataCluster->SetTreeAddress("D,GLT");
    rawData->Raw2Digits(rawReader);
    dataCluster->Fill("D,GLT"); 

    loader->WriteDigits("OVERWRITE");

    //----------------------- digit2cluster & Trigger2Trigger -------------------
    if (!loader->TreeR()) loader->MakeRecPointsContainer();
     
    // tracking branch
    dataCluster->MakeBranch("RC");
    dataCluster->SetTreeAddress("D,RC");
    recoCluster->Digits2Clusters(); 
    dataCluster->Fill("RC"); 

    // trigger branch
    dataCluster->MakeBranch("TC");
    dataCluster->SetTreeAddress("TC");
    recoCluster->Trigger2Trigger(); 
    dataCluster->Fill("TC");

    loader->WriteRecPoints("OVERWRITE");

    //---------------------------- Track & TriggerTrack ---------------------
    if (!loader->TreeT()) loader->MakeTracksContainer();

    // trigger branch
    dataEvent->MakeBranch("RL"); //trigger track
    dataEvent->SetTreeAddress("RL");
    recoEvent->EventReconstructTrigger();
    dataEvent->Fill("RL");

    // tracking branch
    dataEvent->MakeBranch("RT"); //track
    dataEvent->SetTreeAddress("RT");
    recoEvent->EventReconstruct();
    dataEvent->Fill("RT");

    loader->WriteTracks("OVERWRITE");  
  
    //--------------------------- Resetting branches -----------------------
    dataCluster->ResetDigits();
    dataCluster->ResetRawClusters();
    dataCluster->ResetTrigger();

    dataEvent->ResetRawClusters();
    dataEvent->ResetTrigger();
    dataEvent->ResetRecTracks();
    dataEvent->ResetRecTriggerTracks();
  
  }
  loader->UnloadRecPoints();
  loader->UnloadTracks();
  loader->UnloadDigits();

  delete recoCluster;
  delete recoEvent;
}

//_____________________________________________________________________________
void AliMUONReconstructor::FillESD(AliRunLoader* runLoader, AliESD* esd) const
{
  TClonesArray* recTracksArray = 0;
  TClonesArray* recTrigTracksArray = 0;
  
  AliLoader* loader = runLoader->GetLoader("MUONLoader");
  loader->LoadTracks("READ");
  AliMUONData* muonData = new AliMUONData(loader,"MUON","MUON");

   // declaration  
  Int_t iEvent, nPart;
  Int_t nTrackHits, nPrimary;
  Double_t fitFmin;
  TArrayF vertex(3);

  Double_t bendingSlope, nonBendingSlope, inverseBendingMomentum;
  Double_t xRec, yRec, zRec, chi2MatchTrigger;
  Bool_t matchTrigger;

  // setting pointer for tracks, triggertracks & trackparam at vertex
  AliMUONTrack* recTrack = 0;
  AliMUONTrackParam* trackParam = 0;
  AliMUONTriggerTrack* recTriggerTrack = 0;
  TParticle* particle = new TParticle();
  AliGenEventHeader* header = 0;
  iEvent = runLoader->GetEventNumber(); 
  runLoader->GetEvent(iEvent);

  // vertex calculation (maybe it exists already somewhere else)
  vertex[0] = vertex[1] = vertex[2] = 0.;
  nPrimary = 0;
  if ( (header = runLoader->GetHeader()->GenEventHeader()) ) {
    header->PrimaryVertex(vertex);
  } else {
    runLoader->LoadKinematics("READ");
    runLoader->TreeK()->GetBranch("Particles")->SetAddress(&particle);
    nPart = (Int_t)runLoader->TreeK()->GetEntries();
    for(Int_t iPart = 0; iPart < nPart; iPart++) {
      runLoader->TreeK()->GetEvent(iPart);
      if (particle->GetFirstMother() == -1) {
	vertex[0] += particle->Vx();
	vertex[1] += particle->Vy();
	vertex[2] += particle->Vz();
	nPrimary++;
      }
      if (nPrimary) {
	vertex[0] /= (double)nPrimary;
	vertex[1] /= (double)nPrimary;
	vertex[2] /= (double)nPrimary;
      }
    }
  }
  // setting ESD MUON class
  AliESDMuonTrack* theESDTrack = new  AliESDMuonTrack() ;

  //-------------------- trigger tracks-------------
  Long_t trigPat = 0;
  muonData->SetTreeAddress("RL");
  muonData->GetRecTriggerTracks();
  recTrigTracksArray = muonData->RecTriggerTracks();

  // ready global trigger pattern from first track
  if (recTrigTracksArray) 
    recTriggerTrack = (AliMUONTriggerTrack*) recTrigTracksArray->First();
  if (recTriggerTrack) trigPat = recTriggerTrack->GetGTPattern();

  //printf(">>> Event %d Number of Recconstructed tracks %d \n",iEvent, nrectracks);
 
  // -------------------- tracks-------------
  muonData->SetTreeAddress("RT");
  muonData->GetRecTracks();
  recTracksArray = muonData->RecTracks();
        
  Int_t nRecTracks = 0;
  if (recTracksArray)
    nRecTracks = (Int_t) recTracksArray->GetEntriesFast(); //
  
  // loop over tracks
  for (Int_t iRecTracks = 0; iRecTracks <  nRecTracks;  iRecTracks++) {

    // reading info from tracks
    recTrack = (AliMUONTrack*) recTracksArray->At(iRecTracks);

    trackParam = (AliMUONTrackParam*) (recTrack->GetTrackParamAtHit())->First();
    trackParam->ExtrapToVertex(vertex[0],vertex[1],vertex[2]);

    bendingSlope            = trackParam->GetBendingSlope();
    nonBendingSlope         = trackParam->GetNonBendingSlope();
    inverseBendingMomentum = trackParam->GetInverseBendingMomentum();
    xRec  = trackParam->GetNonBendingCoor();
    yRec  = trackParam->GetBendingCoor();
    zRec  = trackParam->GetZ();

    nTrackHits       = recTrack->GetNTrackHits();
    fitFmin          = recTrack->GetFitFMin();
    matchTrigger     = recTrack->GetMatchTrigger();
    chi2MatchTrigger = recTrack->GetChi2MatchTrigger();

    // setting data member of ESD MUON
    theESDTrack->SetInverseBendingMomentum(inverseBendingMomentum);
    theESDTrack->SetThetaX(TMath::ATan(nonBendingSlope));
    theESDTrack->SetThetaY(TMath::ATan(bendingSlope));
    theESDTrack->SetZ(zRec);
    theESDTrack->SetBendingCoor(yRec); // calculate vertex at ESD or Tracking level ?
    theESDTrack->SetNonBendingCoor(xRec);
    theESDTrack->SetChi2(fitFmin);
    theESDTrack->SetNHit(nTrackHits);
    theESDTrack->SetMatchTrigger(matchTrigger);
    theESDTrack->SetChi2MatchTrigger(chi2MatchTrigger);

    // storing ESD MUON Track into ESD Event 
    if (nRecTracks != 0)  
      esd->AddMuonTrack(theESDTrack);
  } // end loop tracks

  // add global trigger pattern
  if (nRecTracks != 0)  
    esd->SetTrigger(trigPat);

  // reset muondata
  muonData->ResetRecTracks();
  muonData->ResetRecTriggerTracks();

  //} // end loop on event  
  loader->UnloadTracks(); 
  if (!header)
    runLoader->UnloadKinematics();
  delete theESDTrack;
  delete muonData;
  delete particle;
  // delete particle;
}
Commit	Line	Data
cac2eb58	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	**************************************************************************/
cac2eb58	15	/* $Id$ */
	16
	17	///////////////////////////////////////////////////////////////////////////////
	18	// //
b002ac70	19	// class for MUON reconstruction //
cac2eb58	20	// //
cac2eb58	21	///////////////////////////////////////////////////////////////////////////////
b002ac70	22	#include <TParticle.h>
30178c30	23	#include <TArrayF.h>
30178c30	24
cac2eb58	25	#include "AliRunLoader.h"
b002ac70	26	#include "AliHeader.h"
b002ac70	27	#include "AliGenEventHeader.h"
cac2eb58	28	#include "AliESD.h"
cf464691	29	#include "AliMUONReconstructor.h"
29f1b13a	30
cac2eb58	31	#include "AliMUONData.h"
29f1b13a	32	#include "AliMUONTrackReconstructor.h"
cac2eb58	33	#include "AliMUONClusterReconstructor.h"
cac2eb58	34	#include "AliMUONClusterFinderVS.h"
	35	#include "AliMUONTrack.h"
	36	#include "AliMUONTrackParam.h"
	37	#include "AliMUONTriggerTrack.h"
	38	#include "AliESDMuonTrack.h"
1197ff51	39	#include "AliMUONRawData.h"
1197ff51	40
cf464691	41	#include "AliRawReader.h"
cac2eb58	42
29f1b13a	43
cac2eb58	44	ClassImp(AliMUONReconstructor)
	45	//_____________________________________________________________________________
	46	AliMUONReconstructor::AliMUONReconstructor()
30178c30	47	: AliReconstructor()
8789635b	48	{
	49	}
	50	//_____________________________________________________________________________
	51	AliMUONReconstructor::~AliMUONReconstructor()
cac2eb58	52	{
	53	}
	54	//_____________________________________________________________________________
	55	void AliMUONReconstructor::Reconstruct(AliRunLoader* runLoader) const
	56	{
	57	// AliLoader
	58	AliLoader* loader = runLoader->GetLoader("MUONLoader");
	59	Int_t nEvents = runLoader->GetNumberOfEvents();
	60
	61	// used local container for each method
	62	// passing fLoader as argument, could be avoided ???
29f1b13a	63	AliMUONTrackReconstructor* recoEvent = new AliMUONTrackReconstructor(loader);
cac2eb58	64	AliMUONData* dataEvent = recoEvent->GetMUONData();
	65
	66	AliMUONClusterReconstructor* recoCluster = new AliMUONClusterReconstructor(loader);
	67	AliMUONData* dataCluster = recoCluster->GetMUONData();
7e4a628d	68	AliMUONClusterFinderVS *recModel = recoCluster->GetRecoModel();
7e4a628d	69	recModel->SetGhostChi2Cut(10);
cac2eb58	70
	71	loader->LoadDigits("READ");
	72	loader->LoadRecPoints("RECREATE");
	73	loader->LoadTracks("RECREATE");
	74
	75	// Loop over events
	76	for(Int_t ievent = 0; ievent < nEvents; ievent++) {
	77	printf("Event %d\n",ievent);
	78	runLoader->GetEvent(ievent);
	79
8210cb40	80	//----------------------- digit2cluster & Trigger2Trigger -------------------
cac2eb58	81	if (!loader->TreeR()) loader->MakeRecPointsContainer();
	82
	83	// tracking branch
	84	dataCluster->MakeBranch("RC");
	85	dataCluster->SetTreeAddress("D,RC");
	86	recoCluster->Digits2Clusters();
	87	dataCluster->Fill("RC");
	88
	89	// trigger branch
ce3e25a8	90	dataCluster->MakeBranch("TC");
ce3e25a8	91	dataCluster->SetTreeAddress("TC");
7e4a628d	92	recoCluster->Trigger2Trigger();
ce3e25a8	93	dataCluster->Fill("TC");
cac2eb58	94
	95	loader->WriteRecPoints("OVERWRITE");
	96
	97	//---------------------------- Track & TriggerTrack ---------------------
	98	if (!loader->TreeT()) loader->MakeTracksContainer();
	99
	100	// trigger branch
	101	dataEvent->MakeBranch("RL"); //trigger track
	102	dataEvent->SetTreeAddress("RL");
	103	recoEvent->EventReconstructTrigger();
	104	dataEvent->Fill("RL");
	105
	106	// tracking branch
	107	dataEvent->MakeBranch("RT"); //track
	108	dataEvent->SetTreeAddress("RT");
	109	recoEvent->EventReconstruct();
	110	dataEvent->Fill("RT");
	111
29f1b13a	112	loader->WriteTracks("OVERWRITE");
cac2eb58	113
	114	//--------------------------- Resetting branches -----------------------
	115	dataCluster->ResetDigits();
	116	dataCluster->ResetRawClusters();
ce3e25a8	117	dataCluster->ResetTrigger();
cac2eb58	118
	119	dataEvent->ResetRawClusters();
	120	dataEvent->ResetTrigger();
29f1b13a	121	dataEvent->ResetRecTracks();
cac2eb58	122	dataEvent->ResetRecTriggerTracks();
29f1b13a	123
cac2eb58	124	}
	125	loader->UnloadDigits();
	126	loader->UnloadRecPoints();
	127	loader->UnloadTracks();
	128
	129	delete recoCluster;
	130	delete recoEvent;
cac2eb58	131	}
cf464691	132
	133	//_____________________________________________________________________________
	134	void AliMUONReconstructor::Reconstruct(AliRunLoader* runLoader, AliRawReader* rawReader) const
	135	{
	136	// AliLoader
	137	AliLoader* loader = runLoader->GetLoader("MUONLoader");
	138
	139	// used local container for each method
	140	// passing fLoader as argument, could be avoided ???
29f1b13a	141	AliMUONTrackReconstructor* recoEvent = new AliMUONTrackReconstructor(loader);
cf464691	142	AliMUONData* dataEvent = recoEvent->GetMUONData();
cf464691	143
1197ff51	144	AliMUONRawData* rawData = new AliMUONRawData(loader);
	145	AliMUONData* dataCluster = rawData->GetMUONData();
	146
	147	AliMUONClusterReconstructor* recoCluster = new AliMUONClusterReconstructor(loader, dataCluster);
7e4a628d	148	AliMUONClusterFinderVS *recModel = recoCluster->GetRecoModel();
7e4a628d	149	recModel->SetGhostChi2Cut(10);
cf464691	150
cf464691	151	loader->LoadRecPoints("RECREATE");
cf464691	152	loader->LoadTracks("RECREATE");
ce6a659c	153	loader->LoadDigits("RECREATE");
	154
	155
cf464691	156	// Loop over events
8210cb40	157	Int_t iEvent = 0;
cf464691	158
8210cb40	159	while (rawReader->NextEvent()) {
cf464691	160	printf("Event %d\n",iEvent);
cf464691	161	runLoader->GetEvent(iEvent++);
ce6a659c	162
	163	//----------------------- raw2digits & raw2trigger-------------------
	164	if (!loader->TreeD()) loader->MakeDigitsContainer();
	165
	166	dataCluster->MakeBranch("D,GLT");
	167	dataCluster->SetTreeAddress("D,GLT");
1197ff51	168	rawData->Raw2Digits(rawReader);
ce6a659c	169	dataCluster->Fill("D,GLT");
	170
	171	loader->WriteDigits("OVERWRITE");
cf464691	172
8210cb40	173	//----------------------- digit2cluster & Trigger2Trigger -------------------
cf464691	174	if (!loader->TreeR()) loader->MakeRecPointsContainer();
	175
	176	// tracking branch
	177	dataCluster->MakeBranch("RC");
	178	dataCluster->SetTreeAddress("D,RC");
1197ff51	179	recoCluster->Digits2Clusters();
cf464691	180	dataCluster->Fill("RC");
	181
	182	// trigger branch
	183	dataCluster->MakeBranch("TC");
	184	dataCluster->SetTreeAddress("TC");
1197ff51	185	recoCluster->Trigger2Trigger();
cf464691	186	dataCluster->Fill("TC");
	187
	188	loader->WriteRecPoints("OVERWRITE");
	189
	190	//---------------------------- Track & TriggerTrack ---------------------
	191	if (!loader->TreeT()) loader->MakeTracksContainer();
	192
	193	// trigger branch
	194	dataEvent->MakeBranch("RL"); //trigger track
	195	dataEvent->SetTreeAddress("RL");
	196	recoEvent->EventReconstructTrigger();
	197	dataEvent->Fill("RL");
	198
	199	// tracking branch
	200	dataEvent->MakeBranch("RT"); //track
	201	dataEvent->SetTreeAddress("RT");
	202	recoEvent->EventReconstruct();
	203	dataEvent->Fill("RT");
	204
	205	loader->WriteTracks("OVERWRITE");
	206
	207	//--------------------------- Resetting branches -----------------------
	208	dataCluster->ResetDigits();
	209	dataCluster->ResetRawClusters();
	210	dataCluster->ResetTrigger();
	211
	212	dataEvent->ResetRawClusters();
	213	dataEvent->ResetTrigger();
	214	dataEvent->ResetRecTracks();
	215	dataEvent->ResetRecTriggerTracks();
	216
	217	}
cf464691	218	loader->UnloadRecPoints();
cf464691	219	loader->UnloadTracks();
ce6a659c	220	loader->UnloadDigits();
cf464691	221
	222	delete recoCluster;
	223	delete recoEvent;
cf464691	224	}
cf464691	225
cac2eb58	226	//_____________________________________________________________________________
	227	void AliMUONReconstructor::FillESD(AliRunLoader* runLoader, AliESD* esd) const
	228	{
	229	TClonesArray* recTracksArray = 0;
	230	TClonesArray* recTrigTracksArray = 0;
	231
	232	AliLoader* loader = runLoader->GetLoader("MUONLoader");
	233	loader->LoadTracks("READ");
cac2eb58	234	AliMUONData* muonData = new AliMUONData(loader,"MUON","MUON");
	235
	236	// declaration
b002ac70	237	Int_t iEvent, nPart;
b002ac70	238	Int_t nTrackHits, nPrimary;
cac2eb58	239	Double_t fitFmin;
b002ac70	240	TArrayF vertex(3);
cac2eb58	241
	242	Double_t bendingSlope, nonBendingSlope, inverseBendingMomentum;
	243	Double_t xRec, yRec, zRec, chi2MatchTrigger;
	244	Bool_t matchTrigger;
	245
b002ac70	246	// setting pointer for tracks, triggertracks & trackparam at vertex
cac2eb58	247	AliMUONTrack* recTrack = 0;
	248	AliMUONTrackParam* trackParam = 0;
	249	AliMUONTriggerTrack* recTriggerTrack = 0;
b002ac70	250	TParticle* particle = new TParticle();
b002ac70	251	AliGenEventHeader* header = 0;
cac2eb58	252	iEvent = runLoader->GetEventNumber();
	253	runLoader->GetEvent(iEvent);
	254
b002ac70	255	// vertex calculation (maybe it exists already somewhere else)
	256	vertex[0] = vertex[1] = vertex[2] = 0.;
	257	nPrimary = 0;
	258	if ( (header = runLoader->GetHeader()->GenEventHeader()) ) {
	259	header->PrimaryVertex(vertex);
	260	} else {
	261	runLoader->LoadKinematics("READ");
	262	runLoader->TreeK()->GetBranch("Particles")->SetAddress(&particle);
	263	nPart = (Int_t)runLoader->TreeK()->GetEntries();
	264	for(Int_t iPart = 0; iPart < nPart; iPart++) {
	265	runLoader->TreeK()->GetEvent(iPart);
	266	if (particle->GetFirstMother() == -1) {
	267	vertex[0] += particle->Vx();
	268	vertex[1] += particle->Vy();
	269	vertex[2] += particle->Vz();
	270	nPrimary++;
	271	}
	272	if (nPrimary) {
	273	vertex[0] /= (double)nPrimary;
	274	vertex[1] /= (double)nPrimary;
	275	vertex[2] /= (double)nPrimary;
	276	}
	277	}
	278	}
cac2eb58	279	// setting ESD MUON class
30178c30	280	AliESDMuonTrack* theESDTrack = new AliESDMuonTrack() ;
cac2eb58	281
	282	//-------------------- trigger tracks-------------
	283	Long_t trigPat = 0;
	284	muonData->SetTreeAddress("RL");
	285	muonData->GetRecTriggerTracks();
	286	recTrigTracksArray = muonData->RecTriggerTracks();
	287
	288	// ready global trigger pattern from first track
	289	if (recTrigTracksArray)
	290	recTriggerTrack = (AliMUONTriggerTrack*) recTrigTracksArray->First();
	291	if (recTriggerTrack) trigPat = recTriggerTrack->GetGTPattern();
	292
	293	//printf(">>> Event %d Number of Recconstructed tracks %d \n",iEvent, nrectracks);
	294
	295	// -------------------- tracks-------------
	296	muonData->SetTreeAddress("RT");
	297	muonData->GetRecTracks();
	298	recTracksArray = muonData->RecTracks();
	299
	300	Int_t nRecTracks = 0;
	301	if (recTracksArray)
	302	nRecTracks = (Int_t) recTracksArray->GetEntriesFast(); //
	303
	304	// loop over tracks
	305	for (Int_t iRecTracks = 0; iRecTracks < nRecTracks; iRecTracks++) {
	306
	307	// reading info from tracks
	308	recTrack = (AliMUONTrack*) recTracksArray->At(iRecTracks);
	309
889a0215	310	trackParam = (AliMUONTrackParam*) (recTrack->GetTrackParamAtHit())->First();
889a0215	311	trackParam->ExtrapToVertex(vertex[0],vertex[1],vertex[2]);
cac2eb58	312
	313	bendingSlope = trackParam->GetBendingSlope();
	314	nonBendingSlope = trackParam->GetNonBendingSlope();
	315	inverseBendingMomentum = trackParam->GetInverseBendingMomentum();
	316	xRec = trackParam->GetNonBendingCoor();
	317	yRec = trackParam->GetBendingCoor();
	318	zRec = trackParam->GetZ();
	319
	320	nTrackHits = recTrack->GetNTrackHits();
	321	fitFmin = recTrack->GetFitFMin();
	322	matchTrigger = recTrack->GetMatchTrigger();
	323	chi2MatchTrigger = recTrack->GetChi2MatchTrigger();
	324
	325	// setting data member of ESD MUON
30178c30	326	theESDTrack->SetInverseBendingMomentum(inverseBendingMomentum);
	327	theESDTrack->SetThetaX(TMath::ATan(nonBendingSlope));
	328	theESDTrack->SetThetaY(TMath::ATan(bendingSlope));
889a0215	329	theESDTrack->SetZ(zRec);
	330	theESDTrack->SetBendingCoor(yRec); // calculate vertex at ESD or Tracking level ?
	331	theESDTrack->SetNonBendingCoor(xRec);
30178c30	332	theESDTrack->SetChi2(fitFmin);
	333	theESDTrack->SetNHit(nTrackHits);
	334	theESDTrack->SetMatchTrigger(matchTrigger);
	335	theESDTrack->SetChi2MatchTrigger(chi2MatchTrigger);
cac2eb58	336
	337	// storing ESD MUON Track into ESD Event
	338	if (nRecTracks != 0)
30178c30	339	esd->AddMuonTrack(theESDTrack);
cac2eb58	340	} // end loop tracks
	341
	342	// add global trigger pattern
	343	if (nRecTracks != 0)
	344	esd->SetTrigger(trigPat);
	345
	346	// reset muondata
	347	muonData->ResetRecTracks();
	348	muonData->ResetRecTriggerTracks();
	349
	350	//} // end loop on event
	351	loader->UnloadTracks();
b002ac70	352	if (!header)
b002ac70	353	runLoader->UnloadKinematics();
30178c30	354	delete theESDTrack;
cac2eb58	355	delete muonData;
ce6a659c	356	delete particle;
b002ac70	357	// delete particle;
cac2eb58	358	}