Changeset 962 for trunk/source/processes/hadronic/cross_sections/src/G4CrossSectionDataStore.cc

Timestamp:

Apr 6, 2009, 12:30:29 PM (15 years ago)

Author:

garnier

Message:

update processes

File:

• trunk/source/processes/hadronic/cross_sections/src/G4CrossSectionDataStore.cc (modified) (7 diffs)

trunk/source/processes/hadronic/cross_sections/src/G4CrossSectionDataStore.cc

@@ -24 +24 @@
 // ********************************************************************
 //
+// $Id: G4CrossSectionDataStore.cc,v 1.16 2009/01/24 11:54:47 vnivanch Exp $
+// GEANT4 tag $Name: geant4-09-02-ref-02 $
+//
+// -------------------------------------------------------------------
+//
+// GEANT4 Class file
+//
+//
+// File name:     G4CrossSectionDataStore
+//
+//
+// Modifications:
+// 23.01.2009 V.Ivanchenko add destruction of data sets
+//
 //
 
@@ -31 +45 @@
 #include "G4HadTmpUtil.hh"
 #include "Randomize.hh"
-
+#include "G4Nucleus.hh" 
+
+G4CrossSectionDataStore::G4CrossSectionDataStore() :
+  NDataSetList(0), verboseLevel(0)
+{}
+
+G4CrossSectionDataStore::~G4CrossSectionDataStore()
+{}
 
 G4double
@@ -140 +161 @@
 
 
-std::pair<G4double, G4double> 
-G4CrossSectionDataStore::SelectRandomIsotope(const G4DynamicParticle* particle,
-                                             const G4Material* aMaterial)
+G4Element* G4CrossSectionDataStore::SampleZandA(const G4DynamicParticle* particle, 
+                                                const G4Material* aMaterial,
+                                                G4Nucleus& target)
+{
+  G4double aTemp = aMaterial->GetTemperature();
+  const G4int nElements = aMaterial->GetNumberOfElements();
+  const G4ElementVector* theElementVector = aMaterial->GetElementVector();
+  G4Element* anElement = (*theElementVector)[0];
+  G4VCrossSectionDataSet* inCharge;
+  G4int i;
+
+  // compounds
+  if(1 < nElements) {
+    G4double* xsec = new G4double [nElements];
+    const G4double* theAtomsPerVolumeVector = aMaterial->GetVecNbOfAtomsPerVolume();
+    G4double cross = 0.0;
+    for(i=0; i<nElements; i++) {
+      anElement= (*theElementVector)[i];
+      inCharge = whichDataSetInCharge(particle, anElement);
+      cross   += theAtomsPerVolumeVector[i]*
+        inCharge->GetCrossSection(particle, anElement, aTemp);
+      xsec[i]  = cross;
+    }
+    cross *= G4UniformRand();
+
+    for(i=0; i<nElements; i++) {
+      if( cross <=  xsec[i] ) {
+        anElement = (*theElementVector)[i];
+        break;
+      }
+    }
+    delete [] xsec;
+  }
+
+  // element have been selected
+  inCharge = whichDataSetInCharge(particle, anElement);
+  G4double ZZ = anElement->GetZ();
+  G4double AA;
+
+  // Collect abundance weighted cross sections and A values for each isotope 
+  // in each element
+ 
+  const G4int nIsoPerElement = anElement->GetNumberOfIsotopes();
+
+  // user-defined isotope abundances
+  if (0 < nIsoPerElement) { 
+    G4IsotopeVector* isoVector = anElement->GetIsotopeVector();
+    AA = G4double((*isoVector)[0]->GetN());
+    if(1 < nIsoPerElement) {
+
+      G4double* xsec  = new G4double [nIsoPerElement];
+      G4double iso_xs = 0.0;
+      G4double cross  = 0.0;
+
+      G4double* abundVector = anElement->GetRelativeAbundanceVector();
+      G4bool elementXS = false;
+      for (i = 0; i<nIsoPerElement; i++) {
+        if (inCharge->IsZAApplicable(particle, ZZ, G4double((*isoVector)[i]->GetN()))) {
+          iso_xs = inCharge->GetIsoCrossSection(particle, (*isoVector)[i], aTemp);
+        } else if (elementXS == false) {
+          iso_xs = inCharge->GetCrossSection(particle, anElement, aTemp);
+          elementXS = true;
+        }
+
+        cross  += abundVector[i]*iso_xs;
+        xsec[i] = cross;
+      }
+      cross *= G4UniformRand();
+      for (i = 0; i<nIsoPerElement; i++) {
+        if(cross <= xsec[i]) {
+          AA = G4double((*isoVector)[i]->GetN());
+          break;
+        }
+      }
+      delete [] xsec;
+    }
+    // natural abundances
+  } else { 
+
+    G4StableIsotopes theDefaultIsotopes;  
+    G4int Z = G4int(ZZ + 0.5);
+    const G4int nIso = theDefaultIsotopes.GetNumberOfIsotopes(Z);
+    G4int index = theDefaultIsotopes.GetFirstIsotope(Z);
+    AA = G4double(theDefaultIsotopes.GetIsotopeNucleonCount(index));
+    
+    if(1 < nIso) {
+
+      G4double* xsec  = new G4double [nIso];
+      G4double iso_xs = 0.0;
+      G4double cross  = 0.0;
+      G4bool elementXS= false;
+
+      for (i = 0; i<nIso; i++) {
+        AA = G4double(theDefaultIsotopes.GetIsotopeNucleonCount(index+i));
+        if (inCharge->IsZAApplicable(particle, ZZ, AA )) {
+          iso_xs = inCharge->GetIsoZACrossSection(particle, ZZ, AA, aTemp);
+        } else if (elementXS == false) {
+          iso_xs = inCharge->GetCrossSection(particle, anElement, aTemp);
+          elementXS = true;
+        }
+        cross  += theDefaultIsotopes.GetAbundance(index+i)*iso_xs;
+        xsec[i] = cross;
+      }
+      cross *= G4UniformRand();
+      for (i = 0; i<nIso; i++) {
+        if(cross <= xsec[i]) {
+          AA = G4double(theDefaultIsotopes.GetIsotopeNucleonCount(index+i));
+          break;
+        }
+      }
+      delete [] xsec;
+    }
+  }
+  //G4cout << "XS: " << particle->GetDefinition()->GetParticleName()
+  //     << " e(GeV)= " << particle->GetKineticEnergy()/GeV
+  //     << " in " << aMaterial->GetName()
+  //     << " ZZ= " << ZZ << " AA= " << AA << "  " << anElement->GetName() << G4endl;
+
+  target.SetParameters(AA, ZZ);
+  return anElement;
+}
+
+/*
+G4Element* G4CrossSectionDataStore::SampleZandA(const G4DynamicParticle* particle, 
+                                                const G4Material* aMaterial,
+                                                G4Nucleus& target)
 {
   static G4StableIsotopes theDefaultIsotopes;  // natural abundances and 
@@ -229 +373 @@
   // cross section is zero
    
+  anElement = (*theElementVector)[0];
+  G4double ZZ = anElement->GetZ();
+  G4double AA = AvaluesPerElement[0][0];
+  if (crossSectionTotal != 0.) {
+    G4double random = G4UniformRand();
+    for(G4int i=0; i < nElements; i++) {
+      if(i!=0) runningSum[i] += runningSum[i-1];
+      if(random <= runningSum[i]/crossSectionTotal) {
+        anElement = (*theElementVector)[i];
+        ZZ = anElement->GetZ();
+
+        // Compare random number to running sum over isotope xc to choose A
+
+        G4int nIso = awicsPerElement[i].size();
+        G4double* running = new G4double[nIso];
+        for (G4int j=0; j < nIso; j++) {
+          running[j] = awicsPerElement[i][j];
+          if(j!=0) running[j] += running[j-1];
+        }
+
+        G4double trial = G4UniformRand(); 
+        for (G4int j=0; j < nIso; j++) {
+          AA = AvaluesPerElement[i][j];
+          if (trial <= running[j]/running[nIso-1]) break;
+        }
+        delete [] running;
+        break;
+      }
+    }
+  }
+
+  //G4cout << "XS: " << particle->GetDefinition()->GetParticleName()
+  //     << " e(GeV)= " << particle->GetKineticEnergy()/GeV
+  //     << " in " << aMaterial->GetName()
+  //     << " ZZ= " << ZZ << " AA= " << AA << "  " << anElement->GetName() << G4endl;
+
+  target.SetParameters(AA, ZZ);
+  return anElement;
+}
+
+std::pair<G4double, G4double> 
+G4CrossSectionDataStore::SelectRandomIsotope(const G4DynamicParticle* particle,
+                                             const G4Material* aMaterial)
+{
+  static G4StableIsotopes theDefaultIsotopes;  // natural abundances and 
+                                               // stable isotopes
+  G4double aTemp = aMaterial->GetTemperature();
+  G4int nElements = aMaterial->GetNumberOfElements();
+  const G4ElementVector* theElementVector = aMaterial->GetElementVector();
+  const G4double* theAtomsPerVolumeVector = aMaterial->GetVecNbOfAtomsPerVolume();
+  std::vector<std::vector<G4double> > awicsPerElement;
+  std::vector<std::vector<G4double> > AvaluesPerElement;
+  G4Element* anElement;
+
+  // Collect abundance weighted cross sections and A values for each isotope 
+  // in each element
+ 
+  for (G4int i = 0; i < nElements; i++) {
+    anElement = (*theElementVector)[i];
+    G4int nIsoPerElement = anElement->GetNumberOfIsotopes();
+    std::vector<G4double> isoholder;
+    std::vector<G4double> aholder;
+    G4double iso_xs = DBL_MIN;
+
+    if (nIsoPerElement) { // user-defined isotope abundances
+      G4IsotopeVector* isoVector = anElement->GetIsotopeVector();
+      G4double* abundVector = anElement->GetRelativeAbundanceVector();
+      G4VCrossSectionDataSet* inCharge = whichDataSetInCharge(particle, anElement);
+      G4bool elementXS = false;
+      for (G4int j = 0; j < nIsoPerElement; j++) {
+        if (inCharge->IsZAApplicable(particle, (*isoVector)[j]->GetZ(), 
+                                               (*isoVector)[j]->GetN() ) ) {
+          iso_xs = inCharge->GetIsoCrossSection(particle, (*isoVector)[j], aTemp);
+        } else if (elementXS == false) {
+          iso_xs = inCharge->GetCrossSection(particle, anElement, aTemp);
+          elementXS = true;
+        }
+
+        isoholder.push_back(abundVector[j]*iso_xs);
+        aholder.push_back(G4double((*isoVector)[j]->GetN()));
+      }
+
+    } else { // natural abundances
+      G4int ZZ = G4lrint(anElement->GetZ());
+      nIsoPerElement = theDefaultIsotopes.GetNumberOfIsotopes(ZZ);
+      G4int index = theDefaultIsotopes.GetFirstIsotope(ZZ);
+      G4double AA;
+      G4double abundance;
+
+      G4VCrossSectionDataSet* inCharge = whichDataSetInCharge(particle, anElement);
+      G4bool elementXS = false;
+
+      for (G4int j = 0; j < nIsoPerElement; j++) {
+        AA = G4double(theDefaultIsotopes.GetIsotopeNucleonCount(index+j));
+        aholder.push_back(AA);
+        if (inCharge->IsZAApplicable(particle, G4double(ZZ), AA) ) {
+          iso_xs = inCharge->GetIsoZACrossSection(particle, G4double(ZZ), AA, aTemp);
+        } else if (elementXS == false) {
+          iso_xs = inCharge->GetCrossSection(particle, anElement, aTemp);
+          elementXS = true;
+        }
+        abundance = theDefaultIsotopes.GetAbundance(index+j)/100.0;
+        isoholder.push_back(abundance*iso_xs);
+      }
+    }
+
+    awicsPerElement.push_back(isoholder);
+    AvaluesPerElement.push_back(aholder);
+  }
+
+  // Calculate running sums for isotope selection
+
+  G4double crossSectionTotal = 0;
+  G4double xSectionPerElement;
+  std::vector<G4double> runningSum;
+
+  for (G4int i=0; i < nElements; i++) {
+    xSectionPerElement = 0;
+    for (G4int j=0; j < G4int(awicsPerElement[i].size()); j++)
+                     xSectionPerElement += awicsPerElement[i][j];
+    runningSum.push_back(theAtomsPerVolumeVector[i]*xSectionPerElement);
+    crossSectionTotal += runningSum[i];
+  }
+
+  // Compare random number to running sum over element xc to choose Z
+
+  // Initialize Z and A to first element and first isotope in case 
+  // cross section is zero
+   
   G4double ZZ = (*theElementVector)[0]->GetZ();
   G4double AA = AvaluesPerElement[0][0];
@@ -259 +532 @@
   return std::pair<G4double, G4double>(ZZ, AA);
 }
-
+*/
 
 void
 G4CrossSectionDataStore::AddDataSet(G4VCrossSectionDataSet* aDataSet)
 {
-   if (NDataSetList == NDataSetMax) {
-      G4cout << "WARNING: G4CrossSectionDataStore::AddDataSet: "<<G4endl;
-      G4cout << "         reached maximum number of data sets";
-      G4cout << "         data set not added !!!!!!!!!!!!!!!!";
-      return;
-   }
-   DataSetList[NDataSetList] = aDataSet;
-   NDataSetList++;
-}
-
+  DataSetList.push_back(aDataSet);
+  NDataSetList++;
+}
 
 void
@@ -279 +545 @@
 BuildPhysicsTable(const G4ParticleDefinition& aParticleType)
 {
-   if (NDataSetList == 0) 
-   {
-     G4Exception("G4CrossSectionDataStore", "007", FatalException,
-                 "BuildPhysicsTable: no data sets registered");
-     return;
-   }
-   for (G4int i = NDataSetList-1; i >= 0; i--) {
+  if(NDataSetList > 0) {
+    for (G4int i=0; i<NDataSetList; i++) {
       DataSetList[i]->BuildPhysicsTable(aParticleType);
-   }
+    } 
+  }
 }
 
@@ -295 +557 @@
 DumpPhysicsTable(const G4ParticleDefinition& aParticleType)
 {
-   if (NDataSetList == 0) {
-      G4cout << "WARNING - G4CrossSectionDataStore::DumpPhysicsTable: no data sets registered"<<G4endl;
-      return;
-   }
-   for (G4int i = NDataSetList-1; i >= 0; i--) {
-      DataSetList[i]->DumpPhysicsTable(aParticleType);
-   }
-}
+  if (NDataSetList == 0) {
+    G4cout << "WARNING - G4CrossSectionDataStore::DumpPhysicsTable: "
+           << " no data sets registered"<<G4endl;
+    return;
+  }
+  for (G4int i = NDataSetList-1; i >= 0; i--) {
+    DataSetList[i]->DumpPhysicsTable(aParticleType);
+  }
+}