TRestGeant4Hits.cxx

 
#include "TRestGeant4Hits.h"
 
#include "TRestGeant4Event.h"
 
using namespace std;
 
ClassImp(TRestGeant4Hits);
 
TRestGeant4Hits::TRestGeant4Hits() : TRestHits() {}
 
TRestGeant4Hits::~TRestGeant4Hits() = default;
 
void TRestGeant4Hits::RemoveG4Hits() {
    RemoveHits();
 
    fProcessID.clear();
    fVolumeID.clear();
    fKineticEnergy.clear();
}
 
Double_t TRestGeant4Hits::GetEnergyInVolume(Int_t volumeID) const {
    Double_t energy = 0;
 
    for (size_t n = 0; n < GetNumberOfHits(); n++) {
        if (fVolumeID[n] == volumeID) {
            energy += GetEnergy(n);
        }
    }
 
    return energy;
}
 
TVector3 TRestGeant4Hits::GetMeanPositionInVolume(Int_t volumeID) const {
    TVector3 pos;
    Double_t energy = 0;
    for (size_t n = 0; n < GetNumberOfHits(); n++)
        if (fVolumeID[n] == volumeID) {
            pos += GetPosition(n) * GetEnergy(n);
            energy += GetEnergy(n);
        }
 
    if (energy == 0) {
        Double_t nan = TMath::QuietNaN();
        return {nan, nan, nan};
    }
 
    pos = (1. / energy) * pos;
    return pos;
}
 
TVector3 TRestGeant4Hits::GetFirstPositionInVolume(Int_t volumeID) const {
    for (size_t n = 0; n < GetNumberOfHits(); n++)
        if (fVolumeID[n] == volumeID) return GetPosition(n);
 
    Double_t nan = TMath::QuietNaN();
    return {nan, nan, nan};
}
 
TVector3 TRestGeant4Hits::GetLastPositionInVolume(Int_t volumeID) const {
    for (int n = GetNumberOfHits() - 1; n >= 0; n--) {
        if (fVolumeID[n] == volumeID) {
            return GetPosition(n);
        }
    }
    Double_t nan = TMath::QuietNaN();
    return {nan, nan, nan};
}
 
size_t TRestGeant4Hits::GetNumberOfHitsInVolume(Int_t volumeID) const {
    size_t result = 0;
    for (size_t n = 0; n < GetNumberOfHits(); n++) {
        if (fVolumeID[n] == volumeID) {
            result++;
        }
    }
    return result;
}
 
TRestGeant4Metadata* TRestGeant4Hits::GetGeant4Metadata() const {
    const TRestGeant4Event* event;
    if (fTrack != nullptr) {
        event = fTrack->GetEvent();
    } else {
        event = fEvent;
    }
    if (event == nullptr) {
        return nullptr;
    }
    return const_cast<TRestGeant4Metadata*>(event->GetGeant4Metadata());
}
 
TString TRestGeant4Hits::GetProcessName(size_t n) const {
    const auto metadata = GetGeant4Metadata();
    return metadata == nullptr ? "" : metadata->GetGeant4PhysicsInfo().GetProcessName(GetProcessId(n));
}
 
TString TRestGeant4Hits::GetVolumeName(size_t n) const {
    const auto metadata = GetGeant4Metadata();
    return metadata == nullptr ? "" : metadata->GetGeant4GeometryInfo().GetVolumeFromID(GetVolumeId(n));
}