mirror of
https://github.com/ParadiseSS13/Paradise.git
synced 2026-07-12 23:53:47 +01:00
Mikhail Dzianishchyts
93319097a1
* Mass screenshot verb * Well, this is actually useless * Apply suggestions from code review Co-authored-by: Luc <89928798+lewcc@users.noreply.github.com> Signed-off-by: Mikhail Dzianishchyts <mikhail.dzianishchyts@gmail.com> --------- Signed-off-by: Mikhail Dzianishchyts <mikhail.dzianishchyts@gmail.com> Co-authored-by: Luc <89928798+lewcc@users.noreply.github.com>
86 lines
3.0 KiB
Python
86 lines
3.0 KiB
Python
#This script processes screenshots from the Mass-Screenshot Debug verb in SS13 into a full map image
|
|
|
|
# Loosely based on https://github.com/vortex1942/telescience/blob/master/src/tools/PhotoProcessor.py
|
|
|
|
# This script uses the Pillow library (PIL) install it with pip install pillow
|
|
|
|
#*****************************************************************
|
|
#******ALL .pngs in the rawimages folder will be processed********
|
|
#********Exported file may be overwritten in the output***********
|
|
#*****************************************************************
|
|
|
|
from PIL import Image
|
|
from os import listdir, path
|
|
|
|
# Selection of Input/Output directories
|
|
rawimgdir = str(input("Directory of RAW images: "))
|
|
if path.exists(rawimgdir) == False:
|
|
print("Directory could not be found!")
|
|
exit(1)
|
|
|
|
imgdir = str(input("Directory for output image (leave blank to use RAW image directory): "))
|
|
if imgdir == "":
|
|
imgdir = rawimgdir
|
|
elif path.exists(imgdir) == False:
|
|
print("Directory could not be found!")
|
|
exit(1)
|
|
|
|
exportfilename = str(input("Filename for Full Image (E.g LV624_Complete): "))
|
|
if exportfilename == "":
|
|
print("Filename is invalid!")
|
|
exit(1)
|
|
|
|
gamearguments = str(input("Values provided after using the Mass-Screenshot verb: "))
|
|
gamearguments = gamearguments.split()
|
|
if len(gamearguments) != 4:
|
|
print("Invalid arguments!")
|
|
exit(1)
|
|
pixelsize = int(gamearguments[3])
|
|
halfchunksize = int(gamearguments[2])
|
|
width = (int(gamearguments[0]) + halfchunksize - 2)
|
|
height = (int(gamearguments[1]) + halfchunksize - 2)
|
|
if width < 1 or height < 1 or pixelsize < 1 or halfchunksize < 1 or halfchunksize * 2 >= width or halfchunksize * 2 >= height:
|
|
print("Invalid arguments!")
|
|
exit(1)
|
|
width *= pixelsize
|
|
height *= pixelsize
|
|
halfchunksize *= pixelsize
|
|
|
|
# Function for saving the image (params: name=Filename, export=Image IMG variable)
|
|
def func_exportfullimage(name, export):
|
|
print("SAVING IMAGE")
|
|
file = (imgdir+"\\"+name+".png")
|
|
print(file)
|
|
file = open(file, "wb")
|
|
export.save(file)
|
|
|
|
# Where the magic happens, Creates a canvas and pastes RAWimages in a grid fashion
|
|
masterexport = Image.new("RGBA", (width,height), color=(0,0,0,255))
|
|
imagelist = [file for file in sorted(listdir(rawimgdir), key=lambda x: path.getmtime(rawimgdir+"\\"+x)) if file.endswith('.png')]
|
|
imagecount = len(imagelist)
|
|
chunksize = halfchunksize * 2 - pixelsize
|
|
x = fc = 0
|
|
y = height - chunksize
|
|
|
|
# For loop stitches RAw images together
|
|
for p in imagelist:
|
|
file = (rawimgdir+"\\"+p)
|
|
photo = Image.open(file).convert("RGBA")
|
|
#Verbose mode [Iteration] [Image Coords] [RAW Filename]
|
|
#print("iter: " f"{fc : >2}", "IMG XY: " f"{x : >4}", f"{y : >4}", "FILE: " f"{p : >13}")
|
|
masterexport.paste(photo, (x, y))
|
|
x += chunksize
|
|
fc += 1
|
|
if x >= width:
|
|
x = 0
|
|
y -= chunksize
|
|
y = max(y, 0)
|
|
progress = fc / imagecount * 100
|
|
print("%.1f" % progress, "%")
|
|
x = min(x, width - chunksize)
|
|
|
|
func_exportfullimage(exportfilename, masterexport)
|
|
|
|
# Hopefully you got this far
|
|
print("COMPLETED :)")
|