1
0
mirror of synced 2024-11-13 17:30:56 +01:00
TakoTako/TJAConvert/TJAMetaData.cs
2023-01-05 19:56:08 -07:00

461 lines
16 KiB
C#

using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Text;
using System.Text.RegularExpressions;
using SimpleHelpers;
namespace TJAConvert;
internal class TJAMetadata
{
public string Id;
public string Title;
public string TitleJA;
public string TitleEN;
public string TitleCN;
public string TitleTW;
public string TitleKO;
public string Detail;
// dunno what to put in here
#pragma warning disable CS0649
public string DetailJA;
public string DetailEN;
public string DetailCN;
public string DetailTW;
public string DetailKO;
#pragma warning restore CS0649
public string Subtitle;
public string SubtitleJA;
public string SubtitleEN;
public string SubtitleCN;
public string SubtitleTW;
public string SubtitleKO;
public string AudioPath;
public float Offset;
public float PreviewTime;
public SongGenre Genre;
public List<Course> Courses = new();
public TJAMetadata(string tjaPath)
{
Id = Path.GetFileNameWithoutExtension(tjaPath);
var encoding = FileEncoding.DetectFileEncoding(tjaPath);
var lines = File.ReadLines(tjaPath, encoding).ToList();
Title = FindAndGetField("TITLE");
TitleJA = FindAndGetField("TITLEJA");
TitleEN = FindAndGetField("TITLEEN");
TitleCN = FindAndGetField("TITLECN");
TitleTW = FindAndGetField("TITLETW");
TitleKO = FindAndGetField("TITLEKO");
Detail = FindAndGetField("MAKER");
ModifySubtitle("SUBTITLE", x => Subtitle = x);
ModifySubtitle("SUBTITLEJA", x => SubtitleJA = x);
ModifySubtitle("SUBTITLEEN", x => SubtitleEN = x);
ModifySubtitle("SUBTITLECN", x => SubtitleCN = x);
ModifySubtitle("SUBTITLETW", x => SubtitleTW = x);
ModifySubtitle("SUBTITLEKO", x => SubtitleKO = x);
void ModifySubtitle(string key, Action<string> setSubtitle)
{
var entry = FindAndGetField(key);
if (string.IsNullOrEmpty(entry))
return;
var subtitle = FindAndGetField(key).TrimStart('-', '+');
setSubtitle(subtitle);
}
AudioPath = FindAndGetField("WAVE");
Offset = float.Parse(FindAndGetField("OFFSET"));
PreviewTime = float.Parse(FindAndGetField("DEMOSTART"));
var genreEntry = FindAndGetField("GENRE");
Genre = GetGenre(genreEntry);
// start finding courses
Course currentCourse = new Course();
// find the last metadata entry
int courseStartIndex = lines.FindLastIndex(x =>
{
var match = TJAKeyValueRegex.Match(x);
if (!match.Success)
return false;
var type = match.Groups["KEY"].Value;
return MainMetadataKeys.Contains(type.ToUpperInvariant());
});
// find where the track starts
int courseEndIndex = lines.FindIndex(courseStartIndex, x => x.StartsWith("#START", StringComparison.InvariantCultureIgnoreCase));
do
{
bool newContent = false;
for (int i = courseStartIndex + 1; i < courseEndIndex; i++)
{
var line = lines[i];
var match = TJAKeyValueRegex.Match(line);
if (!match.Success)
continue;
newContent = true;
var key = match.Groups["KEY"].Value.ToUpperInvariant().Trim();
var value = match.Groups["VALUE"].Value.Trim();
switch (key)
{
case "COURSE":
{
currentCourse.CourseType = GetCourseType(value);
break;
}
case "LEVEL":
{
currentCourse.Level = int.Parse(value);
break;
}
case "STYLE":
{
currentCourse.PlayStyle = (PlayStyle) Enum.Parse(typeof(PlayStyle), value, true);
break;
}
}
if (key.Equals(nameof(Course.OtherMetadata.Balloon).ToUpperInvariant())) currentCourse.Metadata.Balloon = value;
if (key.Equals(nameof(Course.OtherMetadata.ScoreInit).ToUpperInvariant())) currentCourse.Metadata.ScoreInit = value;
if (key.Equals(nameof(Course.OtherMetadata.ScoreDiff).ToUpperInvariant())) currentCourse.Metadata.ScoreDiff = value;
if (key.Equals(nameof(Course.OtherMetadata.BalloonNor).ToUpperInvariant())) currentCourse.Metadata.BalloonNor = value;
if (key.Equals(nameof(Course.OtherMetadata.BalloonExp).ToUpperInvariant())) currentCourse.Metadata.BalloonExp = value;
if (key.Equals(nameof(Course.OtherMetadata.BalloonMas).ToUpperInvariant())) currentCourse.Metadata.BalloonMas = value;
if (key.Equals(nameof(Course.OtherMetadata.Exam1).ToUpperInvariant())) currentCourse.Metadata.Exam1 = value;
if (key.Equals(nameof(Course.OtherMetadata.Exam2).ToUpperInvariant())) currentCourse.Metadata.Exam2 = value;
if (key.Equals(nameof(Course.OtherMetadata.Exam3).ToUpperInvariant())) currentCourse.Metadata.Exam3 = value;
if (key.Equals(nameof(Course.OtherMetadata.GaugeNcr).ToUpperInvariant())) currentCourse.Metadata.GaugeNcr = value;
if (key.Equals(nameof(Course.OtherMetadata.Total).ToUpperInvariant())) currentCourse.Metadata.Total = value;
if (key.Equals(nameof(Course.OtherMetadata.HiddenBranch).ToUpperInvariant())) currentCourse.Metadata.HiddenBranch = value;
}
currentCourse.CourseDataIndexStart = courseStartIndex + 1;
currentCourse.CourseDataIndexEnd = courseEndIndex;
currentCourse.SongDataIndexStart = courseEndIndex;
// find the next end
if (currentCourse.PlayStyle == PlayStyle.Double)
{
// go through p1 and p2
var index = lines.FindIndex(courseEndIndex, x => x.StartsWith("#END", StringComparison.InvariantCultureIgnoreCase));
courseStartIndex = lines.FindIndex(index + 1, x => x.StartsWith("#END", StringComparison.InvariantCultureIgnoreCase));
}
else
{
courseStartIndex = lines.FindIndex(courseEndIndex, x => x.StartsWith("#END", StringComparison.InvariantCultureIgnoreCase));
}
currentCourse.SongDataIndexEnd = courseStartIndex;
// is this branching?
for (int i = currentCourse.SongDataIndexStart; i < currentCourse.SongDataIndexEnd; i++)
{
var line = lines[i];
if (line.Contains("#BRANCH"))
{
currentCourse.IsBranching = true;
break;
}
}
// calculate roughly the amount of song notes in this course
int noteCount = 0;
int branchNoteCount = 0;
int branches = 0;
bool inBranch = false;
for (int i = currentCourse.SongDataIndexStart; i < currentCourse.SongDataIndexEnd; i++)
{
var line = lines[i].Trim();
if (line.Equals("#N", StringComparison.InvariantCultureIgnoreCase)
|| line.Equals("#E", StringComparison.InvariantCultureIgnoreCase)
|| line.Equals("#M", StringComparison.InvariantCultureIgnoreCase))
branches++;
var branchStart = line.StartsWith("#BRANCHSTART", StringComparison.InvariantCultureIgnoreCase);
if (inBranch && (branchStart || line.StartsWith("#BRANCHEND", StringComparison.InvariantCultureIgnoreCase)))
{
noteCount += branchNoteCount / Math.Max(1, branches);
inBranch = false;
}
if (!inBranch && branchStart)
{
inBranch = true;
branchNoteCount = 0;
branches = 0;
}
if (!line.EndsWith(","))
continue;
var notes = line.Count(x => x is '1' or '2' or '3' or '4');
if (inBranch)
branchNoteCount += notes;
else
noteCount += notes;
}
if (currentCourse.PlayStyle == PlayStyle.Double)
currentCourse.EstimatedNotes = noteCount / 2;
else
currentCourse.EstimatedNotes = noteCount;
if (newContent)
Courses.Add(currentCourse);
// duplicate the existing course
currentCourse = new Course(currentCourse);
// find the next start
courseEndIndex = lines.FindIndex(courseStartIndex, x => x.StartsWith("#START", StringComparison.InvariantCultureIgnoreCase));
} while (courseEndIndex > 0);
string FindAndGetField(string fieldName)
{
var tileRegex = new Regex(string.Format(TJAFieldRegexTemplate, fieldName), RegexOptions.IgnoreCase);
var index = lines.FindIndex(x => tileRegex.IsMatch(x));
if (index < 0)
return null;
return tileRegex.Match(lines[index]).Groups["VALUE"].Value;
}
SongGenre GetGenre(string value)
{
if (string.IsNullOrWhiteSpace(value))
return SongGenre.Variety;
switch (value.ToUpperInvariant())
{
case "アニメ":
return SongGenre.Anime;
case "J-POP":
return SongGenre.Pop;
case "どうよう":
return SongGenre.Children;
case "バラエティ":
return SongGenre.Variety;
case "ボーカロイド":
case "VOCALOID":
return SongGenre.Vocaloid;
case "クラシック":
return SongGenre.Classic;
case "ゲームミュージック":
return SongGenre.Game;
case "ナムコオリジナル":
return SongGenre.Namco;
}
return SongGenre.Variety;
}
CourseType GetCourseType(string value)
{
switch (value.ToUpperInvariant())
{
case "EASY":
case "0":
return CourseType.Easy;
case "NORMAL":
case "1":
return CourseType.Normal;
case "HARD":
case "2":
return CourseType.Hard;
case "ONI":
case "3":
return CourseType.Oni;
case "Edit":
case "4":
return CourseType.UraOni;
}
return CourseType.UraOni;
}
}
public const string TJAFieldRegexTemplate = "^{0}:\\s*(?<VALUE>.*?)\\s*$";
public static Regex TJAKeyValueRegex = new("^(?<KEY>.*?):\\s*(?<VALUE>.*?)\\s*$", RegexOptions.IgnoreCase);
public static HashSet<string> MainMetadataKeys = new()
{
"TITLE",
"TITLEEN",
"SUBTITLE",
"SUBTITLEEN",
"BPM",
"WAVE",
"OFFSET",
"DEMOSTART",
"GENRE",
"SCOREMODE",
"MAKER",
"LYRICS",
"SONGVOL",
"SEVOL",
"SIDE",
"LIFE",
"GAME",
"HEADSCROLL",
"BGIMAGE",
"BGMOVIE",
"MOVIEOFFSET",
"TAIKOWEBSKIN",
};
public class Course
{
public CourseType CourseType = CourseType.Oni;
public int Level = 5;
public PlayStyle PlayStyle = PlayStyle.Single;
public bool IsBranching = false;
public OtherMetadata Metadata = new();
public int CourseDataIndexStart;
public int CourseDataIndexEnd;
public int SongDataIndexStart;
public int SongDataIndexEnd;
public int EstimatedNotes = 0;
public Course()
{
}
public Course(Course course)
{
CourseType = course.CourseType;
Level = course.Level;
// PlayStyle = course.PlayStyle;
Metadata = new OtherMetadata(course.Metadata);
}
public class OtherMetadata
{
public string Balloon;
public string ScoreInit;
public string ScoreDiff;
public string BalloonNor;
public string BalloonExp;
public string BalloonMas;
public string Exam1;
public string Exam2;
public string Exam3;
public string GaugeNcr;
public string Total;
public string HiddenBranch;
public OtherMetadata()
{
}
public OtherMetadata(OtherMetadata metadata)
{
Balloon = metadata.Balloon;
ScoreInit = metadata.ScoreInit;
ScoreDiff = metadata.ScoreDiff;
BalloonNor = metadata.BalloonNor;
BalloonExp = metadata.BalloonExp;
BalloonMas = metadata.BalloonMas;
Exam1 = metadata.Exam1;
Exam2 = metadata.Exam2;
Exam3 = metadata.Exam3;
GaugeNcr = metadata.GaugeNcr;
Total = metadata.Total;
HiddenBranch = metadata.HiddenBranch;
}
}
public List<string> MetadataToTJA(PlayStyle? playStyleOverride = null, CourseType? courseTypeOverride = null)
{
List<string> result = new List<string>();
result.Add($"COURSE:{(courseTypeOverride ?? CourseType).ToString()}");
result.Add($"LEVEL:{Level.ToString()}");
result.Add($"STYLE:{(playStyleOverride ?? PlayStyle).ToString()}");
AddIfNotNull(nameof(OtherMetadata.Balloon), Metadata.Balloon);
AddIfNotNull(nameof(OtherMetadata.ScoreInit), Metadata.ScoreInit);
AddIfNotNull(nameof(OtherMetadata.ScoreDiff), Metadata.ScoreDiff);
AddIfNotNull(nameof(OtherMetadata.BalloonNor), Metadata.BalloonNor);
AddIfNotNull(nameof(OtherMetadata.BalloonExp), Metadata.BalloonExp);
AddIfNotNull(nameof(OtherMetadata.BalloonMas), Metadata.BalloonMas);
AddIfNotNull(nameof(OtherMetadata.Exam1), Metadata.Exam1);
AddIfNotNull(nameof(OtherMetadata.Exam2), Metadata.Exam2);
AddIfNotNull(nameof(OtherMetadata.Exam3), Metadata.Exam3);
AddIfNotNull(nameof(OtherMetadata.GaugeNcr), Metadata.GaugeNcr);
AddIfNotNull(nameof(OtherMetadata.Total), Metadata.Total);
AddIfNotNull(nameof(OtherMetadata.HiddenBranch), Metadata.HiddenBranch);
void AddIfNotNull(string name, string value)
{
if (!string.IsNullOrWhiteSpace(value))
result.Add($"{name.ToUpperInvariant()}:{value}");
}
return result;
}
}
public enum PlayStyle
{
None = 0,
Single = 1,
Double = 2,
}
public enum SongGenre
{
Pop,
Anime,
Vocaloid,
Variety,
Children,
Classic,
Game,
Namco,
}
}
public enum CourseType
{
Easy,
Normal,
Hard,
Oni,
UraOni
}
public static class CourseTypeExtensions
{
public static string ToShort(this CourseType courseType)
{
return courseType switch
{
CourseType.Easy => "e",
CourseType.Normal => "n",
CourseType.Hard => "h",
CourseType.Oni => "m",
CourseType.UraOni => "x",
_ => throw new ArgumentOutOfRangeException(nameof(courseType), courseType, null)
};
}
}