Files
geant4/source/processes/hadronic/models/neutron_hp/src/G4NeutronHPLabAngularEnergy.cc
T

352 lines
12 KiB
C++

//
// ********************************************************************
// * DISCLAIMER *
// * *
// * The following disclaimer summarizes all the specific disclaimers *
// * of contributors to this software. The specific disclaimers,which *
// * govern, are listed with their locations in: *
// * http://cern.ch/geant4/license *
// * *
// * Neither the authors of this software system, nor their employing *
// * institutes,nor the agencies providing financial support for this *
// * work make any representation or warranty, express or implied, *
// * regarding this software system or assume any liability for its *
// * use. *
// * *
// * This code implementation is the intellectual property of the *
// * GEANT4 collaboration. *
// * By copying, distributing or modifying the Program (or any work *
// * based on the Program) you indicate your acceptance of this *
// * statement, and all its terms. *
// ********************************************************************
//
// neutron_hp -- source file
// J.P. Wellisch, Nov-1996
// A prototype of the low energy neutron transport model.
//
#include "G4NeutronHPLabAngularEnergy.hh"
#include "G4Gamma.hh"
#include "G4Electron.hh"
#include "G4Positron.hh"
#include "G4Neutron.hh"
#include "G4Proton.hh"
#include "G4Deuteron.hh"
#include "G4Triton.hh"
#include "G4He3.hh"
#include "G4Alpha.hh"
#include "Randomize.hh"
void G4NeutronHPLabAngularEnergy::Init(std::ifstream & aDataFile)
{
aDataFile >> nEnergies;
theManager.Init(aDataFile);
theEnergies = new G4double[nEnergies];
nCosTh = new G4int[nEnergies];
theData = new G4NeutronHPVector * [nEnergies];
theSecondManager = new G4InterpolationManager [nEnergies];
for(G4int i=0; i<nEnergies; i++)
{
aDataFile >> theEnergies[i];
theEnergies[i]*=eV;
aDataFile >> nCosTh[i];
theSecondManager[i].Init(aDataFile);
theData[i] = new G4NeutronHPVector[nCosTh[i]];
G4double label;
for(G4int ii=0; ii<nCosTh[i]; ii++)
{
aDataFile >> label;
theData[i][ii].SetLabel(label);
theData[i][ii].Init(aDataFile, eV);
}
}
}
G4ReactionProduct * G4NeutronHPLabAngularEnergy::Sample(G4double anEnergy, G4double massCode, G4double )
{
G4ReactionProduct * result = new G4ReactionProduct;
G4int Z = static_cast<G4int>(massCode/1000);
G4int A = static_cast<G4int>(massCode-1000*Z);
if(massCode==0)
{
result->SetDefinition(G4Gamma::Gamma());
}
else if(A==0)
{
result->SetDefinition(G4Electron::Electron());
if(Z==1) result->SetDefinition(G4Positron::Positron());
}
else if(A==1)
{
result->SetDefinition(G4Neutron::Neutron());
if(Z==1) result->SetDefinition(G4Proton::Proton());
}
else if(A==2)
{
result->SetDefinition(G4Deuteron::Deuteron());
}
else if(A==3)
{
result->SetDefinition(G4Triton::Triton());
if(Z==2) result->SetDefinition(G4He3::He3());
}
else if(A==4)
{
result->SetDefinition(G4Alpha::Alpha());
if(Z!=2) throw G4HadronicException(__FILE__, __LINE__, "Unknown ion case 1");
}
else
{
throw G4HadronicException(__FILE__, __LINE__, "G4NeutronHPLabAngularEnergy: Unknown ion case 2");
}
// get theta, E
G4double cosTh, secEnergy;
G4int i, it(0);
// find the energy bin
for(i=0; i<nEnergies; i++)
{
it = i;
if(anEnergy<theEnergies[i]) break;
}
if(it==0 || it == nEnergies-1) // it marks the energy bin
{
// integrate the prob for each costh, and select theta.
G4double * running = new G4double [nCosTh[it]];
running[0]=0;
for(i=0;i<nCosTh[it]; i++)
{
if(i!=0) running[i] = running[i-1];
running[i]+=theData[it][i].GetIntegral(); // Does interpolated integral.
}
G4double random = running[nCosTh[it]-1]*G4UniformRand();
G4int ith(0);
for(i=0;i<nCosTh[it]; i++)
{
ith = i;
if(random<running[i]) break;
}
if(ith==0 || ith==nCosTh[it]-1)
{
cosTh = theData[it][ith].GetLabel();
secEnergy = theData[it][ith].Sample();
currentMeanEnergy = theData[it][ith].GetMeanX();
}
else
{
G4double x1 = theData[it][ith-1].GetIntegral();
G4double x2 = theData[it][ith].GetIntegral();
G4double x = random;
G4double y1 = theData[it][ith-1].GetLabel();
G4double y2 = theData[it][ith].GetLabel();
cosTh = theInt.Interpolate(theSecondManager[it].GetInverseScheme(ith),
x, x1, x2, y1, y2);
G4NeutronHPVector theBuff1;
theBuff1.SetInterpolationManager(theData[it][ith-1].GetInterpolationManager());
G4NeutronHPVector theBuff2;
theBuff2.SetInterpolationManager(theData[it][ith].GetInterpolationManager());
x1=y1;
x2=y2;
G4double y, mu;
for(i=0;i<theData[it][ith-1].GetVectorLength(); i++)
{
mu = theData[it][ith-1].GetX(i);
y1 = theData[it][ith-1].GetY(i);
y2 = theData[it][ith].GetY(mu);
y = theInt.Interpolate(theSecondManager[it].GetScheme(ith),
cosTh, x1,x2,y1,y2);
theBuff1.SetData(i, mu, y);
}
for(i=0;i<theData[it][ith].GetVectorLength(); i++)
{
mu = theData[it][ith].GetX(i);
y1 = theData[it][ith-1].GetY(mu);
y2 = theData[it][ith].GetY(i);
y = theInt.Interpolate(theSecondManager[it].GetScheme(ith),
cosTh, x1,x2,y1,y2);
theBuff2.SetData(i, mu, y);
}
G4NeutronHPVector theStore;
theStore.Merge(&theBuff1, &theBuff2);
secEnergy = theStore.Sample();
currentMeanEnergy = theStore.GetMeanX();
}
delete [] running;
}
else // this is the small big else.
{
G4double x, x1, x2, y1, y2, y, tmp, E;
// integrate the prob for each costh, and select theta.
G4NeutronHPVector run1;
run1.SetY(0, 0.);
for(i=0;i<nCosTh[it-1]; i++)
{
if(i!=0) run1.SetY(i, run1.GetY(i-1));
run1.SetX(i, theData[it-1][i].GetLabel());
run1.SetY(i, run1.GetY(i)+theData[it-1][i].GetIntegral());
}
G4NeutronHPVector run2;
run2.SetY(0, 0.);
for(i=0;i<nCosTh[it]; i++)
{
if(i!=0) run2.SetY(i, run2.GetY(i-1));
run2.SetX(i, theData[it][i].GetLabel());
run2.SetY(i, run2.GetY(i)+theData[it][i].GetIntegral());
}
// get the distributions for the correct neutron energy
x = anEnergy;
x1 = theEnergies[it-1];
x2 = theEnergies[it];
G4NeutronHPVector thBuff1; // to be interpolated as run1.
thBuff1.SetInterpolationManager(theSecondManager[it-1]);
for(i=0; i<run1.GetVectorLength(); i++)
{
tmp = run1.GetX(i); //theta
y1 = run1.GetY(i); // integral
y2 = run2.GetY(tmp);
y = theInt.Interpolate(theManager.GetScheme(it), x, x1,x2,y1,y2);
thBuff1.SetData(i, tmp, y);
}
G4NeutronHPVector thBuff2;
thBuff2.SetInterpolationManager(theSecondManager[it]);
for(i=0; i<run2.GetVectorLength(); i++)
{
tmp = run2.GetX(i); //theta
y1 = run1.GetY(tmp); // integral
y2 = run2.GetY(i);
y = theInt.Lin(x, x1,x2,y1,y2);
thBuff2.SetData(i, tmp, y);
}
G4NeutronHPVector theThVec;
theThVec.Merge(&thBuff1 ,&thBuff2); // takes care of interpolation
G4double random = (theThVec.GetY(theThVec.GetVectorLength()-1)
-theThVec.GetY(0)) *G4UniformRand();
G4int ith(0);
for(i=1;i<theThVec.GetVectorLength(); i++)
{
ith = i;
if(random<theThVec.GetY(i)-theThVec.GetY(0)) break;
}
{
// calculate theta
G4double x, x1, x2, y1, y2;
x = random;
x1 = theThVec.GetY(ith-1)-theThVec.GetY(0); // integrals
x2 = theThVec.GetY(ith)-theThVec.GetY(0);
y1 = theThVec.GetX(ith-1); // std::cos(theta)
y2 = theThVec.GetX(ith);
cosTh = theInt.Interpolate(theSecondManager[it].GetScheme(ith),
x, x1,x2,y1,y2);
}
G4int i1(0), i2(0);
// get the indixes of the vectors close to theta for low energy
// first it-1 !!!! i.e. low in energy
for(i=0; i<nCosTh[it-1]; i++)
{
i1 = i;
if(cosTh<theData[it-1][i].GetLabel()) break;
}
// now get the prob at this energy for the right theta value
x = cosTh;
x1 = theData[it-1][i1-1].GetLabel();
x2 = theData[it-1][i1].GetLabel();
G4NeutronHPVector theBuff1a;
theBuff1a.SetInterpolationManager(theData[it-1][i1-1].GetInterpolationManager());
for(i=0;i<theData[it-1][i1-1].GetVectorLength(); i++)
{
E = theData[it-1][i1-1].GetX(i);
y1 = theData[it-1][i1-1].GetY(i);
y2 = theData[it-1][i1].GetY(E);
y = theInt.Lin(x, x1,x2,y1,y2);
theBuff1a.SetData(i, E, y); // wrong E, right theta.
}
G4NeutronHPVector theBuff2a;
theBuff2a.SetInterpolationManager(theData[it-1][i1].GetInterpolationManager());
for(i=0;i<theData[it-1][i1].GetVectorLength(); i++)
{
E = theData[it-1][i1].GetX(i);
y1 = theData[it-1][i1-1].GetY(E);
y2 = theData[it-1][i1].GetY(i);
y = theInt.Lin(x, x1,x2,y1,y2);
theBuff2a.SetData(i, E, y); // wrong E, right theta.
}
G4NeutronHPVector theStore1;
theStore1.Merge(&theBuff1a, &theBuff2a); // wrong E, right theta, complete binning
// get the indixes of the vectors close to theta for high energy
// then it !!!! i.e. high in energy
for(i=0; i<nCosTh[it]; i++)
{
i2 = i;
if(cosTh<theData[it][i2].GetLabel()) break;
} // sonderfaelle mit i1 oder i2 head on fehlen. @@@@@
x1 = theData[it][i2-1].GetLabel();
x2 = theData[it][i2].GetLabel();
G4NeutronHPVector theBuff1b;
theBuff1b.SetInterpolationManager(theData[it][i2-1].GetInterpolationManager());
for(i=0;i<theData[it][i2-1].GetVectorLength(); i++)
{
E = theData[it][i2-1].GetX(i);
y1 = theData[it][i2-1].GetY(i);
y2 = theData[it][i2].GetY(E);
y = theInt.Lin(x, x1,x2,y1,y2);
theBuff1b.SetData(i, E, y); // wrong E, right theta.
}
G4NeutronHPVector theBuff2b;
theBuff2b.SetInterpolationManager(theData[it][i2].GetInterpolationManager());
for(i=0;i<theData[it][i1].GetVectorLength(); i++)
{
E = theData[it][i1].GetX(i);
y1 = theData[it][i1-1].GetY(E);
y2 = theData[it][i1].GetY(i);
y = theInt.Lin(x, x1,x2,y1,y2);
theBuff2b.SetData(i, E, y); // wrong E, right theta.
}
G4NeutronHPVector theStore2;
theStore2.Merge(&theBuff1b, &theBuff2b); // wrong E, right theta, complete binning
// now get to the right energy.
x = anEnergy;
x1 = theEnergies[it-1];
x2 = theEnergies[it];
G4NeutronHPVector theOne1;
theOne1.SetInterpolationManager(theStore1.GetInterpolationManager());
for(i=0; i<theStore1.GetVectorLength(); i++)
{
E = theStore1.GetX(i);
y1 = theStore1.GetY(i);
y2 = theStore2.GetY(E);
y = theInt.Interpolate(theManager.GetScheme(it), x, x1,x2,y1,y2);
theOne1.SetData(i, E, y); // both correct
}
G4NeutronHPVector theOne2;
theOne2.SetInterpolationManager(theStore2.GetInterpolationManager());
for(i=0; i<theStore2.GetVectorLength(); i++)
{
E = theStore2.GetX(i);
y1 = theStore1.GetY(E);
y2 = theStore2.GetY(i);
y = theInt.Interpolate(theManager.GetScheme(it), x, x1,x2,y1,y2);
theOne2.SetData(i, E, y); // both correct
}
G4NeutronHPVector theOne;
theOne.Merge(&theOne1, &theOne2); // both correct, complete binning
secEnergy = theOne.Sample();
currentMeanEnergy = theOne.GetMeanX();
}
// now do random direction in phi, and fill the result.
result->SetKineticEnergy(secEnergy);
G4double phi = twopi*G4UniformRand();
G4double theta = std::acos(cosTh);
G4double sinth = std::sin(theta);
G4double mtot = result->GetTotalMomentum();
G4ThreeVector tempVector(mtot*sinth*std::cos(phi), mtot*sinth*std::sin(phi), mtot*std::cos(theta) );
result->SetMomentum(tempVector);
return result;
}