WAVReader: Make ReadFrames() return optional

That way the caller can detect EOF.
This commit is contained in:
Stenzek
2025-12-22 20:45:54 +10:00
parent 132ca44f05
commit c0277db4cf
3 changed files with 23 additions and 7 deletions

View File

@@ -524,8 +524,11 @@ bool WaveTrackFileInterface::Read(void* buffer, u64 offset, u32 size, Error* err
const u32 num_frames_to_read = std::min(num_frames, m_reader.GetNumFrames() - frame_number);
if (num_frames_to_read > 0)
{
if (!m_reader.SeekToFrame(frame_number, error) || !m_reader.ReadFrames(buffer, num_frames_to_read, error))
if (!m_reader.SeekToFrame(frame_number, error) ||
m_reader.ReadFrames(buffer, num_frames_to_read, error).value_or(0) != num_frames_to_read)
{
return false;
}
}
// Padding.

View File

@@ -64,6 +64,7 @@ WAVReader::WAVReader(WAVReader&& move)
{
m_file = std::exchange(move.m_file, nullptr);
m_frames_start = std::exchange(move.m_frames_start, 0);
m_bytes_per_frame = std::exchange(move.m_bytes_per_frame, 0);
m_sample_rate = std::exchange(move.m_sample_rate, 0);
m_num_channels = std::exchange(move.m_num_channels, 0);
m_num_frames = std::exchange(move.m_num_frames, 0);
@@ -79,6 +80,7 @@ WAVReader& WAVReader::operator=(WAVReader&& move)
{
m_file = std::exchange(move.m_file, nullptr);
m_frames_start = std::exchange(move.m_frames_start, 0);
m_bytes_per_frame = std::exchange(move.m_bytes_per_frame, 0);
m_sample_rate = std::exchange(move.m_sample_rate, 0);
m_num_channels = std::exchange(move.m_num_channels, 0);
m_num_frames = std::exchange(move.m_num_frames, 0);
@@ -182,6 +184,7 @@ bool WAVReader::Open(const char* path, Error* error /*= nullptr*/)
m_file = fp.release();
m_frames_start = FileSystem::FTell64(m_file);
m_sample_rate = format.sample_rate;
m_bytes_per_frame = sizeof(s16) * format.num_channels;
m_num_channels = format.num_channels;
m_num_frames = num_frames;
return true;
@@ -194,7 +197,9 @@ void WAVReader::Close()
std::fclose(m_file);
m_file = nullptr;
m_frames_start = 0;
m_sample_rate = 0;
m_bytes_per_frame = 0;
m_num_channels = 0;
m_num_frames = 0;
}
@@ -204,6 +209,7 @@ std::FILE* WAVReader::TakeFile()
std::FILE* ret = std::exchange(m_file, nullptr);
m_sample_rate = 0;
m_frames_start = 0;
m_bytes_per_frame = 0;
m_num_channels = 0;
m_num_frames = 0;
return ret;
@@ -220,15 +226,19 @@ bool WAVReader::SeekToFrame(u32 num, Error* error)
return FileSystem::FSeek64(m_file, offset, SEEK_SET, error);
}
bool WAVReader::ReadFrames(void* samples, u32 num_frames, Error* error /*= nullptr*/)
std::optional<u32> WAVReader::ReadFrames(void* samples, u32 num_frames, Error* error /*= nullptr*/)
{
if (std::fread(samples, sizeof(s16) * m_num_channels, num_frames, m_file) != num_frames)
const size_t read = std::fread(samples, m_bytes_per_frame, num_frames, m_file);
if (read == 0)
{
Error::SetErrno(error, "fread() failed: ", errno);
return false;
if (std::ferror(m_file))
{
Error::SetErrno(error, "fread() failed: ", errno);
return std::nullopt;
}
}
return true;
return static_cast<u32>(read);
}
WAVWriter::WAVWriter() = default;

View File

@@ -6,6 +6,7 @@
#include "common/types.h"
#include <cstdio>
#include <optional>
class Error;
@@ -23,6 +24,7 @@ public:
ALWAYS_INLINE u32 GetSampleRate() const { return m_sample_rate; }
ALWAYS_INLINE u32 GetNumChannels() const { return m_num_channels; }
ALWAYS_INLINE u32 GetNumFrames() const { return m_num_frames; }
ALWAYS_INLINE u32 GetBytesPerFrame() const { return m_bytes_per_frame; }
ALWAYS_INLINE u64 GetFramesStartOffset() const { return m_frames_start; }
ALWAYS_INLINE bool IsOpen() const { return (m_file != nullptr); }
@@ -34,13 +36,14 @@ public:
bool SeekToFrame(u32 num, Error* error = nullptr);
bool ReadFrames(void* samples, u32 num_frames, Error* error = nullptr);
std::optional<u32> ReadFrames(void* samples, u32 num_frames, Error* error = nullptr);
private:
using SampleType = s16;
std::FILE* m_file = nullptr;
s64 m_frames_start = 0;
u32 m_bytes_per_frame = 0;
u32 m_sample_rate = 0;
u32 m_num_channels = 0;
u32 m_num_frames = 0;