#Python script that replaces the timing values in a Fumen and spits out a new file.
import configparser, struct, sys

#Function to handle converting a float to a binary array
def convert_to_bytearray(f):
    val = hex(struct.unpack('>I', struct.pack('<f', f))[0])
    Fixed_hex = val[2:]
    return bytes.fromhex(Fixed_hex)
    
#Load ini file    
config = configparser.ConfigParser()
config.sections()
config.read('timing.ini')

# Define possible endings of file name and their difficulties.
easy = '_e'
normal = '_n'
hard = '_h'
extreme = '_m'
ura = '_x'

# Check to see if anything (other than the python file) has been entered.
# If two files/the timing window haven't been entered, it'll just throw a normal Python error.
if len(sys.argv) > 1:

    #Check if standard/hitwide/hitnarrow/custom was typed in the console
    if sys.argv[3].lower() == 'standard':
        GOOD = config.getfloat('standard', 'good')
        OK = config.getfloat('standard', 'ok')
        BAD = config.getfloat('standard', 'bad')
        GOOD_EASY = config.getfloat('standard', 'good_easy')
        OK_EASY = config.getfloat('standard', 'ok_easy')
        BAD_EASY = config.getfloat('standard', 'bad_easy')

    else:
        try:
            GOOD = config.getfloat(sys.argv[3].lower(), 'good')
            OK = config.getfloat(sys.argv[3].lower(), 'ok')
            BAD = config.getfloat(sys.argv[3].lower(), 'bad')
            GOOD_EASY = config.getfloat(sys.argv[3].lower(), 'good_easy')
            OK_EASY = config.getfloat(sys.argv[3].lower(), 'ok_easy')
            BAD_EASY = config.getfloat(sys.argv[3].lower(), 'bad_easy')
    
        except:
            print("Invalid Input")
            exit()

    #Convert the floats to a binary array
    GOOD_BYTES = (convert_to_bytearray(GOOD))
    OK_BYTES = (convert_to_bytearray(OK))
    BAD_BYTES = (convert_to_bytearray(BAD))

    GOOD_EASY_BYTES = (convert_to_bytearray(GOOD_EASY))
    OK_EASY_BYTES = (convert_to_bytearray(OK_EASY))
    BAD_EASY_BYTES = (convert_to_bytearray(BAD_EASY))

    #Define binary arrays
    timing_window_hard = (GOOD_BYTES + OK_BYTES + BAD_BYTES) * 36
    timing_window_easy = (GOOD_EASY_BYTES + OK_EASY_BYTES + BAD_EASY_BYTES) * 36    

    # Define the input and output files.
    inFile = sys.argv[1]
    outFile = sys.argv[2]

    # Opens the input file, saves it as the output,
    # This was the only way I managed was able to acomplish the right output.
    with open(inFile, "rb") as old, open(outFile, "wb") as new:
        old.seek(0)
        new.write(old.read())
    
    # Actually replaces the timing windows
    chart_new = open(outFile, "rb+")
    chart_new.seek(0)
    
    # Checks if _e, _n, etc is present in the file name:
    # If it is, it'll use timing_easy,
    # If it isn't present, it will use timing_hard instead.
    if easy in inFile:
        chart_new.write(timing_window_easy)
        chart_new.close()
    elif normal in inFile:
        chart_new.write(timing_window_easy)
        chart_new.close()
    else:
        chart_new.write(timing_window_hard)
        chart_new.close()
    
else:
    print("TaikoFumenTimingReplace\nUsage:",sys.argv[0], "inFile outFile timingWindow",
    "\n\nTiming Options: Standard, Hitnarrow, Hitwide or a User Defined Option.")