SM_SD.cc

Go to the documentation of this file.

//! \file
//!
//! Source file for Symmetric_Moeller 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 Y. Ma based on code for olympus simulation package by D.K. Hasell
//! \version 1.0
//! \date 2010-10-11
//!
//! \ingroup detector

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

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

#include "SM_SD.h"
#include "SM_Hit.h"
#include "SM_Data.h"
#include "SM_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****+****

// SM_SD source code.

SM_SD::SM_SD( G4String name ) : G4VSensitiveDetector( name ) {

   // Insert hits collection name.

   collectionName.insert( "SM_HC" );

   // Create pointer to GT SD Messenger.

   SM_Messenger::Instance()->setSM_SDptr( this );

   // Set defaults.

   SM_threshold = 0. * eV;
   SM_Eresol = 0. * eV;

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

   for( int i = 0; i < N_SM; ++i ) SM_Transform[i] = false;
}

// Destructor.

SM_SD::~SM_SD(){}

// Initialize hits collection.

void SM_SD::Initialize( G4HCofThisEvent * HCE ) {

   SM_HC = new
      SM_HitsCollection( SensitiveDetectorName, collectionName[0] ); 

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

   HCE->AddHitsCollection( HCID, SM_HC );
}

// Process hits.

G4bool SM_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 < SM_threshold ) return false;

   // Get the PreStepPoint and the TouchableHandle.

   G4StepPoint * prestep = step->GetPreStepPoint();

   G4TouchableHandle touchable = prestep->GetTouchableHandle();

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

   G4int copyno = touchable->GetCopyNumber();

   // Get the track.

   G4Track * track = step->GetTrack();

   // Get the global time and track ID.

   G4double time = prestep->GetGlobalTime();
   G4int id = 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( !SM_Transform[copyno] ) {
      SM_WorldtoLocal[copyno] = touchable->GetHistory()->GetTopTransform();
      SM_LocaltoWorld[copyno] = SM_WorldtoLocal[copyno].Inverse();
      SM_Transform[copyno] = true;
   }

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

   // Create local and world coordinates smeared by resolution.

   G4ThreeVector local = truelocal;

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

   // Create a pointer to a new Hit.

   SM_Hit * hit = new SM_Hit();

   // Fill the Hit information.

   hit->copyno  = copyno;
   hit->trackid = id;
   hit->edep    = edep + CLHEP::RandGauss::shoot( 0.0, SM_Eresol );
   hit->time    = time;
   hit->tworld  = trueworld;
   hit->tlocal  = truelocal;
   hit->world   = world;
   hit->local   = local;

   // Add the Hit to the Hit Collection.
  
 SM_HC->insert( hit );
   
   //hit->Print();

   return true;
}

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

void SM_SD::EndOfEvent( G4HCofThisEvent * HCE ){

   SM_Data * smdata = EventAction::smdata;

   smdata->Reset();

   G4int N_Hits = SM_HC->entries();
   
   smdata->nSM = N_Hits;

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

      smdata->id.push_back( ( *SM_HC )[i]->copyno );
      smdata->tr.push_back( ( *SM_HC )[i]->trackid );

      smdata->e.push_back( ( *SM_HC )[i]->edep / MeV );
      smdata->t.push_back( ( *SM_HC )[i]->time / ns );

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

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

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

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

   }

}

void SM_SD::clear(){} 

void SM_SD::DrawAll() {} 

// Print all the hits.

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

// Set/Get energy threshold.

G4double SM_SD::setThreshold( G4double thres ) { return SM_threshold = thres; }
G4double SM_SD::getThreshold() { return SM_threshold; }

// Set/Get Energy resolution.

G4double SM_SD::setEresol( G4double res ) { return SM_Eresol = res; }
G4double SM_SD::getEresol() { return SM_Eresol; }