diff options
| author |  Edoardo Zoni <59625522+EZoni@users.noreply.github.com>
| 2022-05-16 13:18:21 -0700 |
|---|---|---|
| committer |  GitHub <noreply@github.com>
| 2022-05-16 20:18:21 +0000 |
| commit | 7242ebbaa0df9a547c7bbbdf475437fd5b3ddbc6 (patch) | |
| tree | 222526a6e0d3843cf11c7c0d4851a587cc611238 /Python/pywarpx/WarpX.py | |
| parent | eb549caa53f5fc75290311ca5c40069d77f740fd (diff) | |
| download | WarpX-7242ebbaa0df9a547c7bbbdf475437fd5b3ddbc6.tar.gz WarpX-7242ebbaa0df9a547c7bbbdf475437fd5b3ddbc6.tar.zst WarpX-7242ebbaa0df9a547c7bbbdf475437fd5b3ddbc6.zip | |
Add Headers to Input File Written by PICMI (#3093)
* Add Headers to Input File Written by PICMI
* Commit Suggestion by @dpgrote
* Add Comment
* Add Empty Line
* Slightly Better Readable with a newline
Diffstat (limited to 'Python/pywarpx/WarpX.py')
| -rw-r--r-- | Python/pywarpx/WarpX.py | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/Python/pywarpx/WarpX.py b/Python/pywarpx/WarpX.py index 4701fcfb0..175917ab6 100644 --- a/Python/pywarpx/WarpX.py +++ b/Python/pywarpx/WarpX.py @@ -5,6 +5,8 @@ # # License: BSD-3-Clause-LBNL +import re + from . import Particles from .Algo import algo from .Amr import amr @@ -103,7 +105,17 @@ class WarpX(Bucket): with open(filename, 'w') as ff: + prefix_old = '' for arg in argv: + # This prints the name of the input group (prefix) as a header + # before each group to make the input file more human readable + prefix_new = re.split(' |\.', arg)[0] + if prefix_new != prefix_old: + if prefix_old != '': + ff.write('\n') + ff.write(f'# {prefix_new}\n') + prefix_old = prefix_new + ff.write(f'{arg}\n') warpx = WarpX('warpx') |