utils.py

import string
from typing import List, Tuple
ARGSACCEPTED=[
            #接受直接输入的英文
            #角色部分
            "Boy","ColeGuitar","ColeSynth","Farmer","Girl","HoodieBoy","Miner","MrsStevenson","MrStevenson","NicoleCoffee","None","Owl","Parrot","Politician","Samurai",
            #节拍部分
            "Oneshot","Classic",
]

TRANSLATION={
            #方便用中文替换，减少输入英文数量
            #角色部分的翻译(仅翻译了部分常用角色)
            "洛根":"Boy",  
            "吉他科尔":"ColeGuitar", 
            "合成器科尔":"ColeSynth", 
            "农夫":"Farmer", 
            "海莉":"Girl", 
            "科尔":"HoodieBoy", 
            "矿工":"Miner", 
            "史蒂文森太太":"MrsStevenson", 
            "史蒂文森夫人":"MrStevenson", 
            "咖啡妮可":"NicoleCoffee",
            "无":"None", 
            "猫头鹰":"Owl", 
            "鹦鹉":"Parrot", 
            "政客":"Politician", 
            "武士":"Samurai",
            #二拍和七拍的翻译
            "二拍":"Oneshot",
            "七拍":"Classic",
            #Metadata
            "艺术家":"artist",
            "歌曲":"song",
            "作者":"author",
            "描述":"description",
            "癫痫":"seizureWarning",
            "难度":"difficulty",
            "标签":"tags",
            "rank所需错误数":"rankMaxMistakes",
            "rank描述":"rankDescription",
            "简单":"Easy",
            "普通":"Medium",
            "困难":"Tough",
            "噩梦":"VeryTough",
            "开":True,
            "关":False,
            "声音":"Song",
            "动作":"Actions",
            "精灵":"Sprites",
            "房间":"Rooms",
            #预设特效
            "心跳剪影":"SilhouettesOnHBeat",
            "晕影":"Vignette",
            "闪烁晕影":"VignetteFlicker",
            "彩色冲击波":"ColourfulShockwaves",
            "按键重低音":"BassDropOnHit",
            "心跳震屏":"ShakeOnHeartBeat",
            "按键震屏":"ShakeOnHit",
            "轨道浮动":"WavyRows",
            "垂直亮纹":"LightStripVert",
            "过场模式":"CutsceneMode",
            "黑屏":"Blackout",
            "噪点":"Noise",
            "干扰":"GlitchObstruction",
            "落雨":"Rain",
            "矩阵":"Matrix",
            "纸屑":"Confetti",
            "落花":"FallingPetals",
            "落花-即刻":"FallingPetalsInstant",
            "飘雪":"FallingPetalsSnow",
            "雪花":"Snow",
            "高光":"Bloom",
            "橙色高光":"OrangeBloom",
            "蓝色高光":"BlueBloom",
            "镜厅":"HallOfMirrors",
            "自定义屏幕块":"TileN",
            "灰度":"BlackAndWhite",
            "怀旧":"Sepia",
            "自定义滚屏":"CustomScreenScroll",
            "JPEG失真":"JPEG",
            "脉冲计数":"NumbersAbovePulses",
            "马赛克":"Mosaic", 
            "海底波浪":"ScreenWaves",
            "放克":"Funk",
            "电影噪点":"Grain",
            "暴风雪":"Blizzard",
            "素描":"Drawing",
            "色像差":"Aberration",
            "模糊":"Blur",
            "径向模糊":"RadialBlur",
            "色调偏移":"HueShift",
            #闪烁时长
            "短":"Short",
            "中":"Medium",
            "长":"Long",
            
            "颜色":"Color",
            "图片":"Image",
            "填满":"ScaleToFill",
            "拉伸":"AspectFit",
            "居中":"Center",
            "平铺":"Tiled",
            
            "全部":-1,
            "无边框":"None",
            "轮廓":"Outline",
            "发光":"Glow",
        }

def eng(str:str):
    for i in str:
        if i not in string.ascii_lowercase+string.ascii_uppercase:
            return False
        return True

#针对于指令的中英对换，以及检查英文是否输入正确
def replaceString(string:str):
    #中英文检查
    if eng(string):
        try:
            ARGSACCEPTED.index(string)
            return string
        except ValueError:
            raise Exception("您提供的参数\""+string+"\"出错,如果您坚信这是一个错误,请联系硫酸铜进行处理.")
    elif TRANSLATION[string]:
        return TRANSLATION[string]
    else:
        raise Exception("您提供的中文\""+string+"\"并不在程序库中,如果您坚信这是一个错误,请联系硫酸铜进行处理.")

#针对于部分参数的中英兑换
def replaceStringIfNecessary(string:str):
    try:
        return TRANSLATION[string]
    except KeyError:
        return string

class CpbChange:
    def __init__(self, bar: int, beat: int, cpb: int):
        self.bar = bar
        self.beat = beat
        self.cpb = cpb

def getCpbChanges(events: List[Tuple[str, int, int, int]]) -> List[CpbChange]:
    cpbChanges = {0: 8}
    for event in events:
        if event[0] == "SetCrotchetsPerBar":
            cpbChanges[event[1] - 1] = event[3] or 8
    result = []
    bar = 0
    beat = 0
    cpb = 0
    for newBar, newCpb in sorted(cpbChanges.items()):
        if newCpb == cpb:
            continue
        beat += (newBar - bar) * cpb
        bar = newBar
        cpb = newCpb
        result.append(CpbChange(bar, beat, cpb))
    return result

class BarAndBeat:
    def __init__(self, bar: int, beat: int):
        self.bar = bar
        self.beat = beat

def beatToBar(cpbChanges: List[CpbChange], beat: int) -> BarAndBeat:
    lastCpbChange = cpbChanges[0]
    for change in cpbChanges:
        if change.beat > beat:
            break
        lastCpbChange = change
    beatFromLastCpbChange = beat - lastCpbChange.beat
    beatFromStartOfBar = beatFromLastCpbChange % lastCpbChange.cpb
    bar = lastCpbChange.bar + (beatFromLastCpbChange - beatFromStartOfBar) // lastCpbChange.cpb
    return BarAndBeat(bar, beatFromStartOfBar)

class TempoChange:
    def __init__(self, beat: int, time: int, beatLength: int):
        self.beat = beat
        self.time = time
        self.beatLength = beatLength

def getTempoChanges(cpbChanges: List[CpbChange], events: List[Tuple[str, int, int, int]]) -> List[TempoChange]:
    firstSong = next((event for event in events if event[0] == "PlaySong"), None)
    bpmChanges = {0: firstSong[3] if firstSong else 100}
    for event in events:
        if event[0] == "SetBeatsPerMinute":
            beatAndCpb = beatToBar(cpbChanges, event[1] - 1)
            beat = beatAndCpb.beat + (event[2] - 1)
            bpmChanges[beat] = event[3] or 100
    result = []
    beat = 0
    time = 0
    beatLength = 0
    for newBeat, newBpm in sorted(bpmChanges.items()):
        # 根据需要的格式生成 TempoChange 对象并添加到结果列表中
        pass
    return result