Deprecated/UpgradeASCIITests/G4TankSimulatorASCII/G4TankPMTAction.cc

Go to the documentation of this file.

#include "G4TankPMTAction.h"
#include "G4TankTrackingAction.h"
#include "G4TankTrackInformation.h"
#include "G4TankEventAction.h"
#include "G4TankSimulator.h"
#include "G4TankSteppingAction.h"
#include "ParticleCases.h"

#include <sevt/PMT.h>
#include <sevt/PMTSimData.h>
#include <sevt/SEvent.h>
#include <sevt/Station.h>
#include <sevt/StationSimData.h>

#include <utl/ErrorLogger.h>
#include <utl/Particle.h>
#include <utl/Trace.h>

#include <fwk/RunController.h>

#include <G4Event.hh>
#include <G4HCofThisEvent.hh>
#include <G4SDManager.hh>
#include <G4Step.hh>
#include <G4TouchableHistory.hh>
#include <G4Track.hh>
#include <G4Trajectory.hh>
#include <G4TrajectoryContainer.hh>
#include <G4VHitsCollection.hh>

#include <iostream>

using namespace std;
using namespace G4TankSimulatorASCII;
using namespace sevt;
using namespace utl;
using namespace fwk;


G4TankPMTAction::G4TankPMTAction(const G4String name, const int pmtIndex) :
  G4VSensitiveDetector(name),
  fPMTIndex(pmtIndex)
{
  if (fPMTIndex < 1 || fPMTIndex > 3) {
    FATAL("G4PMTAction() : invalid PMT index.");
    exit(EXIT_FAILURE);
  }

  fG4TankSimulator =
    dynamic_cast<G4TankSimulator*>(&RunController::GetInstance().GetModule("G4TankSimulatorASCII"));
}


G4bool
G4TankPMTAction::ProcessHits(G4Step* theStep, G4TouchableHistory* /*R0Hist*/)
{
  const double time = theStep->GetPreStepPoint()->GetGlobalTime();

  // get current particle from G4TankStepping action.
  // use to reject hits registered in sensitive volume
  // in case the particle is not a photon
  const int currentParticle = G4TankSteppingAction::GetCurrentParticleId();

  // This is related to bug #156
  if (time >= 1*utl::second || currentParticle)
    return true;

  const int particle = G4TankEventAction::GetPrimaryId();

  sevt::Station& station = *fG4TankSimulator->GetCurrentEventStationIt();

  PMTSimData& pmtSim = station.GetPMT(fPMTIndex).GetSimData();

  sevt::StationConstants::SignalComponent component = sevt::StationConstants::eTotal;

  switch (particle) {
  case OFFLINE_PHOTON:
    component = sevt::StationConstants::ePhoton;
    if (fG4TankSimulator->GetCurrentParticleIt()->GetSource() == Particle::eShowerFromMuonDecay) {
      if (!pmtSim.HasPETimeDistribution(sevt::StationConstants::eShowerMuonDecayPhoton))
        pmtSim.MakePETimeDistribution(sevt::StationConstants::eShowerMuonDecayPhoton);
      pmtSim.GetPETimeDistribution(sevt::StationConstants::eShowerMuonDecayPhoton).AddTime(time);
    }
    break;
  case OFFLINE_ELECTRONS:
    component = sevt::StationConstants::eElectron;
    if (fG4TankSimulator->GetCurrentParticleIt()->GetSource() == Particle::eShowerFromMuonDecay) {
      if (!pmtSim.HasPETimeDistribution(sevt::StationConstants::eShowerMuonDecayElectron))
        pmtSim.MakePETimeDistribution(sevt::StationConstants::eShowerMuonDecayElectron);
      pmtSim.GetPETimeDistribution(sevt::StationConstants::eShowerMuonDecayElectron).AddTime(time);
    }
    break;
  case OFFLINE_MUONS:
    component = sevt::StationConstants::eMuon;
    break;
  default:
    component = sevt::StationConstants::eHadron;
    break;
  }

  if (!pmtSim.HasPETimeDistribution())
    pmtSim.MakePETimeDistribution();
  pmtSim.GetPETimeDistribution().AddTime(time);

  if (!pmtSim.HasPETimeDistribution(component))
    pmtSim.MakePETimeDistribution(component);
  pmtSim.GetPETimeDistribution(component).AddTime(time);

  const bool IsFromMuonDecay= theStep->GetTrack()->GetUserInformation();
  if ( IsFromMuonDecay ) 
    {
      G4TankTrackingAction::MichelElectronNPE[1]+=1;
      G4TankTrackingAction::MichelElectronNPE_time[1]+=(time);
    }
  
  return true;
}