Global transformation correction (ivana)

[u/mrichter/AliRoot.git] / MUON / MUONTracker.C

diff --git a/MUON/MUONTracker.C b/MUON/MUONTracker.C
index 4d0214230e3d3a709108cedfd5d833255ca07645..ebfd8db9a832879e1cd1f1e4dbfcb70502f8a968 100644 (file)
--- a/MUON/MUONTracker.C
+++ b/MUON/MUONTracker.C
@@ -1,18 +1,33 @@
+/**************************************************************************
+ * 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.                  *
+ **************************************************************************/
+
 // Macro MUONTracker.C (TO BE COMPILED)
 // for testing the C++ reconstruction code
 // Output is using aliroot standard output MUON.Tracks.root
 // The output is a TClonesArray of AliMUONTracks.
-#include <TClonesArray.h>
+// Allow to make track reconstruction directly from AliTrackReference hits 
+// recorded in TrackRefs.root file.
 
+#if !defined(__CINT__) || defined(__MAKECINT__)
 #include "AliRun.h"
 #include "AliMUON.h"
 #include "AliMUONData.h"
 #include "AliMUONEventReconstructor.h"
-#include "AliMUONTrack.h"
-#include "AliMUONTrackHit.h"
-#include "AliMUONTrackParam.h"
+#endif
 
-void MUONTracker (Text_t *FileName = "galice.root", Int_t FirstEvent = 0, Int_t LastEvent = 9999)
+void MUONTracker (Int_t FirstEvent = 0, Int_t LastEvent = 9999, Text_t *FileName = "galice.root")
 {
   //
   cout << "MUONTracker" << endl;
@@ -21,62 +36,78 @@ void MUONTracker (Text_t *FileName = "galice.root", Int_t FirstEvent = 0, Int_t
   cout << "FileName ``" << FileName << "''" << endl;
   
   // Creating Run Loader and openning file containing Hits, Digits and RecPoints
-  AliRunLoader * RunLoader = AliRunLoader::Open(FileName,"Event","UPDATE");
+  AliRunLoader * RunLoader = AliRunLoader::Open(FileName,"MUONLoader","UPDATE");
   if (RunLoader ==0x0) {
     printf(">>> Error : Error Opening %s file \n",FileName);
     return;
   }
   // Loading AliRun master
-  RunLoader->LoadgAlice();
+  if (RunLoader->GetAliRun() == 0x0) RunLoader->LoadgAlice();
   gAlice = RunLoader->GetAliRun();
-  RunLoader->LoadKinematics("READ");
+  RunLoader->LoadTrackRefs("READ");
   
   // Loading MUON subsystem
-  AliMUON * MUON = (AliMUON *) gAlice->GetDetector("MUON");
   AliLoader * MUONLoader = RunLoader->GetLoader("MUONLoader");
-  MUONLoader->LoadHits("READ");
-  MUONLoader->LoadRecPoints("READ");
-  AliMUONData * muondata = MUON->GetMUONData();
-  muondata->SetLoader(MUONLoader);
+ 
+  MUONLoader->LoadTracks("UPDATE");
   
-  Int_t ievent, nevents;
+  Int_t nevents;
   nevents = RunLoader->GetNumberOfEvents();
   
-  AliMUONEventReconstructor *Reco = new AliMUONEventReconstructor();
+  AliMUONEventReconstructor* Reco = new AliMUONEventReconstructor(MUONLoader);
+  AliMUONData* muondata = Reco->GetMUONData();
 
+  // Testing if Tracker has already been done
+  RunLoader->GetEvent(0);
+  if (MUONLoader->TreeT()) {
+    if (muondata->IsTrackBranchesInTree()) {
+      MUONLoader->UnloadTracks();
+      MUONLoader->LoadTracks("RECREATE");
+      printf("Recreating tracks files\n");
+    }
+  }
+  
   // The right place for changing AliMUONEventReconstructor parameters
   // with respect to the default ones
   //   Reco->SetMaxSigma2Distance(100.0);
   //   Reco->SetPrintLevel(20);
-  Reco->SetPrintLevel(1);
+  Reco->SetPrintLevel(0);
   //   Reco->SetBendingResolution(0.0);
   //   Reco->SetNonBendingResolution(0.0);
-  cout << "AliMUONEventReconstructor: actual parameters" << endl;
-  Reco->Dump();
-  //   gObjectTable->Print();
+  Reco->SetRecTrackRefHits(1); // 1: reconst. from track ref. hits 0: from clusters
+
+  if (Reco->GetRecTrackRefHits() == 0)
+    MUONLoader->LoadRecPoints("READ");
 
-  if  (LastEvent>nevents) LastEvent=nevents;
+  if  (LastEvent > nevents-1) LastEvent=nevents-1;
+  
   // Loop over events
-  for (Int_t event = FirstEvent; event < LastEvent; event++) {
+  for (Int_t event = FirstEvent; event <= LastEvent; event++) {
     cout << "Event: " << event << endl;
     RunLoader->GetEvent(event);   
-    muondata->SetTreeAddress("RC");
-    if (MUONLoader->TreeT() == 0x0) MUONLoader->MakeTree("T");
+    // Test if trigger track has already been done before
+    if (MUONLoader->TreeT() == 0x0) {  
+      MUONLoader->MakeTracksContainer();
+    }      else {
+      if (muondata->IsTrackBranchesInTree()){ // Test if track has already been done before
+       if (event==FirstEvent) MUONLoader->UnloadTracks();
+       MUONLoader->MakeTracksContainer();  // Redoing Tracking
+       Info("TrackContainer","Recreating TrackContainer and deleting previous ones");
+      }
+    }
+    
     muondata->MakeBranch("RT");
     muondata->SetTreeAddress("RT");
     Reco->EventReconstruct();
-    // Dump current event
-    Reco->EventDump();
-
-    // Duplicating rectrack data in muondata for output
-    for(Int_t i=0; i<Reco->GetNRecTracks(); i++) {
-      AliMUONTrack * track = (AliMUONTrack*) Reco->GetRecTracksPtr()->At(i);
-      muondata->AddRecTrack(*track);
-    }
-
+    
     muondata->Fill("RT");
-    MUONLoader->WriteTracks("OVERWRITE");
+    MUONLoader->WriteTracks("OVERWRITE");  
     muondata->ResetRecTracks();
+    if (Reco->GetRecTrackRefHits() == 0)
+      muondata->ResetRawClusters();
   } // Event loop
+  
   MUONLoader->UnloadRecPoints();
+  MUONLoader->UnloadTracks();
+  RunLoader->UnloadTrackRefs();
 }