Pol01/src/SteppingAction.cc

0001 //
0002 // ********************************************************************
0003 // * License and Disclaimer                                           *
0004 // *                                                                  *
0005 // * The  Geant4 software  is  copyright of the Copyright Holders  of *
0006 // * the Geant4 Collaboration.  It is provided  under  the terms  and *
0007 // * conditions of the Geant4 Software License,  included in the file *
0008 // * LICENSE and available at  http://cern.ch/geant4/license .  These *
0009 // * include a list of copyright holders.                             *
0010 // *                                                                  *
0011 // * Neither the authors of this software system, nor their employing *
0012 // * institutes,nor the agencies providing financial support for this *
0013 // * work  make  any representation or  warranty, express or implied, *
0014 // * regarding  this  software system or assume any liability for its *
0015 // * use.  Please see the license in the file  LICENSE  and URL above *
0016 // * for the full disclaimer and the limitation of liability.         *
0017 // *                                                                  *
0018 // * This  code  implementation is the result of  the  scientific and *
0019 // * technical work of the GEANT4 collaboration.                      *
0020 // * By using,  copying,  modifying or  distributing the software (or *
0021 // * any work based  on the software)  you  agree  to acknowledge its *
0022 // * use  in  resulting  scientific  publications,  and indicate your *
0023 // * acceptance of all terms of the Geant4 Software license.          *
0024 // ********************************************************************
0025 //
0026 /// \file SteppingAction.cc
0027 /// \brief Implementation of the SteppingAction class
0028
0029 #include "SteppingAction.hh"
0030
0031 #include "DetectorConstruction.hh"
0032 #include "PrimaryGeneratorAction.hh"
0033 #include "RunAction.hh"
0034
0035 #include "G4PolarizationHelper.hh"
0036 #include "G4RunManager.hh"
0037
0038 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
0039
0040 SteppingAction::SteppingAction(DetectorConstruction* det, PrimaryGeneratorAction* prim,
0041                                RunAction* ruAct)
0042   : G4UserSteppingAction(), fDetector(det), fPrimary(prim), fRunAction(ruAct)
0043 {}
0044
0045 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
0046
0047 SteppingAction::~SteppingAction() {}
0048
0049 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
0050
0051 void SteppingAction::UserSteppingAction(const G4Step* aStep)
0052 {
0053   G4StepPoint* prePoint = aStep->GetPreStepPoint();
0054   G4StepPoint* endPoint = aStep->GetPostStepPoint();
0055
0056   G4String procName = endPoint->GetProcessDefinedStep()->GetProcessName();
0057   fRunAction->CountProcesses(procName);
0058
0059   if (prePoint->GetTouchableHandle()->GetVolume() == fDetector->GetBox()
0060       && endPoint->GetTouchableHandle()->GetVolume() == fDetector->GetWorld())
0061   {
0062     G4Track* aTrack = aStep->GetTrack();
0063     const G4ParticleDefinition* part = aTrack->GetDynamicParticle()->GetDefinition();
0064     //    G4cout<<"a "<<particleName<<" left the Box \n";
0065     G4ThreeVector position = endPoint->GetPosition();
0066     G4ThreeVector direction = endPoint->GetMomentumDirection();
0067     G4double kinEnergy = endPoint->GetKineticEnergy();
0068
0069     G4ThreeVector beamDirection = fPrimary->GetParticleGun()->GetParticleMomentumDirection();
0070     G4double polZ = endPoint->GetPolarization().z();
0071
0072     G4double costheta = direction * beamDirection;
0073
0074     G4double xdir = direction * G4PolarizationHelper::GetParticleFrameX(beamDirection);
0075     G4double ydir = direction * G4PolarizationHelper::GetParticleFrameY(beamDirection);
0076
0077     G4double phi = std::atan2(ydir, xdir);
0078     fRunAction->FillData(part, kinEnergy, costheta, phi, polZ);
0079   }
0080 }
0081
0082 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......