GT_SD.cc

Go to the documentation of this file.

//! \file
//!
//! Source file for Gem_Tracker sensitive detector class.
//!
//! The GT_SD class defines the member functions for processing the hit
//! information whenever a hit is simulated in the detector.  These
//! routines initialise the hit collection, process hits, and perform
//! the end of event action which fills the hit information arrays.
//!
//! \author D.K. Hasell
//! \version 1.0
//! \date 2010-10-15
//!
//! \ingroup detector

// *+****1****+****2****+****3****+****4****+****5****+****6****+****7****+****

// Include the GT header files and the user header files referenced here.

#include "GT_SD.h"
#include "GT_Hit.h"
#include "GT_Data.h"
#include "GT_Messenger.h"

#include "EventAction.h"

// Include the GEANT4 header files referenced in this file.

#include "G4VSensitiveDetector.hh"
#include "G4HCofThisEvent.hh"
#include "G4SDManager.hh"
#include "G4Step.hh"
#include "G4TouchableHistory.hh"
#include "G4StepPoint.hh"
#include "G4TouchableHandle.hh"
#include "G4ThreeVector.hh"

#include "Randomize.hh"

// Use the standard namespace.

using namespace std;

// *+****1****+****2****+****3****+****4****+****5****+****6****+****7****+****

// GT_SD source code.

// Constructor.

GT_SD::GT_SD( G4String name ) : G4VSensitiveDetector( name ) {

   // Insert hits collection name.

   collectionName.insert( "GT_HC" );

   // Create pointer to GT_SD Messenger.

   GT_Messenger::Instance()->setGT_SDptr( this );

   // Set defaults.

   GT_threshold =   1.0 * eV;
   GT_Xresol    = 100.0 * um;
   GT_Yresol    = 100.0 * um;

   // Initially transformations are not defined so Boolean is false.

   for( int i = 0; i < N_GT; ++i ) GT_Transform[i] = false;
}

// Destructor.

GT_SD::~GT_SD(){}

// Initialize hits collection.

void GT_SD::Initialize( G4HCofThisEvent * HCE ) {

   GT_HC = new GT_HitsCollection( SensitiveDetectorName, collectionName[0] ); 

   static G4int HCID = -1;
   if( HCID < 0 ) HCID = G4SDManager::GetSDMpointer()->
                     GetCollectionID(collectionName[0]);

   HCE->AddHitsCollection( HCID, GT_HC );
}

// Process hits.

G4bool GT_SD::ProcessHits( G4Step * step, G4TouchableHistory * ROhist ) {

   // Get the total energy deposited in this step.

   G4double edep = step->GetTotalEnergyDeposit();

   // Skip this hit if below threshold.

   if( edep < GT_threshold ) return false;

   // Get the PreStepPoint and TouchableHandle.

   G4StepPoint * prestep = step->GetPreStepPoint();

   G4TouchableHandle touchable = prestep->GetTouchableHandle();

   // Get copy number (used to identify which GT).

   G4int copyno = touchable->GetCopyNumber();

   // Get the track.

   G4Track * track = step->GetTrack();

   // Get the global time and track ID.

   G4double time = prestep->GetGlobalTime();
   G4int tr = track->GetTrackID();

   // Get the true world and local coordinate positions.

   G4ThreeVector trueworld = prestep->GetPosition();

   // If this first time for this detector store transformation and inverse.

   if( !GT_Transform[copyno] ) {
      GT_WorldtoLocal[copyno] = touchable->GetHistory()->GetTopTransform();
      GT_LocaltoWorld[copyno] = GT_WorldtoLocal[copyno].Inverse();
      GT_Transform[copyno] = true;
   }

   G4ThreeVector truelocal = GT_WorldtoLocal[copyno].TransformPoint(trueworld);

   // Create local and world coordinates smeared by resolution.

   G4ThreeVector local(
      truelocal.x() + CLHEP::RandGauss::shoot( 0.0, GT_Xresol ),
      truelocal.y() + CLHEP::RandGauss::shoot( 0.0, GT_Yresol ),
      truelocal.z() );

   G4ThreeVector world = GT_LocaltoWorld[copyno].TransformPoint(local);

   // Create a pointer to a new Hit.

   GT_Hit * hit = new GT_Hit();

   // Fill the Hit information.

   hit->copyno  = copyno;
   hit->trackid = tr;
   hit->edep    = edep;
   hit->time    = time;
   hit->tworld  = trueworld;
   hit->tlocal  = truelocal;
   hit->world   = world;
   hit->local   = local;

   // Add the Hit to the Hit Collection.

   GT_HC->insert( hit );
   
   return true;
}

// Routine to process the hits at the end of an event.

void GT_SD::EndOfEvent( G4HCofThisEvent * HCE ){

   GT_Data * gtdata = EventAction::gtdata;

   gtdata->Reset();

   G4int N_Hits = GT_HC->entries();
   
   gtdata->nGT = N_Hits;

   for( G4int i = 0; i < N_Hits; ++i ) {

      gtdata->id.push_back( ( *GT_HC )[i]->copyno );
      gtdata->tr.push_back( ( *GT_HC )[i]->trackid );

      gtdata->e.push_back( ( *GT_HC )[i]->edep / eV );
      gtdata->t.push_back( ( *GT_HC )[i]->time / ns );

      gtdata->tx.push_back( ( *GT_HC )[i]->tworld.x() / cm );
      gtdata->ty.push_back( ( *GT_HC )[i]->tworld.y() / cm );
      gtdata->tz.push_back( ( *GT_HC )[i]->tworld.z() / cm );

      gtdata->txl.push_back( ( *GT_HC )[i]->tlocal.x() / cm );
      gtdata->tyl.push_back( ( *GT_HC )[i]->tlocal.y() / cm );
      gtdata->tzl.push_back( ( *GT_HC )[i]->tlocal.z() / cm );

      gtdata->x.push_back( ( *GT_HC )[i]->world.x() / cm );
      gtdata->y.push_back( ( *GT_HC )[i]->world.y() / cm );
      gtdata->z.push_back( ( *GT_HC )[i]->world.z() / cm );

      gtdata->xl.push_back( ( *GT_HC )[i]->local.x() / cm );
      gtdata->yl.push_back( ( *GT_HC )[i]->local.y() / cm );
      gtdata->zl.push_back( ( *GT_HC )[i]->local.z() / cm );
   }

}

void GT_SD::clear(){} 

void GT_SD::DrawAll() {} 

// Print all the hits.

void GT_SD::PrintAll() {
   G4int N_Hits = GT_HC->entries();
   G4cout << "\nGT Hits Collection N_Hits = " << N_Hits << "\n" << G4endl; 
   for ( G4int i = 0; i < N_Hits; ++i ) ( *GT_HC )[i]->Print();
} 

// Set/Get energy threshold.

G4double GT_SD::setThreshold( G4double thres ) { return GT_threshold = thres; }
G4double GT_SD::getThreshold() { return GT_threshold; }

// Set/Get X resolution.

G4double GT_SD::setXresol( G4double res ) { return GT_Xresol = res; }
G4double GT_SD::getXresol() { return GT_Xresol; }

// Set/Get Y resolution.

G4double GT_SD::setYresol( G4double res ) { return GT_Yresol = res; }
G4double GT_SD::getYresol() { return GT_Yresol; }