135 lines
4.3 KiB
Python
135 lines
4.3 KiB
Python
from __future__ import annotations
|
|
|
|
import argparse
|
|
import json
|
|
import os
|
|
import shutil
|
|
from dataclasses import asdict
|
|
|
|
from drsxml2json import get_songdata_from_track_id
|
|
from drsxml2json import TRACK_ID_TO_PATH
|
|
from model.dancedash import DDBeatMap
|
|
from model.dancedash import DDBeatMapData
|
|
from model.dancedash import DDBeatMapInfoFile
|
|
from model.dancedash import X_Y
|
|
from model.dancerush import DRSSongData
|
|
from util import get_m4a_and_duration
|
|
from util import get_song_cover_path
|
|
from util import ORDER_COUNT_PER_BEAT
|
|
|
|
|
|
def create_dd_tracks_from_DRSSongData(
|
|
drs_song_data: DRSSongData, target_dir: str,
|
|
) -> bool:
|
|
if not os.path.exists(target_dir):
|
|
os.makedirs(target_dir)
|
|
print(f'Created directory: {target_dir}')
|
|
|
|
dd_bmp = int(drs_song_data.info.bpm_max / 100)
|
|
|
|
song_paths = []
|
|
for attr, difficulty in drs_song_data.difficulties.with_attrs_as_str.items():
|
|
target_difficulty_path = os.path.join(
|
|
target_dir, f'{drs_song_data.song_id}_{attr}.json',
|
|
)
|
|
song_paths.append(target_difficulty_path)
|
|
drs_beat_map = DDBeatMap(
|
|
data=DDBeatMapData(
|
|
name=f'{drs_song_data.info.title_name} {attr}',
|
|
intervalPerSecond=0.0,
|
|
gridSize=X_Y(x=0, y=0),
|
|
planeSize=X_Y(x=0, y=0),
|
|
orderCountPerBeat=ORDER_COUNT_PER_BEAT,
|
|
sphereNodes=[], # TODO: DO THE MAPPING LOL.
|
|
lineNodes=[], # TODO: DO THE MAPPING LOL.
|
|
effectNodes=[],
|
|
roadBlockNodes=[],
|
|
trapNodes=[],
|
|
),
|
|
beatSubs=1,
|
|
BPM=dd_bmp,
|
|
songStartOffset=0.0,
|
|
NPS='0.0',
|
|
developerMode=False,
|
|
noteSpeed=1.0,
|
|
noteJumpOffset=0.0,
|
|
interval=1.0,
|
|
info='',
|
|
)
|
|
with open(target_difficulty_path, 'w') as f:
|
|
json.dump(asdict(drs_beat_map), f, indent=4)
|
|
print(f'Created file: {target_difficulty_path}')
|
|
|
|
folder_path = TRACK_ID_TO_PATH.get(drs_song_data.song_id)
|
|
song_path, song_length = get_m4a_and_duration(folder_path)
|
|
if song_path and song_length:
|
|
shutil.copy(song_path, target_dir)
|
|
|
|
if song_cover_path := get_song_cover_path(folder_path):
|
|
shutil.copy(song_cover_path, target_dir)
|
|
|
|
drs_song_info_json = DDBeatMapInfoFile(
|
|
EditorVersion='1.3.2',
|
|
BeatMapId=drs_song_data.song_id,
|
|
OstId=0,
|
|
CreateTicks=0,
|
|
CreateTime='',
|
|
SongName=drs_song_data.info.title_name,
|
|
SongLength=song_length or '-1',
|
|
SongAuthorName=drs_song_data.info.artist_name,
|
|
LevelAuthorName='https://github.com/thomasasfk/drs2dd',
|
|
SongPreviewSection=0,
|
|
Bpm=str(dd_bmp),
|
|
SongPath=os.path.basename(song_path) if song_path else None,
|
|
OstName=None,
|
|
CoverPath=os.path.basename(
|
|
song_cover_path,
|
|
) if song_cover_path else None,
|
|
# TODO PROPERLY MAP DIFFICULTIES BASED ON DIF NUMBERS
|
|
# ALSO REMOVE 2 PLAYER DIFFICULTIES
|
|
DRS_Easy=os.path.basename(song_paths[0]) if len(
|
|
song_paths,
|
|
) > 0 else None,
|
|
DRS_Normal=os.path.basename(song_paths[1]) if len(
|
|
song_paths,
|
|
) > 1 else None,
|
|
DRS_Hard=os.path.basename(song_paths[2]) if len(
|
|
song_paths,
|
|
) > 2 else None,
|
|
DRS_Expert=os.path.basename(song_paths[3]) if len(
|
|
song_paths,
|
|
) > 3 else None,
|
|
)
|
|
|
|
info_file_path = os.path.join(
|
|
target_dir, f'{drs_song_data.song_id}_info.json',
|
|
)
|
|
with open(info_file_path, 'w') as f:
|
|
json.dump(asdict(drs_song_info_json), f, indent=4)
|
|
print(f'Created file: {info_file_path}')
|
|
|
|
return True
|
|
|
|
|
|
if __name__ == '__main__':
|
|
parser = argparse.ArgumentParser(
|
|
description='Create DD tracks from DRS Song Data',
|
|
)
|
|
parser.add_argument(
|
|
'--song-id', type=int,
|
|
help='ID of the song to process (number of folder)', required=True,
|
|
)
|
|
|
|
args = parser.parse_args()
|
|
|
|
song_data = get_songdata_from_track_id(args.song_id)
|
|
if not song_data:
|
|
print(f'No song data found for song id {args.song_id}')
|
|
raise SystemExit(1)
|
|
|
|
created = create_dd_tracks_from_DRSSongData(
|
|
song_data, song_data.info.title_name,
|
|
)
|
|
|
|
print(f'Song {args.song_id} created?: {created}')
|