PHG4ConeSteppingAction.cc

Go to the documentation of this file.

#include "PHG4ConeSteppingAction.h"
#include "PHG4ConeDetector.h"
 
#include <g4main/PHG4HitContainer.h>
#include <g4main/PHG4Hit.h>
#include <g4main/PHG4Hitv1.h>
#include <g4main/PHG4Shower.h>
#include <g4main/PHG4TrackUserInfoV1.h>
 
#include <phool/getClass.h>
 
#include <Geant4/G4Step.hh>
 
#include <iostream>
 
using namespace std;
//____________________________________________________________________________..
PHG4ConeSteppingAction::PHG4ConeSteppingAction( PHG4ConeDetector* detector ):
  detector_( detector ),
  hits_(NULL),
  hit(NULL),
  saveshower(NULL)
{}
 
PHG4ConeSteppingAction::~PHG4ConeSteppingAction()
{
  // if the last hit was a zero energie deposit hit, it is just reset
  // and the memory is still allocated, so we need to delete it here
  // if the last hit was saved, hit is a NULL pointer which are
  // legal to delete (it results in a no operation)
  delete hit;
}
 
//____________________________________________________________________________..
bool PHG4ConeSteppingAction::UserSteppingAction( const G4Step* aStep, bool )
{
 
  // get volume of the current step
  G4VPhysicalVolume* volume = aStep->GetPreStepPoint()->GetTouchableHandle()->GetVolume();
 
  // collect energy and track length step by step
  G4double edep = aStep->GetTotalEnergyDeposit() / GeV;
 
  const G4Track* aTrack = aStep->GetTrack();
 
  int layer_id =  detector_->get_Layer();
  // make sure we are in a volume
  if ( detector_->IsInConeActive(volume) )
    {
      bool geantino = false;
      // the check for the pdg code speeds things up, I do not want to make 
      // an expensive string compare for every track when we know
      // geantino or chargedgeantino has pid=0
      if (aTrack->GetParticleDefinition()->GetPDGEncoding() == 0 &&
          aTrack->GetParticleDefinition()->GetParticleName().find("geantino") != string::npos)
        {
          geantino = true;
        }
      G4StepPoint * prePoint = aStep->GetPreStepPoint();
      G4StepPoint * postPoint = aStep->GetPostStepPoint();
//       cout << "track id " << aTrack->GetTrackID() << endl;
//       cout << "time prepoint: " << prePoint->GetGlobalTime() << endl;
//       cout << "time postpoint: " << postPoint->GetGlobalTime() << endl;
      switch (prePoint->GetStepStatus())
        {
        case fGeomBoundary:
        case fUndefined:
          if (! hit)
            {
              hit = new PHG4Hitv1();
            }
          //here we set the entrance values in cm
          hit->set_x( 0, prePoint->GetPosition().x() / cm);
          hit->set_y( 0, prePoint->GetPosition().y() / cm );
          hit->set_z( 0, prePoint->GetPosition().z() / cm );
          // time in ns
          hit->set_t( 0, prePoint->GetGlobalTime() / nanosecond );
          //set the track ID
          hit->set_trkid(aTrack->GetTrackID());
          //set the initial energy deposit
          hit->set_edep(0);
 
          if ( G4VUserTrackInformation* p = aTrack->GetUserInformation() )
            {
              if ( PHG4TrackUserInfoV1* pp = dynamic_cast<PHG4TrackUserInfoV1*>(p) )
                {
                  hit->set_trkid(pp->GetUserTrackId());
                  hit->set_shower_id(pp->GetShower()->get_id());
                  saveshower = pp->GetShower();
                }
            }
          
          break;
        default:
          break;
        }
      // here we just update the exit values, it will be overwritten
      // for every step until we leave the volume or the particle
      // ceases to exist
      hit->set_x( 1, postPoint->GetPosition().x() / cm );
      hit->set_y( 1, postPoint->GetPosition().y() / cm );
      hit->set_z( 1, postPoint->GetPosition().z() / cm );
 
      hit->set_t( 1, postPoint->GetGlobalTime() / nanosecond );
      //sum up the energy to get total deposited
      hit->set_edep(hit->get_edep() + edep);
      if (geantino)
        {
          hit->set_edep(-1); // only energy=0 g4hits get dropped, this way geantinos survive the g4hit compression
        }
      if (edep > 0)
        {
            if ( G4VUserTrackInformation* p = aTrack->GetUserInformation() )
              {
                if ( PHG4TrackUserInfoV1* pp = dynamic_cast<PHG4TrackUserInfoV1*>(p) )
                  {
                    pp->SetKeep(1); // we want to keep the track
                  }
              }
        }
 
      // if any of these conditions is true this is the last step in
      // this volume and we need to save the hit
      // postPoint->GetStepStatus() == fGeomBoundary: track leaves this volume
      // postPoint->GetStepStatus() == fWorldBoundary: track leaves this world
      // (not sure if this will ever be the case)
      // aTrack->GetTrackStatus() == fStopAndKill: track ends
      if (postPoint->GetStepStatus() == fGeomBoundary || postPoint->GetStepStatus() == fWorldBoundary|| aTrack->GetTrackStatus() == fStopAndKill)
        {
          // save only hits with energy deposit (or -1 for geantino)
          if (hit->get_edep())
            {
              hits_->AddHit(layer_id, hit);
              if (saveshower)
                {
                  saveshower->add_g4hit_id(hits_->GetID(),hit->get_hit_id());
                }
              // ownership has been transferred to container, set to null
              // so we will create a new hit for the next track
              hit = NULL;
            }
          else
            {
              // if this hit has no energy deposit, just reset it for reuse
              // this means we have to delete it in the dtor. If this was
              // the last hit we processed the memory is still allocated
              hit->Reset();
            }
        }
       // return true to indicate the hit was used
      return true;
 
    }
  else
    {
      return false;
    }
}
 
//____________________________________________________________________________..
void PHG4ConeSteppingAction::SetInterfacePointers( PHCompositeNode* topNode )
{
 
  string hitnodename;
  if (detector_->SuperDetector() != "NONE")
    {
      hitnodename = "G4HIT_" + detector_->SuperDetector();
    }
  else
    {
      hitnodename = "G4HIT_" + detector_->GetName();
    }
 
  //now look for the map and grab a pointer to it.
  hits_ =  findNode::getClass<PHG4HitContainer>( topNode , hitnodename.c_str() );
 
  // if we do not find the node we need to make it.
  if ( ! hits_ )
    { std::cout << "PHG4ConeSteppingAction::SetTopNode - unable to find " << hitnodename << std::endl; }
 
}