diff options
Diffstat (limited to 'PyGan/data/rDragView_proc/get_lcm.py')
| -rw-r--r-- | PyGan/data/rDragView_proc/get_lcm.py | 105 |
1 files changed, 105 insertions, 0 deletions
diff --git a/PyGan/data/rDragView_proc/get_lcm.py b/PyGan/data/rDragView_proc/get_lcm.py new file mode 100644 index 0000000..d83512e --- /dev/null +++ b/PyGan/data/rDragView_proc/get_lcm.py @@ -0,0 +1,105 @@ +#!/usr/bin/env python + +# Purpose: +# Read or create lcm objects. +# +# Copyright: +# Copyright (C) 2025 Polytechnique Montreal +# This library is free software; you can redistribute it and/or modify it under the terms +# of the GNU Lesser General Public License as published by the Free Software Foundation; +# either version 2.1 of the License, or (at your option) any later version +# +# Author(s): Atyab A. Calloo + +from pathlib import Path + +import os +import lifo +import lcm +import cle2000 + + +def read_lcm_obj(verbose): + """ + Reads Geom, Macr, Trck, Flux LCM objects from current working directory. + Expects file name to begin with an underscore and be capitalised. + + Args: + verbose : controls write output level + Returns: + Geom : geometry LCM object + Macr : macrolib LCM object + Trck : tracking LCM object + Flux : flux LCM object + """ + ### FETCH VERBOSE LEVEL + lcm_verbose = verbose-2 if verbose > 2 else 0 + + ### PICK UP LCM OBJECTS + Geom=lcm.new(pytype='LCM_INP',name='GEOM',iact=0,impx=lcm_verbose) + Macr=lcm.new(pytype='LCM_INP',name='MACR',iact=0,impx=lcm_verbose) + Trck=lcm.new(pytype='LCM_INP',name='TRCK',iact=0,impx=lcm_verbose) + Flux=lcm.new(pytype='LCM_INP',name='FLUX',iact=0,impx=lcm_verbose) + + if verbose > 0: + print('>>>>>>> SUCCESSFULLY LOADED LCM OBJECTS.') + print('-'*74 + '\n') + + return Geom, Macr, Trck, Flux + + +def run_pydragon(test_name,verbose): + """ + Executes c2m procedure using DRAGON/DONJON through the cle200 class in PyGan + and returns Geom, Macr, Trck, Flux LCM objects. + + Args: + test_name : file name of the c2m procedure to be computed + verbose : controls write output level + Returns: + Geom : geometry LCM object + Macr : macrolib LCM object + Trck : tracking LCM object + Flux : flux LCM object + Raises: + FileNotFound : if provided file name (test_name) does not match any of the + files in current working directory. + """ + + ### CHECK IF FILE EXISTS + extnsn = '.c2m' + template_file = test_name+extnsn + path_file = Path(template_file) + if not path_file.exists(): + for entry in os.scandir('.'): + if entry.is_file(): + print(entry.name) + raise FileNotFoundError('FILE '+template_file+' NOT FOUND IN CURRENT DIRECTORY.') + + ### CONSTRUCT LIFO STACK + ipLifo1=lifo.new() + ipLifo1.pushEmpty("Geom", "LCM") + ipLifo1.pushEmpty("Macr", "LCM") + ipLifo1.pushEmpty("Trck", "LCM") + ipLifo1.pushEmpty("Syst", "LCM") + ipLifo1.pushEmpty("Flux", "LCM") + + ### RUN CLE-2000 PROCEDURE + CLE2KFILE = cle2000.new(test_name,ipLifo1,1) + CLE2KFILE.exec() + + ### RECOVER OUTPUT LCM OBJECTS + Geom = ipLifo1.node("Geom") + Macr = ipLifo1.node("Macr") + Trck = ipLifo1.node("Trck") + Flux = ipLifo1.node("Flux") + + ### EMPTY LIFO STACK + while ipLifo1.getMax() > 0: + ipLifo1.pop() + + if verbose > 0: + print('>>>>>>> SUCCESSFULLY RAN ',template_file,' DRAGON/DONJON CALCULATION.') + print('-'*74 + '\n') + + return Geom, Macr, Trck, Flux
\ No newline at end of file |