def test_mesh_to_geom(): if not HAVE_PYTAPS: raise SkipTest expected_geom = os.path.join(thisdir, "files_test_alara", "alara_geom.txt") expected_matlib = os.path.join(thisdir, "files_test_alara", "alara_matlib.txt") geom = os.path.join(os.getcwd(), "alara_geom") matlib = os.path.join(os.getcwd(), "alara_matlib") mats = { 0: Material({ 'H1': 1.0, 'K39': 1.0 }, density=1.1), 1: Material({ 'H1': 0.1, 'O16': 1.0 }, density=1.2), 2: Material({'He4': 42.0}, density=1.3), 3: Material({'Tm171': 171.0}, density=1.4), } m = Mesh(structured_coords=[[-1, 0, 1], [-1, 0, 1], [0, 1]], structured=True, mats=mats) mesh_to_geom(m, geom, matlib) with open(expected_geom) as f: written = f.readlines() with open(geom) as f: expected = f.readlines() assert_equal(written, expected) if os.path.isfile(geom): os.remove(geom) with open(expected_matlib) as f: written = f.readlines() with open(matlib) as f: expected = f.readlines() assert_equal(written, expected) if os.path.isfile(matlib): os.remove(matlib)
def test_mesh_to_geom(): if not HAVE_PYTAPS: raise SkipTest expected_geom = os.path.join(thisdir, "files_test_alara", "alara_geom.txt") expected_matlib = os.path.join(thisdir, "files_test_alara", "alara_matlib.txt") geom = os.path.join(os.getcwd(), "alara_geom") matlib = os.path.join(os.getcwd(), "alara_matlib") mats = { 0: Material({'H1': 1.0, 'K39': 1.0}, density=1.1), 1: Material({'H1': 0.1, 'O16': 1.0}, density=1.2), 2: Material({'He4': 42.0}, density=1.3), 3: Material({'Tm171': 171.0}, density=1.4), } m = Mesh(structured_coords=[[-1,0,1],[-1,0,1],[0,1]], structured=True, mats=mats) mesh_to_geom(m, geom, matlib) with open(expected_geom) as f: written = f.readlines() with open(geom) as f: expected = f.readlines() assert_equal(written, expected) if os.path.isfile(geom): os.remove(geom) with open(expected_matlib) as f: written = f.readlines() with open(matlib) as f: expected = f.readlines() assert_equal(written, expected) if os.path.isfile(matlib): os.remove(matlib)
def irradiation_setup(flux_mesh, cell_mats, alara_params, tally_num=4, geom=None, num_rays=10, grid=False, flux_tag="n_flux", fluxin="alara_fluxin", reverse=False, alara_inp="alara_geom", alara_matlib="alara_matlib", output_mesh="r2s_step1.h5m"): """This function is used to setup the irradiation inputs after the first R2S transport step. Parameters ---------- flux_mesh : PyNE Meshtal object, Mesh object, or str The source of the neutron flux information. This can be a PyNE Meshtal object, a pyne Mesh object, or the filename an MCNP meshtal file, or the filename of an unstructured mesh tagged with fluxes. tally_num : int The MCNP FMESH4 tally number of the neutron flux tally within the meshtal file. cell_mats : dict Maps geometry cell numbers to PyNE Material objects. alara_params : str The ALARA input blocks specifying everything except the geometry and materials. This can either be passed as string or as a file name. geom : str, optional The file name of a DAGMC-loadable faceted geometry. This is only necessary if the geometry is not already loaded into memory. num_rays : int, optional The number of rays to fire down a mesh row for geometry discretization. This number must be a perfect square if grid=True. grid : bool, optional The if False, geometry discretization will be done with randomly fired rays. If true, a grid of sqrt(num_rays) x sqrt(num_rays) rays is used for each mesh row. flux_tag : str, optional The iMesh tag for the neutron flux. fluxin : str, optional The name of the ALARA fluxin file to be created. reverse : bool, optional If True the fluxes in the fluxin file will be printed in the reverse order of how they appear within the flux vector tag. Since MCNP and the Meshtal class order fluxes from low energy to high energy, this option should only be true if the transmutation data being used is ordered from high energy to low energy. alara_inp : str, optional The name of the ALARA input file to be created. alara_matlib : str, optional The name of the alara_matlib file to be created. output_mesh : str, optional A mesh containing all the fluxes and materials used for irradiation setup. """ if geom is not None and isfile(geom): load(geom) # flux_mesh is Mesh object if isinstance(flux_mesh, Mesh): m = flux_mesh # flux_mesh is unstructured mesh file elif isinstance(flux_mesh, str) and isfile(flux_mesh) \ and flux_mesh.endswith(".h5m"): m = Mesh(structured=False, mesh=flux_mesh) # flux_mesh is Meshtal or meshtal file else: # flux_mesh is meshtal file if isinstance(flux_mesh, str) and isfile(flux_mesh): flux_mesh = Meshtal(flux_mesh, { tally_num: (flux_tag, flux_tag + "_err", flux_tag + "_total", flux_tag + "_err_total") }, meshes_have_mats=True) m = flux_mesh.tally[tally_num] # flux_mesh is Meshtal object elif instance(flux_mesh, Meshtal): m = flux_mesh.tally[tally_num] else: raise ValueError("meshtal argument not a Mesh object, Meshtal" " object, MCNP meshtal file or meshtal.h5m file.") if m.structured: vol_fracs = discretize_geom(m, num_rays=num_rays, grid=grid) else: vol_fracs = discretize_geom(m) m.cell_fracs_to_mats(vol_fracs, cell_mats) mesh_to_fluxin(m, flux_tag, fluxin, reverse) mesh_to_geom(m, alara_inp, alara_matlib) if isfile(alara_params): with open(alara_params, 'r') as f: alara_params = f.read(alara_params) with open(alara_inp, 'a') as f: f.write("\n" + alara_params) m.write_hdf5(output_mesh)
def irradiation_setup(flux_mesh, cell_mats, alara_params, tally_num=4, geom=None, num_rays=10, grid=False, flux_tag="n_flux", fluxin="alara_fluxin", reverse=False, alara_inp="alara_geom", alara_matlib="alara_matlib", output_mesh="r2s_step1.h5m"): """This function is used to setup the irradiation inputs after the first R2S transport step. Parameters ---------- flux_mesh : PyNE Meshtal object, Mesh object, or str The source of the neutron flux information. This can be a PyNE Meshtal object, a pyne Mesh object, or the filename an MCNP meshtal file, or the filename of an unstructured mesh tagged with fluxes. tally_num : int The MCNP FMESH4 tally number of the neutron flux tally within the meshtal file. cell_mats : dict Maps geometry cell numbers to PyNE Material objects. alara_params : str The ALARA input blocks specifying everything except the geometry and materials. This can either be passed as string or as a file name. geom : str, optional The file name of a DAGMC-loadable faceted geometry. This is only necessary if the geometry is not already loaded into memory. num_rays : int, optional The number of rays to fire down a mesh row for geometry discretization. This number must be a perfect square if grid=True. grid : bool, optional The if False, geometry discretization will be done with randomly fired rays. If true, a grid of sqrt(num_rays) x sqrt(num_rays) rays is used for each mesh row. flux_tag : str, optional The iMesh tag for the neutron flux. fluxin : str, optional The name of the ALARA fluxin file to be created. reverse : bool, optional If True the fluxes in the fluxin file will be printed in the reverse order of how they appear within the flux vector tag. Since MCNP and the Meshtal class order fluxes from low energy to high energy, this option should only be true if the transmutation data being used is ordered from high energy to low energy. alara_inp : str, optional The name of the ALARA input file to be created. alara_matlib : str, optional The name of the alara_matlib file to be created. output_mesh : str, optional A mesh containing all the fluxes and materials used for irradiation setup. """ if geom is not None and isfile(geom): load(geom) # flux_mesh is Mesh object if isinstance(flux_mesh, Mesh): m = flux_mesh # flux_mesh is unstructured mesh file elif isinstance(flux_mesh, str) and isfile(flux_mesh) \ and flux_mesh.endswith(".h5m"): m = Mesh(structured=False, mesh=flux_mesh) # flux_mesh is Meshtal or meshtal file else: # flux_mesh is meshtal file if isinstance(flux_mesh, str) and isfile(flux_mesh): flux_mesh = Meshtal(flux_mesh, {tally_num: (flux_tag, flux_tag + "_err", flux_tag + "_total", flux_tag + "_err_total")}, meshes_have_mats=True) m = flux_mesh.tally[tally_num] # flux_mesh is Meshtal object elif instance(flux_mesh, Meshtal): m = flux_mesh.tally[tally_num] else: raise ValueError("meshtal argument not a Mesh object, Meshtal" " object, MCNP meshtal file or meshtal.h5m file.") if m.structured: vol_fracs = discretize_geom(m, num_rays=num_rays, grid=grid) else: vol_fracs = discretize_geom(m) m.cell_fracs_to_mats(vol_fracs, cell_mats) mesh_to_fluxin(m, flux_tag, fluxin, reverse) mesh_to_geom(m, alara_inp, alara_matlib) if isfile(alara_params): with open(alara_params, 'r') as f: alara_params = f.read(alara_params) with open(alara_inp, 'a') as f: f.write("\n" + alara_params) m.write_hdf5(output_mesh)