36 lines
1.1 KiB
C++
36 lines
1.1 KiB
C++
|
#include "SteppingAction.hh"
|
|||
|
|
|
|||
|
|
#include "DetectorConstruction.hh"
|
|||
|
|
#include "EventAction.hh"
|
|||
|
|
#include "G4Event.hh"
|
|||
|
|
#include "G4LogicalVolume.hh"
|
|||
|
|
#include "G4RunManager.hh"
|
|||
|
|
#include "G4Step.hh"
|
|||
|
|
|
|||
|
|
namespace B1 {
|
|||
|
|
|
|||
|
|
SteppingAction::SteppingAction(EventAction* eventAction) : fEventAction(eventAction) {} // 初始化 eventAction 类
|
|||
|
|
|
|||
|
|
SteppingAction::~SteppingAction() {}
|
|||
|
|
|
|||
|
|
// 每个 step 调用一次
|
|||
|
|
void SteppingAction::UserSteppingAction(const G4Step* step) {
|
|||
|
|
if (!fScoringVolume) {
|
|||
|
|
const DetectorConstruction* detConstruction =
|
|||
|
|
static_cast<const DetectorConstruction*>(G4RunManager::GetRunManager()->GetUserDetectorConstruction());
|
|||
|
|
fScoringVolume = detConstruction->GetScoringVolume();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 获取当前的 Volume
|
|||
|
|
G4LogicalVolume* volume = step->GetPreStepPoint()->GetTouchableHandle()->GetVolume()->GetLogicalVolume();
|
|||
|
|
|
|||
|
|
// 检查是否在计算的 Volume （Shape 2）
|
|||
|
|
if (volume != fScoringVolume) return;
|
|||
|
|
|
|||
|
|
// 获取当前 step 的沉积能量
|
|||
|
|
G4double edepStep = step->GetTotalEnergyDeposit();
|
|||
|
|
fEventAction->AddEdep(edepStep);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
} // namespace B1
|