Method added to start reconstruction from raw data
authorbnandi <bnandi@f7af4fe6-9843-0410-8265-dc069ae4e863>
Thu, 29 Jul 2004 11:53:10 +0000 (11:53 +0000)
committerbnandi <bnandi@f7af4fe6-9843-0410-8265-dc069ae4e863>
Thu, 29 Jul 2004 11:53:10 +0000 (11:53 +0000)
PMD/AliPMDClusterFinder.cxx
PMD/AliPMDClusterFinder.h
PMD/AliPMDDigits2Recpoints.C
PMD/AliPMDReconstructor.cxx
PMD/AliPMDReconstructor.h

index 5c9f370..b139b29 100644 (file)
@@ -104,6 +104,7 @@ void AliPMDClusterFinder::Digits2RecPoints(Int_t ievt)
 
   fRunLoader->GetEvent(ievt);
   //cout << " ***** Beginning::Digits2RecPoints *****" << endl;
+
   fTreeD = fPMDLoader->TreeD();
   if (fTreeD == 0x0)
     {
@@ -114,6 +115,7 @@ void AliPMDClusterFinder::Digits2RecPoints(Int_t ievt)
   branch->SetAddress(&fDigits);
 
   ResetRecpoint();
+
   fTreeR = fPMDLoader->TreeR();
   if (fTreeR == 0x0)
     {
@@ -125,7 +127,7 @@ void AliPMDClusterFinder::Digits2RecPoints(Int_t ievt)
   fTreeR->Branch("PMDRecpoint", &fRecpoints, bufsize); 
 
   Int_t nmodules = (Int_t) fTreeD->GetEntries();
-  
+
   for (Int_t imodule = 0; imodule < nmodules; imodule++)
     {
       ResetCellADC();
@@ -160,7 +162,7 @@ void AliPMDClusterFinder::Digits2RecPoints(Int_t ievt)
          clusdata[2] = pmdcl->GetClusADC();
          clusdata[3] = pmdcl->GetClusCells();
          clusdata[4] = pmdcl->GetClusRadius();
-         
+
          AddRecPoint(idet,ismn,clusdata);
        }
       pmdcont->Clear();
@@ -184,16 +186,12 @@ void AliPMDClusterFinder::Digits2RecPoints(Int_t ievt)
 
 void AliPMDClusterFinder::Digits2RecPoints(Int_t ievt, AliRawReader *rawReader)
 {
-  // Converts digits to recpoints after running clustering
-  // algorithm on CPV plane and PREshower plane
+  // Converts RAW data to recpoints after running clustering
+  // algorithm on CPV and PREshower plane
   //
 
   Float_t  clusdata[5];
 
-
-  rawReader->Reset();
-  AliPMDRawStream pmdinput(rawReader);
-
   TObjArray *pmdcont = new TObjArray();
   AliPMDClustering *pmdclust = new AliPMDClustering();
   pmdclust->SetDebug(fDebug);
@@ -202,80 +200,147 @@ void AliPMDClusterFinder::Digits2RecPoints(Int_t ievt, AliRawReader *rawReader)
   fRunLoader->GetEvent(ievt);
 
   ResetRecpoint();
+
   fTreeR = fPMDLoader->TreeR();
   if (fTreeR == 0x0)
     {
       fPMDLoader->MakeTree("R");
       fTreeR = fPMDLoader->TreeR();
     }
-
   Int_t bufsize = 16000;
   fTreeR->Branch("PMDRecpoint", &fRecpoints, bufsize); 
-  const Int_t kDet = 2;
-  const Int_t kSMN = 24;
+
+  const Int_t kDDL = 6;
   const Int_t kRow = 48;
   const Int_t kCol = 96;
-  Int_t preADC[kSMN][kRow][kCol];
-  Int_t cpvADC[kSMN][kRow][kCol];
 
-  for (Int_t i = 0; i < kSMN; i++)
+  Int_t idet = 0;
+  Int_t iSMN = 0;
+  
+  for (Int_t indexDDL = 0; indexDDL < kDDL; indexDDL++)
     {
-      for (Int_t j = 0; j < kRow; j++)
+      if (indexDDL < 4)
        {
-         for (Int_t k = 0; k < kCol; k++)
-           {
-             preADC[i][j][k] = 0;
-             cpvADC[i][j][k] = 0;
-           }
+         iSMN = 6;
        }
-    }
-  ResetCellADC();
-
-  while(pmdinput.Next())
-    {
-      Int_t det = pmdinput.GetDetector();
-      Int_t smn = pmdinput.GetSMN();
-      //Int_t mcm = pmdinput.GetMCM();
-      //Int_t chno = pmdinput.GetChannel();
-      Int_t row = pmdinput.GetRow();
-      Int_t col = pmdinput.GetColumn();
-      Int_t sig = pmdinput.GetSignal();
-
-      //cout << " det = " << det << " smn = " << smn 
-      //   << " row = " << row << " col = " << col
-      //   << " sig = " << sig << endl;
-
-      if (det == 0)
+      else if (indexDDL >= 4)
        {
-         preADC[smn][row][col] = sig;
+         iSMN = 12;
        }
-      else if (det == 1)
+      Int_t ***precpvADC;
+      precpvADC = new int **[iSMN];
+      for (Int_t i=0; i<iSMN; i++) precpvADC[i] = new int *[kRow];
+      for (Int_t i=0; i<iSMN;i++)
        {
-         cpvADC[smn][row][col] = sig;
+         for (Int_t j=0; j<kRow; j++) precpvADC[i][j] = new int [kCol];
        }
+      for (Int_t i = 0; i < iSMN; i++)
+       {
+         for (Int_t j = 0; j < kRow; j++)
+           {
+             for (Int_t k = 0; k < kCol; k++)
+               {
+                 precpvADC[i][j][k] = 0;
+               }
+           }
+       }
+      ResetCellADC();
+      rawReader->Reset();
+      AliPMDRawStream pmdinput(rawReader);
+      rawReader->Select(12, indexDDL, indexDDL);
+      while(pmdinput.Next())
+       {
+         Int_t det = pmdinput.GetDetector();
+         Int_t smn = pmdinput.GetSMN();
+         //Int_t mcm = pmdinput.GetMCM();
+         //Int_t chno = pmdinput.GetChannel();
+         Int_t row = pmdinput.GetRow();
+         Int_t col = pmdinput.GetColumn();
+         Int_t sig = pmdinput.GetSignal();
+         
+         Int_t indexsmn = 0;
 
-    } // while loop
-
-  for (Int_t idet = 0; idet < kDet; idet++)
-    {
-      for (Int_t ismn = 0; ismn < kSMN; ismn++)
+         if (indexDDL < 4)
+           {
+             if (det != 0)
+               printf(" *** DDL %d and Detector NUMBER %d NOT MATCHING *** ",
+                      indexDDL, det);
+             indexsmn = smn - indexDDL * 6;
+           }
+         else if (indexDDL == 4)
+           {
+             if (det != 1)
+               printf(" *** DDL %d and Detector NUMBER %d NOT MATCHING *** ",
+                      indexDDL, det);
+             if (smn < 6)
+               {
+                 indexsmn = smn;
+               }
+             else if (smn >= 12 && smn < 18)
+               {
+                 indexsmn = smn - 6;
+               }
+           }
+         else if (indexDDL == 5)
+           {
+             if (det != 1)
+               printf(" *** DDL %d and Detector NUMBER %d NOT MATCHING *** ",
+                      indexDDL, det);
+             if (smn >= 6 && smn < 12)
+               {
+                 indexsmn = smn - 6;
+               }
+             else if (smn >= 18 && smn < 24)
+               {
+                 indexsmn = smn - 12;
+               }
+           }         
+         precpvADC[indexsmn][row][col] = sig;
+       } // while loop
+
+      Int_t ismn = 0;
+      for (Int_t indexsmn = 0; indexsmn < iSMN; indexsmn++)
        {
+         ResetCellADC();
          for (Int_t irow = 0; irow < kRow; irow++)
            {
              for (Int_t icol = 0; icol < kCol; icol++)
                {
-                 if (idet == 0)
-                   {
-                     fCellADC[irow][icol] = 
-                       (Double_t) preADC[ismn][irow][icol];
-                   }
-                 else if (idet == 1)
-                   {
-                     fCellADC[irow][icol] = 
-                       (Double_t) cpvADC[ismn][irow][icol];
-                   }
+                 fCellADC[irow][icol] = 
+                   (Double_t) precpvADC[indexsmn][irow][icol];
                } // row
            }     // col
+         if (indexDDL < 4)
+           {
+             ismn = indexsmn + indexDDL * 6;
+             idet = 0;
+           }
+         else if (indexDDL == 4)
+           {
+             if (indexsmn < 6)
+               {
+                 ismn = indexsmn;
+               }
+             else if (indexsmn >= 6 && indexsmn < 12)
+               {
+                 ismn = indexsmn + 6;
+               }
+             idet = 1;
+           }
+         else if (indexDDL == 5)
+           {
+             if (indexsmn < 6)
+               {
+                 ismn = indexsmn + 6;
+               }
+             else if (indexsmn >= 6 && indexsmn < 12)
+               {
+                 ismn = indexsmn + 12;
+               }
+             idet = 1;
+           }
+
+
          pmdclust->DoClust(idet,ismn,fCellADC,pmdcont);
          Int_t nentries1 = pmdcont->GetEntries();
          //      cout << " nentries1 = " << nentries1 << endl;
@@ -290,18 +355,27 @@ void AliPMDClusterFinder::Digits2RecPoints(Int_t ievt, AliRawReader *rawReader)
              clusdata[2] = pmdcl->GetClusADC();
              clusdata[3] = pmdcl->GetClusCells();
              clusdata[4] = pmdcl->GetClusRadius();
-             
+
              AddRecPoint(idet,ismn,clusdata);
            }
          pmdcont->Clear();
          
          fTreeR->Fill();
          ResetRecpoint();
-         
+
+
        } // smn
-    }     // det (pre, cpv)
 
+      for (Int_t i=0; i<iSMN; i++)
+       {
+         for (Int_t j=0; j<kRow; j++) delete [] precpvADC[i][j];
+       }
+      for (Int_t i=0; i<iSMN; i++) delete [] precpvADC[i];
+      delete precpvADC;
+    } // DDL Loop
+  
   ResetCellADC();
+  
   fPMDLoader = fRunLoader->GetLoader("PMDLoader");  
   fPMDLoader->WriteRecPoints("OVERWRITE");
 
@@ -362,6 +436,13 @@ void AliPMDClusterFinder::Load()
   fPMDLoader->LoadRecPoints("recreate");
 }
 // ------------------------------------------------------------------------- //
+void AliPMDClusterFinder::LoadClusters()
+{
+  // Load all the *.root files
+  //
+  fPMDLoader->LoadRecPoints("recreate");
+}
+// ------------------------------------------------------------------------- //
 void AliPMDClusterFinder::UnLoad()
 {
   // Unload all the *.root files
@@ -370,3 +451,10 @@ void AliPMDClusterFinder::UnLoad()
   fPMDLoader->UnloadRecPoints();
 }
 // ------------------------------------------------------------------------- //
+void AliPMDClusterFinder::UnLoadClusters()
+{
+  // Unload all the *.root files
+  //
+  fPMDLoader->UnloadRecPoints();
+}
+// ------------------------------------------------------------------------- //
index 3887c37..0c3b881 100644 (file)
@@ -46,7 +46,9 @@ class AliPMDClusterFinder
   void ResetCellADC();
   void ResetRecpoint();
   void Load();
+  void LoadClusters();
   void UnLoad();
+  void UnLoadClusters();
 
  protected:
   AliRunLoader *fRunLoader; // Pointer to Run Loader
@@ -67,7 +69,7 @@ class AliPMDClusterFinder
   static const Int_t fgkCol = 96; // Total number of cols in one unitmodule
   Double_t fCellADC[fgkRow][fgkCol]; // Array containing individual cell ADC
 
-  ClassDef(AliPMDClusterFinder,5) // To run PMD clustering
+  ClassDef(AliPMDClusterFinder,6) // To run PMD clustering
 };
 #endif
 
index 67c917d..fe7d43a 100644 (file)
@@ -30,26 +30,47 @@ void AliPMDDigits2Recpoints(Int_t nevt=1)
   fRunLoader->LoadgAlice();
   gAlice = fRunLoader->GetAliRun();
 
+  printf(" Do you want reconstruction from Digits file or RAW data \n");
+  printf(" If RAW,    type 0 \n");
+  printf(" If Digits, type 1 \n");
+  Int_t itype;
+  cin >> itype;
 
   // Create the PMD Cluster Finder 
   AliPMDClusterFinder *clus = new AliPMDClusterFinder(fRunLoader);
-  
   clus->SetDebug(1);
-  clus->Load();
-
+  if (itype == 0)
+    {
+      clus->Load();
+    }
+  else if (itype == 1)
+    {
+      clus->LoadClusters();
+    }  
 
 
   for (Int_t ievt = 0; ievt < nevt; ievt++)
     {
-      // from digits data
-      //      clus->Digits2RecPoints(ievt);
-
-      // from raw data
-      AliRawReaderFile reader(ievt);
-      clus->Digits2RecPoints(ievt, &reader);
+      if (itype == 0)
+       {
+         // from digits data
+         clus->Digits2RecPoints(ievt);
+       }
+      else if (itype == 1)
+       {
+         // from raw data
+         AliRawReaderFile reader(ievt);
+         clus->Digits2RecPoints(ievt, &reader);
+       }
+    }
+  if (itype == 0)
+    {
+      clus->UnLoad();
+    }
+  else if (itype == 1)
+    {
+      clus->UnLoadClusters();
     }
-  clus->UnLoad();
-
   timer.Stop();
   timer.Print();
 }
index 1851814..1b5d512 100644 (file)
 #include "AliRun.h"
 #include "AliPMDClusterFinder.h"
 #include "AliPMDtracker.h"
+#include "AliRawReader.h"
 #include "AliESDPmdTrack.h"
 #include "AliESD.h"
 
-
 ClassImp(AliPMDReconstructor)
 
 
 //_____________________________________________________________________________
 void AliPMDReconstructor::Reconstruct(AliRunLoader* runLoader) const
 {
-// reconstruct clusters
+// reconstruct clusters from digits file
 
   AliPMDClusterFinder *pmdClus = new AliPMDClusterFinder(runLoader);
   pmdClus->Load();
+  pmdClus->SetDebug(1);
   for (Int_t iEvent = 0; iEvent < runLoader->GetNumberOfEvents(); iEvent++)
     {
       pmdClus->Digits2RecPoints(iEvent);
@@ -47,6 +48,25 @@ void AliPMDReconstructor::Reconstruct(AliRunLoader* runLoader) const
   delete pmdClus;
 
 }
+//_____________________________________________________________________________
+void AliPMDReconstructor::Reconstruct(AliRunLoader* runLoader,
+                                     AliRawReader *rawReader) const
+{
+// reconstruct clusters from Raw Data
+
+  AliPMDClusterFinder pmdClus(runLoader);
+  pmdClus.LoadClusters();
+
+  Int_t iEvent = 0;
+  while (rawReader->NextEvent()) {
+    pmdClus.SetDebug(1);
+    pmdClus.Digits2RecPoints(iEvent,rawReader);
+    
+    iEvent++;
+  }
+  pmdClus.UnLoadClusters();
+  
+}
 
 // ------------------------------------------------------------------------ //
 
index 5e462c5..3a36dc7 100644 (file)
@@ -7,12 +7,14 @@
 
 class AliPMDReconstructor: public AliReconstructor {
 public:
-  virtual void         Reconstruct(AliRunLoader* runLoader) const;
-  virtual void         FillESD(AliRunLoader* runLoader, AliESD* esd) const;
+  virtual void Reconstruct(AliRunLoader* runLoader) const;
+  virtual void Reconstruct(AliRunLoader* runLoader,
+                          AliRawReader *rawReader) const;
+  virtual void FillESD(AliRunLoader* runLoader, AliESD* esd) const;
 
 private:
 
-  ClassDef(AliPMDReconstructor, 1)   // class for the PMD reconstruction
+  ClassDef(AliPMDReconstructor, 2)   // class for the PMD reconstruction
 };
 
 #endif