- Fix problem with number of events to treat;

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

diff --git a/MUON/MUONAlignment.C b/MUON/MUONAlignment.C
index 827e020ec8f3baa47a53ff4e5cf9fba98262880e..c48266139c78e056b27b05e48b944066fc0ea4d4 100644 (file)
--- a/MUON/MUONAlignment.C
+++ b/MUON/MUONAlignment.C
@@ -34,9 +34,26 @@
 // Authors: B. Becker and J. Castillo
 // ---
 
-void MUONAlignment(Int_t nEvents = 100000, TString fileName = "galice.root", TString fileList = "")
+void MUONAlignment(Int_t nEvents = 100000, char* geoFilename = "geometry.root", TString fileName = "galice.root", TString fileList = "")
 {
  
+  // Import TGeo geometry (needed by AliMUONTrackExtrap::ExtrapToVertex)
+  if (!gGeoManager) {
+    TGeoManager::Import(geoFilename);
+    if (!gGeoManager) {
+      Error("MUONmass_ESD", "getting geometry from file %s failed", geoFilename);
+      return;
+    }
+  }
+  
+  // set  mag field 
+  // waiting for mag field in CDB 
+  printf("Loading field map...\n");
+  AliMagFMaps* field = new AliMagFMaps("Maps","Maps", 1, 1., 10., AliMagFMaps::k5kG);
+  AliTracker::SetFieldMap(field, kFALSE);
+  // set the magnetic field for track extrapolations
+  AliMUONTrackExtrap::SetField(AliTracker::GetFieldMap());
+
   Double_t parameters[3*156];
   Double_t errors[3*156];
   Double_t pulls[3*156];
@@ -72,15 +89,37 @@ void MUONAlignment(Int_t nEvents = 100000, TString fileName = "galice.root", TSt
   alig->InitGlobalParameters(parameters);
 
   AliMUONGeometryTransformer *transform = new AliMUONGeometryTransformer(true);
-  transform->ReadGeometryData("volpath.dat", "transform.dat");
+  transform->ReadGeometryData("volpath.dat", gGeoManager);
   alig->SetGeometryTransformer(transform);
+  
+  // Set tracking station to use
+  Bool_t bStOnOff[5] = {kTRUE,kTRUE,kTRUE,kTRUE,kTRUE};
+
+  // Fix parameters or add constraints here
+  for (Int_t iSt=0; iSt<5; iSt++)
+    if (!bStOnOff[iSt]) alig->FixStation(iSt+1);
+
+  // Left and right sides of the detector are independent, one can choose to align 
+  // only one side
+  Bool_t bSpecLROnOff[2] = {kTRUE,kTRUE};
+  alig->FixHalfSpectrometer(bStOnOff,bSpecLROnOff);
+
+  // Set predifined global constrains: X, Y, P, XvsZ, YvsZ, PvsZ, XvsY, YvsY, PvsY
+  Bool_t bVarXYT[9] = {kTRUE,kTRUE,kTRUE,kTRUE,kTRUE,kTRUE,kTRUE,kTRUE,kTRUE};
+  Bool_t bDetTLBR[4] = {kFALSE,kTRUE,kFALSE,kTRUE};
+  alig->AddConstraints(bStOnOff,bVarXYT,bDetTLBR,bSpecLROnOff);
+
 
   char cFileName[100];  
   AliMUONDataInterface amdi;
 
   Int_t lMaxFile = 1000;
   Int_t iFile = 0;
+  Int_t iEvent = 0;
   bool bKeepLoop = kTRUE;
+  Int_t iTrackTot=0;
+  Int_t iTrackOk=0;
+
   while(bKeepLoop && iFile<lMaxFile){
     iFile++;
     if (bLoop) {
@@ -95,12 +134,19 @@ void MUONAlignment(Int_t nEvents = 100000, TString fileName = "galice.root", TSt
     cout << "Using file: " << cFileName << endl;
     amdi.SetFile(cFileName);
     Int_t nevents = amdi.NumberOfEvents();
+    cout << "... with " << nevents << endl;
     for(Int_t event = 0; event < nevents; event++) {
       amdi.GetEvent(event);
+      if (iEvent >= nEvents){
+       bKeepLoop = kFALSE;
+       break;
+      }
+      iEvent++;
+
       Int_t ntracks = amdi.NumberOfRecTracks();
-      cout << " there are " << ntracks << " tracks in event " << event << endl;
+      if (!event%100)
+       cout << " there are " << ntracks << " tracks in event " << event << endl;
       Int_t iTrack=0;
-      Int_t iTrackOk=0;
       AliMUONTrack* track = amdi.RecTrack(iTrack);
       while(track) {   
        AliMUONTrackParam trackParam(*((AliMUONTrackParam*)(track->GetTrackParamAtHit()->First())));
@@ -109,15 +155,15 @@ void MUONAlignment(Int_t nEvents = 100000, TString fileName = "galice.root", TSt
        fInvBenMom->Fill(invBenMom);
        fBenMom->Fill(1./invBenMom);
        if (TMath::Abs(invBenMom)<=1.04) {
-         cout << "Track " << iTrack << endl;
          alig->ProcessTrack(track);
-         cout << "Calling LocalFit" << endl;
          alig->LocalFit(iTrackOk++,trackParams,0);
        }
        iTrack++;
+       iTrackTot++;
        track = amdi.RecTrack(iTrack);
       }
     }
+    cout << "Processed " << iTrackTot << " Tracks so far." << endl;
   }
   alig->GlobalFit(parameters,errors,pulls);