1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
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
|
