Changed to use VStores (Laurent)
authorhristov <hristov@f7af4fe6-9843-0410-8265-dc069ae4e863>
Sun, 17 Jun 2007 20:54:28 +0000 (20:54 +0000)
committerhristov <hristov@f7af4fe6-9843-0410-8265-dc069ae4e863>
Sun, 17 Jun 2007 20:54:28 +0000 (20:54 +0000)
MUON/AliMUONSDigitizerV2.cxx
MUON/AliMUONTrigger.cxx
MUON/AliMUONTrigger.h

index b68c702..aaa70ba 100644 (file)
 #include "AliLog.h"
 #include "AliMUON.h"
 #include "AliMUONChamber.h"
-#include "AliMUONSimData.h"
-#include "AliMUONDigit.h"
+#include "AliMUONVDigit.h"
+#include "AliMUONDigitStoreV1.h"
 #include "AliMUONHit.h"
 #include "AliMpDEManager.h"
 #include "AliLoader.h"
 #include "AliRun.h"
 #include "AliRunLoader.h"
-#include "TObjArray.h"
-
+#include "AliMUONVHitStore.h"
 
 ///
 /// The sdigitizer performs the transformation from hits (energy deposits by
@@ -83,15 +82,16 @@ AliMUONSDigitizerV2::Exec(Option_t*)
   AliDebug(1,"");
   
   AliRunLoader* runLoader = AliRunLoader::GetRunLoader();
-  AliLoader* fLoader = runLoader->GetLoader("MUONLoader");
+  AliLoader* loader = runLoader->GetDetectorLoader("MUON");
 
-  fLoader->LoadHits("READ");
+  loader->LoadHits("READ");
   
-  AliMUONSimData muonData(fLoader,"MUON","MUON");
-
   AliMUON* muon = static_cast<AliMUON*>(gAlice->GetModule("MUON"));
     
   Int_t nofEvents(runLoader->GetNumberOfEvents());
+  
+  AliMUONVDigitStore* sDigitStore = new AliMUONDigitStoreV1;
+  
   for ( Int_t iEvent = 0; iEvent < nofEvents; ++iEvent ) 
   {    
     // Loop over events.
@@ -100,33 +100,36 @@ AliMUONSDigitizerV2::Exec(Option_t*)
     
     AliDebug(1,Form("iEvent=%d",iEvent));
     runLoader->GetEvent(iEvent);
-    TTree* treeS = fLoader->TreeS();
-    AliDebug(1,Form("TreeS=%p",treeS));
+  
+    loader->MakeSDigitsContainer();
+
+    TTree* treeS = loader->TreeS();
+
     if ( !treeS )
     {
-      AliDebug(1,"MakeSDigitsContainer");
-      fLoader->MakeSDigitsContainer();
-      treeS = fLoader->TreeS();
+      AliFatal("");
     }
-    AliDebug(1,Form("TreeS=%p",treeS));
-    muonData.MakeBranch("S");
-    muonData.SetTreeAddress("S");
+
+    sDigitStore->Connect(*treeS);
+    
+    TTree* treeH = loader->TreeH();
+
+    AliMUONVHitStore* hitStore = AliMUONVHitStore::Create(*treeH);
+    hitStore->Connect(*treeH);
     
-    muonData.SetTreeAddress("H");
-    TTree* treeH = fLoader->TreeH();
-    AliDebug(1,Form("TreeH=%p",treeH));
-             
     Long64_t nofTracks = treeH->GetEntries();
+    
     for ( Long64_t iTrack = 0; iTrack < nofTracks; ++iTrack )
     {
       // Loop over the tracks of this event.
       treeH->GetEvent(iTrack);
-      TClonesArray* hits = muonData.Hits();
-      Int_t nofHits = hits->GetEntriesFast();
-      for ( Int_t ihit = 0; ihit < nofHits; ++ihit )
+
+      AliMUONHit* hit;
+      TIter next(hitStore->CreateIterator());
+      Int_t ihit(0);
+      
+      while ( ( hit = static_cast<AliMUONHit*>(next()) ) )       
       {
-        // Loop over the hits of this track.
-        AliMUONHit* hit = static_cast<AliMUONHit*>(hits->At(ihit)); 
         Int_t chamberId = hit->Chamber()-1;
         AliMUONChamber& chamber = muon->Chamber(chamberId);
         AliMUONResponse* response = chamber.ResponseModel();
@@ -136,35 +139,49 @@ AliMUONSDigitizerV2::Exec(Option_t*)
         response->DisIntegrate(*hit,digits);
         
         TIter next(&digits);
-        AliMUONDigit* d;
-        while ( ( d = (AliMUONDigit*)next() ) )
+        AliMUONVDigit* d;
+        while ( ( d = (AliMUONVDigit*)next() ) )
         {
           // Update some sdigit information that could not be known
           // by the DisIntegrate method
           d->SetHit(ihit);
-          d->AddTrack(iTrack,d->Signal());
+          d->AddTrack(iTrack,d->Charge());
           tdlist.Add(d);
         }
+        ++ihit;
       }
-      muonData.ResetHits();
+      hitStore->Clear();
     } // end of loop on tracks within an event
     
-    for ( Int_t i = 0; i <= tdlist.GetLast(); ++i )
+    TIter next(&tdlist);
+    AliMUONVDigit* d;
+    
+    while ( ( d = static_cast<AliMUONVDigit*>(next()) ) )
     {
-      AliMUONDigit* d = (AliMUONDigit*)tdlist[i];
-      StdoutToAliDebug(1,d->Print(););
-      if ( d->Signal() > 0 ) // that check would be better in the disintegrate
+      if ( d->Charge() > 0 ) // that check would be better in the disintegrate
         // method, but to compare with old sdigitizer, it has to be there.
       {
-        muonData.AddSDigit(AliMpDEManager::GetChamberId(d->DetElemId()),*d);
+        AliMUONVDigit* added = sDigitStore->Add(*d,AliMUONVDigitStore::kMerge);
+        if (!added)
+        {
+          AliError("Could not add digit to digitStore");
+        }
       }
     }
-    muonData.Fill("S");
-    fLoader->WriteSDigits("OVERWRITE");
+
+    treeS->Fill();
+    
+    loader->WriteSDigits("OVERWRITE");
+    
+    sDigitStore->Clear();
+    
+    loader->UnloadSDigits();
+    
+    delete hitStore;
     
-    muonData.ResetSDigits();
-    fLoader->UnloadSDigits();
   } // loop on events
   
-  fLoader->UnloadHits();  
+  loader->UnloadHits();  
+  
+  delete sDigitStore;
 }
index 2d0bd7d..12715c2 100644 (file)
 
 /* $Id$ */
 
-#include <TClonesArray.h>
+#include "AliMUONTrigger.h"
 
 #include "AliLog.h"
+#include "AliMUONGlobalTrigger.h"
+#include "AliMUONVTriggerStore.h"
 #include "AliRun.h"
 #include "AliRunLoader.h"
 #include "AliTriggerInput.h"
 
-#include "AliMUON.h"
-#include "AliMUONSimLoader.h"
-#include "AliMUONData.h"
-#include "AliMUONDigit.h"
-#include "AliMUONGlobalTrigger.h"
-#include "AliMUONTrigger.h"
-
 ///
 /// \class AliMUONTrigger
 ///
@@ -45,7 +40,7 @@ ClassImp(AliMUONTrigger)
 
 //----------------------------------------------------------------------
 AliMUONTrigger::AliMUONTrigger()
-  : AliTriggerDetector() 
+  : AliTriggerDetector(), fTriggerStore(0x0)
 {
 /// Default constructor
 
@@ -56,7 +51,8 @@ AliMUONTrigger::AliMUONTrigger()
 //----------------------------------------------------------------------
 AliMUONTrigger::~AliMUONTrigger()
 {
-/// Destructor
+  /// Destructor
+  delete fTriggerStore;
 }
 
 //----------------------------------------------------------------------
@@ -82,42 +78,56 @@ void AliMUONTrigger::Trigger()
 {
   /// sets the trigger inputs
 
-  AliMUONGlobalTrigger* globalTrigger;
-  TClonesArray* globalTriggerArray;
-
    AliRunLoader* runLoader = gAlice->GetRunLoader();
-
-   AliLoader * muonLoader = runLoader->GetLoader("MUONLoader");
+  
+   AliLoader * muonLoader = runLoader->GetDetectorLoader("MUON");
    muonLoader->LoadDigits("READ");
 
-   // Creating MUON data container
-   AliMUONData* muonData = new AliMUONData(muonLoader,"MUON","MUON");
-
-   // get global info
-   muonData->SetTreeAddress("GLT");
-   muonData->GetTriggerD();
-   globalTriggerArray = muonData->GlobalTrigger(); 
-   if (globalTriggerArray == 0x0) { 
-     AliWarning("No Global Trigger Array available");
+   TTree* treeD = muonLoader->TreeD();
+   
+   if (!treeD)
+   {
+     AliError("No TreeD available. Cannot make trigger");
      return;
    }
-   globalTrigger = (AliMUONGlobalTrigger*)globalTriggerArray->UncheckedAt(0);
+   
+   if (!fTriggerStore) 
+   {  
+     fTriggerStore = AliMUONVTriggerStore::Create(*treeD);
+     if (!fTriggerStore)
+     {
+       AliError("Could not create triggerStore from treeD");
+       return;
+     }     
+   }
 
-   if (globalTrigger == 0x0) { 
-     AliWarning("No Global Trigger available");
+   Bool_t ok = fTriggerStore->Connect(*treeD,kTRUE);
+
+   if (!ok)
+   {
+     AliError("Could not read trigger from TreeD !");
      return;
    }
-   // set CTP
-   if (globalTrigger->SingleLpt())      SetInput("MUON_Single_LPt_L0");
-   if (globalTrigger->SingleHpt())      SetInput("MUON_Single_HPt_L0");
    
-   if (globalTrigger->PairUnlikeLpt())  SetInput("MUON_Unlike_LPt_L0");
-   if (globalTrigger->PairUnlikeHpt())  SetInput("MUON_Unlike_HPt_L0");
+   treeD->GetEvent(0);
    
-   if (globalTrigger->PairLikeLpt())    SetInput("MUON_Like_LPt_L0");
-   if (globalTrigger->PairLikeHpt())    SetInput("MUON_Like_HPt_L0");
-
-   muonData->ResetTrigger();
+   AliMUONGlobalTrigger* globalTrigger = fTriggerStore->Global();
+   if (globalTrigger == 0x0) 
+   { 
+     AliWarning("No Global Trigger available");
+   }
+   else
+   {
+     // set CTP
+     if (globalTrigger->SingleLpt())      SetInput("MUON_Single_LPt_L0");
+     if (globalTrigger->SingleHpt())      SetInput("MUON_Single_HPt_L0");
+     
+     if (globalTrigger->PairUnlikeLpt())  SetInput("MUON_Unlike_LPt_L0");
+     if (globalTrigger->PairUnlikeHpt())  SetInput("MUON_Unlike_HPt_L0");
+     
+     if (globalTrigger->PairLikeLpt())    SetInput("MUON_Like_LPt_L0");
+     if (globalTrigger->PairLikeHpt())    SetInput("MUON_Like_HPt_L0");
+   }
    muonLoader->UnloadDigits();
-
+   fTriggerStore->Clear();
 }
index ffa0637..3b1f8fe 100644 (file)
@@ -12,6 +12,8 @@
 
 #include "AliTriggerDetector.h"
 
+class AliMUONVTriggerStore;
+
 class AliMUONTrigger : public AliTriggerDetector
 {
  public:
@@ -20,6 +22,14 @@ class AliMUONTrigger : public AliTriggerDetector
    virtual void    CreateInputs();
    virtual void    Trigger();
 
+private:
+   /// Not implemented
+   AliMUONTrigger(const AliMUONTrigger&);
+   /// Not implemented
+   AliMUONTrigger& operator=(const AliMUONTrigger&);
+   
+   AliMUONVTriggerStore* fTriggerStore; //!< trigger store
+   
   ClassDef(AliMUONTrigger,1)  // MUON Trigger Detector class
 };
 #endif