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

openmc_data/utils.py

@@ -8,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.6.dist-info → openmc_data-0.2.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: openmc_data
-Version: 0.2.6
+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.6.dist-info → openmc_data-0.2.7.dist-info}/RECORD

@@ -1,19 +1,19 @@
-openmc_data/__init__.py,sha256=RrJuf7nChf2CGexlv60v1tEBQQfloCwQY-0wMk8xmcg,745
-openmc_data/_version.py,sha256=ea_eUoeOtyzu66mqXKQmkHZOdifhlCDVPsMy9xvsneQ,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=nOYLFFNZDKscVw-WLJIAo8XZjTNnSx4LpJ5RDz2YBXs,6535
+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=epk58C0-u7zNdbRJcTeXNs2tKxa1lykVp_L2utkTQWI,1596
 openmc_data/depletion/branching_ratios_pwr.json,sha256=6sLpmQjhkg1-gsgeeH7grSm-nepaYEoZn8rMi857xyM,5498
@@ -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.6.dist-info/LICENSE,sha256=T8b4ZA1twIAqilf_sxR250dAe8AdFhM3k6hqEE0TzBs,1086
-openmc_data-0.2.6.dist-info/METADATA,sha256=6IZl2l0w_uH-YHOu0VWDhD2GIDizAcljkFDMukySEls,9216
-openmc_data-0.2.6.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
-openmc_data-0.2.6.dist-info/entry_points.txt,sha256=zp96W9mML8dY4hMv507oEWYUMygNr-78ugrO046ZLm8,1666
-openmc_data-0.2.6.dist-info/top_level.txt,sha256=ORU-aSVgJUNjwtVsqnZjP4tvU3QUZZpa-XF90ZnE4MQ,12
-openmc_data-0.2.6.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.6.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