New MUON Data container and MUON loader

diff --git a/MUON/AliMUONData.cxx b/MUON/AliMUONData.cxx
new file mode 100644 (file)
index 0000000..16ca8aa
--- /dev/null
+++ b/MUON/AliMUONData.cxx
@@ -0,0 +1,388 @@
+
+
+//Root includes
+
+//AliRoot includes
+#include "AliMUONData.h"
+#include "AliMUONDigit.h"
+#include "AliMUONHit.h"
+#include "AliMUONLocalTrigger.h"
+#include "AliMUONGlobalTrigger.h"
+#include "AliMUONRawCluster.h"
+
+ClassImp(AliMUONData)
+ 
+//_____________________________________________________________________________
+AliMUONData::AliMUONData():TNamed()
+{
+  fLoader        = 0x0;
+  fHits          = 0x0;    // One event in treeH per primary track
+  fDigits        = 0x0;  // One event in treeH per detection plane
+  fRawClusters   = 0x0; //One event in TreeR/RawclusterBranch per tracking detection plane
+  fGlobalTrigger = 0x0; //! List of Global Trigger 1st event in TreeR/GlobalTriggerBranch
+  fLocalTrigger  = 0x0;  //! List of Local Trigger, 1st event in TreeR/LocalTriggerBranch       
+//default constructor
+}
+//_____________________________________________________________________________
+AliMUONData::AliMUONData(AliLoader * loader, const char* name, const char* title):
+  TNamed(name,title)
+{
+  fLoader        = loader;
+  fHits          = new TClonesArray("AliMUONHit",1000);
+  fNhits         = 0;
+  fDigits        = new TObjArray(AliMUONConstants::NCh());
+  fNdigits       = new Int_t[AliMUONConstants::NCh()];
+  for (Int_t iDetectionPlane=0; iDetectionPlane<AliMUONConstants::NCh() ;iDetectionPlane++) {
+    fDigits->AddAt(new TClonesArray("AliMUONDigit",10000),iDetectionPlane); 
+    fNdigits[iDetectionPlane]=0;
+  }
+  fRawClusters   = new TObjArray(AliMUONConstants::NTrackingCh());
+  fNrawclusters  = new Int_t[AliMUONConstants::NTrackingCh()];
+  for (Int_t iDetectionPlane=0; iDetectionPlane<AliMUONConstants::NTrackingCh();iDetectionPlane++) {
+    fRawClusters->AddAt(new TClonesArray("AliMUONRawCluster",10000),iDetectionPlane); 
+    fNrawclusters[iDetectionPlane]=0;
+  }
+  fGlobalTrigger = new TClonesArray("AliMUONGlobalTrigger",1);    
+  fNglobaltrigger =0;
+  fLocalTrigger  = new TClonesArray("AliMUONLocalTrigger",234);   
+  fNlocaltrigger = 0;
+  //default constructor
+}
+//_____________________________________________________________________________
+AliMUONData::AliMUONData(const AliMUONData& rMUONData):TNamed(rMUONData)
+{
+  // Dummy copy constructor
+  ;
+}
+//_____________________________________________________________________________
+AliMUONData::~AliMUONData()
+{
+  if (fHits) {
+    fHits->Delete();
+    delete fHits;
+  }
+  if (fDigits) {
+    fDigits->Delete();
+    delete fDigits;
+  }
+  if (fRawClusters) {
+    fRawClusters->Delete();
+    delete fRawClusters;
+  }
+  if (fGlobalTrigger){
+    fGlobalTrigger->Delete();
+    delete fGlobalTrigger;
+  }  
+  if (fLocalTrigger){
+    fLocalTrigger->Delete();
+    delete fLocalTrigger;
+  }
+  //detructor 
+}
+//_____________________________________________________________________________
+void AliMUONData::AddDigit(Int_t id, Int_t *tracks, Int_t *charges, Int_t *digits)
+{
+  //
+  // Add a MUON digit to the list of Digits of the detection plane id
+  //
+  TClonesArray &ldigits = * ( (TClonesArray*) fDigits->At(id) ); 
+  new(ldigits[fNdigits[id]++]) AliMUONDigit(tracks,charges,digits);
+}
+//_____________________________________________________________________________
+void AliMUONData::AddGlobalTrigger(Int_t *singlePlus, Int_t *singleMinus,
+                                  Int_t *singleUndef,
+                                  Int_t *pairUnlike, Int_t *pairLike)
+{
+  // add a MUON Global Trigger to the list (only one GlobalTrigger per event !)
+  TClonesArray &globalTrigger = *fGlobalTrigger;
+  new(globalTrigger[fNglobaltrigger++]) 
+    AliMUONGlobalTrigger(singlePlus, singleMinus,  singleUndef, pairUnlike, pairLike);
+}
+//_____________________________________________________________________________
+void AliMUONData::AddHit(Int_t fIshunt, Int_t track, Int_t iChamber, 
+                        Int_t idpart, Float_t X, Float_t Y, Float_t Z, 
+                        Float_t tof, Float_t momentum, Float_t theta, 
+                        Float_t phi, Float_t length, Float_t destep)
+{
+  TClonesArray &lhits = *fHits;
+  new(lhits[fNhits++]) AliMUONHit(fIshunt, track, iChamber, 
+                                 idpart, X, Y, Z, 
+                                 tof, momentum, theta, 
+                                 phi, length, destep);
+}
+//____________________________________________________________________________
+void AliMUONData::AddLocalTrigger(Int_t *localtr)
+{
+  // add a MUON Local Trigger to the list
+  TClonesArray &localTrigger = *fLocalTrigger;
+  new(localTrigger[fNlocaltrigger++]) AliMUONLocalTrigger(localtr);
+}
+//_____________________________________________________________________________
+void AliMUONData::AddRawCluster(Int_t id, const AliMUONRawCluster& c)
+{
+  //
+  // Add a MUON rawcluster to the list in the detection plane id
+  //
+  TClonesArray &lrawcl = *((TClonesArray*) fRawClusters->At(id));
+  new(lrawcl[fNrawclusters[id]++]) AliMUONRawCluster(c);
+}
+//_____________________________________________________________________________
+
+//___________________________________________
+void AliMUONData::MakeBranch(Option_t* option)
+{
+  //
+  // Create Tree branches for the MUON.
+  //
+  const Int_t kBufferSize = 4000;
+  char branchname[30];
+  
+  const char *cH   = strstr(option,"H");
+  const char *cD   = strstr(option,"D");   // Digits branches in TreeD
+  const char *cRC  = strstr(option,"RC");  // RawCluster branches in TreeR
+  const char *cGLT = strstr(option,"GLT"); // Global and Local Trigger branches in TreeR
+  const char *cRT  = strstr(option,"RT");  // Reconstructed Track in TreeT
+  const char *cRP  = strstr(option,"RP");  // Reconstructed Particle in TreeP
+  
+  TBranch * branch = 0x0;
+  
+  // Creating Branches for Hits
+  if (TreeH() && cH) {
+    if (fHits == 0x0)  fHits = new TClonesArray("AliMUONHit",1000);
+    fNhits = 0;
+    sprintf(branchname,"%sHits",GetName());  
+    branch = TreeH()->GetBranch(branchname);
+    if (branch) {  
+      Info("MakeBranch","Branch %s is already in tree.",GetName());
+      return ;
+    }
+    branch = TreeH()->Branch(branchname,&fHits,kBufferSize);
+    Info("MakeBranch","Making Branch %s for hits \n",branchname);
+  }  
+  
+  //Creating Branches for Digits
+  if (TreeD() && cD ) {
+    // one branch for digits per chamber
+    if (fDigits  == 0x0) {
+      fDigits = new TObjArray(AliMUONConstants::NCh());
+      fNdigits = new Int_t[AliMUONConstants::NCh()];
+      for (Int_t iDetectionPlane=0; iDetectionPlane<AliMUONConstants::NCh() ;iDetectionPlane++) {
+       fDigits->AddAt(new TClonesArray("AliMUONDigit",10000),iDetectionPlane); 
+       fNdigits[iDetectionPlane]=0;
+      }
+    }
+    for (Int_t iDetectionPlane=0; iDetectionPlane<AliMUONConstants::NCh() ;iDetectionPlane++) {
+      sprintf(branchname,"%sDigits%d",GetName(),iDetectionPlane+1);
+      branch = 0x0;
+      branch = TreeD()->GetBranch(branchname);
+      if (branch) {  
+       Info("MakeBranch","Branch %s is already in tree.",GetName());
+       return;
+      }
+      branch = TreeD()->Branch(branchname, &((*fDigits)[iDetectionPlane]),kBufferSize);
+      Info("MakeBranch","Making Branch %s for digits in detection plane %d\n",branchname,iDetectionPlane+1);
+      }
+  }
+  
+  if (TreeR() && cRC ) {
+    //  one branch for raw clusters per tracking detection plane
+    //        
+    Int_t i;
+    if (fRawClusters == 0x0) {
+      fRawClusters = new TObjArray(AliMUONConstants::NTrackingCh());
+      fNrawclusters= new Int_t[AliMUONConstants::NTrackingCh()];
+      for (Int_t i=0; i<AliMUONConstants::NTrackingCh();i++) {
+       fRawClusters->AddAt(new TClonesArray("AliMUONRawCluster",10000),i); 
+       fNrawclusters[i]=0;
+      }
+    }
+    
+    for (i=0; i<AliMUONConstants::NTrackingCh() ;i++) {
+      sprintf(branchname,"%sRawClusters%d",GetName(),i+1);     
+      branch = 0x0;
+      branch = TreeR()->GetBranch(branchname);
+      if (branch) {  
+       Info("MakeBranch","Branch %s is already in tree.",GetName());
+       return;
+      }
+      branch = TreeR()->Branch(branchname, &((*fRawClusters)[i]),kBufferSize);
+      Info("MakeBranch","Making Branch %s for rawcluster in detection plane %d\n",branchname,i+1);
+    }
+  }
+
+  if (TreeR() && cGLT ) {
+    //
+    // one branch for global trigger
+    //
+    sprintf(branchname,"%sGlobalTrigger",GetName());
+    branch = 0x0;
+    
+    if (fGlobalTrigger == 0x0) {
+      fGlobalTrigger = new TClonesArray("AliMUONGlobalTrigger",1); 
+      fNglobaltrigger = 0;
+    }
+    branch = TreeR()->GetBranch(branchname);
+    if (branch) {  
+      Info("MakeBranch","Branch %s is already in tree.",GetName());
+      return ;
+    }
+    branch = TreeR()->Branch(branchname, &fGlobalTrigger, kBufferSize);
+    Info("MakeBranch", "Making Branch %s for Global Trigger\n",branchname);
+    
+    //
+    // one branch for local trigger
+    //  
+    sprintf(branchname,"%sLocalTrigger",GetName());
+    branch = 0x0;
+    
+    if (fLocalTrigger == 0x0) {
+      fLocalTrigger  = new TClonesArray("AliMUONLocalTrigger",234);
+      fNlocaltrigger = 0;
+    }
+    branch = TreeR()->GetBranch(branchname);
+    if (branch) {  
+      Info("MakeBranch","Branch %s is already in tree.",GetName());
+      return;
+    }
+    branch = TreeR()->Branch(branchname, &fLocalTrigger, kBufferSize);
+    Info("MakeBranch", "Making Branch %s for Global Trigger\n",branchname);  
+  }
+  
+  if (TreeR() && cRT ) {
+    Info("MakeBranch","Making Branch for TreeT is not yet ready. \n");
+  }
+  if (TreeR() && cRP ) {
+    Info("MakeBranch","Making Branch for TreeP is not yet ready. \n");
+  }
+}
+
+//____________________________________________________________________________
+void AliMUONData::ResetDigits()
+{
+    //
+    // Reset number of digits and the digits array for this detector
+    //
+    if (fDigits == 0x0) return;
+    for ( int i=0;i<AliMUONConstants::NCh();i++ ) {
+      if ((*fDigits)[i])    ((TClonesArray*)fDigits->At(i))->Clear();
+      if (fNdigits)  fNdigits[i]=0;
+    }
+}
+//______________________________________________________________________________
+void AliMUONData::ResetHits()
+{
+  // Reset number of clusters and the cluster array for this detector
+  fNhits   = 0;
+  if (fHits) fHits->Clear();
+}
+//_______________________________________________________________________________
+void AliMUONData::ResetRawClusters()
+{
+    // Reset number of raw clusters and the raw clust array for this detector
+    //
+  for ( int i=0;i<AliMUONConstants::NTrackingCh();i++ ) {
+    if ((*fRawClusters)[i])    ((TClonesArray*)fRawClusters->At(i))->Clear();
+    if (fNrawclusters)  fNrawclusters[i]=0;
+  }
+}
+//_______________________________________________________________________________
+void AliMUONData::ResetTrigger()
+{
+  //  Reset Local and Global Trigger 
+  fNglobaltrigger = 0;
+  if (fGlobalTrigger) fGlobalTrigger->Clear();
+  fNlocaltrigger = 0;
+  if (fLocalTrigger) fLocalTrigger->Clear();
+}
+//_____________________________________________________________________________
+void AliMUONData::SetTreeAddress()
+{
+  // Set branch address for the Hits, Digits, RawClusters, GlobalTrigger and LocalTrigger Tree.
+  char branchname[30];
+  TBranch * branch = 0x0;
+
+  //
+  // Branch address for hit tree
+  if ( TreeH() ) {
+    if (fHits == 0x0) fHits     = new TClonesArray("AliMUONHit",1000);
+    fNhits =0;
+  } 
+  if (TreeH() && fHits) {
+    sprintf(branchname,"%sHits",GetName());  
+    branch = TreeH()->GetBranch(branchname);
+    if (branch) {
+      Info("SetTreeAddress","(%s) Setting for Hits",GetName());
+      branch->SetAddress(&fHits);
+    }
+    else { //can be invoked before branch creation
+      Warning("SetTreeAddress","(%s) Failed for Hits. Can not find branch in tree.",GetName());
+    }
+  }
+  
+  //
+  // Branch address for digit tree
+  if ( TreeD() ) {
+    if (fDigits == 0x0) { 
+      fDigits = new TObjArray(AliMUONConstants::NCh());
+      fNdigits= new Int_t[AliMUONConstants::NCh()];
+      for (Int_t i=0; i<AliMUONConstants::NCh() ;i++) {
+       fDigits->AddAt(new TClonesArray("AliMUONDigit",10000),i); 
+       fNdigits[i]=0;
+      }
+    }
+  }
+
+  if (TreeD() && fDigits) {
+    for (int i=0; i<AliMUONConstants::NCh(); i++) {
+      sprintf(branchname,"%sDigits%d",GetName(),i+1);
+      branch = TreeD()->GetBranch(branchname);
+      if (branch) branch->SetAddress(&((*fDigits)[i]));
+      else Warning("SetTreeAddress","(%s) Failed for Digits Detection plane %d. Can not find branch in tree.",GetName(),i);
+    }
+  }
+  
+  //
+  // Branch address for rawclusters, globaltrigger and local trigger tree
+  if (TreeR() ) {
+    if (fRawClusters == 0x0) {
+      fRawClusters = new TObjArray(AliMUONConstants::NTrackingCh());
+      fNrawclusters= new Int_t[AliMUONConstants::NTrackingCh()];
+      for (Int_t i=0; i<AliMUONConstants::NTrackingCh();i++) {
+       fRawClusters->AddAt(new TClonesArray("AliMUONRawCluster",10000),i); 
+       fNrawclusters[i]=0;
+      }
+    }
+    if (fLocalTrigger == 0x0) {
+      fLocalTrigger  = new TClonesArray("AliMUONLocalTrigger",234);
+    }
+    if (fGlobalTrigger== 0x0) {
+        fGlobalTrigger = new TClonesArray("AliMUONGlobalTrigger",1); 
+    }
+
+  }
+  if ( TreeR()  && fRawClusters ) {
+    for (int i=0; i<AliMUONConstants::NTrackingCh(); i++) {
+      sprintf(branchname,"%sRawClusters%d",GetName(),i+1);
+      if (fRawClusters) {
+       branch = TreeR()->GetBranch(branchname);
+       if (branch) branch->SetAddress(&((*fRawClusters)[i]));
+       else Warning("SetTreeAddress","(%s) Failed for RawClusters Detection plane %d. Can not find branch in tree.",GetName(),i);
+      }
+    }
+  }
+  if ( TreeR()  && fLocalTrigger ) {
+    sprintf(branchname,"%sLocalTrigger",GetName());
+    branch = TreeR()->GetBranch(branchname);
+    if (branch) branch->SetAddress(&fLocalTrigger);
+    else Warning("SetTreeAddress","(%s) Failed for LocalTrigger. Can not find branch in tree.",GetName());
+  }
+  if ( TreeR() && fGlobalTrigger) {
+    sprintf(branchname,"%sGlobalTrigger",GetName());
+    branch = TreeR()->GetBranch(branchname);
+    if (branch) branch->SetAddress(&fGlobalTrigger);
+    else Warning("SetTreeAddress","(%s) Failed for LocalTrigger. Can not find branch in tree.",GetName());
+  }
+}
+//_____________________________________________________________________________
+
+

diff --git a/MUON/AliMUONData.h b/MUON/AliMUONData.h
new file mode 100644 (file)
index 0000000..368e11f
--- /dev/null
+++ b/MUON/AliMUONData.h
@@ -0,0 +1,87 @@
+#ifndef ALIMUONDATA_H
+#define ALIMUONDATA_H
+//
+// AliMUONData
+// Class containing MUON data: hits, digits, rawclusters, globaltrigger, localtrigger, etc ...
+// Gines Martinez, Subatech,  September 2003
+//
+
+#include "TNamed.h"
+
+#include "AliLoader.h" 
+#include "AliMUONConstants.h"
+
+class TClonesArray;
+class TObjArray;
+class TTree;
+
+class AliMUONRawCluster;
+
+//__________________________________________________________________
+/////////////////////////////////////////////////////////////////////
+//                                                                 //
+//  class AliMUONData                                              //
+//                                                                 //
+/////////////////////////////////////////////////////////////////////
+
+class AliMUONData : public TNamed {
+ public:
+    AliMUONData();
+    AliMUONData(AliLoader * loader, const char* name, const char* title);
+    AliMUONData(const AliMUONData& rMUONData);
+    virtual ~AliMUONData();  
+    virtual void   AddDigit(Int_t id, Int_t* tracks, Int_t* charges,
+                            Int_t* digits); 
+    virtual void   AddHit(Int_t fIshunt, Int_t track, Int_t iChamber, 
+                         Int_t idpart, Float_t X, Float_t Y, Float_t Z, 
+                         Float_t tof, Float_t momentum, Float_t theta, 
+                         Float_t phi, Float_t length, Float_t destep);
+    virtual void   AddGlobalTrigger(Int_t *singlePlus, Int_t *singleMinus,
+                                   Int_t *singleUndef, Int_t *pairUnlike, 
+                                   Int_t *pairLike);
+    virtual void   AddLocalTrigger(Int_t* ltrigger);
+    virtual void   AddRawCluster(Int_t id, const AliMUONRawCluster& clust);
+    TClonesArray*  Hits() {return fHits;}
+    TClonesArray*  Digits(Int_t DetectionPlane, Int_t /*Cathode*/) 
+      {return ( (TClonesArray*) fDigits->At(DetectionPlane) );}
+    TClonesArray*  LocalTrigger() {return fLocalTrigger;}
+    TClonesArray*  GlobalTrigger() {return fGlobalTrigger;}
+    TClonesArray*  RawClusters(Int_t DetectionPlane)
+      {return ( (TClonesArray*) fRawClusters->At(DetectionPlane) );}
+    
+    virtual void   MakeBranch(Option_t *opt=" ");
+    virtual void   SetTreeAddress();
+
+    virtual void   ResetHits();
+    virtual void   ResetDigits();
+    virtual void   ResetTrigger();
+    virtual void   ResetRawClusters();
+  
+    TTree*         TreeH() {return fLoader->TreeH(); }
+    TTree*         TreeD() {return fLoader->TreeD(); }
+    TTree*         TreeR() {return fLoader->TreeR(); }
+    TTree*         TreeT() {return fLoader->TreeT(); }
+    TTree*         TreeP() {return fLoader->TreeP(); }
+
+ private:  
+    //descendant classes should
+    //use protected interface methods to access these folders
+
+  
+ protected: 
+    AliLoader*  fLoader;
+
+    TClonesArray*   fHits;    // One event in treeH per primary track
+    TObjArray*      fDigits;  // One event in treeD and one branch per detection plane
+    TObjArray*      fRawClusters; //One event in TreeR/Rawcluster and one branch per tracking detection plane
+    TClonesArray*   fGlobalTrigger; //! List of Global Trigger One event in TreeR/GlobalTriggerBranch
+    TClonesArray*   fLocalTrigger;  //! List of Local Trigger, ONe event in TreeR/LocalTriggerBranch       
+    Int_t           fNhits;
+    Int_t*          fNdigits;
+    Int_t*          fNrawclusters;
+    Int_t           fNglobaltrigger;
+    Int_t           fNlocaltrigger;
+
+    ClassDef(AliMUONData,1)
+ };
+#endif

diff --git a/MUON/AliMUONLoader.cxx b/MUON/AliMUONLoader.cxx
index 2e6ad04..af0d516 100644 (file)
--- a/MUON/AliMUONLoader.cxx
+++ b/MUON/AliMUONLoader.cxx
@@ -32,8 +32,7 @@ AliMUONLoader::AliMUONLoader(const Char_t * detname,TFolder* eventfolder):AliLoa
 
 AliMUONLoader::~AliMUONLoader()
 {
-//detructor
- 
+//detructor 
 }
 /*****************************************************************************/ 
 

diff --git a/MUON/AliMUONLoader.h b/MUON/AliMUONLoader.h
index 245f633..97a47c1 100644 (file)
--- a/MUON/AliMUONLoader.h
+++ b/MUON/AliMUONLoader.h
@@ -2,7 +2,6 @@
 #define ALIMUONLOADER_H
 
 #include "AliLoader.h"
-#include "AliDataLoader.h"
 
 //__________________________________________________________________
 /////////////////////////////////////////////////////////////////////
@@ -11,14 +10,17 @@
 //                                                                 //
 /////////////////////////////////////////////////////////////////////
 
+class TClonesArray;
+
+class AliMUONData;
+
 class AliMUONLoader : public AliLoader {
  public:
     AliMUONLoader();
     AliMUONLoader(const Char_t *detname,const Char_t *eventfoldername); //contructor with name of the top folder of the tree
     AliMUONLoader(const Char_t *detname,TFolder* eventfolder);
-
     virtual ~AliMUONLoader();
-   
+
  private:
     //descendant classes should
     //use protected interface methods to access these folders
@@ -31,7 +33,8 @@ class AliMUONLoader : public AliLoader {
     /*********** AliRunLoader as well**************/
     /**********************************************/
  public:
- 
+    AliMUONData * fMUONData; // data for MUON subsystem 
+
     ClassDef(AliMUONLoader,1)
  };