GravityNotes/scoresummaryloader.h at master · realryo1/GravityNotes · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
#pragma once

#include <Windows.h>
#include <algorithm>
#include <fstream>
#include <string>
#include <vector>
#include "framework/nlohmann/json.hpp"
#include "debug_ostream.h"
#include "scoreloader.h"

struct ScoreSummary
{
	std::string jsonname;
	std::string musicname;
	std::string musicauthor;
	std::string scoreauthor;
	float difficulty = 0.0f;
	float bpm = 0.0f;
	std::string thumbnail;
	std::string music;
	int vinylIndex = -1;
};

inline ScoreSummary LoadSingleScoreSummary(const std::string& jsonName, const std::string& directoryPath = "asset\\score")
{
	ScoreSummary summary;

	if (jsonName.empty())
	{
		return summary;
	}

	const std::string jsonPath = (jsonName.find('\\') != std::string::npos || jsonName.find('/') != std::string::npos) ? jsonName : (directoryPath + "\\" + jsonName);
	std::ifstream file(jsonPath);
	if (!file.is_open())
	{
		return summary;
	}

	try
	{

		nlohmann::json jsonData;
		file >> jsonData;

		summary.jsonname = jsonName;
		hal::dout << summary.jsonname << std::endl;
		summary.musicname = jsonData.value("musicname", "");
		hal::dout << summary.musicname << std::endl;
		summary.musicauthor = jsonData.value("musicauthor", "");
		summary.scoreauthor = jsonData.value("scoreauthor", "");
		summary.difficulty = jsonData.value("difficulty", 0.0f);
		summary.bpm = jsonData.value("bpm", 0.0f);
		summary.thumbnail = jsonData.value("thumbnail", "");
		summary.music = jsonData.value("music", "");
	}
	catch (...)
	{
		return ScoreSummary{};
	}

	return summary;
}

inline std::vector<ScoreSummary> LoadScoreSummaries(const std::vector<std::string>& directories = { "asset\\score", "json" })
{
	std::vector<ScoreSummary> summaries;

	for (const auto& directoryPath : directories)
	{
		const std::string pattern = directoryPath + "\\*.json";
		WIN32_FIND_DATAA findData = {};
		HANDLE findHandle = FindFirstFileA(pattern.c_str(), &findData);

		if (findHandle == INVALID_HANDLE_VALUE)
		{
			continue;
		}
		do
		{
			if ((findData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) != 0)
			{
				continue;
			}

			ScoreSummary summary = LoadSingleScoreSummary(findData.cFileName, directoryPath);
			if (!summary.jsonname.empty())
			{
				summary.jsonname = directoryPath + "\\" + findData.cFileName;
				summaries.push_back(summary);
			}

		} while (FindNextFileA(findHandle, &findData) != 0);

		FindClose(findHandle);
	}

	std::sort(summaries.begin(), summaries.end(), [](const ScoreSummary& a, const ScoreSummary& b)
		{
			if (a.difficulty == b.difficulty)
			{
				return a.jsonname < b.jsonname;
			}
			return a.difficulty < b.difficulty;
		});

	for (size_t i = 0; i < summaries.size(); ++i)
	{
		summaries[i].vinylIndex = static_cast<int>(i);
	}

	return summaries;
}