BesTofDigitizer.cc

Go to the documentation of this file.

//---------------------------------------------------------------------------//
//      BOOST --- BESIII Object_Oriented Simulation Tool                     //
//---------------------------------------------------------------------------//
//Description: Chief control center for digitizing modes.Also define 'group hits'.
//Author: Dengzy
//Created: Mar, 2004
//Modified:  Add new digitizer module for endcap TOF. Liu Yong @ Dec,2008
//Comment: 
//---------------------------------------------------------------------------//
//$Id: BesTofDigitizer.cc
 
#include "BesTofDigitizer.hh"
#include "BesTofDigitizerBrV1.hh"
#include "BesTofDigitizerEcV1.hh"
#include "BesTofDigitizerBrV2.hh"
#include "BesTofDigitizerEcV2.hh"
#include "BesTofDigitizerEcV3.hh"
 
#include "BesTofDigitizerEcV4.hh"
 
 
 
 
#include "ScintSingle.hh"
#include "G4DigiManager.hh"
//#include "BesTofDigitizerMessenger.hh"
 
#include "GaudiKernel/ISvcLocator.h"
#include "GaudiKernel/Bootstrap.h"
#include "GaudiKernel/IDataProviderSvc.h"
#include "G4Svc/IG4Svc.h"
#include "G4Svc/G4Svc.h"
#include "ReadBoostRoot.hh"
 
#include "GaudiKernel/MsgStream.h"
 
 
 
BesTofDigitizer::BesTofDigitizer(G4String modName)
        :G4VDigitizerModule(modName)
{
  collectionName.push_back("BesTofDigitsCollection");  
    m_besTofDigitsCollection=0;
    //m_digiMessenger = new BesTofDigitizerMessenger(this);
 
    tofDMBr1 = new BesTofDigitizerBrV1();
    tofDMEc1 = new BesTofDigitizerEcV1();
    tofDMBr2 = new BesTofDigitizerBrV2();
    tofDMEc2 = new BesTofDigitizerEcV2();
    tofDMEc3 = new BesTofDigitizerEcV3();
    
    tofDMEc4 = new BesTofDigitizerEcV4();//MRPC Digitizer
 
    //retrieve G4Svc
    /*ISvcLocator* svcLocator = Gaudi::svcLocator();
    IG4Svc* tmpSvc;
    StatusCode sc=svcLocator->service("G4Svc", tmpSvc);
    G4Svc* g4Svc=dynamic_cast<G4Svc *>(tmpSvc);*/
 
 
    m_versionBr=2;
    
    if(ReadBoostRoot::GetTof() == 3  || ReadBoostRoot::GetTof() == 4 )
    {  
      m_versionEc=4;
      std::cout << std::endl << "BesTofDigitizer       You are using the new EC Tof (MRPC) Digitizer!" << std::endl;
    }  
    else
      {
    m_versionEc=3;
    std::cout << std::endl << "BesTofDigitizer       You are using the old EC Tof Digitizer!" << std::endl;
      }  
 
 
 
}
 
BesTofDigitizer::~BesTofDigitizer()
{
    //delete m_digiMessenger;
  delete tofDMEc4;
}
 
void BesTofDigitizer::Digitize()
{
    m_besTofDigitsCollection=new BesTofDigitsCollection (moduleName, collectionName[0]);
    G4DigiManager* digiManager = G4DigiManager::GetDMpointer();
    G4int THCID = digiManager->GetHitsCollectionID("BesTofHitsCollection");
    m_THC = (BesTofHitsCollection*) (digiManager->GetHitsCollection(THCID));
    
 
 
 
 
 
    if(m_THC) //Haben wir eine Hit-Collection?
     {
 
    
        m_scintGroup = new  vector<ScintSingle*>;//ScintSingle contains: partId, scintnb, edep <--only the basics   
        GroupHits(); //If one finds more hits in one scintillator, they are grouped together 
        G4int size=m_scintGroup->size(); 
 
    //std::cout << "BesTofDigitizer     m_scintGroup->size():  " << size << std::endl;
    
 
        ScintSingle* scint;
        G4int partId;
        for (G4int i=0;i<size;i++) 
        {
            scint = (*m_scintGroup)[i];
            partId = scint->GetPartId();
 
        
 
            if (partId==1) // for barrel part
            {
                if (m_versionBr==1)
                    tofDMBr1->Digitize(scint,m_besTofDigitsCollection);
                else if (m_versionBr==2){
            //std::cout << " tofDMBr2->Digitize" << std::endl;
                    tofDMBr2->Digitize(scint,m_besTofDigitsCollection);
        }
                else
                    tofDMBr2->Digitize(scint,m_besTofDigitsCollection);
            }
            else // for endcap part
            {
                if (m_versionEc==1)
                    tofDMEc1->Digitize(scint,m_besTofDigitsCollection);
                else if (m_versionEc==2)
                    tofDMEc2->Digitize(scint,m_besTofDigitsCollection);
                else if (m_versionEc==3){
            tofDMEc3->Digitize(scint,m_besTofDigitsCollection);}
        else if (m_versionEc==4)
            tofDMEc4->Digitize(scint,m_besTofDigitsCollection);
        else
                {
                    cout << "Warning: BesTofDigitizerEcV2 is utilized otherwise!" << endl;
                    tofDMEc2->Digitize(scint,m_besTofDigitsCollection);
                }
            }//Close endcappart
        }//close for 
        if (m_scintGroup)
        {
            for (size_t i=0;i<m_scintGroup->size();i++)
            {
                delete (*m_scintGroup)[i];
                //std::cout << "i: " << i << std::endl;
                //std::cout << "delete m_scintGroup[i] successfully " << std::endl;
            }
            m_scintGroup->clear();
            delete m_scintGroup;
        }
          
     }//close if(m_THC)
 
    StoreDigiCollection(m_besTofDigitsCollection);
}
 
void BesTofDigitizer::GroupHits()
{
    G4int partId, scinNb,size,flag;
    G4double edep;
    BesTofHit* hit;
    G4int nHits = m_THC->entries();
 
    //group the hits which are in the same scintillator
    for (G4int i=0;i<nHits;i++)
    {
        hit=(*m_THC)[i];
        partId=hit->GetPartId();
        scinNb=hit->GetScinNb();
        edep = hit->GetEdep();
        size=m_scintGroup->size();
        flag=0;
        if (size>0)
        {
            ScintSingle* oldScint;
            for (G4int j=0; j<size;j++)
            {
                oldScint = (*m_scintGroup)[j];
                if (oldScint->GetPartId()==partId &&oldScint->GetScinNb()==scinNb)
                {
                    
          //std::cout << "BesTodDigitizer:       GroupHits: 'Oldhit:'  partID |  scinnb :  "   << partId   << "  |  "  << scinNb << std::endl;   
 
            oldScint->GetHitIndexes()->push_back(i);
                    oldScint->AddEdep(edep);
                    flag=1;
                    break;
                }
            }
        }
        if (flag==0)
        {
            ScintSingle* newScint = new ScintSingle;
            newScint->SetPartId(partId);
            newScint->SetScinNb(scinNb);
            newScint->SetEdep(edep);
            newScint->GetHitIndexes()->push_back(i);
            m_scintGroup->push_back(newScint);
        
        //std::cout << "BesTodDigitizer:        GroupHits:  NewHitGroup:   " << partId  << "   "  << scinNb << std::endl;  
 
     }
    }
}