Work in progress code for combining up to 4 images

combineImg.py

@@ -0,0 +1,86 @@
+from PIL import Image, ImageOps, ImageFilter
+
+# find the highest res image in an array of images
+def findImageWithMostPixels(imageArray):
+    maxPixels = 0
+    maxImage = None
+    for image in imageArray:
+        pixels = image.size[0] * image.size[1]
+        if pixels > maxPixels:
+            maxPixels = pixels
+            maxImage = image
+    return maxImage
+
+def getTotalImgSize(imageArray): # take the image with the most pixels, multiply it by the number of images, and return the width and height
+    maxImage = findImageWithMostPixels(imageArray)
+    if (len(imageArray) == 1):
+        return (maxImage.size[0], maxImage.size[1])
+    elif (len(imageArray) == 2):
+        return (maxImage.size[0] * 2, maxImage.size[1])
+    elif (len(imageArray) == 3):
+        return (maxImage.size[0] * 2, maxImage.size[1]*2)
+    elif (len(imageArray) == 4):
+        return (maxImage.size[0] * 2, maxImage.size[1]*2)
+    else:
+        return (maxImage.size[0] * 2, maxImage.size[1]*2)
+
+def scaleAllImagesToSameSize(imageArray,targetWidth,targetHeight,pad=True): # scale all images in the array to the same size, preserving aspect ratio
+    newImageArray = []
+    for image in imageArray:
+        if pad:
+            image = image.convert('RGBA')
+            newImg = ImageOps.pad(image, (targetWidth, targetHeight),color=(0, 0, 0, 0))
+        else:
+            newImg = ImageOps.fit(image, (targetWidth, targetHeight)) # scale + crop
+        newImageArray.append(newImg)
+    return newImageArray
+
+def blurImage(image, radius):
+    return image.filter(ImageFilter.GaussianBlur(radius=radius))
+
+def combineImages(imageArray, totalWidth, totalHeight,pad=True):
+    x = 0
+    y = 0
+    if (len(imageArray) == 1): # if there is only one image, just return it
+        return imageArray[0]
+    # image generation is needed
+    topImg = findImageWithMostPixels(imageArray)
+    newImage = Image.new("RGBA", (totalWidth, totalHeight),(0, 0, 0, 0))
+    imageArray = scaleAllImagesToSameSize(imageArray,topImg.size[0],topImg.size[1],pad)
+    if (len(imageArray) == 2): # if there are two images, combine them horizontally
+        for image in imageArray:
+            newImage.paste(image, (x, y))
+            x += image.size[0]
+    elif (len(imageArray) == 3): # the elusive 3 image upload
+        # if there are three images, combine the first two horizontally, then combine the last one vertically
+        imageArray[2] = scaleAllImagesToSameSize([imageArray[2]],totalWidth,topImg.size[1],pad)[0] # take the last image, treat it like an image array and scale it to the total width, but same height as all individual images
+        for image in imageArray[0:2]:
+            newImage.paste(image, (x, y))
+            x += image.size[0]
+        y += imageArray[0].size[1]
+        x = 0
+        newImage.paste(imageArray[2], (x, y))
+    elif (len(imageArray) == 4): # if there are four images, combine the first two horizontally, then combine the last two vertically
+        for image in imageArray[0:2]:
+            newImage.paste(image, (x, y))
+            x += image.size[0]
+        y += imageArray[0].size[1]
+        x = 0
+        for image in imageArray[2:4]:
+            newImage.paste(image, (x, y))
+            x += image.size[0]
+    else:
+        for image in imageArray:
+            newImage.paste(image, (x, y))
+            x += image.size[0]
+    return newImage
+
+def saveImage(image, name):
+    image.save(name)
+
+# combine up to four images into a single image
+def genImage(imageArray):
+    combined = combineImages(imageArray, *getTotalImgSize(imageArray))
+    combinedBG = combineImages(imageArray, *getTotalImgSize(imageArray),False)
+    combinedBG = blurImage(combinedBG,50)
+    return Image.alpha_composite(combinedBG,combined)