PHG4TRDSteppingAction.cc

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file PHG4TRDSteppingAction.cc

#include "PHG4TRDSteppingAction.h"
#include "PHG4TRDDetector.h"
#include "PHG4TRDSubsystem.h"

//#include "PHG4StepStatusDecode.h"

#include <phparameter/PHParameters.h>

#include <g4main/PHG4Hit.h>
#include <g4main/PHG4HitContainer.h>
#include <g4main/PHG4Hitv1.h>
#include <g4main/PHG4Shower.h>
#include <g4main/PHG4SteppingAction.h>  // for PHG4SteppingAction
#include <g4main/PHG4TrackUserInfoV1.h>

#include <phool/getClass.h>

#include <Geant4/G4ParticleDefinition.hh>      // for G4ParticleDefinition
#include <Geant4/G4ReferenceCountedHandle.hh>  // for G4ReferenceCountedHandle
#include <Geant4/G4Step.hh>
#include <Geant4/G4StepPoint.hh>   // for G4StepPoint
#include <Geant4/G4StepStatus.hh>  // for fGeomBoundary, fPostSt...
#include <Geant4/G4SystemOfUnits.hh>
#include <Geant4/G4ThreeVector.hh>            // for G4ThreeVector
#include <Geant4/G4TouchableHandle.hh>        // for G4TouchableHandle
#include <Geant4/G4Track.hh>                  // for G4Track
#include <Geant4/G4TrackStatus.hh>            // for fStopAndKill
#include <Geant4/G4Types.hh>                  // for G4double
#include <Geant4/G4VPhysicalVolume.hh>        // for G4VPhysicalVolume
#include <Geant4/G4VTouchable.hh>             // for G4VTouchable
#include <Geant4/G4VUserTrackInformation.hh>  // for G4VUserTrackInformation

#include <boost/io/ios_state.hpp>

#include <cmath>    // for isfinite, copysign
#include <cstdlib>  // for exit
#include <iomanip>
#include <iostream>
#include <string>  // for operator<<, char_traits

class PHCompositeNode;

using namespace std;

//____________________________________________________________________________..
//PHG4TRDSteppingAction::PHG4TRDSteppingAction(PHG4TRDSubsystem *subsys, PHG4TRDDetector* detector, const PHParameters* parameters)
PHG4TRDSteppingAction::PHG4TRDSteppingAction(PHG4TRDDetector* detector, const PHParameters* parameters)
  : PHG4SteppingAction(detector->GetName())
  , m_Detector(detector)
  , m_Params(parameters)
  , m_HitContainer(nullptr)
  , m_ActiveGasHits(nullptr)
  , m_Hit(nullptr)
  , m_SaveHitContainer(nullptr)
  , m_SaveShower(nullptr)
  , m_SaveVolPre(nullptr)
  , m_SaveVolPost(nullptr)
  , m_SaveTrackId(-1)
  , m_SavePreStepStatus(-1)
  , m_SavePostStepStatus(-1)
  , m_BlackHoleFlag(m_Params->get_int_param("blackhole"))
  , m_ActiveFlag(m_Params->get_int_param("active"))
  , m_UseG4StepsFlag(m_Params->get_int_param("use_g4steps"))
  , m_Zmin(m_Params->get_double_param("PosZ") * cm - m_Params->get_double_param("ThicknessZ") * cm / 2.)
  , m_Zmax(m_Params->get_double_param("PosZ") * cm + m_Params->get_double_param("ThicknessZ") * cm / 2.)
  //, m_EdepSum(0)
{
  m_Zmin -= copysign(m_Zmin, 1. / 1e6 * cm);
  m_Zmax += copysign(m_Zmax, 1. / 1e6 * cm);
}

PHG4TRDSteppingAction::~PHG4TRDSteppingAction()
{
  // 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 nullptr pointer which are
  // legal to delete (it results in a no operation)
  delete m_Hit;
}

//____________________________________________________________________________..
bool PHG4TRDSteppingAction::UserSteppingAction(const G4Step* aStep, bool)
{
  // get volume of the current step
  G4TouchableHandle touch = aStep->GetPreStepPoint()->GetTouchableHandle();
  G4TouchableHandle touchpost = aStep->GetPostStepPoint()->GetTouchableHandle();

  G4VPhysicalVolume* volume = touch->GetVolume();
  // G4 just calls  UserSteppingAction for every step (and we loop then over all our
  // steppingactions. First we have to check if we are actually in our volume
  if (!m_Detector->IsInTRD(volume))
  {
    return false;
  }

  int whichactive = m_Detector->IsInTRD(volume);

  // collect energy and track length step by step
  G4double edep = aStep->GetTotalEnergyDeposit() / GeV;

  const G4Track* aTrack = aStep->GetTrack();
  // if this volume stops everything, just put all kinetic energy into edep
  if (m_BlackHoleFlag)
  {
    edep = aTrack->GetKineticEnergy() / GeV;
    G4Track* killtrack = const_cast<G4Track*>(aTrack);
    killtrack->SetTrackStatus(fStopAndKill);
  }
  int layer_id = m_Detector->get_Layer();

  if (m_ActiveFlag)
  {
    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") != std::string::npos)
    {
      geantino = true;
    }
    G4StepPoint* prePoint = aStep->GetPreStepPoint();
    G4StepPoint* postPoint = aStep->GetPostStepPoint();
    //        std::cout << "time prepoint: " << prePoint->GetGlobalTime()/ns << std::endl;
    //        std::cout << "time postpoint: " << postPoint->GetGlobalTime()/ns << std::endl;
    //        std::cout << "kinetic energy: " <<  aTrack->GetKineticEnergy()/GeV << std::endl;
    //       G4ParticleDefinition* def = aTrack->GetDefinition();
    //       std::cout << "Particle: " << def->GetParticleName() << std::endl;
    int prepointstatus = prePoint->GetStepStatus();
    if (prepointstatus == fGeomBoundary ||
        prepointstatus == fUndefined ||
        (prepointstatus == fPostStepDoItProc && m_SavePostStepStatus == fGeomBoundary) ||
        m_UseG4StepsFlag > 0)
    {
      if (!m_Hit)
      {
        m_Hit = new PHG4Hitv1();
      }

      m_Hit->set_layer((unsigned int) layer_id);

      //here we set the entrance values in cm
      m_Hit->set_x(0, prePoint->GetPosition().x() / cm);
      m_Hit->set_y(0, prePoint->GetPosition().y() / cm);
      m_Hit->set_z(0, prePoint->GetPosition().z() / cm);

      m_Hit->set_px(0, prePoint->GetMomentum().x() / GeV);
      m_Hit->set_py(0, prePoint->GetMomentum().y() / GeV);
      m_Hit->set_pz(0, prePoint->GetMomentum().z() / GeV);

      // time in ns
      m_Hit->set_t(0, prePoint->GetGlobalTime() / nanosecond);
      //set and save the track ID
      m_Hit->set_trkid(aTrack->GetTrackID());
      m_SaveTrackId = aTrack->GetTrackID();
      //set the initial energy deposit
      m_Hit->set_edep(0);

      //cout << " which active  : "  << whichactive  <<  " layer ID :" << layer_id  << "  hit id :  " << aTrack->GetTrackID()  <<"  vol name :  " <<    volume->GetName()  <<"  z :  " << prePoint->GetPosition().z()/cm <<   "  end z  :  " << postPoint->GetPosition().z()/cm << endl;
      /*
       if(whichactive > 0 )
  {
    m_SaveHitContainer = m_AbsorberHits;
  }
      else
  {
   m_SaveHitContainer = m_HitContainer; 
  }
      */
      if (whichactive < 0)
      {
        m_SaveHitContainer = m_HitContainer;
      }
      else
      {
        //m_SaveHitContainer = m_AbsorberHits;
        m_SaveHitContainer = m_ActiveGasHits;
      }

      if (!geantino && !m_BlackHoleFlag)
      {
        m_Hit->set_eion(0);
      }
      if (G4VUserTrackInformation* p = aTrack->GetUserInformation())
      {
        if (PHG4TrackUserInfoV1* pp = dynamic_cast<PHG4TrackUserInfoV1*>(p))
        {
          m_Hit->set_trkid(pp->GetUserTrackId());
          m_Hit->set_shower_id(pp->GetShower()->get_id());
          m_SaveShower = pp->GetShower();
        }
      }
      /*
   if (!hasMotherSubsystem() && (m_Hit->get_z(0) * cm > m_Zmax || m_Hit->get_z(0) * cm < m_Zmin))
   {
   boost::io::ios_precision_saver ips(std::cout);
        std::cout << m_Detector->SuperDetector() << std::setprecision(9)
  << " PHG4TRDSteppingAction: Entry hit z " << m_Hit->get_z(0) * cm
  << " outside acceptance,  zmin " << m_Zmin
  << ", zmax " << m_Zmax << ", layer: " << layer_id << std::endl;
  }
       */
    }  //  END ..... ||  m_UseG4StepsFlag > 0)
    /*
    // 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
    // some sanity checks for inconsistencies
    // check if this hit was created, if not print out last post step status
    if (!m_Hit || !std::isfinite(m_Hit->get_x(0)))
      {
  std::cout << GetName() << ": hit was not created" << std::endl;
  std::cout << "prestep status: " << PHG4StepStatusDecode::GetStepStatus(prePoint->GetStepStatus())
      << ", poststep status: " << PHG4StepStatusDecode::GetStepStatus(postPoint->GetStepStatus())
      << ", last pre step status: " << PHG4StepStatusDecode::GetStepStatus(m_SavePreStepStatus)
      << ", last post step status: " << PHG4StepStatusDecode::GetStepStatus(m_SavePostStepStatus) << std::endl;
  std::cout << "last track: " << m_SaveTrackId
      << ", current trackid: " << aTrack->GetTrackID() << std::endl;
  std::cout << "phys pre vol: " << volume->GetName()
      << " post vol : " << touchpost->GetVolume()->GetName() << std::endl;
      std::cout << " previous phys pre vol: " << m_SaveVolPre->GetName()
    << " previous phys post vol: " << m_SaveVolPost->GetName() << std::endl;
      exit(1);
      }
    */
    m_SavePostStepStatus = postPoint->GetStepStatus();
    // check if track id matches the initial one when the hit was created
    if (aTrack->GetTrackID() != m_SaveTrackId)
    {
      std::cout << "hits do not belong to the same track" << std::endl;
      std::cout << "saved track: " << m_SaveTrackId
                << ", current trackid: " << aTrack->GetTrackID()
                << std::endl;
      exit(1);
    }
    m_SavePreStepStatus = prePoint->GetStepStatus();
    m_SavePostStepStatus = postPoint->GetStepStatus();
    m_SaveVolPre = volume;
    m_SaveVolPost = touchpost->GetVolume();

    m_Hit->set_x(1, postPoint->GetPosition().x() / cm);
    m_Hit->set_y(1, postPoint->GetPosition().y() / cm);
    m_Hit->set_z(1, postPoint->GetPosition().z() / cm);

    m_Hit->set_px(1, postPoint->GetMomentum().x() / GeV);
    m_Hit->set_py(1, postPoint->GetMomentum().y() / GeV);
    m_Hit->set_pz(1, postPoint->GetMomentum().z() / GeV);

    m_Hit->set_t(1, postPoint->GetGlobalTime() / nanosecond);
    //sum up the energy to get total deposited
    m_Hit->set_edep(m_Hit->get_edep() + edep);
    /*
    if (!hasMotherSubsystem() && (m_Hit->get_z(1) * cm > m_Zmax || m_Hit->get_z(1) * cm < m_Zmin))
    {
      std::cout << m_Detector->SuperDetector() << std::setprecision(9)
           << " PHG4TRDSteppingAction: Exit hit z " << m_Hit->get_z(1) * cm
           << " outside acceptance zmin " << m_Zmin
           << ", zmax " << m_Zmax << ", layer: " << layer_id << std::endl;
    }
    */
    if (geantino)
    {
      m_Hit->set_edep(-1);  // only energy=0 g4hits get dropped, this way geantinos survive the g4hit compression
    }
    else
    {
      if (!m_BlackHoleFlag)
      {
        double eion = edep - aStep->GetNonIonizingEnergyDeposit() / GeV;
        m_Hit->set_eion(m_Hit->get_eion() + eion);
      }
    }

    // if any of these conditions is true this is the last step in
    // this volume and we need to save the hit
    if (postPoint->GetStepStatus() == fGeomBoundary ||
        postPoint->GetStepStatus() == fWorldBoundary ||
        postPoint->GetStepStatus() == fAtRestDoItProc ||
        aTrack->GetTrackStatus() == fStopAndKill ||
        m_UseG4StepsFlag > 0)
    {
      // save only hits with energy deposit (or -1 for geantino) or if save all hits flag is set
      if (m_Hit->get_edep())
      {
        m_SaveHitContainer->AddHit(layer_id, m_Hit);

        if (m_SaveShower)
        {
          m_SaveShower->add_g4hit_id(m_SaveHitContainer->GetID(), m_Hit->get_hit_id());
        }
        // ownership has been transferred to container, set to null
        // so we will create a new hit for the next track
        m_Hit = nullptr;
      }
      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
        m_Hit->Reset();
      }
    }
    // return true to indicate the hit was used
    return true;
  }  // END Acitve flag condition
  else
  {
    return false;
  }
}

//____________________________________________________________________________..
void PHG4TRDSteppingAction::SetInterfacePointers(PHCompositeNode* topNode)
{
  /*
  // Node Name is passed down from PHG4TRDSubsystem
  //now look for the map and grab a pointer to it.
  m_HitContainer = findNode::getClass<PHG4HitContainer>(topNode, m_HitNodeName);

  // if we do not find the node we need to scream.
  if (!m_HitContainer && !m_BlackHoleFlag)
  {
    std::cout << "PHG4TRDSteppingAction::SetTopNode - unable to find " << m_HitNodeName << std::endl;
  }
  */

  string hitnodename;
  string gasnodename;
  if (m_Detector->SuperDetector() != "NONE")
  {
    hitnodename = "G4HIT_" + m_Detector->SuperDetector();
    gasnodename = "G4HIT_ACTIVEGAS_" + m_Detector->SuperDetector();
  }
  else
  {
    hitnodename = "G4HIT_" + m_Detector->GetName();
    gasnodename = "G4HIT_ACTIVEGAS_" + m_Detector->SuperDetector();
  }

  //now look for the map and grab a pointer to it.
  m_HitContainer = findNode::getClass<PHG4HitContainer>(topNode, hitnodename.c_str());
  m_ActiveGasHits = findNode::getClass<PHG4HitContainer>(topNode, gasnodename.c_str());

  // if we do not find the node it's messed up.
  if (!m_HitContainer)
  {
    std::cout << "PHG4TRDSteppingAction::SetTopNode - unable to find " << hitnodename << std::endl;
  }
  // if (!m_AbsorberHits)
  if (!m_ActiveGasHits)
  {
    if (Verbosity() > 1)
    {
      cout << "PHG4TRDSteppingAction::SetTopNode - unable to find " << gasnodename << endl;
    }
  }
}

/*
bool PHG4TRDSteppingAction::hasMotherSubsystem() const
{
  if (m_Subsystem->GetMotherSubsystem())
  {
    return true;
  }
  return false;
}
*/