BesEmcDigitization.cc

Go to the documentation of this file.

//---------------------------------------------------------------------------//
//      BOOST --- BESIII Object_Oreiented Simulation Tool                    //
//---------------------------------------------------------------------------//
// Descpirtion: EMC detector
// Author: Fu Chengdong
// Created: Dec 15, 2003
// Comment:
//---------------------------------------------------------------------------//
//
#include "EmcSim/BesEmcDigitization.hh"
#include "EmcSim/BesEmcHit.hh"
 
#include "G4Event.hh"
#include "G4EventManager.hh"
#include "G4HCofThisEvent.hh"
#include "G4SDManager.hh"
#include "G4UnitsTable.hh"
#include "G4VHitsCollection.hh"
#include "G4ios.hh"
using namespace CLHEP;
 
BesEmcDigitization::BesEmcDigitization()
    : calorimeterCollID( -1 )
    , fNHits( 0 )
    , fTotECrystal( 0 )
    , fTotLCrystal( 0 )
    , fNCrystalSignal( 0 ) {
  for ( int i = 0; i < 50; i++ )
  {
    for ( int j = 0; j < 150; j++ ) fSingleECrystal[i][j] = 0;
  }
  G4SDManager* SDman = G4SDManager::GetSDMpointer();
  calorimeterCollID  = SDman->GetCollectionID( "CalCollection" );
}
 
BesEmcDigitization::~BesEmcDigitization() {
  delete[] fEnergySignal;
  delete[] fNThetaSignal;
  delete[] fNPhiSignal;
}
 
G4bool BesEmcDigitization::Digitize( const G4Event* evt, G4double ecut ) {
  G4HCofThisEvent*      HCE = evt->GetHCofThisEvent();
  BesEmcHitsCollection* CHC = 0;
 
  if ( HCE ) CHC = (BesEmcHitsCollection*)( HCE->GetHC( calorimeterCollID ) );
  else return false;
 
  G4int i = 0, j = 0;
  if ( CHC )
  {
    fNHits = CHC->entries();
    if ( fNHits )
      for ( i = 0; i < fNHits; i++ )
      {
        fTotECrystal += ( *CHC )[i]->GetEdepCrystal();
        fTotLCrystal += ( *CHC )[i]->GetTrakCrystal();
        G4int nTheta = 45 - ( *CHC )[i]->GetNumThetaCrystal();
        G4int nPhi   = ( *CHC )[i]->GetNumPhiCrystal();
        if ( nTheta > 0 && nTheta < 50 && nPhi > 0 && nPhi < 150 ) // waiting upgrade
        {
          fSingleECrystal[nTheta][nPhi] += ( *CHC )[i]->GetEdepCrystal();
          // G4cout << "EMC Hit in crystal("
          //        << nTheta << "," <<nPhi<< ")" << G4endl;
        }
        else { G4cout << "EMC Hit out range:" << nTheta << ',' << nPhi << G4endl; }
      }
  }
  else return false;
 
  for ( i = 1; i < 50; i++ )
  {
    for ( j = 1; j < 150; j++ )
    {
      if ( fSingleECrystal[i][j] > ecut ) { fNCrystalSignal++; }
    }
  }
  if ( ( fEnergySignal = new G4double[fNCrystalSignal + 1] ) == NULL ||
       ( fNThetaSignal = new G4int[fNCrystalSignal + 1] ) == NULL ||
       ( fNPhiSignal = new G4int[fNCrystalSignal + 1] ) == NULL )
  {
    // G4Exception("BesEmcDigitization:can't allocate memory for Signal");
    G4cout << "BesEmcDigitization:can't allocate memory for Signal" << G4endl;
    exit( -1 );
  }
  G4int n = 0;
  for ( i = 1; i < 50; i++ )
  {
    for ( j = 1; j < 150; j++ )
    {
      if ( fSingleECrystal[i][j] > ecut )
      {
        n++;
        fEnergySignal[n] = fSingleECrystal[i][j];
        fNThetaSignal[n] = 45 - i;
        fNPhiSignal[n]   = j;
      }
    }
  }
 
  return true;
}
 
G4double BesEmcDigitization::GetEmcSignal( G4int nTheta, G4int nPhi ) {
  if ( nTheta > 0 && nTheta < 50 && nPhi > 0 && nPhi < 150 )
    return fSingleECrystal[nTheta][nPhi];
  else return 0;
}
 
void BesEmcDigitization::Print() {
  G4cout << "   Absorber: total energy: " << fTotECrystal / GeV << " GeV" << G4endl
         << "       total track length: " << fTotLCrystal / cm << " cm" << G4endl;
 
  G4cout << "\n     " << fNHits << " hits are stored in BesEmcHitsCollection." << G4endl;
}