source: trunk/examples/advanced/gammaray_telescope/src/GammaRayTelDigitizer.cc @ 1272

//
// ********************************************************************
// * License and Disclaimer                                           *
// *                                                                  *
// * The  Geant4 software  is  copyright of the Copyright Holders  of *
// * the Geant4 Collaboration.  It is provided  under  the terms  and *
// * conditions of the Geant4 Software License,  included in the file *
// * LICENSE and available at  http://cern.ch/geant4/license .  These *
// * include a list of copyright holders.                             *
// *                                                                  *
// * Neither the authors of this software system, nor their employing *
// * institutes,nor the agencies providing financial support for this *
// * work  make  any representation or  warranty, express or implied, *
// * regarding  this  software system or assume any liability for its *
// * use.  Please see the license in the file  LICENSE  and URL above *
// * for the full disclaimer and the limitation of liability.         *
// *                                                                  *
// * This  code  implementation is the result of  the  scientific and *
// * technical work of the GEANT4 collaboration.                      *
// * By using,  copying,  modifying or  distributing the software (or *
// * any work based  on the software)  you  agree  to acknowledge its *
// * use  in  resulting  scientific  publications,  and indicate your *
// * acceptance of all terms of the Geant4 Software license.          *
// ********************************************************************
//
//
// $Id: GammaRayTelDigitizer.cc,v 1.6 2006/06/29 15:56:31 gunter Exp $
// GEANT4 tag $Name: geant4-09-03-cand-01 $
// ------------------------------------------------------------
//      GEANT 4 class implementation file
//      CERN Geneva Switzerland
//
//
//      ------------ GammaRayTelDigitizer  ------
//           by   F.Longo, R.Giannitrapani & G.Santin (13 nov 2000)
//
// ************************************************************

#include "GammaRayTelDigitizer.hh"
#include "GammaRayTelDigi.hh"
#include "GammaRayTelDigitizerMessenger.hh"

#include "GammaRayTelTrackerHit.hh"
#include "GammaRayTelCalorimeterHit.hh"
#include "GammaRayTelAnticoincidenceHit.hh"

#include "G4EventManager.hh"
#include "G4Event.hh"
#include "G4SDManager.hh"
#include "G4DigiManager.hh"
#include "G4ios.hh"

//#include "G4CollectionNameVector.hh"
#include <vector>

//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo....

GammaRayTelDigitizer::GammaRayTelDigitizer(G4String name)
  :G4VDigitizerModule(name)
{


  G4String colName = "DigitsCollection";
  collectionName.push_back(colName);
  Energy_Threshold = 120.*keV;
  TotalEnergy = 0.;
  ACDThreshold = 15*keV;

//create a messenger for this class
  digiMessenger = new GammaRayTelDigitizerMessenger(this);

}

//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo....

GammaRayTelDigitizer::~GammaRayTelDigitizer()
{
  delete digiMessenger;
}

//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo....

void GammaRayTelDigitizer::Digitize()
{
  DigitsCollection = new GammaRayTelDigitsCollection
        ("GammaRayTelDigitizer","DigitsCollection"); // to create the Digi Collection
  
  G4DigiManager* DigiMan = G4DigiManager::GetDMpointer();
  
  G4int THCID; // TrackerCollection
  G4int CHCID; // CalorimeterCollection
  G4int AHCID; // AnticoincidenceCollection

  TotalEnergy = 0.; // at each event
  
  // TKR Hits collection
  
  THCID = DigiMan->GetHitsCollectionID("TrackerCollection");  
  GammaRayTelTrackerHitsCollection* THC = 0; 
  THC = (GammaRayTelTrackerHitsCollection*)
    (DigiMan->GetHitsCollection(THCID));


  // CAL Hits collection

  CHCID = DigiMan->GetHitsCollectionID("CalorimeterCollection");
  GammaRayTelCalorimeterHitsCollection* CHC = 0;
  CHC = (GammaRayTelCalorimeterHitsCollection*)
                 (DigiMan->GetHitsCollection(CHCID));

  // ACD Hits collection

  AHCID = DigiMan->GetHitsCollectionID("AnticoincidenceCollection");  
  GammaRayTelAnticoincidenceHitsCollection* AHC = 0;
  AHC = (GammaRayTelAnticoincidenceHitsCollection*)
                 (DigiMan->GetHitsCollection(AHCID));

  
  if (THC)
    {
      G4int n_hit = THC->entries();

      for (G4int i=0;i<n_hit;i++)
        {
          G4double ESil = (*THC)[i]->GetEdepSil();
          G4int NStrip = (*THC)[i]->GetNStrip();
          G4int NPlane = (*THC)[i]->GetNSilPlane();
          G4int IsX = (*THC)[i]->GetPlaneType();
          
       // digi generation only if energy deposit greater than threshold

          if (ESil>Energy_Threshold)
            {
              GammaRayTelDigi* Digi = new GammaRayTelDigi();
              Digi->SetPlaneNumber(NPlane);
              Digi->SetPlaneType(IsX);
              Digi->SetStripNumber(NStrip);
              Digi->SetDigiType(0);
              Digi->SetEnergy(0.);
              DigitsCollection->insert(Digi);   
            }  
        }
    }
  if (CHC)
    {
      G4int n_hit = CHC->entries();
      
      for (G4int i=0;i<n_hit;i++)
        {
          TotalEnergy +=(*CHC)[i]->GetEdepCAL();
        }
      // digi generation only if energy deposit greater than 0.
      
      if (TotalEnergy>0.)
        {
          GammaRayTelDigi* Digi = new GammaRayTelDigi();
          Digi->SetPlaneNumber(0);
          Digi->SetPlaneType(0);
          Digi->SetStripNumber(0);
          Digi->SetDigiType(1);
          Digi->SetEnergy(TotalEnergy);
          DigitsCollection->insert(Digi);       
        }  

    }

  if (AHC)
    {
      G4int n_hit = AHC->entries();
      
      for (G4int i=0;i<n_hit;i++)
        {
          G4double energy = (*AHC)[i]->GetEdepACD();
          G4int type = (*AHC)[i]->GetACDTileNumber();
          
          // digi generation only if energy deposit greater than 0.
          
          if (energy>ACDThreshold)
            {
              GammaRayTelDigi* Digi = new GammaRayTelDigi();
              Digi->SetPlaneNumber(0);
              Digi->SetPlaneType(0);
              Digi->SetStripNumber(type);
              Digi->SetDigiType(2);
              Digi->SetEnergy(energy);
              DigitsCollection->insert(Digi);   
            }  
          
        }
    }
  
  if (THC||AHC||CHC){
    G4cout << "Number of digits in this event =  " 
           << DigitsCollection->entries()  
      // << " " << DigitsCollection->GetName()  
      // << " " << DigitsCollection->GetDMname() 
           << G4endl;
  }
  
  StoreDigiCollection(DigitsCollection);
  
  G4int DCID = -1;
  if(DCID<0)
    { 
      //          DigiMan->List();
      DCID = DigiMan->GetDigiCollectionID("GammaRayTelDigitizer/DigitsCollection");
    }
  
  
}



//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo....