TF_SD.cc

Go to the documentation of this file.

//! \file
//!
//! Source file for Time_of_Flight sensitive detector class.
//!
//! The TF_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-30
//!
//! \ingroup detector

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

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

#include "TF_SD.h"
#include "TF_Hit.h"
#include "TF_Data.h"
#include "TF_Messenger.h"

#include "EventAction.h"

// Include the GEANT4 header files used below.

#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****+****

// TF_SD source code.

// Constructor.

TF_SD::TF_SD( G4String name ) : G4VSensitiveDetector( name ) {

   // Insert hits collection name.

   collectionName.insert( "TF_HC" );

   // Create pointer to GT SD Messenger.

   TF_Messenger::Instance()->setTF_SDptr( this );

   // Set defaults.

   TF_threshold = 100.0 * eV;
   TF_Tresol    =   1.0 * ns;
   TF_Yresol    =  10.0 * cm;

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

   for( int i = 0; i < N_TF; ++i ) TF_Transform[i] = false;
}

// Destructor.

TF_SD::~TF_SD(){}

// Initialize hits collection.

void TF_SD::Initialize( G4HCofThisEvent * HCE ) {

   TF_HC = new TF_HitsCollection( SensitiveDetectorName, collectionName[0] ); 

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

   HCE->AddHitsCollection( HCID, TF_HC );
}

// Process hits.

G4bool TF_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 < TF_threshold ) return false;

   // Get the PreStepPoint and the TouchableHandle.

   G4StepPoint * prestep = step->GetPreStepPoint();

   G4TouchableHandle touchable = prestep->GetTouchableHandle();

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

   G4int copyno = touchable->GetCopyNumber();

   // Get the track.

   G4Track * track = step->GetTrack();

   // Get the global time and track ID.

   G4double ttime = prestep->GetGlobalTime();
   G4double time  = ttime + CLHEP::RandGauss::shoot( 0.0, TF_Tresol );

   G4int id = track->GetTrackID();

   // Get the true world and local coordinate positions.

   G4ThreeVector trueworld = prestep->GetPosition();

   // If this first time store transformation and inverse.

   if( !TF_Transform[copyno] ) {
      TF_WorldtoLocal[copyno] = touchable->GetHistory()->GetTopTransform();
      TF_LocaltoWorld[copyno] = TF_WorldtoLocal[copyno].Inverse();
      TF_Transform[copyno] = true;
   }

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

   // Create local and world coordinates smeared by resolution.

   G4ThreeVector local( 0.0, 
      truelocal.y() + CLHEP::RandGauss::shoot( 0.0, TF_Yresol ), 0.0 );

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

   // Create a pointer to a new Hit.

   TF_Hit * hit = new TF_Hit();

   // Fill the Hit information.

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

   // Add the Hit to the Hit Collection.

   TF_HC->insert( hit );
   
   //hit->Print();

   return true;
}

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

void TF_SD::EndOfEvent( G4HCofThisEvent * HCE ){

   TF_Data * tfdata = EventAction::tfdata;

   tfdata->Reset();

   G4int N_Hits = TF_HC->entries();
   
   tfdata->nTF = N_Hits;

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

      tfdata->id.push_back( ( *TF_HC )[i]->copyno );
      tfdata->tr.push_back( ( *TF_HC )[i]->trackid );

      tfdata->e.push_back( ( *TF_HC )[i]->edep / MeV );
      tfdata->tt.push_back( ( *TF_HC )[i]->ttime / ns );
      tfdata->t.push_back( ( *TF_HC )[i]->time / ns );

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

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

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

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

}

void TF_SD::clear(){} 

void TF_SD::DrawAll() {} 

// Print all the hits.

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

// Set/Get energy threshold.

G4double TF_SD::setThreshold( G4double thres ) { return TF_threshold = thres; }
G4double TF_SD::getThreshold() { return TF_threshold; }

// Set/Get T resolution.

G4double TF_SD::setTresol( G4double res ) { return TF_Tresol = res; }
G4double TF_SD::getTresol() { return TF_Tresol; }

// Set/Get Y resolution.

G4double TF_SD::setYresol( G4double res ) { return TF_Yresol = res; }
G4double TF_SD::getYresol() { return TF_Yresol; }