HadrontherapyDetectorSD.cc

Go to the documentation of this file.

//
// ********************************************************************
// * License and Disclaimer                                           *
// *                                                                  *
// * The  Geant4 software  is  copyright of the Copyright Holders  of *
// * the Geant4 Collaboration.  It is provided  under  the terms  and *
// * conditions of the Geant4 Software License,  included in the file *
// * LICENSE and available at  http://cern.ch/geant4/license .  These *
// * include a list of copyright holders.                             *
// *                                                                  *
// * Neither the authors of this software system, nor their employing *
// * institutes,nor the agencies providing financial support for this *
// * work  make  any representation or  warranty, express or implied, *
// * regarding  this  software system or assume any liability for its *
// * use.  Please see the license in the file  LICENSE  and URL above *
// * for the full disclaimer and the limitation of liability.         *
// *                                                                  *
// * This  code  implementation is the result of  the  scientific and *
// * technical work of the GEANT4 collaboration.                      *
// * By using,  copying,  modifying or  distributing the software (or *
// * any work based  on the software)  you  agree  to acknowledge its *
// * use  in  resulting  scientific  publications,  and indicate your *
// * acceptance of all terms of the Geant4 Software license.          *
// ********************************************************************
//
// This is the *BASIC* version of Hadrontherapy, a Geant4-based application
// See more at: http://g4advancedexamples.lngs.infn.it/Examples/hadrontherapy
//
// Visit the Hadrontherapy web site (http://www.lns.infn.it/link/Hadrontherapy) to request 
// the *COMPLETE* version of this program, together with its documentation;
// Hadrontherapy (both basic and full version) are supported by the Italian INFN
// Institute in the framework of the MC-INFN Group
//

#include <CLHEP/Units/SystemOfUnits.h>

#include "HadrontherapyDetectorSD.hh"
#include "HadrontherapyAnalysisManager.hh"
#include "HadrontherapyDetectorHit.hh"
#include "G4Step.hh"
#include "G4VTouchable.hh"
#include "G4TouchableHistory.hh"
#include "G4SDManager.hh"
#include "HadrontherapyMatrix.hh"

HadrontherapyDetectorSD::HadrontherapyDetectorSD(G4String name):
    G4VSensitiveDetector(name)
{ 
    G4String HCname;
    collectionName.insert(HCname="HadrontherapyDetectorHitsCollection");
    HitsCollection = NULL; 
    sensitiveDetectorName = name;

}

HadrontherapyDetectorSD::~HadrontherapyDetectorSD()
{ 
}

void HadrontherapyDetectorSD::Initialize(G4HCofThisEvent*)
{
    HitsCollection = new HadrontherapyDetectorHitsCollection(sensitiveDetectorName,
        collectionName[0]);
}

G4bool HadrontherapyDetectorSD::ProcessHits(G4Step* aStep, G4TouchableHistory* ROhist)
{
    //The code doesn't seem to get here if we use the IAEA geometry. FIXME
    if(!ROhist)
    return false;

    if (aStep -> GetPreStepPoint() -> GetPhysicalVolume() -> GetName() != "DetectorPhys")
    return false;
    // Get kinetic energy
    G4Track * theTrack = aStep  ->  GetTrack();
    //G4double kineticEnergy =  theTrack -> GetKineticEnergy();     

    G4ParticleDefinition *particleDef = theTrack -> GetDefinition();
    //Get particle name  
    G4String particleName =  particleDef -> GetParticleName();  
    // G4cout << particleDef -> GetParticleType() << '\n';  
    // Get unique track_id (in an event)
    G4int trackID = theTrack -> GetTrackID();

    G4double energyDeposit = aStep -> GetTotalEnergyDeposit();

    G4int Z = particleDef-> GetAtomicNumber();
    //G4int A = particleDef-> GetAtomicMass();

    // Read voxel indexes: i is the x index, k is the z index
    G4int k  = ROhist -> GetReplicaNumber(0);
    G4int i  = ROhist -> GetReplicaNumber(2);
    G4int j  = ROhist -> GetReplicaNumber(1);

#ifdef G4ANALYSIS_USE_ROOT
    HadrontherapyAnalysisManager* analysis = HadrontherapyAnalysisManager::GetInstance();
#endif

    HadrontherapyMatrix* matrix = HadrontherapyMatrix::GetInstance();

    if (matrix)
    {

    // Increment Fluences & accumulate energy spectra
    // Hit voxels are marked with track_id throught hitTrack matrix
    G4int* hitTrack = matrix -> GetHitTrack(i,j,k); // hitTrack MUST BE cleared at every eventAction!
    if ( *hitTrack != trackID )
    {
        *hitTrack = trackID;
        /*
         * Fill FLUENCE data for every single nuclide 
         * Exclude e-, neutrons, gamma, ...
         */
        if ( Z >= 1)      
            matrix -> Fill(trackID, particleDef, i, j, k, 0, true);
#ifdef G4ANALYSIS_USE_ROOT
/*
        // Fragments kinetic energy (ntuple)
        if (trackID !=1 && Z>=1) 
        {
        // First step kinetic energy for every fragment 
         analysis -> FillKineticFragmentTuple(i, j, k, A, Z, kineticEnergy/MeV);
        }    
        // Kinetic energy spectra for primary particles 

        if ( trackID == 1 && i == 0) 
        {
        // First step kinetic energy for primaries only
        analysis -> FillKineticEnergyPrimaryNTuple(i, j, k, kineticEnergy/MeV);
        }
*/
#endif
    }    

    if(energyDeposit != 0)
    {
/*
 *  This method will fill a dose matrix for every single nuclide. 
 *  A method of the HadrontherapyMatrix class (StoreDoseFluenceAscii())
 *  is called automatically at the end of main (or via the macro command /analysis/writeDoseFile.
 *  It permits to store all dose/fluence data into a single plane ASCII file. 
*/      
        // if (A==1 && Z==1) // primary and sec. protons 
        if ( Z>=1 )    //  exclude e-, neutrons, gamma, ...
            matrix -> Fill(trackID, particleDef, i, j, k, energyDeposit);
        /*
         * Create a hit with the information of position is in the detector     
         */
        HadrontherapyDetectorHit* detectorHit = new HadrontherapyDetectorHit();       
        detectorHit -> SetEdepAndPosition(i, j, k, energyDeposit); 
        HitsCollection -> insert(detectorHit);
    }
    }

#ifdef G4ANALYSIS_USE_ROOT
    if(energyDeposit != 0)
    {  
    if(trackID != 1)
    {
        if (particleName == "proton")
        analysis -> SecondaryProtonEnergyDeposit(i, energyDeposit/CLHEP::MeV);

        else if (particleName == "neutron")
        analysis -> SecondaryNeutronEnergyDeposit(i, energyDeposit/CLHEP::MeV);

        else if (particleName == "alpha")
        analysis -> SecondaryAlphaEnergyDeposit(i, energyDeposit/CLHEP::MeV);

        else if (particleName == "gamma")
        analysis -> SecondaryGammaEnergyDeposit(i, energyDeposit/CLHEP::MeV);

        else if (particleName == "e-")
        analysis -> SecondaryElectronEnergyDeposit(i, energyDeposit/CLHEP::MeV);

        else if (particleName == "triton")
        analysis -> SecondaryTritonEnergyDeposit(i, energyDeposit/CLHEP::MeV);

        else if (particleName == "deuteron")
        analysis -> SecondaryDeuteronEnergyDeposit(i, energyDeposit/CLHEP::MeV);

        else if (particleName == "pi+" || particleName == "pi-" ||  particleName == "pi0")
        analysis -> SecondaryPionEnergyDeposit(i, energyDeposit/CLHEP::MeV);    
    }
    }
#endif

    return true;
}

void HadrontherapyDetectorSD::EndOfEvent(G4HCofThisEvent* HCE)
{
    static G4int HCID = -1;
    if(HCID < 0)
    { 
    HCID = GetCollectionID(0); 
    }

    HCE -> AddHitsCollection(HCID,HitsCollection);
}