MUON/AliMUONSDigitizerV2.cxx

   1 /**************************************************************************
   2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   3 *                                                                        *
   4 * Author: The ALICE Off-line Project.                                    *
   5 * Contributors are mentioned in the code where appropriate.              *
   6 *                                                                        *
   7 * Permission to use, copy, modify and distribute this software and its   *
   8 * documentation strictly for non-commercial purposes is hereby granted   *
   9 * without fee, provided that the above copyright notice appears in all   *
  10 * copies and that both the copyright notice and this permission notice   *
  11 * appear in the supporting documentation. The authors make no claims     *
  12 * about the suitability of this software for any purpose. It is          *
  13 * provided "as is" without express or implied warranty.                  *
  14 **************************************************************************/
  15
  16 // $Id$
  17
  18 #include <TClonesArray.h>
  19
  20 #include "AliMUONSDigitizerV2.h"
  21
  22 #include "AliLog.h"
  23 #include "AliMUON.h"
  24 #include "AliMUONChamber.h"
  25 #include "AliMUONVDigit.h"
  26 #include "AliMUONHit.h"
  27 #include "AliMpDEManager.h"
  28 #include "AliMpCDB.h"
  29 #include "AliLoader.h"
  30 #include "AliRun.h"
  31 #include "AliRunLoader.h"
  32 #include "AliMUONVDigitStore.h"
  33 #include "AliMUONVHitStore.h"
  34
  35 //-----------------------------------------------------------------------------
  36 /// The sdigitizer performs the transformation from hits (energy deposits by
  37 /// the transport code) to sdigits (equivalent of charges on pad).
  38 ///
  39 /// It does so by converting the energy deposit into a charge and then spreading
  40 /// this charge over several pads, according to the response function (a
  41 /// Mathieson distribution, basically).
  42 ///
  43 /// See also AliMUONResponseV0, which is doing the real job (in DisIntegrate
  44 /// method), while this sdigitizer is just "steering" the process.
  45 ///
  46 /// Please note that we do *not* merge sdigits after creation, which means
  47 /// that after sdigitization, a given pad can have several sdigits. This
  48 /// merging is taken care of later on by the digitizer(V3).
  49 //-----------------------------------------------------------------------------
  50
  51 ClassImp(AliMUONSDigitizerV2)
  52
  53 //_____________________________________________________________________________
  54 AliMUONSDigitizerV2::AliMUONSDigitizerV2()
  55 : TTask("AliMUONSDigitizerV2","From Hits to SDigits for MUON")
  56 {
  57   ///
  58   /// ctor.
  59   ///
  60   if ( ! AliMpCDB::LoadMpSegmentation() )
  61   {
  62     AliFatal("Could not access mapping from OCDB !");
  63   }
  64 }
  65
  66 //_____________________________________________________________________________
  67 AliMUONSDigitizerV2::~AliMUONSDigitizerV2()
  68 {
  69   ///
  70   /// dtor.
  71   ///
  72 }
  73
  74 //_____________________________________________________________________________
  75 void
  76 AliMUONSDigitizerV2::Exec(Option_t*)
  77 {
  78   ///
  79   /// Go from hits to sdigits.
  80   ///
  81   /// In the code below, apart from the loop itself (which look complicated
  82   /// but is really only a loop on each hit in the input file) the main
  83   /// work is done in AliMUONResponse::DisIntegrate method, which converts
  84   /// a single hit in (possibly) several sdigits.
  85   ///
  86
  87   AliDebug(1,"");
  88
  89   AliRunLoader* runLoader = AliRunLoader::GetRunLoader();
  90   AliLoader* loader = runLoader->GetDetectorLoader("MUON");
  91
  92   loader->LoadHits("READ");
  93
  94   AliMUON* muon = static_cast<AliMUON*>(gAlice->GetModule("MUON"));
  95
  96   Int_t nofEvents(runLoader->GetNumberOfEvents());
  97
  98   TString classname = muon->DigitStoreClassName();
  99
 100   AliMUONVDigitStore* sDigitStore = AliMUONVDigitStore::Create(classname.Data());
 101
 102   if (!sDigitStore)
 103   {
 104     AliFatal(Form("Could not create digitstore of class %s",classname.Data()));
 105   }
 106
 107   AliInfo(Form("Will use digitStore of type %s",sDigitStore->ClassName()));
 108
 109   for ( Int_t iEvent = 0; iEvent < nofEvents; ++iEvent )
 110   {
 111     // Loop over events.
 112     TObjArray tdlist;
 113     tdlist.SetOwner(kTRUE);
 114
 115     AliDebug(1,Form("iEvent=%d",iEvent));
 116     runLoader->GetEvent(iEvent);
 117
 118     loader->MakeSDigitsContainer();
 119
 120     TTree* treeS = loader->TreeS();
 121
 122     if ( !treeS )
 123     {
 124       AliFatal("");
 125     }
 126
 127     sDigitStore->Connect(*treeS);
 128
 129     TTree* treeH = loader->TreeH();
 130
 131     AliMUONVHitStore* hitStore = AliMUONVHitStore::Create(*treeH);
 132     hitStore->Connect(*treeH);
 133
 134     Long64_t nofTracks = treeH->GetEntries();
 135
 136     for ( Long64_t iTrack = 0; iTrack < nofTracks; ++iTrack )
 137     {
 138       // Loop over the tracks of this event.
 139       treeH->GetEvent(iTrack);
 140
 141       AliMUONHit* hit;
 142       TIter next(hitStore->CreateIterator());
 143       Int_t ihit(0);
 144
 145       while ( ( hit = static_cast<AliMUONHit*>(next()) ) )
 146       {
 147         Int_t chamberId = hit->Chamber()-1;
 148         AliMUONChamber& chamber = muon->Chamber(chamberId);
 149         AliMUONResponse* response = chamber.ResponseModel();
 150
 151         // This is the heart of this method : the dis-integration
 152         TList digits;
 153         response->DisIntegrate(*hit,digits);
 154
 155         TIter next(&digits);
 156         AliMUONVDigit* d;
 157         while ( ( d = (AliMUONVDigit*)next() ) )
 158         {
 159           // Update some sdigit information that could not be known
 160           // by the DisIntegrate method
 161           d->SetHit(ihit);
 162           d->AddTrack(iTrack,d->Charge());
 163           tdlist.Add(d);
 164         }
 165         ++ihit;
 166       }
 167       hitStore->Clear();
 168     } // end of loop on tracks within an event
 169
 170     TIter next(&tdlist);
 171     AliMUONVDigit* d;
 172
 173     while ( ( d = static_cast<AliMUONVDigit*>(next()) ) )
 174     {
 175       if ( d->Charge() > 0 ) // that check would be better in the disintegrate
 176         // method, but to compare with old sdigitizer, it has to be there.
 177       {
 178         AliMUONVDigit* added = sDigitStore->Add(*d,AliMUONVDigitStore::kMerge);
 179         if (!added)
 180         {
 181           AliError("Could not add digit to digitStore");
 182         }
 183       }
 184     }
 185
 186     treeS->Fill();
 187
 188     loader->WriteSDigits("OVERWRITE");
 189
 190     sDigitStore->Clear();
 191
 192     loader->UnloadSDigits();
 193
 194     delete hitStore;
 195
 196   } // loop on events
 197
 198   loader->UnloadHits();
 199
 200   delete sDigitStore;
 201 }