import os

from os import remove

from PIL import Image

 

diskPath = "D:/_FACTORIO/circuit-connector-module/"

inputFolder = "03_Ae/Ae-OUTPUT/Shiftings/"

outputFolder = "04_Python-spritesheeterInput/Shiftings/"

entityName = "_ENTITY-shiftings-ACCUMULATOR"

 

def filterStuff(inputImage, frameNumber):

  shiftingImage = diskPath + inputFolder + inputImage+ "/" + inputImage + frameNumber + ".png"

  i = Image.open(shiftingImage)

 

  newImageRed = Image.new("RGBA", (i.width,i.height), color=(0,0,0,0))

  newImageBlue = Image.new("RGBA", (i.width,i.height), color=(0,0,0,0))

 

  for x in range(0, i.width):

    for y in range(0, i.height):

      pix = i.getpixel((x,y))

      if pix[0] == 255:

        newImageRed.putpixel((x,y),pix)

      if pix[2] == 255:

        newImageBlue.putpixel((x,y),pix)

  os.makedirs(diskPath + outputFolder + inputImage + frameNumber + "/" , exist_ok = True)

  newImageRed.save(diskPath + outputFolder + inputImage + frameNumber + "/" + inputImage + frameNumber + "_boxShifting.png")

  newImageBlue.save(diskPath + outputFolder + inputImage + frameNumber + "/" + inputImage + frameNumber + "_shadowShifting.png")

 

def run():

  filterStuff("_ENTITY-shiftings-ACCUMULATOR", "_000")

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

import traceback

 

try:

 

  run()

 

except Exception as e:

 

  traceback.print_exc()