import sys
import os
import cv2
from wand import image


while True:
    image_directory = input("Image Directory: ")
    if image_directory == "":
        break
    
    image_format = input("Image Format: ")
    if image_format != "jpg" and image_format != "dds":
        break
    
    output_directory = input("Output Directory: ")
    if output_directory == "":
        break
    

    for file in os.listdir(image_directory):
        
        if file[-4:] != ".png" and file[-4:] != ".tga":
            continue
            
        fname = file[:-4]
        
        if image_format == "jpg":
            src = cv2.imread(image_directory + file, cv2.IMREAD_COLOR)
            cv2.imwrite(output_directory + fname + ".jpg", src, [int(cv2.IMWRITE_JPEG_QUALITY), 90]) 
            
            src = cv2.imread(image_directory + file, cv2.IMREAD_UNCHANGED)
            alpha_channel = src[:,:,3]
            cv2.imwrite(output_directory + fname + "_alpha.jpg", alpha_channel, [int(cv2.IMWRITE_JPEG_QUALITY), 100])
        elif image_format == "dds":
            with image.Image(filename=image_directory + file) as img:
                img.compression = 'dxt3'
                img.save(filename=(output_directory + fname + '.dds'))
        print("Saving file: " + fname)
    
    exit(0)