]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - MUON/AliMUONMerger.cxx
Transition to NewIO
[u/mrichter/AliRoot.git] / MUON / AliMUONMerger.cxx
index a29761ec9db53c6a2b95012f7fd4a5a524d1c2f9..69d69102da03d46cd1ead4c2ee0f1eb8e65d208e 100644 (file)
  * provided "as is" without express or implied warranty.                  *
  **************************************************************************/
 
-/*
-$Log$
-Revision 1.2  2001/03/05 08:40:25  morsch
-Method SortTracks(..) imported from AliMUON.
-
-Revision 1.1  2001/02/02 14:11:53  morsch
-AliMUONMerger prototype to be called by the merge manager.
-
-*/
+/* $Id$ */
 
+#include <Riostream.h> 
 #include <TTree.h> 
-#include <TVector.h>
 #include <TObjArray.h>
 #include <TFile.h>
 #include <TDirectory.h>
+#include <TPDGCode.h>
 
-
-// #include "AliMerger.h"
-// #include "AliMergable.h"
-#include "AliMUONMerger.h"
-#include "AliMUONConstants.h"
-#include "AliMUONChamber.h"
 #include "AliHitMap.h"
-#include "AliMUONHitMapA1.h"
 #include "AliMUON.h"
+#include "AliMUONChamber.h"
+#include "AliMUONConstants.h"
+#include "AliMUONDigit.h"
 #include "AliMUONHit.h"
+#include "AliMUONHitMapA1.h"
+#include "AliMUONMerger.h"
 #include "AliMUONPadHit.h"
-#include "AliMUONDigit.h"
 #include "AliMUONTransientDigit.h"
 #include "AliRun.h"
-#include "AliPDG.h"
 
 ClassImp(AliMUONMerger)
 
@@ -62,8 +51,8 @@ AliMUONMerger::AliMUONMerger()
     fPadHitsBgr = 0;
     fHitMap     = 0;
     fList       = 0;
-    fTrList     = 0;
-    fAddress    = 0; 
+    fBgrFile    = 0;
+    fDebug      = 0;
 }
 
 //------------------------------------------------------------------------
@@ -73,74 +62,65 @@ AliMUONMerger::~AliMUONMerger()
     if (fTrH1)       delete fTrH1;
     if (fHitsBgr)    delete fHitsBgr;
     if (fPadHitsBgr) delete fPadHitsBgr;
-    if (fHitMap)     delete fHitMap;
+    if (fHitMap)     delete [] fHitMap;
     if (fList)       delete fList;
-    if (fTrList)     delete fTrList;
-    if (fAddress)    delete fAddress; 
+    if (fBgrFile)    delete fBgrFile;
 }
 
 //------------------------------------------------------------------------
-Bool_t AliMUONMerger::Exists(const AliMUONPadHit *mergable)
+Bool_t AliMUONMerger::Exists(const AliMUONPadHit *padhit) const
 {
-    AliMUONPadHit *padhit = (AliMUONPadHit*) mergable;
+// test if the given padhit was already fired
     return (fHitMap[fNch]->TestHit(padhit->PadX(),padhit->PadY()));
 }
 
 //------------------------------------------------------------------------
-void AliMUONMerger::Update(AliMUONPadHit *mergable)
+void AliMUONMerger::Update(AliMUONPadHit *padhit)
 {
-    AliMUONPadHit *padhit = (AliMUONPadHit*) mergable;    
-    AliMUONTransientDigit* pdigit;
-    Int_t ipx      = padhit->PadX();        // pad number on X
-    Int_t ipy      = padhit->PadY();        // pad number on Y
-    Int_t iqpad    = Int_t(padhit->QPad()); // charge per pad
+// add new contribution to the fired padhit
+    AliMUONTransientDigit *pdigit = 
+      static_cast<AliMUONTransientDigit*>(
+      fHitMap[fNch]->GetHit(padhit->PadX(),padhit->PadY()));
 
-    pdigit = (AliMUONTransientDigit*) fHitMap[fNch]->GetHit(ipx, ipy);
     // update charge
     //
-    (*pdigit).AddSignal(iqpad);
-    (*pdigit).AddPhysicsSignal(iqpad);         
+    Int_t iqpad    = padhit->QPad();        // charge per pad
+    pdigit->AddSignal(iqpad);
+    pdigit->AddPhysicsSignal(iqpad);           
+
     // update list of tracks
     //
-    TObjArray* fTrList = (TObjArray*)pdigit->TrackList();
-    Int_t lastEntry = fTrList->GetLast();
-    TVector *pTrack = (TVector*)fTrList->At(lastEntry);
-    TVector &ptrk   = *pTrack;
-    TVector &trinfo = *((TVector*) (*fAddress)[fCountadr-1]);
-    Int_t lastTrack = Int_t(ptrk(0));
-
-    if (trinfo(0) != kBgTag) {
-       if (lastTrack == fTrack) {
-           Int_t lastCharge = Int_t(ptrk(1));
-           lastCharge += iqpad;
-           fTrList->RemoveAt(lastEntry);
-           trinfo(1) = lastCharge;
-           fTrList->AddAt(&trinfo,lastEntry);
-       } else {
-           fTrList->Add(&trinfo);
-       }
+    Int_t track, charge;    
+    if (fSignal) {
+      track = fTrack;
+      charge = iqpad;
     } else {
-       if (lastTrack != -1) fTrList->Add(&trinfo);
+      track = kBgTag;
+      charge = kBgTag;
     }
+    pdigit->UpdateTrackList(track,charge);
 }
 
 //------------------------------------------------------------------------
-void AliMUONMerger::CreateNew(AliMUONPadHit *mergable)
+void AliMUONMerger::CreateNew(AliMUONPadHit *padhit)
 {
-    AliMUONPadHit *padhit = (AliMUONPadHit*) mergable;    
-    AliMUONTransientDigit* pdigit;
-
-    Int_t ipx      = padhit->PadX();       // pad number on X
-    Int_t ipy      = padhit->PadY();       // pad number on Y
+// add new transient digit to the list, update hit map
     fList->AddAtAndExpand(
        new AliMUONTransientDigit(fNch,fDigits),fCounter);
-    fHitMap[fNch]->SetHit(ipx, ipy, fCounter);
-    fCounter++;
-    pdigit = (AliMUONTransientDigit*)fList->At(fList->GetLast());
+    fHitMap[fNch]->SetHit(padhit->PadX(),padhit->PadY(),fCounter);
+    AliMUONTransientDigit* pdigit = 
+      static_cast<AliMUONTransientDigit*>(fList->Last());
     // list of tracks
-    TObjArray *fTrList = (TObjArray*)pdigit->TrackList();
-    TVector &trinfo    = *((TVector*) (*fAddress)[fCountadr-1]);
-    fTrList->Add(&trinfo);
+    Int_t track, charge;
+    if (fSignal) {
+      track = fTrack;
+      charge = padhit->QPad();
+    } else {
+      track = kBgTag;
+      charge = kBgTag;
+    }
+    pdigit->AddToTrackList(track,charge);
+    fCounter++;
 }
 
 
@@ -148,8 +128,8 @@ void AliMUONMerger::CreateNew(AliMUONPadHit *mergable)
 void AliMUONMerger::Init()
 {
 // Initialisation
-    
-    if (fMerge) fBgrFile = InitBgr();
+    // open only once the background file !!
+    if (fMerge && !fBgrFile) fBgrFile = InitBgr();
 }
 
 
@@ -177,18 +157,15 @@ void AliMUONMerger::Digitise()
     AliSegmentation*  segmentation;
 
     fList = new TObjArray;
-    if(!fAddress) fAddress = new TClonesArray("TVector",10000);
 
     AliMUON *pMUON  = (AliMUON *) gAlice->GetModule("MUON");
     fHitMap= new AliHitMap* [AliMUONConstants::NCh()];
-    for (Int_t i = 0; i < AliMUONConstants::NCh(); i++) {fHitMap[i] = 0;}
     if (fMerge ) {
        fBgrFile->cd();
-       fBgrFile->ls();
         //
        // Get Hits Tree header from file
-       if(fHitsBgr) fHitsBgr->Clear();
-       if(fPadHitsBgr) fPadHitsBgr->Clear();
+       //if(fHitsBgr) fHitsBgr->Clear();     // Useless because line 327
+       //if(fPadHitsBgr) fPadHitsBgr->Clear(); // Useless because line 328
        if(fTrH1) delete fTrH1;
        fTrH1 = 0;
        
@@ -215,11 +192,9 @@ void AliMUONMerger::Digitise()
     //
     // loop over cathodes
     //
-    AliHitMap* hm;
-    fCountadr = 0;
+    fSignal = kTRUE;
     for (int icat = 0; icat < 2; icat++) { 
        fCounter = 0;
-       Int_t * nmuon = new Int_t [AliMUONConstants::NCh()];
        for (Int_t i = 0; i < AliMUONConstants::NCh(); i++) {
            iChamber = &(pMUON->Chamber(i));
            if (iChamber->Nsec() == 1 && icat == 1) {
@@ -228,25 +203,30 @@ void AliMUONMerger::Digitise()
                segmentation = iChamber->SegmentationModel(icat+1);
            }
            fHitMap[i] = new AliMUONHitMapA1(segmentation, fList);
-           nmuon[i] = 0;
        }
 
 //
 //   Loop over tracks
 //
 
-       TTree *treeH  = gAlice->TreeH();
-       Int_t ntracks = (Int_t) treeH->GetEntries();
-       Int_t jj;
+/******************************************************************/
+      TTree* treeH = pMUON->TreeH();
+      if (treeH == 0x0)
+       {
+         cerr<<"AliMUONMerger::Exec: Can not get TreeH"<<endl;
+         return;
+       }
+/******************************************************************/     
 
-       Float_t ** xhit = new Float_t * [AliMUONConstants::NCh()];
-       for (jj = 0; jj < AliMUONConstants::NCh(); jj++) xhit[jj] = new Float_t[2];
-       Float_t ** yhit = new Float_t * [AliMUONConstants::NCh()];
-       for (jj = 0; jj < AliMUONConstants::NCh(); jj++) yhit[jj] = new Float_t[2];
+       
+       
+       Int_t ntracks = (Int_t) treeH->GetEntries();
+       treeH->SetBranchStatus("*",0); // switch off all branches
+        treeH->SetBranchStatus("MUON*",1); // switch on only MUON
 
        for (fTrack = 0; fTrack < ntracks; fTrack++) {
            gAlice->ResetHits();
-           treeH->GetEvent(fTrack);
+           treeH->GetEntry(fTrack,0);
 //
 //   Loop over hits
            for(AliMUONHit* mHit = (AliMUONHit*)pMUON->FirstHit(-1); 
@@ -256,17 +236,6 @@ void AliMUONMerger::Digitise()
                fNch = mHit->Chamber()-1;  // chamber number
                if (fNch > AliMUONConstants::NCh()-1) continue;
                iChamber = &(pMUON->Chamber(fNch));
-               /*
-               if (fMerge) {
-                   if (mHit->Particle() == kMuonPlus || 
-                       mHit->Particle() == kMuonMinus) {
-                       xhit[fNch][nmuon[fNch]] = mHit->X();
-                       yhit[fNch][nmuon[fNch]] = mHit->Y();
-                       nmuon[fNch]++;
-                       if (nmuon[fNch] > 2) printf("MUON: nmuon %d\n",nmuon[fNch]);
-                   }
-               }
-               */
                
 //
 // Loop over pad hits
@@ -276,21 +245,12 @@ void AliMUONMerger::Digitise()
                     mPad = (AliMUONPadHit*)pMUON->NextPad(pMUON->PadHits()))
                {
                    Int_t cathode  = mPad->Cathode();      // cathode number
-                   Int_t ipx      = mPad->PadX();         // pad number on X
-                   Int_t ipy      = mPad->PadY();         // pad number on Y
-                   Int_t iqpad    = Int_t(mPad->QPad());  // charge per pad
                    if (cathode != (icat+1)) continue;
-
-                   segmentation = iChamber->SegmentationModel(cathode);
-
-                   new((*fAddress)[fCountadr++]) TVector(2);
-
-                   TVector &trinfo = *((TVector*) (*fAddress)[fCountadr-1]);
-                   trinfo(0) = (Float_t)fTrack;
-                   trinfo(1) = (Float_t)iqpad;
-
-                   fDigits[0] = ipx;
-                   fDigits[1] = ipy;
+                   Int_t iqpad    = Int_t(mPad->QPad());  // charge per pad
+//                 segmentation = iChamber->SegmentationModel(cathode);
+                   fDigits[0] = mPad->PadX();  
+                   fDigits[1] = mPad->PadY();
+                   if (!(fHitMap[fNch]->ValidateHit(fDigits[0], fDigits[1]))) continue;
                    fDigits[2] = icat;
                    fDigits[3] = iqpad;
                    fDigits[4] = iqpad;
@@ -313,6 +273,7 @@ void AliMUONMerger::Digitise()
        // open the file with background
        
        if (fMerge) {
+            fSignal = kFALSE;
            ntracks = (Int_t)fTrH1->GetEntries();
 //
 //   Loop over tracks
@@ -326,22 +287,11 @@ void AliMUONMerger::Digitise()
 //
 //   Loop over hits
                AliMUONHit* mHit;
-               for(int i = 0; i < fHitsBgr->GetEntriesFast(); ++i) 
+               for(Int_t i = 0; i < fHitsBgr->GetEntriesFast(); ++i) 
                {       
                    mHit   = (AliMUONHit*) (*fHitsBgr)[i];
                    fNch   = mHit->Chamber()-1;  // chamber number
                    iChamber = &(pMUON->Chamber(fNch));
-                    Float_t xbgr = mHit->X();
-                   Float_t ybgr = mHit->Y();
-                   Bool_t cond  = kFALSE;
-                   /*
-                   for (Int_t imuon = 0; imuon < nmuon[fNch]; imuon++) {
-                       Float_t dist = (xbgr-xhit[fNch][imuon])*(xbgr-xhit[fNch][imuon])
-                           +(ybgr-yhit[fNch][imuon])*(ybgr-yhit[fNch][imuon]);
-                       if (dist < 100.) cond = kTRUE;
-                   }
-                   */
-                   cond  = kTRUE;
 //
 // Loop over pad hits
                    for (AliMUONPadHit* mPad =
@@ -353,13 +303,9 @@ void AliMUONMerger::Digitise()
                        Int_t ipx      = mPad->PadX();        // pad number on X
                        Int_t ipy      = mPad->PadY();        // pad number on Y
                        Int_t iqpad    = Int_t(mPad->QPad()); // charge per pad
+                       if (!(fHitMap[fNch]->ValidateHit(ipx, ipy))) continue;
 
                        if (cathode != (icat+1)) continue;
-                       new((*fAddress)[fCountadr++]) TVector(2);
-                       TVector &trinfo = *((TVector*) (*fAddress)[fCountadr-1]);
-                       trinfo(0) = kBgTag;  // tag background
-                       trinfo(1) = kBgTag;
-                       
                        fDigits[0] = ipx;
                        fDigits[1] = ipy;
                        fDigits[2] = icat;
@@ -377,17 +323,15 @@ void AliMUONMerger::Digitise()
                } // hit loop
            } // track loop
 
-           TTree *fAli = gAlice->TreeK();
+           TTree *treeK = gAlice->TreeK();
             TFile *file = NULL;
            
-           if (fAli) file = fAli->GetCurrentFile();
+           if (treeK) file = treeK->GetCurrentFile();
            file->cd();
        } // if fMerge
-       delete [] xhit;
-       delete [] yhit;
 
-       Int_t tracks[10];
-       Int_t charges[10];
+       Int_t tracks[kMAXTRACKS];
+       Int_t charges[kMAXTRACKS];
        Int_t nentries = fList->GetEntriesFast();
        
        for (Int_t nent = 0; nent < nentries; nent++) {
@@ -399,7 +343,7 @@ void AliMUONMerger::Digitise()
 //
 //  Digit Response (noise, threshold, saturation, ...)
            AliMUONResponse * response = iChamber->ResponseModel();
-           q = response->DigitResponse(q);
+           q = response->DigitResponse(q,address);
            
            if (!q) continue;
            
@@ -410,32 +354,29 @@ void AliMUONMerger::Digitise()
            fDigits[4] = address->Physics();
            fDigits[5] = address->Hit();
            
-           TObjArray* fTrList = (TObjArray*)address->TrackList();
-           Int_t nptracks = fTrList->GetEntriesFast();
+           Int_t nptracks = address->GetNTracks();
 
-           // this was changed to accomodate the real number of tracks
-
-           if (nptracks > 10) {
-               printf("\n Attention - nptracks > 10 %d \n", nptracks);
-               nptracks = 10;
+           if (nptracks > kMAXTRACKS) {
+               if (fDebug>0)
+                 printf("\n Attention - nptracks > kMAXTRACKS %d \n", nptracks);
+               nptracks = kMAXTRACKS;
            }
            if (nptracks > 2) {
-               printf("Attention - nptracks > 2  %d \n",nptracks);
-               printf("cat,ich,ix,iy,q %d %d %d %d %d \n",icat,ich,fDigits[0],fDigits[1],q);
+               if (fDebug>0) {
+                 printf("Attention - nptracks > 2  %d \n",nptracks);
+                 printf("cat,ich,ix,iy,q %d %d %d %d %d \n",icat,ich,fDigits[0],fDigits[1],q);
+               }
            }
            for (Int_t tr = 0; tr < nptracks; tr++) {
-               TVector *ppP = (TVector*)fTrList->At(tr);
-               if(!ppP ) printf("ppP - %p\n",ppP);
-               TVector &pp  = *ppP;
-               tracks[tr]   = Int_t(pp(0));
-               charges[tr]  = Int_t(pp(1));
+               tracks[tr]   = address->GetTrack(tr);
+               charges[tr]  = address->GetCharge(tr);
            }      //end loop over list of tracks for one pad
             // Sort list of tracks according to charge
            if (nptracks > 1) {
                SortTracks(tracks,charges,nptracks);
            }
-           if (nptracks < 10 ) {
-               for (Int_t i = nptracks; i < 10; i++) {
+           if (nptracks < kMAXTRACKS ) {
+               for (Int_t i = nptracks; i < kMAXTRACKS; i++) {
                    tracks[i]  = 0;
                    charges[i] = 0;
                }
@@ -443,7 +384,6 @@ void AliMUONMerger::Digitise()
            
            // fill digits
            pMUON->AddDigits(ich,tracks,charges,fDigits);
-           // delete fTrList;
        }
        gAlice->TreeD()->Fill();
        pMUON->ResetDigits();
@@ -452,19 +392,17 @@ void AliMUONMerger::Digitise()
        
        for(Int_t ii = 0; ii < AliMUONConstants::NCh(); ++ii) {
            if (fHitMap[ii]) {
-               hm=fHitMap[ii];
-               delete hm;
+               delete fHitMap[ii];
                fHitMap[ii] = 0;
            }
        }
-       delete [] nmuon;    
     } //end loop over cathodes
     delete [] fHitMap;
     delete fList;
     
-    if (fAddress)    fAddress->Delete();
-    if (fHitsBgr)    fHitsBgr->Delete();
-    if (fPadHitsBgr) fPadHitsBgr->Delete();
+//  no need to delete ... and it makes a crash also
+//    if (fHitsBgr)    fHitsBgr->Delete();
+//    if (fPadHitsBgr) fPadHitsBgr->Delete();
     // gObjectTable->Print();
 }