def main():
geoMan = Geometry.GeometryManager()
partMan = ParticleManager()
partMan.addGeometry(geoMan)
a = Source.SourceManager()
b = Source.Source()
b.setLocation()
b.setEnergy()
b.setDirection()
b.setParticleType()
b.setPopulation(100)
a.addSource(b)
partMan.addParticles(source=a)
print("length of list of all particles :", len(partMan.allParticles))
print(type(partMan.allParticles[0]))
print("Successfully completed \a")
def main():
# Step 1: Initialize ParticleManager
PM = ParticleManager()
PM.setIterationLimit(20)
# Step 2: Generate Data
# This step will hopefully not be explicityly needed in the future
totalXS, scatterXS = DataGenerators.generateElasticElectronData(10)
ionXS = DataGenerators.generateFakeInelasticElectronXS()
xsec = XSections.XSection(totalXS[1], totalXS[0])
diffXS = Distribution.Distribution()
diffXS.setDomain([-np.pi, np.pi])
diffXS.setPdfData(partial(Distribution.diffElasticElectronXS, 10))
ionXSec = XSections.XSection(ionXS[1], ionXS[0])
ionDist = Distribution.Distribution()
ionDist.setDomain([0., 300. * TC._eV_Erg])
ionDist.setPdfData(Distribution.fakeIonizationElectron)
# Step 3: Add Materials
matMan = Material.MaterialManager()
mat = Material.Material(14)
mat.setZNumber(10)
mat.setAtomicDensity(1.0 * (1 / 18.01) * TC._nAvogad)
mat.setElectronScatterDistribution(diffXS.sample)
mat.setElectronElasticXSHandle(xsec.getXSection)
mat.setElectronEnergyLossHandle(ionDist.sample)
mat.setElectronIonizationXSHandle(ionXSec.getXSection)
mat.setElectronPhysics('inelastic')
matMan.addMaterial(mat)
PM.addMaterials(matMan)
# Step 4: Add Geometry
geoMan = Geometry.GeometryManager()
geo = Geometry.Geometry(dim=3,
coordSys='cartesian',
iLimits=(-1e-5, 1e-5),
jLimits=(-1e-5, 1e-5),
kLimits=(0, 1e-5),
coarseMesh=(1, 1, 1),
fineI=[100],
fineJ=[100],
fineK=[100])
geo.setMaterial(14)
geoMan.addGeometry(geo)
PM.addGeometry(geoMan)
# print(PM.matManager.matDict[14].zNumber)
# Step 5: Add Source Particles
soMan = Source.SourceManager()
so = Source.Source()
so.setLocation()
so.setEnergy(1000)
so.setDirection(np.array([0., 0., 1.]))
so.setParticleType()
so.setPopulation(10000)
soMan.addSource(so)
PM.addParticles(source=soMan)
# Step 6: Create and add Tally
tal = Tally.HeatMap(geo)
# tal = Tally.PathTrack(geo)
PM.addTally(tal)
timerStart = time.process_time()
PM.transportParticles()
timerStop = time.process_time()
# tal.createGraphic()
tal.createGraphic(projection='xy')
tal.createGraphic(projection='yz')
tal.createGraphic(projection='xz')
rms = tal.rmsVariance(10000)
print("Root Mean Square variance: ", rms)
elapsedTime = timerStop - timerStart
print("Elapsed Time: ", elapsedTime)
print("FOM: ", 1 / (rms * elapsedTime))
# tal.printEdgesToFile()
# tal.printHeatMapToFile()
return
