def main():
start = time.time()
# set default values
tol = 1.e-8
cell_size = 10.0
solve_method = 'NEM4'
iterations = 100
# create mesh
mesh = Mesh([cell_size,cell_size], [cell_size])
# create fuel
fuel = Material(2, 'fuel')
fuel.setSigmaA([0.005, 0.10])
fuel.setD([1.5, 0.40])
fuel.setNuSigmaF([0.005, 0.15])
fuel.setChi([1.0, 0.0])
fuel.setSigmaS(np.array([[0.0, 0.02],[0.0, 0.0]]))
# create fuel
moderator = Material(2, 'moderator')
moderator.setSigmaA([0.0, 0.01])
moderator.setD([1.5, 0.20])
moderator.setSigmaS(np.array([[0.0, 0.025],[0.0, 0.0]]))
if solve_method == 'NEM4':
order = 4
else:
order = 2
# add materials to cells
mesh.cells[0].setMaterial(fuel, order)
mesh.cells[1].setMaterial(moderator, order)
# mesh = mesh.refineMesh(.1)
mesh.makeSurfaces()
# plot the mesh
pttr.plotMesh(mesh)
# create solver
solver = Solver(mesh, solve_method)
# solve the matrix problem to get flux profile and keff
solver.solve(tol, iterations)
# plot the flux
pttr.plotFlux(solver)
pttr.plotCellFlux(solver)
pttr.plotCurrent(solver)
stop = time.time()
print 'Ran time ' + str(stop-start)[0:5] + ' seconds'
print '----------------------------------------------------------------------'
moderator = Material(2, "moderator")
moderator.setSigmaA([0.0, 0.01])
moderator.setD([1.5, 0.20])
moderator.setSigmaS(np.array([[0.0, 0.025], [0.0, 0.0]]))
if solve_method == "NEM4":
order = 4
else:
order = 2
mesh.cells[0].setMaterial(fuel, order)
mesh.cells[1].setMaterial(moderator, order)
mesh.makeSurfaces()
# plot the mesh
pttr.plotMesh(mesh)
# create solver
solver = Solver(mesh, solve_method)
# solve the matrix problem to get flux profile and keff
if solve_method == "NEM4" or solve_method == "NEM2":
for iteration in range(iterations):
print "CMFD outer iteration " + str(iteration)
solver.computeDs()
solver.makeAM()
solver.solve(tol)
solver.makeN()
