main.py

import datetime
import os
import json
import shutil

def main():
    """
    Main function
    """
    
    image_magick_test = str(input("Please Check if You have ImageMagick installed (y for Yes, n for No) :"))
    if (image_magick_test.lower() == "y"):
        print("ImageMagick is installed")
    else:
        print('Please install ImageMagick from https://imagemagick.org/index.php')
        print('Exiting...')
        return
    maxFrame = 0;

    # fps_of_gifs = int(input("Enter the fps of the gifs: "))

    # Get the path of the current working directory
    path = os.getcwd()
    print("The current working directory is: " + path)

    #get time
    print("Started : "+str(datetime.datetime.now()))

    #get the names of layers
    path = "./layers/"
    
    orderedLayers = json.load(open("./config/layers.json",'r'))
    layers = orderedLayers['layersOrder']

    try:
            
        shutil.rmtree("./layers")
    except:
        pass
    try:
        os.mkdir('build')
    except:
        pass
    try:
        os.mkdir('build/gifs')
    except:
        pass
    os.mkdir('layers')
    for layer in layers:
        path = 'layers/' + layer["name"] 
        try:
            os.mkdir(path)
        except OSError:
            print ("Creation of the directory %s failed" % path)
        else:
            print ("Successfully created the directory %s " % path)


    #get max frame

    for x in range(0,len(layers)):
        layer = layers[x]

        child_files = os.listdir("./unprocessedLayers/"+layer["name"])
        for y in range(0,len(child_files)):

            if (child_files[y][len(child_files[y])-3:len(child_files[y])+1]) == "gif":

                from_dir = os.path.join("unprocessedLayers",layer['name'],child_files[y])

                #get frame count
                cmd_str = 'magick identify "'+ from_dir +'"'
                

                # try:
                #     k = os.popen(cmd_str).read()
                #     frame = int(k.split(']')[-2].split('[')[1]) + 1
                # except:
                k = os.popen(cmd_str[7:]).read()
                frame = int(k.split(']')[-2].split('[')[1]) + 1
                    
                print("Frame count: " + str(frame))
                if (frame > maxFrame):
                    maxFrame = frame
        with open('config/format.json', 'r') as f:
            data = json.load(f)
        with open('internalconfig/format.json', 'w') as f:
            data["width"] = maxFrame*data["width"]
            f.write(json.dumps(data))

    #convert single images to spritesheets of fps_of_gifs length

    #get os path of the images
    for x in range(0,len(layers)):
        layer = layers[x]
        print("Processing layer: " + layer["name"])
        
        #get files in the path

        child_files = os.listdir("./unprocessedLayers/"+layer["name"])
        for y in range(0,len(child_files)):
            if ( not child_files[y].endswith('gif')):

                print('./unprocessedLayers/'+layer['name'] + "/" + child_files[y])

                #converting to sprites
                
                from_dir = os.path.join("unprocessedLayers",layer['name'],child_files[y])
                to_dir = os.path.join("work",child_files[y])

                

                # try:
                #     os.system('copy '+from_dir+' '+to_dir)
                # except:
                os.system('cp '+ '"' +from_dir+'" "'+to_dir+'"')
                
                os.chdir('work')

                cmd_str = "magick convert "


                for z in range(maxFrame):
                    cmd_str+=( "'"+ str(child_files[y]) + "' ")
                cmd_str+=(" +append " + "'final_" + child_files[y] + "'")
                

                # try:
                #     os.system(cmd_str)
                # except:
                print(cmd_str[7:])
                os.system(cmd_str[7:])

                
                
                os.chdir("..")
                from_dir = os.path.join("work","final_"+child_files[y])
                to_dir = os.path.join("layers",layer['name'],child_files[y])

                

                # try:
                #     # print("\n\n\n")
                #     # print("Moving from: " + from_dir)
                #     # print("Moving to: " + to_dir)
                        
                #     # print("\n\n\n")
                #     os.system('move '+from_dir+' '+to_dir)
                # except:
                os.system('mv '+ '"' +from_dir+'" "'+to_dir+'"')
                shutil.rmtree('work')
                os.system('mkdir work')

            elif (child_files[y].endswith('gif')):



                #convert gif to spritesheet
                """magick montage your_gif.gif -tile x1 -geometry +0+0 -alpha On -background "rgba(0, 0, 0, 0.0)" -quality 100 sprites.png"""
                from_dir = os.path.join("unprocessedLayers",layer['name'],child_files[y])

                to_dir = os.path.join("layers",layer["name"],child_files[y][:len(child_files[y])-4].split('#')[0] + ' Animated#'+child_files[y][:len(child_files[y])-4].split('#')[1] +'.' + "png")
                # print("\n\n\n")
                # print("Moving from: " + from_dir)
                # print("Moving to: " + to_dir)
                    
                # print("\n\n\n")
                # try:
                #     os.system('magick montage -coalesce  "'+ from_dir + '" -tile x1 -geometry +0+0 -alpha On -background "rgba(0, 0, 0, 0.0)" -quality 100 "' + to_dir + '"')
                # except:
                #     print("Failed")
                #     pass

                #clear work folder
                
                shutil.rmtree('work')
                os.system('mkdir work')

                #convert to images
                os.system("convert -coalesce '"+ from_dir + "' work/frame_%d.png")

                current_frame_count = 0
                
                k = os.popen("identify '"+ from_dir +"'").read()
                current_frame_count = int(k.split(']')[-2].split('[')[1]) + 1
                
                cmd_str = "convert "
                for x in range(0,maxFrame):
                    cmd_str+= "'work/frame_" + str(x%current_frame_count) + ".png' "
                cmd_str+="+append '" + to_dir + "'"
                # try:
                #     os.system(cmd_str)
                # except:
                print(cmd_str)
                os.system(cmd_str)
                print("Done!")
        
        print("done")
if __name__ == "__main__":
    main()