]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - MUON/AliMUONReconstructor.cxx
Adding other par files
[u/mrichter/AliRoot.git] / MUON / AliMUONReconstructor.cxx
index 79debec71815f2d3b5734db6b990616314623dff..729b132d46e748b53c0c089ff2a46add4a9361b1 100644 (file)
 ///
 /// Implementation of AliReconstructor for MUON subsystem.
 ///
-/// The clustering mode and the associated parameters can be changed by using
-/// AliMUONRecoParam *muonRecoParam = AliMUONRecoParam::GetLow(High)FluxParam();
-/// muonRecoParam->Set...(); // see methods in AliMUONRecoParam.h for details
-/// AliMUONReconstructor::SetRecoParam(muonRecoParam);
+/// The clustering mode and the associated parameters can be changed through the
+/// AliMUONRecoParam object set in the reconstruction macro or read from the CDB
+/// (see methods in AliMUONRecoParam.h file for details)
 ///
 /// Valid modes are :
 ///
@@ -44,6 +43,9 @@
 /// option either, as center-of-gravity is generally not a good estimate
 /// of the cluster position...
 ///
+/// PEAKCOG : COG cluster finder around local maxima
+/// PEAKFIT : fit around local maxima with up to 3 peaks, COG otherwise
+///
 /// NOCLUSTERING : bypass completely the clustering stage
 ///
 /// ------
 ///
 /// TRIGGERDISABLE : disable the treatment of MUON trigger
 ///
+/// ENABLEERRORLOGGING : enable error logging (this activates also warnings in RawStreamTracker)
+///
 /// \author Laurent Aphecetche, Subatech
 //-----------------------------------------------------------------------------
 
 #include "AliMUONReconstructor.h"
 
-#include "AliCDBManager.h"
-#include "AliLog.h"
 #include "AliMUONCalibrationData.h"
 #include "AliMUONClusterFinderCOG.h"
 #include "AliMUONClusterFinderMLEM.h"
 #include "AliMUONClusterFinderSimpleFit.h"
-#include "AliMUONClusterReconstructor.h"
+#include "AliMUONClusterFinderPeakCOG.h"
+#include "AliMUONClusterFinderPeakFit.h"
+#include "AliMUONClusterStoreV1.h"
 #include "AliMUONClusterStoreV2.h"
 #include "AliMUONConstants.h"
 #include "AliMUONDigitCalibrator.h"
 #include "AliMUONPreClusterFinder.h"
 #include "AliMUONPreClusterFinderV2.h"
 #include "AliMUONPreClusterFinderV3.h"
+#include "AliMUONSimpleClusterServer.h"
 #include "AliMUONTracker.h"
-#include "AliMUONVTrackStore.h"
-#include "AliMUONTriggerChamberEff.h"
 #include "AliMUONTriggerCircuit.h"
-#include "AliMUONTriggerCrateStore.h"
 #include "AliMUONTriggerStoreV1.h"
 #include "AliMUONVClusterFinder.h"
-#include "AliMUONRecoParam.h"
+#include "AliMUONVClusterServer.h"
+#include "AliMUONVTrackStore.h"
+
+#include "AliMpArea.h"
 #include "AliMpCDB.h"
+#include "AliMpConstants.h"
+#include "AliMpDDLStore.h"
+#include "AliMpSegmentation.h"
+
 #include "AliRawReader.h"
+#include "AliCDBManager.h"
 #include "AliCodeTimer.h"
+#include "AliLog.h"
+
 #include <Riostream.h>
+#include <TObjArray.h>
 #include <TClonesArray.h>
 #include <TString.h>
 #include <TTree.h>
 ClassImp(AliMUONReconstructor)
 /// \endcond 
 
-AliMUONRecoParam* AliMUONReconstructor::fgRecoParam = 0x0; // reconstruction parameters
-
 //_____________________________________________________________________________
 AliMUONReconstructor::AliMUONReconstructor() : 
 AliReconstructor(),
-fCrateManager(0x0),
 fDigitMaker(0x0),
 fTransformer(new AliMUONGeometryTransformer()),
 fDigitStore(0x0),
 fTriggerCircuit(0x0),
 fCalibrationData(0x0),
 fDigitCalibrator(0x0),
-fClusterReconstructor(0x0),
-fClusterStore(0x0),
+fClusterServer(0x0),
 fTriggerStore(0x0),
 fTrackStore(0x0),
-fTrigChamberEff(0x0)
+fClusterStore(0x0)
 {
   /// normal ctor
 
+  AliDebug(1,"");
+
+  // Unload and delete old mapping
+  AliCDBManager::Instance()->UnloadFromCache("MUON/Calib/Mapping");
+  AliCDBManager::Instance()->UnloadFromCache("MUON/Calib/DDLStore");
+  delete AliMpDDLStore::Instance();
+  delete AliMpSegmentation::Instance();
+
   // Load mapping
   if ( ! AliMpCDB::LoadDDLStore() ) {
     AliFatal("Could not access mapping from OCDB !");
@@ -136,44 +153,28 @@ fTrigChamberEff(0x0)
   // Load geometry data
   fTransformer->LoadGeometryData();
   
-  // initialize reconstruction parameters in not already done
-  if (!fgRecoParam) {
-    AliWarning("Reconstruction parameters not initialized - Use default one");
-    fgRecoParam = AliMUONRecoParam::GetLowFluxParam();
-  }
-  
 }
 
 //_____________________________________________________________________________
 AliMUONReconstructor::~AliMUONReconstructor()
 {
   /// dtor
+
+  AliDebug(1,"");
+
   delete fDigitMaker;
   delete fDigitStore;
   delete fTransformer;
-  delete fCrateManager;
   delete fTriggerCircuit;
-  delete fCalibrationData;
   delete fDigitCalibrator;
-  delete fClusterReconstructor;
-  delete fClusterStore;
+  delete fCalibrationData;
+  delete fClusterServer;
   delete fTriggerStore;
   delete fTrackStore;
-  delete fTrigChamberEff;
-}
+  delete fClusterStore;
 
-//_____________________________________________________________________________
-void AliMUONReconstructor::SetRecoParam(AliMUONRecoParam *param)
-{
-  /// set reconstruction parameters
-  
-  // remove existing parameters
-  if (fgRecoParam) {
-    cout<<"AliMUONReconstructor::SetRecoParam: Reconstruction parameters already initialized - overwrite them"<<endl;
-    delete fgRecoParam;
-  }
-  
-  fgRecoParam = param;
+  delete AliMpSegmentation::Instance(false);
+  delete AliMpDDLStore::Instance(false);
 }
 
 //_____________________________________________________________________________
@@ -185,43 +186,10 @@ AliMUONReconstructor::Calibrate(AliMUONVDigitStore& digitStore) const
   {
     CreateCalibrator();
   }
-  AliCodeTimerAuto(Form("%s::Calibrate(AliMUONVDigitStore*)",fDigitCalibrator->ClassName()))
+  AliCodeTimerAuto(Form("%s::Calibrate(AliMUONVDigitStore*)",fDigitCalibrator->ClassName()),0)
   fDigitCalibrator->Calibrate(digitStore);  
 }
 
-//_____________________________________________________________________________
-void
-AliMUONReconstructor::Clusterize(const AliMUONVDigitStore& digitStore,
-                                 AliMUONVClusterStore& clusterStore) const
-{
-  /// Creates clusters from digits.
-
-  TString sopt(fgRecoParam->GetClusteringMode());
-  sopt.ToUpper();
-  if ( sopt.Contains("NOCLUSTERING") ) return;
-  
-  if  (!fClusterReconstructor) CreateClusterReconstructor();
-  
-  // if the required clustering mode does not exist
-  if  (!fClusterReconstructor) return;
-  
-  AliCodeTimerAuto(Form("%s::Digits2Clusters(const AliMUONVDigitStore&,AliMUONVClusterStore&)",
-                        fClusterReconstructor->ClassName()))
-  fClusterReconstructor->Digits2Clusters(digitStore,clusterStore);  
-}
-
-//_____________________________________________________________________________
-AliMUONVClusterStore*
-AliMUONReconstructor::ClusterStore() const
-{
-  /// Return (and create if necessary) the cluster container
-  if (!fClusterStore) 
-  {
-    fClusterStore = new AliMUONClusterStoreV2;
-  }
-  return fClusterStore;
-}
-
 //_____________________________________________________________________________
 void
 AliMUONReconstructor::ConvertDigits(AliRawReader* rawReader, 
@@ -230,6 +198,14 @@ AliMUONReconstructor::ConvertDigits(AliRawReader* rawReader,
 {
   /// Convert raw data into digit and trigger stores
   CreateDigitMaker();
+
+  // Skip reconstruction if event is Calibration
+  if ( GetRecoParam()->GetEventSpecie() == AliRecoParam::kCalib ) {
+    digitStore->Clear(); // Remove possible digits from previous event
+    triggerStore->Clear(); // Remove possible triggers from previous event
+    AliInfo("Calibration event: do not convert digits");
+    return;
+  }
   
   AliCodeTimerStart(Form("%s::Raw2Digits(AliRawReader*,AliMUONVDigitStore*,AliMUONVTriggerStore*)",
                     fDigitMaker->ClassName()))
@@ -244,7 +220,7 @@ void
 AliMUONReconstructor::ConvertDigits(AliRawReader* rawReader, TTree* digitsTree) const
 {
    /// convert raw data into a digit tree
-  AliCodeTimerAuto("")
+  AliCodeTimerAuto("",0)
 
   Bool_t alone = ( TriggerStore() == 0 );
   
@@ -268,17 +244,6 @@ AliMUONReconstructor::ConvertDigits(AliRawReader* rawReader, TTree* digitsTree)
   }
 }
 
-//_____________________________________________________________________________
-AliMUONTriggerCrateStore*
-AliMUONReconstructor::CrateManager() const
-{
-  /// Return (and create if necessary) the trigger crate store
-  if (fCrateManager) return fCrateManager;
-  fCrateManager = new AliMUONTriggerCrateStore;
-  fCrateManager->ReadFromFile();
-  return fCrateManager;
-}
-
 //_____________________________________________________________________________
 void
 AliMUONReconstructor::CreateDigitMaker() const
@@ -286,9 +251,23 @@ AliMUONReconstructor::CreateDigitMaker() const
   /// Create (and create if necessary) the digit maker
   if (fDigitMaker) return;
 
-  AliCodeTimerAuto("")
+  AliCodeTimerAuto("",0)
+
+  TString option = GetOption();
+  
+  Bool_t enableErrorLogging = kFALSE;
 
-  fDigitMaker = new AliMUONDigitMaker;
+  if (option.Contains("ENABLEERRORLOGGING"))
+  {
+    enableErrorLogging = kTRUE;
+  }
+
+  fDigitMaker = new AliMUONDigitMaker(enableErrorLogging);
+  option.ToUpper();
+  if ( option.Contains("SAVEDIGITS" ))
+    {
+      fDigitMaker->SetMakeTriggerDigits(kTRUE);
+    }
 }
 
 //_____________________________________________________________________________
@@ -298,25 +277,12 @@ AliMUONReconstructor::CreateTriggerCircuit() const
   /// Return (and create if necessary) the trigger circuit object
   if (fTriggerCircuit) return;
 
-  AliCodeTimerAuto("")
+  AliCodeTimerAuto("",0)
 
   fTriggerCircuit = new AliMUONTriggerCircuit(fTransformer);
 
 }
 
-//_____________________________________________________________________________
-void
-AliMUONReconstructor::CreateTriggerChamberEff() const
-{
-  /// Create (and create if necessary) the trigger chamber efficiency class
-  if (fTrigChamberEff) return;
-
-  AliCodeTimerAuto("")
-
-  fTrigChamberEff = new AliMUONTriggerChamberEff(fTransformer,fDigitMaker,kTRUE);
-  //fTrigChamberEff->SetDebugLevel(1);
-}
-
 //_____________________________________________________________________________
 AliTracker* 
 AliMUONReconstructor::CreateTracker() const
@@ -325,26 +291,44 @@ AliMUONReconstructor::CreateTracker() const
   
   CreateTriggerCircuit();
   CreateDigitMaker();
-  CreateTriggerChamberEff();
+  CreateClusterServer();
+
+  AliMUONTracker* tracker(0x0);
+  
+  if ( ! GetRecoParam()->CombineClusterTrackReco() )
+  {
+    tracker = new AliMUONTracker(GetRecoParam(),
+                                0x0,
+                                 *DigitStore(),
+                                 fDigitMaker,
+                                 fTransformer,
+                                 fTriggerCircuit);
+  }
+  else
+  {
+    tracker = new AliMUONTracker(GetRecoParam(),
+                                fClusterServer,
+                                 *DigitStore(),
+                                 fDigitMaker,
+                                 fTransformer,
+                                 fTriggerCircuit);
+  }
   
-  AliMUONTracker* tracker = new AliMUONTracker(fDigitMaker,fTransformer,fTriggerCircuit,fTrigChamberEff);
   
   return tracker;
 }
 
 //_____________________________________________________________________________
-void
-AliMUONReconstructor::CreateClusterReconstructor() const
+AliMUONVClusterFinder*
+AliMUONReconstructor::CreateClusterFinder(const char* clusterFinderType)
 {
-  /// Create cluster reconstructor, depending on clustering mode set in RecoParam
+  /// Create a given cluster finder instance
   
-  AliCodeTimerAuto("")
+  AliCodeTimerAutoGeneral("",0)
 
-  AliDebug(1,"");
-  
   AliMUONVClusterFinder* clusterFinder(0x0);
   
-  TString opt(fgRecoParam->GetClusteringMode());
+  TString opt(clusterFinderType);
   opt.ToUpper();
   
   if ( strstr(opt,"PRECLUSTERV2") )
@@ -359,6 +343,14 @@ AliMUONReconstructor::CreateClusterReconstructor() const
   {
     clusterFinder = new AliMUONPreClusterFinder;
   }  
+  else if ( strstr(opt,"PEAKCOG") )
+  {
+    clusterFinder = new AliMUONClusterFinderPeakCOG(kFALSE,new AliMUONPreClusterFinder);
+  }
+  else if ( strstr(opt,"PEAKFIT") )
+  {
+    clusterFinder = new AliMUONClusterFinderPeakFit(kFALSE,new AliMUONPreClusterFinder);
+  }
   else if ( strstr(opt,"COG") )
   {
     clusterFinder = new AliMUONClusterFinderCOG(new AliMUONPreClusterFinder);
@@ -389,13 +381,30 @@ AliMUONReconstructor::CreateClusterReconstructor() const
   } 
   else
   {
-    AliError(Form("clustering mode \"%s\" does not exist",opt.Data()));
-    return;
+    AliErrorClass(Form("clustering mode \"%s\" does not exist",opt.Data()));
+    return 0x0;
   }
   
+  return clusterFinder;
+}
+
+//_____________________________________________________________________________
+void
+AliMUONReconstructor::CreateClusterServer() const
+{
+  /// Create cluster server
+  
+  if ( fClusterServer ) return;
+  
+  AliCodeTimerAuto("",0);
+    
+  AliMUONVClusterFinder* clusterFinder = CreateClusterFinder(GetRecoParam()->GetClusteringMode());
+  
+  if ( !clusterFinder ) return;
+  
   AliInfo(Form("Will use %s for clusterizing",clusterFinder->ClassName()));
   
-  fClusterReconstructor = new AliMUONClusterReconstructor(clusterFinder,fTransformer);
+  fClusterServer = new AliMUONSimpleClusterServer(clusterFinder,*fTransformer);
 }
 
 //_____________________________________________________________________________
@@ -404,7 +413,7 @@ AliMUONReconstructor::CreateCalibrator() const
 {
   /// Create the calibrator
   
-  AliCodeTimerAuto("")
+  AliCodeTimerAuto("",0)
   
   Int_t runNumber = AliCDBManager::Instance()->GetRun();
 
@@ -435,7 +444,9 @@ AliMUONReconstructor::CreateCalibrator() const
     AliWarning("NOSTATUSMAP is obsolete");
   }
 
-  fDigitCalibrator = new AliMUONDigitCalibrator(*fCalibrationData);
+  TString calibMode = GetRecoParam()->GetCalibrationMode();
+
+  fDigitCalibrator = new AliMUONDigitCalibrator(*fCalibrationData,GetRecoParam(),calibMode.Data());
 }
 
 //_____________________________________________________________________________
@@ -473,40 +484,73 @@ AliMUONReconstructor::DigitStore() const
 //_____________________________________________________________________________
 void
 AliMUONReconstructor::FillTreeR(AliMUONVTriggerStore* triggerStore,
-                                AliMUONVClusterStore* clusterStore,
                                 TTree& clustersTree) const
 {
   /// Write the trigger and cluster information into TreeR
   
-  AliCodeTimerAuto("")
+  AliCodeTimerAuto("",0)
 
   AliDebug(1,"");
   
   Bool_t ok(kFALSE);
+  Bool_t alone(kTRUE); // is trigger the only info in TreeR ?
+  
+  if ( ! GetRecoParam()->CombineClusterTrackReco() )
+  {
+    alone = kFALSE; // we'll get both tracker and trigger information in TreeR
+  }
+  
   if ( triggerStore ) 
   {
-    Bool_t alone = ( clusterStore ? kFALSE : kTRUE );
     ok = triggerStore->Connect(clustersTree,alone);
     if (!ok)
     {
       AliError("Could not create triggerStore branches in TreeR");
     }
   }
-  
-  if ( clusterStore ) 
+
+  if ( !alone )
   {
-    Bool_t alone = ( triggerStore ? kFALSE : kTRUE );
-    ok = clusterStore->Connect(clustersTree,alone);
-    if (!ok)
+    if (!fClusterStore)
     {
-      AliError("Could not create triggerStore branches in TreeR");
-    }    
+      fClusterStore = new AliMUONClusterStoreV2;
+    }
+    
+    CreateClusterServer();
+    
+    TIter next(DigitStore()->CreateIterator());
+    fClusterServer->UseDigits(next,DigitStore());
+
+    AliMpArea area;
+    
+    AliDebug(1,Form("Doing full clusterization in local reconstruction using %s ",fClusterServer->ClassName()));
+    
+    for ( Int_t i = 0; i < AliMpConstants::NofTrackingChambers(); ++i ) 
+    {
+      if (GetRecoParam()->UseChamber(i))
+      {
+        if ( ( i == 6 || i == 7 )  && GetRecoParam()->BypassSt4() ) continue;
+        if ( ( i == 8 || i == 9 )  && GetRecoParam()->BypassSt5() ) continue;
+        
+        fClusterServer->Clusterize(i,*fClusterStore,area,GetRecoParam());
+      }
+    }
+    
+    Bool_t cok = fClusterStore->Connect(clustersTree,alone);
+    
+    if (!cok) AliError("Could not connect clusterStore to clusterTree");
+    
+    AliDebug(1,Form("Number of clusters found = %d",fClusterStore->GetSize()));
+    
+    StdoutToAliDebug(1,fClusterStore->Print());
   }
-  
+         
   if (ok) // at least one type of branches created successfully
   {
     clustersTree.Fill();
   }
+  
+  if (fClusterStore) fClusterStore->Clear();
 }
 
 //_____________________________________________________________________________
@@ -542,9 +586,8 @@ AliMUONReconstructor::Reconstruct(AliRawReader* rawReader, TTree* clustersTree)
   }
   
   ConvertDigits(rawReader,DigitStore(),TriggerStore());
-  Clusterize(*(DigitStore()),*(ClusterStore()));
-    
-  FillTreeR(TriggerStore(),ClusterStore(),*clustersTree);
+
+  FillTreeR(TriggerStore(),*clustersTree);
 }
 
 //_____________________________________________________________________________
@@ -554,7 +597,7 @@ AliMUONReconstructor::Reconstruct(TTree* digitsTree, TTree* clustersTree) const
   /// This method is called by AliReconstruction if HasLocalReconstruction()==kTRUE
   /// AND HasDigitConversion()==kTRUE
   
-  AliCodeTimerAuto("")
+  AliCodeTimerAuto("",0)
   
   AliDebug(1,"");
   
@@ -628,14 +671,13 @@ AliMUONReconstructor::Reconstruct(TTree* digitsTree, TTree* clustersTree) const
     // i.e. w/o going through raw data, this will be the case)
     TIter next(fDigitStore->CreateIterator());
     AliMUONVDigit* digit = static_cast<AliMUONVDigit*>(next());
-    if (!digit->IsCalibrated())
+    if (digit && !digit->IsCalibrated())
     {
       Calibrate(*fDigitStore);
     }
-    Clusterize(*fDigitStore,*(ClusterStore()));
   }
     
-  FillTreeR(fTriggerStore,ClusterStore(),*clustersTree);
+  FillTreeR(fTriggerStore,*clustersTree);
 }
 
 //_____________________________________________________________________________