x_scale.py

#!/usr/bin/python

# SPDX-FileCopyrightText: 2021 Edward Ly <contact@edward.ly>
#
# SPDX-License-Identifier: MIT

"""
File: x_scale.py
Last Updated: 24 March 2021
Python script to (lazily) convert old-scale ratings to X-scale ratings for all
simfiles found in the current directory.

Usage: x_scale.py [-d|--dir <directory>]
"""

import os
import sys, getopt
import math, random
import time

x_scale = [0, 1.5, 3, 4.5, 6, 7.5, 9, 10.5, 12, 13.5, 15.5, 17, 18, 19, 20]

def x_scale_ini(filedata):
    newdata = filedata
    for i, line in enumerate(newdata):
        if newdata[i].find("#METERTYPE:DDR;") != -1:
            newdata[i] = newdata[i].replace("#METERTYPE:DDR;", "#METERTYPE:DDR X;")
            break
        if newdata[i].find("#METERTYPE:ITG;") != -1:
            newdata[i] = newdata[i].replace("#METERTYPE:ITG;", "#METERTYPE:DDR X;")
            break
    return newdata

def classic_to_x(old_meter):
    new_meter = x_scale[old_meter]
    prob = new_meter - math.floor(new_meter)
    if prob > 0:
        if random.random() < prob:
            new_meter = math.floor(new_meter)
        else:
            new_meter = math.ceil(new_meter)
    return str(new_meter)

def x_scale_sm(filedata):
    prefix = "#NOTES:" # ratings are 4 lines down from every "#NOTES:"
    newdata = filedata
    for i, line in enumerate(newdata):
        start = newdata[i - 4].find(prefix)
        if start != -1:
            start += 5
            end = line.find(":", start)
            meter = line[start:end]
            new_meter = classic_to_x(int(meter))
            newdata[i] = newdata[i].replace(meter, new_meter)
    return newdata

def x_scale_ssc(filedata):
    prefix = "#METER:"
    newdata = filedata
    for i, line in enumerate(newdata):
        start = line.find(prefix)
        if start != -1:
            start += 7
            end = line.find(";", start)
            meter = line[start:end]
            new_meter = classic_to_x(int(meter))
            newdata[i] = newdata[i].replace(meter, new_meter)
    return newdata

def modify_file(path, fun):
    with open(path, "r", encoding = "utf8", newline = "") as f:
        filedata = f.readlines()
    newdata = fun(filedata)
    with open(path, "w", encoding = "utf8", newline = "") as f:
        f.writelines(newdata)

def print_help():
    print("x_scale.py [-d|--dir <directory>]")
    print("")
    print("Applies a (semi-random) old-scale to X-scale rating conversion to all .sm and .ssc files found in the specified directory.")
    print("Also changes the #METERTYPE value to \"DDR X\" for all group.ini files found.")
    print("Defaults to the current directory if not provided.")

def print_success(time):
    print(f"Done in {time:0.3f} seconds.")
    print("")
    print("Note: the new ratings may not be completely accurate, so please make any additional adjustments yourself!")

def main(argv):
    # Parse command line arguments
    directory = "."
    try:
        opts, args = getopt.getopt(argv, "hd:", ["help", "dir="])
    except getopt.GetoptError:
        print_help()
        sys.exit(2)
    for opt, arg in opts:
        if opt in ("-h", "--help"):
            print_help()
            sys.exit()
        elif opt in ("-d", "--dir"):
            directory = arg
            if not os.path.isdir(directory):
                print("Error: directory does not exist.")
                sys.exit(3)

    tic = time.perf_counter()

    # Find all .sm/.ssc/.ini files and modify their ratings
    for root, dirs, files in os.walk(directory):
        for file in files:
            if file.endswith("group.ini"):
                modify_file(os.path.join(root, file), x_scale_ini)
            if file.endswith(".sm"):
                modify_file(os.path.join(root, file), x_scale_sm)
            if file.endswith(".ssc"):
                modify_file(os.path.join(root, file), x_scale_ssc)

    toc = time.perf_counter()
    print_success(toc - tic)

if __name__ == "__main__":
    main(sys.argv[1:])