Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history 
for more information, see https://pre-commit.ci
  • Loading branch information
@pre-commit-ci
pre-commit-ci[bot] committed Nov 5, 2023
1 parent c6af986 commit 825bf99
Show file tree
Hide file tree
Showing 2 changed files with 99 additions and 98 deletions.
47 changes: 25 additions & 22 deletions inc/TRestGeant4ParticleSourceCry.h
View file
Original file line number Diff line number Diff line change
Expand Up @@ -11,28 +11,31 @@
#include <TRestGeant4ParticleSource.h>

class TRestGeant4ParticleSourceCry : public TRestGeant4ParticleSource {
private:
Int_t fReturnNeutrons = 1;
Int_t fReturnProtons = 1;
Int_t fReturnGammas = 1;
Int_t fReturnElectrons = 1;
Int_t fReturnPions = 1;
Int_t fReturnKaons = 1;
Int_t fReturnMuons = 1;

Int_t fNParticlesMin = 1;
Int_t fNParticlesMax = 1000000;

Double_t fXOffset = 0;
Double_t fYOffset = 0;
Double_t fZOffset = 0;

std::string fDate = "7-1-2012";

Double_t fLatitude = 90.0;;
Double_t fAltitude = 0.0;;

Double_t fSubBoxLength = 100.0;;
private:
Int_t fReturnNeutrons = 1;
Int_t fReturnProtons = 1;
Int_t fReturnGammas = 1;
Int_t fReturnElectrons = 1;
Int_t fReturnPions = 1;
Int_t fReturnKaons = 1;
Int_t fReturnMuons = 1;

Int_t fNParticlesMin = 1;
Int_t fNParticlesMax = 1000000;

Double_t fXOffset = 0;
Double_t fYOffset = 0;
Double_t fZOffset = 0;

std::string fDate = "7-1-2012";

Double_t fLatitude = 90.0;
;
Double_t fAltitude = 0.0;
;

Double_t fSubBoxLength = 100.0;
;

protected:
#ifdef USE_CRY
Expand Down
150 changes: 74 additions & 76 deletions src/TRestGeant4ParticleSourceCry.cxx
View file
Original file line number Diff line number Diff line change
Expand Up @@ -7,80 +7,77 @@ ClassImp(TRestGeant4ParticleSourceCry);
TRestGeant4ParticleSourceCry::TRestGeant4ParticleSourceCry() {}

void TRestGeant4ParticleSourceCry::PrintMetadata() {

TRestGeant4ParticleSource::PrintMetadata();

RESTMetadata << "Return Neutrons : " << fReturnNeutrons << RESTendl;
RESTMetadata << "Return Protons : " << fReturnProtons << RESTendl;
RESTMetadata << "Return Gammas : " << fReturnGammas << RESTendl;
RESTMetadata << "Return Electrons : " << fReturnElectrons << RESTendl;
RESTMetadata << "Return Pions : " << fReturnPions << RESTendl;
RESTMetadata << "Return Kaons : " << fReturnKaons << RESTendl;
RESTMetadata << "Return Muons : " << fReturnMuons << RESTendl;
RESTMetadata << " ======= " << RESTendl;

RESTMetadata << "N particles min : " << fNParticlesMin << RESTendl;
RESTMetadata << "N particles max : " << fNParticlesMax << RESTendl;
RESTMetadata << " ======= " << RESTendl;

RESTMetadata << "X-offset : " << fXOffset << "m" << RESTendl;
RESTMetadata << "Y-offset : " << fYOffset << "m" << RESTendl;
RESTMetadata << "Z-offset : " << fZOffset << "m" << RESTendl;
RESTMetadata << "SubBoxLength : " << fSubBoxLength << "m" << RESTendl;
RESTMetadata << " ======= " << RESTendl;

RESTMetadata << "Date : " << fDate << RESTendl;
RESTMetadata << "Latitude : " << fLatitude << RESTendl;
RESTMetadata << "Altitude : " << fAltitude << RESTendl;
RESTMetadata << "----------------------" << RESTendl;
TRestGeant4ParticleSource::PrintMetadata();

RESTMetadata << "Return Neutrons : " << fReturnNeutrons << RESTendl;
RESTMetadata << "Return Protons : " << fReturnProtons << RESTendl;
RESTMetadata << "Return Gammas : " << fReturnGammas << RESTendl;
RESTMetadata << "Return Electrons : " << fReturnElectrons << RESTendl;
RESTMetadata << "Return Pions : " << fReturnPions << RESTendl;
RESTMetadata << "Return Kaons : " << fReturnKaons << RESTendl;
RESTMetadata << "Return Muons : " << fReturnMuons << RESTendl;
RESTMetadata << " ======= " << RESTendl;

RESTMetadata << "N particles min : " << fNParticlesMin << RESTendl;
RESTMetadata << "N particles max : " << fNParticlesMax << RESTendl;
RESTMetadata << " ======= " << RESTendl;

RESTMetadata << "X-offset : " << fXOffset << "m" << RESTendl;
RESTMetadata << "Y-offset : " << fYOffset << "m" << RESTendl;
RESTMetadata << "Z-offset : " << fZOffset << "m" << RESTendl;
RESTMetadata << "SubBoxLength : " << fSubBoxLength << "m" << RESTendl;
RESTMetadata << " ======= " << RESTendl;

RESTMetadata << "Date : " << fDate << RESTendl;
RESTMetadata << "Latitude : " << fLatitude << RESTendl;
RESTMetadata << "Altitude : " << fAltitude << RESTendl;
RESTMetadata << "----------------------" << RESTendl;
}

void TRestGeant4ParticleSourceCry::InitFromConfigFile() {

fReturnNeutrons = StringToInteger( GetParameter( "returnNeutrons", "1" ) );
fReturnProtons = StringToInteger( GetParameter( "returnProtons", "1" ) );
fReturnGammas = StringToInteger( GetParameter( "returnGammas", "1" ) );
fReturnElectrons = StringToInteger( GetParameter( "returnElectrons", "1" ) );
fReturnPions = StringToInteger( GetParameter( "returnPions", "1" ) );
fReturnKaons = StringToInteger( GetParameter( "returnKaons", "1" ) );
fReturnMuons = StringToInteger( GetParameter( "returnMuons", "1" ) );

fNParticlesMin = StringToInteger( GetParameter( "nParticlesMin", "1" ) );
fNParticlesMax = StringToInteger( GetParameter( "nParticlesMax", "1000000" ) );

fXOffset = StringToDouble( GetParameter( "xoffset", "0.0" ) );
fYOffset = StringToDouble( GetParameter( "yoffset", "0.0" ) );
fZOffset = StringToDouble( GetParameter( "zoffset", "0.0" ) );
fSubBoxLength = StringToDouble( GetParameter( "subBoxLength", "100.0" ) );

fDate = GetParameter( "date", "7\\1\\2012" );
fDate = REST_StringHelper::Replace( fDate, "\\", "-" );
fLatitude = StringToDouble( GetParameter( "latitude", "90.0" ) );
fAltitude = StringToDouble( GetParameter( "altitude", "0.0" ) );

PrintMetadata();

std::string setupString = "";
setupString += "returnNeutrons " + IntegerToString(fReturnNeutrons);
setupString += " returnProtons " + IntegerToString(fReturnProtons);
setupString += " returnGammas " + IntegerToString(fReturnGammas);
setupString += " returnElectrons " + IntegerToString(fReturnElectrons);
setupString += " returnPions " + IntegerToString(fReturnPions);
setupString += " returnKaons " + IntegerToString(fReturnKaons);
setupString += " returnMuons " + IntegerToString(fReturnMuons);

setupString += " xoffset " + DoubleToString(fXOffset);
setupString += " yoffset " + DoubleToString(fYOffset);
setupString += " zoffset " + DoubleToString(fZOffset);
setupString += " subboxLength " + DoubleToString(fSubBoxLength);

setupString += " date " + fDate;
setupString += " latitude " + DoubleToString(fLatitude);
setupString += " altitude " + DoubleToString(fAltitude);

setupString += " nParticlesMin " + IntegerToString(fNParticlesMin);
setupString += " nParticlesMax " + IntegerToString(fNParticlesMax);

fReturnNeutrons = StringToInteger(GetParameter("returnNeutrons", "1"));
fReturnProtons = StringToInteger(GetParameter("returnProtons", "1"));
fReturnGammas = StringToInteger(GetParameter("returnGammas", "1"));
fReturnElectrons = StringToInteger(GetParameter("returnElectrons", "1"));
fReturnPions = StringToInteger(GetParameter("returnPions", "1"));
fReturnKaons = StringToInteger(GetParameter("returnKaons", "1"));
fReturnMuons = StringToInteger(GetParameter("returnMuons", "1"));

fNParticlesMin = StringToInteger(GetParameter("nParticlesMin", "1"));
fNParticlesMax = StringToInteger(GetParameter("nParticlesMax", "1000000"));

fXOffset = StringToDouble(GetParameter("xoffset", "0.0"));
fYOffset = StringToDouble(GetParameter("yoffset", "0.0"));
fZOffset = StringToDouble(GetParameter("zoffset", "0.0"));
fSubBoxLength = StringToDouble(GetParameter("subBoxLength", "100.0"));

fDate = GetParameter("date", "7\\1\\2012");
fDate = REST_StringHelper::Replace(fDate, "\\", "-");
fLatitude = StringToDouble(GetParameter("latitude", "90.0"));
fAltitude = StringToDouble(GetParameter("altitude", "0.0"));

PrintMetadata();

std::string setupString = "";
setupString += "returnNeutrons " + IntegerToString(fReturnNeutrons);
setupString += " returnProtons " + IntegerToString(fReturnProtons);
setupString += " returnGammas " + IntegerToString(fReturnGammas);
setupString += " returnElectrons " + IntegerToString(fReturnElectrons);
setupString += " returnPions " + IntegerToString(fReturnPions);
setupString += " returnKaons " + IntegerToString(fReturnKaons);
setupString += " returnMuons " + IntegerToString(fReturnMuons);

setupString += " xoffset " + DoubleToString(fXOffset);
setupString += " yoffset " + DoubleToString(fYOffset);
setupString += " zoffset " + DoubleToString(fZOffset);
setupString += " subboxLength " + DoubleToString(fSubBoxLength);

setupString += " date " + fDate;
setupString += " latitude " + DoubleToString(fLatitude);
setupString += " altitude " + DoubleToString(fAltitude);

setupString += " nParticlesMin " + IntegerToString(fNParticlesMin);
setupString += " nParticlesMax " + IntegerToString(fNParticlesMax);

#ifdef USE_CRY
CRYSetup* setup = new CRYSetup(setupString, CRY_DATA_PATH);
Expand All @@ -103,9 +100,10 @@ void TRestGeant4ParticleSourceCry::Update() {

for (const auto& cryParticle : *ev) {
// std::cout << "id: " << cryParticle->id() << std::endl;
// std::cout << "x: " << cryParticle->x() << " y: " << cryParticle->y() << " z: " << cryParticle->z() << std::endl;
// std::cout << "u: " << cryParticle->u() << " v: " << cryParticle->v() << " w: " << cryParticle->w() << std::endl;
// std::cout << "charge: " << cryParticle->charge() << " energy: " << cryParticle->ke() << std::endl;
// std::cout << "x: " << cryParticle->x() << " y: " << cryParticle->y() << " z: " << cryParticle->z()
// << std::endl; std::cout << "u: " << cryParticle->u() << " v: " << cryParticle->v() << " w: " <<
// cryParticle->w() << std::endl; std::cout << "charge: " << cryParticle->charge() << " energy: " <<
// cryParticle->ke() << std::endl;

TRestGeant4Particle particle;

Expand All @@ -115,7 +113,7 @@ void TRestGeant4ParticleSourceCry::Update() {

/// Particle position
TVector3 position(cryParticle->x(), cryParticle->y(), cryParticle->z());
particle.SetOrigin(1000. * position); // In mm (default REST units)
particle.SetOrigin(1000. * position); // In mm (default REST units)

/// Momentum direction
TVector3 momDirection(cryParticle->u(), cryParticle->v(), cryParticle->w());
Expand Down Expand Up @@ -170,7 +168,7 @@ void TRestGeant4ParticleSourceCry::Update() {

AddParticle(particle);
}
//std::cout << "-----" << std::endl;
// std::cout << "-----" << std::endl;
#else
cout << "TRestGeant4ParticleSourceCry - ERROR: Geant4lib was not linked to CRY libraries" << endl;
cout << " " << endl;
Expand Down

0 comments on commit 825bf99

Please sign in to comment.