SimG4Core/CustomPhysics/src/CustomProcessHelper.cc - master - githubmirror/cmssw

SimG4Core/CustomPhysics/src/CustomProcessHelper.cc
432 строки17 KB
Shahzad Malik Muzaffar
[SIMULATION] [LLVM21] Apply clang-tidy changes
26 ноя 2025, 17:10
26 ноя 2025, 17:109591585
#include "SimG4Core/CustomPhysics/interface/CustomProcessHelper.h"
#include "SimG4Core/CustomPhysics/interface/CustomPDGParser.h"
#include "SimG4Core/CustomPhysics/interface/CustomParticle.h"
#include "SimG4Core/CustomPhysics/interface/CustomParticleFactory.h"

#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/ParameterSet/interface/FileInPath.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"

#include "G4ParticleTable.hh"
#include "Randomize.hh"

#include <iostream>
#include <fstream>
#include <string>

using namespace CLHEP;

CustomProcessHelper::CustomProcessHelper(const edm::ParameterSet& p, CustomParticleFactory* ptr) {
  fParticleFactory = ptr;

  particleTable = G4ParticleTable::GetParticleTable();

  theProton = particleTable->FindParticle("proton");
  theNeutron = particleTable->FindParticle("neutron");

  G4String line;

  edm::FileInPath fp = p.getParameter<edm::FileInPath>("processesDef");
  const std::string& processDefFilePath = fp.fullPath();
  std::ifstream process_stream(processDefFilePath.c_str());

  resonant = p.getParameter<bool>("resonant");
  ek_0 = p.getParameter<double>("resonanceEnergy") * GeV;
  gamma = p.getParameter<double>("gamma") * GeV;
  amplitude = p.getParameter<double>("amplitude") * millibarn;
  suppressionfactor = p.getParameter<double>("reggeSuppression");
  hadronlifetime = p.getParameter<double>("hadronLifeTime");
  reggemodel = p.getParameter<bool>("reggeModel");
  mixing = p.getParameter<double>("mixing");

  edm::LogVerbatim("SimG4CoreCustomPhysics")
      << "ProcessHelper: Read in physics parameters:"
      << "\n Resonant = " << resonant << "\n ResonanceEnergy = " << ek_0 / GeV << " GeV"
      << "\n Gamma = " << gamma / GeV << " GeV"
      << "\n Amplitude = " << amplitude / millibarn << " millibarn"
      << "ReggeSuppression = " << 100 * suppressionfactor << " %"
      << "HadronLifeTime = " << hadronlifetime << " s"
      << "ReggeModel = " << reggemodel << "Mixing = " << mixing * 100 << " %";

  checkfraction = 0;
  n_22 = 0;
  n_23 = 0;

  while (getline(process_stream, line)) {
    std::vector<G4String> tokens;
    //Getting a line
    ReadAndParse(line, tokens, "#");
    //Important info
    G4String incident = tokens[0];

    G4ParticleDefinition* incidentDef = particleTable->FindParticle(incident);
    //particleTable->DumpTable();
    G4int incidentPDG = incidentDef->GetPDGEncoding();
    known_particles[incidentDef] = true;

    G4String target = tokens[1];
    edm::LogVerbatim("SimG4CoreCustomPhysics") << "ProcessHelper: Incident " << incident << "; Target " << target;

    // Making a ReactionProduct
    ReactionProduct prod;
    for (std::size_t i = 2; i != tokens.size(); i++) {
      const G4String& part = tokens[i];
      if (particleTable->contains(part)) {
        prod.push_back(particleTable->FindParticle(part)->GetPDGEncoding());
      } else {
        G4Exception("G4ProcessHelper",
                    "UnkownParticle",
                    FatalException,
                    "Initialization: The reaction product list contained an unknown particle");
      }
    }
    if (target == "proton") {
      pReactionMap[incidentPDG].push_back(prod);
    } else if (target == "neutron") {
      nReactionMap[incidentPDG].push_back(prod);
    } else {
      G4Exception("G4ProcessHelper",
                  "IllegalTarget",
                  FatalException,
                  "Initialization: The reaction product list contained an illegal target particle");
    }
  }

  process_stream.close();

  for (auto part : fParticleFactory->getCustomParticles()) {
    CustomParticle* particle = dynamic_cast<CustomParticle*>(part);
    if (particle) {
      edm::LogVerbatim("SimG4CoreCustomPhysics") << "ProcessHelper: Lifetime of " << part->GetParticleName()
                                                 << " set to " << particle->GetPDGLifeTime() / s << " s;"
                                                 << " isStable: " << particle->GetPDGStable();
    }
  }
}

G4bool CustomProcessHelper::ApplicabilityTester(const G4ParticleDefinition& aPart) {
  const G4ParticleDefinition* aP = &aPart;
  return (known_particles[aP]);
}

G4double CustomProcessHelper::GetInclusiveCrossSection(const G4DynamicParticle* aParticle, const G4Element* anElement) {
  //We really do need a dedicated class to handle the cross sections. They might not always be constant

  //Disassemble the PDG-code

  G4int thePDGCode = aParticle->GetDefinition()->GetPDGEncoding();
  double boost = (aParticle->GetKineticEnergy() + aParticle->GetMass()) / aParticle->GetMass();
  //  G4cout<<"thePDGCode: "<<thePDGCode<<G4endl;
  G4double theXsec = 0;
  G4String name = aParticle->GetDefinition()->GetParticleName();
  if (!reggemodel) {
    //Flat cross section
    if (CustomPDGParser::s_isRGlueball(thePDGCode)) {
      theXsec = 24 * millibarn;
    } else {
      std::vector<G4int> nq = CustomPDGParser::s_containedQuarks(thePDGCode);
      //    edm::LogInfo("SimG4CoreCustomPhysics")<<"Number of quarks: "<<nq.size()<<G4endl;
      for (std::vector<G4int>::iterator it = nq.begin(); it != nq.end(); it++) {
        //	  edm::LogInfo("SimG4CoreCustomPhysics")<<"Quarkvector: "<<*it<<G4endl;
        if (*it == 1 || *it == 2)
          theXsec += 12 * millibarn;
        if (*it == 3)
          theXsec += 6 * millibarn;
      }
    }
  } else {  //reggemodel
    double R = Regge(boost);
    double P = Pom(boost);
    if (thePDGCode > 0) {
      if (CustomPDGParser::s_isMesonino(thePDGCode))
        theXsec = (P + R) * millibarn;
      if (CustomPDGParser::s_isSbaryon(thePDGCode))
        theXsec = 2 * P * millibarn;
      if (CustomPDGParser::s_isRMeson(thePDGCode) || CustomPDGParser::s_isRGlueball(thePDGCode))
        theXsec = (R + 2 * P) * millibarn;
      if (CustomPDGParser::s_isRBaryon(thePDGCode))
        theXsec = 3 * P * millibarn;
    } else {
      if (CustomPDGParser::s_isMesonino(thePDGCode))
        theXsec = P * millibarn;
      if (CustomPDGParser::s_isSbaryon(thePDGCode))
        theXsec = (2 * (P + R) + 30 / sqrt(boost)) * millibarn;
      if (CustomPDGParser::s_isRMeson(thePDGCode) || CustomPDGParser::s_isRGlueball(thePDGCode))
        theXsec = (R + 2 * P) * millibarn;
      if (CustomPDGParser::s_isRBaryon(thePDGCode))
        theXsec = 3 * P * millibarn;
    }
  }
  //Adding resonance
  if (resonant) {
    double e_0 = ek_0 + aParticle->GetDefinition()->GetPDGMass();  //Now total energy

    e_0 = sqrt(aParticle->GetDefinition()->GetPDGMass() * aParticle->GetDefinition()->GetPDGMass() +
               theProton->GetPDGMass() * theProton->GetPDGMass() + 2. * e_0 * theProton->GetPDGMass());
    //      edm::LogInfo("SimG4CoreCustomPhysics")<<e_0/GeV<<G4endl;
    //      edm::LogInfo("SimG4CoreCustomPhysics")<<ek_0/GeV<<" "<<aParticle->GetDefinition()->GetPDGMass()/GeV<<" "<<theProton->GetPDGMass()/GeV<<G4endl;
    double sqrts = sqrt(aParticle->GetDefinition()->GetPDGMass() * aParticle->GetDefinition()->GetPDGMass() +
                        theProton->GetPDGMass() * theProton->GetPDGMass() +
                        2 * aParticle->GetTotalEnergy() * theProton->GetPDGMass());

    double res_result = amplitude * (gamma * gamma / 4.) /
                        ((sqrts - e_0) * (sqrts - e_0) + (gamma * gamma / 4.));  //Non-relativistic Breit Wigner

    theXsec += res_result;
    //      if(fabs(aParticle->GetKineticEnergy()/GeV-200)<10)  std::cout<<sqrts/GeV<<" "<<theXsec/millibarn<<std::endl;
  }

  //  std::cout<<"Xsec/nucleon: "<<theXsec/millibarn<<"millibarn, Total Xsec: "<<theXsec * anElement->GetN()/millibarn<<" millibarn"<<std::endl;
  return theXsec * pow(anElement->GetN(), 0.7) * 1.25;  // * 0.523598775598299;
}

ReactionProduct CustomProcessHelper::GetFinalState(const G4Track& aTrack, G4ParticleDefinition*& aTarget) {
  const G4DynamicParticle* aDynamicParticle = aTrack.GetDynamicParticle();

  //-----------------------------------------------
  // Choose n / p as target
  // and get ReactionProductList pointer
  //-----------------------------------------------

  G4Material* aMaterial = aTrack.GetMaterial();
  const G4ElementVector* theElementVector = aMaterial->GetElementVector();
  const G4double* NbOfAtomsPerVolume = aMaterial->GetVecNbOfAtomsPerVolume();

  G4double NumberOfProtons = 0;
  G4double NumberOfNucleons = 0;

  for (size_t elm = 0; elm < aMaterial->GetNumberOfElements(); elm++) {
    //Summing number of protons per unit volume
    NumberOfProtons += NbOfAtomsPerVolume[elm] * (*theElementVector)[elm]->GetZ();
    //Summing nucleons (not neutrons)
    NumberOfNucleons += NbOfAtomsPerVolume[elm] * (*theElementVector)[elm]->GetN();
  }

  if (G4UniformRand() < NumberOfProtons / NumberOfNucleons) {
    theReactionMap = &pReactionMap;
    theTarget = theProton;
  } else {
    theReactionMap = &nReactionMap;
    theTarget = theNeutron;
  }
  aTarget = theTarget;

  G4int theIncidentPDG = aDynamicParticle->GetDefinition()->GetPDGEncoding();

  if (reggemodel && CustomPDGParser::s_isMesonino(theIncidentPDG) && G4UniformRand() * mixing > 0.5 &&
      aDynamicParticle->GetDefinition()->GetPDGCharge() == 0.) {
    //      G4cout<<"Oscillating..."<<G4endl;
    theIncidentPDG *= -1;
  }

  bool baryonise = false;

  if (reggemodel && G4UniformRand() > 0.9 &&
      ((CustomPDGParser::s_isMesonino(theIncidentPDG) && theIncidentPDG > 0) ||
       CustomPDGParser::s_isRMeson(theIncidentPDG)))
    baryonise = true;

  //Making a pointer directly to the ReactionProductList we are looking at. Makes life easier :-)
  ReactionProductList* aReactionProductList = &((*theReactionMap)[theIncidentPDG]);

  //-----------------------------------------------
  // Count processes
  // kinematic check
  // compute number of 2 -> 2 and 2 -> 3 processes
  //-----------------------------------------------

  G4int N22 = 0;  //Number of 2 -> 2 processes
  G4int N23 = 0;  //Number of 2 -> 3 processes. Couldn't think of more informative names

  //This is the list to be populated
  ReactionProductList theReactionProductList;
  std::vector<bool> theChargeChangeList;

  for (ReactionProductList::iterator prod_it = aReactionProductList->begin(); prod_it != aReactionProductList->end();
       prod_it++) {
    G4int secondaries = prod_it->size();
    // If the reaction is not possible we will not consider it
    if (ReactionIsPossible(*prod_it, aDynamicParticle) &&
        (!reggemodel || (baryonise && ReactionGivesBaryon(*prod_it)) ||
         (!baryonise && !ReactionGivesBaryon(*prod_it)) || (CustomPDGParser::s_isSbaryon(theIncidentPDG)) ||
         (CustomPDGParser::s_isRBaryon(theIncidentPDG)))) {
      // The reaction is possible. Let's store and count it
      theReactionProductList.push_back(*prod_it);
      if (secondaries == 2) {
        N22++;
      } else if (secondaries == 3) {
        N23++;
      } else {
        G4cerr << "ReactionProduct has unsupported number of secondaries: " << secondaries << G4endl;
      }
    } /*else {
      edm::LogInfo("SimG4CoreCustomPhysics")<<"There was an impossible process"<<G4endl;
      }*/
  }
  //  edm::LogInfo("SimG4CoreCustomPhysics")<<"The size of the ReactionProductList is: "<<theReactionProductList.size()<<G4endl;

  if (theReactionProductList.empty())
    G4Exception("G4ProcessHelper",
                "NoProcessPossible",
                FatalException,
                "GetFinalState: No process could be selected from the given list.");

  // For the Regge model no phase space considerations. We pick a process at random
  if (reggemodel) {
    int n_rps = theReactionProductList.size();
    int select = (int)(G4UniformRand() * n_rps);
    //      G4cout<<"Possible: "<<n_rps<<", chosen: "<<select<<G4endl;
    return theReactionProductList[select];
  }

  // Fill a probability map. Remember total probability
  // 2->2 is 0.15*1/n_22 2->3 uses phase space
  G4double p22 = 0.15;
  G4double p23 = 1 - p22;  // :-)

  std::vector<G4double> Probabilities;
  std::vector<G4bool> TwotoThreeFlag;

  G4double CumulatedProbability = 0;

  // To each ReactionProduct we assign a cumulated probability and a flag
  // discerning between 2 -> 2 and 2 -> 3
  for (unsigned int i = 0; i != theReactionProductList.size(); i++) {
    if (theReactionProductList[i].size() == 2) {
      CumulatedProbability += p22 / N22;
      TwotoThreeFlag.push_back(false);
    } else {
      CumulatedProbability += p23 / N23;
      TwotoThreeFlag.push_back(true);
    }
    Probabilities.push_back(CumulatedProbability);
    //    edm::LogInfo("SimG4CoreCustomPhysics")<<"Pushing back cumulated probability: "<<CumulatedProbability<<G4endl;
  }

  //Renormalising probabilities
  //  edm::LogInfo("SimG4CoreCustomPhysics")<<"Probs: ";
  for (std::vector<G4double>::iterator it = Probabilities.begin(); it != Probabilities.end(); it++) {
    *it /= CumulatedProbability;
    //      edm::LogInfo("SimG4CoreCustomPhysics")<<*it<<" ";
  }
  //  edm::LogInfo("SimG4CoreCustomPhysics")<<G4endl;

  // Choosing ReactionProduct

  G4bool selected = false;
  G4int tries = 0;

  //Keep looping over the list until we have a choice, or until we have tried 100 times
  unsigned int i;
  while (!selected && tries < 100) {
    i = 0;
    G4double dice = G4UniformRand();
    while (dice > Probabilities[i] && i < theReactionProductList.size()) {
      //      edm::LogInfo("SimG4CoreCustomPhysics")<<"i: "<<i<<G4endl;
      i++;
    }

    if (!TwotoThreeFlag[i]) {
      // 2 -> 2 processes are chosen immediately
      selected = true;
    } else {
      // 2 -> 3 processes require a phase space lookup
      if (PhaseSpace(theReactionProductList[i], aDynamicParticle) > G4UniformRand())
        selected = true;
    }
    //    double suppressionfactor=0.5;
    if (selected && particleTable->FindParticle(theReactionProductList[i][0])->GetPDGCharge() !=
                        aDynamicParticle->GetDefinition()->GetPDGCharge()) {
      if (G4UniformRand() < suppressionfactor)
        selected = false;
    }
    tries++;
  }

  //Updating checkfraction:
  if (theReactionProductList[i].size() == 2) {
    n_22++;
  } else {
    n_23++;
  }

  checkfraction = (1.0 * n_22) / (n_22 + n_23);
  //Return the chosen ReactionProduct
  return theReactionProductList[i];
}

G4double CustomProcessHelper::ReactionProductMass(const ReactionProduct& aReaction,
                                                  const G4DynamicParticle* aDynamicParticle) {
  // Incident energy:
  G4double E_incident = aDynamicParticle->GetTotalEnergy();
  //edm::LogInfo("SimG4CoreCustomPhysics")<<"Total energy: "<<E_incident<<" Kinetic: "<<aDynamicParticle->GetKineticEnergy()<<G4endl;
  // sqrt(s)= sqrt(m_1^2 + m_2^2 + 2 E_1 m_2)
  G4double m_1 = aDynamicParticle->GetDefinition()->GetPDGMass();
  G4double m_2 = theTarget->GetPDGMass();
  //edm::LogInfo("SimG4CoreCustomPhysics")<<"M_R: "<<m_1/GeV<<" GeV, M_np: "<<m_2/GeV<<" GeV"<<G4endl;
  G4double sqrts = sqrt(m_1 * m_1 + m_2 * (m_2 + 2 * E_incident));
  //edm::LogInfo("SimG4CoreCustomPhysics")<<"sqrt(s) = "<<sqrts/GeV<<" GeV"<<G4endl;
  // Sum of rest masses after reaction:
  G4double M_after = 0;
  for (ReactionProduct::const_iterator r_it = aReaction.begin(); r_it != aReaction.end(); r_it++) {
    //edm::LogInfo("SimG4CoreCustomPhysics")<<"Mass contrib: "<<(particleTable->FindParticle(*r_it)->GetPDGMass())/MeV<<" MeV"<<G4endl;
    M_after += particleTable->FindParticle(*r_it)->GetPDGMass();
  }
  //edm::LogInfo("SimG4CoreCustomPhysics")<<"Intending to return this ReactionProductMass: "<<(sqrts - M_after)/MeV<<" MeV"<<G4endl;
  return sqrts - M_after;
}

G4bool CustomProcessHelper::ReactionIsPossible(const ReactionProduct& aReaction,
                                               const G4DynamicParticle* aDynamicParticle) {
  return (ReactionProductMass(aReaction, aDynamicParticle) > 0);
}

G4bool CustomProcessHelper::ReactionGivesBaryon(const ReactionProduct& aReaction) {
  for (ReactionProduct::const_iterator it = aReaction.begin(); it != aReaction.end(); it++)
    if (CustomPDGParser::s_isSbaryon(*it) || CustomPDGParser::s_isRBaryon(*it))
      return true;
  return false;
}

G4double CustomProcessHelper::PhaseSpace(const ReactionProduct& aReaction, const G4DynamicParticle* aDynamicParticle) {
  G4double qValue = ReactionProductMass(aReaction, aDynamicParticle);
  G4double phi = sqrt(1 + qValue / (2 * 0.139 * GeV)) * pow(qValue / (1.1 * GeV), 3. / 2.);
  return (phi / (1 + phi));
}

void CustomProcessHelper::ReadAndParse(const G4String& str, std::vector<G4String>& tokens, const G4String& delimiters) {
  // Skip delimiters at beginning.
  G4String::size_type lastPos = str.find_first_not_of(delimiters, 0);
  // Find first "non-delimiter".
  G4String::size_type pos = str.find_first_of(delimiters, lastPos);

  while (G4String::npos != pos || G4String::npos != lastPos) {
    //Skipping leading / trailing whitespaces
    G4String temp = str.substr(lastPos, pos - lastPos);
    while (temp.c_str()[0] == ' ')
      temp.erase(0, 1);
    while (temp[temp.size() - 1] == ' ')
      temp.erase(temp.size() - 1, 1);
    // Found a token, add it to the vector.
    tokens.push_back(temp);
    // Skip delimiters.  Note the "not_of"
    lastPos = str.find_first_not_of(delimiters, pos);
    // Find next "non-delimiter"
    pos = str.find_first_of(delimiters, lastPos);
  }
}

double CustomProcessHelper::Regge(const double boost) {
  double a = 2.165635078566177;
  double b = 0.1467453738547229;
  double c = -0.9607903711871166;
  return 1.5 * exp(a + b / boost + c * log(boost));
}

double CustomProcessHelper::Pom(const double boost) {
  double a = 4.138224000651535;
  double b = 1.50377557581421;
  double c = -0.05449742257808247;
  double d = 0.0008221235048211401;
  return a + b * sqrt(boost) + c * boost + d * pow(boost, 1.5);
}