MUON/AliMUONESDInterface.cxx

   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 #include "AliMUONESDInterface.h"
  19 #include "AliMUONTrack.h"
  20 #include "AliMUONVTrackStore.h"
  21 #include "AliMUONVCluster.h"
  22 #include "AliMUONVClusterStore.h"
  23 #include "AliMUONVDigit.h"
  24 #include "AliMUONVDigitStore.h"
  25 #include "AliMUON1DMapIterator.h"
  26 #include "AliMUON2DMapIterator.h"
  27 #include "AliMUONTrackParam.h"
  28 #include "AliMUONTrackExtrap.h"
  29 #include "AliMUONConstants.h"
  30
  31 #include "AliMpVSegmentation.h"
  32 #include "AliMpSegmentation.h"
  33 #include "AliMpIntPair.h"
  34 #include "AliMpPad.h"
  35
  36 #include "AliESDEvent.h"
  37 #include "AliESDMuonTrack.h"
  38 #include "AliESDMuonCluster.h"
  39 #include "AliESDMuonPad.h"
  40 #include "AliLog.h"
  41
  42 #include <TClass.h>
  43 #include <TIterator.h>
  44 #include <TMath.h>
  45 #include <TMatrixD.h>
  46 #include <Riostream.h>
  47
  48 //-----------------------------------------------------------------------------
  49 /// \class AliMUONESDInterface
  50 ///
  51 /// There are 2 way of using thid converter between MUON track/cluster/digit
  52 /// and ESDMuon track/cluster/pad:
  53 ///
  54 /// 1) using the static methods converting the objects one by one
  55 ///
  56 /// 2) loading a whole ESDEvent and using the getters and/or the iterators
  57 ///    to access the corresponding MUON objects
  58 ///
  59 /// \author Philippe Pillot
  60 //-----------------------------------------------------------------------------
  61
  62 /// \cond CLASSIMP
  63 ClassImp(AliMUONESDInterface)
  64 /// \endcond
  65
  66 TString AliMUONESDInterface::fgTrackStoreName = "AliMUONTrackStoreV1";
  67 TString AliMUONESDInterface::fgClusterStoreName = "AliMUONClusterStoreV2";
  68 TString AliMUONESDInterface::fgDigitStoreName = "AliMUONDigitStoreV2R";
  69
  70 //_____________________________________________________________________________
  71 AliMUONESDInterface::AliMUONESDInterface()
  72 : TObject(),
  73   fTracks(0x0),
  74   fDigits(0x0),
  75   fTrackMap(0x0),
  76   fClusterMap(0x0),
  77   fDigitMap(0x0)
  78 {
  79   /// Default constructor
  80 }
  81
  82 //_____________________________________________________________________________
  83 AliMUONESDInterface::~AliMUONESDInterface()
  84 {
  85   /// Destructor
  86   delete fTracks;
  87   delete fDigits;
  88   delete fTrackMap;
  89   delete fClusterMap;
  90   delete fDigitMap;
  91 }
  92
  93 //+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++//
  94 //                    methods to play with internal objects                    //
  95 //+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++//
  96
  97 //_____________________________________________________________________________
  98 void AliMUONESDInterface::Clear(Option_t*)
  99 {
 100   /// clear memory
 101   delete fTracks; fTracks = 0x0;
 102   delete fDigits; fDigits = 0x0;
 103   delete fTrackMap; fTrackMap = 0x0;
 104   delete fClusterMap; fClusterMap = 0x0;
 105   delete fDigitMap; fDigitMap = 0x0;
 106 }
 107
 108 //_____________________________________________________________________________
 109 void AliMUONESDInterface::Reset()
 110 {
 111   /// reset stores and maps
 112
 113   if (fTracks) fTracks->Clear("C");
 114   else fTracks = NewTrackStore();
 115
 116   if (fDigits) fDigits->Clear("C");
 117   else fDigits = NewDigitStore();
 118
 119   if (fTrackMap) fTrackMap->Clear();
 120   else fTrackMap = new AliMpExMap(kTRUE);
 121   fTrackMap->SetOwner(kFALSE);
 122
 123   if (fClusterMap) fClusterMap->Clear();
 124   else fClusterMap = new AliMpExMap(kTRUE);
 125   fClusterMap->SetOwner(kTRUE);
 126
 127   if (fDigitMap) fDigitMap->Clear("C");
 128   else fDigitMap = new TClonesArray("AliMpExMap",10);
 129 }
 130
 131 //_____________________________________________________________________________
 132 void AliMUONESDInterface::LoadEvent(AliESDEvent& esdEvent)
 133 {
 134   /// Extract MUON data from the given ESD event
 135
 136   // reset data members
 137   Reset();
 138
 139   // loop over ESD tracks and fill the stores
 140   Int_t nTracks = (Int_t) esdEvent.GetNumberOfMuonTracks();
 141   for (Int_t iTrack = 0; iTrack <  nTracks; iTrack++) {
 142
 143     // get ESD track
 144     AliESDMuonTrack* esdTrack = esdEvent.GetMuonTrack(iTrack);
 145
 146     // add it to track store
 147     AliMUONTrack* track = Add(*esdTrack, *fTracks);
 148
 149     // fill track map
 150     fTrackMap->Add(iTrack, track);
 151
 152     // prepare cluster map
 153     fClusterMap->Add(iTrack, new AliMpExMap(kTRUE));
 154     AliMpExMap* cMap = (AliMpExMap*) fClusterMap->GetObjectFast(iTrack);
 155     cMap->SetOwner(kFALSE);
 156
 157     // prepare digit maps
 158     AliMpExMap* dMaps = new((*fDigitMap)[iTrack]) AliMpExMap(kTRUE);
 159     dMaps->SetOwner(kTRUE);
 160
 161     // loop over ESD clusters
 162     Int_t nClusters = esdTrack->GetNClusters();
 163     for (Int_t iCluster = 0; iCluster <  nClusters; iCluster++) {
 164
 165       // get ESD cluster
 166       AliESDMuonCluster *esdCluster = (AliESDMuonCluster*) esdTrack->GetClusters().UncheckedAt(iCluster);
 167
 168       // get the corresponding MUON cluster
 169       AliMUONVCluster* cluster = FindClusterInTrack(*track, esdCluster->GetUniqueID());
 170
 171       // fill cluster map
 172       cMap->Add(cluster->GetUniqueID(), cluster);
 173
 174       // prepare digit map
 175       dMaps->Add(esdCluster->GetUniqueID(), new AliMpExMap(kTRUE));
 176       AliMpExMap* dMap = (AliMpExMap*) dMaps->GetObjectFast(iCluster);
 177       dMap->SetOwner(kFALSE);
 178
 179       // loop over ESD pads
 180       Int_t nPads = esdCluster->GetNPads();
 181       for (Int_t iPad = 0; iPad <  nPads; iPad++) {
 182
 183         // get ESD pad
 184         AliESDMuonPad *esdPad = (AliESDMuonPad*) esdCluster->GetPads().UncheckedAt(iPad);
 185
 186         // add it to digit store
 187         AliMUONVDigit* digit = Add(*esdPad, *fDigits);
 188
 189         // fill digit map
 190         if (digit) dMap->Add(esdPad->GetUniqueID(), digit);
 191         else dMap->Add(esdPad->GetUniqueID(), fDigits->FindObject(esdPad->GetUniqueID()));
 192
 193       } // end of loop over pads
 194
 195     } // end of loop over clusters
 196
 197   } // end of loop over tracks
 198
 199 }
 200
 201 //___________________________________________________________________________
 202 Int_t AliMUONESDInterface::GetNTracks() const
 203 {
 204   /// return the number of tracks
 205   return fTrackMap ? fTrackMap->GetSize() : 0;
 206 }
 207
 208 //___________________________________________________________________________
 209 Int_t AliMUONESDInterface::GetNClusters() const
 210 {
 211   /// return the number of clusters
 212   Int_t nClusters = 0;
 213   Int_t nTracks = GetNTracks();
 214   for (Int_t i = 0; i < nTracks; i++) nClusters += GetTrackFast(i)->GetNClusters();
 215   return nClusters;
 216 }
 217
 218 //___________________________________________________________________________
 219 Int_t AliMUONESDInterface::GetNClusters(Int_t iTrack) const
 220 {
 221   /// return the number of clusters in track "iTrack"
 222   AliMUONTrack* track = GetTrack(iTrack);
 223   return track ? track->GetNClusters() : 0;
 224 }
 225
 226 //___________________________________________________________________________
 227 Int_t AliMUONESDInterface::GetNDigits() const
 228 {
 229   /// return the number of digits
 230   return fDigits ? fDigits->GetSize() : 0;
 231 }
 232
 233 //___________________________________________________________________________
 234 Int_t AliMUONESDInterface::GetNDigits(Int_t iTrack) const
 235 {
 236   /// return the number of digits in all clusters of track "iTrack"
 237   Int_t nDigits = 0;
 238   Int_t nClusters = GetNClusters(iTrack);
 239   for (Int_t j = 0; j <  nClusters; j++) nDigits += GetClusterFast(iTrack,j)->GetNDigits();
 240   return nDigits;
 241 }
 242
 243 //___________________________________________________________________________
 244 Int_t AliMUONESDInterface::GetNDigits(Int_t iTrack, Int_t iCluster) const
 245 {
 246   /// return the number of digits in cluster numbered "iCluster" of track "iTrack"
 247   AliMUONVCluster* cluster = GetCluster(iTrack, iCluster);
 248   return cluster ? cluster->GetNDigits() : 0;
 249 }
 250
 251 //___________________________________________________________________________
 252 Int_t AliMUONESDInterface::GetNDigitsInCluster(UInt_t clusterId) const
 253 {
 254   /// return the number of digits in cluster "clusterId"
 255   AliMUONVCluster* cluster = FindCluster(clusterId);
 256   return cluster ? cluster->GetNDigits() : 0;
 257 }
 258
 259 //___________________________________________________________________________
 260 AliMUONTrack* AliMUONESDInterface::GetTrack(Int_t iTrack) const
 261 {
 262   /// return MUON track "iTrack" (0x0 if not found)
 263   AliMUONTrack* track = fTrackMap ? (AliMUONTrack*) fTrackMap->GetObject(iTrack) : 0x0;
 264   if (!track) AliWarning(Form("track %d does not exist",iTrack));
 265   return track;
 266 }
 267
 268 //___________________________________________________________________________
 269 AliMUONVCluster* AliMUONESDInterface::GetCluster(Int_t iTrack, Int_t iCluster) const
 270 {
 271   /// return MUON cluster numbered "iCluster" in track "iTrack" (0x0 if not found)
 272   AliMpExMap* cMap = fClusterMap ? (AliMpExMap*) fClusterMap->GetObject(iTrack) : 0x0;
 273   AliMUONVCluster* cluster = cMap ? (AliMUONVCluster*) cMap->GetObject(iCluster) : 0x0;
 274   if (!cluster) AliWarning(Form("cluster #%d in track %d does not exist",iCluster,iTrack));
 275   return cluster;
 276 }
 277
 278 //___________________________________________________________________________
 279 AliMUONVDigit* AliMUONESDInterface::GetDigit(Int_t iTrack, Int_t iCluster, Int_t iDigit) const
 280 {
 281   /// return MUON digit numbered "iDigit" in cluster numbered "iCluster" of track "iTrack" (0x0 if not found)
 282   AliMpExMap* dMaps = fDigitMap ? (AliMpExMap*) fDigitMap->At(iTrack) : 0x0;
 283   AliMpExMap* dMap = dMaps ? (AliMpExMap*) dMaps->GetObject(iCluster) : 0x0;
 284   AliMUONVDigit* digit = dMap ? (AliMUONVDigit*) dMap->GetObject(iDigit) : 0x0;
 285   if (!digit) AliWarning(Form("digit #%d in cluster #%d of track %d does not exist",iDigit,iCluster,iTrack));
 286   return digit;
 287 }
 288
 289 //___________________________________________________________________________
 290 AliMUONVCluster* AliMUONESDInterface::FindCluster(UInt_t clusterId) const
 291 {
 292   /// return cluster "clusterId" (0x0 if not found)
 293   AliMUONVCluster* cluster = 0x0;
 294
 295   Int_t nTracks = GetNTracks();
 296   for (Int_t i = 0; i < nTracks; i++) {
 297
 298     cluster = (AliMUONVCluster*) ((AliMpExMap*) fClusterMap->GetObjectFast(i))->GetValue(clusterId);
 299     if (cluster) break;
 300
 301   }
 302
 303   if (!cluster) AliWarning(Form("cluster %d does not exist",clusterId));
 304   return cluster;
 305 }
 306
 307 //___________________________________________________________________________
 308 AliMUONVDigit* AliMUONESDInterface::FindDigit(UInt_t digitId) const
 309 {
 310   /// return digit "digitId" (0x0 if not found)
 311   AliMUONVDigit *digit = fDigits ? fDigits->FindObject(digitId) : 0x0;
 312   if (!digit) AliWarning(Form("digit %d does not exist",digitId));
 313   return digit;
 314 }
 315
 316 //___________________________________________________________________________
 317 TIterator* AliMUONESDInterface::CreateTrackIterator() const
 318 {
 319   /// return iterator over all tracks
 320   return fTracks ? fTracks->CreateIterator() : 0x0;
 321 }
 322
 323 //___________________________________________________________________________
 324 TIterator* AliMUONESDInterface::CreateClusterIterator() const
 325 {
 326   /// return iterator over all clusters
 327   return fClusterMap ? new AliMUON2DMapIterator(*fClusterMap) : 0x0;
 328 }
 329
 330 //___________________________________________________________________________
 331 TIterator* AliMUONESDInterface::CreateClusterIterator(Int_t iTrack) const
 332 {
 333   /// return iterator over clusters of track "iTrack"
 334   AliMpExMap* cMap = fClusterMap ? (AliMpExMap*) fClusterMap->GetObject(iTrack) : 0x0;
 335   return cMap ? new AliMUON1DMapIterator(*cMap) : 0x0;
 336 }
 337
 338 //___________________________________________________________________________
 339 TIterator* AliMUONESDInterface::CreateDigitIterator() const
 340 {
 341   /// return iterator over all digits
 342   return fDigits ? fDigits->CreateIterator() : 0x0;
 343 }
 344
 345 //___________________________________________________________________________
 346 TIterator* AliMUONESDInterface::CreateDigitIterator(Int_t iTrack) const
 347 {
 348   /// return iterator over all digits of track "iTrack"
 349   AliMpExMap* dMaps = fDigitMap ? (AliMpExMap*) fDigitMap->At(iTrack) : 0x0;
 350   return dMaps ? new AliMUON2DMapIterator(*dMaps) : 0x0;
 351 }
 352
 353 //___________________________________________________________________________
 354 TIterator* AliMUONESDInterface::CreateDigitIterator(Int_t iTrack, Int_t iCluster) const
 355 {
 356   /// return iterator over digits of cluster numbered "iCluster" in track "iTrack"
 357   AliMpExMap* dMaps = fDigitMap ? (AliMpExMap*) fDigitMap->At(iTrack) : 0x0;
 358   AliMpExMap* dMap = dMaps ? (AliMpExMap*) dMaps->GetObject(iCluster) : 0x0;
 359   return dMap ? new AliMUON1DMapIterator(*dMap) : 0x0;
 360 }
 361
 362 //___________________________________________________________________________
 363 TIterator* AliMUONESDInterface::CreateDigitIteratorInCluster(UInt_t clusterId) const
 364 {
 365   /// return iterator over digits of cluster "clusterId"
 366   AliMpExMap* dMap = 0x0;
 367
 368   Int_t nTracks = GetNTracks();
 369   for (Int_t i = 0; i < nTracks; i++) {
 370
 371     dMap = (AliMpExMap*) ((AliMpExMap*) fDigitMap->UncheckedAt(i))->GetValue(clusterId);
 372     if (dMap) break;
 373
 374   }
 375
 376   return dMap ? new AliMUON1DMapIterator(*dMap) : 0x0;
 377 }
 378
 379 //___________________________________________________________________________
 380 AliMUONVCluster* AliMUONESDInterface::FindClusterInTrack(const AliMUONTrack& track, UInt_t clusterId) const
 381 {
 382   /// find the cluster with the given Id into the track
 383
 384   Int_t nClusters = track.GetNClusters();
 385   for (Int_t iCluster = 0; iCluster < nClusters; iCluster++) {
 386
 387     AliMUONVCluster* cluster = ((AliMUONTrackParam*) track.GetTrackParamAtCluster()->UncheckedAt(iCluster))->GetClusterPtr();
 388     if (cluster->GetUniqueID() == clusterId) return cluster;
 389
 390   }
 391
 392   return 0x0;
 393 }
 394
 395 //+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++//
 396 //                                static methods                               //
 397 //+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++//
 398
 399 //_____________________________________________________________________________
 400 AliMUONVTrackStore* AliMUONESDInterface::NewTrackStore()
 401 {
 402   /// Create an empty track store of type fgTrackStoreName
 403   TClass* classPtr = TClass::GetClass(fgTrackStoreName);
 404   if (!classPtr || !classPtr->InheritsFrom("AliMUONVTrackStore")) {
 405     cout<<"E-AliMUONESDInterface::NewTrackStore: Unable to create store of type "<<fgTrackStoreName.Data()<<endl;
 406     return 0x0;
 407   }
 408   return reinterpret_cast<AliMUONVTrackStore*>(classPtr->New());
 409 }
 410
 411 //_____________________________________________________________________________
 412 AliMUONVClusterStore* AliMUONESDInterface::NewClusterStore()
 413 {
 414   /// Create an empty cluster store of type fgClusterStoreName
 415   TClass* classPtr = TClass::GetClass(fgClusterStoreName);
 416   if (!classPtr || !classPtr->InheritsFrom("AliMUONVClusterStore")) {
 417     cout<<"E-AliMUONESDInterface::NewClusterStore: Unable to create store of type "<<fgClusterStoreName.Data()<<endl;
 418     return 0x0;
 419   }
 420   return reinterpret_cast<AliMUONVClusterStore*>(classPtr->New());
 421 }
 422
 423 //_____________________________________________________________________________
 424 AliMUONVDigitStore* AliMUONESDInterface::NewDigitStore()
 425 {
 426   /// Create an empty digit store of type fgDigitStoreName
 427   TClass* classPtr = TClass::GetClass(fgDigitStoreName);
 428   if (!classPtr || !classPtr->InheritsFrom("AliMUONVDigitStore")) {
 429     cout<<"E-AliMUONESDInterface::NewDigitStore: Unable to create store of type "<<fgDigitStoreName.Data()<<endl;
 430     return 0x0;
 431   }
 432   return reinterpret_cast<AliMUONVDigitStore*>(classPtr->New());
 433 }
 434
 435 //_________________________________________________________________________
 436 void AliMUONESDInterface::GetParamAtVertex(const AliESDMuonTrack& esdTrack, AliMUONTrackParam& trackParam)
 437 {
 438   /// Get parameters at vertex from ESDMuon track
 439   trackParam.SetZ(esdTrack.GetZ());
 440   trackParam.SetNonBendingCoor(esdTrack.GetNonBendingCoor());
 441   trackParam.SetNonBendingSlope(TMath::Tan(esdTrack.GetThetaX()));
 442   trackParam.SetBendingCoor(esdTrack.GetBendingCoor());
 443   trackParam.SetBendingSlope(TMath::Tan(esdTrack.GetThetaY()));
 444   trackParam.SetInverseBendingMomentum(esdTrack.GetInverseBendingMomentum());
 445 }
 446
 447 //_________________________________________________________________________
 448 void AliMUONESDInterface::GetParamAtDCA(const AliESDMuonTrack& esdTrack, AliMUONTrackParam& trackParam)
 449 {
 450   /// Get parameters at DCA from ESDMuon track
 451   trackParam.SetZ(esdTrack.GetZ());
 452   trackParam.SetNonBendingCoor(esdTrack.GetNonBendingCoorAtDCA());
 453   trackParam.SetNonBendingSlope(TMath::Tan(esdTrack.GetThetaXAtDCA()));
 454   trackParam.SetBendingCoor(esdTrack.GetBendingCoorAtDCA());
 455   trackParam.SetBendingSlope(TMath::Tan(esdTrack.GetThetaYAtDCA()));
 456   trackParam.SetInverseBendingMomentum(esdTrack.GetInverseBendingMomentumAtDCA());
 457 }
 458
 459 //_________________________________________________________________________
 460 void AliMUONESDInterface::GetParamAtFirstCluster(const AliESDMuonTrack& esdTrack, AliMUONTrackParam& trackParam)
 461 {
 462   /// Get parameters at first cluster from ESDMuon track
 463   trackParam.SetZ(esdTrack.GetZUncorrected());
 464   trackParam.SetNonBendingCoor(esdTrack.GetNonBendingCoorUncorrected());
 465   trackParam.SetNonBendingSlope(TMath::Tan(esdTrack.GetThetaXUncorrected()));
 466   trackParam.SetBendingCoor(esdTrack.GetBendingCoorUncorrected());
 467   trackParam.SetBendingSlope(TMath::Tan(esdTrack.GetThetaYUncorrected()));
 468   trackParam.SetInverseBendingMomentum(esdTrack.GetInverseBendingMomentumUncorrected());
 469 }
 470
 471 //_________________________________________________________________________
 472 void AliMUONESDInterface::GetParamCov(const AliESDMuonTrack& esdTrack, AliMUONTrackParam& trackParam)
 473 {
 474   /// Get parameters covariances from ESD track
 475
 476   // get ESD covariance matrix
 477   TMatrixD covariances(5,5);
 478   esdTrack.GetCovariances(covariances);
 479
 480   // compute Jacobian to change the coordinate system
 481   // from (X,thetaX,Y,thetaY,c/pYZ) to (X,slopeX,Y,slopeY,c/pYZ)
 482   Double_t cosThetaX = TMath::Cos(esdTrack.GetThetaXUncorrected());
 483   Double_t cosThetaY = TMath::Cos(esdTrack.GetThetaYUncorrected());
 484   TMatrixD jacob(5,5);
 485   jacob.Zero();
 486   jacob(0,0) = 1.;
 487   jacob(1,1) = 1. / cosThetaX / cosThetaX;
 488   jacob(2,2) = 1.;
 489   jacob(3,3) = 1. / cosThetaY / cosThetaY;
 490   jacob(4,4) = 1.;
 491
 492   // compute covariance matrix in ESD coordinate system
 493   TMatrixD tmp(covariances,TMatrixD::kMultTranspose,jacob);
 494   trackParam.SetCovariances(TMatrixD(jacob,TMatrixD::kMult,tmp));
 495
 496 }
 497
 498 //_________________________________________________________________________
 499 void AliMUONESDInterface::SetParamAtVertex(const AliMUONTrackParam& trackParam, AliESDMuonTrack& esdTrack)
 500 {
 501   /// Set parameters in ESD track
 502   esdTrack.SetZ(trackParam.GetZ());
 503   esdTrack.SetNonBendingCoor(trackParam.GetNonBendingCoor());
 504   esdTrack.SetThetaX(TMath::ATan(trackParam.GetNonBendingSlope()));
 505   esdTrack.SetBendingCoor(trackParam.GetBendingCoor());
 506   esdTrack.SetThetaY(TMath::ATan(trackParam.GetBendingSlope()));
 507   esdTrack.SetInverseBendingMomentum(trackParam.GetInverseBendingMomentum());
 508 }
 509
 510 //_________________________________________________________________________
 511 void AliMUONESDInterface::SetParamAtDCA(const AliMUONTrackParam& trackParam, AliESDMuonTrack& esdTrack)
 512 {
 513   /// Set parameters in ESD track
 514   esdTrack.SetNonBendingCoorAtDCA(trackParam.GetNonBendingCoor());
 515   esdTrack.SetThetaXAtDCA(TMath::ATan(trackParam.GetNonBendingSlope()));
 516   esdTrack.SetBendingCoorAtDCA(trackParam.GetBendingCoor());
 517   esdTrack.SetThetaYAtDCA(TMath::ATan(trackParam.GetBendingSlope()));
 518   esdTrack.SetInverseBendingMomentumAtDCA(trackParam.GetInverseBendingMomentum());
 519 }
 520
 521 //_________________________________________________________________________
 522 void AliMUONESDInterface::SetParamAtFirstCluster(const AliMUONTrackParam& trackParam, AliESDMuonTrack& esdTrack)
 523 {
 524   /// Set parameters in ESD track
 525   esdTrack.SetZUncorrected(trackParam.GetZ());
 526   esdTrack.SetNonBendingCoorUncorrected(trackParam.GetNonBendingCoor());
 527   esdTrack.SetThetaXUncorrected(TMath::ATan(trackParam.GetNonBendingSlope()));
 528   esdTrack.SetBendingCoorUncorrected(trackParam.GetBendingCoor());
 529   esdTrack.SetThetaYUncorrected(TMath::ATan(trackParam.GetBendingSlope()));
 530   esdTrack.SetInverseBendingMomentumUncorrected(trackParam.GetInverseBendingMomentum());
 531 }
 532
 533 //_________________________________________________________________________
 534 void AliMUONESDInterface::SetParamCov(const AliMUONTrackParam& trackParam, AliESDMuonTrack& esdTrack)
 535 {
 536   /// Set parameters covariances in ESD track
 537
 538   // set null matrix if covariances does not exist
 539   if (!trackParam.CovariancesExist()) {
 540     TMatrixD tmp(5,5);
 541     tmp.Zero();
 542     esdTrack.SetCovariances(tmp);
 543     return;
 544   }
 545
 546   // compute Jacobian to change the coordinate system
 547   // from (X,slopeX,Y,slopeY,c/pYZ) to (X,thetaX,Y,thetaY,c/pYZ)
 548   Double_t cosThetaX = TMath::Cos(TMath::ATan(trackParam.GetNonBendingSlope()));
 549   Double_t cosThetaY = TMath::Cos(TMath::ATan(trackParam.GetBendingSlope()));
 550   TMatrixD jacob(5,5);
 551   jacob.Zero();
 552   jacob(0,0) = 1.;
 553   jacob(1,1) = cosThetaX * cosThetaX;
 554   jacob(2,2) = 1.;
 555   jacob(3,3) = cosThetaY * cosThetaY;
 556   jacob(4,4) = 1.;
 557
 558   // compute covariance matrix in ESD coordinate system
 559   TMatrixD tmp(trackParam.GetCovariances(),TMatrixD::kMultTranspose,jacob);
 560   esdTrack.SetCovariances(TMatrixD(jacob,TMatrixD::kMult,tmp));
 561
 562 }
 563
 564 //_____________________________________________________________________________
 565 void AliMUONESDInterface::ESDToMUON(const AliESDMuonTrack& esdTrack, AliMUONTrack& track)
 566 {
 567   /// Transfert data from ESDMuon track to MUON track
 568
 569   track.Clear("C");
 570
 571   // global info
 572   track.SetUniqueID(esdTrack.GetUniqueID());
 573   track.FitWithVertex(kFALSE);
 574   track.FitWithMCS(kFALSE);
 575   track.SetImproved(kFALSE);
 576   track.SetVertexErrXY2(0.,0.);
 577   track.SetGlobalChi2(esdTrack.GetChi2());
 578   track.SetMatchTrigger(esdTrack.GetMatchTrigger());
 579   track.SetLoTrgNum(-1);
 580   track.SetChi2MatchTrigger(esdTrack.GetChi2MatchTrigger());
 581   track.SetTrackID(0);
 582   track.SetHitsPatternInTrigCh(esdTrack.GetHitsPatternInTrigCh());
 583   track.SetLocalTrigger(esdTrack.LoCircuit(), esdTrack.LoStripX(), esdTrack.LoStripY(),
 584                         esdTrack.LoDev(), esdTrack.LoLpt(), esdTrack.LoHpt());
 585
 586   // track parameters at vertex
 587   AliMUONTrackParam paramAtVertex;
 588   GetParamAtVertex(esdTrack, paramAtVertex);
 589   track.SetTrackParamAtVertex(&paramAtVertex);
 590
 591   // track parameters at first cluster
 592   AliMUONTrackParam param;
 593   GetParamAtFirstCluster(esdTrack, param);
 594   GetParamCov(esdTrack, param);
 595
 596   // create empty cluster
 597   AliMUONVClusterStore* cStore = NewClusterStore();
 598   if (!cStore) return;
 599   AliMUONVCluster* cluster = cStore->CreateCluster(0,0,0);
 600
 601   // fill TrackParamAtCluster with track parameters at each cluster if available
 602   // or with only track parameters at first (fake) cluster if not
 603   if(esdTrack.ClustersStored()) {
 604
 605     // loop over ESD clusters
 606     AliESDMuonCluster *esdCluster = (AliESDMuonCluster*) esdTrack.GetClusters().First();
 607     while (esdCluster) {
 608
 609       // copy cluster information
 610       ESDToMUON(*esdCluster, *cluster);
 611
 612       // only set the Z parameter to avoid error in the AddTrackParamAtCluster(...) method
 613       param.SetZ(cluster->GetZ());
 614
 615       // add common track parameters at current cluster
 616       track.AddTrackParamAtCluster(param, *cluster, kTRUE);
 617
 618       esdCluster = (AliESDMuonCluster*) esdTrack.GetClusters().After(esdCluster);
 619     }
 620
 621     // recompute parameters at first cluster in case of those stored
 622     // in ESD are not related to the most upstream cluster
 623     AliMUONTrackParam *firstTrackParam = (AliMUONTrackParam*) track.GetTrackParamAtCluster()->First();
 624     firstTrackParam->SetZ(esdTrack.GetZUncorrected()); // reset the z to the one stored in ESD
 625     AliMUONTrackExtrap::ExtrapToZCov(firstTrackParam,firstTrackParam->GetClusterPtr()->GetZ());
 626
 627     // Compute track parameters and covariances at each cluster from those at the first one
 628     track.UpdateCovTrackParamAtCluster();
 629
 630   } else {
 631
 632     // get number of the first hit chamber (according to the MUONClusterMap if not empty)
 633     Int_t firstCh = 0;
 634     if (esdTrack.GetMuonClusterMap() != 0) while (!esdTrack.IsInMuonClusterMap(firstCh)) firstCh++;
 635     else firstCh = AliMUONConstants::ChamberNumber(param.GetZ());
 636
 637     // produce fake cluster at this chamber
 638     cluster->SetUniqueID(AliMUONVCluster::BuildUniqueID(firstCh, 0, 0));
 639     cluster->SetXYZ(param.GetNonBendingCoor(), param.GetBendingCoor(), param.GetZ());
 640     cluster->SetErrXY(0., 0.);
 641
 642     // add track parameters at first (fake) cluster
 643     track.AddTrackParamAtCluster(param, *cluster, kTRUE);
 644
 645   }
 646
 647   delete cluster;
 648   delete cStore;
 649
 650 }
 651
 652 //_____________________________________________________________________________
 653 void AliMUONESDInterface::ESDToMUON(const AliESDMuonCluster& esdCluster, AliMUONVCluster& cluster)
 654 {
 655   /// Transfert data from ESDMuon cluster to MUON cluster
 656
 657   cluster.Clear("C");
 658
 659   cluster.SetUniqueID(esdCluster.GetUniqueID());
 660   cluster.SetXYZ(esdCluster.GetX(), esdCluster.GetY(), esdCluster.GetZ());
 661   cluster.SetErrXY(esdCluster.GetErrX(),esdCluster.GetErrY());
 662   cluster.SetCharge(esdCluster.GetCharge());
 663   cluster.SetChi2(esdCluster.GetChi2());
 664
 665   if (esdCluster.PadsStored()) {
 666     Int_t nPads = esdCluster.GetNPads();
 667     for (Int_t iPad = 0; iPad < nPads; iPad++)
 668       cluster.AddDigitId(((AliESDMuonPad*)esdCluster.GetPads().UncheckedAt(iPad))->GetUniqueID());
 669   }
 670
 671 }
 672
 673 //___________________________________________________________________________
 674 void AliMUONESDInterface::ESDToMUON(const AliESDMuonPad& esdPad, AliMUONVDigit& digit)
 675 {
 676   /// Transfert data from ESDMuon pad to MUON digit
 677
 678   const AliMpVSegmentation* seg = AliMpSegmentation::Instance()->GetMpSegmentationByElectronics(esdPad.GetDetElemId(), esdPad.GetManuId());
 679   AliMpPad pad = seg->PadByLocation(AliMpIntPair(esdPad.GetManuId(), esdPad.GetManuChannel()), kFALSE);
 680
 681   digit.Saturated(kFALSE);
 682   digit.Used(kFALSE);
 683   digit.Calibrated(kTRUE);
 684   digit.SetUniqueID(esdPad.GetUniqueID());
 685   digit.SetCharge(esdPad.GetCharge());
 686   digit.SetADC(esdPad.GetADC());
 687   digit.SetPadXY(pad.GetIndices().GetFirst(), pad.GetIndices().GetSecond());
 688
 689 }
 690
 691 //_____________________________________________________________________________
 692 void AliMUONESDInterface::MUONToESD(const AliMUONTrack& track, AliESDMuonTrack& esdTrack, const Double_t vertex[3], const AliMUONVDigitStore* digits)
 693 {
 694   /// Transfert data from MUON track to ESDMuon track
 695   /// Incorporate the ESDPads if the digits are provided
 696
 697   esdTrack.Clear("C");
 698
 699   // set param at first cluster
 700   AliMUONTrackParam* trackParam = static_cast<AliMUONTrackParam*>((track.GetTrackParamAtCluster())->First());
 701   SetParamAtFirstCluster(*trackParam, esdTrack);
 702   SetParamCov(*trackParam, esdTrack);
 703
 704   // set param at vertex
 705   AliMUONTrackParam trackParamAtVtx(*trackParam);
 706   AliMUONTrackExtrap::ExtrapToVertex(&trackParamAtVtx, vertex[0], vertex[1], vertex[2], 0., 0.);
 707   SetParamAtVertex(trackParamAtVtx, esdTrack);
 708
 709   // set param at Distance of Closest Approach
 710   AliMUONTrackParam trackParamAtDCA(*trackParam);
 711   AliMUONTrackExtrap::ExtrapToVertexWithoutBranson(&trackParamAtDCA, vertex[2]);
 712   SetParamAtDCA(trackParamAtDCA, esdTrack);
 713
 714   // set global info
 715   esdTrack.SetUniqueID(track.GetUniqueID());
 716   esdTrack.SetChi2(track.GetGlobalChi2());
 717   esdTrack.SetNHit(track.GetNClusters());
 718   esdTrack.SetLocalTrigger(track.GetLocalTrigger());
 719   esdTrack.SetChi2MatchTrigger(track.GetChi2MatchTrigger());
 720   esdTrack.SetHitsPatternInTrigCh(track.GetHitsPatternInTrigCh());
 721
 722   // set muon cluster info
 723   AliESDMuonCluster esdCluster;
 724   esdTrack.SetMuonClusterMap(0);
 725   while (trackParam) {
 726     MUONToESD(*(trackParam->GetClusterPtr()), esdCluster, digits);
 727     esdTrack.AddCluster(esdCluster);
 728     esdTrack.AddInMuonClusterMap(esdCluster.GetChamberId());
 729     trackParam = static_cast<AliMUONTrackParam*>(track.GetTrackParamAtCluster()->After(trackParam));
 730   }
 731
 732 }
 733
 734 //_____________________________________________________________________________
 735 void AliMUONESDInterface::MUONToESD(const AliMUONVCluster& cluster, AliESDMuonCluster& esdCluster, const AliMUONVDigitStore* digits)
 736 {
 737   /// Transfert data from MUON cluster to ESDMuon cluster
 738   /// Incorporate the ESDPads if the digits are provided
 739
 740   esdCluster.Clear("C");
 741
 742   esdCluster.SetUniqueID(cluster.GetUniqueID());
 743   esdCluster.SetXYZ(cluster.GetX(), cluster.GetY(), cluster.GetZ());
 744   esdCluster.SetErrXY(cluster.GetErrX(), cluster.GetErrY());
 745
 746   if (digits) { // transfert all data...
 747
 748     esdCluster.SetCharge(cluster.GetCharge());
 749     esdCluster.SetChi2(cluster.GetChi2());
 750     AliESDMuonPad esdPad;
 751     for (Int_t i=0; i<cluster.GetNDigits(); i++) {
 752       AliMUONVDigit* digit = digits->FindObject(cluster.GetDigitId(i));
 753       if (!digit) {
 754         cout<<"E-AliMUONESDInterface::MUONToESD: digit "<<cluster.GetDigitId(i)<<" not found"<<endl;
 755         continue;
 756       }
 757       MUONToESD(*digit, esdPad);
 758       esdCluster.AddPad(esdPad);
 759     }
 760
 761   } else { // ...or not
 762
 763     esdCluster.SetCharge(0.);
 764     esdCluster.SetChi2(0.);
 765
 766   }
 767
 768 }
 769
 770 //_____________________________________________________________________________
 771 void AliMUONESDInterface::MUONToESD(const AliMUONVDigit& digit, AliESDMuonPad& esdPad)
 772 {
 773   /// Transfert data from MUON digit to ESDMuon pad
 774   esdPad.SetUniqueID(digit.GetUniqueID());
 775   esdPad.SetADC(digit.ADC());
 776   esdPad.SetCharge(digit.Charge());
 777 }
 778
 779 //___________________________________________________________________________
 780 AliMUONTrack* AliMUONESDInterface::Add(const AliESDMuonTrack& esdTrack, AliMUONVTrackStore& trackStore)
 781 {
 782   /// Create MUON track from ESDMuon track and add it to the store
 783   /// return a pointer to the track into the store
 784   AliMUONTrack* track = trackStore.Add(AliMUONTrack());
 785   ESDToMUON(esdTrack, *track);
 786   return track;
 787 }
 788
 789 //___________________________________________________________________________
 790 AliMUONVCluster* AliMUONESDInterface::Add(const AliESDMuonCluster& esdCluster, AliMUONVClusterStore& clusterStore)
 791 {
 792   /// Create MUON cluster from ESDMuon cluster and add it to the store
 793   /// return a pointer to the cluster into the store (0x0 if the cluster already exist)
 794   AliMUONVCluster* cluster = clusterStore.Add(esdCluster.GetChamberId(), esdCluster.GetDetElemId(), esdCluster.GetClusterIndex());
 795   if (cluster) ESDToMUON(esdCluster, *cluster);
 796   return cluster;
 797 }
 798
 799 //___________________________________________________________________________
 800 AliMUONVDigit* AliMUONESDInterface::Add(const AliESDMuonPad& esdPad, AliMUONVDigitStore& digitStore)
 801 {
 802   /// Create MUON digit from ESDMuon digit and add it to the store
 803   /// return a pointer to the digit into the store (0x0 if the digit already exist)
 804   AliMUONVDigit* digit = digitStore.Add(esdPad.GetDetElemId(), esdPad.GetManuId(), esdPad.GetManuChannel(), esdPad.GetCathode(), AliMUONVDigitStore::kDeny);
 805   if (digit) ESDToMUON(esdPad, *digit);
 806   return digit;
 807 }
 808