]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - STEER/AliReconstruction.cxx
From Ludovic: new files for visualization of V0s.
[u/mrichter/AliRoot.git] / STEER / AliReconstruction.cxx
index 4768924ea48f0a5c704a40a299abea5896f89d06..f90f8891d087f0cc7743298f265099efd63f246b 100644 (file)
 #include "AliRawReaderFile.h"
 #include "AliRawReaderDate.h"
 #include "AliRawReaderRoot.h"
+#include "AliRawEventHeaderBase.h"
 #include "AliESD.h"
 #include "AliESDfriend.h"
 #include "AliESDVertex.h"
+#include "AliMultiplicity.h"
 #include "AliTracker.h"
 #include "AliVertexer.h"
 #include "AliVertexerTracks.h"
@@ -170,6 +172,7 @@ AliReconstruction::AliReconstruction(const char* gAliceFilename, const char* cdb
   fFillESD("ALL"),
   fGAliceFileName(gAliceFilename),
   fInput(""),
+  fEquipIdMap(""),
   fFirstEvent(0),
   fLastEvent(-1),
   fCheckPointLevel(0),
@@ -181,9 +184,11 @@ AliReconstruction::AliReconstruction(const char* gAliceFilename, const char* cdb
   fRawReader(NULL),
 
   fVertexer(NULL),
+  fDiamondProfile(NULL),
 
   fAlignObjArray(NULL),
-  fCDBUri(cdbUri)
+  fCDBUri(cdbUri),
+  fSpecCDBUri()
 {
 // create reconstruction object with default parameters
   
@@ -212,6 +217,7 @@ AliReconstruction::AliReconstruction(const AliReconstruction& rec) :
   fFillESD(rec.fFillESD),
   fGAliceFileName(rec.fGAliceFileName),
   fInput(rec.fInput),
+  fEquipIdMap(rec.fEquipIdMap),
   fFirstEvent(rec.fFirstEvent),
   fLastEvent(rec.fLastEvent),
   fCheckPointLevel(0),
@@ -223,13 +229,15 @@ AliReconstruction::AliReconstruction(const AliReconstruction& rec) :
   fRawReader(NULL),
 
   fVertexer(NULL),
+  fDiamondProfile(NULL),
 
   fAlignObjArray(rec.fAlignObjArray),
-  fCDBUri(rec.fCDBUri)
+  fCDBUri(rec.fCDBUri),
+  fSpecCDBUri()
 {
 // copy constructor
 
-  for (Int_t i = 0; i < fOptions.GetEntriesFast(); i++) {
+  for (Int_t i = 0; i < rec.fOptions.GetEntriesFast(); i++) {
     if (rec.fOptions[i]) fOptions.Add(rec.fOptions[i]->Clone());
   }
   for (Int_t iDet = 0; iDet < fgkNDetectors; iDet++) {
@@ -237,6 +245,9 @@ AliReconstruction::AliReconstruction(const AliReconstruction& rec) :
     fLoader[iDet] = NULL;
     fTracker[iDet] = NULL;
   }
+  for (Int_t i = 0; i < rec.fSpecCDBUri.GetEntriesFast(); i++) {
+    if (rec.fSpecCDBUri[i]) fSpecCDBUri.Add(rec.fSpecCDBUri[i]->Clone());
+  }
 }
 
 //_____________________________________________________________________________
@@ -256,6 +267,7 @@ AliReconstruction::~AliReconstruction()
 
   CleanUp();
   fOptions.Delete();
+  fSpecCDBUri.Delete();
 }
 
 //_____________________________________________________________________________
@@ -266,36 +278,81 @@ void AliReconstruction::InitCDBStorage()
 // to retrieve the calibration and alignment constants
 
   AliCDBManager* man = AliCDBManager::Instance();
-  if (!man->IsDefaultStorageSet())
+  if (man->IsDefaultStorageSet())
   {
     AliWarning("!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!");
-    AliWarning("Default CDB storage not yet set");
-    AliWarning(Form("Using default storage declared in AliSimulation: %s",fCDBUri.Data()));
+    AliWarning("Default CDB storage has been already set !");
+    AliWarning(Form("Ignoring the default storage declared in AliReconstruction: %s",fCDBUri.Data()));
     AliWarning("!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!");
-    SetDefaultStorage(fCDBUri);
-    
-    Int_t cdbRun = AliCDBManager::Instance()->GetRun();
-    if(cdbRun == -1){
-       AliWarning("AliCDBManager's run number temporarily set to 0!!");
-       AliCDBManager::Instance()->SetRun(0);
-    }
-  }  
-  
+    fCDBUri = "";
+  }
+  else {
+    AliDebug(2, "!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!");
+    AliDebug(2, Form("Default CDB storage is set to: %s",fCDBUri.Data()));
+    AliDebug(2, "!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!");
+    man->SetDefaultStorage(fCDBUri);
+  }
+
+  // Now activate the detector specific CDB storage locations
+  for (Int_t i = 0; i < fSpecCDBUri.GetEntriesFast(); i++) {
+    TObject* obj = fSpecCDBUri[i];
+    if (!obj) continue;
+    AliDebug(2, "!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!");
+    AliDebug(2, Form("Specific CDB storage for %s is set to: %s",obj->GetName(),obj->GetTitle()));
+    AliDebug(2, "!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!");
+    man->SetSpecificStorage(obj->GetName(), obj->GetTitle());
+  }
+  man->Print();
 }
 
 //_____________________________________________________________________________
 void AliReconstruction::SetDefaultStorage(const char* uri) {
-// activate a default CDB storage 
+// Store the desired default CDB storage location
+// Activate it later within the Run() method
 
-   AliCDBManager::Instance()->SetDefaultStorage(uri);
+  fCDBUri = uri;
 
 }
 
 //_____________________________________________________________________________
-void AliReconstruction::SetSpecificStorage(const char* detName, const char* uri) {
-// activate a detector-specific CDB storage 
+void AliReconstruction::SetSpecificStorage(const char* calibType, const char* uri) {
+// Store a detector-specific CDB storage location
+// Activate it later within the Run() method
+
+  AliCDBPath aPath(calibType);
+  if(!aPath.IsValid()){
+       // if calibType is not wildcard but it is a valid detector, add "/*" to make it a valid path
+       for (Int_t iDet = 0; iDet < fgkNDetectors; iDet++) {
+               if(!strcmp(calibType, fgkDetectorName[iDet])) {
+                       aPath.SetPath(Form("%s/*", calibType));
+                       AliInfo(Form("Path for specific storage set to %s", aPath.GetPath().Data()));
+                       break;
+               }
+        }
+       if(!aPath.IsValid()){
+               AliError(Form("Not a valid path or detector: %s", calibType));
+               return;
+       }
+  }
 
-   AliCDBManager::Instance()->SetSpecificStorage(detName, uri);
+  // check that calibType refers to a "valid" detector name
+  Bool_t isDetector = kFALSE;
+  for (Int_t iDet = 0; iDet < fgkNDetectors; iDet++) {
+    TString detName = fgkDetectorName[iDet];
+    if(aPath.GetLevel0() == detName) {
+       isDetector = kTRUE;
+       break;
+    }
+  }
+
+  if(!isDetector) {
+       AliError(Form("Not a valid detector: %s", aPath.GetLevel0().Data()));
+       return;
+  }
+
+  TObject* obj = fSpecCDBUri.FindObject(aPath.GetPath().Data());
+  if (obj) fSpecCDBUri.Remove(obj);
+  fSpecCDBUri.Add(new TNamed(aPath.GetPath().Data(), uri));
 
 }
 
@@ -313,8 +370,25 @@ Bool_t AliReconstruction::SetRunNumber()
       return kFALSE;
     }
     // read run number from gAlice
-    AliCDBManager::Instance()->SetRun(fRunLoader->GetAliRun()->GetRunNumber());
-    AliInfo(Form("Run number: %d",AliCDBManager::Instance()->GetRun()));
+    if(fRunLoader->GetAliRun())
+      AliCDBManager::Instance()->SetRun(fRunLoader->GetAliRun()->GetRunNumber());
+    else {
+      if(fRawReader) {
+       if(fRawReader->NextEvent()) {
+         AliCDBManager::Instance()->SetRun(fRawReader->GetRunNumber());
+         fRawReader->RewindEvents();
+       }
+       else {
+         AliError("No raw-data events found !");
+         return kFALSE;
+       }
+      }
+      else {
+       AliError("Neither gAlice nor RawReader objects are found !");
+       return kFALSE;
+      }
+    }
+    AliInfo(Form("CDB Run number: %d",AliCDBManager::Instance()->GetRun()));
   }
   return kTRUE;
 }
@@ -338,7 +412,7 @@ Bool_t AliReconstruction::ApplyAlignObjsToGeom(TObjArray* alObjArray)
     }
 
   if (AliDebugLevelClass() >= 1) {
-    gGeoManager->CheckOverlaps(20);
+    gGeoManager->GetTopNode()->CheckOverlaps(20);
     TObjArray* ovexlist = gGeoManager->GetListOfOverlaps();
     if(ovexlist->GetEntriesFast()){  
       AliError("The application of alignment objects to the geometry caused huge overlaps/extrusions!");
@@ -421,9 +495,9 @@ Bool_t AliReconstruction::MisalignGeometry(const TString& detectors)
   
        if ((detStr.CompareTo("ALL") == 0)) detStr = "";
        dataNotLoaded += detStr;
-       AliInfo(Form("Alignment data loaded for: %s",
+       if(!dataLoaded.IsNull()) AliInfo(Form("Alignment data loaded for: %s",
                          dataLoaded.Data()));
-       AliInfo(Form("Didn't/couldn't load alignment data for: %s",
+       if(!dataNotLoaded.IsNull()) AliInfo(Form("Didn't/couldn't load alignment data for: %s",
                          dataNotLoaded.Data()));
   } // fLoadAlignFromCDB flag
  
@@ -472,8 +546,6 @@ Bool_t AliReconstruction::Run(const char* input,
 {
 // run the reconstruction
 
-  InitCDBStorage();
-
   // set the input
   if (!input) input = fInput.Data();
   TString fileName(input);
@@ -485,10 +557,16 @@ Bool_t AliReconstruction::Run(const char* input,
     fRawReader = new AliRawReaderDate(fileName);
     fRawReader->SelectEvents(7);
   }
+  if (!fEquipIdMap.IsNull() && fRawReader)
+    fRawReader->LoadEquipmentIdsMap(fEquipIdMap);
+
 
   // get the run loader
   if (!InitRunLoader()) return kFALSE;
 
+  // Initialize the CDB storage
+  InitCDBStorage();
+
   // Set run number in CDBManager (if it is not already set by the user)
   if (!SetRunNumber()) if (fStopOnError) return kFALSE;
 
@@ -501,21 +579,6 @@ Bool_t AliReconstruction::Run(const char* input,
   }
   if (!MisalignGeometry(fLoadAlignData)) if (fStopOnError) return kFALSE;
 
-  // Temporary fix by A.Gheata
-  // Could be removed with the next Root version (>5.11)
-  if (gGeoManager) {
-    TIter next(gGeoManager->GetListOfVolumes());
-    TGeoVolume *vol;
-    while ((vol = (TGeoVolume *)next())) {
-      if (vol->GetVoxels()) {
-       if (vol->GetVoxels()->NeedRebuild()) {
-         vol->GetVoxels()->Voxelize();
-         vol->FindOverlaps();
-       }
-      }
-    }
-  }
-
   // local reconstruction
   if (!fRunLocalReconstruction.IsNull()) {
     if (!RunLocalReconstruction(fRunLocalReconstruction)) {
@@ -573,20 +636,15 @@ Bool_t AliReconstruction::Run(const char* input,
   delete esd; delete hltesd;
   esd = NULL; hltesd = NULL;
 
-  // create the file and tree with ESD additions
-  TFile *filef=0; TTree *treef=0; AliESDfriend *esdf=0;
+  // create the branch with ESD additions
+  AliESDfriend *esdf=0;
   if (fWriteESDfriend) {
-     filef = TFile::Open("AliESDfriends.root", "RECREATE");
-     if (!filef->IsOpen()) {
-        AliError("opening AliESDfriends.root failed");
-     }
-     treef = new TTree("esdFriendTree", "Tree with ESD friends");
-     treef->Branch("ESDfriend", "AliESDfriend", &esdf);
+     TBranch *br=tree->Branch("ESDfriend.", "AliESDfriend", &esdf);
+     br->SetFile("AliESDfriends.root");
   }
 
-  gROOT->cd();
-
   AliVertexerTracks tVertexer;
+  if(fDiamondProfile) tVertexer.SetVtxStart(fDiamondProfile);
 
   // loop over events
   if (fRawReader) fRawReader->RewindEvents();
@@ -629,12 +687,10 @@ Bool_t AliReconstruction::Run(const char* input,
     hltesd->SetRunNumber(fRunLoader->GetHeader()->GetRun());
     esd->SetEventNumber(fRunLoader->GetHeader()->GetEventNrInRun());
     hltesd->SetEventNumber(fRunLoader->GetHeader()->GetEventNrInRun());
-    if (gAlice) {
-      esd->SetMagneticField(AliTracker::GetBz());
-      hltesd->SetMagneticField(AliTracker::GetBz());
-    } else {
-      // ???
-    }
+
+    // Set magnetic field from the tracker
+    esd->SetMagneticField(AliTracker::GetBz());
+    hltesd->SetMagneticField(AliTracker::GetBz());
 
     // vertex finder
     if (fRunVertexFinder) {
@@ -672,6 +728,8 @@ Bool_t AliReconstruction::Run(const char* input,
        if (fStopOnError) {CleanUp(file, fileOld); return kFALSE;}
       }
     }
+    // fill Event header information from the RawEventHeader
+    if (fRawReader){FillRawEventHeaderESD(esd);}
 
     // combined PID
     AliESDpid::MakePID(esd);
@@ -686,38 +744,32 @@ Bool_t AliReconstruction::Run(const char* input,
       }
     }
 
-    esd->SetPrimaryVertex(tVertexer.FindVertex(esd));
+    esd->SetPrimaryVertex(tVertexer.FindPrimaryVertex(esd));
 
     // write ESD
-    tree->Fill();
-    // write HLT ESD
-    hlttree->Fill();
-
-    // write ESD friend
     if (fWriteESDfriend) {
        esdf=new AliESDfriend();
        esd->GetESDfriend(esdf);
-       treef->Fill();  
     }
+    tree->Fill();
+
+    // write HLT ESD
+    hlttree->Fill();
 
     if (fCheckPointLevel > 0)  WriteESD(esd, "final"); 
  
-    delete esd; delete hltesd;
-    esd = NULL; hltesd = NULL;
+    delete esd; delete esdf; delete hltesd;
+    esd = NULL; esdf=NULL; hltesd = NULL;
   }
 
   AliInfo(Form("Execution time for filling ESD : R:%.2fs C:%.2fs",
               stopwatch.RealTime(),stopwatch.CpuTime()));
 
   file->cd();
+  tree->SetBranchStatus("ESDfriend*",0);
   tree->Write();
   hlttree->Write();
 
-  if (fWriteESDfriend) {
-     filef->cd();
-     treef->Write(); delete treef; filef->Close(); delete filef; 
-  }
-
   // Create tags for the events in the ESD tree (the ESD tree is always present)
   // In case of empty events the tags will contain dummy values
   CreateTag(file);
@@ -887,6 +939,10 @@ Bool_t AliReconstruction::RunVertexFinder(AliESD*& esd)
     vertex = new AliESDVertex(vtxPos, vtxErr);
   }
   esd->SetVertex(vertex);
+  // if SPD multiplicity has been determined, it is stored in the ESD
+  AliMultiplicity *mult= fVertexer->GetMultiplicity();
+  if(mult)esd->SetMultiplicity(mult);
+
   for (Int_t iDet = 0; iDet < fgkNDetectors; iDet++) {
     if (fTracker[iDet]) fTracker[iDet]->SetVertex(vtxPos, vtxErr);
   }  
@@ -1072,7 +1128,7 @@ Bool_t AliReconstruction::RunTracking(AliESD*& esd)
     AliESDtrack * track = esd->GetTrack(itrack);
     if (!track) continue;
     if (track->IsOn(AliESDtrack::kITSrefit)) continue;
-    track->PropagateTo(kRadius, track->GetMass(),kMaxStep,kTRUE);
+    track->PropagateTo(kRadius, fieldZ, track->GetMass(),kMaxStep,kTRUE);
     track->RelateToVertex(esd->GetVertex(),fieldZ, kMaxD);
   }
   
@@ -1190,6 +1246,30 @@ Bool_t AliReconstruction::FillTriggerESD(AliESD*& esd)
   return kTRUE;
 }
 
+
+
+
+
+//_____________________________________________________________________________
+Bool_t AliReconstruction::FillRawEventHeaderESD(AliESD*& esd)
+{
+  // 
+  // Filling information from RawReader Header
+  // 
+
+  AliInfo("Filling information from RawReader Header");
+  esd->SetTimeStamp(0);
+  esd->SetEventType(0);
+  const AliRawEventHeaderBase * eventHeader = fRawReader->GetEventHeader();
+  if (eventHeader){
+    esd->SetTimeStamp((eventHeader->Get("Timestamp")));  
+    esd->SetEventType((eventHeader->Get("Type")));  
+  }
+
+  return kTRUE;
+}
+
+
 //_____________________________________________________________________________
 Bool_t AliReconstruction::IsSelected(TString detName, TString& detectors) const
 {
@@ -1445,6 +1525,8 @@ void AliReconstruction::CleanUp(TFile* file, TFile* fileOld)
   }
   delete fVertexer;
   fVertexer = NULL;
+  delete fDiamondProfile;
+  fDiamondProfile = NULL;
 
   delete fRunLoader;
   fRunLoader = NULL;
@@ -1550,6 +1632,7 @@ void AliReconstruction::CreateTag(TFile* file)
   Int_t nEl1GeV, nEl3GeV, nEl10GeV;
   Float_t maxPt = .0, meanPt = .0, totalP = .0;
   Int_t fVertexflag;
+  Int_t iRunNumber = 0;
   TString fVertexName("default");
 
   AliRunTag *tag = new AliRunTag();
@@ -1570,8 +1653,9 @@ void AliReconstruction::CreateTag(TFile* file)
   TBranch * b = t->GetBranch("ESD");
   AliESD *esd = 0;
   b->SetAddress(&esd);
-  
-  tag->SetRunId(esd->GetRunNumber());
+
+  b->GetEntry(0);
+  Int_t iInitRunNumber = esd->GetRunNumber();
   
   Int_t iNumberOfEvents = b->GetEntries();
   for (Int_t iEventNumber = 0; iEventNumber < iNumberOfEvents; iEventNumber++) {
@@ -1603,6 +1687,8 @@ void AliReconstruction::CreateTag(TFile* file)
     fVertexflag = 0;
 
     b->GetEntry(iEventNumber);
+    iRunNumber = esd->GetRunNumber();
+    if(iRunNumber != iInitRunNumber) AliFatal("Inconsistency of run numbers in the AliESD!!!");
     const AliESDVertex * vertexIn = esd->GetVertex();
     if (!vertexIn) AliError("ESD has not defined vertex.");
     if (vertexIn) fVertexName = vertexIn->GetName();
@@ -1725,7 +1811,8 @@ void AliReconstruction::CreateTag(TFile* file)
 
     evTag->SetT0VertexZ(esd->GetT0zVertex());
     
-    evTag->SetTrigger(esd->GetTriggerMask());
+    evTag->SetTriggerMask(esd->GetTriggerMask());
+    evTag->SetTriggerCluster(esd->GetTriggerCluster());
     
     evTag->SetZDCNeutron1Energy(esd->GetZDCN1Energy());
     evTag->SetZDCProton1Energy(esd->GetZDCP1Energy());
@@ -1772,6 +1859,7 @@ void AliReconstruction::CreateTag(TFile* file)
     evTag->SetMeanPt(meanPt);
     evTag->SetMaxPt(maxPt);
     
+    tag->SetRunId(iInitRunNumber);
     tag->AddEventTag(*evTag);
   }
   lastEvent = iNumberOfEvents;