#ifndef WAVWRITER_HH
#define WAVWRITER_HH

#include "File.hh"
#include "Mixer.hh"

#include <cassert>
#include <cstdint>
#include <span>
#include <string>

namespace openmsx {

/** Base class for writing WAV files.
  */
class WavWriter
{
public:
	/** Returns false if there has been data written to the wav image.
	 */
	[[nodiscard]] bool isEmpty() const { return bytes == 0; }

	/** Returns the number of bytes (not samples) written so far.
	 */
	[[nodiscard]] uint32_t getBytes() const { return bytes; }

	/** Flush data to file and update header. Try to make (possibly)
	  * incomplete file already usable for external programs.
	  */
	void flush();

protected:
	WavWriter(const std::string& filename,
	          unsigned channels, unsigned bits, unsigned frequency);
	~WavWriter();

protected:
	File file;
	uint32_t bytes = 0;
};

/** Writes 8-bit WAV files.
  */
class Wav8Writer : public WavWriter
{
public:
	Wav8Writer(const std::string& filename, unsigned channels, unsigned frequency)
		: WavWriter(filename, channels, 8, frequency) {}

	void write(std::span<const uint8_t> buffer);
};

/** Writes 16-bit WAV files.
  */
class Wav16Writer : public WavWriter
{
public:
	Wav16Writer(const std::string& filename, unsigned channels, unsigned frequency)
		: WavWriter(filename, channels, 16, frequency) {}

	void write(std::span<const int16_t> buffer);
	void write(std::span<const float> buffer, float amp = 1.0f);
	void write(std::span<const StereoFloat> buffer, float ampLeft = 1.0f, float ampRight = 1.0f);

	void writeSilence(uint32_t samples);
};

} // namespace openmsx

#endif