From e070c5fe084d181654e8090d6e5ee1faef873060 Mon Sep 17 00:00:00 2001 From: Paul Oliver Date: Thu, 29 Feb 2024 01:50:45 +0100 Subject: Updated plugins. Added fast-scroll command. --- plugins/imager.py | 46 +++++++++++++++++++++++----------------------- 1 file changed, 23 insertions(+), 23 deletions(-) (limited to 'plugins/imager.py') 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) -- cgit v1.2.1