openmc-data 0.2.5__py3-none-any.whl → 0.2.7__py3-none-any.whl

openmc_data/utils.py

@@ -1,5 +1,4 @@
 import hashlib
-import imp
 import tarfile
 from typing import Iterable
 import warnings
@@ -9,22 +8,44 @@ import shutil
 from urllib.parse import urlparse
 from urllib.request import Request, urlopen
 import warnings
+from .urls import all_release_details
 
 import openmc.data
 
 _BLOCK_SIZE = 16384
 
-
-def state_download_size(download_size, uncompressed_size, units):
+def state_download_size(compressed_file_size, uncompressed_file_size, units):
     """Prints a standard message to users displaying the amount of storage
     space required to run the script"""
 
-    msg = (f"WARNING: This script will download up to {download_size} {units} "
+    msg = (f"WARNING: This script will download up to {compressed_file_size} {units} "
            "of data. Extracting and processing the data may require as much "
-           f"as {uncompressed_size} {units} of additional free disk space.")
+           f"as {uncompressed_file_size} {units} of additional free disk space.")
     warnings.warn(msg)
 
 
+def get_file_types(particles, script_type='convert'):
+    if script_type == 'convert':
+
+        ft = {}
+        for particle in particles:
+            ft[particle] = {'photon':'endf', 'neutron':'ace'}[particle]
+    return ft
+
+def calculate_download_size(library_name, release, particles, file_type,units='GB'):
+    """Prints a standard message to users displaying the amount of storage
+    space required to run the script"""
+
+    release_details = all_release_details[library_name][release]
+
+    compressed_file_size = 0
+    uncompressed_file_size = 0
+    for p in particles:
+        compressed_file_size += release_details[p][file_type[p]]["compressed_file_size"]
+        uncompressed_file_size += release_details[p][file_type[p]]["uncompressed_file_size"]
+    state_download_size(compressed_file_size, uncompressed_file_size, units)
+ 
+
 def process_neutron(path, output_dir, libver, temperatures=None):
     """Process ENDF neutron sublibrary file into HDF5 and write into a
     specified output directory."""

{openmc_data-0.2.5.dist-info → openmc_data-0.2.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: openmc_data
-Version: 0.2.5
+Version: 0.2.7
 Summary: A Python package containing a collection of scripts for producing and downloading data for OpenMC
 Author-email: Jonathan Shimwell <mail@jshimwell.com>
 License: Copyright (c) 2019-2022 UChicago Argonne LLC and contributors

{openmc_data-0.2.5.dist-info → openmc_data-0.2.7.dist-info}/RECORD

@@ -1,21 +1,21 @@
-openmc_data/__init__.py,sha256=RrJuf7nChf2CGexlv60v1tEBQQfloCwQY-0wMk8xmcg,745
-openmc_data/_version.py,sha256=8n5F0z2KUdXF-kMA1VxiH6w0elyZUUWXosPRzNz_3I0,411
-openmc_data/urls.py,sha256=3hQVPM8g6LUMDmu5eahr3Cp3jtbVtfEsbxiJXH-DysM,12408
+openmc_data/__init__.py,sha256=fOMSqMJm50askApEA9aDzgevM1QC4S-wyAh5xuLdltY,786
+openmc_data/_version.py,sha256=mXPA9v4zB_3yyyQ0dNvl4mba1fMh4hU3qcyCEtD2HlE,411
+openmc_data/urls.py,sha256=L_hwVgNmYllsSNhafvWYm5eon5yRI6ooN90jDTkAJQ4,18860
 openmc_data/urls_chain.py,sha256=4t_uFjcUD_iH58l9U3gBtcm2We1s2HvTt67xd3ZRJHE,2688
 openmc_data/urls_h5.py,sha256=jg6FezS4p-ltNVMzTLw6QVcOdyCzOaiCap7PqbfYFMY,1292
 openmc_data/urls_xml.py,sha256=g08eB9uvAwz_87f1bEBkUbEESjBNcU9U59NGg7zAHso,1421
-openmc_data/utils.py,sha256=F3HNIeQUeBONNJibkTMwBGVEBVcKqIRDWUDz6JXulK0,6546
+openmc_data/utils.py,sha256=7Vx3fbEw6Oakoe1TmlZVEyDjSxYdU4z1tuY8l3EIdCc,7435
 openmc_data/convert/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-openmc_data/convert/convert_endf.py,sha256=mjNsuKi7hmCObsMf5ksDXrrUm_KKOwtewzUJQTQsVBo,7017
-openmc_data/convert/convert_fendl.py,sha256=cQWk8i1CrsUX_eWKitcrB7O2jaIk9enqLiMGIQTWVH8,10836
-openmc_data/convert/convert_jeff32.py,sha256=YamQV_pxXEriu61GADCeTqUP7f6j8IqedPPWEzgp0E4,8568
-openmc_data/convert/convert_jeff33.py,sha256=7WZzEWmU6dnSiUhWcJ3rcoaH1BSq-LcjBSkfU55AnN4,6372
+openmc_data/convert/convert_endf.py,sha256=6pP4wgaPTrCIK4qxOIp2CacFl-tMhQ0XvjDcWoCcI18,6610
+openmc_data/convert/convert_fendl.py,sha256=VZzdJeq9xx5w8IfsZVRvUZ7JqrlaV-s_1LjBIRdyuuA,10507
+openmc_data/convert/convert_jeff32.py,sha256=iFWkOQ_Nsnw-aF3qJT9WuBfjqNZuMEACLhg-hhTKU6Y,8644
+openmc_data/convert/convert_jeff33.py,sha256=CbSCBlgAzk1iALOJIYTmHRNK-z2ZbGSj7d9SNF_aFoo,6448
 openmc_data/convert/convert_lib80x.py,sha256=ZVeMj2AW6HeP3TRUo791-gyfv0MA7JhIJn0Q9TKuPW0,3425
 openmc_data/convert/convert_mcnp70.py,sha256=Y1SKKjxFJIm9A-3VIID_DmjjyCCuPMCyPhkbrsEMVL8,4891
 openmc_data/convert/convert_mcnp71.py,sha256=a2jygs1Y6wXyHcS6HYU9hAAkMhtPOiyUZ-WVp-957V8,4646
-openmc_data/convert/convert_tendl.py,sha256=U8vyQTj023xAoFRKtGHmOlTFuZ2ZVZ3HgZ5oubf92Jo,5589
+openmc_data/convert/convert_tendl.py,sha256=60cicKeYMaxdFqjYp56DshezR6RvLourDwnU3KX4Rc8,5587
 openmc_data/depletion/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-openmc_data/depletion/add_branching_ratios.py,sha256=ORLBltKWhq6BqsZtw2URmoa-FWJAf0YNEwR1aBPbJMA,1498
+openmc_data/depletion/add_branching_ratios.py,sha256=epk58C0-u7zNdbRJcTeXNs2tKxa1lykVp_L2utkTQWI,1596
 openmc_data/depletion/branching_ratios_pwr.json,sha256=6sLpmQjhkg1-gsgeeH7grSm-nepaYEoZn8rMi857xyM,5498
 openmc_data/depletion/branching_ratios_sfr.json,sha256=fRsPf2zYYspGmlKtWnM7ApRkXnkSZLkiTOoLIJBdCnU,6227
 openmc_data/depletion/casl_chain.py,sha256=GgadOJHGviz8r-suqrF4M5TIDgsy1qFa267kYh_SoPw,11173
@@ -30,12 +30,12 @@ openmc_data/depletion/chain_endf_b8.0_pwr.xml,sha256=iVnUWt79n4J5BKU_ETn-Y5F5dnL
 openmc_data/depletion/chain_endf_b8.0_sfr.xml,sha256=qmgyj2QJX1F6_OG5X2BZKr6nyiEMEGj6SleZnnO6cSs,27672964
 openmc_data/depletion/chain_endfb71_pwr.xml,sha256=jAzAlbvI44zzv8xFZgHL1fTf-ZE12U2gDXpPCRBZgaI,1862978
 openmc_data/depletion/chain_endfb71_sfr.xml,sha256=s0OJwslf-xfNiROgPgY96wg7ddvRvBjKBcISNGjN_yw,1863707
-openmc_data/depletion/generate_endf71_chain_casl.py,sha256=Ny0MrXjO9NejP7hJD8pgaikadWuCF29VxNx3bPOyIyM,11156
-openmc_data/depletion/generate_endf_chain.py,sha256=V5tYorpoOKBpaM154SBfGOQ5aUpj4tFPOY709GyYcSY,2869
-openmc_data/depletion/generate_jeff_chain.py,sha256=Ag6tfWxIaGU9SZ7VVV6GVaA86DWF5WoFn1Uy8PpbE70,2597
-openmc_data/depletion/generate_serpent_fissq.py,sha256=jvgk1qID7bBZhPPoAtJL5TR4eSgkU6IkfFUrwHYcsfE,1346
-openmc_data/depletion/generate_tendl_chain.py,sha256=XNpV8phemLvlnlEdQ6yFDUfAbw--vO0jQOFFOBAB1K4,4995
-openmc_data/depletion/reduce_chain.py,sha256=2QCw4dEs4nhZbB9W4o-Yyub_PsPDZsswLtxv1WmsfnU,520
+openmc_data/depletion/generate_endf71_chain_casl.py,sha256=OpQi48fMYpAS7i7M5GfetsSatmPJk4cZfHkQokb2dig,11234
+openmc_data/depletion/generate_endf_chain.py,sha256=cIVuoCWYOap1_I7rKgEkNRSMTJzrazcUG5KlkZuZlNc,2964
+openmc_data/depletion/generate_jeff_chain.py,sha256=utAvQpGlIUkiMJdmN6CSjo7tEau38VK-HG7kQK7bSaE,2692
+openmc_data/depletion/generate_serpent_fissq.py,sha256=IizN7MwhEB3_V-9oOIOa5c5RYUqYfEeR3TexSKeIuSc,1426
+openmc_data/depletion/generate_tendl_chain.py,sha256=owQJLiPPUfgFiSQtuQRsCBlALNOcu-0HT5-eambY7ig,5092
+openmc_data/depletion/reduce_chain.py,sha256=PovyMtc_W_9QhZMU19qlc3w_VJNWgwu0sjWK8do6bHo,1311
 openmc_data/depletion/serpent_fissq.json,sha256=qmXD5cdJJcr4_kNGsSR1jZMGeZRQ1iiBgH1CiH9vMQQ,2383
 openmc_data/depletion/tendl2019_nuclides.json,sha256=aiQK-tQOgpnISlXJbSLsIw_5zO-n1Wdirs8QPunnerA,6266
 openmc_data/depletion/tendl2021_nuclides.json,sha256=aiQK-tQOgpnISlXJbSLsIw_5zO-n1Wdirs8QPunnerA,6266
@@ -45,18 +45,19 @@ openmc_data/download/download_endf_chain.py,sha256=O9uK7sC1yhnLzloGTkxIiGPmVt6tW
 openmc_data/download/download_tendl.py,sha256=Ku1cHcmWXX93LT9s-Grcn-SoFHbqxw7T0zNDUD9Kjgs,2839
 openmc_data/generate/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 openmc_data/generate/generate_cendl.py,sha256=naeFyC99lLXutl6y9n4qebgMc-k2gMt9XC9kb_jL1HQ,6223
-openmc_data/generate/generate_endf.py,sha256=4tm_C6jhJO_pcfQfSswVoFj1xEot8yBqAm4yon7N07U,18185
+openmc_data/generate/generate_endf.py,sha256=PUMKt_3bhQSXCVKcY1v7Y3ECUHIOTNPX2QwOSHcHOfc,17971
+openmc_data/generate/generate_fendl.py,sha256=FuhCKwX25kzFCI6J4aPD9VwuEPAF4DrnB4nRbA3tHLA,4676
 openmc_data/generate/generate_jeff33.py,sha256=nnLHR85jtrmks0O1bZxhOYNbfNRfB94ZLxZ6-pl3gM4,8698
-openmc_data/generate/generate_jendl.py,sha256=zEpxZj7-_jzqhgqkmxbbrMrKHcLXbJ_Cxmsc6BK6Q30,4581
+openmc_data/generate/generate_jendl.py,sha256=3j6ntogp_m10T1Ssz-2aCjdxKRrSVBjrzO4NqI-Qvk8,4589
 openmc_data/other/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 openmc_data/other/combine_libraries.py,sha256=HQvR9ppPL_kLoWC478tFLG_hFfug0EmkvW3bN03FYBk,4250
 openmc_data/other/convert_tendl_rand.py,sha256=F3fq4vQ0kin_FjwAhZm8HP8GzQsk-1gYfWxgKDYN9S0,10619
 openmc_data/other/make_compton.py,sha256=xsY4fLHWceiRA8tUBVavlZ1TbIDk5HKgikWytEqq6vE,2466
 openmc_data/other/make_stopping_powers.py,sha256=4Qy9L6VAHm5sMLrDadWX-xS8TSeT-QTiNHDXmZIcT7I,1956
 openmc_data/other/sample_sandy.py,sha256=uwmZk_XZ59VJOLYqp8cOCruW45qTKzlbeuBwXB6Fk1c,6184
-openmc_data-0.2.5.dist-info/LICENSE,sha256=T8b4ZA1twIAqilf_sxR250dAe8AdFhM3k6hqEE0TzBs,1086
-openmc_data-0.2.5.dist-info/METADATA,sha256=loCjxGQuM4C4U9Au7drQpaJu3zo4k6yYn-zSP_i4iwY,9216
-openmc_data-0.2.5.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
-openmc_data-0.2.5.dist-info/entry_points.txt,sha256=zp96W9mML8dY4hMv507oEWYUMygNr-78ugrO046ZLm8,1666
-openmc_data-0.2.5.dist-info/top_level.txt,sha256=ORU-aSVgJUNjwtVsqnZjP4tvU3QUZZpa-XF90ZnE4MQ,12
-openmc_data-0.2.5.dist-info/RECORD,,
+openmc_data-0.2.7.dist-info/LICENSE,sha256=T8b4ZA1twIAqilf_sxR250dAe8AdFhM3k6hqEE0TzBs,1086
+openmc_data-0.2.7.dist-info/METADATA,sha256=d0Y4vSLM-1-cedvmziY711yErGeWNStVYODOz92uaSY,9216
+openmc_data-0.2.7.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
+openmc_data-0.2.7.dist-info/entry_points.txt,sha256=XHn4QdJU5BA7P4NZ-6SYZuXMq5vKXpRhjGZF1reO2hY,1724
+openmc_data-0.2.7.dist-info/top_level.txt,sha256=ORU-aSVgJUNjwtVsqnZjP4tvU3QUZZpa-XF90ZnE4MQ,12
+openmc_data-0.2.7.dist-info/RECORD,,

{openmc_data-0.2.5.dist-info → openmc_data-0.2.7.dist-info}/entry_points.txt

@@ -17,6 +17,7 @@ generate_cendl = openmc_data.generate.generate_cendl:main
 generate_endf = openmc_data.generate.generate_endf:main
 generate_endf71_chain_casl = openmc_data.depletion.generate_endf71_chain_casl:main
 generate_endf_chain = openmc_data.depletion.generate_endf_chain:main
+generate_fendl = openmc_data.generate.generate_fendl:main
 generate_jeff33 = openmc_data.generate.generate_jeff33:main
 generate_jeff_chain = openmc_data.depletion.generate_jeff_chain:main
 generate_jendl = openmc_data.generate.generate_jendl:main