diff options
Diffstat (limited to 'plugins/imager.py')
| -rwxr-xr-x | plugins/imager.py | 46 |
1 files changed, 23 insertions, 23 deletions
diff --git a/plugins/imager.py b/plugins/imager.py index fc7d77b..20d57d7 100755 --- a/plugins/imager.py +++ b/plugins/imager.py @@ -14,19 +14,20 @@ Calling this will create a PNG image for each file matching the given pattern. By default, this script ignores any existing simulation files whose name matches an existing image on the same directory. You may force it to overwrite -existing images by passing the '-o' flag before the file list parameter. +existing images by passing the '-o' or '--overwrite' flag. All other parameters +are considered part of the file list. ''' import os import sys from ctypes import * from PIL import Image -if sys.argv[1] == '-o': +if '-o' in sys.argv or '--overwrite' in sys.argv: overwrite = True - files = sys.argv[2:] + sys.argv.remove('-o') + sys.argv.remove('--overwrite') else: overwrite = False - files = sys.argv[1:] salis = CDLL('libsalis.so') salis.sm_getSize.restype = c_uint @@ -37,22 +38,21 @@ def makeImage(iname): if not overwrite: if os.path.isfile('./' + oname): print(oname + ' skipped') - return - - salis.s_load(bytes(iname, 'utf-8')) - lsize = int(salis.sm_getSize() ** 0.5) - image = Image.new('L', (lsize, lsize), 'black') - pixels = image.load() - - for y in range(lsize): - for x in range(lsize): - addr = (y * lsize) + x - byte = salis.sm_getByteAt(addr) - pixels[x, y] = byte - - salis.s_quit() - image.save(oname) - print(oname + ' generated') - -for f in files: - makeImage(f) + else: + salis.s_load(bytes(iname, 'utf-8')) + lsize = int(salis.sm_getSize() ** 0.5) + image = Image.new('L', (lsize, lsize), 'black') + pixels = image.load() + + for y in range(lsize): + for x in range(lsize): + addr = (y * lsize) + x + byte = salis.sm_getByteAt(addr) + pixels[x, y] = byte + + salis.s_quit() + image.save(oname) + print(oname + ' generated') + +for fileName in sys.argv[1:]: + makeImage(fileName) |