32 lines
988 B
Python
32 lines
988 B
Python
|
import os
|
||
|
|
import sys
|
||
|
|
|
||
|
|
from .osu_parser.beatmap import Beatmap
|
||
|
|
from .osu.ctb.difficulty import Difficulty
|
||
|
|
from .ppCalc import calculate_pp
|
||
|
|
|
||
|
|
if len(sys.argv) <= 1:
|
||
|
|
beatmap = Beatmap(os.path.dirname(os.path.realpath(__file__)) + "/test.osu") # Yes... this be my test file (Will remove when project is done)
|
||
|
|
else:
|
||
|
|
beatmap = Beatmap(sys.argv[1])
|
||
|
|
|
||
|
|
if len(sys.argv) >= 3:
|
||
|
|
mods = int(sys.argv[2])
|
||
|
|
else:
|
||
|
|
mods = 0
|
||
|
|
|
||
|
|
difficulty = Difficulty(beatmap, mods)
|
||
|
|
print("Calculation:")
|
||
|
|
print("Stars: {}, PP: {}, MaxCombo: {}\n".format(
|
||
|
|
difficulty.star_rating, calculate_pp(difficulty, 1, beatmap.max_combo, 0), beatmap.max_combo
|
||
|
|
))
|
||
|
|
|
||
|
|
"""
|
||
|
|
m = {"NOMOD": 0, "EASY": 2, "HIDDEN": 8, "HARDROCK": 16, "DOUBLETIME": 64, "HALFTIME": 256, "FLASHLIGHT": 1024}
|
||
|
|
for key in m.keys():
|
||
|
|
difficulty = Difficulty(beatmap, m[key])
|
||
|
|
print("Mods: {}".format(key))
|
||
|
|
print("Stars: {}".format(difficulty.star_rating))
|
||
|
|
print("PP: {}\n".format(calculate_pp(difficulty, 1, beatmap.max_combo, 0)))
|
||
|
|
"""
|