Added to repository
This commit is contained in:
paraiso
256
geant4/LEMuSR/src/TDCheck.cc
Normal file
256
geant4/LEMuSR/src/TDCheck.cc
Normal file
@ -0,0 +1,256 @@
|
||||
///§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§//*
|
||||
// LOW ENERGY MUON SPIN RELAXATION, ROTATION, RADIATION
|
||||
//
|
||||
// ID :TDCheck.cc , v 1.2
|
||||
// AUTHOR: Taofiq PARAISO
|
||||
// DATE : 2005-03-01 10:07
|
||||
//§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§//
|
||||
//
|
||||
// & &&&&&&&&&& &&&&&&& &&&&&&&&
|
||||
// & & && && & &&
|
||||
// & & & & & & &&
|
||||
// & &&&&&&& & & &&&&&& &&&&&&&&
|
||||
// & & & && & & &&
|
||||
// & & && & & && && & &
|
||||
// &&&&&&&&&& &&&&&&&&&& & &&&&& && &&&&&&& & &&
|
||||
// &
|
||||
// &
|
||||
// &
|
||||
// &
|
||||
// TDCHECK
|
||||
//§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§§//
|
||||
#include "TDCheck.hh"
|
||||
|
||||
#include "G4SteppingManager.hh"
|
||||
#include "G4Transform3D.hh"
|
||||
#include "G4DynamicParticle.hh"
|
||||
#include "G4UnitsTable.hh"
|
||||
|
||||
|
||||
#include "LEMuSRDetectorConstruction.hh"
|
||||
#include "LEMuSRPrimaryGeneratorAction.hh"
|
||||
|
||||
|
||||
TDCheck::TDCheck()
|
||||
{
|
||||
|
||||
BookRoot();
|
||||
muon.thickness=0;
|
||||
pointer=this ;
|
||||
loop=0;
|
||||
oldz=0;
|
||||
thk_old=0;
|
||||
id=0;
|
||||
old_id=0;
|
||||
loop=0;
|
||||
}
|
||||
|
||||
|
||||
TDCheck::~TDCheck()
|
||||
{
|
||||
WriteRoot();
|
||||
}
|
||||
|
||||
TDCheck* TDCheck::pointer=0;
|
||||
TDCheck* TDCheck::GetInstance()
|
||||
{
|
||||
return pointer;
|
||||
}
|
||||
|
||||
|
||||
void TDCheck::UserSteppingAction(const G4Step* aStep)
|
||||
|
||||
{
|
||||
|
||||
if( aStep->GetPreStepPoint()&& aStep->GetPreStepPoint()->GetPhysicalVolume() )
|
||||
{
|
||||
SetParticleVolumeNames(aStep);
|
||||
|
||||
if(CheckCondition(aStep))
|
||||
{
|
||||
// Get datas
|
||||
SetPositionMomentum(aStep);
|
||||
SetTimeEnergy(aStep);
|
||||
Update();
|
||||
FillRoot();
|
||||
#if defined G4UI_USE_ROOT
|
||||
myFile->Write();
|
||||
#endif
|
||||
}
|
||||
|
||||
|
||||
LoopKiller(aStep);
|
||||
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
|
||||
G4bool TDCheck::CheckCondition(const G4Step* )
|
||||
{
|
||||
G4bool condition=false;
|
||||
|
||||
if(v_name=="lv_CFOIL" )
|
||||
{
|
||||
if(p_name=="Mu" || p_name == "mu+")
|
||||
{
|
||||
condition=true;
|
||||
}
|
||||
}
|
||||
|
||||
if(v_name=="lv_SAH2" )
|
||||
{
|
||||
if(p_name=="Mu" || p_name == "mu+")
|
||||
{
|
||||
condition=true;
|
||||
}
|
||||
}
|
||||
|
||||
return condition;
|
||||
|
||||
}
|
||||
|
||||
|
||||
void TDCheck::LoopKiller(const G4Step *aStep)
|
||||
{
|
||||
// loop killa
|
||||
if(aStep->GetStepLength()<0.1*mm)
|
||||
{
|
||||
loop++;
|
||||
if(loop>20)
|
||||
{
|
||||
aStep->GetTrack()->SetTrackStatus(fStopAndKill);
|
||||
loop=0;
|
||||
}
|
||||
}
|
||||
|
||||
// kill useless particles
|
||||
if(aStep->GetTrack()->GetDefinition()->GetParticleName()=="e-"
|
||||
||aStep->GetTrack()->GetDefinition()->GetParticleName()=="gamma")
|
||||
{
|
||||
// if(aStep->GetTrack()->GetKineticEnergy()<10.*eV)
|
||||
// {
|
||||
aStep->GetTrack()->SetTrackStatus(fStopAndKill);
|
||||
// }
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
void TDCheck::SetParticleVolumeNames(const G4Step* aStep)
|
||||
{
|
||||
// NAMES
|
||||
p_name = aStep->GetTrack()->GetDefinition()->GetParticleName(); // particle name
|
||||
v_name = aStep->GetPreStepPoint()->GetPhysicalVolume()->GetLogicalVolume()->GetName();
|
||||
pv_name = aStep->GetPreStepPoint()->GetPhysicalVolume()->GetName();
|
||||
}
|
||||
|
||||
|
||||
void TDCheck::SetPositionMomentum(const G4Step* aStep)
|
||||
{
|
||||
// POSITION, MOMENTUM
|
||||
position = aStep->GetPreStepPoint()->GetPosition(); // position
|
||||
momentum = aStep->GetPostStepPoint()->GetMomentumDirection(); // momentum
|
||||
momentum_direction = aStep->GetPreStepPoint()->GetMomentumDirection(); // momentum
|
||||
|
||||
}
|
||||
|
||||
|
||||
void TDCheck::SetTimeEnergy(const G4Step* aStep)
|
||||
{
|
||||
|
||||
// ENERGY
|
||||
kenergy= aStep->GetTrack()->GetDynamicParticle()->GetKineticEnergy(); // position
|
||||
|
||||
tenergy= aStep->GetTrack()->GetDynamicParticle()->GetTotalEnergy(); // position
|
||||
|
||||
// TIME
|
||||
localtime = (aStep->GetPreStepPoint()->GetLocalTime()+ aStep->GetPreStepPoint()->GetLocalTime())/2.; // time since track creation
|
||||
globaltime = aStep->GetPreStepPoint()->GetGlobalTime();// time since event creation
|
||||
proptime = aStep->GetPreStepPoint()->GetProperTime(); // proper time of the particle
|
||||
time = proptime;
|
||||
|
||||
}
|
||||
|
||||
void TDCheck::Update()
|
||||
{
|
||||
muon.localtime = localtime/ns ;
|
||||
muon.globaltime= globaltime/ns ;
|
||||
muon.proptime = proptime/ns ;
|
||||
|
||||
muon.positionx = LEMuSRPrimaryGeneratorAction::GetPGA()->X/cm; // position.x()/cm;
|
||||
muon.positiony = LEMuSRPrimaryGeneratorAction::GetPGA()->Y/cm; //position.y()/cm;
|
||||
muon.positionz = position.z()/cm;
|
||||
|
||||
muon.momdirx= LEMuSRPrimaryGeneratorAction::GetPGA()->angle/rad;//momentum_direction.x();
|
||||
muon.momdiry= LEMuSRPrimaryGeneratorAction::GetPGA()->radius/cm; //momentum_direction.y();
|
||||
muon.momdirz= momentum_direction.z();
|
||||
|
||||
muon.tenergy = kenergy/keV;
|
||||
muon.thickness = LEMuSRDetectorConstruction::GetInstance()->cfthk;
|
||||
|
||||
muon.angle=0;
|
||||
|
||||
|
||||
muon.index=0;
|
||||
muon.ienergy= LEMuSRPrimaryGeneratorAction::GetPGA()->energy/keV;
|
||||
|
||||
muon.angle=acos(momentum.z())*180/M_PI;
|
||||
|
||||
/////////////////
|
||||
/*
|
||||
G4cout << "particle name : " << p_name <<" ;\n "
|
||||
<< "position : " << position <<" ;\n "
|
||||
<< "momentum direction: " <<momentum_direction <<" ;\n "
|
||||
<< "time : " << globaltime/ns <<" ;\n "
|
||||
<<std::endl;
|
||||
*/
|
||||
//////////////////
|
||||
|
||||
if( G4RunManager::GetRunManager()->GetCurrentRun()->GetRunID()!=old_id)
|
||||
{
|
||||
id=id+1;
|
||||
old_id= G4RunManager::GetRunManager()->GetCurrentRun()->GetRunID();
|
||||
}
|
||||
|
||||
if( LEMuSRDetectorConstruction::GetInstance()->cfthk!=thk_old)
|
||||
{
|
||||
id=0;
|
||||
thk_old=LEMuSRDetectorConstruction::GetInstance()->cfthk;
|
||||
}
|
||||
muon.id = G4RunManager::GetRunManager()->GetCurrentRun()->GetRunID();
|
||||
|
||||
if(p_name=="Mu") muon.charge=0;
|
||||
else if(p_name=="mu+") muon.charge=1;
|
||||
|
||||
|
||||
}
|
||||
|
||||
|
||||
//--------------------------------------------------------------------------------//
|
||||
// ROOT
|
||||
|
||||
void TDCheck::BookRoot()
|
||||
{
|
||||
myFile = new TFile("gfoil.root", "RECREATE");
|
||||
|
||||
tree = new TTree ("tree","Muons parameters");
|
||||
|
||||
tree->Branch("muon",&muon.ienergy,"Init_Energy/F:kenergy/F:localtime/F:globaltime:propertime/F:positionx/F:positiony:positionz:init_angle:init_radius:momdirz:Angle/F:Plane/I:foil_thickness/F:runID/I:Charge/I");
|
||||
|
||||
}
|
||||
|
||||
|
||||
void TDCheck::FillRoot()
|
||||
{
|
||||
tree->Fill();
|
||||
}
|
||||
|
||||
void TDCheck::WriteRoot()
|
||||
{
|
||||
myFile->Write();
|
||||
myFile->Close();
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user