]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - FMD/AliFMDBaseDA.cxx
Fix to allow new initialization of AliReconstruction (C.H.Christinsen)
[u/mrichter/AliRoot.git] / FMD / AliFMDBaseDA.cxx
index 3e02e304f15242f76f7bd142dd43d344577c643e..8e813ec173bc0abcc10aa3ae7c56a9232c0da6b6 100644 (file)
 //
 
 #include "AliFMDBaseDA.h"
-#include "iostream"
+#include "AliRawReader.h"
+#include "AliFMDDigit.h"
+#include "AliFMDParameters.h"
 #include "AliFMDRawReader.h"
 #include "AliFMDCalibSampleRate.h"
 #include "AliFMDCalibStripRange.h"
 #include "AliLog.h"
 #include "AliRawEventHeaderBase.h"
+#include "AliFMDDigit.h"
+#include <TClonesArray.h>
+#include <TFile.h>
 #include <TDatime.h>
 #include <TSystem.h>
 #include <TH2F.h>
+#include <TStopwatch.h>
+#include <TROOT.h>
+#include <TPluginManager.h>
+#include <iostream>
+#include <fstream>
 
 //_____________________________________________________________________
 ClassImp(AliFMDBaseDA)
@@ -144,10 +154,14 @@ AliFMDBaseDA::AliFMDBaseDA() :
   fRequiredEvents(0),
   fCurrentEvent(0), 
   fRunno(0),
-  fSummaries(0)
+  fSummaries(0),
+  fAll(false)
 {
   //Constructor
-  fSeenDetectors[0] = fSeenDetectors[1] = fSeenDetectors[2] = kFALSE;
+  for(Int_t i = 0; i< 3;i++) {
+    fSeenDetectors[i] = false;
+    fNEventsPerDetector[i] = 0;
+  }
   fDetectorArray.SetOwner();
   Rotate("conditions.csv", 3);
   fConditionsFile.open("conditions.csv");
@@ -166,15 +180,16 @@ AliFMDBaseDA::AliFMDBaseDA(const AliFMDBaseDA & baseDA) :
   fRequiredEvents(baseDA.fRequiredEvents),
   fCurrentEvent(baseDA.fCurrentEvent),
   fRunno(baseDA.fRunno),
-  fSummaries(0)
+  fSummaries(0),
+  fAll(baseDA.fAll)
 {
   //Copy constructor
-  fSeenDetectors[0] = baseDA.fSeenDetectors[0];
-  fSeenDetectors[1] = baseDA.fSeenDetectors[1];
-  fSeenDetectors[2] = baseDA.fSeenDetectors[2];
+  for(Int_t i = 0; i< 3;i++) {
+    fSeenDetectors[i] = baseDA.fSeenDetectors[0];
+    fNEventsPerDetector[i] = baseDA.fNEventsPerDetector[i];
+  }
 
-  fDetectorArray.SetOwner();
-  
+  fDetectorArray.SetOwner();  
 }
 
 
@@ -184,17 +199,44 @@ AliFMDBaseDA::~AliFMDBaseDA()
   //destructor
 }
 
+//_____________________________________________________________________
+Bool_t AliFMDBaseDA::HaveEnough(Int_t nEvents) const
+{
+  // if (!fAll) return nEvents > GetRequiredEvents();
+  if (nEvents <= 1) return false;
+
+  Bool_t ret = true; // Assume we have it 
+  for (Int_t i = 0; i < 3; i++) { 
+    if (!fSeenDetectors[i]) continue;
+    if (Int_t(fNEventsPerDetector[i]) <= GetRequiredEvents()) ret = false;
+  }
+  return ret;
+}
+//_____________________________________________________________________
+UShort_t AliFMDBaseDA::GetProgress(Int_t nEvents) const
+{
+  // if (!fAll) 
+  //  return UShort_t((nEvents *100)/ GetRequiredEvents());
+
+  if (nEvents <= 1) return 0;
+
+  Int_t got = 0;
+  Int_t total = 0;
+  for (Int_t i = 0; i < 3; i++) {
+    if (!fSeenDetectors[i]) continue;
+    got   += fNEventsPerDetector[i];
+    total += GetRequiredEvents();
+  }
+  return UShort_t(total > 0 ? (got * 100.) / total : 0);
+}
+
 //_____________________________________________________________________
 void AliFMDBaseDA::Run(AliRawReader* reader) 
 {
   //Run the FMD DA
   TFile* diagFile = 0;
-  if (fSaveHistograms)
-    diagFile = TFile::Open(fDiagnosticsFilename.Data(),"RECREATE");
-
-  
-  
-  
+  // if (fSaveHistograms)
+  //  diagFile = TFile::Open(fDiagnosticsFilename.Data(),"RECREATE");
   
   reader->Reset();
   fRunno = reader->GetRunNumber();
@@ -205,7 +247,12 @@ void AliFMDBaseDA::Run(AliRawReader* reader)
   Bool_t sodread = kFALSE;
   
   for(Int_t i=0;i<3;i++) {
-    reader->NextEvent(); // Read Start-of-Run / Start-of-Files event
+    if (!reader->NextEvent()) { 
+      // Read Start-of-Run / Start-of-Files event
+      AliWarning(Form("Failed to read the %d%s event",
+                     i+1, (i == 0 ? "st" : (i == 1 ? "nd" : "rd"))));
+      break;
+    }
     
     UInt_t eventType = reader->GetType();
     if(eventType == AliRawEventHeaderBase::kStartOfData || 
@@ -219,37 +266,65 @@ void AliFMDBaseDA::Run(AliRawReader* reader)
   }
   
   InitContainer(diagFile);
+  if (AliLog::GetDebugLevel("FMD","") >= 3) { 
+    fDetectorArray.ls();
+  }
   
   if(!sodread) 
     AliWarning("No SOD event detected!");
   
   int lastProgress = 0;
   
-  
-  
-  for(Int_t n =1;n <= GetRequiredEvents(); n++) {
-    if(!reader->NextEvent()) continue;
+  for(Int_t i = 0; i< 3;i++) fNEventsPerDetector[i] = 0;
+
+  for(Int_t n = 1; !HaveEnough(n); n++) {
+    AliInfoF("Get the next event %d", n);
+    if(!reader->NextEvent()) { n--; continue; }
+    UInt_t eventType = reader->GetType();
+    AliInfoF("Event type is %d", eventType);
+    if(eventType != AliRawEventHeaderBase::kPhysicsEvent) { n--; continue; }
+
     SetCurrentEvent(n);
     digitArray->Clear();
     fmdReader->ReadAdcs(digitArray);
     
+    Bool_t seen[] = { false, false, false };
     for(Int_t i = 0; i<digitArray->GetEntriesFast();i++) {
       AliFMDDigit* digit = static_cast<AliFMDDigit*>(digitArray->At(i));
-      fSeenDetectors[digit->Detector()-1] = kTRUE;
-      FillChannels(digit);
+      UShort_t det = digit->Detector();
+      fSeenDetectors[det-1] = true;
+      seen[det-1]           = true;
+
+      // Only fill if we do not have enough for this detector 
+      if (Int_t(fNEventsPerDetector[det-1]) < GetRequiredEvents()) 
+       FillChannels(digit);
     }
     
-   
+    for(Int_t i = 0; i< 3;i++) 
+      if (seen[i]) (fNEventsPerDetector[i])++;
+      
     FinishEvent();
     
-    int progress = int((n *100)/ GetRequiredEvents()) ;
+    Int_t nReq = GetRequiredEvents();
+    AliInfoF("%9d: %6d/%6d %6d/%6d %6d/%6d", n, 
+            fNEventsPerDetector[0], nReq,
+            fNEventsPerDetector[1], nReq,
+            fNEventsPerDetector[2], nReq);
+
+    int progress = GetProgress(n);
     if (progress <= lastProgress) continue;
     lastProgress = progress;
     std::cout << "Progress: " << lastProgress << " / 100 " << std::endl;
+
+    if (AliLog::GetDebugLevel("FMD","") >= 0) { 
+    }
     
   }
   
-  AliInfo(Form("Looped over %d events",GetCurrentEvent()));
+  AliInfoF("Looped over %d events (%d,%d,%d)",GetCurrentEvent(),
+          fNEventsPerDetector[0], 
+          fNEventsPerDetector[1], 
+          fNEventsPerDetector[2]);
   WriteHeaderToFile();
   
   for(UShort_t det=1;det<=3;det++) {
@@ -270,8 +345,8 @@ void AliFMDBaseDA::Run(AliRawReader* reader)
        }
        std::cout << '.' << std::flush;
       }
-      if(fSaveHistograms)
-       diagFile->Flush();
+      // if(fSaveHistograms)
+      // diagFile->Flush();
       std::cout << "done" << std::endl;
     }
   }
@@ -284,7 +359,9 @@ void AliFMDBaseDA::Run(AliRawReader* reader)
   Terminate(diagFile);
     
   if(fSaveHistograms ) {
-    
+    diagFile = TFile::Open(fDiagnosticsFilename.Data(),"RECREATE");
+    fDetectorArray.Write("FMD", TObject::kSingleKey);
+    fSummaries.Write();
     AliInfo("Closing diagnostics file - please wait ...");
     // diagFile->Write();
     diagFile->Close();
@@ -296,16 +373,13 @@ void AliFMDBaseDA::Run(AliRawReader* reader)
 
 void AliFMDBaseDA::InitContainer(TDirectory* diagFile)
 {
-  //Prepare container for diagnostics
-  TObjArray* detArray;
-  TObjArray* ringArray;
-  TObjArray* sectorArray;
-    
+  //Prepare container for diagnostics    
   TDirectory* savDir   = gDirectory;
 
   for(UShort_t det=1;det<=3;det++) {
-    detArray = new TObjArray();
+    TObjArray* detArray = new TObjArray(det == 1 ? 1 : 2);
     detArray->SetOwner();
+    detArray->SetName(GetDetectorPath(det, false));
     fDetectorArray.AddAtAndExpand(detArray,det);
 
     TDirectory* detDir = 0;
@@ -319,8 +393,9 @@ void AliFMDBaseDA::InitContainer(TDirectory* diagFile)
       Char_t   ring = (ir == 0 ? 'O' : 'I');
       UShort_t nsec = (ir == 0 ? 40  : 20);
       UShort_t nstr = (ir == 0 ? 256 : 512);
-      ringArray = new TObjArray();
+      TObjArray* ringArray = new TObjArray(nsec);
       ringArray->SetOwner();
+      ringArray->SetName(GetRingPath(det, ring, false));
       detArray->AddAtAndExpand(ringArray,ir);
 
 
@@ -332,8 +407,9 @@ void AliFMDBaseDA::InitContainer(TDirectory* diagFile)
       
 
       for(UShort_t sec =0; sec < nsec;  sec++)  {
-       sectorArray = new TObjArray();
+       TObjArray* sectorArray = new TObjArray(nstr);
        sectorArray->SetOwner();
+       sectorArray->SetName(GetSectorPath(det, ring, sec, false));
        ringArray->AddAtAndExpand(sectorArray,sec);
 
 
@@ -348,8 +424,13 @@ void AliFMDBaseDA::InitContainer(TDirectory* diagFile)
            secDir->cd();
            secDir->mkdir(GetStripPath(det, ring, sec, strip, kFALSE));
          }
-         AddChannelContainer(sectorArray, det, ring, sec, strip);
+         TObjArray* stripArray = new TObjArray(0);
+         stripArray->SetOwner(true);
+         stripArray->SetName(GetStripPath(det, ring, sec, strip, false));
+         sectorArray->AddAtAndExpand(stripArray, strip);
+         AddChannelContainer(stripArray, det, ring, sec, strip);
        }
+       AddSectorSummary(sectorArray, det, ring, sec, nstr);
       }
     }
   }
@@ -498,6 +579,15 @@ TH2*
 AliFMDBaseDA::MakeSummaryHistogram(const char* prefix, const char* title, 
                                   UShort_t d, Char_t r) 
 {
+  // 
+  // Utility function for defining summary histograms 
+  // 
+  // Parameters:
+  //    det    Detector 
+  //    ring   Ring identifier 
+  //    prefix Histogram prefix 
+  //    title  Histogram title 
+  //
   Int_t nX = ((d == 1 || r == 'I' || r == 'i') ?  20 :  40);
   Int_t nY = ((d == 1 || r == 'I' || r == 'i') ? 512 : 256);
   
@@ -506,12 +596,285 @@ AliFMDBaseDA::MakeSummaryHistogram(const char* prefix, const char* title,
                      nX, -0.5, nX-0.5, nY, -0.5, nY-0.5);
   ret->SetXTitle("Sector #");
   ret->SetYTitle("Strip #");
-
+  ret->SetDirectory(0);
   // if (!fSummaries) fSummaries = new TObjArray;
   fSummaries.Add(ret);
   return ret;
 }
 
+//_____________________________________________________________________ 
+TObjArray*
+AliFMDBaseDA::GetDetectorArray(UShort_t det)
+{
+  if (det < 1 || det > 3) { 
+    AliErrorF("Detector index %d out of bounds", det);
+    return 0;
+  }
+  return static_cast<TObjArray*>(fDetectorArray.At(det));
+}
+//_____________________________________________________________________ 
+TObjArray*
+AliFMDBaseDA::GetRingArray(UShort_t det, Char_t ring)
+{
+  Int_t idx = (ring == 'O' || ring == 'o' ? 0 : 1);
+  TObjArray* array = GetDetectorArray(det);
+  if (!array) return 0;
+  array = static_cast<TObjArray*>(array->At(idx));
+  if (!array) AliErrorF("No ring array for FMD%d%c (%d)", det, ring, idx);
+  return array;
+}
+//_____________________________________________________________________ 
+TObjArray*
+AliFMDBaseDA::GetSectorArray(UShort_t det, Char_t ring, UShort_t sector)
+{
+  TObjArray* array = GetRingArray(det,ring);
+  if (!array) return 0;
+  array = static_cast<TObjArray*>(array->At(sector));
+  if (!array) AliErrorF("No sector array for FMD%d%c[%02d]", det, ring, sector);
+  return array;
+}
+//_____________________________________________________________________ 
+TObjArray*
+AliFMDBaseDA::GetStripArray(UShort_t det, Char_t ring, 
+                           UShort_t sector, UShort_t strip)
+{
+  TObjArray* array = GetSectorArray(det,ring,sector);
+  if (!array) return 0;
+  array = static_cast<TObjArray*>(array->At(strip));
+  if (!array) AliErrorF("No strip array for FMD%d%c[%02d,%03d]", 
+                       det, ring, sector, strip);
+  return array;
+}
+
+//=====================================================================
+AliFMDBaseDA::Runner::Runner()
+  : fReader(0),
+    fDiagFile(""), 
+    fDiag(false),
+    fAll(false)
+{}
+
+//_____________________________________________________________________ 
+void
+AliFMDBaseDA::Runner::AddHandlers()
+{
+  gROOT->GetPluginManager()->AddHandler("TVirtualStreamerInfo",
+                                       "*",
+                                       "TStreamerInfo",
+                                       "RIO",
+                                       "TStreamerInfo()");
+  gROOT->GetPluginManager()->AddHandler("ROOT::Math::Minimizer", "Minuit", 
+                                       "TMinuitMinimizer",
+                                       "Minuit", 
+                                       "TMinuitMinimizer(const char *)");
+  gROOT->GetPluginManager()->AddHandler("ROOT::Math::Minimizer", 
+                                       "GSLMultiMin", 
+                                       "ROOT::Math::GSLMinimizer",
+                                       "MathMore", 
+                                       "GSLMinimizer(const char *)");
+  gROOT->GetPluginManager()->AddHandler("ROOT::Math::Minimizer", 
+                                       "GSLMultiFit", 
+                                       "ROOT::Math::GSLNLSMinimizer",
+                                       "MathMore", "GSLNLSMinimizer(int)");
+  gROOT->GetPluginManager()->AddHandler("ROOT::Math::Minimizer", 
+                                       "GSLSimAn", 
+                                       "ROOT::Math::GSLSimAnMinimizer",
+                                       "MathMore", 
+                                       "GSLSimAnMinimizer(int)");
+  gROOT->GetPluginManager()->AddHandler("ROOT::Math::Minimizer", 
+                                       "Linear", 
+                                       "TLinearMinimizer",
+                                       "Minuit", 
+                                       "TLinearMinimizer(const char *)");
+  gROOT->GetPluginManager()->AddHandler("ROOT::Math::Minimizer", 
+                                       "Fumili", 
+                                       "TFumiliMinimizer",
+                                       "Fumili", 
+                                       "TFumiliMinimizer(int)");
+}
+//_____________________________________________________________________ 
+void
+AliFMDBaseDA::Runner::ShowUsage(std::ostream& o, const char* progname)
+{
+  o << "Usage: " << progname << " SOURCE [OPTIONS]\n\n"
+    << "Options:\n"
+    << "   -h,--help                Show this help\n"
+    << "   -d,--diagnostics[=FILE]  Write diagnostics to file\n"
+    << "   -D,--debug=LEVEL         Set the debug level\n"
+    << "   -A,--all                 Try to get data from all detectors\n\n"
+    << "SOURCE is one of\n"
+    << " * FILE.raw                Raw data file\n"
+    << " * FILE.root               ROOT'ified raw data file\n"
+    << " * collection://FILE.root  Event list in a ROOT file\n"
+    << " * collection://FILE       File containing list of ROOT files\n"
+    << " * ^FMD                    Monitor source\n"
+    << "There are other options too.  Check the AliRawReader docs\n"
+    << std::endl;
+}
+//_____________________________________________________________________ 
+namespace {
+  Bool_t ExtractValue(const TString& arg, TString& val)
+  {
+    val = "";
+    Int_t eq = arg.Index("=");
+    if (eq == kNPOS) return false;
+    
+    val = arg(eq+1, arg.Length()-eq-1);
+    return true;
+  }
+}
+      
+//_____________________________________________________________________ 
+Int_t
+AliFMDBaseDA::Runner::Init(int argc, char** argv)
+{
+  AddHandlers();
+
+  // --- Process the command line ------------------------------------
+  TString source;
+  Int_t   debugLevel  = 0;
+  Bool_t  help        = false;
+
+  for (int i = 1; i < argc; i++) { 
+    TString arg(argv[i]);
+    Bool_t  badOption   = false;
+    
+    if (arg[0] == '-') { // It's an option 
+      if (arg[1] == '-') { // It's a long option 
+       TString val;
+       if      (arg.EqualTo("--help"))     help = true; 
+       else if (arg.BeginsWith("--debug")) {
+         if (ExtractValue(arg, val))
+           debugLevel = val.Atoi();
+       }
+       else if (arg.BeginsWith("--diagnostics")) { 
+         fDiag = true;
+         if (ExtractValue(arg, val)) 
+           fDiagFile = val;
+       }
+       else if (arg.EqualTo("--all"))  fAll = true;
+       else badOption = true;
+      }
+      else { // Short option 
+       TString next(i < argc-1 ? argv[i+1] : "");
+       switch (arg[1]) { 
+       case 'h': help = true; break;
+       case 'd': fDiag = true; 
+         if (!next.IsNull() && next[0] != '-') {
+           fDiagFile = next;
+           i++;
+         }
+         break;
+       case 'D': 
+         if (!next.IsNull() && next[0] != '-') {
+           debugLevel = next.Atoi();
+           i++;
+         }
+         break;
+       case 'A': fAll = true ; break ;
+       default: badOption = true;
+       }
+      } // End of options
+      if (badOption) { 
+       std::cerr << argv[0] << ": Unknown option " << argv[i] 
+                 << std::endl;
+       return -1;
+      }
+    }
+    else { // source or compatibility debug level 
+      if (source.IsNull()) source = arg;
+      else                 debugLevel = arg.Atoi();
+    }
+  }
+  
+  // --- Check if help was requested ---------------------------------
+  if (help) { 
+    ShowUsage(std::cout, argv[0]);
+    return 1;
+  }
+
+  // --- Check if we have a source -----------------------------------
+  if (source.IsNull()) { 
+    std::cerr << "No source given" << std::endl;
+    return -2;
+  }
+
+  // --- Initialize various things -----------------------------------
+  AliFMDParameters::Instance()->Init(kFALSE,0);
+
+  //This will only work for FDR 1 data. When newer data becomes
+  //available the ! must be removed!
+  Bool_t old = kTRUE;
+  AliFMDParameters::Instance()->UseCompleteHeader(old);
+  
+  AliLog::EnableDebug(debugLevel > 0);
+  AliLog::SetModuleDebugLevel("FMD", debugLevel);
+
+  // --- Make our reader ---------------------------------------------
+  fReader = AliRawReader::Create(source);
+  if (!fReader) { 
+    std::cerr << "Failed to make raw reader for source \"" << source 
+             << "\"" << std::endl;
+    return -3;
+  }
+  return 0;
+}
+
+//_____________________________________________________________________ 
+Int_t
+AliFMDBaseDA::Runner::RunNumber() const
+{ 
+  if (!fReader) return -1;
+  return fReader->GetRunNumber(); 
+}
+
+//_____________________________________________________________________ 
+void
+AliFMDBaseDA::Runner::Exec(AliFMDBaseDA& da)
+{
+  TStopwatch timer;
+  timer.Start();
+
+  da.SetSaveDiagnostics(fDiag || !fDiagFile.IsNull());
+  da.SetTryAll(fAll);
+  if (!fDiagFile.IsNull()) da.SetDiagnosticsFilename(fDiagFile);
+
+  da.Run(fReader);
+
+  timer.Stop();
+  timer.Print();
+
+#ifdef ALI_AMORE
+  try { 
+    amore::da::AmoreDA myAmore(amore::da::AmoreDA::kSender);
+
+    for (UShort_t det = 1; det <= 3; det++) {
+      if (!da.HasSeenDetector(det)) continue;
+      TObject* runNo = new TObject;
+      runNo->SetUniqueID(fReader->GetRunNumber());
+      myAmore.Send(Form("gainRunNoFMD%d", det), runNo);
+    }
+    
+    TIter     next(&da.GetSummaries());
+    TObject*  obj = 0;
+    while ((obj = next())) 
+      myAmore.Send(obj->GetName(), obj);
+    
+  }
+  catch (exception& e) {
+    cerr << "Failed to make AMORE instance: " << e.what() << endl;
+  }
+                              
+#endif
+}
+
+
+  
+  
+  
+//_____________________________________________________________________ 
+
 //_____________________________________________________________________ 
 //
 // EOF