// -------------------------------------------------------------------------
// -----                    CbmDigiManager source file                 -----
// -----                  Created 08/05/07  by V. Friese               -----
// -------------------------------------------------------------------------


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

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

// Includes from cbmroot
#include "CbmDigi.h"
#include "CbmDigiManager.h"
#include "CbmRootManager.h"

// Includes from cbmbase

using namespace std;

// -----   Default constructor   -------------------------------------------
CbmDigiManager::CbmDigiManager() : CbmTask("DigiManager") {
  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() {

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

  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)