videowriter.h

#ifndef LIBCVD_VIDEOWRITER_H
#define LIBCVD_VIDEOWRITER_H

#include "image.h"
#include "rgba.h"
#include "videoffmpeg.h"

#include <libavutil/rational.h>

#include <cmath>
#include <cstdint>
#include <memory>
#include <string>
#include <utility>

namespace CVD
{
namespace internal
{
	class Scaler;
}

// Simple video writer to write RGBA frames to a video file.
class VideoWriter
{
	public:
	VideoWriter(const std::string& filename,
	    const std::string& codec_name,
	    const std::string& pixel_format_name,
	    AVRational timebase,
	    AVRational frame_rate,
	    int rotation_degrees,
	    int threads,
	    int bitrate = -1,
	    std::int64_t duration = -1);
	~VideoWriter();

	std::int64_t seconds_to_timebase(double seconds) const
	{
		return static_cast<std::int64_t>(std::round((seconds * m_timebase.den) / m_timebase.num));
	}
	void put_frame(const BasicImage<Rgba<std::uint8_t>>& image, std::int64_t pts);

	private:
	std::string m_filename;
	AVRational m_timebase;
	int m_threads;
	const AVCodec* m_codec;
	int m_bitrate;
	std::unique_ptr<AVFormatContext, internal::AVFormatContextCloser> m_format_context;
	std::unique_ptr<AVCodecContext, internal::AVCodecContextCloser> m_codec_context;
	std::unique_ptr<internal::Scaler> m_scaler;
};

}

#endif