You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
100 lines
2.8 KiB
100 lines
2.8 KiB
#include "utils/srt_parser.hpp" |
|
|
|
#include <algorithm> |
|
#include <cstdio> |
|
#include <istream> |
|
#include <sstream> |
|
#include <string> |
|
|
|
#include "engine/assets.hpp" |
|
#include "utils/language.h" |
|
#include "utils/log.hpp" |
|
|
|
namespace devilution { |
|
|
|
uint64_t ParseSrtTimestamp(std::string_view timestamp) |
|
{ |
|
unsigned long long h = 0, m = 0, s = 0, ms = 0; |
|
if (sscanf(timestamp.data(), "%llu:%llu:%llu,%llu", &h, &m, &s, &ms) == 4 |
|
|| sscanf(timestamp.data(), "%llu:%llu:%llu.%llu", &h, &m, &s, &ms) == 4) { |
|
return static_cast<uint64_t>(h * 3600000ULL + m * 60000ULL + s * 1000ULL + ms); |
|
} |
|
return 0; |
|
} |
|
|
|
std::vector<SubtitleEntry> LoadSrtFile(std::string_view subtitlePath) |
|
{ |
|
std::vector<SubtitleEntry> subtitles; |
|
|
|
std::string pathStr(subtitlePath); |
|
auto assetData = LoadAsset(pathStr); |
|
if (!assetData.has_value()) { |
|
LogError("Subtitle file not found: {} ({})", subtitlePath, assetData.error()); |
|
return subtitles; |
|
} |
|
|
|
std::string content(assetData->data.get(), assetData->size); |
|
std::istringstream stream(content); |
|
std::string line, text; |
|
uint64_t startTime = 0, endTime = 0; |
|
|
|
while (std::getline(stream, line)) { |
|
// Remove \r if present |
|
if (!line.empty() && line.back() == '\r') |
|
line.pop_back(); |
|
|
|
// Skip empty lines (end of subtitle block) |
|
if (line.empty()) { |
|
if (!text.empty() && startTime < endTime) { |
|
// Remove trailing newline from text |
|
if (!text.empty() && text.back() == '\n') |
|
text.pop_back(); |
|
subtitles.push_back({ startTime, endTime, text }); |
|
text.clear(); |
|
} |
|
continue; |
|
} |
|
|
|
// Check if line is a number (subtitle index) - skip it |
|
if (std::all_of(line.begin(), line.end(), [](char c) { return std::isdigit(static_cast<unsigned char>(c)); })) |
|
continue; |
|
|
|
// Check if line contains --> (timestamp line) |
|
const size_t arrowPos = line.find("-->"); |
|
if (arrowPos != std::string::npos) { |
|
const std::string startStr = line.substr(0, arrowPos); |
|
const std::string endStr = line.substr(arrowPos + 3); |
|
startTime = ParseSrtTimestamp(startStr); |
|
endTime = ParseSrtTimestamp(endStr); |
|
continue; |
|
} |
|
|
|
// Otherwise it's subtitle text |
|
if (!text.empty()) |
|
text += "\n"; |
|
text += line; |
|
} |
|
|
|
// Handle last subtitle if file doesn't end with blank line |
|
if (!text.empty() && startTime < endTime) { |
|
if (!text.empty() && text.back() == '\n') |
|
text.pop_back(); |
|
subtitles.push_back({ startTime, endTime, text }); |
|
} |
|
|
|
return subtitles; |
|
} |
|
|
|
std::string GetSubtitleAtTime(const std::vector<SubtitleEntry> &subtitles, uint64_t videoTimeMs) |
|
{ |
|
for (const auto &entry : subtitles) { |
|
if (videoTimeMs >= entry.startTimeMs && videoTimeMs < entry.endTimeMs) { |
|
// Translate the subtitle text |
|
std::string_view translated = LanguageTranslate(entry.text); |
|
return std::string(translated); |
|
} |
|
} |
|
return ""; |
|
} |
|
|
|
} // namespace devilution
|
|
|