CbmDigiManager.cxx

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file CbmDigiManager.cxx

// -------------------------------------------------------------------------
// -----                    CbmDigiManager source file                 -----
// -----                  Created 08/05/07  by V. Friese               -----
// -------------------------------------------------------------------------
// Includes from cbmroot
#include "CbmDigiManager.h"

#include "CbmDigi.h"

#include "FairRootManager.h"

// Includes from ROOT
#include "TClonesArray.h"

// Includes from c++
#include <iostream>
#include <iomanip>

using std::cout;
using std::endl;
using std::left;
using std::right;
using std::setw;
using std::fixed;
using std::setprecision;
using std::pair;

// -----   Default constructor   -------------------------------------------
CbmDigiManager::CbmDigiManager() 
  : FairTask("DigiManager"),
    fDigiMap(),
    fTimer()
{
  for (Int_t iSys=0; iSys<16; iSys++) fDigis[iSys] = NULL;
  fSystem[0] = "";
  fSystem[1] = "MVD";
  fSystem[2] = "STS";
  fSystem[3] = "RICH";
  fSystem[4] = "MUCH";
  fSystem[5] = "TRD";
  fSystem[6] = "TOF";
  fSystem[7] = "ECAL";
  fSystem[8] = "PSD";
  for (Int_t iSys=9; iSys<16; iSys++) fSystem[iSys] = "";
}
// -------------------------------------------------------------------------



// -----   Destructor   ----------------------------------------------------
CbmDigiManager::~CbmDigiManager() {}
// -------------------------------------------------------------------------



// -----   Virtual public method Exec   ------------------------------------
void CbmDigiManager::Exec(Option_t* opt) {

  // Clear map and timer
  fTimer.Start();
  fDigiMap.clear();

  // Loop over all digi collections
  Int_t nDigis[16];
  for (Int_t iSys=0; iSys<16; iSys++) {
    nDigis[iSys] = 0;
    if ( ! fDigis[iSys] ) continue;

    // Loop over digis in this collection
    nDigis[iSys] = fDigis[iSys]->GetEntriesFast();
    for (Int_t iDigi=0; iDigi<nDigis[iSys]; iDigi++) {
      CbmDigi* digi = (CbmDigi*) fDigis[iSys]->At(iDigi);
      if ( digi->GetSystemId() != iSys ) {
  cout << "-E- " << GetName() << "::Exec: digi with system ID "
       << digi->GetSystemId() << " in digi collection " << iSys
       << "(" << fSystem[iSys] << ")" << endl;
  Fatal("Exec", "Wrong system ID for digi");
      }
      Int_t iDet  = digi->GetDetectorId();
      Int_t iChan = digi->GetChannelNr();
      pair<Int_t, Int_t> a(iDet, iChan);

      // Check for existing entries in the map (should not happen)
      if (fDigiMap.find(a) != fDigiMap.end() ) {
  cout << "-W- " << GetName() << "::Exec: Multiple entry in "
       << fSystem[iSys] << " digi collection; detector " << iDet
       << ", channel " << iChan << endl;
  continue;
      }

      // Enter digi into the map
      fDigiMap[a] = digi;

    }   // Loop over digis in the collection
  }     // Loop over collections

    

  // Control output
  fTimer.Stop();
  cout << "+ ";
  cout << setw(15) << left << fName << ": " << setprecision(4) << setw(8)
       << fixed << right << fTimer.RealTime()
       << " s";
  for (Int_t iSys=0; iSys<15; iSys++)
    if ( fDigis[iSys] ) cout << ", " << fSystem[iSys] 
           << " " << nDigis[iSys];
  cout << endl;

}
// -------------------------------------------------------------------------




// -----   Public method GetDigi   -----------------------------------------
CbmDigi* CbmDigiManager::GetDigi(Int_t iDetector, Int_t iChannel) {

  pair<Int_t, Int_t> a(iDetector, iChannel);

  if ( fDigiMap.find(a) == fDigiMap.end() ) return NULL;
  else return fDigiMap[a];

}
// -------------------------------------------------------------------------



      
// -----   Virtual private method Init   -----------------------------------
InitStatus CbmDigiManager::Init() {

  FairRootManager* ioman = FairRootManager::Instance();
  if ( ! ioman ) Fatal("Init", "No FairRootManager");

  Int_t nColl = 0;
  for (Int_t iSys=0; iSys<16; iSys++) {

    TString digiName = fSystem[iSys];
    digiName += "Digi";
    fDigis[iSys] = (TClonesArray*) ioman->GetObject(digiName.Data());
    if ( fDigis[iSys] ) nColl++;
  }

  cout << "-I- " << GetName() << "::Init: " << nColl 
       << " digi collection";
  if ( nColl != 1 ) cout << "s";
  cout << " found." << endl;

  return kSUCCESS;

}
// -------------------------------------------------------------------------



ClassImp(CbmDigiManager)